refactor(service): 重构服务代码结构并更新配置

common/util/config.go

@@ -0,0 +1,18 @@
+package util
+
+import (
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+// GetModelPrompt 获取请求模型的提示词
+func GetModelPrompt(ctx context.Context, Type int) string {
+	return g.Cfg().MustGet(ctx, "modelPrompts.types."+gconv.String(Type), "").String()
+}
+
+// GetBuildPrompt 获取构建提示词
+func GetBuildPrompt(ctx context.Context, Type int) string {
+	return g.Cfg().MustGet(ctx, "buildProject.types."+gconv.String(Type), "").String()
+}

common/util/files.go

@@ -0,0 +1,88 @@
+package util
+
+import (
+	"path/filepath"
+	"regexp"
+	"strings"
+)
+
+// AllowedMIMEPrefixes 允许的文本类 MIME 类型前缀
+var AllowedMIMEPrefixes = []string{
+	"text/",
+	"application/json",
+	"application/xml",
+	"application/javascript",
+	"application/x-yaml",
+	"application/yaml",
+	"application/toml",
+	"application/x-httpd-php",
+	"application/x-sh",
+	"application/x-python",
+	"application/x-perl",
+	"application/x-ruby",
+}
+
+// BannedExtensions 禁止的文件扩展名
+var BannedExtensions = map[string]bool{
+	".png": true, ".jpg": true, ".jpeg": true, ".gif": true, ".bmp": true,
+	".webp": true, ".svg": true, ".ico": true, ".tiff": true, ".tif": true,
+	".mp3": true, ".wav": true, ".ogg": true, ".flac": true, ".aac": true,
+	".wma": true, ".m4a": true,
+	".mp4": true, ".avi": true, ".mkv": true, ".mov": true, ".wmv": true,
+	".flv": true, ".webm": true,
+	".tar": true, ".gz": true, ".rar": true, ".7z": true,
+	".exe": true, ".dll": true, ".so": true, ".bin": true, ".dat": true,
+	".class": true, ".pyc": true,
+	".pdf": true, ".doc": true, ".docx": true, ".xls": true, ".xlsx": true,
+	".ppt": true, ".pptx": true,
+}
+
+var symbolCleaner = regexp.MustCompile(`[\x00-\x08\x0B\x0C\x0E-\x1F]`)
+
+// SanitizeURL 清洗 URL 字符串
+func SanitizeURL(raw string) string {
+	s := strings.TrimSpace(raw)
+	s = strings.Trim(s, "`\"")
+	return s
+}
+
+// CleanSymbols 清洗文本中的控制字符和多余空行
+func CleanSymbols(text string) string {
+	text = symbolCleaner.ReplaceAllString(text, "")
+	text = strings.ReplaceAll(text, "\r\n", "\n")
+	text = strings.ReplaceAll(text, "\r", "\n")
+	text = regexp.MustCompile(`\n{3,}`).ReplaceAllString(text, "\n\n")
+	return strings.TrimSpace(text)
+}
+
+// IsBannedExtension 判断是否为禁止的文件扩展名
+func IsBannedExtension(url string) bool {
+	ext := strings.ToLower(filepath.Ext(url))
+	if idx := strings.Index(ext, "?"); idx != -1 {
+		ext = ext[:idx]
+	}
+	return BannedExtensions[ext]
+}
+
+// IsZipExtension 判断是否为 zip 文件
+func IsZipExtension(url string) bool {
+	ext := strings.ToLower(filepath.Ext(url))
+	if idx := strings.Index(ext, "?"); idx != -1 {
+		ext = ext[:idx]
+	}
+	return ext == ".zip"
+}
+
+// IsReadableContentType 判断是否为可读的文本类型
+func IsReadableContentType(contentType string) bool {
+	if contentType == "" {
+		return false
+	}
+	ct := strings.ToLower(contentType)
+	for _, prefix := range AllowedMIMEPrefixes {
+		if strings.HasPrefix(ct, prefix) {
+			return true
+		}
+	}
+	return false
+}

common/util/headers.go

@@ -0,0 +1,52 @@
+package util
+
+import (
+	"context"
+
+	"gitea.com/red-future/common/utils"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+// AsyncCtx 固化异步上下文中的 token 和用户信息，避免请求结束后丢失
+func AsyncCtx(ctx context.Context) context.Context {
+	asyncCtx := context.WithoutCancel(ctx)
+	if r := g.RequestFromCtx(ctx); r != nil {
+		if token := r.Header.Get("Authorization"); token != "" {
+			asyncCtx = context.WithValue(asyncCtx, "token", token)
+		}
+		if userInfo := r.Header.Get("X-User-Info"); userInfo != "" {
+			asyncCtx = context.WithValue(asyncCtx, "xUserInfo", userInfo)
+		}
+	}
+	if user, err := utils.GetUserInfo(ctx); err == nil && user != nil {
+		asyncCtx = context.WithValue(asyncCtx, "user", user)
+	}
+	return asyncCtx
+}
+
+// ForwardHeaders 透传调用链路的头信息，优先使用 ctx 中的固化值
+func ForwardHeaders(ctx context.Context) map[string]string {
+	headers := make(map[string]string)
+
+	if token, ok := ctx.Value("token").(string); ok && token != "" {
+		headers["Authorization"] = token
+	}
+	if x, ok := ctx.Value("xUserInfo").(string); ok && x != "" {
+		headers["X-User-Info"] = x
+	}
+
+	// 兜底：从请求头获取
+	if r := g.RequestFromCtx(ctx); r != nil {
+		if headers["Authorization"] == "" {
+			if token := r.Header.Get("Authorization"); token != "" {
+				headers["Authorization"] = token
+			}
+		}
+		if headers["X-User-Info"] == "" {
+			if userInfo := r.Header.Get("X-User-Info"); userInfo != "" {
+				headers["X-User-Info"] = userInfo
+			}
+		}
+	}
+	return headers
+}

common/util/json.go

@@ -0,0 +1,103 @@
+package util
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/gogf/gf/v2/container/gvar"
+	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+// ParseOutput 解析模型输出为 JSON 格式
+func ParseOutput(text string) (map[string]any, error) {
+	j, err := gjson.LoadJson([]byte(text))
+	if err != nil {
+		return nil, fmt.Errorf("解析模型输出失败: %w", err)
+	}
+	return j.Map(), nil
+}
+
+// ConvertToMessages 将原始数据转换为消息列表
+func ConvertToMessages(raw any) []map[string]any {
+	if raw == nil {
+		return nil
+	}
+	j, err := gjson.LoadJson(gconv.Bytes(raw))
+	if err != nil {
+		return nil
+	}
+	// 如果有 messages 字段，直接返回
+	if j.Contains("messages") {
+		return gconv.Maps(j.Get("messages").Array())
+	}
+	// 否则当成单条 message
+	return []map[string]any{
+		j.Map(),
+	}
+}
+
+// IsMessageValid 校验推理结果是否合法
+func IsMessageValid(message map[string]any) bool {
+	if message == nil {
+		return false
+	}
+	return true
+}
+
+// FormToJSON 将表单数据转换为 JSON 字符串
+func FormToJSON(form map[string]any) string {
+	if form == nil {
+		return "{}"
+	}
+	b, _ := json.Marshal(form)
+	return string(b)
+}
+
+// MustMarshal 将对象序列化为 JSON 字符串，失败时返回空对象
+func MustMarshal(v any) string {
+	b, err := json.Marshal(v)
+	if err != nil {
+		return "{}"
+	}
+	return string(b)
+}
+
+// ParseJSONField 解析 JSON 字段
+func ParseJSONField(field any) any {
+	var v *gvar.Var
+	switch val := field.(type) {
+	case *gvar.Var:
+		v = val
+	default:
+		return field
+	}
+
+	if v == nil || v.IsNil() || v.IsEmpty() {
+		return nil
+	}
+
+	str := v.String()
+	var result any
+	if json.Unmarshal([]byte(str), &result) == nil {
+		return result
+	}
+	return str
+}
+
+// JSONPretty 将任意类型转为格式化的 JSON 字符串
+func JSONPretty(v any) string {
+	// 处理 *gvar.Var 类型
+	if gv, ok := v.(*gvar.Var); ok {
+		v = gconv.Map(gv.String())
+	}
+
+	// 统一转 map 再美化
+	var tmp map[string]any
+	if err := gconv.Struct(v, &tmp); err != nil {
+		return gconv.String(v)
+	}
+
+	b, _ := json.MarshalIndent(tmp, "", "  ")
+	return string(b)
+}