Upload 12 files

Dockerfile

@@ -1 +1,45 @@
-FROM linqiu1199/augment2api:latest
+# 使用官方 Golang 镜像作为构建环境
+FROM golang:1.23-alpine AS builder
+
+# 设置工作目录
+WORKDIR /app
+
+# 复制 go.mod 和 go.sum 文件
+COPY go.mod go.sum ./
+
+# 下载依赖
+RUN go mod download
+
+# 复制源代码
+COPY . .
+
+# 构建应用
+RUN CGO_ENABLED=0 GOOS=linux go build -o augment2api
+
+# 使用轻量级的 alpine 镜像
+FROM alpine:latest
+
+# 安装 ca-certificates 以支持 HTTPS
+RUN apk --no-cache add ca-certificates tzdata
+
+# 创建非 root 用户
+RUN adduser -D -g '' appuser
+
+# 从构建阶段复制二进制文件
+COPY --from=builder /app/augment2api /app/augment2api
+
+# 复制静态文件和模板
+COPY --from=builder /app/templates /app/templates
+
+# 设置工作目录
+WORKDIR /app
+
+# 使用非 root 用户运行
+USER appuser
+
+# 暴露端口
+EXPOSE 27080
+
+
+# 运行应用
+CMD ["/app/augment2api"] 

api/auth.go

@@ -0,0 +1,43 @@
+package api
+
+import (
+	"augment2api/config"
+	"augment2api/pkg/logger"
+	"fmt"
+	"net/http"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+)
+
+// AuthMiddleware 验证请求的Authorization header
+func AuthMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		// 如果未设置 AuthToken，则不启用鉴权
+		if config.AppConfig.AuthToken == "" {
+			c.Next()
+			return
+		}
+
+		authHeader := c.GetHeader("Authorization")
+		if authHeader == "" {
+			logger.Log.Error("Authorization is empty")
+			c.JSON(http.StatusUnauthorized, gin.H{"error": "Authorization header is required"})
+			c.Abort()
+			return
+		}
+
+		// 支持 "Bearer <token>" 格式
+		token := strings.TrimPrefix(authHeader, "Bearer ")
+		token = strings.TrimSpace(token)
+
+		if token != config.AppConfig.AuthToken {
+			logger.Log.Error(fmt.Sprintf("Invalid authorization token:%s", token))
+			c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid authorization token"})
+			c.Abort()
+			return
+		}
+
+		c.Next()
+	}
+}

api/handler.go

@@ -0,0 +1,511 @@
+package api
+
+import (
+	"augment2api/config"
+	"bufio"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+// OpenAIRequest OpenAI兼容的请求结构
+type OpenAIRequest struct {
+	Model       string        `json:"model,omitempty"`
+	Messages    []ChatMessage `json:"messages,omitempty"`
+	Stream      bool          `json:"stream,omitempty"`
+	Temperature float64       `json:"temperature,omitempty"`
+	MaxTokens   int           `json:"max_tokens,omitempty"`
+}
+
+// OpenAIResponse OpenAI兼容的响应结构
+type OpenAIResponse struct {
+	ID      string   `json:"id"`
+	Object  string   `json:"object"`
+	Created int64    `json:"created"`
+	Model   string   `json:"model"`
+	Choices []Choice `json:"choices"`
+	Usage   Usage    `json:"usage"`
+}
+
+// OpenAIStreamResponse OpenAI兼容的流式响应结构
+type OpenAIStreamResponse struct {
+	ID      string         `json:"id"`
+	Object  string         `json:"object"`
+	Created int64          `json:"created"`
+	Model   string         `json:"model"`
+	Choices []StreamChoice `json:"choices"`
+}
+
+type StreamChoice struct {
+	Index        int         `json:"index"`
+	Delta        ChatMessage `json:"delta"`
+	FinishReason *string     `json:"finish_reason"`
+}
+
+type Choice struct {
+	Index        int         `json:"index"`
+	Message      ChatMessage `json:"message"`
+	FinishReason *string     `json:"finish_reason"`
+}
+
+type ChatMessage struct {
+	Role    string      `json:"role"`
+	Content interface{} `json:"content"`
+}
+
+// GetContent 添加一个辅助方法来获取消息内容
+func (m ChatMessage) GetContent() string {
+	switch v := m.Content.(type) {
+	case string:
+		return v
+	case []interface{}:
+		var result string
+		for _, item := range v {
+			if contentMap, ok := item.(map[string]interface{}); ok {
+				if text, exists := contentMap["text"]; exists {
+					if textStr, ok := text.(string); ok {
+						result += textStr
+					}
+				}
+			}
+		}
+		return result
+	default:
+		return ""
+	}
+}
+
+type Usage struct {
+	PromptTokens     int `json:"prompt_tokens"`
+	CompletionTokens int `json:"completion_tokens"`
+	TotalTokens      int `json:"total_tokens"`
+}
+
+// AugmentRequest Augment API请求结构
+type AugmentRequest struct {
+	ChatHistory []AugmentChatHistory `json:"chat_history"`
+	Message     string               `json:"message"`
+	Mode        string               `json:"mode"`
+}
+
+type AugmentChatHistory struct {
+	ResponseText   string `json:"response_text"`
+	RequestMessage string `json:"request_message"`
+}
+
+// AugmentResponse Augment API响应结构
+type AugmentResponse struct {
+	Text string `json:"text"`
+	Done bool   `json:"done"`
+}
+
+// CodeResponse 用于解析从授权服务返回的代码
+type CodeResponse struct {
+	Code      string `json:"code"`
+	State     string `json:"state"`
+	TenantURL string `json:"tenant_url"`
+}
+
+// ModelObject OpenAI模型对象结构
+type ModelObject struct {
+	ID      string `json:"id"`
+	Object  string `json:"object"`
+	Created int    `json:"created"`
+	OwnedBy string `json:"owned_by"`
+}
+
+// ModelsResponse OpenAI模型列表响应结构
+type ModelsResponse struct {
+	Object string        `json:"object"`
+	Data   []ModelObject `json:"data"`
+}
+
+// 全局变量
+var (
+	accessToken string
+	tenantURL   string
+)
+
+// SetAuthInfo 设置认证信息
+func SetAuthInfo(token, tenant string) {
+	accessToken = token
+	tenantURL = tenant
+}
+
+// GetAuthInfo 获取认证信息
+func GetAuthInfo() (string, string) {
+	if config.AppConfig.CodingMode == "true" {
+		// 调试模式
+		return config.AppConfig.CodingToken, config.AppConfig.TenantURL
+	}
+
+	// 随机获取一个token
+	token, tenantURL := GetRandomToken()
+	if token != "" && tenantURL != "" {
+		return token, tenantURL
+	}
+
+	// 如果没有可用的token，则使用内存中的token
+	return accessToken, tenantURL
+}
+
+// convertToAugmentRequest 将OpenAI请求转换为Augment请求
+func convertToAugmentRequest(req OpenAIRequest) AugmentRequest {
+	augmentReq := AugmentRequest{
+		Mode: "CHAT",
+	}
+
+	if len(req.Messages) > 0 {
+		lastMsg := req.Messages[len(req.Messages)-1]
+		augmentReq.Message = lastMsg.GetContent()
+	}
+
+	var history []AugmentChatHistory
+	for i := 0; i < len(req.Messages)-1; i += 2 {
+		if i+1 < len(req.Messages) {
+			history = append(history, AugmentChatHistory{
+				RequestMessage: req.Messages[i].GetContent(),
+				ResponseText:   req.Messages[i+1].GetContent(),
+			})
+		}
+	}
+
+	augmentReq.ChatHistory = history
+	return augmentReq
+}
+
+// AuthHandler 处理授权请求
+func AuthHandler(c *gin.Context, authorizeURL string) {
+	c.JSON(http.StatusOK, gin.H{
+		"authorize_url": authorizeURL,
+	})
+}
+
+// CallbackHandler 处理回调请求
+func CallbackHandler(c *gin.Context, getAccessTokenFunc func(string, string, string) (string, error)) {
+	// 1. 解析请求数据
+	var codeResp CodeResponse
+	if err := c.ShouldBindJSON(&codeResp); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "无效的请求数据"})
+		return
+	}
+
+	// 2. 使用授权码获取访问令牌
+	token, err := getAccessTokenFunc(codeResp.TenantURL, "", codeResp.Code)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	// 3. 保存令牌和租户URL
+	SetAuthInfo(token, codeResp.TenantURL)
+
+	// 4. 保存到Redis
+	if err := SaveTokenToRedis(token, codeResp.TenantURL); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "保存token到Redis失败: " + err.Error()})
+		return
+	}
+
+	// 5. 返回成功响应
+	c.JSON(http.StatusOK, gin.H{
+		"status": "success",
+		"token":  token,
+	})
+}
+
+// ModelsHandler 处理模型请求
+func ModelsHandler(c *gin.Context) {
+	// 创建符合OpenAI格式的模型列表响应
+	response := ModelsResponse{
+		Object: "list",
+		Data: []ModelObject{
+			{
+				ID:      "claude-3-7-sonnet-20250219",
+				Object:  "model",
+				Created: 1708387201,
+				OwnedBy: "anthropic",
+			},
+			{
+				ID:      "claude-3.7",
+				Object:  "model",
+				Created: 1708387200,
+				OwnedBy: "anthropic",
+			},
+		},
+	}
+
+	c.JSON(http.StatusOK, response)
+}
+
+// ChatCompletionsHandler 处理聊天完成请求
+func ChatCompletionsHandler(c *gin.Context) {
+	token, tenant := GetAuthInfo()
+	if token == "" || tenant == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "无可用Token,请先在管理页面获取"})
+		return
+	}
+
+	var openAIReq OpenAIRequest
+	if err := c.ShouldBindJSON(&openAIReq); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "无效的请求数据"})
+		return
+	}
+
+	// 转换为Augment请求格式
+	augmentReq := convertToAugmentRequest(openAIReq)
+
+	// 处理流式请求
+	if openAIReq.Stream {
+		handleStreamRequest(c, augmentReq, openAIReq.Model)
+		return
+	}
+
+	// 处理非流式请求
+	handleNonStreamRequest(c, augmentReq, openAIReq.Model)
+}
+
+// 处理流式请求
+func handleStreamRequest(c *gin.Context, augmentReq AugmentRequest, model string) {
+	c.Header("Content-Type", "text/event-stream")
+	c.Header("Cache-Control", "no-cache")
+	c.Header("Connection", "keep-alive")
+
+	// 获取token和tenant_url
+	token, tenant := GetAuthInfo()
+	if token == "" || tenant == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "无可用Token,请先在管理页面获取"})
+		return
+	}
+
+	// 准备请求数据
+	jsonData, err := json.Marshal(augmentReq)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "序列化请求失败"})
+		return
+	}
+
+	// 创建请求 - 使用获取到的tenant_url
+	req, err := http.NewRequest("POST", tenant+"chat-stream", strings.NewReader(string(jsonData)))
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "创建请求失败"})
+		return
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+token) // 使用获取到的token
+
+	// 发送请求
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "请求失败: " + err.Error()})
+		return
+	}
+	defer resp.Body.Close()
+
+	// 设置刷新器以确保数据立即发送
+	flusher, ok := c.Writer.(http.Flusher)
+	if !ok {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "流式传输不支持"})
+		return
+	}
+
+	// 读取并转发响应
+	reader := bufio.NewReader(resp.Body)
+	responseID := fmt.Sprintf("chatcmpl-%d", time.Now().Unix())
+
+	var fullText string
+	for {
+		line, err := reader.ReadString('\n')
+		if err != nil {
+			if err == io.EOF {
+				break
+			}
+			log.Printf("读取响应失败: %v", err)
+			break
+		}
+
+		line = strings.TrimSpace(line)
+		if line == "" {
+			continue
+		}
+
+		var augmentResp AugmentResponse
+		if err := json.Unmarshal([]byte(line), &augmentResp); err != nil {
+			log.Printf("解析响应失败: %v", err)
+			continue
+		}
+
+		fullText += augmentResp.Text
+
+		// 创建OpenAI兼容的流式响应
+		streamResp := OpenAIStreamResponse{
+			ID:      responseID,
+			Object:  "chat.completion.chunk",
+			Created: time.Now().Unix(),
+			Model:   model,
+			Choices: []StreamChoice{
+				{
+					Index: 0,
+					Delta: ChatMessage{
+						Role:    "assistant",
+						Content: augmentResp.Text,
+					},
+					FinishReason: nil,
+				},
+			},
+		}
+
+		// 如果是最后一条消息，设置完成原因
+		if augmentResp.Done {
+			finishReason := "stop"
+			streamResp.Choices[0].FinishReason = &finishReason
+		}
+
+		// 序列化并发送响应
+		jsonResp, err := json.Marshal(streamResp)
+		if err != nil {
+			log.Printf("序列化响应失败: %v", err)
+			continue
+		}
+
+		fmt.Fprintf(c.Writer, "data: %s\n\n", jsonResp)
+		flusher.Flush()
+
+		// 如果完成，发送最后的[DONE]标记
+		if augmentResp.Done {
+			fmt.Fprintf(c.Writer, "data: [DONE]\n\n")
+			flusher.Flush()
+			break
+		}
+	}
+}
+
+// estimateTokenCount 粗略估计文本中的token数量
+// 这是一个简单的估算方法，实际token数量取决于具体的分词算法
+func estimateTokenCount(text string) int {
+	// 英文单词和标点符号大约是1个token
+	// 中文字符大约是1.5个token（每个字符约为0.75个token）
+	// 按空格分割英文单词
+	words := strings.Fields(text)
+	wordCount := len(words)
+
+	// 计算中文字符数量
+	chineseCount := 0
+	for _, r := range text {
+		if r >= 0x4E00 && r <= 0x9FFF {
+			chineseCount++
+		}
+	}
+
+	// 粗略估计：英文单词按1个token计算，中文字符按0.75个token计算
+	return wordCount + int(float64(chineseCount)*0.75)
+}
+
+// 处理非流式请求
+func handleNonStreamRequest(c *gin.Context, augmentReq AugmentRequest, model string) {
+	// 获取token和tenant_url
+	token, tenant := GetAuthInfo()
+	if token == "" || tenant == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "无可用Token,请先在管理页面获取"})
+		return
+	}
+
+	// 准备请求数据
+	jsonData, err := json.Marshal(augmentReq)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "序列化请求失败"})
+		return
+	}
+
+	// 创建请求 - 使用获取到的tenant_url
+	req, err := http.NewRequest("POST", tenant+"chat-stream", strings.NewReader(string(jsonData)))
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "创建请求失败"})
+		return
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+token) // 使用获取到的token
+
+	// 发送请求
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "请求失败: " + err.Error()})
+		return
+	}
+	defer resp.Body.Close()
+
+	// 读取完整响应
+	reader := bufio.NewReader(resp.Body)
+	var fullText string
+
+	for {
+		line, err := reader.ReadString('\n')
+		if err != nil {
+			if err == io.EOF {
+				break
+			}
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "读取响应失败: " + err.Error()})
+			return
+		}
+
+		line = strings.TrimSpace(line)
+		if line == "" {
+			continue
+		}
+
+		var augmentResp AugmentResponse
+		if err := json.Unmarshal([]byte(line), &augmentResp); err != nil {
+			continue
+		}
+
+		fullText += augmentResp.Text
+
+		if augmentResp.Done {
+			break
+		}
+	}
+
+	// 创建OpenAI兼容的响应
+	finishReason := "stop"
+
+	// 估算token数量
+	promptTokens := estimateTokenCount(augmentReq.Message)
+	for _, history := range augmentReq.ChatHistory {
+		promptTokens += estimateTokenCount(history.RequestMessage)
+		promptTokens += estimateTokenCount(history.ResponseText)
+	}
+	completionTokens := estimateTokenCount(fullText)
+
+	openAIResp := OpenAIResponse{
+		ID:      fmt.Sprintf("chatcmpl-%d", time.Now().Unix()),
+		Object:  "chat.completion",
+		Created: time.Now().Unix(),
+		Model:   model,
+		Choices: []Choice{
+			{
+				Index: 0,
+				Message: ChatMessage{
+					Role:    "assistant",
+					Content: fullText,
+				},
+				FinishReason: &finishReason,
+			},
+		},
+		Usage: Usage{
+			PromptTokens:     promptTokens,
+			CompletionTokens: completionTokens,
+			TotalTokens:      promptTokens + completionTokens,
+		},
+	}
+
+	c.JSON(http.StatusOK, openAIResp)
+}

api/token_handler.go

@@ -0,0 +1,183 @@
+package api
+
+import (
+	"augment2api/config"
+	"math/rand"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+// TokenInfo 存储token信息
+type TokenInfo struct {
+	Token     string `json:"token"`
+	TenantURL string `json:"tenant_url"`
+}
+
+// GetRedisTokenHandler 从Redis获取token列表
+func GetRedisTokenHandler(c *gin.Context) {
+	// 获取所有token的key (使用通配符模式)
+	keys, err := config.RedisKeys("token:*")
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"status": "error",
+			"error":  "获取token列表失败: " + err.Error(),
+		})
+		return
+	}
+
+	// 如果没有token
+	if len(keys) == 0 {
+		c.JSON(http.StatusOK, gin.H{
+			"status": "success",
+			"tokens": []TokenInfo{},
+		})
+		return
+	}
+
+	// 构建token列表
+	var tokenList []TokenInfo
+	for _, key := range keys {
+		// 从key中提取token (格式: "token:{token}")
+		token := key[6:] // 去掉前缀 "token:"
+
+		// 获取对应的tenant_url
+		tenantURL, err := config.RedisHGet(key, "tenant_url")
+		if err != nil {
+			continue // 跳过无效的token
+		}
+
+		tokenList = append(tokenList, TokenInfo{
+			Token:     token,
+			TenantURL: tenantURL,
+		})
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"status": "success",
+		"tokens": tokenList,
+	})
+}
+
+// SaveTokenToRedis 保存token到Redis
+func SaveTokenToRedis(token, tenantURL string) error {
+	// 创建一个唯一的key，包含token和tenant_url
+	tokenKey := "token:" + token
+
+	// 将tenant_url存储在token对应的哈希表中
+	return config.RedisHSet(tokenKey, "tenant_url", tenantURL)
+}
+
+// GetRandomToken 从Redis中随机获取一个token
+func GetRandomToken() (string, string) {
+	// 获取所有token的key
+	keys, err := config.RedisKeys("token:*")
+	if err != nil || len(keys) == 0 {
+		return "", ""
+	}
+
+	// 随机选择一个token
+	randomIndex := rand.Intn(len(keys))
+	randomKey := keys[randomIndex]
+
+	// 从key中提取token
+	token := randomKey[6:] // 去掉前缀 "token:"
+
+	// 获取对应的tenant_url
+	tenantURL, err := config.RedisHGet(randomKey, "tenant_url")
+	if err != nil {
+		return "", ""
+	}
+
+	return token, tenantURL
+}
+
+// DeleteTokenHandler 删除指定的token
+func DeleteTokenHandler(c *gin.Context) {
+	token := c.Param("token")
+	if token == "" {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"status": "error",
+			"error":  "未指定token",
+		})
+		return
+	}
+
+	tokenKey := "token:" + token
+
+	// 检查token是否存在
+	exists, err := config.RedisExists(tokenKey)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"status": "error",
+			"error":  "检查token失败: " + err.Error(),
+		})
+		return
+	}
+
+	if !exists {
+		c.JSON(http.StatusNotFound, gin.H{
+			"status": "error",
+			"error":  "token不存在",
+		})
+		return
+	}
+
+	// 删除token
+	if err := config.RedisDel(tokenKey); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"status": "error",
+			"error":  "删除token失败: " + err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"status": "success",
+	})
+}
+
+// UseTokenHandler 设置指定的token为当前活跃token
+func UseTokenHandler(c *gin.Context) {
+	token := c.Param("token")
+	if token == "" {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"status": "error",
+			"error":  "未指定token",
+		})
+		return
+	}
+
+	tokenKey := "token:" + token
+
+	// 检查token是否存在
+	exists, err := config.RedisExists(tokenKey)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"status": "error",
+			"error":  "检查token失败: " + err.Error(),
+		})
+		return
+	}
+
+	if !exists {
+		c.JSON(http.StatusNotFound, gin.H{
+			"status": "error",
+			"error":  "token不存在",
+		})
+		return
+	}
+
+	// 设置当前活跃token
+	if err := config.RedisSet("current_token", token, 0); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"status": "error",
+			"error":  "设置当前token失败: " + err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"status": "success",
+	})
+}

config/config.go

@@ -0,0 +1,50 @@
+package config
+
+import (
+	"augment2api/pkg/logger"
+	"os"
+)
+
+type Config struct {
+	RedisConnString string
+	AuthToken       string
+	CodingMode      string
+	CodingToken     string
+	TenantURL       string
+}
+
+var AppConfig Config
+
+func InitConfig() error {
+	// 从环境变量读取配置
+	AppConfig = Config{
+		// 必填配置
+		RedisConnString: getEnv("REDIS_CONN_STRING", ""),
+		// 非必填配置
+		AuthToken:   getEnv("AUTH_TOKEN", ""),
+		CodingMode:  getEnv("CODING_MODE", "false"),
+		CodingToken: getEnv("CODING_TOKEN", ""),
+		TenantURL:   getEnv("TENANT_URL", ""),
+	}
+
+	// redis连接字符串 示例: redis://default:pwd@localhost:6379
+	if AppConfig.RedisConnString == "" {
+		logger.Log.Fatalln("未配置环境变量 REDIS_CONN_STRING")
+	}
+
+	logger.Log.Info("Augment2Api配置加载完成:\n" +
+		"----------------------------------------\n" +
+		"AuthToken:    " + AppConfig.AuthToken + "\n" +
+		"RedisConnString: " + AppConfig.RedisConnString + "\n" +
+		"----------------------------------------")
+
+	return nil
+}
+
+func getEnv(key, defaultValue string) string {
+	value := os.Getenv(key)
+	if value == "" {
+		return defaultValue
+	}
+	return value
+}

config/redis.go

@@ -0,0 +1,94 @@
+package config
+
+import (
+	"augment2api/pkg/logger"
+	"context"
+	"os"
+	"time"
+
+	"github.com/go-redis/redis/v8"
+)
+
+var RDB redis.Cmdable
+
+// InitRedisClient This function is called after init()
+func InitRedisClient() (err error) {
+
+	RedisConnString := AppConfig.RedisConnString
+	if RedisConnString == "" {
+		logger.Log.Debug("REDIS_CONN_STRING not set, Redis is not enabled")
+		return nil
+	}
+
+	opt, err := redis.ParseURL(RedisConnString)
+	if err != nil {
+		logger.Log.Fatalln("failed to parse Redis connection string: " + err.Error())
+	}
+	RDB = redis.NewClient(opt)
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	_, err = RDB.Ping(ctx).Result()
+	if err != nil {
+		logger.Log.Fatalln("Redis ping test failed: " + err.Error())
+	}
+	return err
+}
+
+func ParseRedisOption() *redis.Options {
+	opt, err := redis.ParseURL(os.Getenv("REDIS_CONN_STRING"))
+	if err != nil {
+		logger.Log.Fatalln("failed to parse Redis connection string: " + err.Error())
+	}
+	return opt
+}
+
+func RedisSet(key string, value string, expiration time.Duration) error {
+	ctx := context.Background()
+	return RDB.Set(ctx, key, value, expiration).Err()
+}
+
+func RedisGet(key string) (string, error) {
+	ctx := context.Background()
+	return RDB.Get(ctx, key).Result()
+}
+
+func RedisDel(key string) error {
+	ctx := context.Background()
+	return RDB.Del(ctx, key).Err()
+}
+
+// RedisHSet 设置哈希表字段值
+func RedisHSet(key, field, value string) error {
+	ctx := context.Background()
+	return RDB.HSet(ctx, key, field, value).Err()
+}
+
+// RedisHGet 获取哈希表字段值
+func RedisHGet(key, field string) (string, error) {
+	ctx := context.Background()
+	return RDB.HGet(ctx, key, field).Result()
+}
+
+// RedisExpire 设置键的过期时间
+func RedisExpire(key string, expiration time.Duration) error {
+	ctx := context.Background()
+	return RDB.Expire(ctx, key, expiration).Err()
+}
+
+// RedisKeys 获取匹配指定模式的所有键
+func RedisKeys(pattern string) ([]string, error) {
+	ctx := context.Background()
+	return RDB.Keys(ctx, pattern).Result()
+}
+
+// RedisExists 检查键是否存在
+func RedisExists(key string) (bool, error) {
+	ctx := context.Background()
+	result, err := RDB.Exists(ctx, key).Result()
+	if err != nil {
+		return false, err
+	}
+	return result > 0, nil
+}

go.mod

@@ -0,0 +1,44 @@
+module augment2api
+
+go 1.23.0
+
+toolchain go1.23.6
+
+require (
+	github.com/gin-contrib/cors v1.7.4
+	github.com/gin-gonic/gin v1.10.0
+	github.com/go-redis/redis/v8 v8.11.5
+	github.com/sirupsen/logrus v1.9.3
+)
+
+require (
+	github.com/bytedance/sonic v1.12.6 // indirect
+	github.com/bytedance/sonic/loader v0.2.1 // indirect
+	github.com/cespare/xxhash/v2 v2.1.2 // indirect
+	github.com/cloudwego/base64x v0.1.4 // indirect
+	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
+	github.com/gabriel-vasile/mimetype v1.4.7 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-playground/validator/v10 v10.23.0 // indirect
+	github.com/goccy/go-json v0.10.4 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.9 // indirect
+	github.com/kr/text v0.2.0 // indirect
+	github.com/leodido/go-urn v1.4.0 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
+	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
+	github.com/ugorji/go/codec v1.2.12 // indirect
+	golang.org/x/arch v0.12.0 // indirect
+	golang.org/x/crypto v0.36.0 // indirect
+	golang.org/x/net v0.37.0 // indirect
+	golang.org/x/sys v0.31.0 // indirect
+	golang.org/x/text v0.23.0 // indirect
+	google.golang.org/protobuf v1.36.1 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,116 @@
+github.com/bytedance/sonic v1.12.6 h1:/isNmCUF2x3Sh8RAp/4mh4ZGkcFAX/hLrzrK3AvpRzk=
+github.com/bytedance/sonic v1.12.6/go.mod h1:B8Gt/XvtZ3Fqj+iSKMypzymZxw/FVwgIGKzMzT9r/rk=
+github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/bytedance/sonic/loader v0.2.1 h1:1GgorWTqf12TA8mma4DDSbaQigE2wOgQo7iCjjJv3+E=
+github.com/bytedance/sonic/loader v0.2.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE=
+github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
+github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
+github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
+github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
+github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
+github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+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/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
+github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
+github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
+github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
+github.com/gabriel-vasile/mimetype v1.4.7 h1:SKFKl7kD0RiPdbht0s7hFtjl489WcQ1VyPW8ZzUMYCA=
+github.com/gabriel-vasile/mimetype v1.4.7/go.mod h1:GDlAgAyIRT27BhFl53XNAFtfjzOkLaF35JdEG0P7LtU=
+github.com/gin-contrib/cors v1.7.4 h1:/fC6/wk7rCRtqKqki8lLr2Xq+hnV49aXDLIuSek9g4k=
+github.com/gin-contrib/cors v1.7.4/go.mod h1:vGc/APSgLMlQfEJV5NAzkrAHb0C8DetL3K6QZuvGii0=
+github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
+github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
+github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
+github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
+github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
+github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
+github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
+github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
+github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
+github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
+github.com/go-playground/validator/v10 v10.23.0 h1:/PwmTwZhS0dPkav3cdK9kV1FsAmrL8sThn8IHr/sO+o=
+github.com/go-playground/validator/v10 v10.23.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
+github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=
+github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
+github.com/goccy/go-json v0.10.4 h1:JSwxQzIqKfmFX1swYPpUThQZp/Ka4wzJdK0LWVytLPM=
+github.com/goccy/go-json v0.10.4/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
+github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
+github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.2.9 h1:66ze0taIn2H33fBvCkXuv9BmCwDfafmiIVpKV9kKGuY=
+github.com/klauspost/cpuid/v2 v2.2.9/go.mod h1:rqkxqrZ1EhYM9G+hXH7YdowN5R5RGN6NK4QwQ3WMXF8=
+github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
+github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
+github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
+github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
+github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
+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/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
+github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=
+github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
+github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=
+github.com/onsi/ginkgo v1.16.5/go.mod h1:+E8gABHa3K6zRBolWtd+ROzc/U5bkGt0FwiG042wbpU=
+github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=
+github.com/onsi/gomega v1.18.1/go.mod h1:0q+aL8jAiMXy9hbwj2mr5GziHiwhAIQpFmmtT5hitRs=
+github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
+github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
+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/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
+github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
+github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
+github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
+github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
+github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
+github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+golang.org/x/arch v0.12.0 h1:UsYJhbzPYGsT0HbEdmYcqtCv8UNGvnaL561NnIUvaKg=
+golang.org/x/arch v0.12.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+golang.org/x/crypto v0.36.0 h1:AnAEvhDddvBdpY+uR+MyHmuZzzNqXSe/GvuDeob5L34=
+golang.org/x/crypto v0.36.0/go.mod h1:Y4J0ReaxCR1IMaabaSMugxJES1EpwhBHhv2bDHklZvc=
+golang.org/x/net v0.37.0 h1:1zLorHbz+LYj7MQlSf1+2tPIIgibq2eL5xkrGk6f+2c=
+golang.org/x/net v0.37.0/go.mod h1:ivrbrMbzFq5J41QOQh0siUuly180yBYtLp+CKbEaFx8=
+golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.31.0 h1:ioabZlmFYtWhL+TRYpcnNlLwhyxaM9kWTDEmfnprqik=
+golang.org/x/sys v0.31.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
+golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY=
+golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk=
+google.golang.org/protobuf v1.36.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
+gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=

main.go

@@ -0,0 +1,212 @@
+package main
+
+import (
+	"augment2api/api"
+	"augment2api/config"
+	"augment2api/middleware"
+	"augment2api/pkg/logger"
+	"crypto/rand"
+	"crypto/sha256"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+const clientID = "v"
+
+// OAuthState 存储OAuth状态信息
+type OAuthState struct {
+	CodeVerifier  string    `json:"code_verifier"`
+	CodeChallenge string    `json:"code_challenge"`
+	State         string    `json:"state"`
+	CreationTime  time.Time `json:"creation_time"`
+}
+
+// 全局变量存储OAuth状态
+var (
+	globalOAuthState OAuthState
+)
+
+// base64URLEncode 编码Buffer为base64 URL安全格式
+func base64URLEncode(data []byte) string {
+	encoded := base64.StdEncoding.EncodeToString(data)
+	encoded = strings.ReplaceAll(encoded, "+", "-")
+	encoded = strings.ReplaceAll(encoded, "/", "_")
+	encoded = strings.ReplaceAll(encoded, "=", "")
+	return encoded
+}
+
+// sha256Hash 计算SHA256哈希
+func sha256Hash(input []byte) []byte {
+	hash := sha256.Sum256(input)
+	return hash[:]
+}
+
+// createOAuthState 创建OAuth状态
+func createOAuthState() OAuthState {
+	codeVerifierBytes := make([]byte, 32)
+	_, err := rand.Read(codeVerifierBytes)
+	if err != nil {
+		log.Fatalf("生成随机字节失败: %v", err)
+	}
+
+	codeVerifier := base64URLEncode(codeVerifierBytes)
+	codeChallenge := base64URLEncode(sha256Hash([]byte(codeVerifier)))
+
+	stateBytes := make([]byte, 8)
+	_, err = rand.Read(stateBytes)
+	if err != nil {
+		log.Fatalf("生成随机状态失败: %v", err)
+	}
+	state := base64URLEncode(stateBytes)
+
+	return OAuthState{
+		CodeVerifier:  codeVerifier,
+		CodeChallenge: codeChallenge,
+		State:         state,
+		CreationTime:  time.Now(),
+	}
+}
+
+// generateAuthorizeURL 生成授权URL
+func generateAuthorizeURL(oauthState OAuthState) string {
+	params := url.Values{}
+	params.Add("response_type", "code")
+	params.Add("code_challenge", oauthState.CodeChallenge)
+	params.Add("client_id", clientID)
+	params.Add("state", oauthState.State)
+	params.Add("prompt", "login")
+
+	authorizeURL := fmt.Sprintf("https://auth.augmentcode.com/authorize?%s", params.Encode())
+	return authorizeURL
+}
+
+// getAccessToken 获取访问令牌
+func getAccessToken(tenantURL, codeVerifier, code string) (string, error) {
+	data := map[string]string{
+		"grant_type":    "authorization_code",
+		"client_id":     clientID,
+		"code_verifier": codeVerifier,
+		"redirect_uri":  "",
+		"code":          code,
+	}
+
+	jsonData, err := json.Marshal(data)
+	if err != nil {
+		return "", fmt.Errorf("序列化数据失败: %v", err)
+	}
+
+	resp, err := http.Post(tenantURL+"token", "application/json", strings.NewReader(string(jsonData)))
+	if err != nil {
+		return "", fmt.Errorf("请求令牌失败: %v", err)
+	}
+	defer resp.Body.Close()
+
+	var result map[string]interface{}
+	if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
+		return "", fmt.Errorf("解析响应失败: %v", err)
+	}
+
+	token, ok := result["access_token"].(string)
+	if !ok {
+		return "", fmt.Errorf("响应中没有访问令牌")
+	}
+
+	return token, nil
+}
+
+// 初始化路由
+func setupRouter() *gin.Engine {
+	r := gin.Default()
+
+	// 跨域
+	r.Use(middleware.CORS())
+
+	// 初始化OAuth状态
+	globalOAuthState = createOAuthState()
+
+	// 静态文件服务
+	r.Static("/static", "./static")
+	r.LoadHTMLGlob("templates/*")
+
+	r.GET("/", func(c *gin.Context) {
+		c.HTML(http.StatusOK, "admin.html", gin.H{})
+	})
+
+	// 授权端点
+	r.GET("/auth", func(c *gin.Context) {
+		authorizeURL := generateAuthorizeURL(globalOAuthState)
+		api.AuthHandler(c, authorizeURL)
+	})
+
+	// 获取token
+	r.GET("/api/tokens", api.GetRedisTokenHandler)
+
+	// 删除token
+	r.DELETE("/api/token/:token", api.DeleteTokenHandler)
+
+	// 回调端点，用于处理授权码
+	r.POST("/callback", func(c *gin.Context) {
+		api.CallbackHandler(c, func(tenantURL, _, code string) (string, error) {
+			return getAccessToken(tenantURL, globalOAuthState.CodeVerifier, code)
+		})
+	})
+
+	// 鉴权路由组
+	authGroup := r.Group("")
+	authGroup.Use(api.AuthMiddleware())
+	{
+		// OpenAI兼容的聊天端点
+		authGroup.POST("/v1/chat/completions", api.ChatCompletionsHandler)
+		authGroup.POST("/v1", api.ChatCompletionsHandler)
+		authGroup.POST("/v1/chat", api.ChatCompletionsHandler)
+
+		// OpenAI兼容的模型接口
+		authGroup.GET("/v1/models", api.ModelsHandler)
+	}
+
+	return r
+}
+
+func main() {
+	// 设置全局时区为东八区（CST）
+	time.Local = time.FixedZone("CST", 8*3600)
+
+	// 设置 Gin 为发布模式
+	gin.SetMode(gin.ReleaseMode)
+
+	// 初始化日志
+	logger.Init()
+
+	// 初始化配置
+	err := config.InitConfig()
+	if err != nil {
+		logger.Log.Fatalln("failed to initialize config: " + err.Error())
+		return
+	}
+
+	// 初始化Redis
+	err = config.InitRedisClient()
+	if err != nil {
+		logger.Log.Fatalln("failed to initialize Redis: " + err.Error())
+	}
+
+	r := setupRouter()
+
+	// 启动服务器
+	if err := r.Run(":27080"); err != nil {
+		logger.Log.Fatalf("启动服务失败: %v", err)
+	}
+
+	logger.Log.WithFields(map[string]interface{}{
+		"port": 27080,
+		"mode": gin.Mode(),
+	}).Info("Augment2API 服务启动成功")
+}

middleware/cors.go

@@ -0,0 +1,15 @@
+package middleware
+
+import (
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+)
+
+func CORS() gin.HandlerFunc {
+	config := cors.DefaultConfig()
+	config.AllowAllOrigins = true
+	config.AllowCredentials = true
+	config.AllowMethods = []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"}
+	config.AllowHeaders = []string{"*"}
+	return cors.New(config)
+}

pkg/logger/logger.go

@@ -0,0 +1,61 @@
+package logger
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"os"
+	"strings"
+	"time"
+)
+
+// CustomFormatter 自定义格式化器
+type CustomFormatter struct {
+	TimestampFormat string
+}
+
+func (f *CustomFormatter) Format(entry *logrus.Entry) ([]byte, error) {
+	// 将时间调整为东八区
+	localTime := entry.Time.In(time.FixedZone("CST", 8*3600))
+	// 构建日志消息
+	timestamp := localTime.Format(f.TimestampFormat)
+	level := strings.ToUpper(entry.Level.String())
+
+	// 将所有字段合并到一个字符串中，添加适当的分隔
+	var fieldsStr string
+	if len(entry.Data) > 0 {
+		pairs := make([]string, 0, len(entry.Data))
+		for k, v := range entry.Data {
+			pairs = append(pairs, fmt.Sprintf("%s: %v", k, v))
+		}
+		fieldsStr = " | " + strings.Join(pairs, " | ")
+	}
+
+	// 简化的日志格式，移除文件名和行号
+	logMsg := fmt.Sprintf("[%s] %-5s %s%s\n",
+		timestamp,
+		level,
+		entry.Message,
+		fieldsStr,
+	)
+
+	return []byte(logMsg), nil
+}
+
+var Log = logrus.New()
+
+func Init() {
+	// 使用自定义格式化器
+	Log.SetFormatter(&CustomFormatter{
+		TimestampFormat: "2006-01-02 15:04:05",
+	})
+
+	// 设置输出到标准输出
+	Log.SetOutput(os.Stdout)
+
+	// 设置日志级别
+	if os.Getenv("DEBUG") == "true" {
+		Log.SetLevel(logrus.DebugLevel)
+	} else {
+		Log.SetLevel(logrus.InfoLevel)
+	}
+}

templates/admin.html

@@ -0,0 +1,660 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Augment2Api</title>
+    <style>
+        :root {
+            --primary-color: #4361ee;
+            --secondary-color: #3f37c9;
+            --success-color: #4caf50;
+            --error-color: #f44336;
+            --bg-color: #f8f9fa;
+            --card-bg: #ffffff;
+            --text-color: #333333;
+            --border-color: #e0e0e0;
+            --shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            --radius: 8px;
+        }
+
+        * {
+            box-sizing: border-box;
+            margin: 0;
+            padding: 0;
+        }
+
+        body {
+            font-family: 'PingFang SC', 'Microsoft YaHei', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, 'Open Sans', 'Helvetica Neue', sans-serif;
+            background-color: var(--bg-color);
+            color: var(--text-color);
+            line-height: 1.6;
+            padding: 20px;
+            min-height: 100vh;
+            display: flex;
+            flex-direction: column;
+        }
+
+        .container {
+            max-width: 1200px;
+            margin: 0 auto;
+            flex: 1;
+            display: flex;
+            flex-direction: column;
+        }
+
+        header {
+            text-align: center;
+            margin-bottom: 30px;
+            padding-bottom: 15px;
+            border-bottom: 1px solid var(--border-color);
+        }
+
+        h1 {
+            font-size: 28px;
+            font-weight: 600;
+            color: black;
+        }
+
+        h2 {
+            font-size: 20px;
+            font-weight: 500;
+            margin-bottom: 15px;
+            color: var(--secondary-color);
+        }
+
+        .dashboard {
+            display: flex;
+            flex-direction: row;
+            gap: 20px;
+            align-items: stretch;
+            min-height: 500px;
+        }
+
+        @media (max-width: 768px) {
+            .dashboard {
+                flex-direction: column;
+            }
+        }
+
+        .panel {
+            background: var(--card-bg);
+            border-radius: var(--radius);
+            box-shadow: var(--shadow);
+            padding: 25px;
+            display: flex;
+            flex-direction: column;
+        }
+        
+        .panel-left {
+            width: 50%;
+            max-width: 580px;
+            flex: 1;
+        }
+        
+        .panel-right {
+            width: 50%;
+            max-width: 580px;
+            flex: 1;
+            position: sticky;
+            top: 20px;
+        }
+        
+        @media (max-width: 768px) {
+            .panel-left, .panel-right {
+                width: 100%;
+                max-width: none;
+            }
+        }
+
+        .panel-title {
+            display: flex;
+            align-items: center;
+            margin-bottom: 20px;
+        }
+
+        .panel-title i {
+            margin-right: 10px;
+            font-size: 20px;
+            color: var(--primary-color);
+        }
+
+        .panel-title h2 {
+            margin-bottom: 0;
+        }
+
+        .token-display {
+            background: var(--bg-color);
+            padding: 15px;
+            border-radius: var(--radius);
+            word-break: break-all;
+            margin: 15px 0;
+            border: 1px solid var(--border-color);
+            font-family: monospace;
+            font-size: 14px;
+            position: relative;
+        }
+
+        .token-label {
+            font-weight: 500;
+            margin-bottom: 5px;
+            color: #666;
+        }
+
+        button {
+            background: var(--primary-color);
+            color: white;
+            border: none;
+            padding: 10px 15px;
+            border-radius: var(--radius);
+            cursor: pointer;
+            font-size: 14px;
+            margin-top: 10px;
+            transition: background 0.3s;
+            display: inline-flex;
+            align-items: center;
+            justify-content: center;
+        }
+
+        button:hover {
+            background: var(--secondary-color);
+        }
+
+        button i {
+            margin-right: 8px;
+        }
+
+        input, textarea {
+            width: 100%;
+            padding: 12px;
+            margin: 10px 0;
+            border: 1px solid var(--border-color);
+            border-radius: var(--radius);
+            font-size: 14px;
+            transition: border 0.3s;
+        }
+
+        input:focus, textarea:focus {
+            outline: none;
+            border-color: var(--primary-color);
+        }
+
+        textarea {
+            height: 120px;
+            resize: vertical;
+        }
+
+        .error {
+            color: var(--error-color);
+            margin: 10px 0;
+            padding: 10px;
+            background-color: rgba(244, 67, 54, 0.1);
+            border-radius: var(--radius);
+            display: none;
+        }
+
+        .success {
+            color: var(--success-color);
+            margin: 10px 0;
+            padding: 10px;
+            background-color: rgba(76, 175, 80, 0.1);
+            border-radius: var(--radius);
+            display: none;
+        }
+
+        .step {
+            margin-bottom: 20px;
+            padding-bottom: 20px;
+            border-bottom: 1px dashed var(--border-color);
+        }
+
+        .step:last-child {
+            border-bottom: none;
+        }
+
+        .step-number {
+            display: inline-block;
+            width: 24px;
+            height: 24px;
+            background-color: var(--primary-color);
+            color: white;
+            border-radius: 50%;
+            text-align: center;
+            line-height: 24px;
+            margin-right: 10px;
+            font-size: 14px;
+        }
+
+        /* 添加一些图标 */
+        .icon {
+            display: inline-block;
+            width: 1em;
+            height: 1em;
+            stroke-width: 0;
+            stroke: currentColor;
+            fill: currentColor;
+            vertical-align: middle;
+        }
+
+        /* 添加Token列表样式 */
+        .token-list {
+            display: flex;
+            flex-direction: column;
+            gap: 15px;
+            margin-bottom: 15px;
+            max-height: 500px;
+            overflow-y: auto;
+            padding-right: 10px;
+        }
+        
+        .token-item {
+            background: var(--bg-color);
+            border-radius: var(--radius);
+            border: 1px solid var(--border-color);
+            overflow: hidden;
+            transition: all 0.3s ease;
+        }
+        
+        .current-token {
+            border: 2px solid var(--primary-color);
+        }
+        
+        .token-header {
+            display: flex;
+            padding: 12px 15px;
+            cursor: pointer;
+            align-items: center;
+            background: #f0f2f5;
+        }
+        
+        .token-number {
+            background: var(--primary-color);
+            color: white;
+            padding: 6px 10px;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            font-weight: bold;
+            min-width: 30px;
+            border-radius: 4px;
+            margin-right: 12px;
+        }
+        
+        .token-summary {
+            flex: 1;
+            font-weight: 500;
+            white-space: nowrap;
+            overflow: hidden;
+            text-overflow: ellipsis;
+        }
+        
+        .token-toggle {
+            margin-left: 10px;
+            transition: transform 0.3s;
+        }
+        
+        .token-toggle.open {
+            transform: rotate(180deg);
+        }
+        
+        .token-details {
+            padding: 0;
+            max-height: 0;
+            overflow: hidden;
+            transition: all 0.3s ease;
+        }
+        
+        .token-details.open {
+            padding: 12px;
+            max-height: none;
+            overflow: visible;
+        }
+        
+        .token-details .token-label {
+            font-size: 13px;
+            margin-bottom: 3px;
+        }
+        
+        .token-details .token-display {
+            font-size: 13px;
+            padding: 10px;
+            margin: 5px 0 10px 0;
+        }
+        
+        .token-actions {
+            display: flex;
+            justify-content: flex-end;
+            margin-top: 8px;
+        }
+        
+        .token-actions button {
+            margin-left: 10px;
+            padding: 6px 12px;
+            font-size: 13px;
+        }
+        
+        .no-tokens {
+            padding: 20px;
+            text-align: center;
+            background: var(--bg-color);
+            border-radius: var(--radius);
+            color: #666;
+            margin-bottom: 15px;
+        }
+
+        /* 添加加载动画样式 */
+        .loading {
+            position: relative;
+            pointer-events: none;
+        }
+        
+        .loading:after {
+            content: '';
+            display: inline-block;
+            width: 1em;
+            height: 1em;
+            border: 2px solid rgba(255, 255, 255, 0.3);
+            border-radius: 50%;
+            border-top-color: white;
+            animation: spin 0.8s linear infinite;
+            margin-left: 8px;
+            vertical-align: middle;
+        }
+        
+        @keyframes spin {
+            to {
+                transform: rotate(360deg);
+            }
+        }
+        
+        .btn-text {
+            display: inline-block;
+        }
+        
+        .loading .btn-icon {
+            display: none;
+        }
+
+        /* 添加页脚样式 */
+        footer {
+            text-align: center;
+            margin-top: 30px;
+            padding: 15px 0;
+            color: #666;
+            font-size: 14px;
+            border-top: 1px solid var(--border-color);
+            margin-top: auto;
+        }
+        
+        footer a {
+            color: var(--primary-color);
+            text-decoration: none;
+            transition: color 0.3s;
+        }
+        
+        footer a:hover {
+            color: var(--secondary-color);
+            text-decoration: underline;
+        }
+    </style>
+    <!-- 添加图标 -->
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.10.0/font/bootstrap-icons.css">
+</head>
+<body>
+    <div class="container">
+        <header>
+            <h1>Augment面板</h1>
+        </header>
+
+        <div class="dashboard">
+            <!-- 左侧面板：查看当前Token -->
+            <div class="panel panel-left">
+                <div class="panel-title">
+                    <i class="bi bi-key-fill"></i>
+                    <h2>当前Token列表</h2>
+                </div>
+                
+                <div id="token-list">加载中...</div>
+                <button id="refresh-token"><i class="bi bi-arrow-clockwise btn-icon"></i> <span class="btn-text">刷新</span></button>
+            </div>
+
+            <!-- 右侧面板：授权获取Token -->
+            <div class="panel panel-right">
+                <div class="panel-title">
+                    <i class="bi bi-shield-lock-fill"></i>
+                    <h2>授权获取Token</h2>
+                </div>
+
+                <div class="auth-steps">
+                    <div class="step">
+                        <h3><span class="step-number">1</span> 获取授权地址</h3>
+                        <p>点击下方按钮获取授权地址，然后在浏览器中打开该地址进行授权。</p>
+                        <button id="get-auth-url"><i class="bi bi-link-45deg" class="btn-icon"></i> <span class="btn-text">获取授权地址</span></button>
+                        <div id="auth-url" class="token-display" style="display: none;"></div>
+                    </div>
+
+                    <div class="step">
+                        <h3><span class="step-number">2</span> 提交授权响应</h3>
+                        <p>完成授权后，将获得的授权响应粘贴到下面的文本框中：</p>
+                        <textarea id="auth-response" placeholder='{"code":"_000baec407c57c4bf9xxxxxxxxxxxxxx","state":"0uXxxxxxxxx","tenant_url":"https://d20.api.augmentcode.com/"}'></textarea>
+                        <div id="validation-message" class="error"></div>
+                        <button id="submit-auth"><i class="bi bi-check2-circle" class="btn-icon"></i> <span class="btn-text">获取Token</span></button>
+                        <div id="submit-result" class="success"></div>
+                    </div>
+                </div>
+            </div>
+        </div>
+        
+        <!-- 添加页脚 -->
+        <footer>
+            Designed by <a href="https://linux.do/u/bifang/summary" target="_blank">彼方</a>
+        </footer>
+    </div>
+
+    <script>
+        document.addEventListener('DOMContentLoaded', function() {
+            // 获取当前Token列表
+            function fetchCurrentToken() {
+                fetch('/api/tokens')
+                .then(response => response.json())
+                .then(data => {
+                    if (data.status === 'success') {
+                        const tokenListElement = document.getElementById('token-list');
+                        
+                        if (data.tokens.length === 0) {
+                            tokenListElement.innerHTML = '<div class="no-tokens">暂无可用Token，请点击右侧面板获取</div>';
+                            return;
+                        }
+                        
+                        // 创建token列表容器
+                        tokenListElement.innerHTML = '<div class="token-list"></div>';
+                        const listContainer = tokenListElement.querySelector('.token-list');
+                        
+                        // 添加每个token项
+                        data.tokens.forEach((tokenInfo, index) => {
+                            // 截断token以便显示
+                            const shortToken = tokenInfo.token.length > 15 
+                                ? tokenInfo.token.substring(0, 15) + '...' 
+                                : tokenInfo.token;
+                            
+                            const tokenItem = document.createElement('div');
+                            tokenItem.className = 'token-item';
+                            tokenItem.innerHTML = `
+                                <div class="token-header">
+                                    <div class="token-number">${index + 1}</div>
+                                    <div class="token-summary">${shortToken}</div>
+                                    <div class="token-toggle"><i class="bi bi-chevron-down"></i></div>
+                                </div>
+                                <div class="token-details">
+                                    <div class="token-label">Token:</div>
+                                    <div class="token-display">${tokenInfo.token}</div>
+                                    <div class="token-label">租户URL:</div>
+                                    <div class="token-display">${tokenInfo.tenant_url}</div>
+                                    <div class="token-actions">
+                                        <button class="delete-token" data-token="${tokenInfo.token}">
+                                            <i class="bi bi-trash"></i> 删除
+                                        </button>
+                                    </div>
+                                </div>
+                            `;
+                            
+                            listContainer.appendChild(tokenItem);
+                            
+                            // 添加点击事件
+                            const header = tokenItem.querySelector('.token-header');
+                            const details = tokenItem.querySelector('.token-details');
+                            const toggle = tokenItem.querySelector('.token-toggle');
+                            
+                            header.addEventListener('click', function() {
+                                details.classList.toggle('open');
+                                toggle.classList.toggle('open');
+                            });
+                        });
+                    } else {
+                        showError('获取Token列表失败: ' + (data.error || '未知错误'));
+                    }
+                })
+                .catch(error => {
+                    document.getElementById('token-list').innerHTML = 
+                        '<div class="error" style="display:block;">请求失败: ' + error.message + '</div>';
+                });
+                
+                return Promise.resolve(); // 确保可以使用.finally()
+            }
+            
+            // 显示错误信息的辅助函数
+            function showError(message) {
+                const tokenListElement = document.getElementById('token-list');
+                tokenListElement.innerHTML = `<div class="error" style="display:block;">${message}</div>`;
+            }
+            
+            // 获取授权地址
+            document.getElementById('get-auth-url').addEventListener('click', function() {
+                const button = this;
+                button.classList.add('loading');
+                
+                fetch('/auth')
+                    .then(response => response.json())
+                    .then(data => {
+                        const authUrlElement = document.getElementById('auth-url');
+                        authUrlElement.textContent = data.authorize_url;
+                        authUrlElement.style.display = 'block';
+                    })
+                    .catch(error => {
+                        alert('获取授权地址失败: ' + error.message);
+                    })
+                    .finally(() => {
+                        button.classList.remove('loading');
+                    });
+            });
+            
+            // 验证授权响应格式
+            function validateAuthResponse(response) {
+                try {
+                    const data = JSON.parse(response);
+                    if (!data.code || !data.state || !data.tenant_url) {
+                        return { valid: false, message: '缺少必要字段: code, state 或 tenant_url' };
+                    }
+                    return { valid: true };
+                } catch (e) {
+                    return { valid: false, message: 'JSON格式无效: ' + e.message };
+                }
+            }
+            
+            // 提交授权响应
+            document.getElementById('submit-auth').addEventListener('click', function() {
+                const button = this;
+                const authResponse = document.getElementById('auth-response').value.trim();
+                const validationMessage = document.getElementById('validation-message');
+                
+                // 检查是否为空
+                if (!authResponse) {
+                    validationMessage.textContent = '请输入授权响应';
+                    validationMessage.style.display = 'block';
+                    return;
+                }
+                
+                const validationResult = validateAuthResponse(authResponse);
+                
+                if (!validationResult.valid) {
+                    validationMessage.textContent = validationResult.message;
+                    validationMessage.style.display = 'block';
+                    return;
+                }
+                
+                validationMessage.style.display = 'none';
+                button.classList.add('loading');
+                
+                fetch('/callback', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: authResponse
+                })
+                .then(response => response.json())
+                .then(data => {
+                    const submitResult = document.getElementById('submit-result');
+                    if (data.status === 'success') {
+                        submitResult.textContent = 'Token获取成功！';
+                        submitResult.className = 'success';
+                        fetchCurrentToken(); // 刷新当前Token显示
+                    } else {
+                        submitResult.textContent = '获取失败: ' + (data.error || '未知错误');
+                        submitResult.className = 'error';
+                    }
+                    submitResult.style.display = 'block';
+                })
+                .catch(error => {
+                    const submitResult = document.getElementById('submit-result');
+                    submitResult.textContent = '请求失败: ' + error.message;
+                    submitResult.className = 'error';
+                    submitResult.style.display = 'block';
+                })
+                .finally(() => {
+                    button.classList.remove('loading');
+                });
+            });
+            
+            // 刷新Token
+            document.getElementById('refresh-token').addEventListener('click', function() {
+                const button = this;
+                button.classList.add('loading');
+                
+                fetchCurrentToken()
+                    .finally(() => {
+                        button.classList.remove('loading');
+                    });
+            });
+            
+            // 绑定删除按钮事件
+            document.addEventListener('click', function(e) {
+                if (e.target.closest('.delete-token')) {
+                    const button = e.target.closest('.delete-token');
+                    const token = button.dataset.token;
+                    
+                    if (confirm('确定要删除此Token吗？')) {
+                        fetch(`/api/token/${token}`, {
+                            method: 'DELETE'
+                        })
+                        .then(response => response.json())
+                        .then(data => {
+                            if (data.status === 'success') {
+                                fetchCurrentToken(); // 刷新列表
+                            } else {
+                                alert('删除失败: ' + (data.error || '未知错误'));
+                            }
+                        })
+                        .catch(error => {
+                            alert('请求失败: ' + error.message);
+                        });
+                    }
+                }
+            });
+            
+            // 初始加载
+            fetchCurrentToken();
+        });
+    </script>
+</body>
+</html>