modified: serv_nginx/api_bb/internal/handlers/avatar.go

add to response ok message
2025-10-13 01:17:58 +05:00
parent e13545c5f1
commit 39df128e1e
1 changed files with 55 additions and 51 deletions
@@ -33,65 +33,69 @@ func (h *AvatarHandler) Routes() chi.Router {
 }

 func (h *AvatarHandler) UploadAvatar(w http.ResponseWriter, r *http.Request) {
-	user, ok := middleware.GetUserFromContext(r.Context())
-	if !ok {
-		utils.RespondWithError(w, http.StatusUnauthorized, "Authentication required")
-		return
-	}
+    user, ok := middleware.GetUserFromContext(r.Context())
+    if !ok {
+        utils.RespondWithError(w, http.StatusUnauthorized, "Authentication required")
+        return
+    }

-	// Парсим multipart форму
-	if err := r.ParseMultipartForm(10 << 20); err != nil { // 10MB limit
-		utils.RespondWithError(w, http.StatusBadRequest, "Failed to parse form: "+err.Error())
-		return
-	}
+    // Парсим multipart форму
+    if err := r.ParseMultipartForm(10 << 20); err != nil { // 10MB limit
+        utils.RespondWithError(w, http.StatusBadRequest, "Failed to parse form: "+err.Error())
+        return
+    }

-	file, header, err := r.FormFile("avatar")
-	if err != nil {
-		utils.RespondWithError(w, http.StatusBadRequest, "Failed to get file: "+err.Error())
-		return
-	}
-	defer file.Close()
+    file, header, err := r.FormFile("avatar")
+    if err != nil {
+        utils.RespondWithError(w, http.StatusBadRequest, "Failed to get file: "+err.Error())
+        return
+    }
+    defer file.Close()

-	// Проверяем тип файла
-	allowedTypes := map[string]bool{
-		"image/jpeg": true,
-		"image/jpg":  true,
-		"image/png":  true,
-		"image/gif":  true,
-	}
-	if !allowedTypes[header.Header.Get("Content-Type")] {
-		utils.RespondWithError(w, http.StatusBadRequest, "Only JPEG, PNG and GIF images are allowed")
-		return
-	}
+    // Проверяем тип файла
+    allowedTypes := map[string]bool{
+        "image/jpeg": true,
+        "image/jpg":  true,
+        "image/png":  true,
+        "image/gif":  true,
+    }
+    if !allowedTypes[header.Header.Get("Content-Type")] {
+        utils.RespondWithError(w, http.StatusBadRequest, "Only JPEG, PNG and GIF images are allowed")
+        return
+    }

-	// Загружаем аватар
-	avatarPath, err := h.avatarService.UploadAvatar(user.ID, file, header)
-	if err != nil {
-		h.logger.Error("Failed to upload avatar", zap.Error(err))
-		utils.RespondWithError(w, http.StatusInternalServerError, "Failed to upload avatar: "+err.Error())
-		return
-	}
+    // Загружаем аватар
+    avatarPath, err := h.avatarService.UploadAvatar(user.ID, file, header)
+    if err != nil {
+        h.logger.Error("Failed to upload avatar", zap.Error(err))
+        utils.RespondWithError(w, http.StatusInternalServerError, "Failed to upload avatar: "+err.Error())
+        return
+    }

-	utils.RespondWithJSON(w, http.StatusOK, map[string]interface{}{
-		"message": "Avatar uploaded successfully",
-		"avatar":  avatarPath,
-	})
+    // Возвращаем ответ с полем success
+    utils.RespondWithJSON(w, http.StatusOK, map[string]interface{}{
+        "success": true,
+        "message": "Avatar uploaded successfully",
+        "avatar":  avatarPath,
+    })
 }

 func (h *AvatarHandler) DeleteAvatar(w http.ResponseWriter, r *http.Request) {
-	user, ok := middleware.GetUserFromContext(r.Context())
-	if !ok {
-		utils.RespondWithError(w, http.StatusUnauthorized, "Authentication required")
-		return
-	}
+    user, ok := middleware.GetUserFromContext(r.Context())
+    if !ok {
+        utils.RespondWithError(w, http.StatusUnauthorized, "Authentication required")
+        return
+    }

-	if err := h.avatarService.DeleteAvatar(user.ID); err != nil {
-		h.logger.Error("Failed to delete avatar", zap.Error(err))
-		utils.RespondWithError(w, http.StatusInternalServerError, "Failed to delete avatar: "+err.Error())
-		return
-	}
+    if err := h.avatarService.DeleteAvatar(user.ID); err != nil {
+        h.logger.Error("Failed to delete avatar", zap.Error(err))
+        utils.RespondWithError(w, http.StatusInternalServerError, "Failed to delete avatar: "+err.Error())
+        return
+    }

-	utils.RespondWithJSON(w, http.StatusOK, map[string]interface{}{
-		"message": "Avatar deleted successfully",
-	})
+    // Возвращаем ответ с полем success
+    utils.RespondWithJSON(w, http.StatusOK, map[string]interface{}{
+        "success": true,
+        "message": "Avatar deleted successfully",
+    })
 }