From 4538aa0be40316a967d6fbcbc08bc9670b9bbcf0 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sat, 13 Feb 2021 15:46:02 +0800
Subject: feat: Add timeline manageable.

---
 BackEnd/Timeline/Controllers/HighlightTimelineController.cs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'BackEnd/Timeline/Controllers/HighlightTimelineController.cs')

diff --git a/BackEnd/Timeline/Controllers/HighlightTimelineController.cs b/BackEnd/Timeline/Controllers/HighlightTimelineController.cs
index ea012f76..f582e74b 100644
--- a/BackEnd/Timeline/Controllers/HighlightTimelineController.cs
+++ b/BackEnd/Timeline/Controllers/HighlightTimelineController.cs
@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Threading.Tasks;
 using Timeline.Auth;
+using Timeline.Entities;
 using Timeline.Models.Http;
 using Timeline.Models.Mapper;
 using Timeline.Models.Validation;
@@ -28,6 +29,11 @@ namespace Timeline.Controllers
             _timelineMapper = timelineMapper;
         }
 
+        private Task<List<HttpTimeline>> Map(List<TimelineEntity> timelines)
+        {
+            return _timelineMapper.MapToHttp(timelines, Url, this.GetOptionalUserId(), this.UserHasPermission(UserPermission.AllTimelineManagement));
+        }
+
         /// <summary>
         /// Get all highlight timelines.
         /// </summary>
@@ -38,7 +44,7 @@ namespace Timeline.Controllers
         {
             var ids = await _service.GetHighlightTimelines();
             var timelines = await _timelineService.GetTimelineList(ids);
-            return await _timelineMapper.MapToHttp(timelines, Url, this.GetOptionalUserId());
+            return await Map(timelines);
         }
 
         /// <summary>
-- 
cgit v1.2.3