Files
slogan/styleagent/service/file_storage.go
T
adminandClaude Opus 4.7 ef22f7672a feat: slogan-agent MVP 服务端完整实现
- 用户域:注册/登录(JWT)/修改密码/个人资料
- 照片/衣橱/身形/化身:上传存储 + 3D 化身模板匹配
- 穿搭生成:天气(高德+和风+缓存) → 规则预筛 → LLM 规划(1次调用)
  → 规则评分(5维100分制) → 全低分触发 LLM 兜底创作 → 异步任务状态机
- 效果图:选主方案后异步生成 3 视角(mock/wanx 供应商 + 内容 hash 缓存 + 每日限额)
- 商业化:合作门店列表(seed 4 家)
- 冒烟:全链路端到端验证通过(mock LLM/天气),23 个 API 端点

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-07-31 12:15:11 +08:00

52 lines
1.4 KiB
Go

package service
import (
"errors"
"fmt"
"os"
"path/filepath"
"strings"
"time"
"github.com/gogf/gf/v2/net/ghttp"
)
var allowedImageExt = map[string]bool{".jpg": true, ".jpeg": true, ".png": true, ".webp": true}
// SaveUploadedFile 保存上传文件到 workspace/{subDir},返回访问路径 /workspace/{subDir}/{filename}
func SaveUploadedFile(file *ghttp.UploadFile, subDir string) (string, error) {
if file == nil {
return "", errors.New("未收到文件")
}
ext := strings.ToLower(filepath.Ext(file.Filename))
if !allowedImageExt[ext] {
return "", errors.New("仅支持 jpg/jpeg/png/webp 格式")
}
if file.Size > 10*1024*1024 {
return "", errors.New("单张图片不能超过 10MB")
}
dir := filepath.Join("workspace", subDir)
if err := os.MkdirAll(dir, 0o755); err != nil {
return "", err
}
filename := fmt.Sprintf("%d%s", time.Now().UnixNano(), ext)
path := filepath.Join(dir, filename)
if _, err := file.Save(path); err != nil {
return "", err
}
return "/" + filepath.ToSlash(filepath.Join("workspace", subDir, filename)), nil
}
// RemoveWorkspaceFile 删除 workspace 下文件(路径穿越防护)
func RemoveWorkspaceFile(url string) error {
rel := strings.TrimPrefix(url, "/workspace/")
if rel == "" || strings.Contains(rel, "..") {
return errors.New("非法文件路径")
}
abs := filepath.Join("workspace", rel)
if _, err := os.Stat(abs); os.IsNotExist(err) {
return nil
}
return os.Remove(abs)
}