This commit is contained in:
commit
5e4b42493d
|
|
@ -0,0 +1,2 @@
|
|||
/.idea
|
||||
/outputs
|
||||
|
|
@ -0,0 +1,71 @@
|
|||
# ============================================================
|
||||
# 阶段1: 构建
|
||||
# ============================================================
|
||||
FROM golang:1.26-alpine AS builder
|
||||
|
||||
# 设置工作目录
|
||||
WORKDIR /app
|
||||
|
||||
# 安装必要的构建工具
|
||||
RUN apk add --no-cache git make
|
||||
|
||||
# 复制 go.mod 和 go.sum 并下载依赖（利用 Docker 缓存）
|
||||
COPY go.mod go.sum ./
|
||||
RUN go mod download && go mod verify
|
||||
|
||||
# 复制源代码
|
||||
COPY . .
|
||||
|
||||
# 构建二进制文件
|
||||
RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build \
|
||||
-ldflags="-s -w" \
|
||||
-o /app/sdk-generator \
|
||||
./cmd/server
|
||||
|
||||
# ============================================================
|
||||
# 阶段2: 运行
|
||||
# ============================================================
|
||||
FROM alpine:latest
|
||||
|
||||
# 安装运行时依赖
|
||||
RUN apk add --no-cache ca-certificates tzdata git
|
||||
|
||||
# 设置时区
|
||||
ENV TZ=Asia/Shanghai
|
||||
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
||||
|
||||
# 创建非 root 用户
|
||||
RUN adduser -D -g '' appuser
|
||||
|
||||
# 设置工作目录
|
||||
WORKDIR /app
|
||||
|
||||
# 从构建阶段复制二进制文件
|
||||
COPY --from=builder /app/sdk-generator /app/sdk-generator
|
||||
|
||||
# 复制 Web 静态文件和模板
|
||||
COPY --chown=appuser:appuser ./web /app/web
|
||||
|
||||
# 创建数据目录并设置权限
|
||||
RUN mkdir -p /app/outputs /app/uploads && \
|
||||
chown -R appuser:appuser /app
|
||||
|
||||
# 切换到非 root 用户
|
||||
USER appuser
|
||||
|
||||
# 暴露端口
|
||||
EXPOSE 5002
|
||||
|
||||
# 健康检查
|
||||
HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
|
||||
CMD wget --no-verbose --tries=1 --spider http://localhost:8080/health || exit 1
|
||||
|
||||
# 启动
|
||||
ENTRYPOINT ["/app/sdk-generator"]
|
||||
|
||||
|
||||
#docker run -d \
|
||||
# -p 8080:8080 \
|
||||
# -v /var/www/sdk-generator/outputs:/app/outputs \
|
||||
# --name sdk-generator \
|
||||
# sdk-generator
|
||||
|
|
@ -0,0 +1,31 @@
|
|||
module sdk-generator
|
||||
|
||||
go 1.26.2
|
||||
|
||||
require (
|
||||
code.gitea.io/sdk/gitea v0.25.1
|
||||
github.com/gofiber/fiber/v2 v2.52.13
|
||||
github.com/gofiber/template/html/v2 v2.1.3
|
||||
github.com/google/uuid v1.6.0
|
||||
github.com/sashabaranov/go-openai v1.28.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/42wim/httpsig v1.2.4 // indirect
|
||||
github.com/andybalholm/brotli v1.1.0 // indirect
|
||||
github.com/davidmz/go-pageant v1.0.2 // indirect
|
||||
github.com/go-fed/httpsig v1.1.0 // indirect
|
||||
github.com/gofiber/template v1.8.3 // indirect
|
||||
github.com/gofiber/utils v1.1.0 // indirect
|
||||
github.com/hashicorp/go-version v1.9.0 // indirect
|
||||
github.com/klauspost/compress v1.17.9 // indirect
|
||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
||||
github.com/rivo/uniseg v0.2.0 // indirect
|
||||
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
||||
github.com/valyala/fasthttp v1.51.0 // indirect
|
||||
github.com/valyala/tcplisten v1.0.0 // indirect
|
||||
golang.org/x/crypto v0.50.0 // indirect
|
||||
golang.org/x/sys v0.43.0 // indirect
|
||||
)
|
||||
|
|
@ -0,0 +1,69 @@
|
|||
code.gitea.io/sdk/gitea v0.25.1 h1:yywxWwoV+SdjHtbC6unBiXojWdZOtoHuGhEazEXeWuE=
|
||||
code.gitea.io/sdk/gitea v0.25.1/go.mod h1:uDFWYBU8dgZsgOHwe6C/6olxvf8FHguNB3wW1i83fgg=
|
||||
github.com/42wim/httpsig v1.2.4 h1:mI5bH0nm4xn7K18fo1K3okNDRq8CCJ0KbBYWyA6r8lU=
|
||||
github.com/42wim/httpsig v1.2.4/go.mod h1:yKsYfSyTBEohkPik224QPFylmzEBtda/kjyIAJjh3ps=
|
||||
github.com/andybalholm/brotli v1.1.0 h1:eLKJA0d02Lf0mVpIDgYnqXcUn0GqVmEFny3VuID1U3M=
|
||||
github.com/andybalholm/brotli v1.1.0/go.mod h1:sms7XGricyQI9K10gOSf56VKKWS4oLer58Q+mhRPtnY=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davidmz/go-pageant v1.0.2 h1:bPblRCh5jGU+Uptpz6LgMZGD5hJoOt7otgT454WvHn0=
|
||||
github.com/davidmz/go-pageant v1.0.2/go.mod h1:P2EDDnMqIwG5Rrp05dTRITj9z2zpGcD9efWSkTNKLIE=
|
||||
github.com/go-fed/httpsig v1.1.0 h1:9M+hb0jkEICD8/cAiNqEB66R87tTINszBRTjwjQzWcI=
|
||||
github.com/go-fed/httpsig v1.1.0/go.mod h1:RCMrTZvN1bJYtofsG4rd5NaO5obxQ5xBkdiS7xsT7bM=
|
||||
github.com/gofiber/fiber/v2 v2.52.13 h1:TOKP64iqC9b5P49VrBW5tHhUOvDyrtJ0xePEfzJbCbk=
|
||||
github.com/gofiber/fiber/v2 v2.52.13/go.mod h1:YEcBbO/FB+5M1IZNBP9FO3J9281zgPAreiI1oqg8nDw=
|
||||
github.com/gofiber/template v1.8.3 h1:hzHdvMwMo/T2kouz2pPCA0zGiLCeMnoGsQZBTSYgZxc=
|
||||
github.com/gofiber/template v1.8.3/go.mod h1:bs/2n0pSNPOkRa5VJ8zTIvedcI/lEYxzV3+YPXdBvq8=
|
||||
github.com/gofiber/template/html/v2 v2.1.3 h1:n1LYBtmr9C0V/k/3qBblXyMxV5B0o/gpb6dFLp8ea+o=
|
||||
github.com/gofiber/template/html/v2 v2.1.3/go.mod h1:U5Fxgc5KpyujU9OqKzy6Kn6Qup6Tm7zdsISR+VpnHRE=
|
||||
github.com/gofiber/utils v1.1.0 h1:vdEBpn7AzIUJRhe+CiTOJdUcTg4Q9RK+pEa0KPbLdrM=
|
||||
github.com/gofiber/utils v1.1.0/go.mod h1:poZpsnhBykfnY1Mc0KeEa6mSHrS3dV0+oBWyeQmb2e0=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/hashicorp/go-version v1.9.0 h1:CeOIz6k+LoN3qX9Z0tyQrPtiB1DFYRPfCIBtaXPSCnA=
|
||||
github.com/hashicorp/go-version v1.9.0/go.mod h1:fltr4n8CU8Ke44wwGCBoEymUuxUHl09ZGVZPK5anwXA=
|
||||
github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA=
|
||||
github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw=
|
||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/sashabaranov/go-openai v1.28.0 h1:WS9F9BriSvtHvknPQy2Oi3b+8zkmJdEXcycrWqrSicQ=
|
||||
github.com/sashabaranov/go-openai v1.28.0/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
|
||||
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
|
||||
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
|
||||
github.com/valyala/fasthttp v1.51.0 h1:8b30A5JlZ6C7AS81RsWjYMQmrZG6feChmgAolCl1SqA=
|
||||
github.com/valyala/fasthttp v1.51.0/go.mod h1:oI2XroL+lI7vdXyYoQk03bXBThfFl2cVdIA3Xl7cH8g=
|
||||
github.com/valyala/tcplisten v1.0.0 h1:rBHj/Xf+E1tRGZyWIWwJDiRY0zc1Js+CV5DqwacVSA8=
|
||||
github.com/valyala/tcplisten v1.0.0/go.mod h1:T0xQ8SeCZGxckz9qRXTfG43PvQ/mcWh7FwZEA7Ioqkc=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||
golang.org/x/crypto v0.0.0-20210513164829-c07d793c2f9a/go.mod h1:P+XmwS30IXTQdn5tA2iutPOUgjI07+tq3H3K9MVA1s8=
|
||||
golang.org/x/crypto v0.50.0 h1:zO47/JPrL6vsNkINmLoo/PH1gcxpls50DNogFvB5ZGI=
|
||||
golang.org/x/crypto v0.50.0/go.mod h1:3muZ7vA7PBCE6xgPX7nkzzjiUq87kRItoJQM1Yo8S+Q=
|
||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.43.0 h1:Rlag2XtaFTxp19wS8MXlJwTvoh8ArU6ezoyFsMyCTNI=
|
||||
golang.org/x/sys v0.43.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.42.0 h1:UiKe+zDFmJobeJ5ggPwOshJIVt6/Ft0rcfrXZDLWAWY=
|
||||
golang.org/x/term v0.42.0/go.mod h1:Dq/D+snpsbazcBG5+F9Q1n2rXV8Ma+71xEjTRufARgY=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
|
@ -0,0 +1,50 @@
|
|||
package call
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/sashabaranov/go-openai"
|
||||
)
|
||||
|
||||
type LlmCallSet struct {
|
||||
ApiKey string
|
||||
BaseUrl string
|
||||
ModelName string
|
||||
}
|
||||
|
||||
type CallLLM struct {
|
||||
ModelName string
|
||||
openaiClient *openai.Client
|
||||
}
|
||||
|
||||
func NewCallLLM(LlmCallSet *LlmCallSet) *CallLLM {
|
||||
clientConfig := openai.DefaultConfig(LlmCallSet.ApiKey)
|
||||
clientConfig.BaseURL = LlmCallSet.BaseUrl
|
||||
client := openai.NewClientWithConfig(clientConfig)
|
||||
return &CallLLM{
|
||||
openaiClient: client,
|
||||
ModelName: LlmCallSet.ModelName,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *CallLLM) Do(ctx context.Context, RequestOption openai.ChatCompletionRequest, modelName ...string) (string, error) {
|
||||
if modelName == nil {
|
||||
modelName = append(modelName, s.ModelName)
|
||||
}
|
||||
if modelName[0] == "" {
|
||||
return "", fmt.Errorf("model name is nil")
|
||||
}
|
||||
RequestOption.Model = modelName[0]
|
||||
resp, err := s.openaiClient.CreateChatCompletion(ctx, RequestOption)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
if len(resp.Choices) == 0 {
|
||||
return "", fmt.Errorf("模型未返回任何内容")
|
||||
}
|
||||
content := resp.Choices[0].Message.Content
|
||||
//content := test.Res
|
||||
return content, nil
|
||||
}
|
||||
|
|
@ -0,0 +1,54 @@
|
|||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
// 服务配置
|
||||
Port string
|
||||
UploadDir string
|
||||
OutputDir string
|
||||
|
||||
// 大模型配置
|
||||
APIKey string
|
||||
BaseURL string
|
||||
Model string
|
||||
MaxFixAttempts int
|
||||
Timeout time.Duration
|
||||
|
||||
// 限流配置
|
||||
MaxFileSize int64
|
||||
MaxTasks int
|
||||
|
||||
//gitea配置
|
||||
GiteaUrl string
|
||||
GiteaToken string
|
||||
OrgName string
|
||||
}
|
||||
|
||||
func Load() *Config {
|
||||
return &Config{
|
||||
Port: getEnv("PORT", "5002"),
|
||||
UploadDir: getEnv("UPLOAD_DIR", "./uploads"),
|
||||
OutputDir: getEnv("OUTPUT_DIR", "./outputs"),
|
||||
APIKey: getEnv("DOUBAO_API_KEY", "ark-3f942f9b-7428-4880-b939-026087b99957-062f1"),
|
||||
BaseURL: getEnv("DOUBAO_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3"),
|
||||
Model: getEnv("DOUBAO_MODEL", "ep-20260717134040-w9qhc"),
|
||||
Timeout: 5 * time.Minute,
|
||||
MaxFileSize: 10 * 1024 * 1024, // 10MB
|
||||
MaxTasks: 100,
|
||||
MaxFixAttempts: 3,
|
||||
GiteaUrl: "https://gitea.cdlsxd.cn",
|
||||
GiteaToken: "406ebea224965f9be4c25e7ff2614588827fd31f",
|
||||
OrgName: "ai_sdk",
|
||||
}
|
||||
}
|
||||
|
||||
func getEnv(key, defaultVal string) string {
|
||||
if val := os.Getenv(key); val != "" {
|
||||
return val
|
||||
}
|
||||
return defaultVal
|
||||
}
|
||||
|
|
@ -0,0 +1,151 @@
|
|||
package extractor
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type File struct {
|
||||
Path string
|
||||
Content string
|
||||
}
|
||||
|
||||
// Extract 从模型响应中提取代码文件
|
||||
// 使用纯字符串解析，不依赖 goldmark AST
|
||||
func Extract(response string) ([]File, error) {
|
||||
var files []File
|
||||
seen := make(map[string]bool)
|
||||
|
||||
// 按 // File: 分割
|
||||
parts := strings.Split(response, "// File:")
|
||||
|
||||
for _, part := range parts {
|
||||
part = strings.TrimSpace(part)
|
||||
if part == "" {
|
||||
continue
|
||||
}
|
||||
|
||||
// 第一行是文件名
|
||||
lines := strings.SplitN(part, "\n", 2)
|
||||
if len(lines) < 2 {
|
||||
continue
|
||||
}
|
||||
|
||||
fileName := strings.TrimSpace(lines[0])
|
||||
code := strings.TrimSpace(lines[1])
|
||||
|
||||
if fileName == "" || code == "" {
|
||||
continue
|
||||
}
|
||||
|
||||
// 清理代码块标记（移除 ```go 或 ```）
|
||||
code = strings.TrimPrefix(code, "```go")
|
||||
code = strings.TrimPrefix(code, "```")
|
||||
code = strings.TrimSuffix(code, "```")
|
||||
code = strings.TrimSpace(code)
|
||||
|
||||
// 智能识别文件类型
|
||||
fileName = smartFileName(fileName, code)
|
||||
|
||||
// 规范化路径
|
||||
fullPath := normalizePath(fileName)
|
||||
|
||||
// 去重
|
||||
if seen[fullPath] {
|
||||
continue
|
||||
}
|
||||
seen[fullPath] = true
|
||||
|
||||
files = append(files, File{
|
||||
Path: fullPath,
|
||||
Content: code,
|
||||
})
|
||||
}
|
||||
|
||||
if len(files) == 0 {
|
||||
return nil, fmt.Errorf("未找到任何代码块，请确保响应格式为：// File: 文件名\n```go\n代码\n```")
|
||||
}
|
||||
|
||||
return files, nil
|
||||
}
|
||||
|
||||
// smartFileName 智能识别文件名
|
||||
func smartFileName(fileName, code string) string {
|
||||
fileName = strings.TrimSpace(fileName)
|
||||
fileName = strings.TrimLeft(fileName, "0123456789.- ")
|
||||
|
||||
// 如果文件名是 code.go 或类似通用名，根据内容推断
|
||||
base := filepath.Base(fileName)
|
||||
if base == "code.go" || base == "code" || base == "main.go" {
|
||||
// 根据内容推断
|
||||
if strings.Contains(code, "package sdk_test") {
|
||||
return "example_test.go"
|
||||
}
|
||||
if strings.Contains(code, "module ") {
|
||||
return "go.mod"
|
||||
}
|
||||
if strings.Contains(code, "package sdk") && strings.Contains(code, "func (c *Client)") {
|
||||
// 检查是哪个 API 文件
|
||||
if strings.Contains(code, "CreateKeyOrder") || strings.Contains(code, "OrderKey") {
|
||||
return "api_key.go"
|
||||
}
|
||||
if strings.Contains(code, "BatchOrder") || strings.Contains(code, "BatchQuery") {
|
||||
return "api_batch.go"
|
||||
}
|
||||
if strings.Contains(code, "QueryKey") || strings.Contains(code, "DiscardKey") {
|
||||
return "api_key.go"
|
||||
}
|
||||
}
|
||||
if strings.Contains(code, "package sdk") && strings.Contains(code, "type Client struct") {
|
||||
return "client.go"
|
||||
}
|
||||
if strings.Contains(code, "package sdk") && (strings.Contains(code, "type APIError") || strings.Contains(code, "CodeSuccess")) {
|
||||
return "errors.go"
|
||||
}
|
||||
if strings.Contains(code, "package sdk") && (strings.Contains(code, "ParseRSAPrivateKey") || strings.Contains(code, "AESEncryptECB")) {
|
||||
return "crypto.go"
|
||||
}
|
||||
if strings.Contains(code, "package sdk") && (strings.Contains(code, "type RequestEnvelope") || strings.Contains(code, "type KeyOrderInfo")) {
|
||||
return "types.go"
|
||||
}
|
||||
}
|
||||
|
||||
return fileName
|
||||
}
|
||||
|
||||
func normalizePath(path string) string {
|
||||
path = strings.TrimSpace(path)
|
||||
path = strings.TrimLeft(path, "0123456789.- ")
|
||||
|
||||
// 确保文件扩展名正确
|
||||
if !strings.HasSuffix(path, ".go") && !strings.HasSuffix(path, ".mod") {
|
||||
// 如果是 go.mod 或 go.work，保持原样
|
||||
if path != "go.mod" && path != "go.work" {
|
||||
path = path + ".go"
|
||||
}
|
||||
}
|
||||
|
||||
// 确保路径以 sdk/ 开头
|
||||
//if !strings.HasPrefix(path, "sdk/") && !strings.HasPrefix(path, "output/") {
|
||||
// path = filepath.Join("sdk", path)
|
||||
//}
|
||||
return path
|
||||
}
|
||||
|
||||
// WriteFiles 将文件写入本地
|
||||
func WriteFiles(files []File, outputDir string) error {
|
||||
for _, f := range files {
|
||||
fullPath := filepath.Join(outputDir, f.Path)
|
||||
|
||||
if err := os.MkdirAll(filepath.Dir(fullPath), 0755); err != nil {
|
||||
return fmt.Errorf("创建目录失败: %w", err)
|
||||
}
|
||||
|
||||
if err := os.WriteFile(fullPath, []byte(f.Content), 0644); err != nil {
|
||||
return fmt.Errorf("写入文件失败: %w", fullPath, err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -0,0 +1,102 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ConvertResponse 定义接口响应结构
|
||||
type ConvertResponse struct {
|
||||
Filename string `json:"filename"`
|
||||
LlmUsed bool `json:"llm_used"`
|
||||
Markdown string `json:"markdown"`
|
||||
Size int `json:"size"`
|
||||
Success bool `json:"success"`
|
||||
Title *string `json:"title"`
|
||||
}
|
||||
|
||||
// ConvertRequest 定义请求参数结构
|
||||
type ConvertRequest struct {
|
||||
File multipart.File
|
||||
FileName string
|
||||
LlmModel string
|
||||
LlmApiKey string
|
||||
LlmBaseUrl string
|
||||
LlmPrompt string
|
||||
}
|
||||
|
||||
const url = "http://121.199.38.107:5001/convert"
|
||||
|
||||
// ConvertFile 调用 /convert 接口识别本地文件内容
|
||||
func ConvertFile(req ConvertRequest) (*ConvertResponse, error) {
|
||||
defer req.File.Close()
|
||||
// 创建 multipart 表单
|
||||
body := &bytes.Buffer{}
|
||||
writer := multipart.NewWriter(body)
|
||||
// 添加文件字段
|
||||
part, err := writer.CreateFormFile("file", req.FileName)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建文件表单字段失败: %w", err)
|
||||
}
|
||||
if _, err := io.Copy(part, req.File); err != nil {
|
||||
return nil, fmt.Errorf("复制文件内容失败: %w", err)
|
||||
}
|
||||
// 添加可选参数（如果提供）
|
||||
if req.LlmModel != "" {
|
||||
_ = writer.WriteField("llm_model", req.LlmModel)
|
||||
}
|
||||
if req.LlmApiKey != "" {
|
||||
_ = writer.WriteField("llm_api_key", req.LlmApiKey)
|
||||
}
|
||||
if req.LlmBaseUrl != "" {
|
||||
_ = writer.WriteField("llm_base_url", req.LlmBaseUrl)
|
||||
}
|
||||
if req.LlmPrompt != "" {
|
||||
_ = writer.WriteField("llm_prompt", req.LlmPrompt)
|
||||
}
|
||||
|
||||
// 关闭 multipart writer
|
||||
if err = writer.Close(); err != nil {
|
||||
return nil, fmt.Errorf("关闭表单写入器失败: %w", err)
|
||||
}
|
||||
|
||||
// 创建 HTTP 请求
|
||||
|
||||
httpReq, err := http.NewRequest("POST", url, body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建HTTP请求失败: %w", err)
|
||||
}
|
||||
httpReq.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
// 发送请求
|
||||
client := &http.Client{Timeout: 60 * time.Second}
|
||||
resp, err := client.Do(httpReq)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("发送请求失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// 读取响应
|
||||
respBody, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取响应失败: %w", err)
|
||||
}
|
||||
|
||||
// 检查 HTTP 状态码
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, fmt.Errorf("请求返回非200状态码: %d, 响应: %s", resp.StatusCode, string(respBody))
|
||||
}
|
||||
|
||||
// 解析响应 JSON
|
||||
var result ConvertResponse
|
||||
if err := json.Unmarshal(respBody, &result); err != nil {
|
||||
return nil, fmt.Errorf("解析响应JSON失败: %w, 原始响应: %s", err, string(respBody))
|
||||
}
|
||||
|
||||
return &result, nil
|
||||
}
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"github.com/gofiber/fiber/v2"
|
||||
)
|
||||
|
||||
type PageHandler struct{}
|
||||
|
||||
func NewPageHandler() *PageHandler {
|
||||
return &PageHandler{}
|
||||
}
|
||||
|
||||
// Index 首页
|
||||
func (h *PageHandler) Index(c *fiber.Ctx) error {
|
||||
return c.SendFile("./web/templates/index.html")
|
||||
}
|
||||
|
|
@ -0,0 +1,295 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sdk-generator/internal/call"
|
||||
"sdk-generator/internal/models"
|
||||
"sdk-generator/internal/service"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
)
|
||||
|
||||
type SDKHandler struct {
|
||||
service *service.SDKGeneratorService
|
||||
}
|
||||
|
||||
func NewSDKHandler(service *service.SDKGeneratorService) *SDKHandler {
|
||||
return &SDKHandler{service: service}
|
||||
}
|
||||
|
||||
// GenerateSDK 上传文档并生成 SDK
|
||||
// POST /api/v1/generate
|
||||
func (h *SDKHandler) GenerateSDK(c *fiber.Ctx) error {
|
||||
//获取上传文件
|
||||
file, err := c.FormFile("file")
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "请上传文件 (field: file)",
|
||||
})
|
||||
}
|
||||
|
||||
// 检查文件大小
|
||||
if file.Size > 10*1024*1024 {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "文件大小超过 10MB 限制",
|
||||
})
|
||||
}
|
||||
var content []byte
|
||||
// 检查文件类型
|
||||
filename := file.Filename
|
||||
ext := strings.ToLower(filepath.Ext(filename))
|
||||
src, err := file.Open()
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": "打开文件失败",
|
||||
})
|
||||
}
|
||||
llmSet := call.LlmCallSet{
|
||||
ApiKey: c.FormValue("llm_api_key", ""),
|
||||
BaseUrl: c.FormValue("llm_base_url", ""),
|
||||
ModelName: c.FormValue("llm_model", ""),
|
||||
}
|
||||
if llmSet.ModelName == "" || llmSet.ApiKey == "" || llmSet.BaseUrl == "" {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "缺少 llm_model 或 llm_api_key 或 llm_base_url",
|
||||
})
|
||||
}
|
||||
defer src.Close()
|
||||
switch ext {
|
||||
case ".md", ".txt", ".markdown":
|
||||
// 读取文件内容
|
||||
content, err = io.ReadAll(src)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": "读取文件失败",
|
||||
})
|
||||
}
|
||||
break
|
||||
case ".docx", ".doc", ".pdf":
|
||||
res, err := ConvertFile(ConvertRequest{
|
||||
File: src,
|
||||
FileName: filename,
|
||||
LlmModel: llmSet.ModelName,
|
||||
LlmApiKey: llmSet.ApiKey,
|
||||
LlmBaseUrl: llmSet.BaseUrl,
|
||||
//LlmPrompt: c.FormValue("llm_prompt", ""),
|
||||
})
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": "读取文件失败" + err.Error(),
|
||||
})
|
||||
}
|
||||
content = []byte(res.Markdown)
|
||||
break
|
||||
default:
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "不支持的文件类型"})
|
||||
}
|
||||
|
||||
sdkName := c.FormValue("sdk_name", time.Now().Format(time.RFC3339))
|
||||
// 调用服务生成
|
||||
taskID, err := h.service.Generate(c.Context(), content, sdkName, &llmSet)
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.Status(fiber.StatusAccepted).JSON(models.GenerateResponse{
|
||||
TaskID: taskID,
|
||||
Status: string(models.StatusPending),
|
||||
Message: "SDK 生成任务已提交，请通过 /api/v1/tasks/:task_id 查询状态",
|
||||
})
|
||||
}
|
||||
|
||||
// GetTaskStatus 查询任务状态
|
||||
// GET /api/v1/tasks/:task_id
|
||||
func (h *SDKHandler) GetTaskStatus(c *fiber.Ctx) error {
|
||||
taskID := c.Params("task_id")
|
||||
if taskID == "" {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "缺少 task_id",
|
||||
})
|
||||
}
|
||||
|
||||
task, ok := h.service.GetTask(taskID)
|
||||
if !ok {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": "任务不存在",
|
||||
})
|
||||
}
|
||||
task.StatusName = task.Status.Desc()
|
||||
task.Percent = task.Status.Percent()
|
||||
response := models.TaskResponse{
|
||||
Task: *task,
|
||||
}
|
||||
|
||||
// 如果任务完成，列出生成的文件
|
||||
if task.Status == models.StatusCompleted {
|
||||
files, _ := listFiles(task.OutputDir)
|
||||
response.Files = files
|
||||
}
|
||||
|
||||
return c.JSON(response)
|
||||
}
|
||||
|
||||
// DownloadSDK 下载生成的 SDK
|
||||
// GET /api/v1/tasks/:task_id/download
|
||||
func (h *SDKHandler) DownloadSDK(c *fiber.Ctx) error {
|
||||
taskID := c.Params("task_id")
|
||||
if taskID == "" {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "缺少 task_id",
|
||||
})
|
||||
}
|
||||
|
||||
task, ok := h.service.GetTask(taskID)
|
||||
if !ok {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": "任务不存在",
|
||||
})
|
||||
}
|
||||
|
||||
if task.Status != models.StatusCompleted {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": fmt.Sprintf("任务未完成，当前状态: %s", task.Status),
|
||||
})
|
||||
}
|
||||
|
||||
// 检查输出目录是否存在
|
||||
if _, err := os.Stat(task.OutputDir); os.IsNotExist(err) {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": "SDK 文件不存在",
|
||||
})
|
||||
}
|
||||
|
||||
// 创建 ZIP 文件并流式返回
|
||||
zipFilename := fmt.Sprintf("sdk_%s.zip", taskID)
|
||||
c.Set("Content-Type", "application/zip")
|
||||
c.Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", zipFilename))
|
||||
|
||||
// 使用 Fiber 的 Response 直接写入
|
||||
zipWriter := zip.NewWriter(c.Response().BodyWriter())
|
||||
defer zipWriter.Close()
|
||||
|
||||
// 遍历目录添加文件到 ZIP
|
||||
err := filepath.Walk(task.OutputDir, func(path string, info os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if info.IsDir() {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 计算相对路径
|
||||
relPath, err := filepath.Rel(task.OutputDir, path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 创建 ZIP 条目
|
||||
zipFile, err := zipWriter.Create(relPath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 读取文件内容
|
||||
content, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = zipFile.Write(content)
|
||||
return err
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||
"error": fmt.Sprintf("打包 SDK 失败: %v", err),
|
||||
})
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ListTasks 列出所有任务
|
||||
// GET /api/v1/tasks
|
||||
func (h *SDKHandler) ListTasks(c *fiber.Ctx) error {
|
||||
tasks := h.service.ListTasks()
|
||||
|
||||
// 分页
|
||||
page, _ := strconv.Atoi(c.Query("page", "1"))
|
||||
size, _ := strconv.Atoi(c.Query("size", "20"))
|
||||
|
||||
if page < 1 {
|
||||
page = 1
|
||||
}
|
||||
if size < 1 {
|
||||
size = 20
|
||||
}
|
||||
if size > 100 {
|
||||
size = 100
|
||||
}
|
||||
|
||||
total := len(tasks)
|
||||
start := (page - 1) * size
|
||||
end := start + size
|
||||
|
||||
if start > total {
|
||||
start = total
|
||||
}
|
||||
if end > total {
|
||||
end = total
|
||||
}
|
||||
|
||||
return c.JSON(models.ListTasksResponse{
|
||||
Data: tasks[start:end],
|
||||
Total: total,
|
||||
Page: page,
|
||||
Size: size,
|
||||
})
|
||||
}
|
||||
|
||||
// DeleteTask 删除任务
|
||||
// DELETE /api/v1/tasks/:task_id
|
||||
func (h *SDKHandler) DeleteTask(c *fiber.Ctx) error {
|
||||
taskID := c.Params("task_id")
|
||||
if taskID == "" {
|
||||
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
|
||||
"error": "缺少 task_id",
|
||||
})
|
||||
}
|
||||
|
||||
if err := h.service.DeleteTask(taskID); err != nil {
|
||||
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
|
||||
return c.JSON(fiber.Map{
|
||||
"message": "任务已删除",
|
||||
})
|
||||
}
|
||||
|
||||
// 辅助函数：列出目录下的文件
|
||||
func listFiles(dir string) ([]string, error) {
|
||||
var files []string
|
||||
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !info.IsDir() {
|
||||
relPath, _ := filepath.Rel(dir, path)
|
||||
files = append(files, relPath)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
return files, err
|
||||
}
|
||||
|
|
@ -0,0 +1,104 @@
|
|||
package models
|
||||
|
||||
import (
|
||||
"sdk-generator/internal/call"
|
||||
"sdk-generator/internal/extractor"
|
||||
"time"
|
||||
)
|
||||
|
||||
type TaskStatus string
|
||||
|
||||
const (
|
||||
StatusPending TaskStatus = "pending"
|
||||
StatusProcessing TaskStatus = "processing"
|
||||
StatusMkdir TaskStatus = "mkdir"
|
||||
StatusAnaMd TaskStatus = "anaMd"
|
||||
StatusGenerateCode TaskStatus = "generateCode"
|
||||
StatusCreatFile TaskStatus = "creatFile"
|
||||
StatusValid TaskStatus = "valid"
|
||||
StatusProcess TaskStatus = "process"
|
||||
StatusFix TaskStatus = "fix"
|
||||
StatusSendToGit TaskStatus = "sendToGit "
|
||||
StatusMsgSend TaskStatus = "msgSend"
|
||||
StatusCompleted TaskStatus = "completed"
|
||||
StatusFailed TaskStatus = "failed"
|
||||
)
|
||||
|
||||
func (s TaskStatus) Desc() string {
|
||||
return StatusName[s]
|
||||
}
|
||||
|
||||
func (s TaskStatus) Percent() int {
|
||||
return StatusPercent[s]
|
||||
}
|
||||
|
||||
var StatusPercent = map[TaskStatus]int{
|
||||
StatusPending: 0,
|
||||
StatusProcessing: 1,
|
||||
StatusMkdir: 10,
|
||||
StatusAnaMd: 20,
|
||||
StatusGenerateCode: 60,
|
||||
StatusCreatFile: 50,
|
||||
StatusValid: 70,
|
||||
StatusProcess: 62,
|
||||
StatusFix: 80,
|
||||
StatusSendToGit: 85,
|
||||
StatusMsgSend: 95,
|
||||
StatusCompleted: 100,
|
||||
StatusFailed: 100,
|
||||
}
|
||||
|
||||
var StatusName = map[TaskStatus]string{
|
||||
StatusPending: "准备中",
|
||||
StatusProcessing: "开始创建",
|
||||
StatusMkdir: "新建项目文件夹",
|
||||
StatusAnaMd: "分析文档",
|
||||
StatusGenerateCode: "生成代码",
|
||||
StatusCreatFile: "创建代码目录",
|
||||
StatusValid: "验证功能",
|
||||
StatusProcess: "编译检测",
|
||||
StatusFix: "问题修复",
|
||||
StatusSendToGit: "上传代码仓库",
|
||||
StatusMsgSend: "发送钉钉消息",
|
||||
StatusCompleted: "任务完成",
|
||||
StatusFailed: "任务失败",
|
||||
}
|
||||
|
||||
type Task struct {
|
||||
ID string `json:"id"`
|
||||
Status TaskStatus `json:"status"`
|
||||
StatusName string `json:"status_name"`
|
||||
Percent int `json:"percent"`
|
||||
SdkName string `json:"sdk_name"`
|
||||
OutputDir string `json:"output_dir"`
|
||||
Error string `json:"error,omitempty"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
||||
CallLLM *call.CallLLM
|
||||
SdkPackage string
|
||||
MarkdownContent string
|
||||
Refine string
|
||||
Resp string
|
||||
Valid string
|
||||
Files []extractor.File
|
||||
RepoURL string
|
||||
}
|
||||
|
||||
type GenerateResponse struct {
|
||||
TaskID string `json:"task_id"`
|
||||
Status string `json:"status"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
type TaskResponse struct {
|
||||
Task
|
||||
Files []string `json:"files,omitempty"`
|
||||
}
|
||||
|
||||
type ListTasksResponse struct {
|
||||
Data []*Task `json:"data"`
|
||||
Total int `json:"total"`
|
||||
Page int `json:"page"`
|
||||
Size int `json:"size"`
|
||||
}
|
||||
|
|
@ -0,0 +1,46 @@
|
|||
package msg
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
var MesSendUrl = "https://oapi.dingtalk.com/robot/send?access_token=d8367c13499dfb164b1e02a4c024024f6857a892e7406265be08d5b5e06fc590"
|
||||
|
||||
func SendDingTalkAlert(ctx context.Context, sdkName, url string) {
|
||||
client := &http.Client{
|
||||
Timeout: 10 * time.Second,
|
||||
}
|
||||
|
||||
// ✅ 正确的 link 消息格式
|
||||
message := map[string]interface{}{
|
||||
"msgtype": "link",
|
||||
"link": map[string]interface{}{
|
||||
"title": "[ai_sdk]🎉 " + sdkName + " 代码生成完成，点此进入仓库",
|
||||
"text": url + ".git",
|
||||
"messageUrl": url,
|
||||
"picUrl": "", // 可选：图片 URL
|
||||
},
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(message)
|
||||
if err != nil {
|
||||
log.Printf("JSON序列化失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := client.Post(MesSendUrl, "application/json", bytes.NewBuffer(jsonData))
|
||||
if err != nil {
|
||||
log.Printf("发送钉钉告警失败: %v", err)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
data, _ := io.ReadAll(resp.Body)
|
||||
log.Printf("钉钉告警发送结果，状态码: %s,内容:%s", resp.Status, data)
|
||||
return
|
||||
}
|
||||
|
|
@ -0,0 +1,17 @@
|
|||
package msg
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestSendDingTalkAlert(t *testing.T) {
|
||||
// 测试正常发送
|
||||
ctx := context.Background()
|
||||
sdkName := "ymt_v3_generate"
|
||||
url := "https://gitea.cdlsxd.cn/ai_sdk/ymt_v3_generate-20260721-095556"
|
||||
|
||||
SendDingTalkAlert(ctx, sdkName, url)
|
||||
|
||||
t.Log("所有测试完成")
|
||||
}
|
||||
|
|
@ -0,0 +1,56 @@
|
|||
package postprocess
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func Process(ctx context.Context, sdkPackage string) error {
|
||||
// 确保 outputDir 存在
|
||||
if err := os.MkdirAll(sdkPackage, 0755); err != nil {
|
||||
return fmt.Errorf("创建输出目录失败: %w", err)
|
||||
}
|
||||
|
||||
// 检查是否有 go.mod，没有则初始化
|
||||
goModPath := filepath.Join(sdkPackage, "go.mod")
|
||||
if _, err := os.Stat(goModPath); os.IsNotExist(err) {
|
||||
cmd := exec.Command("go", "mod", "init", sdkPackage)
|
||||
cmd.Dir = sdkPackage
|
||||
if err := cmd.Run(); err != nil {
|
||||
// 忽略错误，可能已经有 go.mod 了
|
||||
}
|
||||
}
|
||||
|
||||
commands := []struct {
|
||||
name string
|
||||
args []string
|
||||
desc string
|
||||
}{
|
||||
{"go", []string{"fmt", "./..."}, "格式化代码"},
|
||||
{"go", []string{"mod", "tidy"}, "整理依赖"},
|
||||
{"go", []string{"build", "./..."}, "编译检查"},
|
||||
}
|
||||
|
||||
var errors []string
|
||||
for _, cmdInfo := range commands {
|
||||
c := exec.Command(cmdInfo.name, cmdInfo.args...)
|
||||
c.Dir = sdkPackage
|
||||
|
||||
var stderr bytes.Buffer
|
||||
c.Stderr = &stderr
|
||||
|
||||
if err := c.Run(); err != nil {
|
||||
errors = append(errors, fmt.Sprintf("%s: %v\n%s", cmdInfo.desc, err, stderr.String()))
|
||||
}
|
||||
}
|
||||
|
||||
if len(errors) > 0 {
|
||||
return fmt.Errorf("\n%s", strings.Join(errors, "\n"))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -0,0 +1,58 @@
|
|||
// internal/prompts/fix_prompt.go
|
||||
package prompts
|
||||
|
||||
import "github.com/sashabaranov/go-openai"
|
||||
|
||||
func FixPrompt(codeContent, errorMsg, sdkName, refineDoc string) openai.ChatCompletionRequest {
|
||||
return openai.ChatCompletionRequest{
|
||||
Messages: []openai.ChatCompletionMessage{
|
||||
{
|
||||
Role: openai.ChatMessageRoleSystem,
|
||||
Content: "你是一个资深的 Go 语言工程师。生成的 SDK 代码在编译/格式化时出现错误，请修复。",
|
||||
},
|
||||
{
|
||||
Role: openai.ChatMessageRoleUser,
|
||||
Content: GetFixPrompt(codeContent, errorMsg, sdkName, refineDoc),
|
||||
},
|
||||
},
|
||||
Temperature: 0.1, // 更低，保证确定性输出
|
||||
TopP: 0.9, // 核采样，平衡多样性和质量
|
||||
MaxTokens: 65536, // 加大到 64k，确保足够
|
||||
FrequencyPenalty: 0.0, // 代码生成不需要
|
||||
PresencePenalty: 0.0, // 代码生成不需要
|
||||
Stop: nil, // 不设置，让模型完整输出
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// GetFixPrompt 构建修复代码的提示词
|
||||
func GetFixPrompt(codeContent, errorMsg, refineDoc, sdkName string) string {
|
||||
return `
|
||||
## 错误信息
|
||||
` + errorMsg + `
|
||||
|
||||
## 文档（供参考）
|
||||
` + refineDoc + `
|
||||
|
||||
## 当前代码
|
||||
` + codeContent + `
|
||||
|
||||
## ⚠️ 重要：输出格式要求（必须严格遵守）
|
||||
每个文件必须以以下格式输出：
|
||||
|
||||
// File: ` + sdkName + `/文件名.go
|
||||
` + "```go" + `
|
||||
package ` + sdkName + `
|
||||
// ... 代码内容 ...
|
||||
` + "```" + `
|
||||
|
||||
|
||||
## 修复要求
|
||||
1. 根据错误信息修复代码
|
||||
2. 确保代码符合 Go 语法规范
|
||||
3. 不要改变接口定义和业务逻辑
|
||||
4. 保持原有代码风格
|
||||
5. 请直接输出修复后的完整代码，不要有任何额外文字
|
||||
|
||||
请返回修复后的完整代码。`
|
||||
}
|
||||
|
|
@ -0,0 +1,145 @@
|
|||
package prompts
|
||||
|
||||
import "github.com/sashabaranov/go-openai"
|
||||
|
||||
func SDKGeneratorPrompt(prompt string) openai.ChatCompletionRequest {
|
||||
return openai.ChatCompletionRequest{
|
||||
Messages: []openai.ChatCompletionMessage{
|
||||
{
|
||||
Role: openai.ChatMessageRoleSystem,
|
||||
Content: "你是一个资深的 Go 语言工程师，擅长生成高质量、可直接编译的 SDK 代码。",
|
||||
},
|
||||
{
|
||||
Role: openai.ChatMessageRoleUser,
|
||||
Content: prompt,
|
||||
},
|
||||
},
|
||||
Temperature: 0.1, // 更低，保证确定性输出
|
||||
TopP: 0.9, // 核采样，平衡多样性和质量
|
||||
MaxTokens: 65536, // 加大到 64k，确保足够
|
||||
FrequencyPenalty: 0.0, // 代码生成不需要
|
||||
PresencePenalty: 0.0, // 代码生成不需要
|
||||
Stop: nil, // 不设置，让模型完整输出
|
||||
}
|
||||
}
|
||||
|
||||
const GenSDKPrompt = `你是一个资深的 Go 语言工程师，擅长生成高质量的 SDK 代码。
|
||||
|
||||
## 任务
|
||||
根据用户提供的 API 接口文档（Markdown 格式），生成一个完整的 Go SDK 代码工程。
|
||||
|
||||
## ⚠️ 重要：输出格式要求（必须严格遵守）
|
||||
每个文件必须以以下格式输出：
|
||||
|
||||
// File: {{sdk_name}}/文件名.go
|
||||
` + "```go" + `
|
||||
package {{sdk_name}}
|
||||
// ... 代码内容 ...
|
||||
` + "```" + `
|
||||
|
||||
## ⚠️ 重要：导入路径规范（必须遵守）
|
||||
生成的 SDK 代码中，**示例代码（example_test.go）必须使用本地导入路径**，而不是远程 GitHub 路径：
|
||||
|
||||
✅ 正确示例：
|
||||
` + "```go" + `
|
||||
package {{sdk_name}}
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"net/http"
|
||||
)
|
||||
` + "```" + `
|
||||
|
||||
❌ 错误示例（禁止使用）：
|
||||
` + "```go" + `
|
||||
import sdk "github.com/bluebrothers/marketing-sdk-go" // 禁止！这会拉取远程包
|
||||
` + "```" + `
|
||||
|
||||
## 文件结构
|
||||
{{sdk_name}}/
|
||||
├── go.mod # 模块名为 generated-sdk
|
||||
├── client.go
|
||||
├── types.go
|
||||
├── crypto.go
|
||||
├── errors.go
|
||||
├── api_key.go
|
||||
└── example_test.go # 使用本地导入路径
|
||||
|
||||
## 代码规范
|
||||
1. 遵循 Effective Go 规范
|
||||
2. 所有导出类型/方法必须有 godoc 注释
|
||||
3. context.Context 作为方法的第一个参数
|
||||
4. 错误处理统一返回 error
|
||||
|
||||
## 客户端设计
|
||||
1. 提供 NewClient 构造函数，支持 Option 模式
|
||||
2. 支持 WithBaseURL、WithTimeout、WithHTTPClient 等配置
|
||||
3. 根据文档中描述的认证方式实现认证逻辑
|
||||
|
||||
## ⚠️ 加密/签名实现规范（必须遵守）
|
||||
|
||||
**核心原则：使用成熟的第三方库，不要自己实现加解密算法。**
|
||||
|
||||
### 推荐使用的第三方库
|
||||
|
||||
**AES 加密：**
|
||||
` + "```go" + `
|
||||
import "crypto/aes" // Go 标准库
|
||||
import "crypto/cipher" // Go 标准库
|
||||
// 或使用更高级的封装
|
||||
import "github.com/tink-crypto/tink-go/v2/aead"
|
||||
` + "```" + `
|
||||
|
||||
**RSA 加密/签名：**
|
||||
` + "```go" + `
|
||||
import "crypto/rsa" // Go 标准库
|
||||
import "crypto/rand" // Go 标准库
|
||||
` + "```" + `
|
||||
|
||||
**HMAC 签名：**
|
||||
` + "```go" + `
|
||||
import "crypto/hmac"
|
||||
import "crypto/sha256"
|
||||
` + "```" + `
|
||||
|
||||
**SM2/SM3/SM4（国密）：**
|
||||
` + "```go" + `
|
||||
import "github.com/tjfoc/gmsm/sm2"
|
||||
import "github.com/tjfoc/gmsm/sm3"
|
||||
import "github.com/tjfoc/gmsm/sm4"
|
||||
` + "```" + `
|
||||
|
||||
**JWT：**
|
||||
` + "```go" + `
|
||||
import "github.com/golang-jwt/jwt/v5"
|
||||
` + "```" + `
|
||||
|
||||
### 禁止的行为
|
||||
|
||||
❌ **不要自己实现**：
|
||||
- 不要自己写 AES 加密函数（除非调用标准库）
|
||||
- 不要自己写 RSA 签名函数（除非调用标准库）
|
||||
- 不要自己实现 PKCS7 填充（使用标准库或成熟库）
|
||||
- 不要自己实现任何加解密算法
|
||||
|
||||
✅ **正确做法**：
|
||||
- 调用 Go 标准库 "crypto/*" 包
|
||||
- 使用成熟的开源库（如 gmsm、jwt 等）
|
||||
- 如果文档要求特殊算法，先查找是否有现成的 Go 实现
|
||||
- 只在标准库和成熟库都不支持时才考虑自己实现
|
||||
|
||||
## ⚠️ 单元测试规范（必须严格遵守）
|
||||
1. 所有单元测试必须写在 example_test.go 文件中
|
||||
2. 必须使用标准库 testing 包
|
||||
3. **每个 API 接口必须对应一个独立的单元测试函数，格式必须为：**
|
||||
|
||||
## 关键要求
|
||||
1. 生成的代码必须可直接编译运行
|
||||
2. go的版本为1.21
|
||||
3. **go.mod 的 module 名使用 {{sdk_name}}**
|
||||
4. 不要引用任何不存在的远程 GitHub 仓库
|
||||
|
||||
## 接口文档
|
||||
%s`
|
||||
|
|
@ -0,0 +1,70 @@
|
|||
package prompts
|
||||
|
||||
import "github.com/sashabaranov/go-openai"
|
||||
|
||||
// BuildRefinePrompt 构建文档精炼的请求
|
||||
func BuildRefinePrompt(rawDoc string) openai.ChatCompletionRequest {
|
||||
return openai.ChatCompletionRequest{
|
||||
Messages: []openai.ChatCompletionMessage{
|
||||
{
|
||||
Role: openai.ChatMessageRoleSystem,
|
||||
Content: `你是一个专业的 API 文档分析专家，擅长从各种格式的文档中提取技术信息。你的输出必须准确、完整，因为后续会基于你的输出生成可运行的代码。`,
|
||||
},
|
||||
{
|
||||
Role: openai.ChatMessageRoleUser,
|
||||
Content: RefinePrompt(rawDoc),
|
||||
},
|
||||
},
|
||||
Temperature: 0.1,
|
||||
TopP: 0.9,
|
||||
MaxTokens: 65536,
|
||||
FrequencyPenalty: 0.0,
|
||||
PresencePenalty: 0.0,
|
||||
}
|
||||
}
|
||||
|
||||
// RefinePrompt 文档精炼提示词模板
|
||||
func RefinePrompt(rawDoc string) string {
|
||||
return `分析以下 API 文档，为后续代码生成做准备。
|
||||
|
||||
## 你的任务
|
||||
|
||||
1. 判断文档类型：是"调用文档"（生成客户端 SDK）还是"对接文档"（生成服务端骨架）
|
||||
2. 完整提取所有接口信息（路径、方法、参数、响应）
|
||||
3. 完整提取认证/加密/签名细节
|
||||
4. 去除营销文案、流程说明等噪音，但保留所有技术信息
|
||||
|
||||
## 必须保留的内容
|
||||
|
||||
- 所有接口（一个不漏）
|
||||
- 所有参数和字段（包括类型、必填、描述）
|
||||
- 所有错误码
|
||||
- 认证、加密、签名的完整流程
|
||||
- 所有示例
|
||||
|
||||
## 可以去掉的内容
|
||||
|
||||
- 公司介绍、平台介绍
|
||||
- 接入流程步骤
|
||||
- 联调建议
|
||||
- 重复内容
|
||||
|
||||
## 输出格式
|
||||
|
||||
` + "```" + `
|
||||
文档类型：[调用文档/对接文档/混合型]
|
||||
|
||||
[接口列表...保持完整]
|
||||
[认证加密签名...保持完整]
|
||||
[错误码...保持完整]
|
||||
` + "```" + `
|
||||
|
||||
## 关键原则
|
||||
|
||||
1. **完整性优先**：不确定是否重要的信息，保留
|
||||
2. **忠于原文**：类型、字段名、示例值保持原样，不要修改
|
||||
3. **不编造**：文档中没有的信息，不要补充
|
||||
|
||||
原始文档：
|
||||
` + rawDoc
|
||||
}
|
||||
|
|
@ -0,0 +1,73 @@
|
|||
// internal/prompts/validate_prompt.go
|
||||
package prompts
|
||||
|
||||
import (
|
||||
"github.com/sashabaranov/go-openai"
|
||||
)
|
||||
|
||||
// GetValidatePrompt 验证代码完整性，直接返回修复后的代码
|
||||
func GetValidatePrompt(refineDoc, sdkName, codeContent string) openai.ChatCompletionRequest {
|
||||
return openai.ChatCompletionRequest{
|
||||
Messages: []openai.ChatCompletionMessage{
|
||||
{
|
||||
Role: openai.ChatMessageRoleSystem,
|
||||
Content: "你是代码审查专家。检查生成的 SDK 代码是否完整实现了文档中的所有内容。",
|
||||
},
|
||||
{
|
||||
Role: openai.ChatMessageRoleUser,
|
||||
Content: ValidPrompt(refineDoc, sdkName, codeContent),
|
||||
},
|
||||
},
|
||||
Temperature: 0.1, // 更低，保证确定性输出
|
||||
TopP: 0.9, // 核采样，平衡多样性和质量
|
||||
MaxTokens: 65536, // 加大到 64k，确保足够
|
||||
FrequencyPenalty: 0.0, // 代码生成不需要
|
||||
PresencePenalty: 0.0, // 代码生成不需要
|
||||
Stop: nil, // 不设置，让模型完整输出
|
||||
}
|
||||
}
|
||||
|
||||
func ValidPrompt(codeContent, sdkName, refineDoc string) string {
|
||||
|
||||
return `你是代码审查专家。检查生成的 SDK 代码是否完整实现了文档中的所有内容。
|
||||
|
||||
## 精炼文档
|
||||
` + refineDoc + `
|
||||
|
||||
## 当前代码
|
||||
` + codeContent + `
|
||||
|
||||
## 检查清单
|
||||
|
||||
1. 文档中的所有接口是否都已实现？
|
||||
2. 每个接口的请求参数是否完整（字段名、类型、必填）？
|
||||
3. 每个接口的响应字段是否完整？
|
||||
4. 认证方式是否已实现？
|
||||
5. 加密方式是否已实现？
|
||||
6. 签名方式是否已实现？
|
||||
7. 错误码是否已定义？
|
||||
|
||||
## 输出规则（严格遵守）
|
||||
|
||||
**情况一：审查通过，没有任何问题**
|
||||
只输出两个字符：` + "`OK`" + `
|
||||
|
||||
**情况二：存在问题**
|
||||
输出修复后的**全量代码**，每个文件用以下格式：
|
||||
|
||||
// File: ` + sdkName + `/文件名.go
|
||||
` + "```go" + `
|
||||
package ` + sdkName + `
|
||||
// ... 代码内容 ...
|
||||
` + "```" + `
|
||||
|
||||
|
||||
|
||||
## 重要提醒
|
||||
|
||||
1. 如果存在问题，必须输出**所有文件**的完整代码，不能只输出修改的部分
|
||||
2. 保持原有代码风格不变
|
||||
3. 补全所有遗漏的接口、参数、加密/签名方式
|
||||
4. 不要加任何描述性注释
|
||||
`
|
||||
}
|
||||
|
|
@ -0,0 +1,348 @@
|
|||
package push
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"code.gitea.io/sdk/gitea"
|
||||
)
|
||||
|
||||
var (
|
||||
// 全局客户端实例（单例）
|
||||
defaultClient *GiteaClient
|
||||
once sync.Once
|
||||
initErr error
|
||||
)
|
||||
|
||||
// GiteaClient Gitea 客户端封装
|
||||
type GiteaClient struct {
|
||||
client *gitea.Client
|
||||
}
|
||||
|
||||
// RepoConfig 仓库配置
|
||||
type RepoConfig struct {
|
||||
Name string // 仓库名称
|
||||
Description string // 仓库描述
|
||||
Private bool // 是否私有，false 为公共仓库
|
||||
AutoInit bool // 是否自动初始化
|
||||
Gitignores string // .gitignore 模板，如 "Go", "Python"
|
||||
License string // 许可证，如 "MIT", "Apache-2.0"
|
||||
Readme string // README 模板，如 "Default"
|
||||
}
|
||||
|
||||
// FileConfig 文件配置
|
||||
type FileConfig struct {
|
||||
Path string // 文件路径，如 "main.go"
|
||||
Content string // 文件内容（原始字符串）
|
||||
Message string // 提交信息
|
||||
Branch string // 分支名，为空则使用默认分支
|
||||
}
|
||||
|
||||
// TagConfig Tag 配置
|
||||
type TagConfig struct {
|
||||
TagName string // tag 名称，如 "v1.0.0"
|
||||
Message string // tag 说明
|
||||
}
|
||||
|
||||
// GetClient 获取全局客户端实例（单例模式）
|
||||
func GetClient() (*GiteaClient, error) {
|
||||
//if defaultClient == nil {
|
||||
// once.Do(func() {
|
||||
// defaultClient, initErr = NewGiteaClient(
|
||||
// "https://gitea.cdlsxd.cn/api/v1",
|
||||
// "rzy_lsxd_2026",
|
||||
// )
|
||||
// })
|
||||
// if initErr != nil {
|
||||
// return nil, initErr
|
||||
// }
|
||||
//}
|
||||
return defaultClient, nil
|
||||
}
|
||||
|
||||
// NewGiteaClient 创建 Gitea 客户端实例
|
||||
func NewGiteaClient(url, token string) (*GiteaClient, error) {
|
||||
client, err := gitea.NewClient(url, gitea.SetToken(token))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建 Gitea 客户端失败: %v", err)
|
||||
}
|
||||
defaultClient = &GiteaClient{
|
||||
client: client,
|
||||
}
|
||||
return defaultClient, nil
|
||||
}
|
||||
|
||||
// CreateRepo 创建仓库（支持个人和组织）
|
||||
func (g *GiteaClient) CreateRepo(cfg RepoConfig, orgName ...string) (*gitea.Repository, error) {
|
||||
if g.client == nil {
|
||||
return nil, fmt.Errorf("客户端未初始化")
|
||||
}
|
||||
|
||||
opts := gitea.CreateRepoOption{
|
||||
Name: cfg.Name,
|
||||
Description: cfg.Description,
|
||||
Private: cfg.Private,
|
||||
AutoInit: cfg.AutoInit,
|
||||
Gitignores: cfg.Gitignores,
|
||||
License: cfg.License,
|
||||
//Readme: cfg.Readme,
|
||||
}
|
||||
|
||||
var repo *gitea.Repository
|
||||
var err error
|
||||
|
||||
// 如果传入了组织名，则为组织创建仓库
|
||||
if len(orgName) > 0 && orgName[0] != "" {
|
||||
repo, _, err = g.client.CreateOrgRepo(orgName[0], opts)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("为组织 %s 创建仓库失败: %v", orgName[0], err)
|
||||
}
|
||||
} else {
|
||||
repo, _, err = g.client.CreateRepo(opts)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建仓库失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
return repo, nil
|
||||
}
|
||||
|
||||
// PushFile 推送单个文件
|
||||
func (g *GiteaClient) PushFile(owner, repoName string, fileCfg FileConfig) error {
|
||||
if g.client == nil {
|
||||
return fmt.Errorf("客户端未初始化")
|
||||
}
|
||||
// Base64 编码文件内容
|
||||
contentBase64 := base64.StdEncoding.EncodeToString([]byte(fileCfg.Content))
|
||||
|
||||
opts := gitea.CreateFileOptions{
|
||||
FileOptions: gitea.FileOptions{
|
||||
Message: fileCfg.Message,
|
||||
BranchName: fileCfg.Branch,
|
||||
},
|
||||
Content: contentBase64,
|
||||
}
|
||||
|
||||
_, _, err := g.client.CreateFile(owner, repoName, fileCfg.Path, opts)
|
||||
if err != nil {
|
||||
return fmt.Errorf("推送文件 %s 失败: %v", fileCfg.Path, err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// PushFiles 批量推送多个文件
|
||||
func (g *GiteaClient) PushFiles(owner, repoName string, files []FileConfig) error {
|
||||
if g.client == nil {
|
||||
return fmt.Errorf("客户端未初始化")
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
if err := g.PushFile(owner, repoName, file); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CreateTag 创建标签
|
||||
func (g *GiteaClient) CreateTag(owner, repoName string, tagCfg TagConfig) (*gitea.Tag, error) {
|
||||
if g.client == nil {
|
||||
return nil, fmt.Errorf("客户端未初始化")
|
||||
}
|
||||
|
||||
opts := gitea.CreateTagOption{
|
||||
TagName: tagCfg.TagName,
|
||||
Message: tagCfg.Message,
|
||||
}
|
||||
|
||||
tag, _, err := g.client.CreateTag(owner, repoName, opts)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建标签 %s 失败: %v", tagCfg.TagName, err)
|
||||
}
|
||||
|
||||
return tag, nil
|
||||
}
|
||||
|
||||
// CreateRepoAndPush 创建仓库并推送代码（完整流程）
|
||||
func (g *GiteaClient) CreateRepoAndPush(cfg RepoConfig, files []FileConfig, tagCfg *TagConfig, orgName ...string) error {
|
||||
if g.client == nil {
|
||||
return fmt.Errorf("客户端未初始化")
|
||||
}
|
||||
|
||||
// 1. 创建仓库
|
||||
repo, err := g.CreateRepo(cfg, orgName...)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 等待仓库初始化完成
|
||||
if cfg.AutoInit {
|
||||
fmt.Println("⏳ 等待仓库初始化...")
|
||||
time.Sleep(2 * time.Second)
|
||||
}
|
||||
|
||||
// 2. 推送代码
|
||||
if len(files) > 0 {
|
||||
fmt.Println("📤 开始推送文件...")
|
||||
if err := g.PushFiles(repo.Owner.UserName, repo.Name, files); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 创建标签（如果需要）
|
||||
if tagCfg != nil && tagCfg.TagName != "" {
|
||||
fmt.Println("🏷️ 创建标签...")
|
||||
if _, err := g.CreateTag(repo.Owner.UserName, repo.Name, *tagCfg); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Println("🎉 全部操作完成！")
|
||||
return nil
|
||||
}
|
||||
|
||||
// ==================== 便捷函数（使用全局单例） ====================
|
||||
|
||||
// CreateRepoQuick 使用全局客户端快速创建仓库
|
||||
func CreateRepoQuick(cfg RepoConfig, orgName ...string) (*gitea.Repository, error) {
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return client.CreateRepo(cfg, orgName...)
|
||||
}
|
||||
|
||||
// PushFileQuick 使用全局客户端快速推送文件
|
||||
func PushFileQuick(owner, repoName string, fileCfg FileConfig) error {
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return client.PushFile(owner, repoName, fileCfg)
|
||||
}
|
||||
|
||||
// PushFilesQuick 使用全局客户端快速批量推送文件
|
||||
func PushFilesQuick(owner, repoName string, files []FileConfig) error {
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return client.PushFiles(owner, repoName, files)
|
||||
}
|
||||
|
||||
// CreateTagQuick 使用全局客户端快速创建标签
|
||||
func CreateTagQuick(owner, repoName string, tagCfg TagConfig) (*gitea.Tag, error) {
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return client.CreateTag(owner, repoName, tagCfg)
|
||||
}
|
||||
|
||||
// PushQuick 使用全局客户端快速完整推送
|
||||
func PushQuick(repoName, description string, files []FileConfig, tagCfg *TagConfig, orgName ...string) error {
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cfg := RepoConfig{
|
||||
Name: repoName,
|
||||
Description: description,
|
||||
Private: false,
|
||||
AutoInit: true,
|
||||
Gitignores: "Go",
|
||||
License: "MIT",
|
||||
Readme: "Default",
|
||||
}
|
||||
|
||||
return client.CreateRepoAndPush(cfg, files, tagCfg, orgName...)
|
||||
}
|
||||
|
||||
// ==================== 示例 ====================
|
||||
|
||||
// ExampleUsage 使用示例
|
||||
func ExampleUsage() {
|
||||
// 准备文件
|
||||
files := []FileConfig{
|
||||
{
|
||||
Path: "main.go",
|
||||
Content: `package main\n\nimport "fmt"\n\nfunc main() {\n\tfmt.Println("Hello, Gitea!")\n}\n`,
|
||||
Message: "添加主程序",
|
||||
Branch: "main",
|
||||
},
|
||||
{
|
||||
Path: "README.md",
|
||||
Content: `# Auto Created Repo\n\nThis repo is created by Gitea API.`,
|
||||
Message: "添加 README",
|
||||
Branch: "main",
|
||||
},
|
||||
}
|
||||
|
||||
tagCfg := &TagConfig{
|
||||
TagName: "v1.0.0",
|
||||
Message: "首次发布",
|
||||
}
|
||||
|
||||
// 方式1：使用快捷函数（自动使用全局单例）
|
||||
err := PushQuick("my-auto-repo", "通过 API 自动创建的公共仓库", files, tagCfg)
|
||||
if err != nil {
|
||||
fmt.Printf("❌ 操作失败: %v\n", err)
|
||||
}
|
||||
|
||||
// 方式2：手动获取客户端实例（更灵活）
|
||||
client, err := GetClient()
|
||||
if err != nil {
|
||||
fmt.Printf("获取客户端失败: %v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 创建私有仓库
|
||||
cfg := RepoConfig{
|
||||
Name: "private-repo",
|
||||
Description: "私有仓库",
|
||||
Private: true,
|
||||
AutoInit: true,
|
||||
}
|
||||
|
||||
// 为组织创建仓库
|
||||
err = client.CreateRepoAndPush(cfg, files, nil, "my-org")
|
||||
if err != nil {
|
||||
fmt.Printf("❌ 操作失败: %v\n", err)
|
||||
}
|
||||
|
||||
// 方式3：单独使用各个功能
|
||||
// 只创建仓库
|
||||
repoCfg := RepoConfig{
|
||||
Name: "test-repo",
|
||||
Private: false,
|
||||
AutoInit: true,
|
||||
}
|
||||
repo, err := CreateRepoQuick(repoCfg)
|
||||
if err != nil {
|
||||
fmt.Printf("创建仓库失败: %v\n", err)
|
||||
}
|
||||
|
||||
// 只推送文件
|
||||
err = PushFileQuick(repo.Owner.UserName, repo.Name, FileConfig{
|
||||
Path: "test.txt",
|
||||
Content: "Hello World",
|
||||
Message: "添加测试文件",
|
||||
Branch: "main",
|
||||
})
|
||||
if err != nil {
|
||||
fmt.Printf("推送文件失败: %v\n", err)
|
||||
}
|
||||
|
||||
// 只创建标签
|
||||
_, err = CreateTagQuick(repo.Owner.UserName, repo.Name, TagConfig{
|
||||
TagName: "v0.1.0",
|
||||
Message: "测试版本",
|
||||
})
|
||||
if err != nil {
|
||||
fmt.Printf("创建标签失败: %v\n", err)
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,409 @@
|
|||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sdk-generator/internal/call"
|
||||
"sdk-generator/internal/config"
|
||||
"sdk-generator/internal/extractor"
|
||||
"sdk-generator/internal/models"
|
||||
"sdk-generator/internal/msg"
|
||||
"sdk-generator/internal/postprocess"
|
||||
"sdk-generator/internal/prompts"
|
||||
"sdk-generator/internal/push"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type SDKGeneratorService struct {
|
||||
config *config.Config
|
||||
tasks map[string]*models.Task
|
||||
tasksMu sync.RWMutex
|
||||
}
|
||||
|
||||
func NewSDKGeneratorService(cfg *config.Config) *SDKGeneratorService {
|
||||
// 创建 OpenAI 客户端配置
|
||||
return &SDKGeneratorService{
|
||||
config: cfg,
|
||||
tasks: make(map[string]*models.Task),
|
||||
}
|
||||
}
|
||||
|
||||
// Generate 异步生成 SDK
|
||||
func (s *SDKGeneratorService) Generate(ctx context.Context, content []byte, sdkName string, llmSet *call.LlmCallSet) (string, error) {
|
||||
// 限制任务数量
|
||||
s.tasksMu.RLock()
|
||||
if len(s.tasks) >= s.config.MaxTasks {
|
||||
s.tasksMu.RUnlock()
|
||||
return "", fmt.Errorf("任务队列已满，请稍后重试")
|
||||
}
|
||||
s.tasksMu.RUnlock()
|
||||
|
||||
taskID := uuid.New().String()
|
||||
task := &models.Task{
|
||||
ID: taskID,
|
||||
Status: models.StatusPending,
|
||||
SdkName: sdkName,
|
||||
OutputDir: filepath.Join(s.config.OutputDir, taskID),
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
CallLLM: call.NewCallLLM(llmSet),
|
||||
MarkdownContent: string(content),
|
||||
}
|
||||
|
||||
s.tasksMu.Lock()
|
||||
s.tasks[taskID] = task
|
||||
s.tasksMu.Unlock()
|
||||
|
||||
// 异步处理
|
||||
go func() {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
s.failTask(task, fmt.Sprintf("panic: %v", r))
|
||||
}
|
||||
}()
|
||||
s.processTask(context.Background(), task, string(content))
|
||||
}()
|
||||
|
||||
return taskID, nil
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) processTask(ctx context.Context, task *models.Task, markdownContent string) {
|
||||
s.processTaskStatus(task, models.StatusProcessing)
|
||||
|
||||
//0.创建目录
|
||||
s.processTaskStatus(task, models.StatusMkdir)
|
||||
err := s.mkdir(ctx, task)
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("[mkdir]: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 1. 分析文档
|
||||
s.processTaskStatus(task, models.StatusAnaMd)
|
||||
err = s.anaMd(ctx, task)
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("[anaMd]: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 2. 生成代码
|
||||
s.processTaskStatus(task, models.StatusGenerateCode)
|
||||
err = s.generateCode(ctx, task)
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("[generateCode]: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 4. 验证是否存在缺失
|
||||
s.processTaskStatus(task, models.StatusValid)
|
||||
err = s.valid(ctx, task)
|
||||
if err != nil {
|
||||
// 后处理失败不中断流程
|
||||
s.failTask(task, fmt.Sprintf("[valid]: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 3. 处理文件
|
||||
s.processTaskStatus(task, models.StatusCreatFile)
|
||||
err = s.creatFile(ctx, task)
|
||||
if err != nil {
|
||||
// 后处理失败不中断流程
|
||||
s.failTask(task, fmt.Sprintf("[creatFile]: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
// 5. 后处理
|
||||
s.processTaskStatus(task, models.StatusProcess)
|
||||
err = postprocess.Process(ctx, task.SdkPackage)
|
||||
if err != nil {
|
||||
|
||||
//如果运行报错则多轮重试
|
||||
s.processTaskStatus(task, models.StatusFix)
|
||||
err = s.fix(ctx, task, err.Error(), 1)
|
||||
}
|
||||
|
||||
// 6. 上传仓库
|
||||
s.processTaskStatus(task, models.StatusSendToGit)
|
||||
err = s.sendToGit(ctx, task)
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("[sendToGit]: %v", err))
|
||||
}
|
||||
|
||||
// 6. 发送钉钉消息
|
||||
s.processTaskStatus(task, models.StatusMsgSend)
|
||||
msg.SendDingTalkAlert(ctx, task.SdkName, task.RepoURL)
|
||||
|
||||
// 5. 完成
|
||||
now := time.Now()
|
||||
task.Status = models.StatusCompleted
|
||||
task.UpdatedAt = now
|
||||
task.CompletedAt = &now
|
||||
s.updateTask(task)
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) sendToGit(ctx context.Context, task *models.Task) error {
|
||||
// 1. 获取 Gitea 客户端
|
||||
giteaClient, err := push.GetClient()
|
||||
if err != nil || giteaClient == nil {
|
||||
return fmt.Errorf("[GetClient]: %v", err)
|
||||
}
|
||||
|
||||
// 2. 准备文件列表
|
||||
var files []push.FileConfig
|
||||
|
||||
// 如果有提取的文件列表，推送所有文件
|
||||
if len(task.Files) > 0 {
|
||||
for _, file := range task.Files {
|
||||
files = append(files, push.FileConfig{
|
||||
Path: file.Path,
|
||||
Content: file.Content,
|
||||
Message: fmt.Sprintf("添加文件: %s", file.Path),
|
||||
Branch: "main",
|
||||
})
|
||||
}
|
||||
} else if task.Resp != "" {
|
||||
// 如果只有 Resp 内容，作为 main.go 推送
|
||||
files = append(files, push.FileConfig{
|
||||
Path: "main.go",
|
||||
Content: task.Resp,
|
||||
Message: "Initial commit: SDK code",
|
||||
Branch: "main",
|
||||
})
|
||||
}
|
||||
|
||||
//3. 如果有 MarkdownContent，添加为 README.md
|
||||
if task.MarkdownContent != "" {
|
||||
files = append(files, push.FileConfig{
|
||||
Path: "README.md",
|
||||
Content: task.MarkdownContent,
|
||||
Message: "添加 README 文档",
|
||||
Branch: "main",
|
||||
})
|
||||
}
|
||||
|
||||
// 4. 如果没有文件，返回错误
|
||||
if len(files) == 0 {
|
||||
|
||||
return fmt.Errorf("没有可推送的文件内容")
|
||||
}
|
||||
|
||||
// 5. 生成仓库名称（使用时间戳确保唯一性）
|
||||
repoName := fmt.Sprintf("%s-%s", task.SdkName, task.CreatedAt.Format("20060102-150405"))
|
||||
|
||||
// 6. 配置标签（可选）
|
||||
var tagCfg *push.TagConfig
|
||||
if task.CompletedAt != nil {
|
||||
tagCfg = &push.TagConfig{
|
||||
TagName: fmt.Sprintf("v%s", task.CompletedAt.Format("20060102.150405")),
|
||||
Message: fmt.Sprintf("SDK 生成完成: %s", task.SdkName),
|
||||
}
|
||||
}
|
||||
|
||||
// 7. 创建仓库并推送代码
|
||||
repoCfg := push.RepoConfig{
|
||||
Name: repoName,
|
||||
Description: task.SdkName,
|
||||
Private: false, // 公共仓库
|
||||
AutoInit: false,
|
||||
Gitignores: "",
|
||||
License: "",
|
||||
Readme: "",
|
||||
}
|
||||
|
||||
err = giteaClient.CreateRepoAndPush(repoCfg, files, tagCfg, s.config.OrgName)
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("推送代码到 Gitea 失败: %v", err))
|
||||
return fmt.Errorf("推送代码到 Gitea 失败: %v", err)
|
||||
}
|
||||
|
||||
task.RepoURL = fmt.Sprintf("%s/%s/%s", s.config.GiteaUrl, s.config.OrgName, repoName)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) mkdir(ctx context.Context, task *models.Task) (err error) {
|
||||
if _, err = os.Stat(task.OutputDir); err == nil {
|
||||
if err = os.RemoveAll(task.OutputDir); err != nil {
|
||||
return fmt.Errorf("删除旧目录失败: %v", err)
|
||||
}
|
||||
}
|
||||
if err = os.MkdirAll(task.OutputDir, 0755); err != nil {
|
||||
|
||||
return fmt.Errorf("创建输出目录失败: %v", err)
|
||||
}
|
||||
task.SdkPackage = filepath.Join(task.OutputDir, task.SdkName)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) generateCode(ctx context.Context, task *models.Task) (err error) {
|
||||
promptSetSdtName := strings.Replace(prompts.GenSDKPrompt, "{{sdk_name}}", task.SdkName, -1)
|
||||
doc := task.Refine
|
||||
if len(doc) == 0 {
|
||||
doc = task.MarkdownContent
|
||||
}
|
||||
prompt := fmt.Sprintf(promptSetSdtName, doc)
|
||||
task.Resp, err = task.CallLLM.Do(ctx, prompts.SDKGeneratorPrompt(prompt))
|
||||
if err != nil {
|
||||
s.failTask(task, fmt.Sprintf("调用大模型失败: %v", err))
|
||||
return
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) anaMd(ctx context.Context, task *models.Task) (err error) {
|
||||
task.Refine, err = task.CallLLM.Do(ctx, prompts.BuildRefinePrompt(task.MarkdownContent))
|
||||
if err != nil {
|
||||
|
||||
return fmt.Errorf("调用大模型失败: %v", err)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) valid(ctx context.Context, task *models.Task) (err error) {
|
||||
|
||||
validRes, err := task.CallLLM.Do(ctx, prompts.GetValidatePrompt(task.Refine, task.SdkName, task.Resp))
|
||||
if err != nil {
|
||||
return fmt.Errorf("调用大模型失败: %v", err)
|
||||
}
|
||||
if validRes != "OK" {
|
||||
task.Valid = validRes
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) fix(ctx context.Context, task *models.Task, errMsg string, fixCount int) (err error) {
|
||||
log.Printf("代码存在问题，需要修复: %s,当前修复次数:%d", errMsg, fixCount)
|
||||
if fixCount >= s.config.MaxFixAttempts {
|
||||
return fmt.Errorf("修复失败，已达到最大尝试次数: %v", fixCount)
|
||||
}
|
||||
fix, err := task.CallLLM.Do(ctx, prompts.FixPrompt(task.Resp, errMsg, task.SdkName, task.Refine))
|
||||
if err != nil {
|
||||
return fmt.Errorf("调用大模型失败: %v", err)
|
||||
}
|
||||
if len(task.Valid) > 0 && task.Valid != "OK" {
|
||||
task.Valid = fix
|
||||
} else {
|
||||
task.Resp = fix
|
||||
}
|
||||
if err = s.creatFile(ctx, task); err != nil {
|
||||
// 后处理失败不中断流程
|
||||
return fmt.Errorf("创建文件失败: %v", err)
|
||||
}
|
||||
|
||||
if err = postprocess.Process(ctx, task.SdkPackage); err != nil {
|
||||
err = s.fix(ctx, task, err.Error(), fixCount+1)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) creatFile(ctx context.Context, task *models.Task) error {
|
||||
// 3. 提取代码
|
||||
extract := task.Resp
|
||||
if task.Valid != "" {
|
||||
extract = task.Valid
|
||||
}
|
||||
files, err := extractor.Extract(extract)
|
||||
if err != nil {
|
||||
|
||||
return fmt.Errorf("提取代码失败: %v", err)
|
||||
}
|
||||
|
||||
if len(files) == 0 {
|
||||
return fmt.Errorf("未提取到任何代码文件")
|
||||
}
|
||||
files = append(files,
|
||||
extractor.File{
|
||||
Path: filepath.Join(task.SdkName, "doc.md"),
|
||||
Content: task.MarkdownContent,
|
||||
},
|
||||
extractor.File{
|
||||
Path: filepath.Join(task.SdkName, "refine.md"),
|
||||
Content: task.Refine,
|
||||
},
|
||||
extractor.File{
|
||||
Path: filepath.Join(task.SdkName, "generate.md"),
|
||||
Content: task.Resp,
|
||||
},
|
||||
extractor.File{
|
||||
Path: filepath.Join(task.SdkName, "valid.md"),
|
||||
Content: task.Valid,
|
||||
},
|
||||
)
|
||||
// 3. 写入文件
|
||||
if err = extractor.WriteFiles(files, task.OutputDir); err != nil {
|
||||
return fmt.Errorf("写入文件失败: %v", err)
|
||||
}
|
||||
task.Files = files
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) failTask(task *models.Task, errMsg string) {
|
||||
task.Status = models.StatusFailed
|
||||
task.Error = errMsg
|
||||
task.UpdatedAt = time.Now()
|
||||
s.updateTask(task)
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) processTaskStatus(task *models.Task, process models.TaskStatus) {
|
||||
task.Status = process
|
||||
task.UpdatedAt = time.Now()
|
||||
s.tasksMu.Lock()
|
||||
defer s.tasksMu.Unlock()
|
||||
s.tasks[task.ID] = task
|
||||
}
|
||||
|
||||
func (s *SDKGeneratorService) updateTask(task *models.Task) {
|
||||
s.tasksMu.Lock()
|
||||
defer s.tasksMu.Unlock()
|
||||
s.tasks[task.ID] = task
|
||||
}
|
||||
|
||||
// GetTask 获取任务状态
|
||||
func (s *SDKGeneratorService) GetTask(taskID string) (*models.Task, bool) {
|
||||
s.tasksMu.RLock()
|
||||
defer s.tasksMu.RUnlock()
|
||||
task, ok := s.tasks[taskID]
|
||||
return task, ok
|
||||
}
|
||||
|
||||
// ListTasks 列出所有任务
|
||||
func (s *SDKGeneratorService) ListTasks() []*models.Task {
|
||||
s.tasksMu.RLock()
|
||||
defer s.tasksMu.RUnlock()
|
||||
|
||||
tasks := make([]*models.Task, 0, len(s.tasks))
|
||||
for _, task := range s.tasks {
|
||||
tasks = append(tasks, task)
|
||||
}
|
||||
return tasks
|
||||
}
|
||||
|
||||
// DeleteTask 删除任务
|
||||
func (s *SDKGeneratorService) DeleteTask(taskID string) error {
|
||||
s.tasksMu.Lock()
|
||||
defer s.tasksMu.Unlock()
|
||||
|
||||
task, ok := s.tasks[taskID]
|
||||
if !ok {
|
||||
return fmt.Errorf("任务不存在")
|
||||
}
|
||||
|
||||
// 删除输出目录
|
||||
if err := os.RemoveAll(task.OutputDir); err != nil {
|
||||
return fmt.Errorf("删除输出目录失败: %w", err)
|
||||
}
|
||||
|
||||
delete(s.tasks, taskID)
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetTaskOutputDir 获取任务输出目录
|
||||
func (s *SDKGeneratorService) GetTaskOutputDir(taskID string) string {
|
||||
return filepath.Join(s.config.OutputDir, taskID)
|
||||
}
|
||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
|
|
@ -0,0 +1,92 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"sdk-generator/internal/config"
|
||||
"sdk-generator/internal/handler"
|
||||
"sdk-generator/internal/push"
|
||||
"sdk-generator/internal/service"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/gofiber/fiber/v2/middleware/cors"
|
||||
"github.com/gofiber/fiber/v2/middleware/logger"
|
||||
"github.com/gofiber/fiber/v2/middleware/recover"
|
||||
"github.com/gofiber/template/html/v2"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 加载配置
|
||||
cfg := config.Load()
|
||||
|
||||
// 创建服务
|
||||
sdkService := service.NewSDKGeneratorService(cfg)
|
||||
|
||||
// 创建 Handler
|
||||
sdkHandler := handler.NewSDKHandler(sdkService)
|
||||
pageHandler := handler.NewPageHandler()
|
||||
|
||||
// 配置模板引擎
|
||||
engine := html.New("./web/templates", ".html")
|
||||
engine.Reload(true) // 开发时热重载
|
||||
|
||||
// 创建 Fiber 应用
|
||||
app := fiber.New(fiber.Config{
|
||||
AppName: "SDK Generator API",
|
||||
Views: engine,
|
||||
Prefork: false,
|
||||
ServerHeader: "Fiber",
|
||||
BodyLimit: 10 * 1024 * 1024, // 10MB
|
||||
})
|
||||
|
||||
// 全局中间件
|
||||
app.Use(recover.New())
|
||||
app.Use(logger.New(logger.Config{
|
||||
Format: "[${time}] ${status} - ${method} ${path} ${latency}\n",
|
||||
}))
|
||||
app.Use(cors.New(cors.Config{
|
||||
AllowOrigins: "*",
|
||||
AllowMethods: "GET,POST,PUT,DELETE,OPTIONS",
|
||||
AllowHeaders: "Content-Type,Authorization",
|
||||
}))
|
||||
|
||||
// 健康检查
|
||||
app.Get("/health", func(c *fiber.Ctx) error {
|
||||
return c.JSON(fiber.Map{
|
||||
"status": "ok",
|
||||
"app": "SDK Generator API",
|
||||
})
|
||||
})
|
||||
|
||||
// 静态文件
|
||||
app.Static("/static", "./web/static")
|
||||
|
||||
// 页面路由
|
||||
app.Get("/", pageHandler.Index)
|
||||
|
||||
// API 路由
|
||||
api := app.Group("/api/v1")
|
||||
|
||||
// 上传文档生成 SDK
|
||||
api.Post("/generate", sdkHandler.GenerateSDK)
|
||||
|
||||
// 查询生成任务状态
|
||||
api.Get("/tasks/:task_id", sdkHandler.GetTaskStatus)
|
||||
|
||||
// 下载生成的 SDK
|
||||
api.Get("/tasks/:task_id/download", sdkHandler.DownloadSDK)
|
||||
|
||||
// 列出所有任务
|
||||
api.Get("/tasks", sdkHandler.ListTasks)
|
||||
|
||||
// 删除任务
|
||||
api.Delete("/tasks/:task_id", sdkHandler.DeleteTask)
|
||||
//初始化代码仓库客户端
|
||||
_, err := push.NewGiteaClient(cfg.GiteaUrl, cfg.GiteaToken)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to initialize Gitea client: %v", err)
|
||||
}
|
||||
// 启动服务请访问以下链接获取接口“生成”的接口定义信息：https://api.apifox.cn/temp-links/api/488609283?t=cb680e83-e054-41e5-a167-715db384e1d1
|
||||
if err := app.Listen(":" + cfg.Port); err != nil {
|
||||
log.Fatalf("Failed to start server: %v", err)
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,747 @@
|
|||
# 识别内容(本地文件)
|
||||
|
||||
## OpenAPI Specification
|
||||
|
||||
```yaml
|
||||
openapi: 3.0.1
|
||||
info:
|
||||
title: ''
|
||||
version: 1.0.0
|
||||
paths:
|
||||
/convert:
|
||||
post:
|
||||
summary: 识别内容(本地文件)
|
||||
deprecated: false
|
||||
description: ''
|
||||
tags:
|
||||
- any2md
|
||||
parameters: []
|
||||
requestBody:
|
||||
content:
|
||||
multipart/form-data:
|
||||
schema:
|
||||
type: object
|
||||
properties:
|
||||
file:
|
||||
example: file://C:\Users\Administrator\Desktop\sucai\bl.png
|
||||
type: string
|
||||
format: binary
|
||||
llm_model:
|
||||
example: doubao-seed-2-0-mini-260428
|
||||
type: string
|
||||
llm_api_key:
|
||||
example: '******'
|
||||
type: string
|
||||
llm_base_url:
|
||||
example: https://ark.cn-beijing.volces.com/api/v3
|
||||
type: string
|
||||
llm_prompt:
|
||||
example: 这个图片合法吗
|
||||
type: string
|
||||
required:
|
||||
- file
|
||||
example:
|
||||
file: file://C:\Users\Administrator\Desktop\sucai\bl.png
|
||||
llm_model: doubao-seed-2-0-mini-260428
|
||||
llm_api_key: '******'
|
||||
llm_base_url: https://ark.cn-beijing.volces.com/api/v3
|
||||
llm_prompt: 这个图片合法吗
|
||||
required: true
|
||||
responses:
|
||||
'200':
|
||||
description: ''
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: object
|
||||
properties:
|
||||
filename:
|
||||
type: string
|
||||
llm_used:
|
||||
type: boolean
|
||||
markdown:
|
||||
type: string
|
||||
size:
|
||||
type: integer
|
||||
success:
|
||||
type: boolean
|
||||
title:
|
||||
type: 'null'
|
||||
required:
|
||||
- filename
|
||||
- llm_used
|
||||
- markdown
|
||||
- size
|
||||
- success
|
||||
- title
|
||||
example:
|
||||
filename: 蓝色兄弟营销开放API V3.docx
|
||||
llm_used: false
|
||||
markdown: >-
|
||||
# **蓝色兄弟营销开放API V3**
|
||||
|
||||
|
||||
## 接入流程
|
||||
|
||||
|
||||
1. **注册开发应用账号**：联系平台商务或技术支持，提交接入申请。
|
||||
|
||||
2. **获取应用接入信息**：创建应用成功后，可获取应用 ID、应用私钥配置要求、平台公钥及业务加密 Key。
|
||||
|
||||
3. **阅读接口文档**：仔细阅读本文档中的接口说明，确认请求加密、签名、响应解密方式。
|
||||
|
||||
4. **联调测试**：在正式使用前，完成单发、查询、作废、批量、异常场景验证。
|
||||
|
||||
|
||||
### 环境配置
|
||||
|
||||
|
||||
测试环境地址：https://gateway.dev.cdlsxd.cn
|
||||
|
||||
|
||||
正式环境地址：https://market.api.86698.cn
|
||||
|
||||
|
||||
### 测试参数
|
||||
|
||||
|
||||
# 客户应用id
|
||||
|
||||
app\_id: "xxx"
|
||||
|
||||
# 应用客户私钥，用于请求签名
|
||||
|
||||
private\_key: "xxx"
|
||||
|
||||
# 应用平台公钥，用于平台响应或回调验签
|
||||
|
||||
public\_key: "xxx"
|
||||
|
||||
# 业务参数加密key
|
||||
|
||||
key: "xxxx"
|
||||
|
||||
# 活动编号
|
||||
|
||||
activity\_no: "xxxx"
|
||||
|
||||
# 签名类型
|
||||
|
||||
sign\_type: "RSA"
|
||||
|
||||
|
||||
## 概述
|
||||
|
||||
|
||||
### V3 接入说明
|
||||
|
||||
|
||||
V3 版本统一采用 ciphertext 传输业务报文，默认接入方式与现有开放 API 保持一致。
|
||||
|
||||
|
||||
### 业务参数
|
||||
|
||||
|
||||
将业务参数组装为 JSON 明文 plaintext
|
||||
|
||||
再使用应用 key 加密，得到 ciphertext
|
||||
|
||||
|
||||
### 签名规则
|
||||
|
||||
|
||||
拼接签名字符串：分配给开发者的应用ID + 发送请求的时间 + ciphertext
|
||||
|
||||
使用应用私钥生成签名字符串 sign
|
||||
|
||||
|
||||
### 请求模式
|
||||
|
||||
|
||||
V3 统一采用 Header 鉴权模式：
|
||||
|
||||
|
||||
* Header 传：Appid、Timestamp、Sign
|
||||
|
||||
* Body 传：ciphertext
|
||||
|
||||
|
||||
### 响应说明
|
||||
|
||||
|
||||
V3 统一采用严格密文响应模式，业务响应数据通过 ciphertext 返回。
|
||||
|
||||
|
||||
### 时间戳规则
|
||||
|
||||
|
||||
时间格式：yyyy-MM-dd HH:mm:ss
|
||||
|
||||
请求时间与服务端时间误差不能超过 3 分钟
|
||||
|
||||
|
||||
### SDK
|
||||
|
||||
|
||||
开发者可参考以下 SDK：
|
||||
|
||||
|
||||
// Go
|
||||
|
||||
https://gitee.com/chengdu\_blue\_brothers/ymt-openapi-go-sdk.git
|
||||
|
||||
// Java
|
||||
|
||||
https://codeup.aliyun.com/lsxd/marketing/ymt-openapi-java-sdk.git
|
||||
|
||||
|
||||
### 公共 Header 请求参数
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | **示例值** |
|
||||
|
||||
| Appid | string | 分配给开发者的应用 ID | 123456 |
|
||||
|
||||
| Timestamp | string | 发送请求的时间，格式 yyyy-MM-dd HH:mm:ss |
|
||||
2026-06-22 15:30:00 |
|
||||
|
||||
| Sign | string | 请求签名串 | 详见 SDK 示例 |
|
||||
|
||||
| Content-Type | string | 请求数据格式 | application/json |
|
||||
|
||||
|
||||
#### 公共 Header 请求参数示例
|
||||
|
||||
|
||||
{
|
||||
"Sign": ["签名字符串"],
|
||||
"Appid": ["应用ID"],
|
||||
"Timestamp": ["请求时间"],
|
||||
"Content-Type": ["application/json"]
|
||||
}
|
||||
|
||||
|
||||
### 公共请求参数
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | **示例值** |
|
||||
|
||||
| ciphertext | string | 请求业务参数加密串 | 详见 SDK 示例 |
|
||||
|
||||
|
||||
#### 公共请求参数示例
|
||||
|
||||
|
||||
{
|
||||
"ciphertext": "加密后的业务报文"
|
||||
}
|
||||
|
||||
|
||||
### 公共响应参数
|
||||
|
||||
|
||||
| | | |
|
||||
|
||||
| --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** |
|
||||
|
||||
| code | int32 | 200 成功 |
|
||||
|
||||
| message | string | 请求描述 |
|
||||
|
||||
| reason | string | 错误原因，错误时返回 |
|
||||
|
||||
| data.ciphertext | string | 业务响应加密串 |
|
||||
|
||||
|
||||
### 公共错误码
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **code状态码** | **reason** | **错误原因** | **解决** |
|
||||
|
||||
| 500 | PANIC/其它 | 系统错误 | 联系平台处理 |
|
||||
|
||||
| 400 | INVALID\_PAYLOAD | 请求外壳格式错误 | 请检查请求 JSON 结构 |
|
||||
|
||||
| 400 | MISSING\_PARAM | 缺少必要参数 | 请检查
|
||||
app\_id、timestamp、sign、ciphertext |
|
||||
|
||||
| 400 | INVALID\_TIMESTAMP | 时间格式错误 | 请检查时间格式 |
|
||||
|
||||
| 400 | DECRYPT\_FAILED | 业务参数解密失败 | 请检查加密方式与密钥 |
|
||||
|
||||
| 401 | APP\_NOT\_FOUND | 应用不存在 | 请检查应用 ID |
|
||||
|
||||
| 401 | INVALID\_SIGNATURE | 签名错误 | 请检查签名串与私钥 |
|
||||
|
||||
| 401 | EXPIRED\_TIMESTAMP | 请求已过期 | 请检查客户端时间 |
|
||||
|
||||
| 429 | DUPLICATE\_REQUEST | 重复请求，请稍后重试 | 请避免短时间内重复提交同一业务号 |
|
||||
|
||||
|
||||
### 业务错误码[汇总]
|
||||
|
||||
|
||||
备注：若出现其它未知状态码，请联系平台技术人员。
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **code状态码** | **reason** | **错误原因** | **解决** |
|
||||
|
||||
| 401 | ACTIVITY\_NOT\_AUTH | 活动未授权 | 请检查活动授权状态 |
|
||||
|
||||
| 401 | MERCHANT\_NOT\_EXIST | 客户不存在 | 请检查客户是否存在 |
|
||||
|
||||
| 401 | MERCHANT\_NOT\_AUTH | 客户冻结 | 请检查客户授权状态 |
|
||||
|
||||
| 401 | MERCHANT\_APP\_INCOMPLETE | 客户应用配置未完善 | 请检查客户应用配置 |
|
||||
|
||||
| 401 | MERCHANT\_APP\_NOT\_AUTH | 应用不存在或未授权 | 请检查客户应用授权状态 |
|
||||
|
||||
| 403 | ACTIVITY\_EXPIRE | 活动已结束 | 请检查活动是否有效 |
|
||||
|
||||
| 403 | ACTIVITY\_OUT\_OF\_STOCK | 活动剩余量不足 | 请检查活动库存 |
|
||||
|
||||
| 404 | ACTIVITY\_NOT\_EXIST | 活动不存在 | 请检查活动编号 |
|
||||
|
||||
| 404 | MERCHANT\_ORDER\_NOT\_EXIST | 订单不存在 | 请检查交易号或外部业务号 |
|
||||
|
||||
| 404 | KEY\_NOT\_EXIST | key码不存在 | 请检查订单信息 |
|
||||
|
||||
| 400 | PARAM\_FAIL | 参数错误 | 请检查业务参数 |
|
||||
|
||||
| 400 | PARAM\_DECRYPT\_FAIL | 明文参数格式错误 | 请检查密文解密后的业务报文 |
|
||||
|
||||
|
||||
### 获取券码
|
||||
|
||||
|
||||
* 请求方式：POST
|
||||
|
||||
* Content-Type：application/json
|
||||
|
||||
* 请求路由：/openapi/v3/key/order
|
||||
|
||||
|
||||
#### 业务请求参数
|
||||
|
||||
|
||||
| | | | | |
|
||||
|
||||
| --- | --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | 是否必填 | **示例值** |
|
||||
|
||||
| out\_biz\_no | string | 外部业务号，幂等 | M | 123456 |
|
||||
|
||||
| activity\_no | string | 活动编号 | M | ACT20260622001 |
|
||||
|
||||
| account | string | 账号，按活动类型透传 | N | 18666666666 |
|
||||
|
||||
| notify\_url | string | 回调通知地址 | N |
|
||||
https://notify.example.com/openapi |
|
||||
|
||||
|
||||
#### 业务响应参数（解密后）
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | 是否必填 | **描述** |
|
||||
|
||||
| out\_biz\_no | string | M | 外部业务号 |
|
||||
|
||||
| trade\_no | string | M | 交易号 |
|
||||
|
||||
| key | string | N | 卡密 |
|
||||
|
||||
| url | string | N | 链接型活动返回短链接，key/url 不会同时为空 |
|
||||
|
||||
| valid\_begin\_time | string | N | 生效时间 |
|
||||
|
||||
| valid\_end\_time | string | N | 失效时间 |
|
||||
|
||||
| usable\_num | uint32 | M | 可用次数 |
|
||||
|
||||
| usage\_num | uint32 | M | 已使用次数 |
|
||||
|
||||
| status | uint32 | M | 状态：1 正常，2 已核销，3 已作废 |
|
||||
|
||||
| settlement\_price | float | N | 结算价 |
|
||||
|
||||
| account | string | N | 上报账号 |
|
||||
|
||||
|
||||
#### 请求示例
|
||||
|
||||
|
||||
明文业务参数：
|
||||
|
||||
|
||||
{
|
||||
"out\_biz\_no": "order\_001",
|
||||
"activity\_no": "ACT20260622001",
|
||||
"account": "18666666666",
|
||||
"notify\_url": "https://notify.example.com/openapi"
|
||||
}
|
||||
|
||||
|
||||
请求体示例：
|
||||
|
||||
|
||||
{
|
||||
"ciphertext": "加密后的业务报文"
|
||||
}
|
||||
|
||||
|
||||
#### 响应示例
|
||||
|
||||
|
||||
异常
|
||||
|
||||
|
||||
{
|
||||
"code": 401,
|
||||
"message": "Signature verification failed",
|
||||
"reason": "INVALID\_SIGNATURE"
|
||||
}
|
||||
|
||||
|
||||
成功
|
||||
|
||||
|
||||
{
|
||||
"code": 200,
|
||||
"data": {
|
||||
"ciphertext": "加密后的响应报文"
|
||||
},
|
||||
"message": "成功"
|
||||
}
|
||||
|
||||
|
||||
解密后示例
|
||||
|
||||
|
||||
{
|
||||
"out\_biz\_no": "order\_001",
|
||||
"trade\_no": "7251449503000383488",
|
||||
"key": "aZKdU9BymzR6qGRzJM",
|
||||
"url": "",
|
||||
"valid\_begin\_time": "2026-06-22 15:30:00",
|
||||
"valid\_end\_time": "2026-12-31 23:59:59",
|
||||
"usable\_num": 1,
|
||||
"usage\_num": 0,
|
||||
"status": 1,
|
||||
"settlement\_price": 9.9,
|
||||
"account": "18666666666"
|
||||
}
|
||||
|
||||
|
||||
### 券码查询
|
||||
|
||||
|
||||
* 请求方式：POST
|
||||
|
||||
* Content-Type：application/json
|
||||
|
||||
* 请求路由：/openapi/v3/key/query
|
||||
|
||||
|
||||
#### 业务请求参数
|
||||
|
||||
|
||||
| | | | | |
|
||||
|
||||
| --- | --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | 是否必填 | **示例值** |
|
||||
|
||||
| out\_biz\_no | string | 外部业务号，与 trade\_no 二选一 | N |
|
||||
order\_001 |
|
||||
|
||||
| trade\_no | string | 交易号，与 out\_biz\_no 二选一 | N |
|
||||
7251449503000383488 |
|
||||
|
||||
|
||||
#### 业务响应参数
|
||||
|
||||
|
||||
与“获取券码”响应参数一致。
|
||||
|
||||
|
||||
#### 响应示例
|
||||
|
||||
|
||||
{
|
||||
"code": 200,
|
||||
"data": {
|
||||
"out\_biz\_no": "order\_001",
|
||||
"trade\_no": "7251449503000383488",
|
||||
"key": "aZKdU9BymzR6qGRzJM",
|
||||
"url": "",
|
||||
"valid\_begin\_time": "2026-06-22 15:30:00",
|
||||
"valid\_end\_time": "2026-12-31 23:59:59",
|
||||
"usable\_num": 1,
|
||||
"usage\_num": 0,
|
||||
"status": 1,
|
||||
"settlement\_price": 9.9,
|
||||
"account": "18666666666",
|
||||
"ciphertext": ""
|
||||
},
|
||||
"message": "成功"
|
||||
}
|
||||
|
||||
|
||||
### 券码作废
|
||||
|
||||
|
||||
* 请求方式：POST
|
||||
|
||||
* Content-Type：application/json
|
||||
|
||||
* 请求路由：/openapi/v3/key/discard
|
||||
|
||||
|
||||
#### 业务请求参数
|
||||
|
||||
|
||||
| | | | | |
|
||||
|
||||
| --- | --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | 是否必填 | **示例值** |
|
||||
|
||||
| out\_biz\_no | string | 外部业务号，与 trade\_no 二选一 | N |
|
||||
order\_001 |
|
||||
|
||||
| trade\_no | string | 交易号，与 out\_biz\_no 二选一 | N |
|
||||
7251449503000383488 |
|
||||
|
||||
|
||||
#### 业务响应参数
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | 是否必填 | **描述** |
|
||||
|
||||
| out\_biz\_no | string | M | 外部业务号 |
|
||||
|
||||
| trade\_no | string | M | 交易号 |
|
||||
|
||||
| status | uint32 | M | 3 表示已作废 |
|
||||
|
||||
|
||||
#### 响应示例
|
||||
|
||||
|
||||
{
|
||||
"code": 200,
|
||||
"data": {
|
||||
"out\_biz\_no": "order\_001",
|
||||
"trade\_no": "7251449503000383488",
|
||||
"status": 3,
|
||||
"ciphertext": ""
|
||||
},
|
||||
"message": "成功"
|
||||
}
|
||||
|
||||
|
||||
### 批量发卡
|
||||
|
||||
|
||||
* 请求方式：POST
|
||||
|
||||
* Content-Type：application/json
|
||||
|
||||
* 请求路由：/openapi/v3/key/batch\_order
|
||||
|
||||
|
||||
#### 业务请求参数
|
||||
|
||||
|
||||
| | | | | |
|
||||
|
||||
| --- | --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | 是否必填 | **示例值** |
|
||||
|
||||
| out\_biz\_no | string | 外部业务号，幂等 | M | batch\_001 |
|
||||
|
||||
| activity\_no | string | 活动编号 | M | ACT20260622001 |
|
||||
|
||||
| number | int32 | 发卡数量 | M | 100 |
|
||||
|
||||
| notify\_url | string | 回调通知地址 | N |
|
||||
https://notify.example.com/openapi |
|
||||
|
||||
|
||||
#### 业务响应参数
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | 是否必填 | **描述** |
|
||||
|
||||
| out\_biz\_no | string | M | 外部业务号 |
|
||||
|
||||
| trade\_no | string | M | 交易号 |
|
||||
|
||||
| status | string | M | 任务状态，初始返回 processing |
|
||||
|
||||
|
||||
#### 响应示例
|
||||
|
||||
|
||||
{
|
||||
"code": 200,
|
||||
"data": {
|
||||
"ciphertext": "加密后的响应报文"
|
||||
},
|
||||
"message": "成功"
|
||||
}
|
||||
|
||||
|
||||
解密后示例：
|
||||
|
||||
|
||||
{
|
||||
"out\_biz\_no": "batch\_001",
|
||||
"trade\_no": "7251449503000383499",
|
||||
"status": "processing"
|
||||
}
|
||||
|
||||
|
||||
### 批量查询
|
||||
|
||||
|
||||
* 请求方式：POST
|
||||
|
||||
* Content-Type：application/json
|
||||
|
||||
* 请求路由：/openapi/v3/key/batch\_query
|
||||
|
||||
|
||||
#### 业务请求参数
|
||||
|
||||
|
||||
| | | | | |
|
||||
|
||||
| --- | --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | **描述** | 是否必填 | **示例值** |
|
||||
|
||||
| out\_biz\_no | string | 外部业务号，与 trade\_no 二选一 | N |
|
||||
batch\_001 |
|
||||
|
||||
| trade\_no | string | 交易号，与 out\_biz\_no 二选一 | N |
|
||||
7251449503000383499 |
|
||||
|
||||
|
||||
#### 业务响应参数
|
||||
|
||||
|
||||
| | | | |
|
||||
|
||||
| --- | --- | --- | --- |
|
||||
|
||||
| **字段名称** | **类型** | 是否必填 | **描述** |
|
||||
|
||||
| out\_biz\_no | string | M | 外部业务号 |
|
||||
|
||||
| trade\_no | string | M | 交易号 |
|
||||
|
||||
| status | string | M | processing / success / failed |
|
||||
|
||||
| download\_url | string | N | 批量任务成功后返回下载地址 |
|
||||
|
||||
| zip\_password | string | N | 批量任务成功后返回压缩包密码 |
|
||||
|
||||
|
||||
#### 响应示例
|
||||
|
||||
|
||||
{
|
||||
"code": 200,
|
||||
"data": {
|
||||
"ciphertext": "加密后的响应报文"
|
||||
},
|
||||
"message": "成功"
|
||||
}
|
||||
|
||||
|
||||
解密后示例：
|
||||
|
||||
|
||||
{
|
||||
"out\_biz\_no": "batch\_001",
|
||||
"trade\_no": "7251449503000383499",
|
||||
"status": "success",
|
||||
"download\_url": "https://oss.example.com/openapi\\_7251449503000383499.zip",
|
||||
"zip\_password": "123456"
|
||||
}
|
||||
|
||||
|
||||
### 高级能力说明
|
||||
|
||||
|
||||
如客户有特殊签名套件、加密套件适配需求，可通过商户应用 GatewayConfig 做高级配置。
|
||||
|
||||
|
||||
默认接入无需关注该项；如需启用，请联系平台技术人员对接。
|
||||
|
||||
|
||||
### 联调建议
|
||||
|
||||
|
||||
1. 先验证单发、查询、作废三个基础接口。
|
||||
|
||||
2. 再验证批量发卡与批量查询。
|
||||
|
||||
3. 重点验证响应密文解密结果是否正确。
|
||||
|
||||
4. 批量发卡需验证 processing -> success/failed 状态流转。
|
||||
|
||||
5. 批量成功后需验证 download\_url、zip\_password 是否可用。
|
||||
size: 12582
|
||||
success: true
|
||||
title: null
|
||||
headers: {}
|
||||
x-apifox-name: 成功
|
||||
x-apifox-ordering: 0
|
||||
security: []
|
||||
x-apifox-folder: any2md
|
||||
x-apifox-status: developing
|
||||
x-run-in-apifox: https://app.apifox.com/web/project/8591432/apis/api-488221832-run
|
||||
components:
|
||||
schemas: {}
|
||||
responses: {}
|
||||
securitySchemes: {}
|
||||
servers:
|
||||
- url: 127.0.0.1:5002
|
||||
description: generator_api
|
||||
security: []
|
||||
|
||||
```
|
||||
|
|
@ -0,0 +1,373 @@
|
|||
/* ===== 全局重置 ===== */
|
||||
* {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||||
background: #f5f7fa;
|
||||
color: #1a1a2e;
|
||||
min-height: 100vh;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
padding: 40px 20px;
|
||||
}
|
||||
|
||||
.container {
|
||||
max-width: 820px;
|
||||
width: 100%;
|
||||
background: #ffffff;
|
||||
border-radius: 24px;
|
||||
padding: 40px 48px;
|
||||
box-shadow: 0 20px 60px rgba(0, 0, 0, 0.08);
|
||||
}
|
||||
|
||||
/* ===== 头部 ===== */
|
||||
.header {
|
||||
text-align: center;
|
||||
margin-bottom: 32px;
|
||||
}
|
||||
|
||||
.header h1 {
|
||||
font-size: 28px;
|
||||
font-weight: 700;
|
||||
background: linear-gradient(135deg, #6366f1, #8b5cf6);
|
||||
-webkit-background-clip: text;
|
||||
-webkit-text-fill-color: transparent;
|
||||
}
|
||||
|
||||
.subtitle {
|
||||
color: #6b7280;
|
||||
font-size: 15px;
|
||||
margin-top: 4px;
|
||||
}
|
||||
|
||||
/* ===== 上传区域 ===== */
|
||||
.upload-section {
|
||||
margin-bottom: 24px;
|
||||
}
|
||||
|
||||
.upload-zone {
|
||||
border: 2px dashed #d1d5db;
|
||||
border-radius: 16px;
|
||||
padding: 48px 24px;
|
||||
text-align: center;
|
||||
cursor: pointer;
|
||||
transition: all 0.25s ease;
|
||||
background: #fafbfc;
|
||||
}
|
||||
|
||||
.upload-zone:hover {
|
||||
border-color: #8b5cf6;
|
||||
background: #f5f3ff;
|
||||
}
|
||||
|
||||
.upload-zone.dragover {
|
||||
border-color: #8b5cf6;
|
||||
background: #ede9fe;
|
||||
}
|
||||
|
||||
.upload-icon {
|
||||
font-size: 48px;
|
||||
margin-bottom: 12px;
|
||||
}
|
||||
|
||||
.upload-text {
|
||||
font-size: 16px;
|
||||
font-weight: 500;
|
||||
color: #1f2937;
|
||||
}
|
||||
|
||||
.upload-hint {
|
||||
font-size: 13px;
|
||||
color: #9ca3af;
|
||||
margin-top: 4px;
|
||||
}
|
||||
|
||||
#fileInput {
|
||||
display: none;
|
||||
}
|
||||
|
||||
.file-info {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 12px;
|
||||
padding: 12px 16px;
|
||||
background: #f0fdf4;
|
||||
border-radius: 12px;
|
||||
border: 1px solid #86efac;
|
||||
margin-top: 12px;
|
||||
}
|
||||
|
||||
.file-name {
|
||||
font-weight: 500;
|
||||
color: #166534;
|
||||
flex: 1;
|
||||
}
|
||||
|
||||
.file-size {
|
||||
color: #6b7280;
|
||||
font-size: 13px;
|
||||
}
|
||||
|
||||
.btn-remove {
|
||||
background: none;
|
||||
border: none;
|
||||
color: #ef4444;
|
||||
font-size: 18px;
|
||||
cursor: pointer;
|
||||
padding: 0 4px;
|
||||
}
|
||||
|
||||
/* ===== 配置区域 ===== */
|
||||
.config-section {
|
||||
display: grid;
|
||||
grid-template-columns: 1fr 1fr;
|
||||
gap: 16px 24px;
|
||||
margin-bottom: 24px;
|
||||
}
|
||||
|
||||
.form-group {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 4px;
|
||||
}
|
||||
|
||||
.form-group label {
|
||||
font-size: 13px;
|
||||
font-weight: 500;
|
||||
color: #374151;
|
||||
}
|
||||
|
||||
.form-group .required {
|
||||
color: #ef4444;
|
||||
}
|
||||
|
||||
.form-group input,
|
||||
.form-group select {
|
||||
padding: 10px 14px;
|
||||
border: 1px solid #d1d5db;
|
||||
border-radius: 10px;
|
||||
font-size: 14px;
|
||||
transition: border 0.2s;
|
||||
background: #fafbfc;
|
||||
}
|
||||
|
||||
.form-group input:focus,
|
||||
.form-group select:focus {
|
||||
outline: none;
|
||||
border-color: #8b5cf6;
|
||||
box-shadow: 0 0 0 3px rgba(139, 92, 246, 0.15);
|
||||
}
|
||||
|
||||
.form-group input:invalid {
|
||||
border-color: #fca5a5;
|
||||
}
|
||||
|
||||
.form-group small {
|
||||
font-size: 12px;
|
||||
color: #9ca3af;
|
||||
}
|
||||
|
||||
/* ===== 操作按钮 ===== */
|
||||
.action-section {
|
||||
text-align: center;
|
||||
margin-bottom: 24px;
|
||||
}
|
||||
|
||||
.btn-generate {
|
||||
padding: 14px 48px;
|
||||
font-size: 16px;
|
||||
font-weight: 600;
|
||||
border: none;
|
||||
border-radius: 12px;
|
||||
background: linear-gradient(135deg, #6366f1, #8b5cf6);
|
||||
color: #fff;
|
||||
cursor: pointer;
|
||||
transition: all 0.2s ease;
|
||||
display: inline-flex;
|
||||
align-items: center;
|
||||
gap: 8px;
|
||||
}
|
||||
|
||||
.btn-generate:hover:not(:disabled) {
|
||||
transform: translateY(-2px);
|
||||
box-shadow: 0 8px 30px rgba(99, 102, 241, 0.35);
|
||||
}
|
||||
|
||||
.btn-generate:disabled {
|
||||
opacity: 0.5;
|
||||
cursor: not-allowed;
|
||||
transform: none;
|
||||
}
|
||||
|
||||
.btn-spinner {
|
||||
display: inline-block;
|
||||
animation: spin 0.8s linear infinite;
|
||||
}
|
||||
|
||||
@keyframes spin {
|
||||
to { transform: rotate(360deg); }
|
||||
}
|
||||
|
||||
/* ===== 进度区域 ===== */
|
||||
.progress-section {
|
||||
margin-bottom: 24px;
|
||||
padding: 20px 0;
|
||||
}
|
||||
|
||||
.progress-bar {
|
||||
height: 6px;
|
||||
background: #e5e7eb;
|
||||
border-radius: 4px;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.progress-fill {
|
||||
height: 100%;
|
||||
background: linear-gradient(90deg, #6366f1, #8b5cf6);
|
||||
border-radius: 4px;
|
||||
transition: width 0.4s ease;
|
||||
}
|
||||
|
||||
.progress-status {
|
||||
text-align: center;
|
||||
font-size: 14px;
|
||||
color: #6b7280;
|
||||
margin-top: 10px;
|
||||
}
|
||||
|
||||
.progress-logs {
|
||||
margin-top: 12px;
|
||||
background: #1f2937;
|
||||
border-radius: 10px;
|
||||
padding: 12px 16px;
|
||||
max-height: 180px;
|
||||
overflow-y: auto;
|
||||
font-family: 'Menlo', 'Consolas', monospace;
|
||||
font-size: 12px;
|
||||
color: #d1d5db;
|
||||
}
|
||||
|
||||
.progress-logs .log-line {
|
||||
padding: 2px 0;
|
||||
opacity: 0.8;
|
||||
}
|
||||
|
||||
.progress-logs .log-line.info { color: #60a5fa; }
|
||||
.progress-logs .log-line.success { color: #34d399; }
|
||||
.progress-logs .log-line.warning { color: #fbbf24; }
|
||||
.progress-logs .log-line.error { color: #f87171; }
|
||||
|
||||
/* ===== 结果区域 ===== */
|
||||
.result-section {
|
||||
padding: 20px 0;
|
||||
border-top: 1px solid #e5e7eb;
|
||||
}
|
||||
|
||||
.result-header {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 10px;
|
||||
margin-bottom: 12px;
|
||||
}
|
||||
|
||||
.result-icon {
|
||||
font-size: 24px;
|
||||
}
|
||||
|
||||
.result-title {
|
||||
font-size: 18px;
|
||||
font-weight: 600;
|
||||
}
|
||||
|
||||
.result-files {
|
||||
display: flex;
|
||||
flex-wrap: wrap;
|
||||
gap: 8px;
|
||||
margin-bottom: 16px;
|
||||
}
|
||||
|
||||
.result-files .file-tag {
|
||||
background: #f3f4f6;
|
||||
padding: 4px 14px;
|
||||
border-radius: 20px;
|
||||
font-size: 13px;
|
||||
color: #374151;
|
||||
}
|
||||
|
||||
.result-actions {
|
||||
display: flex;
|
||||
gap: 12px;
|
||||
}
|
||||
|
||||
.result-actions button {
|
||||
padding: 10px 24px;
|
||||
border: none;
|
||||
border-radius: 10px;
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
cursor: pointer;
|
||||
transition: all 0.2s;
|
||||
}
|
||||
|
||||
.btn-download {
|
||||
background: #6366f1;
|
||||
color: #fff;
|
||||
}
|
||||
|
||||
.btn-download:hover {
|
||||
background: #4f46e5;
|
||||
transform: translateY(-1px);
|
||||
}
|
||||
|
||||
.btn-view {
|
||||
background: #f3f4f6;
|
||||
color: #374151;
|
||||
}
|
||||
|
||||
.btn-view:hover {
|
||||
background: #e5e7eb;
|
||||
}
|
||||
|
||||
.result-error {
|
||||
margin-top: 12px;
|
||||
padding: 12px 16px;
|
||||
background: #fef2f2;
|
||||
border-radius: 10px;
|
||||
border: 1px solid #fca5a5;
|
||||
color: #dc2626;
|
||||
}
|
||||
|
||||
/* ===== 响应式 ===== */
|
||||
@media (max-width: 600px) {
|
||||
.container {
|
||||
padding: 24px 16px;
|
||||
}
|
||||
|
||||
.config-section {
|
||||
grid-template-columns: 1fr;
|
||||
}
|
||||
|
||||
.upload-zone {
|
||||
padding: 32px 16px;
|
||||
}
|
||||
|
||||
.btn-generate {
|
||||
width: 100%;
|
||||
justify-content: center;
|
||||
}
|
||||
}
|
||||
|
||||
/* 查看代码仓库按钮 */
|
||||
.btn-view-repo {
|
||||
background: #10b981;
|
||||
color: #fff;
|
||||
}
|
||||
|
||||
.btn-view-repo:hover {
|
||||
background: #059669;
|
||||
transform: translateY(-1px);
|
||||
}
|
||||
|
|
@ -0,0 +1,496 @@
|
|||
// ============================================================
|
||||
// 状态管理
|
||||
// ============================================================
|
||||
const state = {
|
||||
file: null,
|
||||
taskId: null,
|
||||
status: 'idle',
|
||||
pollInterval: null,
|
||||
currentStatusName: '',
|
||||
repoURL: '',
|
||||
dotCount: 0,
|
||||
dotInterval: null,
|
||||
progressSimInterval: null,
|
||||
};
|
||||
|
||||
// ============================================================
|
||||
// DOM 引用
|
||||
// ============================================================
|
||||
const uploadZone = document.getElementById('uploadZone');
|
||||
const fileInput = document.getElementById('fileInput');
|
||||
const fileInfo = document.getElementById('fileInfo');
|
||||
const fileName = document.getElementById('fileName');
|
||||
const fileSize = document.getElementById('fileSize');
|
||||
const removeFile = document.getElementById('removeFile');
|
||||
|
||||
const sdkName = document.getElementById('sdkName');
|
||||
const llmModel = document.getElementById('llmModel');
|
||||
const apiKey = document.getElementById('apiKey');
|
||||
const baseUrl = document.getElementById('baseUrl');
|
||||
|
||||
const generateBtn = document.getElementById('generateBtn');
|
||||
const btnText = document.querySelector('.btn-text');
|
||||
const btnSpinner = document.querySelector('.btn-spinner');
|
||||
|
||||
const progressSection = document.getElementById('progressSection');
|
||||
const progressFill = document.getElementById('progressFill');
|
||||
const progressStatus = document.getElementById('progressStatus');
|
||||
const progressLogs = document.getElementById('progressLogs');
|
||||
|
||||
const resultSection = document.getElementById('resultSection');
|
||||
const resultFiles = document.getElementById('resultFiles');
|
||||
const resultError = document.getElementById('resultError');
|
||||
const downloadBtn = document.getElementById('downloadBtn');
|
||||
const viewRepoBtn = document.getElementById('viewRepoBtn');
|
||||
|
||||
// ============================================================
|
||||
// 动态效果函数
|
||||
// ============================================================
|
||||
|
||||
// 1. 进度条闪烁动画
|
||||
function addProgressGlow() {
|
||||
const existingStyle = document.getElementById('progressGlowStyle');
|
||||
if (existingStyle) return;
|
||||
|
||||
const style = document.createElement('style');
|
||||
style.id = 'progressGlowStyle';
|
||||
style.textContent = `
|
||||
.progress-fill.glow {
|
||||
background: linear-gradient(90deg, #6366f1, #8b5cf6, #6366f1);
|
||||
background-size: 200% 100%;
|
||||
animation: glowMove 1.5s ease-in-out infinite;
|
||||
position: relative;
|
||||
}
|
||||
.progress-fill.glow::after {
|
||||
content: '';
|
||||
position: absolute;
|
||||
right: -4px;
|
||||
top: -4px;
|
||||
width: 14px;
|
||||
height: 14px;
|
||||
background: #8b5cf6;
|
||||
border-radius: 50%;
|
||||
box-shadow: 0 0 20px 8px rgba(139, 92, 246, 0.6);
|
||||
animation: pulse 1s ease-in-out infinite;
|
||||
}
|
||||
@keyframes glowMove {
|
||||
0% { background-position: 0% 50%; }
|
||||
50% { background-position: 100% 50%; }
|
||||
100% { background-position: 0% 50%; }
|
||||
}
|
||||
@keyframes pulse {
|
||||
0%, 100% { transform: scale(1); opacity: 1; }
|
||||
50% { transform: scale(1.3); opacity: 0.6; }
|
||||
}
|
||||
`;
|
||||
document.head.appendChild(style);
|
||||
}
|
||||
|
||||
// 2. 状态文字动态点点点
|
||||
function startDotAnimation() {
|
||||
if (state.dotInterval) {
|
||||
clearInterval(state.dotInterval);
|
||||
}
|
||||
|
||||
let dotCount = 0;
|
||||
|
||||
state.dotInterval = setInterval(() => {
|
||||
dotCount = (dotCount + 1) % 4;
|
||||
const dots = '.'.repeat(dotCount);
|
||||
if (state.status === 'generating' || state.status === 'pending') {
|
||||
const currentText = progressStatus.textContent.replace(/\.\.\.$/, '').trim();
|
||||
if (!currentText.includes('✅') && !currentText.includes('❌')) {
|
||||
progressStatus.textContent = currentText + dots;
|
||||
}
|
||||
}
|
||||
}, 500);
|
||||
}
|
||||
|
||||
function stopDotAnimation() {
|
||||
if (state.dotInterval) {
|
||||
clearInterval(state.dotInterval);
|
||||
state.dotInterval = null;
|
||||
}
|
||||
}
|
||||
|
||||
// 3. 进度条模拟前进
|
||||
function startProgressSimulation() {
|
||||
if (state.progressSimInterval) {
|
||||
clearInterval(state.progressSimInterval);
|
||||
}
|
||||
|
||||
let lastPercent = parseInt(progressFill.style.width) || 0;
|
||||
state.progressSimInterval = setInterval(() => {
|
||||
if (state.status === 'generating' && lastPercent < 95) {
|
||||
const increment = 0.5 + Math.random() * 1.5;
|
||||
lastPercent = Math.min(lastPercent + increment, 95);
|
||||
progressFill.style.width = lastPercent + '%';
|
||||
}
|
||||
}, 2000);
|
||||
}
|
||||
|
||||
function stopProgressSimulation() {
|
||||
if (state.progressSimInterval) {
|
||||
clearInterval(state.progressSimInterval);
|
||||
state.progressSimInterval = null;
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 日志管理（只让最新的一条闪烁）
|
||||
// ============================================================
|
||||
let logCounter = 0;
|
||||
|
||||
function addLog(type, message) {
|
||||
const line = document.createElement('div');
|
||||
line.className = `log-line ${type}`;
|
||||
line.textContent = message;
|
||||
progressLogs.appendChild(line);
|
||||
progressLogs.scrollTop = progressLogs.scrollHeight;
|
||||
|
||||
// 移除之前所有日志的闪烁效果
|
||||
document.querySelectorAll('.log-line .log-dot-blink').forEach(el => {
|
||||
el.classList.remove('log-dot-blink');
|
||||
});
|
||||
|
||||
// 为最新日志添加闪烁点（在行首添加）
|
||||
const dotSpan = document.createElement('span');
|
||||
dotSpan.className = 'log-dot-blink';
|
||||
dotSpan.textContent = '● ';
|
||||
dotSpan.style.cssText = `
|
||||
color: #60a5fa;
|
||||
animation: logBlink 1s ease-in-out infinite;
|
||||
`;
|
||||
line.prepend(dotSpan);
|
||||
|
||||
// 确保动画样式存在
|
||||
const styleId = 'logBlinkStyle';
|
||||
if (!document.getElementById(styleId)) {
|
||||
const style = document.createElement('style');
|
||||
style.id = styleId;
|
||||
style.textContent = `
|
||||
@keyframes logBlink {
|
||||
0%, 100% { opacity: 1; transform: scale(1); }
|
||||
50% { opacity: 0.2; transform: scale(0.8); }
|
||||
}
|
||||
`;
|
||||
document.head.appendChild(style);
|
||||
}
|
||||
}
|
||||
|
||||
function clearLogs() {
|
||||
progressLogs.innerHTML = '';
|
||||
logCounter = 0;
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 上传区域事件
|
||||
// ============================================================
|
||||
uploadZone.addEventListener('click', () => fileInput.click());
|
||||
|
||||
uploadZone.addEventListener('dragover', (e) => {
|
||||
e.preventDefault();
|
||||
uploadZone.classList.add('dragover');
|
||||
});
|
||||
|
||||
uploadZone.addEventListener('dragleave', () => {
|
||||
uploadZone.classList.remove('dragover');
|
||||
});
|
||||
|
||||
uploadZone.addEventListener('drop', (e) => {
|
||||
e.preventDefault();
|
||||
uploadZone.classList.remove('dragover');
|
||||
if (e.dataTransfer.files.length > 0) {
|
||||
handleFile(e.dataTransfer.files[0]);
|
||||
}
|
||||
});
|
||||
|
||||
fileInput.addEventListener('change', (e) => {
|
||||
if (e.target.files.length > 0) {
|
||||
handleFile(e.target.files[0]);
|
||||
}
|
||||
});
|
||||
|
||||
removeFile.addEventListener('click', () => {
|
||||
state.file = null;
|
||||
fileInfo.style.display = 'none';
|
||||
uploadZone.style.display = 'block';
|
||||
generateBtn.disabled = true;
|
||||
fileInput.value = '';
|
||||
});
|
||||
|
||||
// ============================================================
|
||||
// 文件处理
|
||||
// ============================================================
|
||||
function handleFile(file) {
|
||||
const validTypes = ['.md', '.txt', '.doc', '.docx', '.pdf'];
|
||||
const ext = '.' + file.name.split('.').pop().toLowerCase();
|
||||
|
||||
if (!validTypes.includes(ext)) {
|
||||
alert('不支持的文件类型，请上传 .md .txt .doc .docx .pdf');
|
||||
return;
|
||||
}
|
||||
|
||||
if (file.size > 50 * 1024 * 1024) {
|
||||
alert('文件大小超过 50MB 限制');
|
||||
return;
|
||||
}
|
||||
|
||||
state.file = file;
|
||||
fileName.textContent = file.name;
|
||||
fileSize.textContent = formatFileSize(file.size);
|
||||
fileInfo.style.display = 'flex';
|
||||
uploadZone.style.display = 'none';
|
||||
generateBtn.disabled = false;
|
||||
}
|
||||
|
||||
function formatFileSize(bytes) {
|
||||
if (bytes < 1024) return bytes + ' B';
|
||||
if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(1) + ' KB';
|
||||
return (bytes / (1024 * 1024)).toFixed(1) + ' MB';
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 生成按钮
|
||||
// ============================================================
|
||||
generateBtn.addEventListener('click', generateSDK);
|
||||
|
||||
async function generateSDK() {
|
||||
if (!state.file) return;
|
||||
|
||||
const sdkNameVal = sdkName.value.trim();
|
||||
const modelVal = llmModel.value.trim();
|
||||
const apiKeyVal = apiKey.value.trim();
|
||||
const baseUrlVal = baseUrl.value.trim();
|
||||
|
||||
if (!sdkNameVal) {
|
||||
alert('请输入 SDK 名称');
|
||||
sdkName.focus();
|
||||
return;
|
||||
}
|
||||
if (!modelVal) {
|
||||
alert('请输入大模型名称');
|
||||
llmModel.focus();
|
||||
return;
|
||||
}
|
||||
if (!apiKeyVal) {
|
||||
alert('请输入 API Key');
|
||||
apiKey.focus();
|
||||
return;
|
||||
}
|
||||
if (!baseUrlVal) {
|
||||
alert('请输入 Base URL');
|
||||
baseUrl.focus();
|
||||
return;
|
||||
}
|
||||
|
||||
// 重置状态
|
||||
state.status = 'generating';
|
||||
state.taskId = null;
|
||||
state.currentStatusName = '';
|
||||
state.repoURL = '';
|
||||
resultSection.style.display = 'none';
|
||||
resultError.style.display = 'none';
|
||||
resultFiles.innerHTML = '';
|
||||
progressSection.style.display = 'block';
|
||||
progressFill.style.width = '0%';
|
||||
progressStatus.textContent = '准备中...';
|
||||
clearLogs();
|
||||
generateBtn.disabled = true;
|
||||
btnText.textContent = '生成中';
|
||||
btnSpinner.style.display = 'inline-block';
|
||||
|
||||
viewRepoBtn.style.display = 'none';
|
||||
|
||||
// 启动动态效果
|
||||
addProgressGlow();
|
||||
progressFill.classList.add('glow');
|
||||
startDotAnimation();
|
||||
startProgressSimulation();
|
||||
|
||||
const formData = new FormData();
|
||||
formData.append('file', state.file);
|
||||
formData.append('sdk_name', sdkNameVal);
|
||||
formData.append('llm_model', modelVal);
|
||||
formData.append('llm_api_key', apiKeyVal);
|
||||
formData.append('llm_base_url', baseUrlVal);
|
||||
|
||||
try {
|
||||
const response = await fetch('/api/v1/generate', {
|
||||
method: 'POST',
|
||||
body: formData,
|
||||
});
|
||||
|
||||
const data = await response.json();
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(data.error || '生成失败');
|
||||
}
|
||||
|
||||
state.taskId = data.task_id;
|
||||
addLog('info', '📤 任务已提交: ' + data.task_id);
|
||||
|
||||
pollTaskStatus();
|
||||
|
||||
} catch (err) {
|
||||
showError(err.message);
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 轮询任务状态
|
||||
// ============================================================
|
||||
function pollTaskStatus() {
|
||||
if (state.pollInterval) {
|
||||
clearInterval(state.pollInterval);
|
||||
}
|
||||
|
||||
state.pollInterval = setInterval(async () => {
|
||||
try {
|
||||
const response = await fetch(`/api/v1/tasks/${state.taskId}`);
|
||||
const data = await response.json();
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(data.error || '查询状态失败');
|
||||
}
|
||||
|
||||
updateProgress(data);
|
||||
|
||||
if (data.status === 'generateComplete' || data.status === 'completed') {
|
||||
clearInterval(state.pollInterval);
|
||||
state.pollInterval = null;
|
||||
onComplete(data);
|
||||
} else if (data.status === 'failed') {
|
||||
clearInterval(state.pollInterval);
|
||||
state.pollInterval = null;
|
||||
showError(data.error || '生成失败');
|
||||
}
|
||||
|
||||
} catch (err) {
|
||||
clearInterval(state.pollInterval);
|
||||
state.pollInterval = null;
|
||||
showError(err.message);
|
||||
}
|
||||
}, 2000);
|
||||
}
|
||||
|
||||
function updateProgress(data) {
|
||||
const percent = data.percent || 0;
|
||||
progressFill.style.width = percent + '%';
|
||||
|
||||
const statusName = data.status_name || '';
|
||||
if (statusName && statusName !== state.currentStatusName) {
|
||||
state.currentStatusName = statusName;
|
||||
addLog('info', '🔄 ' + statusName);
|
||||
}
|
||||
|
||||
// ✅ 更新状态文字（带闪烁前缀）
|
||||
const statusTextMap = {
|
||||
'pending': '⏳ 等待中',
|
||||
'generateCode': '🔄 生成代码',
|
||||
'validateCode': '🔍 验证代码',
|
||||
'generateComplete': '✅ 完成！',
|
||||
'completed': '✅ 完成！',
|
||||
'failed': '❌ 失败',
|
||||
};
|
||||
const baseText = statusTextMap[data.status] || (statusName || data.status);
|
||||
|
||||
// ✅ 只有进行中的状态才让文字闪烁
|
||||
const isProcessing = data.status !== 'generateComplete' && data.status !== 'completed' && data.status !== 'failed';
|
||||
if (isProcessing) {
|
||||
progressStatus.innerHTML = `<span class="status-blink">●</span> ${baseText}`;
|
||||
} else {
|
||||
progressStatus.textContent = baseText;
|
||||
stopDotAnimation();
|
||||
}
|
||||
|
||||
// ✅ 确保闪烁样式存在
|
||||
const styleId = 'statusBlinkStyle';
|
||||
if (!document.getElementById(styleId) && isProcessing) {
|
||||
const style = document.createElement('style');
|
||||
style.id = styleId;
|
||||
style.textContent = `
|
||||
.status-blink {
|
||||
color: #60a5fa;
|
||||
display: inline-block;
|
||||
animation: statusBlink 1s ease-in-out infinite;
|
||||
margin-right: 6px;
|
||||
}
|
||||
@keyframes statusBlink {
|
||||
0%, 100% { opacity: 1; transform: scale(1); }
|
||||
50% { opacity: 0.2; transform: scale(0.8); }
|
||||
}
|
||||
`;
|
||||
document.head.appendChild(style);
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 完成处理
|
||||
// ============================================================
|
||||
function onComplete(data) {
|
||||
stopDotAnimation();
|
||||
stopProgressSimulation();
|
||||
progressFill.classList.remove('glow');
|
||||
|
||||
generateBtn.disabled = false;
|
||||
btnText.textContent = '生成 SDK';
|
||||
btnSpinner.style.display = 'none';
|
||||
|
||||
resultSection.style.display = 'block';
|
||||
resultFiles.innerHTML = '';
|
||||
|
||||
const viewBtn = document.getElementById('viewBtn');
|
||||
if (viewBtn) {
|
||||
viewBtn.style.display = 'none';
|
||||
}
|
||||
|
||||
downloadBtn.onclick = () => {
|
||||
window.location.href = `/api/v1/tasks/${state.taskId}/download`;
|
||||
};
|
||||
|
||||
const repoURL = data.RepoURL || data.repo_url || '';
|
||||
if (repoURL) {
|
||||
viewRepoBtn.style.display = 'inline-block';
|
||||
viewRepoBtn.onclick = () => {
|
||||
window.open(repoURL, '_blank');
|
||||
};
|
||||
} else {
|
||||
viewRepoBtn.style.display = 'none';
|
||||
}
|
||||
|
||||
// ✅ 完成状态不带闪烁
|
||||
progressStatus.textContent = '✅ 生成完成！';
|
||||
progressFill.style.width = '100%';
|
||||
addLog('success', '🎉 SDK 生成完成！');
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 错误处理
|
||||
// ============================================================
|
||||
function showError(message) {
|
||||
stopDotAnimation();
|
||||
stopProgressSimulation();
|
||||
progressFill.classList.remove('glow');
|
||||
|
||||
state.status = 'error';
|
||||
generateBtn.disabled = false;
|
||||
btnText.textContent = '生成 SDK';
|
||||
btnSpinner.style.display = 'none';
|
||||
|
||||
resultSection.style.display = 'block';
|
||||
resultError.style.display = 'block';
|
||||
resultError.textContent = '❌ ' + message;
|
||||
|
||||
viewRepoBtn.style.display = 'none';
|
||||
|
||||
// ✅ 错误状态不带闪烁
|
||||
progressStatus.textContent = '❌ 失败';
|
||||
progressFill.style.width = '100%';
|
||||
progressFill.style.background = '#f87171';
|
||||
|
||||
if (state.pollInterval) {
|
||||
clearInterval(state.pollInterval);
|
||||
state.pollInterval = null;
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,94 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8" />
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||
<title>SDK 生成器</title>
|
||||
<link rel="stylesheet" href="/static/css/style.css" />
|
||||
</head>
|
||||
<body>
|
||||
|
||||
<div class="container">
|
||||
<!-- 头部 -->
|
||||
<header class="header">
|
||||
<h1>🚀 SDK 生成器</h1>
|
||||
<p class="subtitle">上传 API 文档，自动生成 Go SDK</p>
|
||||
</header>
|
||||
|
||||
<!-- 上传区域 -->
|
||||
<section class="upload-section">
|
||||
<div class="upload-zone" id="uploadZone">
|
||||
<div class="upload-icon">📄</div>
|
||||
<p class="upload-text">拖拽文档到此处，或点击上传</p>
|
||||
<p class="upload-hint">支持 .md .txt .doc .docx .pdf</p>
|
||||
<input type="file" id="fileInput" accept=".md,.txt,.doc,.docx,.pdf" />
|
||||
</div>
|
||||
|
||||
<div class="file-info" id="fileInfo" style="display:none;">
|
||||
<span class="file-name" id="fileName"></span>
|
||||
<span class="file-size" id="fileSize"></span>
|
||||
<button class="btn-remove" id="removeFile">✕</button>
|
||||
</div>
|
||||
</section>
|
||||
|
||||
<!-- 配置区域 -->
|
||||
<section class="config-section">
|
||||
<div class="form-group">
|
||||
<label for="sdkName">SDK 名称 <span class="required">*</span></label>
|
||||
<input type="text" id="sdkName" placeholder="例如：marketing-sdk" value="my-sdk" required />
|
||||
<small>将作为 Go 模块名和包名</small>
|
||||
</div>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="llmModel">大模型名称 <span class="required">*</span></label>
|
||||
<input type="text" id="llmModel" placeholder="例如：doubao-seed-evolving" value="doubao-seed-evolving" required />
|
||||
<small>请输入完整的大模型标识符</small>
|
||||
</div>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="apiKey">API Key <span class="required">*</span></label>
|
||||
<input type="password" id="apiKey" placeholder="请输入 API Key" required />
|
||||
</div>
|
||||
|
||||
<div class="form-group">
|
||||
<label for="baseUrl">Base URL <span class="required">*</span></label>
|
||||
<input type="text" id="baseUrl" placeholder="https://ark.cn-beijing.volces.com/api/v3" value="https://ark.cn-beijing.volces.com/api/v3" required />
|
||||
</div>
|
||||
</section>
|
||||
|
||||
<!-- 操作按钮 -->
|
||||
<section class="action-section">
|
||||
<button class="btn-generate" id="generateBtn" disabled>
|
||||
<span class="btn-text">生成 SDK</span>
|
||||
<span class="btn-spinner" style="display:none;">⟳</span>
|
||||
</button>
|
||||
</section>
|
||||
|
||||
<!-- 进度区域 -->
|
||||
<section class="progress-section" id="progressSection" style="display:none;">
|
||||
<div class="progress-bar">
|
||||
<div class="progress-fill" id="progressFill" style="width: 0%;"></div>
|
||||
</div>
|
||||
<div class="progress-status" id="progressStatus">准备中...</div>
|
||||
<div class="progress-logs" id="progressLogs"></div>
|
||||
</section>
|
||||
|
||||
<!-- 结果区域 -->
|
||||
<section class="result-section" id="resultSection" style="display:none;">
|
||||
<div class="result-header">
|
||||
<span class="result-icon">✅</span>
|
||||
<span class="result-title">生成完成！</span>
|
||||
</div>
|
||||
<!-- 文件列表隐藏 -->
|
||||
<div class="result-files" id="resultFiles" style="display:none;"></div>
|
||||
<div class="result-actions">
|
||||
<button class="btn-download" id="downloadBtn">📥 下载 SDK</button>
|
||||
<button class="btn-view-repo" id="viewRepoBtn" style="display:none;">📂 查看代码仓库</button>
|
||||
</div>
|
||||
<div class="result-error" id="resultError" style="display:none;"></div>
|
||||
</section>
|
||||
</div>
|
||||
|
||||
<script src="/static/js/app.js"></script>
|
||||
</body>
|
||||
</html>
|
||||
Loading…
Reference in New Issue