feat(backend): 授权改为时长兑换券体系 + 退役 ed25519/HMAC + 平台生码工具
- 新增 license_codes 码池表 + model.LicenseCode;licenses 加 tier 档位列 - LicenseService.Redeem:单事务 FOR UPDATE 校验码未用 → 时长叠加(可叠加,0=永久) → 写 type/tier/max_devices → 绑设备(超限整笔回滚) → 标记已用 → 即时失效 phase 缓存 路由仍 POST /license/activate,客户端零破坏 - util.GenerateRedeemCode/NormalizeCode:JIUKU-XXXX-XXXX 短码(crypto/rand) - cmd/gencode:平台批量生成兑换码并落库;删除 cmd/issue、cmd/genkey - 退役 ed25519 + HMAC:删 util/license_key、GenerateKey、License 全部 config 字段 及生产启动私钥校验;trial 改直接建行(无需私钥、去 Fatal) - tier 档位钩子默认 standard,分档消费模式后续设计 - 测试:Redeem 全场景(叠加/过期重置/永久/一码一次/无效/设备上限回滚) Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,90 @@
|
|||||||
|
// gencode — 平台方批量生成兑换券(激活码)并写入码池 license_codes。
|
||||||
|
//
|
||||||
|
// 用法(在 backend/ 目录下执行,读 config/env 取数据库 DSN):
|
||||||
|
//
|
||||||
|
// go run ./cmd/gencode -type annual -days 365 -count 100 -batch 2026-summer
|
||||||
|
// go run ./cmd/gencode -days 30 -count 10 -note "试用补偿"
|
||||||
|
// go run ./cmd/gencode -type lifetime -days 0 -count 1 # 永久授权码
|
||||||
|
//
|
||||||
|
// 生成的码以 JIUKU-XXXX-XXXX 格式打印(status=unused),分发给用户在 App 内兑换。
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
stdlog "log"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gorm.io/driver/mysql"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/logger"
|
||||||
|
|
||||||
|
"github.com/wangjia/jiu/backend/config"
|
||||||
|
"github.com/wangjia/jiu/backend/internal/model"
|
||||||
|
"github.com/wangjia/jiu/backend/internal/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
licType := flag.String("type", "annual", "计费/时长标签:trial|monthly|annual|lifetime")
|
||||||
|
tier := flag.String("tier", "standard", "档位(当前仅 standard)")
|
||||||
|
days := flag.Int("days", 365, "授予时长(天);0 = 永久")
|
||||||
|
devices := flag.Int("devices", 0, "授予设备上限;0 = 兑换时不改变门店现值")
|
||||||
|
count := flag.Int("count", 1, "生成数量")
|
||||||
|
batch := flag.String("batch", "", "发放批次/活动名(便于追踪)")
|
||||||
|
note := flag.String("note", "", "备注")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
if *count < 1 {
|
||||||
|
fmt.Fprintln(os.Stderr, "error: -count 必须 >= 1")
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
config.Load()
|
||||||
|
db, err := gorm.Open(mysql.Open(config.C.Database.DSN), &gorm.Config{
|
||||||
|
Logger: logger.New(
|
||||||
|
stdlog.New(os.Stdout, "\r\n", stdlog.LstdFlags),
|
||||||
|
logger.Config{LogLevel: logger.Warn, IgnoreRecordNotFoundError: true},
|
||||||
|
),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
stdlog.Fatalf("连接数据库失败: %v", err)
|
||||||
|
}
|
||||||
|
if err := db.AutoMigrate(&model.LicenseCode{}); err != nil {
|
||||||
|
stdlog.Fatalf("迁移 license_codes 失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
display := make([]string, 0, *count)
|
||||||
|
for i := 0; i < *count; i++ {
|
||||||
|
// 重试避免极小概率的唯一冲突
|
||||||
|
var shown string
|
||||||
|
for attempt := 0; attempt < 5; attempt++ {
|
||||||
|
shown = util.GenerateRedeemCode()
|
||||||
|
lc := model.LicenseCode{
|
||||||
|
Code: util.NormalizeCode(shown),
|
||||||
|
Type: *licType,
|
||||||
|
Tier: *tier,
|
||||||
|
DurationDays: *days,
|
||||||
|
MaxDevices: *devices,
|
||||||
|
Status: "unused",
|
||||||
|
Batch: *batch,
|
||||||
|
Note: *note,
|
||||||
|
}
|
||||||
|
err := db.Create(&lc).Error
|
||||||
|
if err == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if attempt == 4 {
|
||||||
|
stdlog.Fatalf("写入兑换码失败(连续冲突): %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
display = append(display, shown)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("\n✅ 已生成 %d 个兑换码(type=%s tier=%s 时长=%d天 设备=%d batch=%q)于 %s:\n\n",
|
||||||
|
*count, *licType, *tier, *days, *devices, *batch, time.Now().Format("2006-01-02 15:04:05"))
|
||||||
|
for _, c := range display {
|
||||||
|
fmt.Println(" " + c)
|
||||||
|
}
|
||||||
|
fmt.Println()
|
||||||
|
}
|
||||||
@@ -1,59 +0,0 @@
|
|||||||
// genkey generates an Ed25519 keypair for license signing.
|
|
||||||
// Run once; store the private key in Bitwarden and set the public key in config.
|
|
||||||
//
|
|
||||||
// Usage: go run ./cmd/genkey
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/internal/util"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
priv, pub, err := util.GenerateEd25519KeyPair()
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "failed to generate keypair: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Println("=== Ed25519 License Keypair ===")
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println("[Bitwarden] Private key (keep secret, never commit):")
|
|
||||||
fmt.Println(priv)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println("[Config / LICENSE_ED25519_PUBLIC_KEY] Public key:")
|
|
||||||
fmt.Println(pub)
|
|
||||||
fmt.Println()
|
|
||||||
|
|
||||||
// Demo: issue and verify a sample token to confirm the keypair works
|
|
||||||
now := time.Now()
|
|
||||||
exp := now.Add(30 * 24 * time.Hour).Unix()
|
|
||||||
sample := util.LicensePayload{
|
|
||||||
ShopID: 1,
|
|
||||||
LicenseID: 1,
|
|
||||||
Type: "trial",
|
|
||||||
IssuedAt: now.Unix(),
|
|
||||||
ExpiresAt: &exp,
|
|
||||||
MaxDevices: 3,
|
|
||||||
}
|
|
||||||
token, err := util.IssueLicenseToken(sample, priv)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "demo sign failed: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
verified, err := util.VerifyLicenseToken(token, pub)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "demo verify failed: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
out, _ := json.MarshalIndent(verified, "", " ")
|
|
||||||
fmt.Println("[Demo] Sample token (30-day trial, shop_id=1):")
|
|
||||||
fmt.Println(token)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println("[Demo] Verified payload:")
|
|
||||||
fmt.Println(string(out))
|
|
||||||
}
|
|
||||||
@@ -1,72 +0,0 @@
|
|||||||
// issue signs a license token for a specific shop.
|
|
||||||
// Usage: go run ./cmd/issue -shop 1 -days 365 -type annual -key <base64-private-key>
|
|
||||||
// Or use env var: LICENSE_ED25519_PRIVATE_KEY=<key> go run ./cmd/issue -shop 1 -days 365
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/internal/util"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
shopID := flag.Uint64("shop", 0, "shop ID (required)")
|
|
||||||
licenseID := flag.Uint64("license", 0, "license record ID (optional, 0 = omit)")
|
|
||||||
days := flag.Int("days", 365, "validity days; 0 = perpetual (no expiry)")
|
|
||||||
licType := flag.String("type", "annual", "license type: trial | annual | lifetime")
|
|
||||||
maxDevices := flag.Int("devices", 3, "max devices")
|
|
||||||
privKey := flag.String("key", "", "Ed25519 private key (base64); falls back to LICENSE_ED25519_PRIVATE_KEY env")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
if *shopID == 0 {
|
|
||||||
fmt.Fprintln(os.Stderr, "error: -shop is required")
|
|
||||||
flag.Usage()
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
key := *privKey
|
|
||||||
if key == "" {
|
|
||||||
key = os.Getenv("LICENSE_ED25519_PRIVATE_KEY")
|
|
||||||
}
|
|
||||||
if key == "" {
|
|
||||||
fmt.Fprintln(os.Stderr, "error: provide -key or set LICENSE_ED25519_PRIVATE_KEY")
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
now := time.Now()
|
|
||||||
payload := util.LicensePayload{
|
|
||||||
ShopID: *shopID,
|
|
||||||
Type: *licType,
|
|
||||||
IssuedAt: now.Unix(),
|
|
||||||
MaxDevices: *maxDevices,
|
|
||||||
}
|
|
||||||
if *licenseID > 0 {
|
|
||||||
payload.LicenseID = *licenseID
|
|
||||||
}
|
|
||||||
if *days > 0 {
|
|
||||||
exp := now.Add(time.Duration(*days) * 24 * time.Hour).Unix()
|
|
||||||
payload.ExpiresAt = &exp
|
|
||||||
}
|
|
||||||
|
|
||||||
token, err := util.IssueLicenseToken(payload, key)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "sign failed: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
out, _ := json.MarshalIndent(payload, "", " ")
|
|
||||||
fmt.Println("=== License Token ===")
|
|
||||||
fmt.Println(token)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println("=== Payload ===")
|
|
||||||
fmt.Println(string(out))
|
|
||||||
if payload.ExpiresAt != nil {
|
|
||||||
fmt.Printf("\nExpires: %s\n", time.Unix(*payload.ExpiresAt, 0).Format("2006-01-02 15:04:05"))
|
|
||||||
} else {
|
|
||||||
fmt.Println("\nExpires: never (perpetual)")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -26,6 +26,7 @@ var allModels = []any{
|
|||||||
&model.Shop{},
|
&model.Shop{},
|
||||||
&model.User{},
|
&model.User{},
|
||||||
&model.License{},
|
&model.License{},
|
||||||
|
&model.LicenseCode{},
|
||||||
&model.ProductCategory{},
|
&model.ProductCategory{},
|
||||||
&model.Product{},
|
&model.Product{},
|
||||||
&model.ProductNameOption{},
|
&model.ProductNameOption{},
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ type Config struct {
|
|||||||
Server ServerConfig
|
Server ServerConfig
|
||||||
Database DatabaseConfig
|
Database DatabaseConfig
|
||||||
JWT JWTConfig
|
JWT JWTConfig
|
||||||
License LicenseConfig
|
|
||||||
Storage StorageConfig
|
Storage StorageConfig
|
||||||
Session SessionConfig
|
Session SessionConfig
|
||||||
}
|
}
|
||||||
@@ -34,12 +33,6 @@ type JWTConfig struct {
|
|||||||
RefreshExpireH int `mapstructure:"refresh_expire_h"` // Refresh Token 有效小时数
|
RefreshExpireH int `mapstructure:"refresh_expire_h"` // Refresh Token 有效小时数
|
||||||
}
|
}
|
||||||
|
|
||||||
type LicenseConfig struct {
|
|
||||||
HMACSecret string `mapstructure:"hmac_secret"` // legacy, kept for backward compat
|
|
||||||
Ed25519PublicKey string `mapstructure:"ed25519_public_key"` // base64 Ed25519 public key for token verification
|
|
||||||
Ed25519PrivateKey string `mapstructure:"ed25519_private_key"` // base64 Ed25519 private key for token signing (keep in Bitwarden)
|
|
||||||
}
|
|
||||||
|
|
||||||
// SessionConfig 登录会话与并发限制(全局默认,可被每店 session_policy 覆盖)。
|
// SessionConfig 登录会话与并发限制(全局默认,可被每店 session_policy 覆盖)。
|
||||||
type SessionConfig struct {
|
type SessionConfig struct {
|
||||||
LimitDesktop int `mapstructure:"limit_desktop"` // 桌面端(win/mac/linux)最大并发会话,0=禁止
|
LimitDesktop int `mapstructure:"limit_desktop"` // 桌面端(win/mac/linux)最大并发会话,0=禁止
|
||||||
@@ -72,9 +65,6 @@ func Load() {
|
|||||||
// 显式绑定没有默认值的 key,确保 AutomaticEnv 能找到对应 env var
|
// 显式绑定没有默认值的 key,确保 AutomaticEnv 能找到对应 env var
|
||||||
_ = viper.BindEnv("database.dsn", "DATABASE_DSN")
|
_ = viper.BindEnv("database.dsn", "DATABASE_DSN")
|
||||||
_ = viper.BindEnv("jwt.secret", "JWT_SECRET")
|
_ = viper.BindEnv("jwt.secret", "JWT_SECRET")
|
||||||
_ = viper.BindEnv("license.hmac_secret", "LICENSE_HMAC_SECRET")
|
|
||||||
_ = viper.BindEnv("license.ed25519_public_key", "LICENSE_ED25519_PUBLIC_KEY")
|
|
||||||
_ = viper.BindEnv("license.ed25519_private_key", "LICENSE_ED25519_PRIVATE_KEY")
|
|
||||||
_ = viper.BindEnv("storage.upload_dir", "STORAGE_UPLOAD_DIR")
|
_ = viper.BindEnv("storage.upload_dir", "STORAGE_UPLOAD_DIR")
|
||||||
_ = viper.BindEnv("storage.base_url", "STORAGE_BASE_URL")
|
_ = viper.BindEnv("storage.base_url", "STORAGE_BASE_URL")
|
||||||
_ = viper.BindEnv("storage.public_url", "STORAGE_PUBLIC_URL")
|
_ = viper.BindEnv("storage.public_url", "STORAGE_PUBLIC_URL")
|
||||||
|
|||||||
@@ -11,10 +11,8 @@ import (
|
|||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/config"
|
|
||||||
"github.com/wangjia/jiu/backend/internal/middleware"
|
"github.com/wangjia/jiu/backend/internal/middleware"
|
||||||
"github.com/wangjia/jiu/backend/internal/service"
|
"github.com/wangjia/jiu/backend/internal/service"
|
||||||
"github.com/wangjia/jiu/backend/internal/util"
|
|
||||||
"github.com/wangjia/jiu/backend/testutil"
|
"github.com/wangjia/jiu/backend/testutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -301,16 +299,13 @@ func TestAuthHandler_Login_ResponseContainsUserInfo(t *testing.T) {
|
|||||||
// #8 心跳 /auth/ping 回带授权概况,客户端据此免去单独轮询 /license/info。
|
// #8 心跳 /auth/ping 回带授权概况,客户端据此免去单独轮询 /license/info。
|
||||||
func TestAuthHandler_Ping_ReturnsLicense(t *testing.T) {
|
func TestAuthHandler_Ping_ReturnsLicense(t *testing.T) {
|
||||||
db := testutil.SetupTestDB()
|
db := testutil.SetupTestDB()
|
||||||
priv, _, err := util.GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
config.C.License.Ed25519PrivateKey = priv
|
|
||||||
|
|
||||||
shop := testutil.CreateTestShop(db, "AHPING")
|
shop := testutil.CreateTestShop(db, "AHPING")
|
||||||
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
|
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
|
||||||
|
|
||||||
authSvc := service.NewAuthService(db)
|
authSvc := service.NewAuthService(db)
|
||||||
// 登录触发首登自动 trial。
|
// 登录触发首登自动 trial。
|
||||||
_, _, err = authSvc.Login("AHPING", "admin", "password123", service.DeviceInfo{Platform: "windows"})
|
_, _, err := authSvc.Login("AHPING", "admin", "password123", service.DeviceInfo{Platform: "windows"})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
licSvc := service.NewLicenseService(db)
|
licSvc := service.NewLicenseService(db)
|
||||||
|
|||||||
@@ -17,11 +17,12 @@ func NewLicenseHandler(svc *service.LicenseService) *LicenseHandler {
|
|||||||
return &LicenseHandler{svc: svc}
|
return &LicenseHandler{svc: svc}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Activate POST /api/v1/license/activate
|
// Activate POST /api/v1/license/activate — 兑换激活码(时长券),把时长叠加到门店授权。
|
||||||
|
// 路由名保留 activate 以兼容客户端;内部走 Redeem 逻辑。
|
||||||
func (h *LicenseHandler) Activate(c *gin.Context) {
|
func (h *LicenseHandler) Activate(c *gin.Context) {
|
||||||
shopID := middleware.GetShopID(c)
|
shopID := middleware.GetShopID(c)
|
||||||
var req struct {
|
var req struct {
|
||||||
LicenseKey string `json:"license_key" binding:"required"`
|
LicenseKey string `json:"license_key" binding:"required"` // 承载激活码(短码)
|
||||||
DeviceID string `json:"device_id" binding:"required"`
|
DeviceID string `json:"device_id" binding:"required"`
|
||||||
DeviceName string `json:"device_name"`
|
DeviceName string `json:"device_name"`
|
||||||
Platform string `json:"platform"`
|
Platform string `json:"platform"`
|
||||||
@@ -31,7 +32,7 @@ func (h *LicenseHandler) Activate(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
lic, err := h.svc.Activate(shopID, req.LicenseKey, req.DeviceID, req.DeviceName, req.Platform)
|
lic, err := h.svc.Redeem(shopID, req.LicenseKey, req.DeviceID, req.DeviceName, req.Platform)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
return
|
return
|
||||||
|
|||||||
@@ -19,28 +19,32 @@ func TestLicenseHandler_Activate_Success(t *testing.T) {
|
|||||||
token := getAuthToken(user.ID, shop.ID, "admin")
|
token := getAuthToken(user.ID, shop.ID, "admin")
|
||||||
r := setupProtectedRouter(db)
|
r := setupProtectedRouter(db)
|
||||||
|
|
||||||
expiry := time.Now().Add(30 * 24 * time.Hour)
|
// 码池里放一张未用兑换码(30 天)
|
||||||
lic := &model.License{
|
require.NoError(t, db.Create(&model.LicenseCode{
|
||||||
ShopID: shop.ID,
|
Code: "JIUKUTEST0001", Type: "annual", Tier: "standard",
|
||||||
LicenseKey: "LHACT-BBBBB-CCCCC-DDDDD",
|
DurationDays: 30, MaxDevices: 3, Status: "unused",
|
||||||
IsActive: true,
|
}).Error)
|
||||||
ExpiresAt: &expiry,
|
|
||||||
MaxDevices: 3,
|
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
|
|
||||||
|
// 请求带连字符/会被归一化为 JIUKUTEST0001
|
||||||
w := makeRequest(r, "POST", "/api/v1/license/activate", token, map[string]interface{}{
|
w := makeRequest(r, "POST", "/api/v1/license/activate", token, map[string]interface{}{
|
||||||
"license_key": "LHACT-BBBBB-CCCCC-DDDDD",
|
"license_key": "JIUKU-TEST-0001",
|
||||||
"device_id": "device-123",
|
"device_id": "device-123",
|
||||||
"device_name": "Test Machine",
|
"device_name": "Test Machine",
|
||||||
"platform": "windows",
|
"platform": "windows",
|
||||||
})
|
})
|
||||||
assert.Equal(t, http.StatusOK, w.Code)
|
assert.Equal(t, http.StatusOK, w.Code)
|
||||||
|
|
||||||
// Verify device was recorded in license_devices
|
// 兑换后门店生成授权行 + 设备绑定
|
||||||
|
var lic model.License
|
||||||
|
require.NoError(t, db.Where("shop_id = ?", shop.ID).First(&lic).Error)
|
||||||
var dev model.LicenseDevice
|
var dev model.LicenseDevice
|
||||||
require.NoError(t, db.Where("license_id = ? AND device_id = ?", lic.ID, "device-123").First(&dev).Error)
|
require.NoError(t, db.Where("license_id = ? AND device_id = ?", lic.ID, "device-123").First(&dev).Error)
|
||||||
assert.Equal(t, "Test Machine", dev.DeviceName)
|
assert.Equal(t, "Test Machine", dev.DeviceName)
|
||||||
|
|
||||||
|
// 码被标记已用
|
||||||
|
var lc model.LicenseCode
|
||||||
|
require.NoError(t, db.Where("code = ?", "JIUKUTEST0001").First(&lc).Error)
|
||||||
|
assert.Equal(t, "redeemed", lc.Status)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLicenseHandler_Activate_MissingFields(t *testing.T) {
|
func TestLicenseHandler_Activate_MissingFields(t *testing.T) {
|
||||||
@@ -84,40 +88,31 @@ func TestLicenseHandler_Activate_DeviceLimitExceeded(t *testing.T) {
|
|||||||
token := getAuthToken(user.ID, shop.ID, "admin")
|
token := getAuthToken(user.ID, shop.ID, "admin")
|
||||||
r := setupProtectedRouter(db)
|
r := setupProtectedRouter(db)
|
||||||
|
|
||||||
|
// 既有授权:1 个设备名额已占满
|
||||||
|
future := time.Now().Add(30 * 24 * time.Hour)
|
||||||
lic := &model.License{
|
lic := &model.License{
|
||||||
ShopID: shop.ID, LicenseKey: "LHBND-BBBBB-CCCCC-DDDDD", IsActive: true, MaxDevices: 1,
|
ShopID: shop.ID, LicenseKey: "TRIAL-LH004", IsActive: true, MaxDevices: 1, ExpiresAt: &future,
|
||||||
}
|
}
|
||||||
require.NoError(t, db.Create(lic).Error)
|
require.NoError(t, db.Create(lic).Error)
|
||||||
// Fill the single allowed slot
|
|
||||||
require.NoError(t, db.Create(&model.LicenseDevice{
|
require.NoError(t, db.Create(&model.LicenseDevice{
|
||||||
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "existing-device",
|
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "existing-device",
|
||||||
}).Error)
|
}).Error)
|
||||||
|
// 码不提升设备上限(max_devices=0)
|
||||||
|
require.NoError(t, db.Create(&model.LicenseCode{
|
||||||
|
Code: "JIUKUDEVLIMIT1", Type: "annual", Tier: "standard",
|
||||||
|
DurationDays: 365, MaxDevices: 0, Status: "unused",
|
||||||
|
}).Error)
|
||||||
|
|
||||||
w := makeRequest(r, "POST", "/api/v1/license/activate", token, map[string]interface{}{
|
w := makeRequest(r, "POST", "/api/v1/license/activate", token, map[string]interface{}{
|
||||||
"license_key": "LHBND-BBBBB-CCCCC-DDDDD",
|
"license_key": "JIUKU-DEVL-IMIT1",
|
||||||
"device_id": "different-device",
|
"device_id": "different-device",
|
||||||
})
|
})
|
||||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
assert.Equal(t, http.StatusBadRequest, w.Code)
|
||||||
}
|
|
||||||
|
|
||||||
func TestLicenseHandler_Activate_Expired(t *testing.T) {
|
// 设备超限 → 整笔回滚:码仍未使用
|
||||||
db := testutil.SetupTestDB()
|
var lc model.LicenseCode
|
||||||
shop := testutil.CreateTestShop(db, "LH005")
|
require.NoError(t, db.Where("code = ?", "JIUKUDEVLIMIT1").First(&lc).Error)
|
||||||
user := testutil.CreateTestUser(db, shop.ID, "admin", "pass", "admin")
|
assert.Equal(t, "unused", lc.Status)
|
||||||
token := getAuthToken(user.ID, shop.ID, "admin")
|
|
||||||
r := setupProtectedRouter(db)
|
|
||||||
|
|
||||||
expiry := time.Now().Add(-24 * time.Hour)
|
|
||||||
lic := &model.License{
|
|
||||||
ShopID: shop.ID, LicenseKey: "LHEXP-BBBBB-CCCCC-DDDDD", IsActive: true, ExpiresAt: &expiry, MaxDevices: 3,
|
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
|
|
||||||
w := makeRequest(r, "POST", "/api/v1/license/activate", token, map[string]interface{}{
|
|
||||||
"license_key": "LHEXP-BBBBB-CCCCC-DDDDD",
|
|
||||||
"device_id": "device-123",
|
|
||||||
})
|
|
||||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLicenseHandler_Activate_NoAuth(t *testing.T) {
|
func TestLicenseHandler_Activate_NoAuth(t *testing.T) {
|
||||||
|
|||||||
@@ -60,7 +60,7 @@ func TestLicenseGuardUsesLiveDBPhase(t *testing.T) {
|
|||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
|
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
|
||||||
shop_id INTEGER NOT NULL,
|
shop_id INTEGER NOT NULL,
|
||||||
license_key TEXT, type TEXT, expires_at DATETIME,
|
license_key TEXT, type TEXT, tier TEXT DEFAULT 'standard', expires_at DATETIME,
|
||||||
is_active INTEGER DEFAULT 1, max_devices INTEGER DEFAULT 3,
|
is_active INTEGER DEFAULT 1, max_devices INTEGER DEFAULT 3,
|
||||||
features TEXT, device_id TEXT, activated_at DATETIME
|
features TEXT, device_id TEXT, activated_at DATETIME
|
||||||
)`).Error)
|
)`).Error)
|
||||||
@@ -125,7 +125,7 @@ func TestLicenseGuardRevokedAndInvalidation(t *testing.T) {
|
|||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
|
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
|
||||||
shop_id INTEGER NOT NULL,
|
shop_id INTEGER NOT NULL,
|
||||||
license_key TEXT, type TEXT, expires_at DATETIME,
|
license_key TEXT, type TEXT, tier TEXT DEFAULT 'standard', expires_at DATETIME,
|
||||||
is_active INTEGER DEFAULT 1, max_devices INTEGER DEFAULT 3,
|
is_active INTEGER DEFAULT 1, max_devices INTEGER DEFAULT 3,
|
||||||
features TEXT, device_id TEXT, activated_at DATETIME
|
features TEXT, device_id TEXT, activated_at DATETIME
|
||||||
)`).Error)
|
)`).Error)
|
||||||
|
|||||||
@@ -7,7 +7,9 @@ type License struct {
|
|||||||
ShopID uint64 `gorm:"not null;index" json:"shop_id"`
|
ShopID uint64 `gorm:"not null;index" json:"shop_id"`
|
||||||
LicenseKey string `gorm:"size:768;uniqueIndex" json:"license_key"`
|
LicenseKey string `gorm:"size:768;uniqueIndex" json:"license_key"`
|
||||||
Type string `gorm:"type:enum('trial','monthly','annual','lifetime');default:'trial'" json:"type"`
|
Type string `gorm:"type:enum('trial','monthly','annual','lifetime');default:'trial'" json:"type"`
|
||||||
ExpiresAt *time.Time `json:"expires_at"`
|
// Tier 当前权益档位(标准/pro/max…),随兑换码写入;当前默认 'standard',暂不据此做能力差异。
|
||||||
|
Tier string `gorm:"size:32;not null;default:'standard'" json:"tier"`
|
||||||
|
ExpiresAt *time.Time `json:"expires_at"`
|
||||||
IsActive bool `gorm:"default:true" json:"is_active"`
|
IsActive bool `gorm:"default:true" json:"is_active"`
|
||||||
MaxDevices int `gorm:"default:3" json:"max_devices"`
|
MaxDevices int `gorm:"default:3" json:"max_devices"`
|
||||||
Features JSON `gorm:"type:json" json:"features,omitempty"`
|
Features JSON `gorm:"type:json" json:"features,omitempty"`
|
||||||
|
|||||||
@@ -0,0 +1,27 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// LicenseCode 兑换券(激活码)。由平台方批量生成,用户购买/活动获得后在自己门店兑换。
|
||||||
|
// 每张码代表一段时长(duration_days),兑换时叠加到门店当前到期时间上;一码一次(status 控制)。
|
||||||
|
type LicenseCode struct {
|
||||||
|
ID uint64 `gorm:"primaryKey;autoIncrement" json:"id"`
|
||||||
|
Code string `gorm:"size:32;uniqueIndex;not null" json:"code"` // 归一化大写无连字符存储
|
||||||
|
|
||||||
|
Type string `gorm:"type:enum('trial','monthly','annual','lifetime');default:'annual'" json:"type"` // 计费/时长标签
|
||||||
|
// Tier 档位钩子(类似 pro/max)。当前只用 'standard' 一种,分档能力/消费模式后续设计。
|
||||||
|
Tier string `gorm:"size:32;not null;default:'standard'" json:"tier"`
|
||||||
|
DurationDays int `gorm:"not null;default:0" json:"duration_days"` // 授予时长;0 = 永久(lifetime)
|
||||||
|
MaxDevices int `gorm:"not null;default:0" json:"max_devices"` // 授予设备上限;0 = 不改变现值
|
||||||
|
|
||||||
|
Status string `gorm:"type:enum('unused','redeemed','void');default:'unused';index" json:"status"`
|
||||||
|
RedeemedShopID *uint64 `gorm:"index" json:"redeemed_shop_id,omitempty"`
|
||||||
|
RedeemedAt *time.Time `json:"redeemed_at,omitempty"`
|
||||||
|
RedeemedDeviceID string `gorm:"size:255" json:"redeemed_device_id,omitempty"`
|
||||||
|
|
||||||
|
Batch string `gorm:"size:64" json:"batch,omitempty"` // 发放批次/活动名
|
||||||
|
Note string `gorm:"size:255" json:"note,omitempty"`
|
||||||
|
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
@@ -7,9 +7,7 @@ import (
|
|||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/config"
|
|
||||||
"github.com/wangjia/jiu/backend/internal/model"
|
"github.com/wangjia/jiu/backend/internal/model"
|
||||||
"github.com/wangjia/jiu/backend/internal/util"
|
|
||||||
"github.com/wangjia/jiu/backend/testutil"
|
"github.com/wangjia/jiu/backend/testutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -31,9 +29,6 @@ func TestAuthService_Login_Success(t *testing.T) {
|
|||||||
|
|
||||||
func TestAuthService_Login_AutoTrialOnFirstUse(t *testing.T) {
|
func TestAuthService_Login_AutoTrialOnFirstUse(t *testing.T) {
|
||||||
db := testutil.SetupTestDB()
|
db := testutil.SetupTestDB()
|
||||||
priv, _, err := util.GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
config.C.License.Ed25519PrivateKey = priv
|
|
||||||
|
|
||||||
shop := testutil.CreateTestShop(db, "TRIAL001")
|
shop := testutil.CreateTestShop(db, "TRIAL001")
|
||||||
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
|
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
|
||||||
@@ -46,7 +41,7 @@ func TestAuthService_Login_AutoTrialOnFirstUse(t *testing.T) {
|
|||||||
svc := NewAuthService(db)
|
svc := NewAuthService(db)
|
||||||
|
|
||||||
// 首次登录 → 自动签发 30 天 trial
|
// 首次登录 → 自动签发 30 天 trial
|
||||||
_, _, err = svc.Login("TRIAL001", "admin", "password123", DeviceInfo{Platform: "windows"})
|
_, _, err := svc.Login("TRIAL001", "admin", "password123", DeviceInfo{Platform: "windows"})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
var lic model.License
|
var lic model.License
|
||||||
|
|||||||
@@ -1,18 +1,13 @@
|
|||||||
package service
|
package service
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/hmac"
|
|
||||||
"crypto/sha256"
|
|
||||||
"encoding/base32"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"log"
|
"log"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/config"
|
|
||||||
"github.com/wangjia/jiu/backend/internal/middleware"
|
"github.com/wangjia/jiu/backend/internal/middleware"
|
||||||
"github.com/wangjia/jiu/backend/internal/model"
|
"github.com/wangjia/jiu/backend/internal/model"
|
||||||
"github.com/wangjia/jiu/backend/internal/util"
|
"github.com/wangjia/jiu/backend/internal/util"
|
||||||
@@ -22,7 +17,12 @@ var (
|
|||||||
ErrLicenseNotFound = errors.New("license not found")
|
ErrLicenseNotFound = errors.New("license not found")
|
||||||
ErrLicenseInactive = errors.New("license is inactive")
|
ErrLicenseInactive = errors.New("license is inactive")
|
||||||
ErrLicenseExpired = errors.New("license has expired")
|
ErrLicenseExpired = errors.New("license has expired")
|
||||||
ErrDeviceLimitExceed = errors.New("device limit reached — deactivate another device first")
|
ErrDeviceLimitExceed = errors.New("设备数已达上限,请先在其它设备上解绑后再试")
|
||||||
|
|
||||||
|
// 兑换券(激活码)相关错误,消息直接面向用户。
|
||||||
|
ErrCodeNotFound = errors.New("无效激活码")
|
||||||
|
ErrCodeUsed = errors.New("该激活码已被使用")
|
||||||
|
ErrCodeVoid = errors.New("该激活码已失效")
|
||||||
)
|
)
|
||||||
|
|
||||||
type LicenseService struct {
|
type LicenseService struct {
|
||||||
@@ -33,69 +33,150 @@ func NewLicenseService(db *gorm.DB) *LicenseService {
|
|||||||
return &LicenseService{db: db}
|
return &LicenseService{db: db}
|
||||||
}
|
}
|
||||||
|
|
||||||
// GenerateKey 生成许可证激活码
|
// Redeem 兑换一张激活码(时长券):校验码有效且未使用 → 把时长叠加到门店当前授权的到期时间
|
||||||
// 格式:HMAC-SHA256(shopID+licenseType+expiry, secret) → base32, 每5字符加'-'
|
// → 标记码已用 → 绑定本设备。整个过程在单事务内,check-then-act 用 FOR UPDATE 锁行,
|
||||||
func GenerateKey(shopID uint64, licenseType string, expiresAt *time.Time) string {
|
// 保证一码只能被成功兑换一次(并发下恰一个成功)。
|
||||||
payload := fmt.Sprintf("%d:%s", shopID, licenseType)
|
//
|
||||||
if expiresAt != nil {
|
// 时长叠加规则:新到期 = max(今天, 当前到期) + duration_days;duration_days=0 视为永久(到期置 NULL)。
|
||||||
payload += ":" + expiresAt.Format("20060102")
|
func (s *LicenseService) Redeem(shopID uint64, rawCode, deviceID, deviceName, platform string) (*model.License, error) {
|
||||||
}
|
code := util.NormalizeCode(rawCode)
|
||||||
mac := hmac.New(sha256.New, []byte(config.C.License.HMACSecret))
|
if code == "" {
|
||||||
mac.Write([]byte(payload))
|
return nil, ErrCodeNotFound
|
||||||
raw := base32.StdEncoding.WithPadding(base32.NoPadding).EncodeToString(mac.Sum(nil))
|
|
||||||
// 截取前20字符,分4段,每段5字符
|
|
||||||
raw = strings.ToUpper(raw)[:20]
|
|
||||||
return fmt.Sprintf("%s-%s-%s-%s", raw[0:5], raw[5:10], raw[10:15], raw[15:20])
|
|
||||||
}
|
|
||||||
|
|
||||||
// Activate 激活许可证并绑定设备到 license_devices 表。
|
|
||||||
// 若该设备已绑定,则更新 last_seen_at(幂等)。
|
|
||||||
// 若是新设备,则校验是否超出 max_devices 上限。
|
|
||||||
func (s *LicenseService) Activate(shopID uint64, licenseKey, deviceID, deviceName, platform string) (*model.License, error) {
|
|
||||||
var lic model.License
|
|
||||||
if err := s.db.Where("license_key = ? AND shop_id = ?", licenseKey, shopID).First(&lic).Error; err != nil {
|
|
||||||
return nil, ErrLicenseNotFound
|
|
||||||
}
|
|
||||||
if !lic.IsActive {
|
|
||||||
return nil, ErrLicenseInactive
|
|
||||||
}
|
|
||||||
if lic.ExpiresAt != nil && time.Now().After(*lic.ExpiresAt) {
|
|
||||||
return nil, ErrLicenseExpired
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 激活成功即清除该店 phase 缓存:续费/换新授权码后写权限即时恢复,不必等 30s TTL。
|
var result model.License
|
||||||
defer middleware.InvalidateLicensePhase(shopID)
|
err := s.db.Transaction(func(tx *gorm.DB) error {
|
||||||
|
// 1) 锁定并校验兑换码
|
||||||
var existing model.LicenseDevice
|
var lc model.LicenseCode
|
||||||
err := s.db.Where("license_id = ? AND device_id = ?", lic.ID, deviceID).First(&existing).Error
|
if err := tx.Set("gorm:query_option", "FOR UPDATE").
|
||||||
if err == nil {
|
Where("code = ?", code).First(&lc).Error; err != nil {
|
||||||
// Device already bound — just touch last_seen_at (handled by autoUpdateTime)
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
if err := s.db.Model(&existing).Update("device_name", deviceName).Error; err != nil {
|
return ErrCodeNotFound
|
||||||
return nil, err
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch lc.Status {
|
||||||
|
case "redeemed":
|
||||||
|
return ErrCodeUsed
|
||||||
|
case "void":
|
||||||
|
return ErrCodeVoid
|
||||||
}
|
}
|
||||||
return &lic, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// New device — enforce max_devices
|
now := time.Now()
|
||||||
var count int64
|
|
||||||
if err := s.db.Model(&model.LicenseDevice{}).Where("license_id = ?", lic.ID).Count(&count).Error; err != nil {
|
// 2) 取本店最新有效授权行(锁行);无则新建一行
|
||||||
|
var lic model.License
|
||||||
|
err := tx.Set("gorm:query_option", "FOR UPDATE").
|
||||||
|
Where("shop_id = ? AND is_active = ?", shopID, true).
|
||||||
|
Order("id DESC").First(&lic).Error
|
||||||
|
creating := errors.Is(err, gorm.ErrRecordNotFound)
|
||||||
|
if err != nil && !creating {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3) 计算叠加后的到期时间
|
||||||
|
var newExpires *time.Time
|
||||||
|
if lc.DurationDays > 0 {
|
||||||
|
base := now
|
||||||
|
if lic.ExpiresAt != nil && lic.ExpiresAt.After(now) {
|
||||||
|
base = *lic.ExpiresAt
|
||||||
|
}
|
||||||
|
t := base.Add(time.Duration(lc.DurationDays) * 24 * time.Hour)
|
||||||
|
newExpires = &t
|
||||||
|
} // duration_days==0 → 永久授权,newExpires 保持 nil
|
||||||
|
|
||||||
|
maxDevices := lic.MaxDevices
|
||||||
|
if lc.MaxDevices > maxDevices {
|
||||||
|
maxDevices = lc.MaxDevices
|
||||||
|
}
|
||||||
|
if maxDevices == 0 {
|
||||||
|
maxDevices = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
if creating {
|
||||||
|
lic = model.License{
|
||||||
|
ShopID: shopID,
|
||||||
|
LicenseKey: "REDEEM-" + uuid.New().String(),
|
||||||
|
Type: lc.Type,
|
||||||
|
Tier: lc.Tier,
|
||||||
|
ExpiresAt: newExpires,
|
||||||
|
IsActive: true,
|
||||||
|
MaxDevices: maxDevices,
|
||||||
|
}
|
||||||
|
if err := tx.Create(&lic).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 不动 license_key(保留原值,licenses 行只表示"当前权益")
|
||||||
|
if err := tx.Model(&lic).Updates(map[string]any{
|
||||||
|
"type": lc.Type,
|
||||||
|
"tier": lc.Tier,
|
||||||
|
"expires_at": newExpires,
|
||||||
|
"is_active": true,
|
||||||
|
"max_devices": maxDevices,
|
||||||
|
}).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
lic.ExpiresAt = newExpires
|
||||||
|
lic.Type = lc.Type
|
||||||
|
lic.Tier = lc.Tier
|
||||||
|
lic.MaxDevices = maxDevices
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4) 绑定本设备(幂等 + max_devices 上限校验)
|
||||||
|
if err := bindDevice(tx, &lic, deviceID, deviceName, platform); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 5) 标记码已用
|
||||||
|
if err := tx.Model(&model.LicenseCode{}).Where("id = ?", lc.ID).Updates(map[string]any{
|
||||||
|
"status": "redeemed",
|
||||||
|
"redeemed_shop_id": shopID,
|
||||||
|
"redeemed_at": now,
|
||||||
|
"redeemed_device_id": deviceID,
|
||||||
|
}).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
result = lic
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if int(count) >= lic.MaxDevices {
|
|
||||||
return nil, ErrDeviceLimitExceed
|
|
||||||
}
|
|
||||||
|
|
||||||
dev := model.LicenseDevice{
|
// 事务提交后再失效 phase 缓存:此刻新到期对其它连接已可见,写权限即时恢复,不必等 30s TTL。
|
||||||
|
middleware.InvalidateLicensePhase(shopID)
|
||||||
|
return &result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// bindDevice 把设备绑定到许可证(幂等:已绑则更新名称;新设备则校验 max_devices 上限)。
|
||||||
|
func bindDevice(tx *gorm.DB, lic *model.License, deviceID, deviceName, platform string) error {
|
||||||
|
if deviceID == "" {
|
||||||
|
return nil // 无设备信息(如服务端工具调用)则跳过绑定
|
||||||
|
}
|
||||||
|
var existing model.LicenseDevice
|
||||||
|
err := tx.Where("license_id = ? AND device_id = ?", lic.ID, deviceID).First(&existing).Error
|
||||||
|
if err == nil {
|
||||||
|
return tx.Model(&existing).Update("device_name", deviceName).Error
|
||||||
|
}
|
||||||
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var count int64
|
||||||
|
if err := tx.Model(&model.LicenseDevice{}).Where("license_id = ?", lic.ID).Count(&count).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if int(count) >= lic.MaxDevices {
|
||||||
|
return ErrDeviceLimitExceed
|
||||||
|
}
|
||||||
|
return tx.Create(&model.LicenseDevice{
|
||||||
LicenseID: lic.ID,
|
LicenseID: lic.ID,
|
||||||
ShopID: shopID,
|
ShopID: lic.ShopID,
|
||||||
DeviceID: deviceID,
|
DeviceID: deviceID,
|
||||||
DeviceName: deviceName,
|
DeviceName: deviceName,
|
||||||
Platform: platform,
|
Platform: platform,
|
||||||
}
|
}).Error
|
||||||
if err := s.db.Create(&dev).Error; err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &lic, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Verify 验证设备许可证(客户端启动时调用)。
|
// Verify 验证设备许可证(客户端启动时调用)。
|
||||||
@@ -188,33 +269,15 @@ func (s *LicenseService) Deactivate(shopID uint64, deviceID string) error {
|
|||||||
Delete(&model.LicenseDevice{}).Error
|
Delete(&model.LicenseDevice{}).Error
|
||||||
}
|
}
|
||||||
|
|
||||||
// issueTrialLicense 为门店签发并写入一条 30 天 trial license。
|
// issueTrialLicense 为门店写入一条 30 天 trial license。直接建行(无需签名/私钥),
|
||||||
// 私钥未配置或签发/落库失败时返回 error,由调用方决定如何处理(注册路径 Fatal、
|
// license_key 用合成唯一值占位以满足 NOT NULL/UNIQUE。落库失败返回 error。
|
||||||
// 登录路径仅记日志)。
|
|
||||||
func issueTrialLicense(db *gorm.DB, shopID uint64) error {
|
func issueTrialLicense(db *gorm.DB, shopID uint64) error {
|
||||||
privKey := config.C.License.Ed25519PrivateKey
|
|
||||||
if privKey == "" {
|
|
||||||
return fmt.Errorf("ed25519 private key not configured")
|
|
||||||
}
|
|
||||||
|
|
||||||
expiresAt := time.Now().Add(30 * 24 * time.Hour)
|
expiresAt := time.Now().Add(30 * 24 * time.Hour)
|
||||||
expiresUnix := expiresAt.Unix()
|
|
||||||
payload := util.LicensePayload{
|
|
||||||
ShopID: shopID,
|
|
||||||
Type: "trial",
|
|
||||||
IssuedAt: time.Now().Unix(),
|
|
||||||
ExpiresAt: &expiresUnix,
|
|
||||||
MaxDevices: 1,
|
|
||||||
}
|
|
||||||
token, err := util.IssueLicenseToken(payload, privKey)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
lic := model.License{
|
lic := model.License{
|
||||||
ShopID: shopID,
|
ShopID: shopID,
|
||||||
LicenseKey: token,
|
LicenseKey: "TRIAL-" + uuid.New().String(),
|
||||||
Type: "trial",
|
Type: "trial",
|
||||||
|
Tier: "standard",
|
||||||
ExpiresAt: &expiresAt,
|
ExpiresAt: &expiresAt,
|
||||||
IsActive: true,
|
IsActive: true,
|
||||||
MaxDevices: 1,
|
MaxDevices: 1,
|
||||||
@@ -228,12 +291,8 @@ func issueTrialLicense(db *gorm.DB, shopID uint64) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// createTrialLicense 在注册事务中为新门店签发 30 天 trial license。
|
// createTrialLicense 在注册事务中为新门店写入 30 天 trial license。失败仅记日志、不阻断注册。
|
||||||
// 私钥未配置时 Fatal,防止生产环境静默跳过导致新注册门店无 license。
|
|
||||||
func createTrialLicense(tx *gorm.DB, shopID uint64) {
|
func createTrialLicense(tx *gorm.DB, shopID uint64) {
|
||||||
if config.C.License.Ed25519PrivateKey == "" {
|
|
||||||
log.Fatalf("[license] Ed25519 private key not configured — cannot issue trial for shop %d; set License.Ed25519PrivateKey in config", shopID)
|
|
||||||
}
|
|
||||||
if err := issueTrialLicense(tx, shopID); err != nil {
|
if err := issueTrialLicense(tx, shopID); err != nil {
|
||||||
log.Printf("[license] failed to create trial license for shop %d: %v", shopID, err)
|
log.Printf("[license] failed to create trial license for shop %d: %v", shopID, err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,155 +6,185 @@ import (
|
|||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
|
||||||
"github.com/wangjia/jiu/backend/internal/model"
|
"github.com/wangjia/jiu/backend/internal/model"
|
||||||
|
"github.com/wangjia/jiu/backend/internal/util"
|
||||||
"github.com/wangjia/jiu/backend/testutil"
|
"github.com/wangjia/jiu/backend/testutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestLicenseService_Activate_Success(t *testing.T) {
|
// createCode 在码池写入一张 unused 兑换码(默认 type=annual / tier=standard)。
|
||||||
db := testutil.SetupTestDB()
|
func createCode(t *testing.T, db *gorm.DB, code string, durationDays, maxDevices int) {
|
||||||
shop := testutil.CreateTestShop(db, "LIC001")
|
t.Helper()
|
||||||
|
require.NoError(t, db.Create(&model.LicenseCode{
|
||||||
expiry := time.Now().Add(30 * 24 * time.Hour)
|
Code: util.NormalizeCode(code),
|
||||||
lic := &model.License{
|
Type: "annual",
|
||||||
ShopID: shop.ID,
|
Tier: "standard",
|
||||||
LicenseKey: "AAAAA-BBBBB-CCCCC-DDDDD",
|
DurationDays: durationDays,
|
||||||
IsActive: true,
|
MaxDevices: maxDevices,
|
||||||
ExpiresAt: &expiry,
|
Status: "unused",
|
||||||
MaxDevices: 3,
|
}).Error)
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
|
||||||
result, err := svc.Activate(shop.ID, "AAAAA-BBBBB-CCCCC-DDDDD", "device-001", "Test PC", "windows")
|
|
||||||
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.NotNil(t, result)
|
|
||||||
|
|
||||||
// Verify device record was created
|
|
||||||
var dev model.LicenseDevice
|
|
||||||
require.NoError(t, db.Where("license_id = ? AND device_id = ?", lic.ID, "device-001").First(&dev).Error)
|
|
||||||
assert.Equal(t, "Test PC", dev.DeviceName)
|
|
||||||
assert.Equal(t, "windows", dev.Platform)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLicenseService_Activate_SameDeviceIdempotent(t *testing.T) {
|
func daysFromNow(t *time.Time) float64 {
|
||||||
db := testutil.SetupTestDB()
|
if t == nil {
|
||||||
shop := testutil.CreateTestShop(db, "LIC002")
|
return 0
|
||||||
|
|
||||||
lic := &model.License{
|
|
||||||
ShopID: shop.ID,
|
|
||||||
LicenseKey: "EEEEE-FFFFF-GGGGG-HHHHH",
|
|
||||||
IsActive: true,
|
|
||||||
MaxDevices: 3,
|
|
||||||
}
|
}
|
||||||
require.NoError(t, db.Create(lic).Error)
|
return time.Until(*t).Hours() / 24
|
||||||
// Pre-bind the device
|
}
|
||||||
require.NoError(t, db.Create(&model.LicenseDevice{
|
|
||||||
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "same-device",
|
// 首次兑换(门店尚无授权行):新建一行,到期 = 今天 + 时长,码标记已用并绑定设备。
|
||||||
|
func TestLicenseService_Redeem_NewShop(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM001")
|
||||||
|
createCode(t, db, "JIUKU-AAAA-BBBB", 365, 3)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
// 传小写 + 连字符,验证归一化
|
||||||
|
lic, err := svc.Redeem(shop.ID, "jiuku-aaaa-bbbb", "dev-1", "Test PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, lic)
|
||||||
|
assert.InDelta(t, 365, daysFromNow(lic.ExpiresAt), 1)
|
||||||
|
assert.Equal(t, 3, lic.MaxDevices)
|
||||||
|
assert.Equal(t, "standard", lic.Tier)
|
||||||
|
assert.Equal(t, "annual", lic.Type)
|
||||||
|
|
||||||
|
var lc model.LicenseCode
|
||||||
|
require.NoError(t, db.Where("code = ?", "JIUKUAAAABBBB").First(&lc).Error)
|
||||||
|
assert.Equal(t, "redeemed", lc.Status)
|
||||||
|
require.NotNil(t, lc.RedeemedShopID)
|
||||||
|
assert.Equal(t, shop.ID, *lc.RedeemedShopID)
|
||||||
|
|
||||||
|
var dev model.LicenseDevice
|
||||||
|
require.NoError(t, db.Where("license_id = ? AND device_id = ?", lic.ID, "dev-1").First(&dev).Error)
|
||||||
|
assert.Equal(t, "Test PC", dev.DeviceName)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 在既有未过期授权上兑换:到期时间叠加在原到期之后。
|
||||||
|
func TestLicenseService_Redeem_ExtendsExisting(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM002")
|
||||||
|
expiry := time.Now().Add(10 * 24 * time.Hour)
|
||||||
|
require.NoError(t, db.Create(&model.License{
|
||||||
|
ShopID: shop.ID, LicenseKey: "TRIAL-RDM002", Type: "trial", Tier: "standard",
|
||||||
|
ExpiresAt: &expiry, IsActive: true, MaxDevices: 3,
|
||||||
|
}).Error)
|
||||||
|
createCode(t, db, "JIUKU-CCCC-DDDD", 365, 0)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
lic, err := svc.Redeem(shop.ID, "JIUKU-CCCC-DDDD", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
// 原剩 10 天 + 365 ≈ 375
|
||||||
|
assert.InDelta(t, 375, daysFromNow(lic.ExpiresAt), 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 叠加:连兑两张码,时长累加。
|
||||||
|
func TestLicenseService_Redeem_Stacks(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM003")
|
||||||
|
createCode(t, db, "JIUKU-1111-1111", 365, 1)
|
||||||
|
createCode(t, db, "JIUKU-2222-2222", 365, 1)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
_, err := svc.Redeem(shop.ID, "JIUKU-1111-1111", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
lic, err := svc.Redeem(shop.ID, "JIUKU-2222-2222", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.InDelta(t, 730, daysFromNow(lic.ExpiresAt), 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 已过期门店兑换:从今天起算,不在过去叠加。
|
||||||
|
func TestLicenseService_Redeem_ExpiredBase(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM004")
|
||||||
|
past := time.Now().Add(-5 * 24 * time.Hour)
|
||||||
|
require.NoError(t, db.Create(&model.License{
|
||||||
|
ShopID: shop.ID, LicenseKey: "TRIAL-RDM004", Type: "trial", Tier: "standard",
|
||||||
|
ExpiresAt: &past, IsActive: true, MaxDevices: 1,
|
||||||
|
}).Error)
|
||||||
|
createCode(t, db, "JIUKU-EEEE-FFFF", 30, 0)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
lic, err := svc.Redeem(shop.ID, "JIUKU-EEEE-FFFF", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.InDelta(t, 30, daysFromNow(lic.ExpiresAt), 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 永久码(duration=0):到期置 NULL。
|
||||||
|
func TestLicenseService_Redeem_Lifetime(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM005")
|
||||||
|
createCode(t, db, "JIUKU-LIFE-TIME", 0, 1)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
lic, err := svc.Redeem(shop.ID, "JIUKU-LIFE-TIME", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.Nil(t, lic.ExpiresAt)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 一码一次:同码兑换两次,第二次失败。
|
||||||
|
func TestLicenseService_Redeem_AlreadyUsed(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM006")
|
||||||
|
createCode(t, db, "JIUKU-USED-ONCE", 365, 1)
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
_, err := svc.Redeem(shop.ID, "JIUKU-USED-ONCE", "dev-1", "PC", "windows")
|
||||||
|
require.NoError(t, err)
|
||||||
|
_, err = svc.Redeem(shop.ID, "JIUKU-USED-ONCE", "dev-1", "PC", "windows")
|
||||||
|
assert.Equal(t, ErrCodeUsed, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLicenseService_Redeem_NotFound(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM007")
|
||||||
|
|
||||||
|
svc := NewLicenseService(db)
|
||||||
|
_, err := svc.Redeem(shop.ID, "JIUKU-NONE-XXXX", "dev-1", "", "")
|
||||||
|
assert.Equal(t, ErrCodeNotFound, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLicenseService_Redeem_Void(t *testing.T) {
|
||||||
|
db := testutil.SetupTestDB()
|
||||||
|
shop := testutil.CreateTestShop(db, "RDM008")
|
||||||
|
require.NoError(t, db.Create(&model.LicenseCode{
|
||||||
|
Code: "JIUKUVOIDCODE0", Type: "annual", Tier: "standard",
|
||||||
|
DurationDays: 365, Status: "void",
|
||||||
}).Error)
|
}).Error)
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
svc := NewLicenseService(db)
|
||||||
// Re-activating same device should succeed (idempotent)
|
_, err := svc.Redeem(shop.ID, "JIUKU-VOID-CODE0", "dev-1", "", "")
|
||||||
result, err := svc.Activate(shop.ID, "EEEEE-FFFFF-GGGGG-HHHHH", "same-device", "Updated Name", "windows")
|
assert.Equal(t, ErrCodeVoid, err)
|
||||||
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.NotNil(t, result)
|
|
||||||
|
|
||||||
// Still only one device record
|
|
||||||
var count int64
|
|
||||||
db.Model(&model.LicenseDevice{}).Where("license_id = ?", lic.ID).Count(&count)
|
|
||||||
assert.Equal(t, int64(1), count)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLicenseService_Activate_DeviceLimitExceeded(t *testing.T) {
|
// 设备超上限:兑换整体回滚——码保持 unused、到期不变。
|
||||||
|
func TestLicenseService_Redeem_DeviceLimitRollsBack(t *testing.T) {
|
||||||
db := testutil.SetupTestDB()
|
db := testutil.SetupTestDB()
|
||||||
shop := testutil.CreateTestShop(db, "LIC003")
|
shop := testutil.CreateTestShop(db, "RDM009")
|
||||||
|
expiry := time.Now().Add(10 * 24 * time.Hour)
|
||||||
lic := &model.License{
|
lic := &model.License{
|
||||||
ShopID: shop.ID,
|
ShopID: shop.ID, LicenseKey: "TRIAL-RDM009", Type: "trial", Tier: "standard",
|
||||||
LicenseKey: "IIIII-JJJJJ-KKKKK-LLLLL",
|
ExpiresAt: &expiry, IsActive: true, MaxDevices: 1,
|
||||||
IsActive: true,
|
|
||||||
MaxDevices: 2,
|
|
||||||
}
|
}
|
||||||
require.NoError(t, db.Create(lic).Error)
|
require.NoError(t, db.Create(lic).Error)
|
||||||
// Fill up the device limit
|
// 已占满 1 个设备名额
|
||||||
require.NoError(t, db.Create(&model.LicenseDevice{LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "dev-1"}).Error)
|
require.NoError(t, db.Create(&model.LicenseDevice{
|
||||||
require.NoError(t, db.Create(&model.LicenseDevice{LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "dev-2"}).Error)
|
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "old-dev",
|
||||||
|
}).Error)
|
||||||
|
createCode(t, db, "JIUKU-DEVL-IMIT", 365, 0) // 不提升设备上限
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
svc := NewLicenseService(db)
|
||||||
result, err := svc.Activate(shop.ID, "IIIII-JJJJJ-KKKKK-LLLLL", "dev-3", "", "")
|
_, err := svc.Redeem(shop.ID, "JIUKU-DEVL-IMIT", "new-dev", "New PC", "windows")
|
||||||
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, ErrDeviceLimitExceed, err)
|
assert.Equal(t, ErrDeviceLimitExceed, err)
|
||||||
assert.Nil(t, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLicenseService_Activate_NotFound(t *testing.T) {
|
// 码仍未使用(回滚)
|
||||||
db := testutil.SetupTestDB()
|
var lc model.LicenseCode
|
||||||
shop := testutil.CreateTestShop(db, "LIC004")
|
require.NoError(t, db.Where("code = ?", "JIUKUDEVLIMIT").First(&lc).Error)
|
||||||
|
assert.Equal(t, "unused", lc.Status)
|
||||||
svc := NewLicenseService(db)
|
// 到期不变
|
||||||
result, err := svc.Activate(shop.ID, "NONEX-ISTEN-TTTTT-LICCC", "device-001", "", "")
|
var after model.License
|
||||||
|
require.NoError(t, db.First(&after, lic.ID).Error)
|
||||||
assert.Error(t, err)
|
assert.InDelta(t, 10, daysFromNow(after.ExpiresAt), 1)
|
||||||
assert.Equal(t, ErrLicenseNotFound, err)
|
|
||||||
assert.Nil(t, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLicenseService_Activate_WrongShop(t *testing.T) {
|
|
||||||
db := testutil.SetupTestDB()
|
|
||||||
shop := testutil.CreateTestShop(db, "LIC004B")
|
|
||||||
otherShop := testutil.CreateTestShop(db, "LIC004C")
|
|
||||||
|
|
||||||
lic := &model.License{
|
|
||||||
ShopID: shop.ID, LicenseKey: "OTHSH-BBBBB-CCCCC-DDDDD", IsActive: true, MaxDevices: 3,
|
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
|
||||||
// otherShop cannot activate a license belonging to shop
|
|
||||||
result, err := svc.Activate(otherShop.ID, "OTHSH-BBBBB-CCCCC-DDDDD", "device-001", "", "")
|
|
||||||
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, ErrLicenseNotFound, err)
|
|
||||||
assert.Nil(t, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLicenseService_Activate_Inactive(t *testing.T) {
|
|
||||||
db := testutil.SetupTestDB()
|
|
||||||
shop := testutil.CreateTestShop(db, "LIC005")
|
|
||||||
|
|
||||||
lic := &model.License{
|
|
||||||
ShopID: shop.ID, LicenseKey: "MMMMM-NNNNN-OOOOO-PPPPP", IsActive: true, MaxDevices: 3,
|
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
require.NoError(t, db.Model(lic).Update("is_active", false).Error)
|
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
|
||||||
result, err := svc.Activate(shop.ID, "MMMMM-NNNNN-OOOOO-PPPPP", "device-001", "", "")
|
|
||||||
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, ErrLicenseInactive, err)
|
|
||||||
assert.Nil(t, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLicenseService_Activate_Expired(t *testing.T) {
|
|
||||||
db := testutil.SetupTestDB()
|
|
||||||
shop := testutil.CreateTestShop(db, "LIC006")
|
|
||||||
|
|
||||||
expiry := time.Now().Add(-24 * time.Hour)
|
|
||||||
lic := &model.License{
|
|
||||||
ShopID: shop.ID, LicenseKey: "QQQQQ-RRRRR-SSSSS-TTTTT", IsActive: true, ExpiresAt: &expiry, MaxDevices: 3,
|
|
||||||
}
|
|
||||||
require.NoError(t, db.Create(lic).Error)
|
|
||||||
|
|
||||||
svc := NewLicenseService(db)
|
|
||||||
result, err := svc.Activate(shop.ID, "QQQQQ-RRRRR-SSSSS-TTTTT", "device-001", "", "")
|
|
||||||
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, ErrLicenseExpired, err)
|
|
||||||
assert.Nil(t, result)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLicenseService_Verify_Success(t *testing.T) {
|
func TestLicenseService_Verify_Success(t *testing.T) {
|
||||||
@@ -229,17 +259,3 @@ func TestLicenseService_Verify_NoExpiry(t *testing.T) {
|
|||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
require.NotNil(t, result)
|
require.NotNil(t, result)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestGenerateKey(t *testing.T) {
|
|
||||||
testutil.InitConfig()
|
|
||||||
|
|
||||||
expiry := time.Now().Add(30 * 24 * time.Hour)
|
|
||||||
key := GenerateKey(1, "annual", &expiry)
|
|
||||||
|
|
||||||
assert.NotEmpty(t, key)
|
|
||||||
// 格式:XXXXX-XXXXX-XXXXX-XXXXX
|
|
||||||
assert.Equal(t, 23, len(key)) // 4*5 + 3 dashes = 23
|
|
||||||
assert.Equal(t, '-', rune(key[5]))
|
|
||||||
assert.Equal(t, '-', rune(key[11]))
|
|
||||||
assert.Equal(t, '-', rune(key[17]))
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,106 +0,0 @@
|
|||||||
package util
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/ed25519"
|
|
||||||
"crypto/rand"
|
|
||||||
"encoding/base64"
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
ErrInvalidLicenseToken = errors.New("invalid license token")
|
|
||||||
ErrInvalidLicenseSignature = errors.New("invalid license token signature")
|
|
||||||
)
|
|
||||||
|
|
||||||
// LicensePayload is the verified content extracted from a signed license token.
|
|
||||||
type LicensePayload struct {
|
|
||||||
ShopID uint64 `json:"shop_id"`
|
|
||||||
LicenseID uint64 `json:"license_id,omitempty"`
|
|
||||||
Type string `json:"type"` // trial | monthly | annual | lifetime
|
|
||||||
IssuedAt int64 `json:"issued_at"`
|
|
||||||
ExpiresAt *int64 `json:"expires_at,omitempty"` // unix seconds; nil = perpetual
|
|
||||||
MaxDevices int `json:"max_devices"`
|
|
||||||
Features map[string]any `json:"features,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// GenerateEd25519KeyPair generates a new Ed25519 keypair.
|
|
||||||
// Returns standard base64-encoded private key (64 bytes) and public key (32 bytes).
|
|
||||||
// The private key must be stored securely (Bitwarden); the public key goes in config.
|
|
||||||
func GenerateEd25519KeyPair() (privKeyB64, pubKeyB64 string, err error) {
|
|
||||||
pub, priv, err := ed25519.GenerateKey(rand.Reader)
|
|
||||||
if err != nil {
|
|
||||||
return "", "", err
|
|
||||||
}
|
|
||||||
return base64.StdEncoding.EncodeToString(priv),
|
|
||||||
base64.StdEncoding.EncodeToString(pub),
|
|
||||||
nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// IssueLicenseToken signs a LicensePayload with the Ed25519 private key and returns
|
|
||||||
// a compact token: base64url(header).base64url(payload).base64url(signature).
|
|
||||||
// privKeyB64 is the standard base64-encoded 64-byte Ed25519 private key.
|
|
||||||
func IssueLicenseToken(payload LicensePayload, privKeyB64 string) (string, error) {
|
|
||||||
privKeyBytes, err := base64.StdEncoding.DecodeString(privKeyB64)
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("decode private key: %w", err)
|
|
||||||
}
|
|
||||||
if len(privKeyBytes) != ed25519.PrivateKeySize {
|
|
||||||
return "", fmt.Errorf("private key must be %d bytes, got %d", ed25519.PrivateKeySize, len(privKeyBytes))
|
|
||||||
}
|
|
||||||
privKey := ed25519.PrivateKey(privKeyBytes)
|
|
||||||
|
|
||||||
header := rawB64([]byte(`{"alg":"EdDSA","typ":"LIC"}`))
|
|
||||||
payloadJSON, err := json.Marshal(payload)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
body := rawB64(payloadJSON)
|
|
||||||
signingInput := header + "." + body
|
|
||||||
sig := ed25519.Sign(privKey, []byte(signingInput))
|
|
||||||
return signingInput + "." + rawB64(sig), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// VerifyLicenseToken verifies the Ed25519 signature of a license token and returns
|
|
||||||
// the decoded payload. Does NOT check expiry — callers must check ExpiresAt themselves.
|
|
||||||
// pubKeyB64 is the standard base64-encoded 32-byte Ed25519 public key.
|
|
||||||
func VerifyLicenseToken(token, pubKeyB64 string) (*LicensePayload, error) {
|
|
||||||
parts := strings.Split(token, ".")
|
|
||||||
if len(parts) != 3 {
|
|
||||||
return nil, ErrInvalidLicenseToken
|
|
||||||
}
|
|
||||||
|
|
||||||
pubKeyBytes, err := base64.StdEncoding.DecodeString(pubKeyB64)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("decode public key: %w", err)
|
|
||||||
}
|
|
||||||
if len(pubKeyBytes) != ed25519.PublicKeySize {
|
|
||||||
return nil, fmt.Errorf("public key must be %d bytes, got %d", ed25519.PublicKeySize, len(pubKeyBytes))
|
|
||||||
}
|
|
||||||
pubKey := ed25519.PublicKey(pubKeyBytes)
|
|
||||||
|
|
||||||
signingInput := parts[0] + "." + parts[1]
|
|
||||||
sigBytes, err := base64.RawURLEncoding.DecodeString(parts[2])
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrInvalidLicenseToken
|
|
||||||
}
|
|
||||||
if !ed25519.Verify(pubKey, []byte(signingInput), sigBytes) {
|
|
||||||
return nil, ErrInvalidLicenseSignature
|
|
||||||
}
|
|
||||||
|
|
||||||
payloadJSON, err := base64.RawURLEncoding.DecodeString(parts[1])
|
|
||||||
if err != nil {
|
|
||||||
return nil, ErrInvalidLicenseToken
|
|
||||||
}
|
|
||||||
var p LicensePayload
|
|
||||||
if err := json.Unmarshal(payloadJSON, &p); err != nil {
|
|
||||||
return nil, ErrInvalidLicenseToken
|
|
||||||
}
|
|
||||||
return &p, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func rawB64(data []byte) string {
|
|
||||||
return base64.RawURLEncoding.EncodeToString(data)
|
|
||||||
}
|
|
||||||
@@ -1,110 +0,0 @@
|
|||||||
package util
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestLicenseKeyRoundTrip(t *testing.T) {
|
|
||||||
priv, pub, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
exp := time.Now().Add(30 * 24 * time.Hour).Unix()
|
|
||||||
payload := LicensePayload{
|
|
||||||
ShopID: 42,
|
|
||||||
LicenseID: 7,
|
|
||||||
Type: "annual",
|
|
||||||
IssuedAt: time.Now().Unix(),
|
|
||||||
ExpiresAt: &exp,
|
|
||||||
MaxDevices: 3,
|
|
||||||
}
|
|
||||||
|
|
||||||
token, err := IssueLicenseToken(payload, priv)
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.NotEmpty(t, token)
|
|
||||||
|
|
||||||
got, err := VerifyLicenseToken(token, pub)
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, payload.ShopID, got.ShopID)
|
|
||||||
assert.Equal(t, payload.Type, got.Type)
|
|
||||||
assert.Equal(t, payload.MaxDevices, got.MaxDevices)
|
|
||||||
assert.Equal(t, *payload.ExpiresAt, *got.ExpiresAt)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyLicenseToken_TamperedPayload(t *testing.T) {
|
|
||||||
priv, pub, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
exp := time.Now().Add(30 * 24 * time.Hour).Unix()
|
|
||||||
token, err := IssueLicenseToken(LicensePayload{
|
|
||||||
ShopID: 1, Type: "trial", IssuedAt: time.Now().Unix(), ExpiresAt: &exp, MaxDevices: 1,
|
|
||||||
}, priv)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
// Flip the last byte of the signature to simulate tampering
|
|
||||||
tampered := token[:len(token)-2] + "XX"
|
|
||||||
_, err = VerifyLicenseToken(tampered, pub)
|
|
||||||
assert.Error(t, err, "tampered token must be rejected")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyLicenseToken_WrongKey(t *testing.T) {
|
|
||||||
priv, _, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
_, otherPub, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
exp := time.Now().Add(30 * 24 * time.Hour).Unix()
|
|
||||||
token, err := IssueLicenseToken(LicensePayload{
|
|
||||||
ShopID: 1, Type: "trial", IssuedAt: time.Now().Unix(), ExpiresAt: &exp, MaxDevices: 1,
|
|
||||||
}, priv)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
_, err = VerifyLicenseToken(token, otherPub)
|
|
||||||
assert.ErrorIs(t, err, ErrInvalidLicenseSignature)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyLicenseToken_InvalidFormat(t *testing.T) {
|
|
||||||
_, pub, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
_, err = VerifyLicenseToken("not-a-valid-token", pub)
|
|
||||||
assert.ErrorIs(t, err, ErrInvalidLicenseToken)
|
|
||||||
}
|
|
||||||
|
|
||||||
// TestLicenseTokenFitsColumnLimit verifies that a realistic (even worst-case)
|
|
||||||
// license token fits within the VARCHAR(768) column limit imposed by the
|
|
||||||
// InnoDB index constraint (768 chars × 4 bytes/char = 3072 bytes max).
|
|
||||||
func TestLicenseTokenFitsColumnLimit(t *testing.T) {
|
|
||||||
const columnLimit = 768
|
|
||||||
|
|
||||||
priv, _, err := GenerateEd25519KeyPair()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
// Use a large features map to simulate a worst-case payload.
|
|
||||||
exp := time.Now().Add(365 * 24 * time.Hour).Unix()
|
|
||||||
payload := LicensePayload{
|
|
||||||
ShopID: 999999999,
|
|
||||||
LicenseID: 999999999,
|
|
||||||
Type: "lifetime",
|
|
||||||
IssuedAt: time.Now().Unix(),
|
|
||||||
ExpiresAt: &exp,
|
|
||||||
MaxDevices: 99,
|
|
||||||
Features: map[string]any{
|
|
||||||
"finance": true,
|
|
||||||
"inventory": true,
|
|
||||||
"reports": true,
|
|
||||||
"export": true,
|
|
||||||
"multi_shop": true,
|
|
||||||
"api_access": true,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
token, err := IssueLicenseToken(payload, priv)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
assert.LessOrEqual(t, len(token), columnLimit,
|
|
||||||
"license token length %d exceeds VARCHAR(%d) column limit", len(token), columnLimit)
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,40 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// codeAlphabet 兑换码字母表:剔除易混字符 0/O/1/I/L,避免人工抄录歧义。
|
||||||
|
const codeAlphabet = "23456789ABCDEFGHJKMNPQRSTUVWXYZ"
|
||||||
|
|
||||||
|
// codePrefix 兑换码固定前缀,便于一眼识别归属。
|
||||||
|
const codePrefix = "JIUKU"
|
||||||
|
|
||||||
|
// GenerateRedeemCode 生成一个随机兑换码,展示格式 JIUKU-XXXX-XXXX(8 位随机段)。
|
||||||
|
// 使用 crypto/rand + 无歧义字母表;归一化(NormalizeCode)后入库与比对。
|
||||||
|
func GenerateRedeemCode() string {
|
||||||
|
const n = 8
|
||||||
|
buf := make([]byte, n)
|
||||||
|
if _, err := rand.Read(buf); err != nil {
|
||||||
|
// crypto/rand 失败属系统级异常,调用方(CLI 生码)应直接失败而非产出弱码。
|
||||||
|
panic("crypto/rand failed: " + err.Error())
|
||||||
|
}
|
||||||
|
out := make([]byte, n)
|
||||||
|
for i, b := range buf {
|
||||||
|
out[i] = codeAlphabet[int(b)%len(codeAlphabet)]
|
||||||
|
}
|
||||||
|
return codePrefix + "-" + string(out[:4]) + "-" + string(out[4:])
|
||||||
|
}
|
||||||
|
|
||||||
|
// NormalizeCode 归一化兑换码:转大写、去掉连字符/空格,得到入库与查表用的规范形式。
|
||||||
|
func NormalizeCode(code string) string {
|
||||||
|
var b strings.Builder
|
||||||
|
for _, r := range strings.ToUpper(code) {
|
||||||
|
if r == '-' || r == ' ' || r == '\t' || r == '\n' || r == '\r' {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
b.WriteRune(r)
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
+1
-3
@@ -24,9 +24,6 @@ func main() {
|
|||||||
if config.C.Server.CORSOrigin == "*" {
|
if config.C.Server.CORSOrigin == "*" {
|
||||||
log.Fatal("server.cors_origin must not be '*' in production — set it to the actual frontend origin")
|
log.Fatal("server.cors_origin must not be '*' in production — set it to the actual frontend origin")
|
||||||
}
|
}
|
||||||
if config.C.License.Ed25519PrivateKey == "" {
|
|
||||||
log.Fatal("license.ed25519_private_key is required in production — store the key in Bitwarden and inject via env LICENSE_ED25519PRIVATEKEY")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 初始化数据库
|
// 初始化数据库
|
||||||
@@ -101,6 +98,7 @@ func autoMigrate(db *gorm.DB) {
|
|||||||
&model.User{},
|
&model.User{},
|
||||||
&model.License{},
|
&model.License{},
|
||||||
&model.LicenseDevice{},
|
&model.LicenseDevice{},
|
||||||
|
&model.LicenseCode{},
|
||||||
&model.UserSession{},
|
&model.UserSession{},
|
||||||
&model.LoginAttempt{},
|
&model.LoginAttempt{},
|
||||||
&model.ProductCategory{},
|
&model.ProductCategory{},
|
||||||
|
|||||||
@@ -107,6 +107,7 @@ CREATE TABLE IF NOT EXISTS `licenses` (
|
|||||||
`license_key` VARCHAR(768) NOT NULL COMMENT 'Ed25519 signed token',
|
`license_key` VARCHAR(768) NOT NULL COMMENT 'Ed25519 signed token',
|
||||||
`device_id` VARCHAR(255) DEFAULT NULL COMMENT 'deprecated: use license_devices',
|
`device_id` VARCHAR(255) DEFAULT NULL COMMENT 'deprecated: use license_devices',
|
||||||
`type` ENUM('trial','monthly','annual','lifetime') NOT NULL DEFAULT 'trial',
|
`type` ENUM('trial','monthly','annual','lifetime') NOT NULL DEFAULT 'trial',
|
||||||
|
`tier` VARCHAR(32) NOT NULL DEFAULT 'standard' COMMENT '权益档位钩子(pro/max-like),当前默认 standard',
|
||||||
`expires_at` DATETIME DEFAULT NULL COMMENT 'NULL=永久',
|
`expires_at` DATETIME DEFAULT NULL COMMENT 'NULL=永久',
|
||||||
`is_active` TINYINT(1) NOT NULL DEFAULT 1,
|
`is_active` TINYINT(1) NOT NULL DEFAULT 1,
|
||||||
`max_devices` INT NOT NULL DEFAULT 3 COMMENT '最大绑定设备数',
|
`max_devices` INT NOT NULL DEFAULT 3 COMMENT '最大绑定设备数',
|
||||||
@@ -134,6 +135,28 @@ CREATE TABLE IF NOT EXISTS `license_devices` (
|
|||||||
KEY `idx_license_id` (`license_id`)
|
KEY `idx_license_id` (`license_id`)
|
||||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='许可证设备绑定';
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='许可证设备绑定';
|
||||||
|
|
||||||
|
-- 兑换券(激活码)码池:平台方批量生成,用户兑换后叠加时长到门店授权
|
||||||
|
CREATE TABLE IF NOT EXISTS `license_codes` (
|
||||||
|
`id` BIGINT UNSIGNED NOT NULL AUTO_INCREMENT,
|
||||||
|
`code` VARCHAR(32) NOT NULL COMMENT '归一化大写无连字符',
|
||||||
|
`type` ENUM('trial','monthly','annual','lifetime') NOT NULL DEFAULT 'annual' COMMENT '计费/时长标签',
|
||||||
|
`tier` VARCHAR(32) NOT NULL DEFAULT 'standard' COMMENT '档位钩子(pro/max-like),当前默认 standard',
|
||||||
|
`duration_days` INT NOT NULL DEFAULT 0 COMMENT '授予时长;0=永久(lifetime)',
|
||||||
|
`max_devices` INT NOT NULL DEFAULT 0 COMMENT '授予设备上限;0=不改变现值',
|
||||||
|
`status` ENUM('unused','redeemed','void') NOT NULL DEFAULT 'unused',
|
||||||
|
`redeemed_shop_id` BIGINT UNSIGNED DEFAULT NULL COMMENT '被哪个门店兑换',
|
||||||
|
`redeemed_at` DATETIME DEFAULT NULL,
|
||||||
|
`redeemed_device_id` VARCHAR(255) DEFAULT NULL COMMENT '兑换设备(审计)',
|
||||||
|
`batch` VARCHAR(64) DEFAULT NULL COMMENT '发放批次/活动名',
|
||||||
|
`note` VARCHAR(255) DEFAULT NULL,
|
||||||
|
`created_at` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
`updated_at` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
|
||||||
|
PRIMARY KEY (`id`),
|
||||||
|
UNIQUE KEY `uk_code` (`code`),
|
||||||
|
KEY `idx_status` (`status`),
|
||||||
|
KEY `idx_redeemed_shop` (`redeemed_shop_id`)
|
||||||
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='兑换券码池';
|
||||||
|
|
||||||
-- ------------------------------------------------------------
|
-- ------------------------------------------------------------
|
||||||
-- 商品分类
|
-- 商品分类
|
||||||
-- ------------------------------------------------------------
|
-- ------------------------------------------------------------
|
||||||
|
|||||||
@@ -27,9 +27,6 @@ func InitConfig() {
|
|||||||
AccessExpireMin: 60,
|
AccessExpireMin: 60,
|
||||||
RefreshExpireH: 168,
|
RefreshExpireH: 168,
|
||||||
},
|
},
|
||||||
License: config.LicenseConfig{
|
|
||||||
HMACSecret: "test-license-hmac-secret",
|
|
||||||
},
|
|
||||||
Session: config.SessionConfig{
|
Session: config.SessionConfig{
|
||||||
LimitDesktop: 2,
|
LimitDesktop: 2,
|
||||||
LimitMobile: 2,
|
LimitMobile: 2,
|
||||||
@@ -125,12 +122,29 @@ func SetupTestDB() *gorm.DB {
|
|||||||
license_key TEXT UNIQUE,
|
license_key TEXT UNIQUE,
|
||||||
device_id TEXT,
|
device_id TEXT,
|
||||||
type TEXT DEFAULT 'trial',
|
type TEXT DEFAULT 'trial',
|
||||||
|
tier TEXT DEFAULT 'standard',
|
||||||
expires_at DATETIME,
|
expires_at DATETIME,
|
||||||
is_active INTEGER DEFAULT 1,
|
is_active INTEGER DEFAULT 1,
|
||||||
max_devices INTEGER DEFAULT 3,
|
max_devices INTEGER DEFAULT 3,
|
||||||
features TEXT,
|
features TEXT,
|
||||||
activated_at DATETIME
|
activated_at DATETIME
|
||||||
)`,
|
)`,
|
||||||
|
`CREATE TABLE IF NOT EXISTS license_codes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
code TEXT UNIQUE NOT NULL,
|
||||||
|
type TEXT DEFAULT 'annual',
|
||||||
|
tier TEXT DEFAULT 'standard',
|
||||||
|
duration_days INTEGER NOT NULL DEFAULT 0,
|
||||||
|
max_devices INTEGER NOT NULL DEFAULT 0,
|
||||||
|
status TEXT NOT NULL DEFAULT 'unused',
|
||||||
|
redeemed_shop_id INTEGER,
|
||||||
|
redeemed_at DATETIME,
|
||||||
|
redeemed_device_id TEXT,
|
||||||
|
batch TEXT,
|
||||||
|
note TEXT,
|
||||||
|
created_at DATETIME,
|
||||||
|
updated_at DATETIME
|
||||||
|
)`,
|
||||||
`CREATE TABLE IF NOT EXISTS license_devices (
|
`CREATE TABLE IF NOT EXISTS license_devices (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
license_id INTEGER NOT NULL,
|
license_id INTEGER NOT NULL,
|
||||||
|
|||||||
Reference in New Issue
Block a user