chore: remove direct avatar upload endpoint (POST /profile/avatar);

feat: add endpoints for presigned upload URLs (GET /upload/avatar, GET /upload/image); refactor: replace ProfileDto with NewProfileDto in update profile endpoint; feat: implement S3 integration for avatar management; fix: update database queries to handle new avatar upload flow; chore: add new dependencies for S3 handling (golang.org/x/time); refactor: rename UploadService to S3Service; refactor: change return type for func LocalizeS3Url(originalURL string) (*url.URL, error); feat: add custom validator for upload_id
2025-08-01 04:34:06 +03:00
parent 8dba0f79aa
commit 669349e020
15 changed files with 405 additions and 222 deletions
--- a/backend/internal/controllers/s3.go
+++ b/backend/internal/controllers/s3.go
@@ -0,0 +1,108 @@
+// Copyright (c) 2025 Nikolai Papin
+//
+// This file is part of Easywish
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
+// the GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package controllers
+
+import (
+	"easywish/internal/middleware"
+	"easywish/internal/models"
+	"easywish/internal/services"
+	"easywish/internal/utils/enums"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"go.uber.org/zap"
+	"golang.org/x/time/rate"
+)
+
+type S3Controller struct {
+	log *zap.Logger
+	s3  services.S3Service
+}
+
+func NewS3Controller(_log *zap.Logger, _us services.S3Service) Controller {
+	ctrl := S3Controller{log: _log, s3: _us}
+
+	return &controllerImpl{
+		Path: "/upload",
+		Middleware: []gin.HandlerFunc{
+			middleware.RateLimitMiddleware(rate.Every(2*time.Second), 1),
+		},
+		Methods: []ControllerMethod{
+			{
+				HttpMethod:   GET,
+				Path:         "/avatar",
+				Authorization: enums.UserRole,
+				Middleware:   []gin.HandlerFunc{},
+				Function:     ctrl.getAvatarUploadUrl,
+			},
+			{
+				HttpMethod:   GET,
+				Path:         "/image",
+				Authorization: enums.UserRole,
+				Middleware:   []gin.HandlerFunc{},
+				Function:     ctrl.getImageUploadUrl,
+			},
+		},
+	}
+}
+
+// XXX: untested
+//	@Summary Get presigned URL for avatar upload
+//	@Tags Upload
+//	@Accept		json
+//	@Produce	json
+//	@Security	JWT
+//	@Success	200		{object}	models.PresignedUploadResponse	"Presigned URL and form data"
+//	@Failure	500		"Internal server error"
+//	@Router		/upload/avatar [get]
+func (ctrl *S3Controller) getAvatarUploadUrl(c *gin.Context) {
+	url, formData, err := ctrl.s3.CreateAvatarUrl()
+	if err != nil {
+		ctrl.log.Error("Failed to generate avatar upload URL", zap.Error(err))
+		c.Status(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, models.PresignedUploadResponse{
+		Url:    *url,
+		Fields: *formData,
+	})
+}
+
+// XXX: untested
+//	@Summary Get presigned URL for image upload
+//	@Tags Upload
+//	@Accept		json
+//	@Produce	json
+//	@Security	JWT
+//	@Success	200		{object}	models.PresignedUploadResponse	"Presigned URL and form data"
+//	@Failure	500		"Internal server error"
+//	@Router		/upload/image [get]
+func (ctrl *S3Controller) getImageUploadUrl(c *gin.Context) {
+	url, formData, err := ctrl.s3.CreateImageUrl()
+	if err != nil {
+		c.Status(http.StatusInternalServerError)
+		return
+	}
+
+	c.JSON(http.StatusOK, models.PresignedUploadResponse{
+		Url: *url,
+		Fields: *formData,
+	})
+}