From 2f9da0048893aacb7ffbfcae50ea5e0c8f840dbc Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Tue, 10 Mar 2020 03:10:34 +0800
Subject: ...

---
 Timeline/Controllers/TimelineController.cs | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

(limited to 'Timeline/Controllers')

diff --git a/Timeline/Controllers/TimelineController.cs b/Timeline/Controllers/TimelineController.cs
index 00e9a860..e3baff12 100644
--- a/Timeline/Controllers/TimelineController.cs
+++ b/Timeline/Controllers/TimelineController.cs
@@ -154,9 +154,10 @@ namespace TimelineApp.Controllers
             {
                 return StatusCode(StatusCodes.Status403Forbidden, ErrorResponse.Common.Forbid());
             }
-            await _service.ChangeProperty(name, body);
-            var timeline = (await _service.GetTimeline(name)).FillLinks(Url);
-            return Ok(timeline);
+            await _service.ChangeProperty(name, _mapper.Map<TimelineChangePropertyRequest>(body));
+            var timeline = await _service.GetTimeline(name);
+            var result = _mapper.Map<TimelineInfo>(timeline);
+            return result;
         }
 
         [HttpPut("timelines/{name}/members/{member}")]
@@ -207,8 +208,9 @@ namespace TimelineApp.Controllers
 
             try
             {
-                var timelineInfo = (await _service.CreateTimeline(body.Name, userId)).FillLinks(Url);
-                return Ok(timelineInfo);
+                var timeline = await _service.CreateTimeline(body.Name, userId);
+                var result = _mapper.Map<TimelineInfo>(timeline);
+                return result;
             }
             catch (ConflictException)
             {
@@ -218,7 +220,7 @@ namespace TimelineApp.Controllers
 
         [HttpDelete("timelines/{name}")]
         [Authorize]
-        public async Task<ActionResult<TimelineInfo>> TimelineDelete([FromRoute][TimelineName] string name)
+        public async Task<ActionResult<CommonDeleteResponse>> TimelineDelete([FromRoute][TimelineName] string name)
         {
             if (!this.IsAdministrator() && !(await _service.HasManagePermission(name, this.GetUserId())))
             {
@@ -228,11 +230,11 @@ namespace TimelineApp.Controllers
             try
             {
                 await _service.DeleteTimeline(name);
-                return Ok(CommonDeleteResponse.Delete());
+                return CommonDeleteResponse.Delete();
             }
             catch (TimelineNotExistException)
             {
-                return Ok(CommonDeleteResponse.NotExist());
+                return CommonDeleteResponse.NotExist();
             }
         }
     }
-- 
cgit v1.2.3