Merge pull request #4 from zimotech10/feat/auth

boiler plate completed

Extensions/AuthEndpoints.cs

@@ -54,7 +54,8 @@ private static async Task<IResult> Register(
 
         await userManager.AddToRoleAsync(user, role);
 
-        return Results.Ok("User registered successfully with role " + role);
+        // Return success message with the new user's ID
+        return Results.Ok(new { Message = "User registered successfully with role " + role, UserId = user.Id });
     }
 
     private static async Task<IResult> Login(
@@ -99,7 +100,9 @@ private static string GenerateJwtToken(Users user, IConfiguration configuration)
         return new JwtSecurityTokenHandler().WriteToken(token);
     }
 
-    private static async Task<IResult> UploadAvatar(IFormFile avatarFile)
+    private static async Task<IResult> UploadAvatar(IFormFile avatarFile, UserManager<Users> userManager,
+    ApplicationDbContext dbContext,
+    string userId) // Pass userId to identify the user
     {
         if (avatarFile == null || avatarFile.Length == 0)
         {
@@ -131,6 +134,27 @@ private static async Task<IResult> UploadAvatar(IFormFile avatarFile)
             await avatarFile.CopyToAsync(stream);
         }
 
+        // Update user's avatar path in the database
+        var user = await userManager.FindByIdAsync(userId);
+        if (user == null)
+        {
+            return Results.NotFound("User not found.");
+        }
+
+        user.PhotoUrl = $"/uploads/{fileName}";
+
+        var result = await userManager.UpdateAsync(user);
+        if (!result.Succeeded)   //If operation is failed delete uploaded file
+        {
+            // Handle update failure (rollback file creation, etc.)
+            File.Delete(filePath);
+            return Results.BadRequest(result.Errors);
+        }
+
+        // Save changes to the database
+        await dbContext.SaveChangesAsync();
+
+
         return Results.Ok(new { FilePath = $"/uploads/{fileName}" });
     }
 

Migrations/20240829041317_AddPhotoUrlToUsers.cs

@@ -0,0 +1,28 @@
+using Microsoft.EntityFrameworkCore.Migrations;
+
+#nullable disable
+
+namespace freelancerAPI.Migrations
+{
+    /// <inheritdoc />
+    public partial class AddPhotoUrlToUsers : Migration
+    {
+        /// <inheritdoc />
+        protected override void Up(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.AddColumn<string>(
+                name: "PhotoUrl",
+                table: "AspNetUsers",
+                type: "nvarchar(max)",
+                nullable: true);
+        }
+
+        /// <inheritdoc />
+        protected override void Down(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.DropColumn(
+                name: "PhotoUrl",
+                table: "AspNetUsers");
+        }
+    }
+}

Migrations/ApplicationDbContextModelSnapshot.cs

@@ -216,6 +216,9 @@ protected override void BuildModel(ModelBuilder modelBuilder)
                     b.Property<bool>("PhoneNumberConfirmed")
                         .HasColumnType("bit");
 
+                    b.Property<string>("PhotoUrl")
+                        .HasColumnType("nvarchar(max)");
+
                     b.Property<string>("SecurityStamp")
                         .HasColumnType("nvarchar(max)");
 

Models/Users.cs

@@ -7,4 +7,5 @@ public class Users : IdentityUser
     public required string Country { get; set; }
     public required string Bio { get; set; }
     public required string BioTitle { get; set; }
+    public string? PhotoUrl { get; set; }
 }