feat: fully implement profile controller;
feat: implement file upload handling in controller with size and type validation; feat: add custom validation rules for bio and color hex fields; refactor: enhance request handling with dedicated client info extraction; chore: update profile DTOs with validation tags; docs: profile controller swagger
This commit is contained in:
@@ -22,10 +22,14 @@ import (
|
||||
"easywish/internal/services"
|
||||
"easywish/internal/utils/enums"
|
||||
"easywish/internal/validation"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/go-playground/validator/v10"
|
||||
"github.com/google/uuid"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
@@ -75,12 +79,45 @@ func (ctrl *controllerImpl) Setup(group *gin.RouterGroup, log *zap.Logger, auth
|
||||
}
|
||||
}),
|
||||
method.Function)...,
|
||||
)}
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
func GetFile(c *gin.Context, name string, maxSize int64, allowedTypes map[string]bool) (*string, error) {
|
||||
file, err := c.FormFile(name); if err != nil {
|
||||
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("File '%s': not provided", name)})
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if file.Size > int64(maxSize) {
|
||||
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("File '%s': file too large", name)})
|
||||
return nil, fmt.Errorf("File too large")
|
||||
}
|
||||
|
||||
fileType := file.Header.Get("Content-Type")
|
||||
if len(allowedTypes) > 0 && !allowedTypes[fileType] {
|
||||
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("File '%s': forbidden file type: %s", name, fileType)})
|
||||
return nil, fmt.Errorf("Wrong file type")
|
||||
}
|
||||
|
||||
folderPath := "/tmp/uploads"
|
||||
if _, err := os.Stat(folderPath); os.IsNotExist(err) {
|
||||
os.MkdirAll(folderPath, 0700)
|
||||
}
|
||||
|
||||
filePath := fmt.Sprintf("%s/%s-%s", folderPath, uuid.New().String(), filepath.Base(file.Filename))
|
||||
if err := c.SaveUploadedFile(file, filePath); err != nil {
|
||||
c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": "Error saving file"})
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &filePath, nil
|
||||
}
|
||||
|
||||
func GetRequest[ModelT any](c *gin.Context) (*dto.Request[ModelT], error) {
|
||||
|
||||
var body ModelT
|
||||
|
||||
if err := c.ShouldBindJSON(&body); err != nil {
|
||||
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return nil, err
|
||||
@@ -97,6 +134,16 @@ func GetRequest[ModelT any](c *gin.Context) (*dto.Request[ModelT], error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
cinfo := GetClientInfo(c)
|
||||
|
||||
return &dto.Request[ModelT]{
|
||||
Body: body,
|
||||
User: cinfo,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func GetClientInfo(c * gin.Context) (dto.ClientInfo) {
|
||||
|
||||
cinfoFromCtx, ok := c.Get("client_info"); if !ok {
|
||||
c.AbortWithStatusJSON(
|
||||
http.StatusInternalServerError,
|
||||
@@ -105,8 +152,5 @@ func GetRequest[ModelT any](c *gin.Context) (*dto.Request[ModelT], error) {
|
||||
}
|
||||
cinfo := cinfoFromCtx.(dto.ClientInfo)
|
||||
|
||||
return &dto.Request[ModelT]{
|
||||
Body: body,
|
||||
User: cinfo,
|
||||
}, nil
|
||||
return cinfo
|
||||
}
|
||||
|
||||
232
backend/internal/controllers/profile.go
Normal file
232
backend/internal/controllers/profile.go
Normal file
@@ -0,0 +1,232 @@
|
||||
// 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/dto"
|
||||
errs "easywish/internal/errors"
|
||||
"easywish/internal/services"
|
||||
"easywish/internal/utils/enums"
|
||||
"errors"
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
type ProfileController struct {
|
||||
log *zap.Logger
|
||||
ps services.ProfileService
|
||||
}
|
||||
|
||||
func NewProfileController(_log *zap.Logger, _ps services.ProfileService) Controller {
|
||||
|
||||
ctrl := ProfileController{log: _log, ps: _ps}
|
||||
|
||||
return &controllerImpl{
|
||||
Path: "/profile",
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Methods: []ControllerMethod{
|
||||
{
|
||||
HttpMethod: GET,
|
||||
Path: "",
|
||||
Authorization: enums.UserRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.getMyProfile,
|
||||
},
|
||||
{
|
||||
HttpMethod: GET,
|
||||
Path: "/:username",
|
||||
Authorization: enums.GuestRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.getProfileByUsername,
|
||||
},
|
||||
{
|
||||
HttpMethod: GET,
|
||||
Path: "/settings",
|
||||
Authorization: enums.UserRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.getProfileSettings,
|
||||
},
|
||||
{
|
||||
HttpMethod: PATCH,
|
||||
Path: "",
|
||||
Authorization: enums.UserRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.updateProfile,
|
||||
},
|
||||
{
|
||||
HttpMethod: PATCH,
|
||||
Path: "/settings",
|
||||
Authorization: enums.UserRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.updateProfileSettings,
|
||||
},
|
||||
{
|
||||
HttpMethod: POST,
|
||||
Path: "/avatar",
|
||||
Authorization: enums.UserRole,
|
||||
Middleware: []gin.HandlerFunc{},
|
||||
Function: ctrl.uploadAvatar,
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// @Summary Get your profile
|
||||
// @Tags Profile
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Security JWT
|
||||
// @Success 200 {object} dto.ProfileDto " "
|
||||
// @Router /profile [get]
|
||||
func (ctrl *ProfileController) getMyProfile(c *gin.Context) {
|
||||
cinfo := GetClientInfo(c)
|
||||
|
||||
response, err := ctrl.ps.GetMyProfile(cinfo); if err != nil {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, response)
|
||||
}
|
||||
|
||||
// @Summary Get profile by username
|
||||
// @Tags Profile
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Security JWT
|
||||
// @Param username path string true " "
|
||||
// @Success 200 {object} dto.ProfileDto " "
|
||||
// @Failure 404 "Profile not found"
|
||||
// @Failure 403 "Restricted profile"
|
||||
// @Router /profile/{username} [get]
|
||||
func (ctrl *ProfileController) getProfileByUsername(c *gin.Context) {
|
||||
cinfo := GetClientInfo(c)
|
||||
|
||||
username := c.Param("username"); if username == "" {
|
||||
c.Status(http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
response, err := ctrl.ps.GetProfileByUsername(cinfo, username); if err != nil {
|
||||
if errors.Is(err, errs.ErrNotFound) {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "Profile not found"})
|
||||
} else if errors.Is(err, errs.ErrForbidden) {
|
||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access restricted by profile's privacy settings"})
|
||||
} else {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, response)
|
||||
|
||||
print(cinfo.Username)
|
||||
panic("Not implemented")
|
||||
}
|
||||
|
||||
// @Summary Get your profile settings
|
||||
// @Tags Profile
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Security JWT
|
||||
// @Success 200 {object} dto.ProfileSettingsDto " "
|
||||
// @Router /profile/settings [get]
|
||||
func (ctrl *ProfileController) getProfileSettings(c *gin.Context) {
|
||||
cinfo := GetClientInfo(c)
|
||||
|
||||
response, err := ctrl.ps.GetProfileSettings(cinfo); if err != nil {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, response)
|
||||
}
|
||||
|
||||
// @Summary Update your profile
|
||||
// @Tags Profile
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Security JWT
|
||||
// @Param request body dto.ProfileDto true " "
|
||||
// @Success 200 {object} bool " "
|
||||
// @Router /profile [patch]
|
||||
func (ctrl *ProfileController) updateProfile(c *gin.Context) {
|
||||
request, err := GetRequest[dto.ProfileDto](c); if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
response, err := ctrl.ps.UpdateProfile(request.User, request.Body); if err != nil || !response {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, response)
|
||||
}
|
||||
|
||||
// @Summary Update your profile's settings
|
||||
// @Tags Profile
|
||||
// @Accept json
|
||||
// @Produce json
|
||||
// @Security JWT
|
||||
// @Param request body dto.ProfileSettingsDto true " "
|
||||
// @Success 200 {object} bool " "
|
||||
// @Router /profile/settings [patch]
|
||||
func (ctrl *ProfileController) updateProfileSettings(c *gin.Context) {
|
||||
request, err := GetRequest[dto.ProfileSettingsDto](c); if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
response, err := ctrl.ps.UpdateProfileSettings(request.User, request.Body); if err != nil || !response {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, response)
|
||||
}
|
||||
|
||||
// @Summary Upload an avatar
|
||||
// @Tags Profile
|
||||
// @Accept mpfd
|
||||
// @Produce plain
|
||||
// @Security JWT
|
||||
// @Param file formData file true "Avatar image file"
|
||||
// @Success 200 {object} string "Uploaded image url"
|
||||
// @Router /profile/avatar [post]
|
||||
func (ctrl *ProfileController) uploadAvatar(c *gin.Context) {
|
||||
cinfo := GetClientInfo(c)
|
||||
|
||||
allowedTypes := map[string]bool{
|
||||
"image/jpeg": true,
|
||||
"image/png": true,
|
||||
"image/webp": true,
|
||||
}
|
||||
fileName, err := GetFile(c, "file", 8*1024*1024, allowedTypes); if err != nil {
|
||||
return
|
||||
}
|
||||
defer os.Remove(*fileName)
|
||||
|
||||
link, err := ctrl.ps.UploadAvatar(cinfo, *fileName); if err != nil {
|
||||
c.Status(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
c.String(http.StatusOK, link)
|
||||
}
|
||||
@@ -65,8 +65,9 @@ func setupControllers(p SetupControllersParams) {
|
||||
|
||||
var Module = fx.Module("controllers",
|
||||
fx.Provide(
|
||||
fx.Annotate(NewAuthController, fx.ResultTags(`group:"controllers"`)),
|
||||
fx.Annotate(NewServiceController, fx.ResultTags(`group:"controllers"`)),
|
||||
fx.Annotate(NewAuthController, fx.ResultTags(`group:"controllers"`)),
|
||||
fx.Annotate(NewProfileController, fx.ResultTags(`group:"controllers"`)),
|
||||
),
|
||||
fx.Invoke(setupControllers),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user