feat: Add automated ChatGPT account registration with backend API, TLS client, and fingerprinting, alongside new frontend pages for configuration, monitoring, and upload.
This commit is contained in:
172
backend/internal/api/codex_proxy.go
Normal file
172
backend/internal/api/codex_proxy.go
Normal file
@@ -0,0 +1,172 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"codex-pool/internal/database"
|
||||
)
|
||||
|
||||
// HandleCodexProxies 处理代理池请求
|
||||
func HandleCodexProxies(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
listCodexProxies(w, r)
|
||||
case http.MethodPost:
|
||||
addCodexProxy(w, r)
|
||||
case http.MethodDelete:
|
||||
deleteCodexProxy(w, r)
|
||||
case http.MethodPut:
|
||||
toggleCodexProxy(w, r)
|
||||
default:
|
||||
Error(w, http.StatusMethodNotAllowed, "方法不允许")
|
||||
}
|
||||
}
|
||||
|
||||
// listCodexProxies 获取代理列表
|
||||
func listCodexProxies(w http.ResponseWriter, r *http.Request) {
|
||||
proxies, err := database.Instance.GetCodexProxies()
|
||||
if err != nil {
|
||||
Error(w, http.StatusInternalServerError, "获取代理列表失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
stats := database.Instance.GetCodexProxyStats()
|
||||
|
||||
Success(w, map[string]interface{}{
|
||||
"proxies": proxies,
|
||||
"stats": stats,
|
||||
})
|
||||
}
|
||||
|
||||
// AddProxyRequest 添加代理请求
|
||||
type AddProxyRequest struct {
|
||||
ProxyURL string `json:"proxy_url"`
|
||||
Description string `json:"description"`
|
||||
Proxies []string `json:"proxies"` // 批量添加
|
||||
}
|
||||
|
||||
// addCodexProxy 添加代理
|
||||
func addCodexProxy(w http.ResponseWriter, r *http.Request) {
|
||||
var req AddProxyRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
Error(w, http.StatusBadRequest, "请求参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
// 批量添加
|
||||
if len(req.Proxies) > 0 {
|
||||
// 过滤空行和格式化
|
||||
var validProxies []string
|
||||
for _, p := range req.Proxies {
|
||||
p = strings.TrimSpace(p)
|
||||
if p != "" {
|
||||
// 如果没有协议前缀,自动添加 http://
|
||||
if !strings.HasPrefix(p, "http://") && !strings.HasPrefix(p, "https://") && !strings.HasPrefix(p, "socks5://") {
|
||||
p = "http://" + p
|
||||
}
|
||||
validProxies = append(validProxies, p)
|
||||
}
|
||||
}
|
||||
|
||||
if len(validProxies) == 0 {
|
||||
Error(w, http.StatusBadRequest, "没有有效的代理地址")
|
||||
return
|
||||
}
|
||||
|
||||
count, err := database.Instance.AddCodexProxies(validProxies)
|
||||
if err != nil {
|
||||
Error(w, http.StatusInternalServerError, "批量添加代理失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
Success(w, map[string]interface{}{
|
||||
"added": count,
|
||||
"total": len(validProxies),
|
||||
"message": "批量添加成功",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 单个添加
|
||||
if req.ProxyURL == "" {
|
||||
Error(w, http.StatusBadRequest, "代理地址不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
proxyURL := strings.TrimSpace(req.ProxyURL)
|
||||
if !strings.HasPrefix(proxyURL, "http://") && !strings.HasPrefix(proxyURL, "https://") && !strings.HasPrefix(proxyURL, "socks5://") {
|
||||
proxyURL = "http://" + proxyURL
|
||||
}
|
||||
|
||||
id, err := database.Instance.AddCodexProxy(proxyURL, req.Description)
|
||||
if err != nil {
|
||||
if strings.Contains(err.Error(), "UNIQUE constraint failed") {
|
||||
Error(w, http.StatusConflict, "代理地址已存在")
|
||||
return
|
||||
}
|
||||
Error(w, http.StatusInternalServerError, "添加代理失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
Success(w, map[string]interface{}{
|
||||
"id": id,
|
||||
"message": "添加成功",
|
||||
})
|
||||
}
|
||||
|
||||
// deleteCodexProxy 删除代理
|
||||
func deleteCodexProxy(w http.ResponseWriter, r *http.Request) {
|
||||
idStr := r.URL.Query().Get("id")
|
||||
if idStr == "" {
|
||||
// 如果没有 id 参数,检查是否要清空所有
|
||||
if r.URL.Query().Get("all") == "true" {
|
||||
err := database.Instance.ClearCodexProxies()
|
||||
if err != nil {
|
||||
Error(w, http.StatusInternalServerError, "清空代理失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
Success(w, map[string]string{"message": "已清空所有代理"})
|
||||
return
|
||||
}
|
||||
Error(w, http.StatusBadRequest, "缺少代理 ID")
|
||||
return
|
||||
}
|
||||
|
||||
id, err := strconv.ParseInt(idStr, 10, 64)
|
||||
if err != nil {
|
||||
Error(w, http.StatusBadRequest, "无效的代理 ID")
|
||||
return
|
||||
}
|
||||
|
||||
if err := database.Instance.DeleteCodexProxy(id); err != nil {
|
||||
Error(w, http.StatusInternalServerError, "删除代理失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
Success(w, map[string]string{"message": "删除成功"})
|
||||
}
|
||||
|
||||
// toggleCodexProxy 切换代理启用状态
|
||||
func toggleCodexProxy(w http.ResponseWriter, r *http.Request) {
|
||||
idStr := r.URL.Query().Get("id")
|
||||
if idStr == "" {
|
||||
Error(w, http.StatusBadRequest, "缺少代理 ID")
|
||||
return
|
||||
}
|
||||
|
||||
id, err := strconv.ParseInt(idStr, 10, 64)
|
||||
if err != nil {
|
||||
Error(w, http.StatusBadRequest, "无效的代理 ID")
|
||||
return
|
||||
}
|
||||
|
||||
if err := database.Instance.ToggleCodexProxy(id); err != nil {
|
||||
Error(w, http.StatusInternalServerError, "切换代理状态失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
Success(w, map[string]string{"message": "状态已切换"})
|
||||
}
|
||||
Reference in New Issue
Block a user