package admin

import (
	"net/http"
	"server/internal/model/common"
	"server/internal/pkg/httputil"
	"server/internal/router"
	"server/internal/service/admin"

	"github.com/go-chi/chi/v5"
)

type FileHandler struct {
	service *admin.FileService
}

var _ router.Registrar = (*FileHandler)(nil)

func NewFileHandler(service *admin.FileService) *FileHandler {
	return &FileHandler{
		service: service,
	}
}

func (h *FileHandler) Register(r chi.Router) {
	r.Route("/files", func(r chi.Router) {
		r.Get("/", h.List)
		r.Post("/", h.Upload)
	})
}

func (h *FileHandler) List(w http.ResponseWriter, r *http.Request) {
	pagination := httputil.Pagination(r)

	result, err := h.service.List(r.Context(), pagination)
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	resp := common.PageResponse{
		Page:     pagination.Page,
		PageSize: pagination.PageSize,
		List:     result.List,
		Total:    result.Total,
	}

	httputil.OkWithPage(w, &resp)
}

func (h *FileHandler) Upload(w http.ResponseWriter, r *http.Request) {
	folder := r.FormValue("folder")
	if folder == "" {
		folder = "/"
	}

	_, header, err := r.FormFile("file")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	file, err := h.service.Upload(r.Context(), folder, header)
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w, file)
}