// 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 UploadController struct {
	log *zap.Logger
	us  services.UploadService
}

func NewUploadController(_log *zap.Logger, _us services.UploadService) Controller {
	ctrl := UploadController{log: _log, us: _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 *UploadController) getAvatarUploadUrl(c *gin.Context) {
	url, formData, err := ctrl.us.GetAvatarUrl()
	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 *UploadController) getImageUploadUrl(c *gin.Context) {
	url, formData, err := ctrl.us.GetImageUrl()
	if err != nil {
		c.Status(http.StatusInternalServerError)
		return
	}

	c.JSON(http.StatusOK, models.PresignedUploadResponse{
		Url: *url,
		Fields: *formData,
	})
}