李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package admin
|
||
|
||
import (
|
||
"strconv"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/response"
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// CNProviderHandler 暴露国产供应商（kimi/zhipu/deepseek）的额度与余额查询端点。
|
||
//
|
||
// - GET /admin/cn-providers/accounts/:id/quota Coding Plan 滚动窗口用量（kimi/zhipu）
|
||
// - GET /admin/cn-providers/accounts/:id/balance payg 账号余额（kimi/deepseek）
|
||
//
|
||
// 智谱（zhipu）无余额端点，故同一账号仅 quota 或 balance 其一可用：服务端按账号
|
||
// platform + account_mode 校验并返回明确错误（见 CNProvider*Service 的 load*Account）。
|
||
type CNProviderHandler struct {
|
||
quotaService *service.CNProviderQuotaService
|
||
balanceService *service.CNProviderBalanceService
|
||
}
|
||
|
||
func NewCNProviderHandler(
|
||
quotaService *service.CNProviderQuotaService,
|
||
balanceService *service.CNProviderBalanceService,
|
||
) *CNProviderHandler {
|
||
return &CNProviderHandler{
|
||
quotaService: quotaService,
|
||
balanceService: balanceService,
|
||
}
|
||
}
|
||
|
||
// QueryQuota 查询 Coding Plan 滚动窗口用量（5h + weekly）。
|
||
func (h *CNProviderHandler) QueryQuota(c *gin.Context) {
|
||
accountID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
response.BadRequest(c, "Invalid account ID")
|
||
return
|
||
}
|
||
if h == nil || h.quotaService == nil {
|
||
response.BadRequest(c, "cn provider quota service is not enabled")
|
||
return
|
||
}
|
||
result, err := h.quotaService.QueryUsage(c.Request.Context(), accountID)
|
||
if err != nil {
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
response.Success(c, result)
|
||
}
|
||
|
||
// QueryBalance 查询 payg 账号余额。
|
||
func (h *CNProviderHandler) QueryBalance(c *gin.Context) {
|
||
accountID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
||
if err != nil {
|
||
response.BadRequest(c, "Invalid account ID")
|
||
return
|
||
}
|
||
if h == nil || h.balanceService == nil {
|
||
response.BadRequest(c, "cn provider balance service is not enabled")
|
||
return
|
||
}
|
||
result, err := h.balanceService.QueryBalance(c.Request.Context(), accountID)
|
||
if err != nil {
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
response.Success(c, result)
|
||
}
|