xiaolajiaoyyds
6f71df4849
多平台账号批量注册系统，支持 OpenAI / Grok / Kiro / Gemini 平台的全自动注册。 Features: - 用户名/密码注册登录（首个用户自动管理员） - JWT 认证 + bcrypt 密码加密 - 积分系统与多用户管理 - HuggingFace Space 弹性 Worker 节点管理 - Cloudflare Worker 负载均衡 - 实时日志推送（SSE/WebSocket） - 管理后台（用户管理、系统设置、代理池、卡池） Tech Stack: Go 1.25 / Gin / GORM / PostgreSQL / Vue 3 / Vite / TailwindCSS
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package newapi
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
// Client 封装 New-API 管理员 API
|
||
type Client struct {
|
||
baseURL string
|
||
adminToken string
|
||
adminUserID int // 管理员自身的 New-API 用户 ID（用于 New-Api-User 鉴权头）
|
||
http *http.Client
|
||
}
|
||
|
||
// NewClient 创建 New-API 客户端
|
||
// adminUserID: 管理员在 New-API 中的用户 ID，所有请求的 New-Api-User 头必须设为此值
|
||
func NewClient(baseURL, adminToken string, adminUserID int) *Client {
|
||
return &Client{
|
||
baseURL: baseURL,
|
||
adminToken: adminToken,
|
||
adminUserID: adminUserID,
|
||
http: &http.Client{
|
||
Timeout: 10 * time.Second,
|
||
},
|
||
}
|
||
}
|
||
|
||
// setAdminAuth 设置管理员鉴权头
|
||
// New-API 的 authHelper 中间件要求 New-Api-User 与 token 对应的用户 ID 一致
|
||
func (c *Client) setAdminAuth(req *http.Request) {
|
||
req.Header.Set("Authorization", "Bearer "+c.adminToken)
|
||
req.Header.Set("New-Api-User", fmt.Sprintf("%d", c.adminUserID))
|
||
}
|
||
|
||
// userResponse New-API 用户接口响应
|
||
type userResponse struct {
|
||
Success bool `json:"success"`
|
||
Data struct {
|
||
ID int `json:"id"`
|
||
Quota int `json:"quota"`
|
||
} `json:"data"`
|
||
Message string `json:"message"`
|
||
}
|
||
|
||
// genericResponse New-API 通用响应
|
||
type genericResponse struct {
|
||
Success bool `json:"success"`
|
||
Message string `json:"message"`
|
||
}
|
||
|
||
// GetUserQuota 实时查询用户 quota（GET /api/user/{id}）
|
||
func (c *Client) GetUserQuota(newapiUserID int) (int, error) {
|
||
req, err := http.NewRequest("GET", fmt.Sprintf("%s/api/user/%d", c.baseURL, newapiUserID), nil)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("创建请求失败: %w", err)
|
||
}
|
||
c.setAdminAuth(req)
|
||
|
||
resp, err := c.http.Do(req)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("请求 New-API 失败: %w", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
body, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("读取响应失败: %w", err)
|
||
}
|
||
|
||
if resp.StatusCode != http.StatusOK {
|
||
return 0, fmt.Errorf("New-API 返回错误 %d: %s", resp.StatusCode, string(body))
|
||
}
|
||
|
||
var result userResponse
|
||
if err := json.Unmarshal(body, &result); err != nil {
|
||
return 0, fmt.Errorf("解析响应失败: %w", err)
|
||
}
|
||
|
||
if !result.Success {
|
||
return 0, fmt.Errorf("New-API 错误: %s", result.Message)
|
||
}
|
||
|
||
return result.Data.Quota, nil
|
||
}
|
||
|
||
// DeductQuota 原子扣减 quota（POST /api/user/{id}/quota-deduct）
|
||
// 由 New-API 保证原子性，无需客户端加锁
|
||
func (c *Client) DeductQuota(newapiUserID int, amount int) error {
|
||
payload := map[string]interface{}{
|
||
"amount": amount,
|
||
"remark": "regplatform 购买注册次数",
|
||
}
|
||
body, _ := json.Marshal(payload)
|
||
|
||
req, err := http.NewRequest("POST", fmt.Sprintf("%s/api/user/%d/quota-deduct", c.baseURL, newapiUserID), bytes.NewReader(body))
|
||
if err != nil {
|
||
return fmt.Errorf("创建请求失败: %w", err)
|
||
}
|
||
c.setAdminAuth(req)
|
||
req.Header.Set("Content-Type", "application/json")
|
||
|
||
resp, err := c.http.Do(req)
|
||
if err != nil {
|
||
return fmt.Errorf("请求 New-API 失败: %w", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
respBody, _ := io.ReadAll(resp.Body)
|
||
|
||
if resp.StatusCode != http.StatusOK {
|
||
return fmt.Errorf("New-API 扣减失败 %d: %s", resp.StatusCode, string(respBody))
|
||
}
|
||
|
||
var result genericResponse
|
||
if err := json.Unmarshal(respBody, &result); err != nil {
|
||
return fmt.Errorf("解析响应失败: %w", err)
|
||
}
|
||
|
||
if !result.Success {
|
||
return fmt.Errorf("New-API 扣减失败: %s", result.Message)
|
||
}
|
||
|
||
return nil
|
||
}
|