From ffe44ba70c9e5c6a01179c7e2f4185543cbc441c Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 7 Jan 2021 20:12:00 +0800
Subject: refactor: Make mapper a service. Fix #202.

---
 BackEnd/Timeline/Controllers/TokenController.cs | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

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

diff --git a/BackEnd/Timeline/Controllers/TokenController.cs b/BackEnd/Timeline/Controllers/TokenController.cs
index e695a10e..b3675aad 100644
--- a/BackEnd/Timeline/Controllers/TokenController.cs
+++ b/BackEnd/Timeline/Controllers/TokenController.cs
@@ -1,4 +1,3 @@
-using AutoMapper;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
@@ -26,14 +25,16 @@ namespace Timeline.Controllers
         private readonly IUserCredentialService _userCredentialService;
         private readonly IUserTokenManager _userTokenManager;
         private readonly ILogger<TokenController> _logger;
+        private readonly UserMapper _userMapper;
         private readonly IClock _clock;
 
         /// <summary></summary>
-        public TokenController(IUserCredentialService userCredentialService, IUserTokenManager userTokenManager, ILogger<TokenController> logger, IClock clock)
+        public TokenController(IUserCredentialService userCredentialService, IUserTokenManager userTokenManager, ILogger<TokenController> logger, UserMapper userMapper, IClock clock)
         {
             _userCredentialService = userCredentialService;
             _userTokenManager = userTokenManager;
             _logger = logger;
+            _userMapper = userMapper;
             _clock = clock;
         }
 
@@ -69,11 +70,11 @@ namespace Timeline.Controllers
                     ("Username", request.Username),
                     ("Expire At", expireTime?.ToString(CultureInfo.CurrentCulture.DateTimeFormat) ?? "default")
                 ));
-                return Ok(new HttpCreateTokenResponse
+                return new HttpCreateTokenResponse
                 {
                     Token = result.Token,
-                    User = result.User.MapToHttp(Url)
-                });
+                    User = await _userMapper.MapToHttp(result.User, Url)
+                };
             }
             catch (UserNotExistException e)
             {
@@ -111,10 +112,10 @@ namespace Timeline.Controllers
                 var result = await _userTokenManager.VerifyToken(request.Token);
                 _logger.LogInformation(Log.Format(LogVerifySuccess,
                     ("Username", result.Username), ("Token", request.Token)));
-                return Ok(new HttpVerifyTokenResponse
+                return new HttpVerifyTokenResponse
                 {
-                    User = result.MapToHttp(Url)
-                });
+                    User = await _userMapper.MapToHttp(result, Url)
+                };
             }
             catch (UserTokenTimeExpireException e)
             {
-- 
cgit v1.2.3