refactor(backend): split project write responsibilities

This commit is contained in:
2026-07-21 15:27:11 +08:00
parent 9e862f2a7f
commit 1fcbb31301
18 changed files with 1025 additions and 649 deletions

View File

@@ -0,0 +1,101 @@
package files
import (
"errors"
"net/http"
"time"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"senlinai-agent/backend/internal/httpx"
"senlinai-agent/backend/internal/logic/auth"
"senlinai-agent/backend/internal/logic/projects"
"senlinai-agent/backend/internal/models"
)
// SourceDTO 是文件资料写接口的稳定响应,仅包含可公开的相对存储路径。
type SourceDTO struct {
ID string `json:"id"`
ProjectID string `json:"projectId"`
Kind string `json:"kind"`
Title string `json:"title"`
FilePath string `json:"filePath"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
}
// Handler 负责 multipart 边界,文件名清理和本地路径构造始终委托给 Service。
type Handler struct {
service *Service
}
// NewHandler 创建文件资料 HTTP registrar。
func NewHandler(service *Service) *Handler {
return &Handler{service: service}
}
// Register 将文件资料上传接口注册到上层提供的 /api/v1 路由组。
func (h *Handler) Register(router gin.IRouter) {
router.POST("/projects/:id/sources", h.upload)
}
func (h *Handler) upload(c *gin.Context) {
userID, ok := auth.CurrentUserID(c)
if !ok {
httpx.Error(c, http.StatusUnauthorized, "unauthorized", "未登录或登录已失效")
return
}
projectIdentity, ok := httpx.IdentityParam(c, "id")
if !ok {
return
}
// 所有权必须在写磁盘前校验,防止越权请求留下孤立文件。
project, err := projects.FindOwnedProject(userID, projectIdentity)
if err != nil {
writeSourceError(c, err)
return
}
fileHeader, err := c.FormFile("file")
if err != nil {
httpx.Error(c, http.StatusBadRequest, "invalid_request", "请选择要上传的文件")
return
}
file, err := fileHeader.Open()
if err != nil {
httpx.Error(c, http.StatusBadRequest, "invalid_request", "文件读取失败")
return
}
defer file.Close()
// 先由文件服务保存内容再用其返回的相对路径创建资料记录handler 不拼接任何本地路径。
stored, err := h.service.Save(project.ID, fileHeader.Filename, file)
if err != nil {
httpx.Error(c, http.StatusInternalServerError, "internal_error", "文件保存失败")
return
}
source, err := h.service.CreateSource(userID, project, c.PostForm("title"), stored)
if err != nil {
writeSourceError(c, err)
return
}
c.JSON(http.StatusCreated, sourceDTO(*source))
}
func sourceDTO(source models.SenlinAgentSource) SourceDTO {
return SourceDTO{
ID: source.Identity, ProjectID: source.ProjectIdentity, Kind: source.Kind,
Title: source.Title, FilePath: source.FilePath,
CreatedAt: source.CreatedAt.UTC(), UpdatedAt: source.UpdatedAt.UTC(),
}
}
func writeSourceError(c *gin.Context, err error) {
switch {
case errors.Is(err, gorm.ErrRecordNotFound):
httpx.Error(c, http.StatusNotFound, "not_found", "项目不存在")
case errors.Is(err, ErrSourceTitleRequired), errors.Is(err, ErrSourcePathRequired):
httpx.Error(c, http.StatusBadRequest, "invalid_request", "资料参数无效")
default:
httpx.Error(c, http.StatusInternalServerError, "internal_error", "资料创建失败")
}
}