pocket-id-pocket-id-2/backend/internal/controller/app_config_controller.go

package controller

import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/stonith404/pocket-id/backend/internal/common"
	"github.com/stonith404/pocket-id/backend/internal/middleware"
	"github.com/stonith404/pocket-id/backend/internal/model"
	"github.com/stonith404/pocket-id/backend/internal/service"
	"github.com/stonith404/pocket-id/backend/internal/utils"
	"net/http"
)

func NewApplicationConfigurationController(
	group *gin.RouterGroup,
	jwtAuthMiddleware *middleware.JwtAuthMiddleware,
	appConfigService *service.AppConfigService) {

	acc := &ApplicationConfigurationController{
		appConfigService: appConfigService,
	}
	group.GET("/application-configuration", acc.listApplicationConfigurationHandler)
	group.GET("/application-configuration/all", jwtAuthMiddleware.Add(true), acc.listAllApplicationConfigurationHandler)
	group.PUT("/application-configuration", acc.updateApplicationConfigurationHandler)

	group.GET("/application-configuration/logo", acc.getLogoHandler)
	group.GET("/application-configuration/background-image", acc.getBackgroundImageHandler)
	group.GET("/application-configuration/favicon", acc.getFaviconHandler)
	group.PUT("/application-configuration/logo", jwtAuthMiddleware.Add(true), acc.updateLogoHandler)
	group.PUT("/application-configuration/favicon", jwtAuthMiddleware.Add(true), acc.updateFaviconHandler)
	group.PUT("/application-configuration/background-image", jwtAuthMiddleware.Add(true), acc.updateBackgroundImageHandler)
}

type ApplicationConfigurationController struct {
	appConfigService *service.AppConfigService
}

func (acc *ApplicationConfigurationController) listApplicationConfigurationHandler(c *gin.Context) {
	configuration, err := acc.appConfigService.ListApplicationConfiguration(false)
	if err != nil {
		utils.UnknownHandlerError(c, err)
		return
	}

	c.JSON(200, configuration)
}

func (acc *ApplicationConfigurationController) listAllApplicationConfigurationHandler(c *gin.Context) {
	configuration, err := acc.appConfigService.ListApplicationConfiguration(true)
	if err != nil {
		utils.UnknownHandlerError(c, err)
		return
	}

	c.JSON(200, configuration)
}

func (acc *ApplicationConfigurationController) updateApplicationConfigurationHandler(c *gin.Context) {
	var input model.AppConfigUpdateDto
	if err := c.ShouldBindJSON(&input); err != nil {
		utils.HandlerError(c, http.StatusBadRequest, common.ErrInvalidBody.Error())
		return
	}

	savedConfigVariables, err := acc.appConfigService.UpdateApplicationConfiguration(input)
	if err != nil {
		utils.UnknownHandlerError(c, err)
		return
	}

	c.JSON(http.StatusOK, savedConfigVariables)
}

func (acc *ApplicationConfigurationController) getLogoHandler(c *gin.Context) {
	imageType := acc.appConfigService.DbConfig.LogoImageType.Value
	acc.getImage(c, "logo", imageType)
}

func (acc *ApplicationConfigurationController) getFaviconHandler(c *gin.Context) {
	acc.getImage(c, "favicon", "ico")
}

func (acc *ApplicationConfigurationController) getBackgroundImageHandler(c *gin.Context) {
	imageType := acc.appConfigService.DbConfig.BackgroundImageType.Value
	acc.getImage(c, "background", imageType)
}

func (acc *ApplicationConfigurationController) updateLogoHandler(c *gin.Context) {
	imageType := acc.appConfigService.DbConfig.LogoImageType.Value
	acc.updateImage(c, "logo", imageType)
}

func (acc *ApplicationConfigurationController) updateFaviconHandler(c *gin.Context) {
	file, err := c.FormFile("file")
	if err != nil {
		utils.HandlerError(c, http.StatusBadRequest, common.ErrInvalidBody.Error())
		return
	}

	fileType := utils.GetFileExtension(file.Filename)
	if fileType != "ico" {
		utils.HandlerError(c, http.StatusBadRequest, "File must be of type .ico")
		return
	}
	acc.updateImage(c, "favicon", "ico")
}

func (acc *ApplicationConfigurationController) updateBackgroundImageHandler(c *gin.Context) {
	imageType := acc.appConfigService.DbConfig.BackgroundImageType.Value
	acc.updateImage(c, "background", imageType)
}

func (acc *ApplicationConfigurationController) getImage(c *gin.Context, name string, imageType string) {
	imagePath := fmt.Sprintf("%s/application-images/%s.%s", common.EnvConfig.UploadPath, name, imageType)
	mimeType := utils.GetImageMimeType(imageType)

	c.Header("Content-Type", mimeType)
	c.File(imagePath)
}

func (acc *ApplicationConfigurationController) updateImage(c *gin.Context, imageName string, oldImageType string) {
	file, err := c.FormFile("file")
	if err != nil {
		utils.HandlerError(c, http.StatusBadRequest, common.ErrInvalidBody.Error())
		return
	}

	err = acc.appConfigService.UpdateImage(file, imageName, oldImageType)
	if err != nil {
		if errors.Is(err, common.ErrFileTypeNotSupported) {
			utils.HandlerError(c, http.StatusBadRequest, err.Error())
		} else {
			utils.UnknownHandlerError(c, err)
		}
		return
	}

	c.Status(http.StatusNoContent)
}