102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
// handlers/avatar.go
|
|
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"api_bb/internal/service"
|
|
"api_bb/pkg/logger"
|
|
"api_bb/pkg/middleware"
|
|
"api_bb/pkg/utils"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type AvatarHandler struct {
|
|
logger logger.LoggerInterface
|
|
avatarService service.AvatarService
|
|
}
|
|
|
|
func NewAvatarHandler(avatarService service.AvatarService) *AvatarHandler {
|
|
return &AvatarHandler{
|
|
logger: logger.NewWrapper(logger.Get().With(zap.String("handler", "avatar"))),
|
|
avatarService: avatarService,
|
|
}
|
|
}
|
|
|
|
func (h *AvatarHandler) Routes() chi.Router {
|
|
r := chi.NewRouter()
|
|
r.Post("/upload", h.UploadAvatar)
|
|
r.Delete("/delete", h.DeleteAvatar)
|
|
return r
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
// Парсим 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()
|
|
|
|
// Проверяем тип файла
|
|
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
|
|
}
|
|
|
|
// Возвращаем ответ с полем 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
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
// Возвращаем ответ с полем success
|
|
utils.RespondWithJSON(w, http.StatusOK, map[string]interface{}{
|
|
"success": true,
|
|
"message": "Avatar deleted successfully",
|
|
})
|
|
}
|