项目初始化

2026-01-13 17:00:49 +08:00
commit d2352318f4
27 changed files with 4343 additions and 0 deletions
--- a/internal/api/public/pickup.go
+++ b/internal/api/public/pickup.go
@@ -0,0 +1,162 @@
+package public
+
+import (
+	"FileRelay/internal/api/middleware"
+	"FileRelay/internal/bootstrap"
+	"FileRelay/internal/model"
+	"FileRelay/internal/service"
+	"FileRelay/internal/storage"
+	"archive/zip"
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+type PickupResponse struct {
+	Remark        string           `json:"remark"`
+	ExpireAt      *time.Time       `json:"expire_at"`
+	ExpireType    string           `json:"expire_type"`
+	DownloadCount int              `json:"download_count"`
+	MaxDownloads  int              `json:"max_downloads"`
+	Files         []model.FileItem `json:"files"`
+}
+
+// DownloadBatch 批量下载文件 (ZIP)
+// @Summary      批量下载文件
+// @Description  根据取件码将批次内的所有文件打包为 ZIP 格式一次性下载
+// @Tags         Public
+// @Param        pickup_code  path  string  true  "取件码"
+// @Produce      application/zip
+// @Success      200  {file}    file
+// @Failure      404  {object}  model.Response
+// @Router       /api/download/batch/{pickup_code} [get]
+func (h *PickupHandler) DownloadBatch(c *gin.Context) {
+	code := c.Param("pickup_code")
+	batch, err := h.batchService.GetBatchByPickupCode(code)
+	if err != nil {
+		c.JSON(http.StatusNotFound, model.ErrorResponse(model.CodeNotFound, "batch not found or expired"))
+		return
+	}
+
+	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"batch_%s.zip\"", code))
+	c.Header("Content-Type", "application/zip")
+
+	zw := zip.NewWriter(c.Writer)
+	defer zw.Close()
+
+	for _, item := range batch.FileItems {
+		reader, err := storage.GlobalStorage.Open(c.Request.Context(), item.StoragePath)
+		if err != nil {
+			continue // Skip failed files
+		}
+
+		f, err := zw.Create(item.OriginalName)
+		if err != nil {
+			reader.Close()
+			continue
+		}
+
+		_, _ = io.Copy(f, reader)
+		reader.Close()
+	}
+
+	// 增加下载次数
+	h.batchService.IncrementDownloadCount(batch.ID)
+}
+
+type PickupHandler struct {
+	batchService *service.BatchService
+}
+
+func NewPickupHandler() *PickupHandler {
+	return &PickupHandler{
+		batchService: service.NewBatchService(),
+	}
+}
+
+// Pickup 获取批次信息
+// @Summary      获取批次信息
+// @Description  根据取件码获取文件批次详详情和文件列表
+// @Tags         Public
+// @Produce      json
+// @Param        pickup_code  path  string  true  "取件码"
+// @Success      200  {object}  model.Response{data=PickupResponse}
+// @Failure      404  {object}  model.Response
+// @Router       /api/pickup/{pickup_code} [get]
+func (h *PickupHandler) Pickup(c *gin.Context) {
+	code := c.Param("pickup_code")
+	if code == "" {
+		c.JSON(http.StatusBadRequest, model.ErrorResponse(model.CodeBadRequest, "pickup code required"))
+		return
+	}
+
+	batch, err := h.batchService.GetBatchByPickupCode(code)
+	if err != nil {
+		middleware.RecordPickupFailure(c.ClientIP(), code)
+		c.JSON(http.StatusNotFound, model.ErrorResponse(model.CodeNotFound, "batch not found or expired"))
+		return
+	}
+
+	c.JSON(http.StatusOK, model.SuccessResponse(PickupResponse{
+		Remark:        batch.Remark,
+		ExpireAt:      batch.ExpireAt,
+		ExpireType:    batch.ExpireType,
+		DownloadCount: batch.DownloadCount,
+		MaxDownloads:  batch.MaxDownloads,
+		Files:         batch.FileItems,
+	}))
+}
+
+// DownloadFile 下载单个文件
+// @Summary      下载单个文件
+// @Description  根据文件 ID 下载单个文件
+// @Tags         Public
+// @Param        file_id  path  int  true  "文件 ID"
+// @Produce      application/octet-stream
+// @Success      200  {file}    file
+// @Failure      404  {object}  model.Response
+// @Failure      410  {object}  model.Response
+// @Router       /api/download/file/{file_id} [get]
+func (h *PickupHandler) DownloadFile(c *gin.Context) {
+	fileIDStr := c.Param("file_id")
+	fileID, _ := strconv.ParseUint(fileIDStr, 10, 32)
+
+	var item model.FileItem
+	if err := bootstrap.DB.First(&item, fileID).Error; err != nil {
+		c.JSON(http.StatusNotFound, model.ErrorResponse(model.CodeNotFound, "file not found"))
+		return
+	}
+
+	var batch model.FileBatch
+	if err := bootstrap.DB.First(&batch, item.BatchID).Error; err != nil {
+		c.JSON(http.StatusNotFound, model.ErrorResponse(model.CodeNotFound, "batch not found"))
+		return
+	}
+
+	if h.batchService.IsExpired(&batch) {
+		h.batchService.MarkAsExpired(&batch)
+		c.JSON(http.StatusGone, model.ErrorResponse(model.CodeGone, "batch expired"))
+		return
+	}
+
+	// 打开文件
+	reader, err := storage.GlobalStorage.Open(c.Request.Context(), item.StoragePath)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "failed to open file"))
+		return
+	}
+	defer reader.Close()
+
+	// 增加下载次数
+	h.batchService.IncrementDownloadCount(batch.ID)
+
+	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", item.OriginalName))
+	c.Header("Content-Type", item.MimeType)
+	c.Header("Content-Length", strconv.FormatInt(item.Size, 10))
+
+	io.Copy(c.Writer, reader)
+}
--- a/internal/api/public/upload.go
+++ b/internal/api/public/upload.go
@@ -0,0 +1,96 @@
+package public
+
+import (
+	"FileRelay/internal/config"
+	"FileRelay/internal/model"
+	"FileRelay/internal/service"
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+type UploadHandler struct {
+	uploadService *service.UploadService
+}
+
+func NewUploadHandler() *UploadHandler {
+	return &UploadHandler{
+		uploadService: service.NewUploadService(),
+	}
+}
+
+type UploadResponse struct {
+	PickupCode string     `json:"pickup_code"`
+	ExpireAt   *time.Time `json:"expire_at"`
+	BatchID    uint       `json:"batch_id"`
+}
+
+// Upload 上传文件并生成取件码
+// @Summary      上传文件
+// @Description  上传一个或多个文件并创建一个提取批次
+// @Tags         Public
+// @Accept       multipart/form-data
+// @Produce      json
+// @Param        files       formData  file    true   "文件列表"
+// @Param        remark      formData  string  false  "备注"
+// @Param        expire_type formData  string  false  "过期类型 (time/download/permanent)"
+// @Param        expire_days formData  int     false  "过期天数 (针对 time 类型)"
+// @Param        max_downloads formData int    false  "最大下载次数 (针对 download 类型)"
+// @Success      200  {object}  model.Response{data=UploadResponse}
+// @Failure      400  {object}  model.Response
+// @Failure      500  {object}  model.Response
+// @Router       /api/upload [post]
+func (h *UploadHandler) Upload(c *gin.Context) {
+	form, err := c.MultipartForm()
+	if err != nil {
+		c.JSON(http.StatusBadRequest, model.ErrorResponse(model.CodeBadRequest, "invalid form"))
+		return
+	}
+
+	files := form.File["files"]
+	if len(files) == 0 {
+		c.JSON(http.StatusBadRequest, model.ErrorResponse(model.CodeBadRequest, "no files uploaded"))
+		return
+	}
+
+	if len(files) > config.GlobalConfig.Upload.MaxBatchFiles {
+		c.JSON(http.StatusBadRequest, model.ErrorResponse(model.CodeBadRequest, "too many files"))
+		return
+	}
+
+	remark := c.PostForm("remark")
+	expireType := c.PostForm("expire_type") // time / download / permanent
+	if expireType == "" {
+		expireType = "time"
+	}
+
+	var expireValue interface{}
+	switch expireType {
+	case "time":
+		days, _ := strconv.Atoi(c.PostForm("expire_days"))
+		if days <= 0 {
+			days = config.GlobalConfig.Upload.MaxRetentionDays
+		}
+		expireValue = days
+	case "download":
+		max, _ := strconv.Atoi(c.PostForm("max_downloads"))
+		if max <= 0 {
+			max = 1
+		}
+		expireValue = max
+	}
+
+	batch, err := h.uploadService.CreateBatch(c.Request.Context(), files, remark, expireType, expireValue)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, err.Error()))
+		return
+	}
+
+	c.JSON(http.StatusOK, model.SuccessResponse(UploadResponse{
+		PickupCode: batch.PickupCode,
+		ExpireAt:   batch.ExpireAt,
+		BatchID:    batch.ID,
+	}))
+}