Nikolai Papin
669349e020
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
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
// 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,
|
|
})
|
|
}
|