147 lines
3.4 KiB
Go
147 lines
3.4 KiB
Go
// Package filehdl 是文件上传服务的 HTTP 接入层，分别提供后台 admin 与小程序 ma 两套入口。
|
|
package filehdl
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"portal-mini-go/internal/middleware"
|
|
"portal-mini-go/internal/pkg/errors"
|
|
"portal-mini-go/internal/pkg/response"
|
|
filesvc "portal-mini-go/internal/service/file"
|
|
)
|
|
|
|
// Handler 文件上传请求处理。
|
|
type Handler struct {
|
|
svc *filesvc.Service
|
|
}
|
|
|
|
func NewHandler(svc *filesvc.Service) *Handler { return &Handler{svc: svc} }
|
|
|
|
// 表单字段名。
|
|
const formFieldFile = "file"
|
|
|
|
// Upload 接收 multipart 单文件上传。上传者身份由路由分组决定（admin 或 ma）。
|
|
func (h *Handler) Upload(c *gin.Context) {
|
|
fh, err := c.FormFile(formFieldFile)
|
|
if err != nil {
|
|
c.Error(errors.ErrBadRequest)
|
|
return
|
|
}
|
|
f, err := fh.Open()
|
|
if err != nil {
|
|
c.Error(errors.Wrap(50000, "open uploaded file", err))
|
|
return
|
|
}
|
|
defer f.Close()
|
|
|
|
typ, id := resolveUploader(c)
|
|
if id == 0 {
|
|
c.Error(errors.ErrUnauthorized)
|
|
return
|
|
}
|
|
|
|
res, err := h.svc.Upload(c.Request.Context(), filesvc.UploadInput{
|
|
Reader: f,
|
|
Size: fh.Size,
|
|
Filename: fh.Filename,
|
|
UploaderType: typ,
|
|
UploaderID: id,
|
|
})
|
|
if err != nil {
|
|
c.Error(err)
|
|
return
|
|
}
|
|
response.OK(c, res)
|
|
}
|
|
|
|
// Info 返回文件元数据。
|
|
func (h *Handler) Info(c *gin.Context) {
|
|
id, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
f, err := h.svc.Get(c.Request.Context(), id)
|
|
if err != nil {
|
|
c.Error(err)
|
|
return
|
|
}
|
|
response.OK(c, f)
|
|
}
|
|
|
|
// Raw 以流形式返回文件内容，按 MIME 设置 Content-Type，供预览/下载。
|
|
func (h *Handler) Raw(c *gin.Context) {
|
|
id, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
rc, f, err := h.svc.Download(c.Request.Context(), id)
|
|
if err != nil {
|
|
c.Error(err)
|
|
return
|
|
}
|
|
defer rc.Close()
|
|
extra := map[string]string{
|
|
"Content-Disposition": `inline; filename="` + f.Filename + `"`,
|
|
}
|
|
c.DataFromReader(http.StatusOK, f.Size, f.MimeType, rc, extra)
|
|
}
|
|
|
|
// Delete 删除文件（存储内容 + 元数据）。
|
|
func (h *Handler) Delete(c *gin.Context) {
|
|
id, ok := parseID(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
if err := h.svc.Delete(c.Request.Context(), id); err != nil {
|
|
c.Error(err)
|
|
return
|
|
}
|
|
response.OK(c, gin.H{"id": id})
|
|
}
|
|
|
|
// resolveUploader 从 context 解析上传者身份：后台走 admin_id，小程序走 user_id。
|
|
// 两者均不存在（未认证）返回 ("", 0)，由调用方拒绝。
|
|
func resolveUploader(c *gin.Context) (string, uint64) {
|
|
if aid, ok := middleware.AdminIDUint(c); ok && aid != 0 {
|
|
return "admin", aid
|
|
}
|
|
if uid := middleware.UserID(c); uid != 0 {
|
|
return "ma", uid
|
|
}
|
|
return "", 0
|
|
}
|
|
|
|
func parseID(c *gin.Context) (uint64, bool) {
|
|
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
|
if err != nil {
|
|
c.Error(errors.ErrBadRequest)
|
|
return 0, false
|
|
}
|
|
return id, true
|
|
}
|
|
|
|
// RegisterAdminUpload 在已鉴权（AdminAuth + PermissionLoader）的后台分组下挂载上传接口。
|
|
func RegisterAdminUpload(g *gin.RouterGroup, h *Handler) {
|
|
rg := g.Group("/files")
|
|
{
|
|
rg.POST("/upload", h.Upload)
|
|
rg.GET("/:id", h.Info)
|
|
rg.GET("/:id/raw", h.Raw)
|
|
rg.DELETE("/:id", h.Delete)
|
|
}
|
|
}
|
|
|
|
// RegisterMiniAppUpload 在已鉴权（MiniAppAuth + MiniUserAuth）的小程序分组下挂载上传接口。
|
|
func RegisterMiniAppUpload(g *gin.RouterGroup, h *Handler) {
|
|
rg := g.Group("/files")
|
|
{
|
|
rg.POST("/upload", h.Upload)
|
|
rg.GET("/:id", h.Info)
|
|
rg.GET("/:id/raw", h.Raw)
|
|
rg.DELETE("/:id", h.Delete)
|
|
}
|
|
}
|