From d45f776470a7cece565a46655f286fcb06cb3a87 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sat, 13 Feb 2021 15:09:21 +0800
Subject: feat: Post info add editable field.

---
 BackEnd/Timeline/Models/Http/HttpTimelinePost.cs | 12 +++++++-
 BackEnd/Timeline/Models/Mapper/TimelineMapper.cs | 37 ++++++++++++++++++++----
 2 files changed, 43 insertions(+), 6 deletions(-)

(limited to 'BackEnd/Timeline/Models')

diff --git a/BackEnd/Timeline/Models/Http/HttpTimelinePost.cs b/BackEnd/Timeline/Models/Http/HttpTimelinePost.cs
index 26e1a92d..5e069821 100644
--- a/BackEnd/Timeline/Models/Http/HttpTimelinePost.cs
+++ b/BackEnd/Timeline/Models/Http/HttpTimelinePost.cs
@@ -11,7 +11,7 @@ namespace Timeline.Models.Http
     {
         public HttpTimelinePost() { }
 
-        public HttpTimelinePost(long id, List<HttpTimelinePostDataDigest> dataList, bool deleted, DateTime time, HttpUser? author, string? color, DateTime lastUpdated)
+        public HttpTimelinePost(long id, List<HttpTimelinePostDataDigest> dataList, bool deleted, DateTime time, HttpUser? author, string? color, DateTime lastUpdated, string timelineName, bool editable)
         {
             Id = id;
             DataList = dataList;
@@ -20,6 +20,8 @@ namespace Timeline.Models.Http
             Author = author;
             Color = color;
             LastUpdated = lastUpdated;
+            TimelineName = timelineName;
+            Editable = editable;
         }
 
         /// <summary>
@@ -52,5 +54,13 @@ namespace Timeline.Models.Http
         /// Last updated time.
         /// </summary>
         public DateTime LastUpdated { get; set; } = default!;
+        /// <summary>
+        /// Timeline name.
+        /// </summary>
+        public string TimelineName { get; set; } = default!;
+        /// <summary>
+        /// True if you can edit this post.
+        /// </summary>
+        public bool Editable { get; set; }
     }
 }
diff --git a/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs b/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
index 5c46fa81..191e2f70 100644
--- a/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
+++ b/BackEnd/Timeline/Models/Mapper/TimelineMapper.cs
@@ -16,13 +16,17 @@ namespace Timeline.Models.Mapper
         private readonly UserMapper _userMapper;
         private readonly IHighlightTimelineService _highlightTimelineService;
         private readonly IBookmarkTimelineService _bookmarkTimelineService;
+        private readonly ITimelineService _timelineService;
+        private readonly ITimelinePostService _timelinePostService;
 
-        public TimelineMapper(DatabaseContext database, UserMapper userMapper, IHighlightTimelineService highlightTimelineService, IBookmarkTimelineService bookmarkTimelineService)
+        public TimelineMapper(DatabaseContext database, UserMapper userMapper, IHighlightTimelineService highlightTimelineService, IBookmarkTimelineService bookmarkTimelineService, ITimelineService timelineService, ITimelinePostService timelinePostService)
         {
             _database = database;
             _userMapper = userMapper;
             _highlightTimelineService = highlightTimelineService;
             _bookmarkTimelineService = bookmarkTimelineService;
+            _timelineService = timelineService;
+            _timelinePostService = timelinePostService;
         }
 
         public async Task<HttpTimeline> MapToHttp(TimelineEntity entity, IUrlHelper urlHelper, long? userId)
@@ -64,7 +68,7 @@ namespace Timeline.Models.Mapper
         }
 
 
-        public async Task<HttpTimelinePost> MapToHttp(TimelinePostEntity entity, string timelineName, IUrlHelper urlHelper)
+        public async Task<HttpTimelinePost> MapToHttp(TimelinePostEntity entity, string timelineName, IUrlHelper urlHelper, long? userId, bool isAdministrator)
         {
             _ = timelineName;
 
@@ -79,6 +83,22 @@ namespace Timeline.Models.Mapper
                 author = await _userMapper.MapToHttp(entity.Author, urlHelper);
             }
 
+            bool editable;
+
+            if (userId is null)
+            {
+                editable = false;
+            }
+            else if (isAdministrator)
+            {
+                editable = true;
+            }
+            else
+            {
+                editable = await _timelinePostService.HasPostModifyPermission(entity.TimelineId, entity.LocalId, userId.Value);
+            }
+
+
             return new HttpTimelinePost(
                     id: entity.LocalId,
                     dataList: dataDigestList,
@@ -86,18 +106,25 @@ namespace Timeline.Models.Mapper
                     author: author,
                     color: entity.Color,
                     deleted: entity.Deleted,
-                    lastUpdated: entity.LastUpdated
+                    lastUpdated: entity.LastUpdated,
+                    timelineName: timelineName,
+                    editable: editable
                 );
         }
 
-        public async Task<List<HttpTimelinePost>> MapToHttp(List<TimelinePostEntity> entities, string timelineName, IUrlHelper urlHelper)
+        public async Task<List<HttpTimelinePost>> MapToHttp(List<TimelinePostEntity> entities, string timelineName, IUrlHelper urlHelper, long? userId, bool isAdministrator)
         {
             var result = new List<HttpTimelinePost>();
             foreach (var entity in entities)
             {
-                result.Add(await MapToHttp(entity, timelineName, urlHelper));
+                result.Add(await MapToHttp(entity, timelineName, urlHelper, userId, isAdministrator));
             }
             return result;
         }
+
+        internal Task MapToHttp(TimelinePostEntity post, string timeline, IUrlHelper url)
+        {
+            throw new System.NotImplementedException();
+        }
     }
 }
-- 
cgit v1.2.3