From 45873d9115840c9db596c2dffebc7bb29df13686 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 15 Nov 2020 20:48:28 +0800
Subject: feat: Deleting root user now returns 400.

---
 BackEnd/Timeline/Controllers/UserController.cs | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

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

diff --git a/BackEnd/Timeline/Controllers/UserController.cs b/BackEnd/Timeline/Controllers/UserController.cs
index da34cb1b..8edae139 100644
--- a/BackEnd/Timeline/Controllers/UserController.cs
+++ b/BackEnd/Timeline/Controllers/UserController.cs
@@ -138,15 +138,23 @@ namespace Timeline.Controllers
         /// <returns>Info of deletion.</returns>
         [HttpDelete("users/{username}"), PermissionAuthorize(UserPermission.UserManagement)]
         [ProducesResponseType(StatusCodes.Status200OK)]
+        [ProducesResponseType(StatusCodes.Status400BadRequest)]
         [ProducesResponseType(StatusCodes.Status401Unauthorized)]
         [ProducesResponseType(StatusCodes.Status403Forbidden)]
         public async Task<ActionResult<CommonDeleteResponse>> Delete([FromRoute][Username] string username)
         {
-            var delete = await _userDeleteService.DeleteUser(username);
-            if (delete)
-                return Ok(CommonDeleteResponse.Delete());
-            else
-                return Ok(CommonDeleteResponse.NotExist());
+            try
+            {
+                var delete = await _userDeleteService.DeleteUser(username);
+                if (delete)
+                    return Ok(CommonDeleteResponse.Delete());
+                else
+                    return Ok(CommonDeleteResponse.NotExist());
+            }
+            catch (InvalidOperationOnRootUserException)
+            {
+                return BadRequest(ErrorResponse.UserController.Delete_RootUser());
+            }
         }
 
         /// <summary>
-- 
cgit v1.2.3