mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-19 16:17:56 +00:00
feeeei
9a51333101
- groups 新增 i18n JSONB 列（迁移 242）：语言代码 -> {name, description}，只存译文；
name/description 原列语义不变，仍用于唯一性校验、搜索与各类日志
- 用户侧接口由后端按请求语言输出：取 Accept-Language 中权重最高的一项，先按完整
语言代码、再按主语言匹配；该语言有译文的字段换成译文，否则保持原字段，且不再带出
译文表。覆盖密钥、可用分组、订阅（含进度与汇总）、兑换、使用记录、可用渠道、
模型广场、套餐列表
- 管理端接口一律返回原字段；分组管理接口额外带 i18n 供编辑
- 创建/编辑分组：名称、描述的标题行右侧新增语言下拉，同一个输入框在默认内容与各语言
版本之间切换；语种跟随前端 availableLocales
- 管理端写入：创建/更新接受 i18n（更新时缺省不修改，传入则整体替换），校验语言代码
形态与名称长度并丢弃空译文；简易模式同样放行；复制分组只带走描述译文
- 切换界面语言后整页刷新，让已加载的数据换成新语言；模型广场响应加
Vary: Accept-Language
- 网关路径，以及只带 group_name 字符串的统计、错误记录出口不做本地化
267 lines
10 KiB
Go
267 lines
10 KiB
Go
package handler
|
||
|
||
import (
|
||
"log/slog"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/response"
|
||
"github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// ModelPlazaHandler 处理「模型广场」查询。
|
||
//
|
||
// 广场路由挂 OptionalJWT 中间件：匿名可访问（除非 require_auth 开启），带 token 则
|
||
// 识别用户。可见性规则（橱窗语义，与「可用渠道」的可绑定语义不同）：
|
||
// - 匿名：仅非专属分组（订阅型照常展示）；
|
||
// - 登录：非专属分组 + user_allowed_groups 授权或持有有效订阅的专属分组；
|
||
// 若该用户开启了公开分组限制，则公开分组同样需要落在授权集合内。
|
||
type ModelPlazaHandler struct {
|
||
plazaService *service.ModelPlazaService
|
||
apiKeyService *service.APIKeyService
|
||
settingService *service.SettingService
|
||
}
|
||
|
||
// NewModelPlazaHandler 创建模型广场 handler。
|
||
func NewModelPlazaHandler(
|
||
plazaService *service.ModelPlazaService,
|
||
apiKeyService *service.APIKeyService,
|
||
settingService *service.SettingService,
|
||
) *ModelPlazaHandler {
|
||
return &ModelPlazaHandler{
|
||
plazaService: plazaService,
|
||
apiKeyService: apiKeyService,
|
||
settingService: settingService,
|
||
}
|
||
}
|
||
|
||
// modelPlazaOfficialPricing 官方参考价（USD per token，与计费目录同源）。
|
||
type modelPlazaOfficialPricing struct {
|
||
InputPrice *float64 `json:"input_price"`
|
||
OutputPrice *float64 `json:"output_price"`
|
||
CacheWritePrice *float64 `json:"cache_write_price"`
|
||
CacheWrite1hPrice *float64 `json:"cache_write_1h_price,omitempty"`
|
||
CacheReadPrice *float64 `json:"cache_read_price"`
|
||
// Intervals 官方长上下文阶梯，仅多档模型给出。
|
||
Intervals []userPricingIntervalDTO `json:"intervals,omitempty"`
|
||
}
|
||
|
||
// modelPlazaTimePricingPeriod 分时倍率时段（配置时区当天 [start, end)）。
|
||
type modelPlazaTimePricingPeriod struct {
|
||
StartTime string `json:"start_time"`
|
||
EndTime string `json:"end_time"`
|
||
Multiplier float64 `json:"multiplier"`
|
||
}
|
||
|
||
// modelPlazaTimePricing 计费会生效的分时倍率（仅倍率 ≠ 1 的时段）。
|
||
// WeekdaysOnly 为 true 时时段仅周一至周五生效，周末整天按标准价计费。
|
||
type modelPlazaTimePricing struct {
|
||
Timezone string `json:"timezone"`
|
||
WeekdaysOnly bool `json:"weekdays_only,omitempty"`
|
||
Periods []modelPlazaTimePricingPeriod `json:"periods"`
|
||
}
|
||
|
||
// modelPlazaModel 广场模型条目：实收口径展示定价（白名单形态）+ 官方参考价。
|
||
type modelPlazaModel struct {
|
||
Name string `json:"name"`
|
||
Platform string `json:"platform"`
|
||
Pricing *userSupportedModelPricing `json:"pricing"`
|
||
OfficialPricing *modelPlazaOfficialPricing `json:"official_pricing"`
|
||
// LongContextBasis 多档时的计价基准："whole_request"（整单按档）| "marginal"（仅超出部分）。
|
||
LongContextBasis string `json:"long_context_basis,omitempty"`
|
||
// TimePricing 分时倍率时段，落在时段内的请求整单乘倍率；无分时省略。
|
||
TimePricing *modelPlazaTimePricing `json:"time_pricing,omitempty"`
|
||
}
|
||
|
||
// modelPlazaGroup 广场分组条目（白名单字段）。
|
||
type modelPlazaGroup struct {
|
||
ID int64 `json:"id"`
|
||
Name string `json:"name"`
|
||
Description string `json:"description"`
|
||
Platform string `json:"platform"`
|
||
SubscriptionType string `json:"subscription_type"`
|
||
RateMultiplier float64 `json:"rate_multiplier"`
|
||
UserRateMultiplier *float64 `json:"user_rate_multiplier,omitempty"`
|
||
PeakRateEnabled bool `json:"peak_rate_enabled"`
|
||
PeakStart string `json:"peak_start"`
|
||
PeakEnd string `json:"peak_end"`
|
||
PeakRateMultiplier float64 `json:"peak_rate_multiplier"`
|
||
IsExclusive bool `json:"is_exclusive"`
|
||
// 生图独立倍率：为 true 时图片计费模型的实付倍率取 ImageRateMultiplier，
|
||
// 不取分组/用户专属倍率。
|
||
ImageRateIndependent bool `json:"image_rate_independent"`
|
||
ImageRateMultiplier float64 `json:"image_rate_multiplier"`
|
||
VideoRateIndependent bool `json:"video_rate_independent"`
|
||
VideoRateMultiplier float64 `json:"video_rate_multiplier"`
|
||
// 分组是否启用长上下文阶梯计费；关闭时模型实付列只展示最低档/基础价。
|
||
LongContextPricingEnabled bool `json:"long_context_pricing_enabled"`
|
||
Models []modelPlazaModel `json:"models"`
|
||
}
|
||
|
||
// modelPlazaResponse 广场页响应。
|
||
type modelPlazaResponse struct {
|
||
Description string `json:"description"`
|
||
Groups []modelPlazaGroup `json:"groups"`
|
||
}
|
||
|
||
// Get 返回模型广场数据。
|
||
// GET /api/v1/model-plaza
|
||
func (h *ModelPlazaHandler) Get(c *gin.Context) {
|
||
if h.settingService == nil {
|
||
response.NotFound(c, "Model plaza is not enabled")
|
||
return
|
||
}
|
||
rt := h.settingService.GetModelPlazaRuntime(c.Request.Context())
|
||
if !rt.Enabled {
|
||
response.NotFound(c, "Model plaza is not enabled")
|
||
return
|
||
}
|
||
|
||
subject, authed := middleware.GetAuthSubjectFromContext(c)
|
||
if rt.RequireAuth && !authed {
|
||
response.Unauthorized(c, "Authentication required")
|
||
return
|
||
}
|
||
|
||
groups, err := h.plazaService.ListGroups(c.Request.Context())
|
||
if err != nil {
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
|
||
// allowedGroups == nil 表示匿名；登录用户恒为非 nil（可能为空集合）。
|
||
var allowedGroups map[int64]struct{}
|
||
var restrictPublicGroups bool
|
||
var userRates map[int64]float64
|
||
if authed {
|
||
allowedGroups, restrictPublicGroups, err = h.apiKeyService.GetUserGroupVisibility(c.Request.Context(), subject.UserID)
|
||
if err != nil {
|
||
// 可见性数据拿不到时不能静默降级成匿名视图（会错漏专属分组），直接报错。
|
||
response.ErrorFrom(c, err)
|
||
return
|
||
}
|
||
userRates, err = h.apiKeyService.GetUserGroupRates(c.Request.Context(), subject.UserID)
|
||
if err != nil {
|
||
// 专属倍率仅是展示增强，失败降级为分组默认倍率。
|
||
slog.Warn("model_plaza_user_rates_failed", "error", err, "user_id", subject.UserID)
|
||
userRates = nil
|
||
}
|
||
}
|
||
|
||
visible := filterPlazaVisibleGroups(groups, allowedGroups, restrictPublicGroups)
|
||
|
||
// 广场可匿名访问，分组文案随请求语言变化，提示中间缓存按语言区分。
|
||
c.Header("Vary", "Accept-Language")
|
||
out := toLocalizedModelPlazaGroups(visible, userRates, c.GetHeader("Accept-Language"))
|
||
response.Success(c, modelPlazaResponse{
|
||
Description: rt.Description,
|
||
Groups: out,
|
||
})
|
||
}
|
||
|
||
// filterPlazaVisibleGroups 按登录态裁剪分组可见性。
|
||
// allowedGroups == nil 表示匿名（仅非专属）；非 nil 包含普通授权及有效订阅分组。
|
||
// restrictPublicGroups 为 true 时，公开分组也必须落在 allowedGroups 内，否则用户会
|
||
// 在广场看到自己实际绑定不了的分组。
|
||
func filterPlazaVisibleGroups(
|
||
groups []service.PlazaGroup,
|
||
allowedGroups map[int64]struct{},
|
||
restrictPublicGroups bool,
|
||
) []service.PlazaGroup {
|
||
visible := make([]service.PlazaGroup, 0, len(groups))
|
||
for _, g := range groups {
|
||
if g.IsExclusive || (restrictPublicGroups && allowedGroups != nil) {
|
||
if allowedGroups == nil {
|
||
continue
|
||
}
|
||
if _, ok := allowedGroups[g.ID]; !ok {
|
||
continue
|
||
}
|
||
}
|
||
visible = append(visible, g)
|
||
}
|
||
return visible
|
||
}
|
||
|
||
// toLocalizedModelPlazaGroups 转换可见分组，并把名称与描述换成请求首选语言的版本。
|
||
func toLocalizedModelPlazaGroups(visible []service.PlazaGroup, userRates map[int64]float64, acceptLanguage string) []modelPlazaGroup {
|
||
out := make([]modelPlazaGroup, 0, len(visible))
|
||
for i := range visible {
|
||
group := toModelPlazaGroupDTO(&visible[i], userRates)
|
||
group.Name, group.Description = visible[i].I18n.Localize(acceptLanguage, group.Name, group.Description)
|
||
out = append(out, group)
|
||
}
|
||
return out
|
||
}
|
||
|
||
// toModelPlazaGroupDTO 将 service 层广场分组映射为白名单 DTO,并合并用户专属倍率。
|
||
func toModelPlazaGroupDTO(g *service.PlazaGroup, userRates map[int64]float64) modelPlazaGroup {
|
||
models := make([]modelPlazaModel, 0, len(g.Models))
|
||
for i := range g.Models {
|
||
m := &g.Models[i]
|
||
models = append(models, modelPlazaModel{
|
||
Name: m.Name,
|
||
Platform: m.Platform,
|
||
Pricing: toUserPricing(m.Pricing),
|
||
OfficialPricing: toModelPlazaOfficialPricing(m.OfficialPricing),
|
||
LongContextBasis: string(m.LongContextBasis),
|
||
TimePricing: toModelPlazaTimePricing(m.TimePricing),
|
||
})
|
||
}
|
||
dto := modelPlazaGroup{
|
||
ID: g.ID,
|
||
Name: g.Name,
|
||
Description: g.Description,
|
||
Platform: g.Platform,
|
||
SubscriptionType: g.SubscriptionType,
|
||
RateMultiplier: g.RateMultiplier,
|
||
PeakRateEnabled: g.PeakRateEnabled,
|
||
PeakStart: g.PeakStart,
|
||
PeakEnd: g.PeakEnd,
|
||
PeakRateMultiplier: g.PeakRateMultiplier,
|
||
IsExclusive: g.IsExclusive,
|
||
ImageRateIndependent: g.ImageRateIndependent,
|
||
ImageRateMultiplier: g.ImageRateMultiplier,
|
||
VideoRateIndependent: g.VideoRateIndependent,
|
||
VideoRateMultiplier: g.VideoRateMultiplier,
|
||
LongContextPricingEnabled: g.LongContextPricingEnabled,
|
||
Models: models,
|
||
}
|
||
if rate, ok := userRates[g.ID]; ok {
|
||
dto.UserRateMultiplier = &rate
|
||
}
|
||
return dto
|
||
}
|
||
|
||
// toModelPlazaTimePricing 转换分时倍率；nil 透传（JSON 省略）。
|
||
func toModelPlazaTimePricing(p *service.TimePricingSchedule) *modelPlazaTimePricing {
|
||
if p == nil || len(p.Periods) == 0 {
|
||
return nil
|
||
}
|
||
periods := make([]modelPlazaTimePricingPeriod, 0, len(p.Periods))
|
||
for _, period := range p.Periods {
|
||
periods = append(periods, modelPlazaTimePricingPeriod{
|
||
StartTime: period.StartTime,
|
||
EndTime: period.EndTime,
|
||
Multiplier: period.Multiplier,
|
||
})
|
||
}
|
||
return &modelPlazaTimePricing{Timezone: p.Timezone, WeekdaysOnly: p.WeekdaysOnly, Periods: periods}
|
||
}
|
||
|
||
// toModelPlazaOfficialPricing 转换官方参考价；nil 透传（前端显示 "-"）。
|
||
func toModelPlazaOfficialPricing(p *service.PlazaOfficialPricing) *modelPlazaOfficialPricing {
|
||
if p == nil {
|
||
return nil
|
||
}
|
||
return &modelPlazaOfficialPricing{
|
||
InputPrice: p.InputPrice,
|
||
OutputPrice: p.OutputPrice,
|
||
CacheWritePrice: p.CacheWritePrice,
|
||
CacheWrite1hPrice: p.CacheWrite1hPrice,
|
||
CacheReadPrice: p.CacheReadPrice,
|
||
Intervals: toUserPricingIntervals(p.Intervals),
|
||
}
|
||
}
|