package restricted

import (
	"strconv"

	"git.ma-al.com/goc_daniel/b2b/app/config"
	"git.ma-al.com/goc_daniel/b2b/app/delivery/middleware"
	"git.ma-al.com/goc_daniel/b2b/app/delivery/middleware/perms"
	"git.ma-al.com/goc_daniel/b2b/app/service/storageService"
	"git.ma-al.com/goc_daniel/b2b/app/utils/i18n"
	"git.ma-al.com/goc_daniel/b2b/app/utils/localeExtractor"
	"git.ma-al.com/goc_daniel/b2b/app/utils/nullable"
	"git.ma-al.com/goc_daniel/b2b/app/utils/response"
	"git.ma-al.com/goc_daniel/b2b/app/utils/responseErrors"
	"github.com/gofiber/fiber/v3"
)

type StorageHandler struct {
	storageService *storageService.StorageService
	config         *config.Config
}

func NewStorageHandler() *StorageHandler {
	return &StorageHandler{
		storageService: storageService.New(),
		config:         config.Get(),
	}
}

func StorageHandlerRoutes(r fiber.Router) fiber.Router {
	handler := NewStorageHandler()

	// for all users
	r.Get("/list-content/*", handler.ListContent)
	r.Get("/download-file/*", handler.DownloadFile)

	// for admins only
	r.Get("/create-new-webdav-token", middleware.Require(perms.WebdavCreateToken), handler.CreateNewWebdavToken)

	return r
}

// accepted path looks like e.g. "/folder1/" or "folder1"
func (h *StorageHandler) ListContent(c fiber.Ctx) error {
	// relative path defaults to root directory
	abs_path, err := h.storageService.AbsPath(h.config.Storage.RootFolder, c.Params("*"))
	if err != nil {
		return c.Status(responseErrors.GetErrorStatus(err)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
	}

	entries_in_list, err := h.storageService.ListContent(abs_path)

	if err != nil {
		return c.Status(responseErrors.GetErrorStatus(err)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
	}

	return c.JSON(response.Make(entries_in_list, 0, i18n.T_(c, response.Message_OK)))
}

func (h *StorageHandler) DownloadFile(c fiber.Ctx) error {
	abs_path, err := h.storageService.AbsPath(h.config.Storage.RootFolder, c.Params("*"))
	if err != nil {
		return c.Status(responseErrors.GetErrorStatus(err)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
	}

	f, filename, filesize, err := h.storageService.DownloadFilePrep(abs_path)
	if err != nil {
		return c.Status(responseErrors.GetErrorStatus(err)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
	}

	c.Attachment(filename)
	c.Set("Content-Length", strconv.FormatInt(filesize, 10))
	c.Set("Content-Type", "application/octet-stream")
	return c.SendStream(f, int(filesize))
}

func (h *StorageHandler) CreateNewWebdavToken(c fiber.Ctx) error {
	userID, ok := localeExtractor.GetUserID(c)
	if !ok {
		return c.Status(responseErrors.GetErrorStatus(responseErrors.ErrInvalidBody)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, responseErrors.ErrInvalidBody)))
	}

	new_token, err := h.storageService.NewWebdavToken(userID)
	if err != nil {
		return c.Status(responseErrors.GetErrorStatus(err)).
			JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
	}

	return c.JSON(response.Make(&new_token, 0, i18n.T_(c, response.Message_OK)))
}