dev: solved possible null reference

ProdMoon · Jul 30, 2024 · cdab19e · cdab19e
1 parent 1b3cf8d
commit cdab19e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Controllers/Api/DliibController.cs b/Controllers/Api/DliibController.cs
@@ -32,6 +32,10 @@ public async Task<ActionResult<IEnumerable<DliibModel>>> GetDliibs()
     [Authorize]
     public async Task<ActionResult<IEnumerable<DliibModel>>> GetMyDliibs()
     {
+        if (User.Identity == null)
+        {
+            return Unauthorized();
+        }
         var user = await db.Users.FirstOrDefaultAsync(x => x.NormalizedUserName == User.Identity.Name);
         return await db.Dliibs
             .Where(x => x.Author == user)
@@ -93,6 +97,10 @@ public async Task<IActionResult> PutDliib(int id, Dliib dliib)
     [Authorize]
     public async Task<ActionResult<Dliib>> PostDliib(Dliib dliib)
     {
+        if (User.Identity == null)
+        {
+            return Unauthorized();
+        }
         var user = await db.Users.FirstOrDefaultAsync(x => x.NormalizedUserName == User.Identity.Name);
         dliib.Author = user;
         db.Dliibs.Add(dliib);
@@ -111,6 +119,10 @@ public async Task<IActionResult> DeleteDliib(int id)
             return NotFound();
         }
 
+        if (User.Identity == null)
+        {
+            return Unauthorized();
+        }
         var user = await db.Users.FirstOrDefaultAsync(x => x.NormalizedUserName == User.Identity.Name);
         if (dliib.Author != user)
         {