Add CreatedBy field to identity user who created comment

vectorNull · Jun 26, 2024 · bee2958 · bee2958
1 parent cb27f26
commit bee2958
Show file tree

Hide file tree

Showing 9 changed files with 555 additions and 8 deletions.
diff --git a/api/Controllers/CommentController.cs b/api/Controllers/CommentController.cs
@@ -1,6 +1,9 @@
 using api.DTOs.Comment;
+using api.Extensions;
 using api.Interfaces;
 using api.Mappers;
+using api.Models;
+using Microsoft.AspNetCore.Identity;
 using Microsoft.AspNetCore.Mvc;
 
 namespace api.Controllers
@@ -11,11 +14,13 @@ public class CommentController : ControllerBase
     {
         private readonly ICommentRepo _commentRepo;
         private readonly IStockRepo _stockRepo;
+        private readonly UserManager<AppUser> _userManager;
 
-        public CommentController(ICommentRepo commentRepo, IStockRepo stockRepo)
+        public CommentController(ICommentRepo commentRepo, IStockRepo stockRepo, UserManager<AppUser> userManager)
         {
             _commentRepo = commentRepo;
             _stockRepo = stockRepo;
+            _userManager = userManager;
         }
 
         [HttpGet]
@@ -71,7 +76,21 @@ public async Task<IActionResult> CreateComment([FromRoute] int stockId, [FromBod
                 return BadRequest("Stock does not exist.");
             }
 
-            var commentModel = commentDto.FromCreatedDtoToComment(stockId);
+            var username = User.GetUsername();
+
+            if (username == null)
+            {
+                return Unauthorized();
+            }
+
+            var user = await _userManager.FindByNameAsync(username);
+
+            if (user == null)
+            {
+                return NotFound();
+            }
+
+            var commentModel = commentDto.FromCreatedDtoToComment(stockId, user.Id);
 
             await _commentRepo.CreateAsync(commentModel);
 

diff --git a/api/DTOs/Comment/CommentDto.cs b/api/DTOs/Comment/CommentDto.cs
@@ -6,6 +6,7 @@ public class CommentDto
         public string Title { get; set; } = string.Empty;
         public string Content { get; set; } = string.Empty;
         public DateTime CreatedOn { get; set; }
+        public string? CreatedBy { get; set; } = string.Empty;
         public int? StockId { get; set; }
     }
 }
diff --git a/api/Mappers/CommentMappers.cs b/api/Mappers/CommentMappers.cs
@@ -13,17 +13,19 @@ public static CommentDto ToCommentDto(this Comment commentModel)
                 Title = commentModel.Title,
                 Content = commentModel.Content,
                 CreatedOn = commentModel.CreatedOn,
-                StockId = commentModel.StockId
+                CreatedBy = commentModel.AppUser?.UserName,
+                StockId = commentModel.StockId,
             };
         }
 
-        public static Comment FromCreatedDtoToComment(this CreateCommentRequestDto commentDto, int stockId)
+        public static Comment FromCreatedDtoToComment(this CreateCommentRequestDto commentDto, int stockId, string userId)
         {
             return new Comment
             {
                 Title = commentDto.Title,
                 Content = commentDto.Content,
                 StockId = stockId,
+                AppUserId = userId
             };
         }