增加续杯激活
This commit is contained in:
parent
6b8651af25
commit
6788d49f47
750
controllers/platform_cursor_activation_code.go
Normal file
750
controllers/platform_cursor_activation_code.go
Normal file
@ -0,0 +1,750 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"encoding/csv"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
"server/pkg/jwtutil"
|
||||
|
||||
"github.com/beego/beego/v2/client/orm"
|
||||
beego "github.com/beego/beego/v2/server/web"
|
||||
)
|
||||
|
||||
// PlatformCursorActivationCodeController 平台端 Cursor 激活码管理
|
||||
type PlatformCursorActivationCodeController struct {
|
||||
beego.Controller
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) platformClaims() (*jwtutil.Claims, error) {
|
||||
auth := c.Ctx.Request.Header.Get("Authorization")
|
||||
if auth == "" {
|
||||
return nil, fmt.Errorf("未登录")
|
||||
}
|
||||
parts := strings.SplitN(auth, " ", 2)
|
||||
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||
return nil, fmt.Errorf("认证信息格式错误")
|
||||
}
|
||||
claims, err := jwtutil.ParseToken(parts[1])
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无效的token")
|
||||
}
|
||||
if claims.UserType != "platform" {
|
||||
return nil, fmt.Errorf("无权访问")
|
||||
}
|
||||
return claims, nil
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) jsonErr(httpStatus, bizCode int, msg string) {
|
||||
c.Ctx.Output.SetStatus(httpStatus)
|
||||
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) ok(data interface{}) {
|
||||
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": data}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
func cursorActivationCodeTrimPtr(value *string) *string {
|
||||
if value == nil {
|
||||
return nil
|
||||
}
|
||||
v := strings.TrimSpace(*value)
|
||||
if v == "" {
|
||||
return nil
|
||||
}
|
||||
return &v
|
||||
}
|
||||
|
||||
func cursorActivationCodeTimePtr(value *string) *time.Time {
|
||||
if value == nil {
|
||||
return nil
|
||||
}
|
||||
v := strings.TrimSpace(*value)
|
||||
if v == "" {
|
||||
return nil
|
||||
}
|
||||
layouts := []string{
|
||||
time.RFC3339,
|
||||
"2006-01-02 15:04:05",
|
||||
"2006-01-02 15:04",
|
||||
"2006-01-02",
|
||||
}
|
||||
for _, layout := range layouts {
|
||||
if t, err := time.ParseInLocation(layout, v, time.Local); err == nil {
|
||||
return &t
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func cursorActivationCodeStatusValid(status int8) bool {
|
||||
return status == 0 || status == 1 || status == 2 || status == 3
|
||||
}
|
||||
|
||||
func cursorActivationCodeTypeName(cardType int) string {
|
||||
switch cardType {
|
||||
case 1:
|
||||
return "天卡"
|
||||
case 7:
|
||||
return "周卡"
|
||||
case 30:
|
||||
return "月卡"
|
||||
case 90:
|
||||
return "季卡"
|
||||
case 365:
|
||||
return "年卡"
|
||||
case 0:
|
||||
return "自定义"
|
||||
default:
|
||||
return fmt.Sprintf("%d天", cardType)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) rowToMap(row *models.PlatformCursorActivationCode) map[string]interface{} {
|
||||
bindStatus := 0
|
||||
if row.BindAccount != nil || row.BindDeviceID != nil || row.MachineCode != nil {
|
||||
bindStatus = 1
|
||||
}
|
||||
|
||||
return map[string]interface{}{
|
||||
"id": row.ID,
|
||||
"code": row.Code,
|
||||
"type": row.Type,
|
||||
"typeName": cursorActivationCodeTypeName(row.Type),
|
||||
"status": row.Status,
|
||||
"durationDays": row.DurationDays,
|
||||
"bindAccount": row.BindAccount,
|
||||
"bindDeviceId": row.BindDeviceID,
|
||||
"bindStatus": bindStatus,
|
||||
"machineCode": row.MachineCode,
|
||||
"deviceInfo": row.DeviceInfo,
|
||||
"ownerUserId": row.OwnerUserID,
|
||||
"ownerUserName": row.OwnerUserName,
|
||||
"activatedAt": row.ActivatedAt,
|
||||
"expiredAt": row.ExpiredAt,
|
||||
"createdAt": row.CreateTime,
|
||||
"updatedAt": row.UpdateTime,
|
||||
"createTime": row.CreateTime,
|
||||
"updateTime": row.UpdateTime,
|
||||
"remark": row.Remark,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) filteredQuery() orm.QuerySeter {
|
||||
keyword := strings.TrimSpace(c.GetString("keyword"))
|
||||
statusText := strings.TrimSpace(c.GetString("status"))
|
||||
typeText := strings.TrimSpace(c.GetString("type"))
|
||||
bindStatusText := strings.TrimSpace(c.GetString("bindStatus"))
|
||||
|
||||
qs := models.Orm.QueryTable(new(models.PlatformCursorActivationCode)).Filter("delete_time__isnull", true)
|
||||
|
||||
if keyword != "" {
|
||||
cond := orm.NewCondition().
|
||||
Or("code__icontains", keyword).
|
||||
Or("bind_account__icontains", keyword).
|
||||
Or("machine_code__icontains", keyword).
|
||||
Or("device_info__icontains", keyword).
|
||||
Or("owner_user_name__icontains", keyword).
|
||||
Or("remark__icontains", keyword)
|
||||
qs = qs.SetCond(cond)
|
||||
qs = qs.Filter("delete_time__isnull", true)
|
||||
}
|
||||
|
||||
if statusText != "" {
|
||||
status, err := strconv.ParseInt(statusText, 10, 8)
|
||||
if err == nil && cursorActivationCodeStatusValid(int8(status)) {
|
||||
qs = qs.Filter("status", int8(status))
|
||||
}
|
||||
}
|
||||
|
||||
if typeText != "" {
|
||||
cardType, err := strconv.Atoi(typeText)
|
||||
if err == nil {
|
||||
qs = qs.Filter("type", cardType)
|
||||
}
|
||||
}
|
||||
|
||||
if bindStatusText != "" {
|
||||
bindStatus, err := strconv.Atoi(bindStatusText)
|
||||
if err == nil {
|
||||
if bindStatus == 0 {
|
||||
qs = qs.Filter("bind_account__isnull", true).Filter("bind_device_id__isnull", true).Filter("machine_code__isnull", true)
|
||||
} else if bindStatus == 1 {
|
||||
cond := orm.NewCondition().
|
||||
Or("bind_account__isnull", false).
|
||||
Or("bind_device_id__isnull", false).
|
||||
Or("machine_code__isnull", false)
|
||||
qs = qs.SetCond(cond)
|
||||
qs = qs.Filter("delete_time__isnull", true)
|
||||
if statusText != "" {
|
||||
status, err := strconv.ParseInt(statusText, 10, 8)
|
||||
if err == nil && cursorActivationCodeStatusValid(int8(status)) {
|
||||
qs = qs.Filter("status", int8(status))
|
||||
}
|
||||
}
|
||||
if typeText != "" {
|
||||
cardType, err := strconv.Atoi(typeText)
|
||||
if err == nil {
|
||||
qs = qs.Filter("type", cardType)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return qs
|
||||
}
|
||||
|
||||
// List GET /platform/cursor/activationcode/list
|
||||
func (c *PlatformCursorActivationCodeController) List() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
page, _ := c.GetInt("page", 1)
|
||||
pageSize, _ := c.GetInt("pageSize", 20)
|
||||
if page < 1 {
|
||||
page = 1
|
||||
}
|
||||
if pageSize < 1 {
|
||||
pageSize = 20
|
||||
}
|
||||
if pageSize > 200 {
|
||||
pageSize = 200
|
||||
}
|
||||
|
||||
qs := c.filteredQuery()
|
||||
total, _ := qs.Count()
|
||||
|
||||
var rows []models.PlatformCursorActivationCode
|
||||
_, err := qs.OrderBy("-id").Limit(pageSize, (page-1)*pageSize).All(&rows)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "获取激活码列表失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
list := make([]map[string]interface{}, 0, len(rows))
|
||||
for i := range rows {
|
||||
list = append(list, c.rowToMap(&rows[i]))
|
||||
}
|
||||
|
||||
c.ok(map[string]interface{}{
|
||||
"list": list,
|
||||
"total": total,
|
||||
"page": page,
|
||||
"pageSize": pageSize,
|
||||
})
|
||||
}
|
||||
|
||||
// Detail GET /platform/cursor/activationcode/detail/:id
|
||||
func (c *PlatformCursorActivationCodeController) Detail() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
|
||||
if err != nil || id == 0 {
|
||||
c.jsonErr(400, 400, "无效ID")
|
||||
return
|
||||
}
|
||||
|
||||
var row models.PlatformCursorActivationCode
|
||||
err = models.Orm.QueryTable(new(models.PlatformCursorActivationCode)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
One(&row)
|
||||
if err != nil {
|
||||
c.jsonErr(404, 404, "激活码不存在")
|
||||
return
|
||||
}
|
||||
|
||||
c.ok(c.rowToMap(&row))
|
||||
}
|
||||
|
||||
type platformCursorActivationCodePayload struct {
|
||||
ID *uint64 `json:"id"`
|
||||
Code *string `json:"code"`
|
||||
Type *int `json:"type"`
|
||||
Status *int8 `json:"status"`
|
||||
DurationDays *int `json:"durationDays"`
|
||||
BindAccount *string `json:"bindAccount"`
|
||||
BindDeviceID *uint64 `json:"bindDeviceId"`
|
||||
OwnerUserID *uint64 `json:"ownerUserId"`
|
||||
OwnerUserName *string `json:"ownerUserName"`
|
||||
ActivatedAt *string `json:"activatedAt"`
|
||||
ExpiredAt *string `json:"expiredAt"`
|
||||
Remark *string `json:"remark"`
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) readPayload() (*platformCursorActivationCodePayload, error) {
|
||||
body, _ := io.ReadAll(c.Ctx.Request.Body)
|
||||
var p platformCursorActivationCodePayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &p, nil
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) fillDeviceSnapshot(up map[string]interface{}, bindDeviceID *uint64) {
|
||||
if bindDeviceID == nil || *bindDeviceID == 0 {
|
||||
up["bind_device_id"] = nil
|
||||
up["machine_code"] = nil
|
||||
up["device_info"] = nil
|
||||
return
|
||||
}
|
||||
|
||||
var device models.PlatformCursorEquipment
|
||||
err := models.Orm.QueryTable(new(models.PlatformCursorEquipment)).
|
||||
Filter("id", *bindDeviceID).
|
||||
Filter("delete_time__isnull", true).
|
||||
One(&device)
|
||||
if err == nil {
|
||||
up["bind_device_id"] = *bindDeviceID
|
||||
up["machine_code"] = device.MachineCode
|
||||
up["device_info"] = device.DeviceInfo
|
||||
return
|
||||
}
|
||||
|
||||
up["bind_device_id"] = *bindDeviceID
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) payloadToUpdateMap(p *platformCursorActivationCodePayload, includeCode bool) (map[string]interface{}, error) {
|
||||
up := map[string]interface{}{}
|
||||
|
||||
if includeCode {
|
||||
if p.Code == nil || strings.TrimSpace(*p.Code) == "" {
|
||||
return nil, fmt.Errorf("激活码不能为空")
|
||||
}
|
||||
up["code"] = strings.TrimSpace(*p.Code)
|
||||
} else if p.Code != nil {
|
||||
if strings.TrimSpace(*p.Code) == "" {
|
||||
return nil, fmt.Errorf("激活码不能为空")
|
||||
}
|
||||
up["code"] = strings.TrimSpace(*p.Code)
|
||||
}
|
||||
|
||||
if p.Type != nil {
|
||||
if *p.Type < 0 {
|
||||
return nil, fmt.Errorf("卡密类型不合法")
|
||||
}
|
||||
up["type"] = *p.Type
|
||||
}
|
||||
if p.Status != nil {
|
||||
if !cursorActivationCodeStatusValid(*p.Status) {
|
||||
return nil, fmt.Errorf("状态不合法，支持：0 未使用、1 已使用、2 已过期、3 已禁用")
|
||||
}
|
||||
up["status"] = *p.Status
|
||||
}
|
||||
if p.DurationDays != nil {
|
||||
if *p.DurationDays < 0 || *p.DurationDays > 9999 {
|
||||
return nil, fmt.Errorf("有效天数范围为 0-9999")
|
||||
}
|
||||
up["duration_days"] = *p.DurationDays
|
||||
}
|
||||
if p.BindAccount != nil {
|
||||
up["bind_account"] = cursorActivationCodeTrimPtr(p.BindAccount)
|
||||
}
|
||||
if p.BindDeviceID != nil {
|
||||
c.fillDeviceSnapshot(up, p.BindDeviceID)
|
||||
}
|
||||
if p.OwnerUserID != nil {
|
||||
if *p.OwnerUserID == 0 {
|
||||
up["owner_user_id"] = nil
|
||||
} else {
|
||||
up["owner_user_id"] = *p.OwnerUserID
|
||||
}
|
||||
}
|
||||
if p.OwnerUserName != nil {
|
||||
up["owner_user_name"] = cursorActivationCodeTrimPtr(p.OwnerUserName)
|
||||
}
|
||||
if p.ActivatedAt != nil {
|
||||
up["activated_at"] = cursorActivationCodeTimePtr(p.ActivatedAt)
|
||||
}
|
||||
if p.ExpiredAt != nil {
|
||||
up["expired_at"] = cursorActivationCodeTimePtr(p.ExpiredAt)
|
||||
}
|
||||
if p.Remark != nil {
|
||||
up["remark"] = cursorActivationCodeTrimPtr(p.Remark)
|
||||
}
|
||||
|
||||
return up, nil
|
||||
}
|
||||
|
||||
// Add POST /platform/cursor/activationcode/add
|
||||
func (c *PlatformCursorActivationCodeController) Add() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
p, err := c.readPayload()
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
up, err := c.payloadToUpdateMap(p, true)
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
row := models.PlatformCursorActivationCode{
|
||||
Code: up["code"].(string),
|
||||
Type: 30,
|
||||
Status: 0,
|
||||
DurationDays: 30,
|
||||
BindAccount: cursorActivationCodeTrimPtr(p.BindAccount),
|
||||
BindDeviceID: p.BindDeviceID,
|
||||
OwnerUserID: p.OwnerUserID,
|
||||
OwnerUserName: cursorActivationCodeTrimPtr(p.OwnerUserName),
|
||||
ActivatedAt: cursorActivationCodeTimePtr(p.ActivatedAt),
|
||||
ExpiredAt: cursorActivationCodeTimePtr(p.ExpiredAt),
|
||||
Remark: cursorActivationCodeTrimPtr(p.Remark),
|
||||
CreateTime: time.Now(),
|
||||
}
|
||||
|
||||
if p.Type != nil {
|
||||
row.Type = *p.Type
|
||||
}
|
||||
if p.Status != nil {
|
||||
row.Status = *p.Status
|
||||
}
|
||||
if p.DurationDays != nil {
|
||||
row.DurationDays = *p.DurationDays
|
||||
}
|
||||
if row.BindDeviceID != nil && *row.BindDeviceID == 0 {
|
||||
row.BindDeviceID = nil
|
||||
}
|
||||
if row.OwnerUserID != nil && *row.OwnerUserID == 0 {
|
||||
row.OwnerUserID = nil
|
||||
}
|
||||
if row.BindDeviceID != nil {
|
||||
var device models.PlatformCursorEquipment
|
||||
if err := models.Orm.QueryTable(new(models.PlatformCursorEquipment)).
|
||||
Filter("id", *row.BindDeviceID).
|
||||
Filter("delete_time__isnull", true).
|
||||
One(&device); err == nil {
|
||||
row.MachineCode = &device.MachineCode
|
||||
row.DeviceInfo = device.DeviceInfo
|
||||
}
|
||||
}
|
||||
|
||||
id, err := models.Orm.Insert(&row)
|
||||
if err != nil {
|
||||
if strings.Contains(strings.ToLower(err.Error()), "duplicate") {
|
||||
c.jsonErr(400, 400, "激活码已存在")
|
||||
return
|
||||
}
|
||||
c.jsonErr(500, 500, "新增激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.ok(map[string]interface{}{"id": id})
|
||||
}
|
||||
|
||||
// Update POST /platform/cursor/activationcode/update
|
||||
func (c *PlatformCursorActivationCodeController) Update() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
p, err := c.readPayload()
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
if p.ID == nil || *p.ID == 0 {
|
||||
c.jsonErr(400, 400, "无效ID")
|
||||
return
|
||||
}
|
||||
|
||||
up, err := c.payloadToUpdateMap(p, false)
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
if len(up) == 0 {
|
||||
c.jsonErr(400, 400, "无更新字段")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
up["update_time"] = now
|
||||
|
||||
n, err := models.Orm.QueryTable(new(models.PlatformCursorActivationCode)).
|
||||
Filter("id", *p.ID).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(up)
|
||||
if err != nil {
|
||||
if strings.Contains(strings.ToLower(err.Error()), "duplicate") {
|
||||
c.jsonErr(400, 400, "激活码已存在")
|
||||
return
|
||||
}
|
||||
c.jsonErr(500, 500, "更新激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "激活码不存在")
|
||||
return
|
||||
}
|
||||
|
||||
c.ok(nil)
|
||||
}
|
||||
|
||||
// Delete POST /platform/cursor/activationcode/delete/:id
|
||||
func (c *PlatformCursorActivationCodeController) Delete() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
|
||||
if err != nil || id == 0 {
|
||||
c.jsonErr(400, 400, "无效ID")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
n, err := models.Orm.QueryTable(new(models.PlatformCursorActivationCode)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(map[string]interface{}{"delete_time": now, "update_time": now})
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "删除激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "激活码不存在")
|
||||
return
|
||||
}
|
||||
|
||||
c.ok(nil)
|
||||
}
|
||||
|
||||
type platformCursorActivationCodeGeneratePayload struct {
|
||||
Count int `json:"count"`
|
||||
Type int `json:"type"`
|
||||
DurationDays int `json:"durationDays"`
|
||||
OwnerUserID *uint64 `json:"ownerUserId"`
|
||||
OwnerUserName *string `json:"ownerUserName"`
|
||||
Remark *string `json:"remark"`
|
||||
}
|
||||
|
||||
func randomCursorActivationCode() (string, error) {
|
||||
b := make([]byte, 12)
|
||||
if _, err := rand.Read(b); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return "CUR-" + strings.ToUpper(hex.EncodeToString(b)), nil
|
||||
}
|
||||
|
||||
// Generate POST /platform/cursor/activationcode/generate
|
||||
func (c *PlatformCursorActivationCodeController) Generate() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
body, _ := io.ReadAll(c.Ctx.Request.Body)
|
||||
var p platformCursorActivationCodeGeneratePayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
if p.Count < 1 {
|
||||
p.Count = 1
|
||||
}
|
||||
if p.Count > 10000 {
|
||||
c.jsonErr(400, 400, "单次最多生成 10000 个激活码")
|
||||
return
|
||||
}
|
||||
if p.Type < 0 {
|
||||
c.jsonErr(400, 400, "卡密类型不合法")
|
||||
return
|
||||
}
|
||||
if p.DurationDays < 0 || p.DurationDays > 9999 {
|
||||
c.jsonErr(400, 400, "有效天数范围为 0-9999")
|
||||
return
|
||||
}
|
||||
if p.Type == 0 && p.DurationDays == 0 {
|
||||
p.DurationDays = 30
|
||||
}
|
||||
if p.Type > 0 && p.DurationDays == 0 {
|
||||
p.DurationDays = p.Type
|
||||
}
|
||||
|
||||
createdIDs := make([]int64, 0, p.Count)
|
||||
codes := make([]string, 0, p.Count)
|
||||
now := time.Now()
|
||||
|
||||
for len(createdIDs) < p.Count {
|
||||
code, err := randomCursorActivationCode()
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "生成激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
row := models.PlatformCursorActivationCode{
|
||||
Code: code,
|
||||
Type: p.Type,
|
||||
Status: 0,
|
||||
DurationDays: p.DurationDays,
|
||||
OwnerUserID: p.OwnerUserID,
|
||||
OwnerUserName: cursorActivationCodeTrimPtr(p.OwnerUserName),
|
||||
Remark: cursorActivationCodeTrimPtr(p.Remark),
|
||||
CreateTime: now,
|
||||
}
|
||||
if row.OwnerUserID != nil && *row.OwnerUserID == 0 {
|
||||
row.OwnerUserID = nil
|
||||
}
|
||||
|
||||
id, err := models.Orm.Insert(&row)
|
||||
if err != nil {
|
||||
if strings.Contains(strings.ToLower(err.Error()), "duplicate") {
|
||||
continue
|
||||
}
|
||||
c.jsonErr(500, 500, "生成激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
createdIDs = append(createdIDs, id)
|
||||
codes = append(codes, code)
|
||||
}
|
||||
|
||||
c.ok(map[string]interface{}{
|
||||
"count": len(createdIDs),
|
||||
"ids": createdIDs,
|
||||
"codes": codes,
|
||||
})
|
||||
}
|
||||
|
||||
// Enable POST /platform/cursor/activationcode/enable/:id
|
||||
func (c *PlatformCursorActivationCodeController) Enable() {
|
||||
c.changeStatus(0, "启用激活码失败")
|
||||
}
|
||||
|
||||
// Disable POST /platform/cursor/activationcode/disable/:id
|
||||
func (c *PlatformCursorActivationCodeController) Disable() {
|
||||
c.changeStatus(3, "禁用激活码失败")
|
||||
}
|
||||
|
||||
func (c *PlatformCursorActivationCodeController) changeStatus(status int8, failMsg string) {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
|
||||
if err != nil || id == 0 {
|
||||
c.jsonErr(400, 400, "无效ID")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
n, err := models.Orm.QueryTable(new(models.PlatformCursorActivationCode)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(map[string]interface{}{
|
||||
"status": status,
|
||||
"update_time": now,
|
||||
})
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, failMsg+": "+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "激活码不存在")
|
||||
return
|
||||
}
|
||||
|
||||
c.ok(nil)
|
||||
}
|
||||
|
||||
// Export GET /platform/cursor/activationcode/export
|
||||
func (c *PlatformCursorActivationCodeController) Export() {
|
||||
if _, err := c.platformClaims(); err != nil {
|
||||
c.jsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
var rows []models.PlatformCursorActivationCode
|
||||
_, err := c.filteredQuery().OrderBy("-id").Limit(50000).All(&rows)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "导出激活码失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
filename := fmt.Sprintf("cursor-activation-code-%s.csv", time.Now().Format("20060102150405"))
|
||||
c.Ctx.Output.Header("Content-Type", "text/csv; charset=utf-8")
|
||||
c.Ctx.Output.Header("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filename))
|
||||
|
||||
_, _ = c.Ctx.ResponseWriter.Write([]byte{0xEF, 0xBB, 0xBF})
|
||||
writer := csv.NewWriter(c.Ctx.ResponseWriter)
|
||||
_ = writer.Write([]string{
|
||||
"ID", "激活码", "类型", "有效天数", "状态", "绑定账号", "绑定设备ID", "机器码", "归属用户ID", "归属用户", "激活时间", "过期时间", "创建时间", "备注",
|
||||
})
|
||||
|
||||
statusText := map[int8]string{
|
||||
0: "未使用",
|
||||
1: "已使用",
|
||||
2: "已过期",
|
||||
3: "已禁用",
|
||||
}
|
||||
|
||||
for i := range rows {
|
||||
row := rows[i]
|
||||
_ = writer.Write([]string{
|
||||
strconv.FormatUint(row.ID, 10),
|
||||
row.Code,
|
||||
cursorActivationCodeTypeName(row.Type),
|
||||
strconv.Itoa(row.DurationDays),
|
||||
statusText[row.Status],
|
||||
stringPtrValue(row.BindAccount),
|
||||
uint64PtrValue(row.BindDeviceID),
|
||||
stringPtrValue(row.MachineCode),
|
||||
uint64PtrValue(row.OwnerUserID),
|
||||
stringPtrValue(row.OwnerUserName),
|
||||
timePtrValue(row.ActivatedAt),
|
||||
timePtrValue(row.ExpiredAt),
|
||||
row.CreateTime.Format("2006-01-02 15:04:05"),
|
||||
stringPtrValue(row.Remark),
|
||||
})
|
||||
}
|
||||
|
||||
writer.Flush()
|
||||
}
|
||||
|
||||
func stringPtrValue(value *string) string {
|
||||
if value == nil {
|
||||
return ""
|
||||
}
|
||||
return *value
|
||||
}
|
||||
|
||||
func uint64PtrValue(value *uint64) string {
|
||||
if value == nil {
|
||||
return ""
|
||||
}
|
||||
return strconv.FormatUint(*value, 10)
|
||||
}
|
||||
|
||||
func timePtrValue(value *time.Time) string {
|
||||
if value == nil {
|
||||
return ""
|
||||
}
|
||||
return value.Format("2006-01-02 15:04:05")
|
||||
}
|
||||
31
docs/sql/yz_platform_cursor_activation_code.sql
Normal file
31
docs/sql/yz_platform_cursor_activation_code.sql
Normal file
@ -0,0 +1,31 @@
|
||||
-- Cursor 激活码管理
|
||||
-- status: 0 未使用 1 已使用 2 已过期 3 已禁用
|
||||
-- type: 0 自定义 1 天卡 7 周卡 30 月卡 90 季卡 365 年卡
|
||||
|
||||
CREATE TABLE IF NOT EXISTS `yz_platform_cursor_activation_code` (
|
||||
`id` bigint unsigned NOT NULL AUTO_INCREMENT COMMENT '主键ID',
|
||||
`code` varchar(128) NOT NULL COMMENT '激活码',
|
||||
`type` int NOT NULL DEFAULT 30 COMMENT '卡密类型：0自定义 1天卡 7周卡 30月卡 90季卡 365年卡',
|
||||
`status` tinyint NOT NULL DEFAULT 0 COMMENT '状态：0未使用 1已使用 2已过期 3已禁用',
|
||||
`duration_days` int NOT NULL DEFAULT 30 COMMENT '有效天数',
|
||||
`bind_account` varchar(128) DEFAULT NULL COMMENT '绑定账号',
|
||||
`bind_device_id` bigint unsigned DEFAULT NULL COMMENT '绑定设备ID，关联 yz_platform_cursor_equipment.id',
|
||||
`machine_code` varchar(128) DEFAULT NULL COMMENT '绑定设备机器码',
|
||||
`device_info` varchar(1000) DEFAULT NULL COMMENT '绑定设备信息',
|
||||
`owner_user_id` bigint unsigned DEFAULT NULL COMMENT '归属用户ID',
|
||||
`owner_user_name` varchar(128) DEFAULT NULL COMMENT '归属用户名称',
|
||||
`activated_at` datetime DEFAULT NULL COMMENT '激活时间',
|
||||
`expired_at` datetime DEFAULT NULL COMMENT '过期时间',
|
||||
`remark` varchar(1000) DEFAULT NULL COMMENT '备注',
|
||||
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
|
||||
`update_time` datetime DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
|
||||
`delete_time` datetime DEFAULT NULL COMMENT '删除时间',
|
||||
PRIMARY KEY (`id`),
|
||||
UNIQUE KEY `uk_code` (`code`),
|
||||
KEY `idx_status_delete` (`status`,`delete_time`),
|
||||
KEY `idx_type_status` (`type`,`status`),
|
||||
KEY `idx_bind_account` (`bind_account`),
|
||||
KEY `idx_bind_device_id` (`bind_device_id`),
|
||||
KEY `idx_owner_user_id` (`owner_user_id`),
|
||||
KEY `idx_expired_at` (`expired_at`)
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='Cursor续杯激活码';
|
||||
@ -57,6 +57,7 @@ func Init(_ string) {
|
||||
new(PlatformComplaint),
|
||||
new(SystemSoftwareUpgrade),
|
||||
new(PlatformCursorEquipment),
|
||||
new(PlatformCursorActivationCode),
|
||||
new(PlatformAccountPoolKiro),
|
||||
new(PlatformAccountPoolWindsurf),
|
||||
new(PlatformAccountPoolCursor),
|
||||
|
||||
28
models/platform_cursor_activation_code.go
Normal file
28
models/platform_cursor_activation_code.go
Normal file
@ -0,0 +1,28 @@
|
||||
package models
|
||||
|
||||
import "time"
|
||||
|
||||
// PlatformCursorActivationCode Cursor 续杯激活码 yz_platform_cursor_activation_code
|
||||
type PlatformCursorActivationCode struct {
|
||||
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
||||
Code string `orm:"column(code);size(128);unique" json:"code"`
|
||||
Type int `orm:"column(type);default(30)" json:"type"`
|
||||
Status int8 `orm:"column(status);default(0)" json:"status"`
|
||||
DurationDays int `orm:"column(duration_days);default(30)" json:"durationDays"`
|
||||
BindAccount *string `orm:"column(bind_account);size(128);null" json:"bindAccount"`
|
||||
BindDeviceID *uint64 `orm:"column(bind_device_id);null" json:"bindDeviceId"`
|
||||
MachineCode *string `orm:"column(machine_code);size(128);null" json:"machineCode"`
|
||||
DeviceInfo *string `orm:"column(device_info);size(1000);null" json:"deviceInfo"`
|
||||
OwnerUserID *uint64 `orm:"column(owner_user_id);null" json:"ownerUserId"`
|
||||
OwnerUserName *string `orm:"column(owner_user_name);size(128);null" json:"ownerUserName"`
|
||||
ActivatedAt *time.Time `orm:"column(activated_at);type(datetime);null" json:"activatedAt"`
|
||||
ExpiredAt *time.Time `orm:"column(expired_at);type(datetime);null" json:"expiredAt"`
|
||||
Remark *string `orm:"column(remark);size(1000);null" json:"remark"`
|
||||
CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"createTime"`
|
||||
UpdateTime *time.Time `orm:"column(update_time);auto_now;type(datetime);null" json:"updateTime"`
|
||||
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"deleteTime"`
|
||||
}
|
||||
|
||||
func (m *PlatformCursorActivationCode) TableName() string {
|
||||
return "yz_platform_cursor_activation_code"
|
||||
}
|
||||
@ -172,6 +172,17 @@ func Register() {
|
||||
beego.Router("/platform/cursor/equipment/activationRecords", &controllers.PlatformCursorEquipmentController{}, "get:ActivationRecords")
|
||||
beego.Router("/platform/cursor/equipment/extractRecords", &controllers.PlatformCursorEquipmentController{}, "get:ExtractRecords")
|
||||
|
||||
// Cursor 激活码管理（yz_platform_cursor_activation_code）
|
||||
beego.Router("/platform/cursor/activationcode/list", &controllers.PlatformCursorActivationCodeController{}, "get:List")
|
||||
beego.Router("/platform/cursor/activationcode/detail/:id", &controllers.PlatformCursorActivationCodeController{}, "get:Detail")
|
||||
beego.Router("/platform/cursor/activationcode/add", &controllers.PlatformCursorActivationCodeController{}, "post:Add")
|
||||
beego.Router("/platform/cursor/activationcode/update", &controllers.PlatformCursorActivationCodeController{}, "post:Update")
|
||||
beego.Router("/platform/cursor/activationcode/delete/:id", &controllers.PlatformCursorActivationCodeController{}, "post:Delete")
|
||||
beego.Router("/platform/cursor/activationcode/generate", &controllers.PlatformCursorActivationCodeController{}, "post:Generate")
|
||||
beego.Router("/platform/cursor/activationcode/enable/:id", &controllers.PlatformCursorActivationCodeController{}, "post:Enable")
|
||||
beego.Router("/platform/cursor/activationcode/disable/:id", &controllers.PlatformCursorActivationCodeController{}, "post:Disable")
|
||||
beego.Router("/platform/cursor/activationcode/export", &controllers.PlatformCursorActivationCodeController{}, "get:Export")
|
||||
|
||||
// 账号池管理（cursor/windsurf/krio）
|
||||
beego.Router("/platform/accountPool/cursor/list", &controllers.PlatformAccountPoolCursorController{}, "get:List")
|
||||
beego.Router("/platform/accountPool/cursor/add", &controllers.PlatformAccountPoolCursorController{}, "post:Add")
|
||||
|
||||
Loading…
Reference in New Issue
Block a user