6b39b344d8
为用户在 anthropic/openai/gemini/antigravity 四个平台上提供日/周/月 三个窗口的 USD 配额管控。配额语义：未设置=不限制，0=禁用，>0=美元上限。 两层模型： - 配置层：系统默认配额，以及 email/linuxdo/oidc/wechat/github/google/ dingtalk 七个鉴权来源的默认配额，存于 settings，以嵌套 JSON 整体读写 （系统 1 个 key + 每个来源 1 个 key），整体替换语义。 - 运行时层：user_platform_quota 表按用户记录实际配额，与配置层解耦。 后端：新增 ent schema 与 140_user_platform_quotas.sql 迁移、repository 与 service 端口、计费链路集成、管理端与用户端读写接口。 前端：管理端设置页配额编辑、用户配额管理 Modal、用户 Dashboard 展示、 中英文案。 Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
_ "github.com/Wei-Shaw/sub2api/ent/runtime"
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
"github.com/Wei-Shaw/sub2api/internal/repository"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
func main() {
|
|
email := flag.String("email", "", "Admin email to issue a JWT for (defaults to first active admin)")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.LoadForBootstrap()
|
|
if err != nil {
|
|
log.Fatalf("failed to load config: %v", err)
|
|
}
|
|
|
|
client, sqlDB, err := repository.InitEnt(cfg)
|
|
if err != nil {
|
|
log.Fatalf("failed to init db: %v", err)
|
|
}
|
|
defer func() {
|
|
if err := client.Close(); err != nil {
|
|
log.Printf("failed to close db: %v", err)
|
|
}
|
|
}()
|
|
|
|
userRepo := repository.NewUserRepository(client, sqlDB)
|
|
authService := service.NewAuthService(client, userRepo, nil, nil, cfg, nil, nil, nil, nil, nil, nil, nil, nil)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
var user *service.User
|
|
if *email != "" {
|
|
user, err = userRepo.GetByEmail(ctx, *email)
|
|
} else {
|
|
user, err = userRepo.GetFirstAdmin(ctx)
|
|
}
|
|
if err != nil {
|
|
log.Fatalf("failed to resolve admin user: %v", err)
|
|
}
|
|
|
|
token, err := authService.GenerateToken(user)
|
|
if err != nil {
|
|
log.Fatalf("failed to generate token: %v", err)
|
|
}
|
|
|
|
fmt.Printf("ADMIN_EMAIL=%s\nADMIN_USER_ID=%d\nJWT=%s\n", user.Email, user.ID, token)
|
|
}
|