102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
|
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", "资料创建失败")
|
|||
|
|
}
|
|||
|
|
}
|