From 0bb2cc098506963ebf9ee06ec94b43c8d388543c Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 7 Jan 2021 22:10:58 +0800
Subject: feat: Timeline info contains bookmark and highlight flag.

---
 BackEnd/Timeline/Models/Mapper/TimelineMapper.cs | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

(limited to 'BackEnd/Timeline/Models/Mapper')

diff --git a/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs b/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
index 14ca8fe9..95418573 100644
--- a/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
+++ b/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
@@ -15,14 +15,18 @@ namespace Timeline.Models.Mapper
     {
         private readonly DatabaseContext _database;
         private readonly UserMapper _userMapper;
+        private readonly IHighlightTimelineService _highlightTimelineService;
+        private readonly IBookmarkTimelineService _bookmarkTimelineService;
 
-        public TimelineMapper(DatabaseContext database, UserMapper userMapper)
+        public TimelineMapper(DatabaseContext database, UserMapper userMapper, IHighlightTimelineService highlightTimelineService, IBookmarkTimelineService bookmarkTimelineService)
         {
             _database = database;
             _userMapper = userMapper;
+            _highlightTimelineService = highlightTimelineService;
+            _bookmarkTimelineService = bookmarkTimelineService;
         }
 
-        public async Task<HttpTimeline> MapToHttp(TimelineEntity entity, IUrlHelper urlHelper)
+        public async Task<HttpTimeline> MapToHttp(TimelineEntity entity, IUrlHelper urlHelper, long? userId)
         {
             await _database.Entry(entity).Reference(e => e.Owner).LoadAsync();
             await _database.Entry(entity).Collection(e => e.Members).Query().Include(m => m.User).LoadAsync();
@@ -40,6 +44,8 @@ namespace Timeline.Models.Mapper
                 members: await _userMapper.MapToHttp(entity.Members.Select(m => m.User).ToList(), urlHelper),
                 createTime: entity.CreateTime,
                 lastModified: entity.LastModified,
+                isHighlight: await _highlightTimelineService.IsHighlightTimeline(entity.Id),
+                isBookmark: userId is not null && await _bookmarkTimelineService.IsBookmark(userId.Value, entity.Id, false, false),
                 links: new HttpTimelineLinks(
                     self: urlHelper.ActionLink(nameof(TimelineController.TimelineGet), nameof(TimelineController)[0..^nameof(Controller).Length], new { timeline = timelineName }),
                     posts: urlHelper.ActionLink(nameof(TimelineController.PostListGet), nameof(TimelineController)[0..^nameof(Controller).Length], new { timeline = timelineName })
@@ -47,12 +53,12 @@ namespace Timeline.Models.Mapper
             );
         }
 
-        public async Task<List<HttpTimeline>> MapToHttp(List<TimelineEntity> entities, IUrlHelper urlHelper)
+        public async Task<List<HttpTimeline>> MapToHttp(List<TimelineEntity> entities, IUrlHelper urlHelper, long? userId)
         {
             var result = new List<HttpTimeline>();
             foreach (var entity in entities)
             {
-                result.Add(await MapToHttp(entity, urlHelper));
+                result.Add(await MapToHttp(entity, urlHelper, userId));
             }
             return result;
         }
-- 
cgit v1.2.3