71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
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)
|
|
}
|