Compare commits

...

14 Commits

Author SHA1 Message Date
wangjia 7bbc944ae2 chore: release server-v1.0.62
Deploy Server / release-deploy-server (push) Successful in 51s
服务端安全加固:多维限流(按 IP/按门店)+ 敏感接口独立速率上限抵御 DDoS/刷接口;
登录暴力破解新增按来源 IP 锁定;反代后正确识别真实客户端 IP;
门店 custom_fields 轻量配置(录入默认值)透传保存。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 20:20:44 +08:00
wangjia 0d967e899a chore: release client-v1.0.59
Deploy Client / build-client-web (push) Successful in 41s
Deploy Client / build-windows (push) Successful in 2m36s
Deploy Client / build-macos (push) Successful in 2m17s
Deploy Client / build-android (push) Successful in 1m36s
Deploy Client / build-ios (push) Successful in 2m46s
Deploy Client / release-deploy-client (push) Successful in 1m24s
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 14:07:16 +08:00
wangjia 0a21ea5b2a chore: release server-v1.0.61
Deploy Server / release-deploy-server (push) Successful in 1m8s
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 14:06:03 +08:00
wangjia 9b6dba4c2b test(backend): 补 redeem_code 单测 + gitignore todo/ 与 gencode 产物
覆盖 NormalizeCode 边界/幂等、GenerateRedeemCode 格式/无歧义字母表
(剔除 0/O/1/I/L)/无碰撞。.gitignore 忽略全局 todo skill 本地数据
与 backend/gencode 构建产物。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 14:04:22 +08:00
wangjia 54451f0e52 feat(deploy): gencode 纳入 server 发版产物 + EC2 生码封装
server 流水线编译并发布 jiu-gencode 二进制(回滚路径随 release 资产
自动下载),部署到 /opt/jiu/backend/,对旧版回滚做存在性判断优雅跳过。
新增 deploy/jiu-gencode.sh 封装自动读 production.env,线上生码从此
一条 ssh 即可,无需本地交叉编译 + scp。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 14:04:14 +08:00
wangjia 8fc24b5d36 chore(deploy): 移除退役的 ed25519/HMAC license 死配置
兑换券改造后运行时已不读签名相关 env,清理 production.env.template
的 LICENSE_HMAC_SECRET / LICENSE_ED25519_* 三项,并从 render-env.sh
的 SECRET_KEYS 去掉对应密钥(渲染 env 时不再白拉 Bitwarden)。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 14:04:06 +08:00
wangjia 6ac6b52717 chore: 移除两个废弃 skill(archive-design / design-review)
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y2Wdwo7SmgBJU37cBrkhPK
2026-06-19 12:52:30 +08:00
wangjia c72e4171fc chore: release server-v1.0.60
Deploy Server / release-deploy-server (push) Successful in 42s
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 12:15:28 +08:00
wangjia f955391830 docs: license-design 增补 v2 时长兑换券模型说明
顶部新增 v2 节,标注 v1 的 Ed25519 token 方案已废弃,记录兑换券
模型/数据表/Redeem 流程/与 v1 差异;运行时降级 phase 两版共用。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 12:14:35 +08:00
wangjia f4e5f935e2 feat(client): 授权 Tab 文案改为兑换券语义
激活卡片标题/说明/输入提示/按钮/成功提示对齐时长券:
"兑换激活码"、hint JIUKU-XXXX-XXXX、"时长叠加到当前授权之后"。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 12:14:35 +08:00
wangjia 23dff69c62 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>
2026-06-19 12:14:27 +08:00
wangjia 914e2fb533 chore(scripts): 新增授权到期时间测试工具 set-license-expiry.sh
按相对天数设置门店授权 expires_at,快速切换 normal/grace/readonly/locked
四个阶段验证客户端表现。SSH 到线上 jiu_mysql 执行,DB 密码仅在远端从
production.env 读取,不打印/外传。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 09:58:31 +08:00
wangjia 58266a2095 feat(client): 登录页品牌化 + 授权过期续费入口 + 全平台应用图标统一
- 登录页文案与 Logo 改为「岩美酒库」品牌:标题/副标题/页脚 + 内嵌品牌 mark(替换原酒杯渐变方块)
- 授权过期锁定时,登录页给出续费入口:前往官网续费 + 复制客服邮箱
- 应用图标全平台统一为岩美品牌 mark:macOS(16-1024) / Web(favicon+icons+maskable)
  / Android mipmap(48-192) / iOS AppIcon(20-1024,满底无 alpha 过审核)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 09:58:25 +08:00
wangjia ca093c3c22 chore: 移除项目级 Figma MCP 与自带 /todo 命令,TODO 统一走全局 skill
- 删除 .mcp.json(项目级 Figma MCP 注册,改用全局)
- 删除 .claude/commands/todo.md(项目自带 /todo slash command)
- CLAUDE.md「项目 TODO 管理」改写:统一使用全局 todo 工具
  (~/.claude/skills/todo/),数据落到本项目 todo/ 目录

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-19 09:57:17 +08:00
82 changed files with 1621 additions and 1198 deletions
-158
View File
@@ -1,158 +0,0 @@
---
description: 项目 TODO 管理(list / add / status / done / reject / reopen / rm / sub
argument-hint: "list | add <描述> | status <id> <open|doing|done> | done <id> [version] | reject <id> <原因> | reopen <id> | rm <id> | sub add <parent_id> ... | sub status <sid> ..."
model: claude-sonnet-4-6
allowed-tools: Bash, Read
---
管理项目 `todo/todo.json` 待办清单,所有写操作会自动重渲染 `todo/todo.html`
**重要规则**:所有待办项必须通过本命令管理,禁止使用其它 todo 工具(TaskCreate/TodoWrite 等)。
**状态流转**`open`(待开始)→ `doing`(开发中)→ `done`(待验收)→ `accepted`(已验收)
- `done` 状态可被用户 `reject`(拒绝),退回 `open` 并升为最高优先级
- Claude 可设:`open` / `doing` / `done`;用户专属:`accepted``/todo done`)、`reject`
**子任务(subtask**:复杂任务(tier-1/2)可拆出子任务,SID 格式为 `{parentId}{Letter}`(如 `21A``21B`)。
- 父任务在子任务全部 `done` 前**不得**手动设为 `done`;全部完成时**自动**转为 `done`
- 子任务状态与父任务独立,自身有 `open` / `doing` / `done` 三态
- 子任务可声明依赖(`deps`),HTML 中以绿/红徽章展示依赖完成情况
---
## 改动等级 (tier) 与开发流程
每个 todo 有两个独立维度:**重要度 `level`**(优先级高低)和 **改动等级 `tier`**(复杂度/工作量)。
| tier | 名称 | 判定标准 | 开发流程 |
|------|------|---------|---------|
| 1 | 一级 | 涉及接口设计 / 数据库 schema 变更,**或**改动跨越 **2 个以上模块** | **必须先进 plan 模式**,用户批准方案后才能编码;完成后同步更新 `docs/architecture/``docs/api/` 等设计文档 |
| 2 | 二级 | 模块内较大改动,涉及 **5 个文件以上** | 可直接开发,完成后本地验证(build/test/analyze |
| 3 | 三级 | 小改动、局部优化、bugfix | 可直接开发,完成后本地验证 |
**判定关键词指引:**
- 一级:「新增接口」「改 schema」「新表」「跨模块」「涉及后端+前端」「新增 API」
- 二级:「重构」「整页改版」「涉及多个 screen」「较大」
- 三级:「修复」「bugfix」「优化」「调整文案」「小改」「单文件」
**重要**:开发完成后只做本地验证(build / test / analyze),**不要**自动触发 `/release`。发版由用户决定。
---
根据 `$ARGUMENTS` 分派,缺省等同 `list`
## list(或无参数)
运行以下命令并将终端 summary 转述给用户,同时提示 HTML 路径便于浏览:
```bash
node todo/todo.mjs list
```
## add <描述>
`$ARGUMENTS` 去掉首词 `add` 后的内容解析为待办描述,推断:
- `--title`:核心一句话标题(简洁,去掉平台/类别信息)
- `--level``high`(阻断/紧急)/ `mid`(重要,默认)/ `low`(一般/优化)
- 关键词「紧急」「阻断」「必须」「严重」→ high;「重要」「需要」→ mid;其余 → low
- `--tier`:**必须推断**,见上方「改动等级」判定标准,取值 `1`/`2`/`3`
- `--tags`:从描述推断平台/类别,可多个逗号分隔,常用值:`前端,后端,Web,mac,Windows,Android,iOS,数据库,CI/CD,文档`
- `--desc`:补充说明(可选,如有具体文件/路径/上下文则填入)
然后运行:
```bash
node todo/todo.mjs add --title "..." --level mid --tier 2 --tags "前端,Web" --desc "..."
```
转述:「已添加 #id [级别·等级] 标题 标签」,并显示 summary。
## status <id> <open|doing|done>
**Claude 在开发过程中主动调用**,更新条目的开发状态:
- 开始处理某个 todo → `status <id> doing`(**先确认 tier**:一级须先 plan 模式,二/三级直接开发)
- 开发完成提交验收 → `status <id> done`
- 退回重做 → `status <id> open`
```bash
node todo/todo.mjs status <id> <open|doing|done>
```
转述:「#id 状态已更新为 xxx」,并显示 summary。
注意:`accepted` 不可通过此命令设置,仅用户可标记。
## done <id> [version]
**用户调用**,标记条目已验收交付,记录版本号。从 `$ARGUMENTS` 提取 id 和可选 version(格式 vX.Y.Z):
- 有 version`node todo/todo.mjs done <id> --version <version>`
- 无 version`node todo/todo.mjs done <id>`(脚本自动用 `git describe --tags --abbrev=0`
转述:「#id 已验收,记入版本 vX.Y.Z」,并显示 summary。
## reject <id> <原因>
**用户调用**,拒绝验收处于 `done` 状态的条目。拒绝后:
- 状态退回 `open`
- 优先级强制升为 `high`
- 拒绝原因记录在条目上,HTML 卡片中可见
`$ARGUMENTS` 提取 id 和原因文字:
```bash
node todo/todo.mjs reject <id> --reason "<原因>"
```
转述:「#id 已拒绝,优先级升为最高,原因:xxx」,并显示 summary。
## reopen <id>
重新开启条目(清空拒绝记录和验收信息),退回 `open` 状态:
```bash
node todo/todo.mjs reopen <id>
```
转述「#id 已重新开启」。
## rm <id>
```bash
node todo/todo.mjs rm <id>
```
转述「#id 已删除」。
---
## sub add <parent_id> --title "..." [--tier N] [--deps "21A,21B"]
**Claude 在开发过程中主动调用**,为复杂任务(tier-1/2)添加子任务:
- `parent_id`:父任务数字 id(如 `21`
- `--title`:子任务标题
- `--tier`:改动等级(`1`/`2`/`3`,默认 `2`
- `--deps`:依赖的其他子任务 SID,逗号分隔(如 `"21A,21B"`);被依赖项必须已存在
SID 自动分配:第一个子任务为 `{parent_id}A`,依此类推(21A, 21B, 21C…)。
父任务若为 `open` 状态,添加第一个子任务时自动升为 `doing`
```bash
node todo/todo.mjs sub add <parent_id> --title "..." [--tier N] [--deps "21A,21B"]
```
转述:「已添加子任务 {SID} [等级]「{title}」→ #parent 依赖: ...」,并显示 summary。
## sub status <sid> <open|doing|done>
**Claude 在开发过程中主动调用**,更新子任务状态:
- SID 格式:`21A``21B`(不区分大小写)
- 当**最后一个子任务**标记为 `done` 时,父任务**自动**转为 `done`
```bash
node todo/todo.mjs sub status <sid> <open|doing|done>
```
转述:「{SID}「{title}」→ {状态}」,若触发父任务自动转换则一并提示。
---
所有命令完成后,在终端显示来自脚本的完整 summary(按 open / doing / done / accepted 分组)。
-152
View File
@@ -1,152 +0,0 @@
---
name: archive-design
description: 将已确认上线的 Figma 设计稿存档到 docs/design/archive/。接收 Figma 文件链接,通过 Figma MCP 读取设计内容,自动整理并写入存档文档。在设计确认、功能上线后调用。
allowed-tools: Read, Write, Glob, Grep
user-invocable: true
argument-hint: <figma-url> <功能名称> [版本号,默认v1]
---
# 设计存档工作流
你正在执行 `/archive-design` 命令,将已确认的 Figma 设计存档。
## 参数解析
```
$ARGUMENTS 格式:<figma-url> <功能名称> [版本]
示例:https://figma.com/file/xxx 入库单 v1
```
`$ARGUMENTS` 中提取:
- `FIGMA_URL`Figma 文件链接
- `FEATURE_NAME`:功能名称(用于目录和文件命名)
- `VERSION`:版本号,未提供则默认 `v1`
## 执行步骤
### Step 1:读取 Figma 文件
使用 Figma MCP 工具读取设计文件内容:
- 获取所有页面列表
- 获取主要页面的组件结构
- 获取设计 Token(颜色、文字样式、间距)
- 获取组件列表和说明
如果 Figma MCP 不可用,跳过此步骤,仅记录链接和手动填写的信息。
### Step 2:读取已有需求和架构文档
读取以下文件,补充存档上下文:
- `docs/requirements/{功能名称}.md`(如存在)
- `docs/api/{功能名称}.md`(如存在)
- `docs/design/{功能名称}-design-brief.md`(如存在)
### Step 3:写入存档文档
创建 `docs/design/archive/{功能名称}/{版本}/design-archive.md`
```markdown
# {功能名称} — 设计存档 {版本}
**存档时间**{当前日期}
**Figma 文件**{FIGMA_URL}
**状态**:✅ 已确认上线
---
## 设计概览
### 页面清单
| 页面名称 | Figma 画板 | 说明 |
|---------|-----------|------|
| (从 Figma 读取或手动填写)|
### 核心设计决策
(从 docs/design/{功能名称}-design-brief.md 中提取设计目标和关键决策)
---
## 设计规范
### 颜色使用
```yaml
(从 Figma 读取,或按项目基础规范填写)
主色: #1565C0 用于:顶栏、主按钮、链接
成功色: #2E7D32 用于:已审核状态标签
危险色: #C62828 用于:删除操作、库存不足警告
```
### 组件清单
```yaml
(列出本功能使用的主要组件)
复用现有组件:
- AppDataTable:通用数据表格
- StatusBadge:状态标签
- FormDialog:弹窗表单
新增组件:
- (从 Figma 读取或手动填写)
```
### 关键尺寸
```yaml
(从 Figma 读取,或手动填写关键布局尺寸)
弹窗宽度: 720px
表格行高: 48px
工具栏高: 52px
```
---
## 交互说明
(描述关键交互逻辑,如状态流转、操作确认弹窗等)
---
## 与需求的对应关系
| 用户故事 | 设计实现 |
|---------|---------|
| (从需求文档提取,对应到具体设计页面)|
---
## 历史版本
| 版本 | 时间 | 变更说明 |
|------|------|---------|
| {版本} | {当前日期} | 初始版本 |
```
### Step 4:更新设计索引
在 `docs/design/archive/INDEX.md` 中追加一行记录(如文件不存在则创建):
```markdown
# 设计存档索引
| 功能 | 版本 | 存档时间 | Figma 链接 | 存档文档 |
|------|------|---------|-----------|---------|
| {功能名称} | {版本} | {日期} | [查看]({FIGMA_URL}) | [文档](/{功能名称}/{版本}/design-archive.md) |
```
### Step 5:完成提示
输出:
```
✅ 设计存档完成
📁 存档路径:docs/design/archive/{功能名称}/{版本}/design-archive.md
🔗 Figma 链接:{FIGMA_URL}
📋 索引已更新:docs/design/archive/INDEX.md
下一步建议:
- 运行 /design-review 对比实现与设计稿的差异
- 提交存档文件到 gitgit add docs/design/archive/
```
-127
View File
@@ -1,127 +0,0 @@
---
name: design-review
description: 对比 Figma 设计稿与 Flutter 实现代码,输出还原度差异报告。在 flutter-coder 实现完成后、上线前调用,确保实现与设计一致。
allowed-tools: Read, Write, Glob, Grep
user-invocable: true
argument-hint: <figma-url> <功能名称>
---
# 设计还原度检查工作流
你正在执行 `/design-review` 命令,对比 Figma 设计稿与 Flutter 实现的差异。
## 参数解析
```
$ARGUMENTS 格式:<figma-url> <功能名称>
示例:https://figma.com/file/xxx 入库单
```
## 执行步骤
### Step 1:读取 Figma 设计规范
使用 Figma MCP 读取:
- 各页面的组件结构和层级
- 颜色值(精确到 hex
- 字体大小、字重
- 间距数值(padding、margin、gap
- 组件状态(hover、disabled、selected 等)
如果 Figma MCP 不可用,读取 `docs/design/{功能名称}-design-brief.md` 中的设计规范作为对照基准。
### Step 2:读取 Flutter 实现代码
查找并读取相关实现文件:
```
client/lib/screens/ → 找到功能名称相关的 screen 文件
client/lib/widgets/ → 找到使用的组件文件
client/lib/models/ → 数据模型
```
重点检查:
- 颜色是否使用了正确的 hex 值或 Theme 变量
- 字体大小、字重是否与设计一致
- 间距(padding/margin)数值是否匹配
- 组件状态样式是否完整实现
- 布局结构是否与设计稿一致
### Step 3:生成差异报告
写入 `docs/review/{功能名称}-design-review.md`
```markdown
# {功能名称} — 设计还原度报告
**检查时间**{当前日期}
**Figma 设计稿**{FIGMA_URL}
**检查范围**{列出检查的 Flutter 文件}
---
## 总体评分
还原度:**X / 10**
(10分=完全一致,8分=细节有出入,6分=布局基本对但样式差异明显)
---
## 必须修复(影响视觉一致性)
### [DR-001] 颜色不一致
**页面**:入库单列表 — 状态标签
**设计稿**:已审核状态背景色 `#E8F5E9`,文字 `#2E7D32`
**实现**:背景色 `Colors.green.shade100`(实际值 `#DCEDC8`,偏浅)
**文件**`client/lib/widgets/status_badge.dart:34`
**修复**:改为 `Color(0xFFE8F5E9)`
### [DR-002] 间距不一致
**页面**:新建入库单弹窗
**设计稿**:表单字段间距 16px
**实现**`SizedBox(height: 12)` — 差了 4px
**文件**`client/lib/screens/stock_in/stock_in_form.dart:67`
**修复**:改为 `SizedBox(height: 16)`
---
## 建议调整(不影响主要功能,但影响精细度)
### [DR-003] 字体字重
**页面**:表格标题行
**设计稿**`FontWeight.w600`
**实现**`FontWeight.bold`(实际是 w700,略粗)
**建议**:统一改为 `FontWeight.w600`
---
## 通过检查项
- ✅ 整体布局结构与设计稿一致
- ✅ 主色 #1565C0 使用正确
- ✅ 顶栏高度 56px 正确
- ✅ 侧边栏宽度 200px 正确
- ✅ 状态流转交互与设计一致
---
## 修复建议
优先处理 DR-001、DR-002(颜色和间距是最直观的差异)。
修复后可再次运行 `/design-review` 验证。
```
### Step 4:完成提示
输出:
```
✅ 设计还原度检查完成
📋 报告路径:docs/review/{功能名称}-design-review.md
🔍 发现问题:必须修复 X 项,建议调整 Y 项
后续步骤:
1. 将报告发给 flutter-coder 修复(必须修复项)
2. 修复后再次运行 /design-review 验证
3. 全部通过后运行 /archive-design 存档设计稿
```
+4
View File
@@ -55,6 +55,10 @@ coverage/
# 编译产物(后端可执行文件)
backend/issue
backend/gencode
# 全局 todo skill 的本地数据(看板/任务,不入库)
/todo/
# Claude 私有文件
.claude/commands/
-8
View File
@@ -1,8 +0,0 @@
{
"mcpServers": {
"figma": {
"type": "http",
"url": "https://mcp.figma.com/mcp"
}
}
}
+11
View File
@@ -5,6 +5,17 @@ All notable changes to this project will be documented in this file.
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
## [1.0.59] - 2026-06-19
### 新功能
- 授权到期后可在界面内直接续费:输入兑换码(形如 JIUKU-XXXX-XXXX)即可为门店叠加授权时长
### 改进
- 品牌焕新:登录页启用「岩美酒库」品牌标识,桌面 / 移动 / 网页各端应用图标统一为品牌图标
- 「激活授权」改为「兑换激活码」语义,文案与提示更贴合按时长续期的使用方式
- 授权状态实时生效:续费、到期或授权被调整后无需重新登录即可刷新;账号被停用或在其他设备登录时及时自动退出
- 「退出登录」收入右上角用户名下拉菜单,界面更聚焦
## [1.0.58] - 2026-06-18
### 修复
+25 -1
View File
@@ -5,7 +5,31 @@
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
## [1.0.59] - 2026-06-19
## [1.0.62] - 2026-06-19
### 新功能
- 服务端新增多维限流防护:按客户端 IP、按门店分别限速,登录/注册/刷新令牌等敏感接口与公开查询接口各有独立速率上限,有效抵御恶意刷接口与流量冲击(DDoS),正常使用不受影响
- 登录暴力破解加固:除按账号锁定外,新增按来源 IP 的失败次数限制,单一 IP 连续撞库多个账号会被整体拦截
### 改进
- 在反向代理后正确识别真实客户端 IP,使限流与失败锁定按真实来源计数,避免误伤同一出口的正常用户
- 门店可保存录入默认值(默认系列/规格等)等轻量配置,刷新页面后依然保留
## [1.0.61] - 2026-06-19
### 改进
- 平台侧兑换券生成工具随服务端一并发布并部署到服务器,可直接在服务器上批量生成激活码,无需每次本地编译上传;同步清理已退役的授权签名相关配置项
## [1.0.60] - 2026-06-19
### 新功能
- 激活码升级为「时长兑换券」:每张码代表一段时长(如 +1 年 / +6 个月),兑换后叠加到当前授权之后,可购买多张累计续期;过期后兑换则从兑换当天重新起算
- 激活码改为简短易输入的形式(形如 JIUKU-XXXX-XXXX),告别此前需复制粘贴的超长字符串
- 同一张激活码只能被成功兑换一次,重复或无效的码会明确提示
### 改进
- 续费/兑换成功后授权状态即时刷新,无需退出重新登录
- 新门店首次使用仍自动获得 30 天试用,开箱即用
### 新功能
- 会话安全加固:刷新令牌每次续期自动轮换并具备盗用检测,旧令牌一旦被重放即吊销整条会话;新增失败登录记录便于审计风控
+4 -3
View File
@@ -324,10 +324,11 @@ CI/CDForgejo)按 tag 前缀触发对应 workflow,自动:编译 → 测
### 项目 TODO 管理
**本项目已停用 todo。** 即便全局 `~/.claude/CLAUDE.md` 要求用 `/todo` 记录待办,**本项目一律不记 todo、不调用 `/todo`、不创建 `todo/` 数据**(此项目规则覆盖全局规则)。需要追踪的事项直接在对话里说明即可
**统一使用全局 todo 工具**`~/.claude/skills/todo/`,遵循全局 `~/.claude/CLAUDE.md` 的规则)。本项目**不再维护自己的 `/todo` slash command**(已删除),一律调用全局 todo skill
- **改动等级仍把控**:接口/schema/跨 2+ 模块的「大改」先进 plan 模式经用户批准,完成后同步更新设计文档;小 bugfix 直接做
- **禁止**使用 TaskCreate/TodoWrite 等内置 todo 工具。
- **数据落地**:全局 skill 把待办存到**当前项目**的 `todo/` 目录(`todo.json` + 看板 `todo/todo.html`,首次使用自动创建);命令须在项目根目录运行,例如 `node ~/.claude/skills/todo/todo.mjs list`
- **禁止**使用 TaskCreate / TodoWrite 等任何内置 todo 工具,一律走全局 todo skill
- **改动等级把控**:接口/schema/跨 2+ 模块的「大改」先进 plan 模式经用户批准,完成后同步更新设计文档;小 bugfix 直接做。
---
+90
View File
@@ -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()
}
-59
View File
@@ -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))
}
-72
View File
@@ -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)")
}
}
+1
View File
@@ -26,6 +26,7 @@ var allModels = []any{
&model.Shop{},
&model.User{},
&model.License{},
&model.LicenseCode{},
&model.ProductCategory{},
&model.Product{},
&model.ProductNameOption{},
+38 -22
View File
@@ -8,17 +8,17 @@ import (
)
type Config struct {
Server ServerConfig
Database DatabaseConfig
JWT JWTConfig
License LicenseConfig
Storage StorageConfig
Session SessionConfig
Server ServerConfig
Database DatabaseConfig
JWT JWTConfig
Storage StorageConfig
Session SessionConfig
RateLimit RateLimitConfig
}
type ServerConfig struct {
Port string `mapstructure:"port"`
Mode string `mapstructure:"mode"` // debug | release
Mode string `mapstructure:"mode"` // debug | release
CORSOrigin string `mapstructure:"cors_origin"` // 允许的 CORS 来源,生产设为具体域名
}
@@ -34,20 +34,29 @@ type JWTConfig struct {
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 覆盖)。
type SessionConfig struct {
LimitDesktop int `mapstructure:"limit_desktop"` // 桌面端(win/mac/linux)最大并发会话,0=禁止
LimitMobile int `mapstructure:"limit_mobile"` // 移动端(android/ios)最大并发会话,0=禁止
LimitWeb int `mapstructure:"limit_web"` // web 端最大并发会话,0=禁止
MaxFailures int `mapstructure:"max_failures"` // 连续登录失败几次后锁定
LockMinutes int `mapstructure:"lock_minutes"` // 锁定时长(分钟
RetentionDays int `mapstructure:"retention_days"` // 已撤销/过期会话与失败登录记录的保留天数,过期后台清理
LimitDesktop int `mapstructure:"limit_desktop"` // 桌面端(win/mac/linux)最大并发会话,0=禁止
LimitMobile int `mapstructure:"limit_mobile"` // 移动端(android/ios)最大并发会话,0=禁止
LimitWeb int `mapstructure:"limit_web"` // web 端最大并发会话,0=禁止
MaxFailures int `mapstructure:"max_failures"` // 同一账号连续登录失败几次后锁定
IPMaxFailures int `mapstructure:"ip_max_failures"` // 同一 IP 跨账号累计失败几次后锁定该 IP(防单 IP 撞多账号
LockMinutes int `mapstructure:"lock_minutes"` // 锁定时长(分钟)
RetentionDays int `mapstructure:"retention_days"` // 已撤销/过期会话与失败登录记录的保留天数,过期后台清理
}
// RateLimitConfig 应用层限流(按真实客户端 IP / 按门店)。各 *PerMin 为每分钟允许次数,
// burst 取同值(允许一分钟额度的突发,之后按速率回补)。Enabled=false 时所有限流放行。
type RateLimitConfig struct {
Enabled bool `mapstructure:"enabled"`
LoginPerMin int `mapstructure:"login_per_min"` // 未鉴权 /auth/login,按 IP
RefreshPerMin int `mapstructure:"refresh_per_min"` // 未鉴权 /auth/refresh,按 IP
RegisterPerMin int `mapstructure:"register_per_min"` // 未鉴权 /public/register,按 IP
ErrorsPerMin int `mapstructure:"errors_per_min"` // 未鉴权 /public/errors,按 IP
ShopListPerMin int `mapstructure:"shop_list_per_min"` // 公开商品列表(主爬取入口),按 IP,最紧
PublicReadPerMin int `mapstructure:"public_read_per_min"` // 其余公开读接口(单品/release),按 IP
ShopRPS int `mapstructure:"shop_rps"` // 认证流量每店每秒,按 shop_id
ShopBurst int `mapstructure:"shop_burst"` // 认证流量每店突发
}
type StorageConfig struct {
@@ -72,9 +81,6 @@ func Load() {
// 显式绑定没有默认值的 key,确保 AutomaticEnv 能找到对应 env var
_ = viper.BindEnv("database.dsn", "DATABASE_DSN")
_ = 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.base_url", "STORAGE_BASE_URL")
_ = viper.BindEnv("storage.public_url", "STORAGE_PUBLIC_URL")
@@ -90,8 +96,18 @@ func Load() {
viper.SetDefault("session.limit_mobile", 2)
viper.SetDefault("session.limit_web", 2) // 默认不禁 web(官网挂着 Web 版 app);设 0 可禁
viper.SetDefault("session.max_failures", 5)
viper.SetDefault("session.ip_max_failures", 20)
viper.SetDefault("session.lock_minutes", 15)
viper.SetDefault("session.retention_days", 90)
viper.SetDefault("ratelimit.enabled", true)
viper.SetDefault("ratelimit.login_per_min", 10)
viper.SetDefault("ratelimit.refresh_per_min", 20)
viper.SetDefault("ratelimit.register_per_min", 5)
viper.SetDefault("ratelimit.errors_per_min", 30)
viper.SetDefault("ratelimit.shop_list_per_min", 30)
viper.SetDefault("ratelimit.public_read_per_min", 60)
viper.SetDefault("ratelimit.shop_rps", 20)
viper.SetDefault("ratelimit.shop_burst", 40)
viper.SetDefault("database.max_idle_conns", 10)
viper.SetDefault("database.max_open_conns", 100)
viper.SetDefault("storage.upload_dir", "./uploads/images")
+1
View File
@@ -71,5 +71,6 @@ require (
golang.org/x/net v0.51.0 // indirect
golang.org/x/sys v0.42.0 // indirect
golang.org/x/text v0.35.0 // indirect
golang.org/x/time v0.15.0 // indirect
google.golang.org/protobuf v1.36.10 // indirect
)
+2
View File
@@ -150,6 +150,8 @@ golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
golang.org/x/text v0.35.0 h1:JOVx6vVDFokkpaq1AEptVzLTpDe9KGpj5tR4/X+ybL8=
golang.org/x/text v0.35.0/go.mod h1:khi/HExzZJ2pGnjenulevKNX1W67CUy0AsXcNubPGCA=
golang.org/x/time v0.15.0 h1:bbrp8t3bGUeFOx08pvsMYRTCVSMk89u4tKbNOZbp88U=
golang.org/x/time v0.15.0/go.mod h1:Y4YMaQmXwGQZoFaVFk4YpCt4FLQMYKZe9oeV/f4MSno=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE=
google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
+1 -6
View File
@@ -11,10 +11,8 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/wangjia/jiu/backend/config"
"github.com/wangjia/jiu/backend/internal/middleware"
"github.com/wangjia/jiu/backend/internal/service"
"github.com/wangjia/jiu/backend/internal/util"
"github.com/wangjia/jiu/backend/testutil"
)
@@ -301,16 +299,13 @@ func TestAuthHandler_Login_ResponseContainsUserInfo(t *testing.T) {
// #8 心跳 /auth/ping 回带授权概况,客户端据此免去单独轮询 /license/info。
func TestAuthHandler_Ping_ReturnsLicense(t *testing.T) {
db := testutil.SetupTestDB()
priv, _, err := util.GenerateEd25519KeyPair()
require.NoError(t, err)
config.C.License.Ed25519PrivateKey = priv
shop := testutil.CreateTestShop(db, "AHPING")
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
authSvc := service.NewAuthService(db)
// 登录触发首登自动 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)
licSvc := service.NewLicenseService(db)
+4 -3
View File
@@ -17,11 +17,12 @@ func NewLicenseHandler(svc *service.LicenseService) *LicenseHandler {
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) {
shopID := middleware.GetShopID(c)
var req struct {
LicenseKey string `json:"license_key" binding:"required"`
LicenseKey string `json:"license_key" binding:"required"` // 承载激活码(短码)
DeviceID string `json:"device_id" binding:"required"`
DeviceName string `json:"device_name"`
Platform string `json:"platform"`
@@ -31,7 +32,7 @@ func (h *LicenseHandler) Activate(c *gin.Context) {
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 {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
+28 -33
View File
@@ -19,28 +19,32 @@ func TestLicenseHandler_Activate_Success(t *testing.T) {
token := getAuthToken(user.ID, shop.ID, "admin")
r := setupProtectedRouter(db)
expiry := time.Now().Add(30 * 24 * time.Hour)
lic := &model.License{
ShopID: shop.ID,
LicenseKey: "LHACT-BBBBB-CCCCC-DDDDD",
IsActive: true,
ExpiresAt: &expiry,
MaxDevices: 3,
}
require.NoError(t, db.Create(lic).Error)
// 码池里放一张未用兑换码(30 天)
require.NoError(t, db.Create(&model.LicenseCode{
Code: "JIUKUTEST0001", Type: "annual", Tier: "standard",
DurationDays: 30, MaxDevices: 3, Status: "unused",
}).Error)
// 请求带连字符/会被归一化为 JIUKUTEST0001
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_name": "Test Machine",
"platform": "windows",
})
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
require.NoError(t, db.Where("license_id = ? AND device_id = ?", lic.ID, "device-123").First(&dev).Error)
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) {
@@ -84,40 +88,31 @@ func TestLicenseHandler_Activate_DeviceLimitExceeded(t *testing.T) {
token := getAuthToken(user.ID, shop.ID, "admin")
r := setupProtectedRouter(db)
// 既有授权:1 个设备名额已占满
future := time.Now().Add(30 * 24 * time.Hour)
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)
// Fill the single allowed slot
require.NoError(t, db.Create(&model.LicenseDevice{
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "existing-device",
}).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{}{
"license_key": "LHBND-BBBBB-CCCCC-DDDDD",
"license_key": "JIUKU-DEVL-IMIT1",
"device_id": "different-device",
})
assert.Equal(t, http.StatusBadRequest, w.Code)
}
func TestLicenseHandler_Activate_Expired(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LH005")
user := testutil.CreateTestUser(db, shop.ID, "admin", "pass", "admin")
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)
// 设备超限 → 整笔回滚:码仍未使用
var lc model.LicenseCode
require.NoError(t, db.Where("code = ?", "JIUKUDEVLIMIT1").First(&lc).Error)
assert.Equal(t, "unused", lc.Status)
}
func TestLicenseHandler_Activate_NoAuth(t *testing.T) {
+20 -6
View File
@@ -42,12 +42,13 @@ func (h *ShopHandler) UpdateInfo(c *gin.Context) {
shopID := middleware.GetShopID(c)
var req struct {
Name string `json:"name"`
Address string `json:"address"`
Phone string `json:"phone"`
ManagerName string `json:"manager_name"`
LogoURL string `json:"logo_url"`
WechatID string `json:"wechat_id"`
Name string `json:"name"`
Address string `json:"address"`
Phone string `json:"phone"`
ManagerName string `json:"manager_name"`
LogoURL string `json:"logo_url"`
WechatID string `json:"wechat_id"`
CustomFields map[string]interface{} `json:"custom_fields"`
}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
@@ -64,6 +65,19 @@ func (h *ShopHandler) UpdateInfo(c *gin.Context) {
if req.LogoURL != "" {
updates["logo_url"] = req.LogoURL
}
// custom_fields 增量 merge:保留已有键(如其它店级配置),只覆盖本次传入的键。
if req.CustomFields != nil {
var cur model.Shop
h.db.Select("custom_fields").Where("id = ?", shopID).First(&cur)
merged := model.JSON{}
for k, v := range cur.CustomFields {
merged[k] = v
}
for k, v := range req.CustomFields {
merged[k] = v
}
updates["custom_fields"] = merged
}
if err := h.db.Model(&model.Shop{}).Where("id = ?", shopID).Updates(updates).Error; err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
+64
View File
@@ -0,0 +1,64 @@
package handler
import (
"net/http"
"testing"
"github.com/gin-gonic/gin"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gorm.io/gorm"
"github.com/wangjia/jiu/backend/internal/middleware"
"github.com/wangjia/jiu/backend/internal/model"
"github.com/wangjia/jiu/backend/testutil"
)
func setupShopRouter(db *gorm.DB) *gin.Engine {
shopH := NewShopHandler(db)
r := gin.New()
r.Use(gin.Recovery())
api := r.Group("/api/v1")
api.Use(middleware.JWT(db))
shop := api.Group("/shop")
shop.GET("/info", shopH.GetInfo)
shop.PUT("/info", shopH.UpdateInfo)
return r
}
// TestUpdateInfo_MergesCustomFields 钉死「设为默认」依赖的契约:
// 传 custom_fields 会 merge 进本店已有 custom_fields(保留旧键),且只动本店。
func TestUpdateInfo_MergesCustomFields(t *testing.T) {
gin.SetMode(gin.TestMode)
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "S_CF1")
admin := testutil.CreateTestUser(db, shop.ID, "admin", "pw", "admin")
// 预置已有 custom_fields,验证 merge 不抹掉其它键
require.NoError(t, db.Model(&model.Shop{}).Where("id = ?", shop.ID).
Update("custom_fields", model.JSON{"foo": "bar"}).Error)
// 另一店做隔离对照
other := testutil.CreateTestShop(db, "S_CF2")
require.NoError(t, db.Model(&model.Shop{}).Where("id = ?", other.ID).
Update("custom_fields", model.JSON{"x": "y"}).Error)
r := setupShopRouter(db)
token := getAuthToken(admin.ID, shop.ID, "admin")
w := makeRequest(r, http.MethodPut, "/api/v1/shop/info", token, jsonBody(
"name", "改名后",
"custom_fields", map[string]interface{}{"default_series_id": 7},
))
assert.Equal(t, http.StatusOK, w.Code)
var got model.Shop
require.NoError(t, db.First(&got, shop.ID).Error)
assert.Equal(t, "改名后", got.Name)
assert.Equal(t, "bar", got.CustomFields["foo"]) // 旧键保留
assert.EqualValues(t, 7, got.CustomFields["default_series_id"]) // 新键写入
// 隔离:另一店的 custom_fields 不受影响
var o model.Shop
require.NoError(t, db.First(&o, other.ID).Error)
assert.Equal(t, "y", o.CustomFields["x"])
_, leaked := o.CustomFields["default_series_id"]
assert.False(t, leaked)
}
@@ -60,7 +60,7 @@ func TestLicenseGuardUsesLiveDBPhase(t *testing.T) {
id INTEGER PRIMARY KEY AUTOINCREMENT,
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
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,
features TEXT, device_id TEXT, activated_at DATETIME
)`).Error)
@@ -125,7 +125,7 @@ func TestLicenseGuardRevokedAndInvalidation(t *testing.T) {
id INTEGER PRIMARY KEY AUTOINCREMENT,
created_at DATETIME, updated_at DATETIME, deleted_at DATETIME,
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,
features TEXT, device_id TEXT, activated_at DATETIME
)`).Error)
+127
View File
@@ -0,0 +1,127 @@
package middleware
import (
"net/http"
"strconv"
"sync"
"time"
"github.com/gin-gonic/gin"
"golang.org/x/time/rate"
"github.com/wangjia/jiu/backend/config"
)
// 限流器内存条目的清理参数:每 5 分钟扫一次,淘汰超过 10 分钟未活动的 key。
// 保证 map 不随攻击者构造的随机 key(IP/shop)无限增长。
const (
rateLimitSweep = 5 * time.Minute
rateLimitIdleTTL = 10 * time.Minute
rateLimitRetryHdr = "60" // Retry-After 秒数(提示客户端退避)
)
// keyedLimiter 按任意字符串 keyIP 或 shop_id)维护独立令牌桶,内存有界(带 janitor)。
// 单实例进程内状态,重启即清零;多实例水平扩展时需改为 Redis(见方案「暂不做」)。
type keyedLimiter struct {
mu sync.Mutex
entries map[string]*limiterBucket
r rate.Limit
burst int
}
type limiterBucket struct {
lim *rate.Limiter
lastSeen time.Time
}
func newKeyedLimiter(r rate.Limit, burst int) *keyedLimiter {
kl := &keyedLimiter{entries: map[string]*limiterBucket{}, r: r, burst: burst}
go kl.janitor()
return kl
}
// get 取(或惰性创建)该 key 的令牌桶并刷新活动时间。
func (kl *keyedLimiter) get(key string) *rate.Limiter {
kl.mu.Lock()
defer kl.mu.Unlock()
b := kl.entries[key]
if b == nil {
b = &limiterBucket{lim: rate.NewLimiter(kl.r, kl.burst)}
kl.entries[key] = b
}
b.lastSeen = time.Now()
return b.lim
}
func (kl *keyedLimiter) janitor() {
t := time.NewTicker(rateLimitSweep)
defer t.Stop()
for range t.C {
kl.sweep(rateLimitIdleTTL)
}
}
// sweep 淘汰超过 ttl 未活动的 key。拆出便于测试。
func (kl *keyedLimiter) sweep(ttl time.Duration) {
now := time.Now()
kl.mu.Lock()
defer kl.mu.Unlock()
for k, b := range kl.entries {
if now.Sub(b.lastSeen) > ttl {
delete(kl.entries, k)
}
}
}
// PerMinute 把「每分钟 n 次」转成 rate.Limit(令牌/秒)。
func PerMinute(n int) rate.Limit {
return rate.Limit(float64(n) / 60.0)
}
// PerSecond 把「每秒 n 次」转成 rate.Limit。
func PerSecond(n int) rate.Limit {
return rate.Limit(n)
}
// rateLimit 通用工厂:keyFn 抽取限流维度的 key(返回空串表示无法判定 → 放行,不误伤)。
// config.C.RateLimit.Enabled=false 时整体放行(应急/测试开关)。
func rateLimit(r rate.Limit, burst int, keyFn func(*gin.Context) string) gin.HandlerFunc {
kl := newKeyedLimiter(r, burst)
return func(c *gin.Context) {
if !config.C.RateLimit.Enabled {
c.Next()
return
}
key := keyFn(c)
if key == "" {
c.Next()
return
}
if !kl.get(key).Allow() {
c.Header("Retry-After", rateLimitRetryHdr)
c.AbortWithStatusJSON(http.StatusTooManyRequests, gin.H{"error": "请求过于频繁,请稍后再试"})
return
}
c.Next()
}
}
// RateLimitByIP 按真实客户端 IP 限流(依赖 main.go 的 SetTrustedProxies + RemoteIPHeaders
// 使 c.ClientIP() 返回不可伪造的真实 IP)。
func RateLimitByIP(r rate.Limit, burst int) gin.HandlerFunc {
return rateLimit(r, burst, func(c *gin.Context) string {
return "ip:" + c.ClientIP()
})
}
// RateLimitByShop 按门店(shop_id,取自 JWT)限流,须挂在 JWT 中间件之后。
// 未取到 shop_id 时放行(交由 JWT 中间件拦截非法 token)。
func RateLimitByShop(r rate.Limit, burst int) gin.HandlerFunc {
return rateLimit(r, burst, func(c *gin.Context) string {
id := GetShopID(c)
if id == 0 {
return ""
}
return "shop:" + strconv.FormatUint(id, 10)
})
}
@@ -0,0 +1,194 @@
package middleware
import (
"net/http"
"net/http/httptest"
"strconv"
"testing"
"time"
"github.com/gin-gonic/gin"
"golang.org/x/time/rate"
"github.com/wangjia/jiu/backend/config"
)
func init() {
gin.SetMode(gin.TestMode)
}
// enableRateLimit 临时打开限流开关,返回恢复函数。
func enableRateLimit(t *testing.T) {
t.Helper()
prev := config.C.RateLimit.Enabled
config.C.RateLimit.Enabled = true
t.Cleanup(func() { config.C.RateLimit.Enabled = prev })
}
// hit 发一个带指定客户端 IP 的请求,返回状态码。SetTrustedProxies 让 X-Forwarded-For 被采信,
// 便于在测试里模拟不同来源 IP。
func hit(r *gin.Engine, ip string) *httptest.ResponseRecorder {
w := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "/", nil)
req.RemoteAddr = ip + ":12345"
r.ServeHTTP(w, req)
return w
}
func newIPRouter(h gin.HandlerFunc) *gin.Engine {
r := gin.New()
r.GET("/", h, func(c *gin.Context) { c.String(http.StatusOK, "ok") })
return r
}
func TestRateLimitByIP_BurstThen429(t *testing.T) {
enableRateLimit(t)
// 每分钟 3 次,burst 3:前 3 个放行,第 4 个 429。
r := newIPRouter(RateLimitByIP(PerMinute(3), 3))
for i := 0; i < 3; i++ {
if w := hit(r, "1.1.1.1"); w.Code != http.StatusOK {
t.Fatalf("第 %d 个请求应放行,得到 %d", i+1, w.Code)
}
}
w := hit(r, "1.1.1.1")
if w.Code != http.StatusTooManyRequests {
t.Fatalf("超出 burst 应 429,得到 %d", w.Code)
}
if ra := w.Header().Get("Retry-After"); ra == "" {
t.Fatal("429 响应应带 Retry-After 头")
}
}
func TestRateLimitByIP_KeysIndependent(t *testing.T) {
enableRateLimit(t)
r := newIPRouter(RateLimitByIP(PerMinute(1), 1))
// 不同 IP 各有独立令牌桶,互不影响。
if w := hit(r, "2.2.2.2"); w.Code != http.StatusOK {
t.Fatalf("IP A 首次应放行,得到 %d", w.Code)
}
if w := hit(r, "3.3.3.3"); w.Code != http.StatusOK {
t.Fatalf("IP B 首次应放行(独立桶),得到 %d", w.Code)
}
// IP A 再来一发应 429(桶已空)。
if w := hit(r, "2.2.2.2"); w.Code != http.StatusTooManyRequests {
t.Fatalf("IP A 第二发应 429,得到 %d", w.Code)
}
}
func TestRateLimitByIP_Refill(t *testing.T) {
enableRateLimit(t)
// 每秒 50 次:耗尽 burst 后等一小会儿令牌补回,能再次放行。
r := newIPRouter(RateLimitByIP(rate.Limit(50), 1))
if w := hit(r, "4.4.4.4"); w.Code != http.StatusOK {
t.Fatalf("首发应放行,得到 %d", w.Code)
}
if w := hit(r, "4.4.4.4"); w.Code != http.StatusTooManyRequests {
t.Fatalf("紧接第二发应 429,得到 %d", w.Code)
}
time.Sleep(40 * time.Millisecond) // 50/s → 20ms 补一个令牌
if w := hit(r, "4.4.4.4"); w.Code != http.StatusOK {
t.Fatalf("等待补充后应放行,得到 %d", w.Code)
}
}
func TestRateLimitDisabledPassthrough(t *testing.T) {
prev := config.C.RateLimit.Enabled
config.C.RateLimit.Enabled = false
defer func() { config.C.RateLimit.Enabled = prev }()
r := newIPRouter(RateLimitByIP(PerMinute(1), 1))
// 关闭时远超额度也全放行。
for i := 0; i < 10; i++ {
if w := hit(r, "5.5.5.5"); w.Code != http.StatusOK {
t.Fatalf("限流关闭时第 %d 发应放行,得到 %d", i+1, w.Code)
}
}
}
func TestKeyedLimiterSweepEvictsIdle(t *testing.T) {
kl := newKeyedLimiter(PerMinute(60), 1)
kl.get("ip:a")
kl.get("ip:b")
if len(kl.entries) != 2 {
t.Fatalf("应有 2 个 entry,得到 %d", len(kl.entries))
}
// 把 a 的活动时间推到很久以前,sweep 应只淘汰 a。
kl.mu.Lock()
kl.entries["ip:a"].lastSeen = time.Now().Add(-time.Hour)
kl.mu.Unlock()
kl.sweep(10 * time.Minute)
kl.mu.Lock()
defer kl.mu.Unlock()
if _, ok := kl.entries["ip:a"]; ok {
t.Fatal("空闲 key a 应被淘汰")
}
if _, ok := kl.entries["ip:b"]; !ok {
t.Fatal("活跃 key b 不应被淘汰")
}
}
// TestTrustedProxyRealIP 复刻 main.go 的可信代理配置:只信任本机写的 X-Real-IP,
// 客户端伪造的 X-Forwarded-For 不被采信 → c.ClientIP() 返回真实 IP,限流不可被请求头绕过。
func TestTrustedProxyRealIP(t *testing.T) {
r := gin.New()
_ = r.SetTrustedProxies([]string{"127.0.0.1", "::1"})
r.RemoteIPHeaders = []string{"X-Real-IP"}
var got string
r.GET("/", func(c *gin.Context) {
got = c.ClientIP()
c.Status(http.StatusOK)
})
w := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "/", nil)
req.RemoteAddr = "127.0.0.1:9999" // 本机 nginx
req.Header.Set("X-Real-IP", "9.9.9.9") // nginx 写入的真实 IP
req.Header.Set("X-Forwarded-For", "1.2.3.4") // 客户端伪造,应被忽略
r.ServeHTTP(w, req)
if got != "9.9.9.9" {
t.Fatalf("ClientIP 应取可信的 X-Real-IP=9.9.9.9,得到 %q(伪造的 XFF 不应生效)", got)
}
}
func TestRateLimitByShop_NoShopPassthrough(t *testing.T) {
enableRateLimit(t)
// 未设置 shop_id(无 JWT 上下文)时放行,交由 JWT 中间件拦截。
r := gin.New()
r.GET("/", RateLimitByShop(PerSecond(1), 1), func(c *gin.Context) { c.String(http.StatusOK, "ok") })
for i := 0; i < 5; i++ {
if w := hit(r, "6.6.6.6"); w.Code != http.StatusOK {
t.Fatalf("无 shop_id 时第 %d 发应放行,得到 %d", i+1, w.Code)
}
}
}
func TestRateLimitByShop_PerShop(t *testing.T) {
enableRateLimit(t)
r := gin.New()
// 模拟 JWT 已注入 shop_id(取自 ?shop= 查询参数)。
r.GET("/", func(c *gin.Context) {
id, _ := strconv.ParseUint(c.Query("shop"), 10, 64)
c.Set(CtxShopID, id)
}, RateLimitByShop(PerSecond(1), 1), func(c *gin.Context) { c.String(http.StatusOK, "ok") })
shopHit := func(shop string) int {
w := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "/?shop="+shop, nil)
r.ServeHTTP(w, req)
return w.Code
}
if c := shopHit("1"); c != http.StatusOK {
t.Fatalf("shop1 首发应放行,得到 %d", c)
}
if c := shopHit("2"); c != http.StatusOK {
t.Fatalf("shop2 首发应放行(独立桶),得到 %d", c)
}
if c := shopHit("1"); c != http.StatusTooManyRequests {
t.Fatalf("shop1 第二发应 429,得到 %d", c)
}
}
+3 -1
View File
@@ -7,7 +7,9 @@ type License struct {
ShopID uint64 `gorm:"not null;index" json:"shop_id"`
LicenseKey string `gorm:"size:768;uniqueIndex" json:"license_key"`
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"`
MaxDevices int `gorm:"default:3" json:"max_devices"`
Features JSON `gorm:"type:json" json:"features,omitempty"`
+27
View File
@@ -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"`
}
+36 -23
View File
@@ -4,6 +4,7 @@ import (
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"github.com/wangjia/jiu/backend/config"
"github.com/wangjia/jiu/backend/internal/handler"
"github.com/wangjia/jiu/backend/internal/middleware"
"github.com/wangjia/jiu/backend/internal/service"
@@ -46,31 +47,43 @@ func Setup(r *gin.Engine, db *gorm.DB) {
// 版本信息(无需认证,用于客户端更新检查)
r.GET("/version", handler.GetVersion)
// 限流维度(按真实客户端 IP,依赖 main.go 的 SetTrustedProxies)。
// 防爆破/防刷写/防爬:未鉴权接口逐条挂 per-IP 限流,公开商品列表最紧。
rl := config.C.RateLimit
publicReadIP := middleware.RateLimitByIP(middleware.PerMinute(rl.PublicReadPerMin), rl.PublicReadPerMin)
shopListIP := middleware.RateLimitByIP(middleware.PerMinute(rl.ShopListPerMin), rl.ShopListPerMin)
loginIP := middleware.RateLimitByIP(middleware.PerMinute(rl.LoginPerMin), rl.LoginPerMin)
refreshIP := middleware.RateLimitByIP(middleware.PerMinute(rl.RefreshPerMin), rl.RefreshPerMin)
registerIP := middleware.RateLimitByIP(middleware.PerMinute(rl.RegisterPerMin), rl.RegisterPerMin)
errorsIP := middleware.RateLimitByIP(middleware.PerMinute(rl.ErrorsPerMin), rl.ErrorsPerMin)
// 公开商品详情页(注入 OG 标签的 Flutter index.html,供社交分享爬虫读取)
r.GET("/product/:public_id", publicH.ProductPage)
r.GET("/product/:public_id", publicReadIP, publicH.ProductPage)
v1 := r.Group("/api/v1")
// 公开路由(无需登录)
// 公开路由(无需登录):登录/刷新按 IP 限流(与 service 层失败锁定互补)
auth := v1.Group("/auth")
{
auth.POST("/login", authH.Login)
auth.POST("/refresh", authH.Refresh)
auth.POST("/login", loginIP, authH.Login)
auth.POST("/refresh", refreshIP, authH.Refresh)
}
// 公开接口(无需登录)
// 公开接口(无需登录):读接口防爬、写接口防刷
public := v1.Group("/public")
{
public.GET("/products/:public_id", publicH.GetProduct)
public.GET("/shops/:shop_code/products", publicH.ListShopProducts)
public.GET("/release", publicH.GetRelease)
public.POST("/errors", errorReportH.Submit)
public.POST("/register", authH.Register)
public.GET("/products/:public_id", publicReadIP, publicH.GetProduct)
public.GET("/shops/:shop_code/products", shopListIP, publicH.ListShopProducts)
public.GET("/release", publicReadIP, publicH.GetRelease)
public.POST("/errors", errorsIP, errorReportH.Submit)
public.POST("/register", registerIP, authH.Register)
}
// 需要 JWT 的基础路由组
api := v1.Group("")
api.Use(middleware.JWT(db))
// 每店限流(按 shop_id,挂在 JWT 之后):防单店打爆共享后端,保障多租户公平
api.Use(middleware.RateLimitByShop(middleware.PerSecond(rl.ShopRPS), rl.ShopBurst))
// 会话/在线状态(豁免 LicenseGuard:登出/心跳/在线列表在锁定期也要可用)
{
@@ -179,11 +192,11 @@ func Setup(r *gin.Engine, db *gorm.DB) {
// 财务
finance := api.Group("/finance")
{
finance.GET("/records", financeH.ListRecords)
finance.POST("/records", financeH.Create)
finance.PUT("/records/:id/close", financeH.Close)
finance.PUT("/records/close-by-ref", financeH.CloseByRef)
finance.GET("/summary", financeH.Summary)
finance.GET("/records", financeH.ListRecords)
finance.POST("/records", financeH.Create)
finance.PUT("/records/:id/close", financeH.Close)
finance.PUT("/records/close-by-ref", financeH.CloseByRef)
finance.GET("/summary", financeH.Summary)
}
// 酒行信息
@@ -211,15 +224,15 @@ func Setup(r *gin.Engine, db *gorm.DB) {
// 导入
imp := api.Group("/import")
{
imp.POST("/products", importH.ImportProducts)
imp.POST("/partners", importH.ImportPartners)
imp.POST("/product-names", importH.ImportProductNames)
imp.POST("/products", importH.ImportProducts)
imp.POST("/partners", importH.ImportPartners)
imp.POST("/product-names", importH.ImportProductNames)
imp.POST("/product-series", importH.ImportProductSeries)
imp.POST("/product-specs", importH.ImportProductSpecs)
imp.POST("/product-codes", importH.ImportProductCodes)
imp.POST("/stock-in", importH.ImportStockIn)
imp.POST("/stock-out", importH.ImportStockOut)
imp.POST("/inventory", importH.ImportInventory)
imp.POST("/product-specs", importH.ImportProductSpecs)
imp.POST("/product-codes", importH.ImportProductCodes)
imp.POST("/stock-in", importH.ImportStockIn)
imp.POST("/stock-out", importH.ImportStockOut)
imp.POST("/inventory", importH.ImportInventory)
}
// 基础数据选项(名称/系列/规格)
+57 -10
View File
@@ -40,16 +40,23 @@ type DeviceInfo struct {
}
// loginLimiter 内存登录失败限流器(单实例,重启即清零)。
// 两个维度共用同一张表:账号维度 key="<shopCode>|<username>"IP 维度 key="ip|<addr>"
// 分别用不同阈值锁定。带 janitor 清理空闲 entry,避免攻击者用随机 key 灌爆内存。
type loginLimiter struct {
mu sync.Mutex
entries map[string]*limiterEntry
mu sync.Mutex
entries map[string]*limiterEntry
janitorOnce sync.Once
}
type limiterEntry struct {
failures int
lockedTill time.Time
lastSeen time.Time
}
// loginLimiterIdleTTL:已解锁且超过该时长未活动的 entry 会被 janitor 清理。
const loginLimiterIdleTTL = 30 * time.Minute
var loginLim = &loginLimiter{entries: map[string]*limiterEntry{}}
// locked 返回该 key 是否处于锁定中。
@@ -57,11 +64,16 @@ func (l *loginLimiter) locked(key string) bool {
l.mu.Lock()
defer l.mu.Unlock()
e := l.entries[key]
return e != nil && time.Now().Before(e.lockedTill)
if e == nil {
return false
}
e.lastSeen = time.Now()
return time.Now().Before(e.lockedTill)
}
// recordFailure 记一次失败,达到阈值则锁定
func (l *loginLimiter) recordFailure(key string) {
// recordFailure 记一次失败,达到 max 阈值则锁定(max<=0 表示该维度不锁)
func (l *loginLimiter) recordFailure(key string, max int) {
l.startJanitor()
l.mu.Lock()
defer l.mu.Unlock()
e := l.entries[key]
@@ -69,8 +81,8 @@ func (l *loginLimiter) recordFailure(key string) {
e = &limiterEntry{}
l.entries[key] = e
}
e.lastSeen = time.Now()
e.failures++
max := config.C.Session.MaxFailures
if max > 0 && e.failures >= max {
e.lockedTill = time.Now().Add(time.Duration(config.C.Session.LockMinutes) * time.Minute)
e.failures = 0
@@ -84,6 +96,26 @@ func (l *loginLimiter) reset(key string) {
delete(l.entries, key)
}
// startJanitor 惰性启动后台清理(仅一次):每 5 分钟淘汰「未锁定且超过 TTL 未活动」的 entry。
func (l *loginLimiter) startJanitor() {
l.janitorOnce.Do(func() {
go func() {
t := time.NewTicker(5 * time.Minute)
defer t.Stop()
for range t.C {
now := time.Now()
l.mu.Lock()
for k, e := range l.entries {
if now.After(e.lockedTill) && now.Sub(e.lastSeen) > loginLimiterIdleTTL {
delete(l.entries, k)
}
}
l.mu.Unlock()
}
}()
})
}
type AuthService struct {
db *gorm.DB
}
@@ -101,15 +133,27 @@ type TokenPair struct {
// Login 账号密码登录
func (s *AuthService) Login(shopCode, username, password string, dev DeviceInfo) (*TokenPair, *model.User, error) {
// 两个限流维度:账号维度防针对单账号的密码爆破;IP 维度(阈值更高)防单 IP 撞多个账号、
// 以及用随机账号灌爆内存。IP 为空(本地/测试)时退化为仅账号维度。
limiterKey := shopCode + "|" + username
if loginLim.locked(limiterKey) {
ipKey := ""
if dev.IP != "" {
ipKey = "ip|" + dev.IP
}
recordFail := func() {
loginLim.recordFailure(limiterKey, config.C.Session.MaxFailures)
if ipKey != "" {
loginLim.recordFailure(ipKey, config.C.Session.IPMaxFailures)
}
}
if loginLim.locked(limiterKey) || (ipKey != "" && loginLim.locked(ipKey)) {
s.recordLoginAttempt(shopCode, username, dev, false, "locked")
return nil, nil, ErrTooManyAttempts
}
var shop model.Shop
if err := s.db.Where("code = ?", shopCode).First(&shop).Error; err != nil {
loginLim.recordFailure(limiterKey)
recordFail()
s.recordLoginAttempt(shopCode, username, dev, false, "invalid_shop")
return nil, nil, ErrInvalidCredentials
}
@@ -117,7 +161,7 @@ func (s *AuthService) Login(shopCode, username, password string, dev DeviceInfo)
var user model.User
if err := s.db.Where("shop_id = ? AND username = ? AND deleted_at IS NULL", shop.ID, username).
First(&user).Error; err != nil {
loginLim.recordFailure(limiterKey)
recordFail()
s.recordLoginAttempt(shopCode, username, dev, false, "invalid_user")
return nil, nil, ErrInvalidCredentials
}
@@ -128,7 +172,7 @@ func (s *AuthService) Login(shopCode, username, password string, dev DeviceInfo)
}
if err := bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(password)); err != nil {
loginLim.recordFailure(limiterKey)
recordFail()
s.recordLoginAttempt(shopCode, username, dev, false, "bad_password")
return nil, nil, ErrInvalidCredentials
}
@@ -193,6 +237,9 @@ func (s *AuthService) Login(shopCode, username, password string, dev DeviceInfo)
}
loginLim.reset(limiterKey)
if ipKey != "" {
loginLim.reset(ipKey)
}
user.LastLoginAt = &now
pair, err := s.issueTokens(user.ID, shop.ID, user.Role, sid, jti)
+46 -5
View File
@@ -1,6 +1,7 @@
package service
import (
"strconv"
"testing"
"time"
@@ -9,7 +10,6 @@ import (
"github.com/wangjia/jiu/backend/config"
"github.com/wangjia/jiu/backend/internal/model"
"github.com/wangjia/jiu/backend/internal/util"
"github.com/wangjia/jiu/backend/testutil"
)
@@ -31,9 +31,6 @@ func TestAuthService_Login_Success(t *testing.T) {
func TestAuthService_Login_AutoTrialOnFirstUse(t *testing.T) {
db := testutil.SetupTestDB()
priv, _, err := util.GenerateEd25519KeyPair()
require.NoError(t, err)
config.C.License.Ed25519PrivateKey = priv
shop := testutil.CreateTestShop(db, "TRIAL001")
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
@@ -46,7 +43,7 @@ func TestAuthService_Login_AutoTrialOnFirstUse(t *testing.T) {
svc := NewAuthService(db)
// 首次登录 → 自动签发 30 天 trial
_, _, err = svc.Login("TRIAL001", "admin", "password123", DeviceInfo{Platform: "windows"})
_, _, err := svc.Login("TRIAL001", "admin", "password123", DeviceInfo{Platform: "windows"})
require.NoError(t, err)
var lic model.License
@@ -151,6 +148,50 @@ func TestAuthService_RefreshTokens_Invalid(t *testing.T) {
assert.Nil(t, newPair)
}
// TestLogin_AccountLockoutAfterMaxFailures 同一账号连续失败达阈值后锁定(回归)。
func TestLogin_AccountLockoutAfterMaxFailures(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LOCK_ACC")
testutil.CreateTestUser(db, shop.ID, "admin", "password123", "admin")
config.C.Session.MaxFailures = 3
svc := NewAuthService(db)
defer loginLim.reset("LOCK_ACC|admin")
for i := 0; i < 3; i++ {
_, _, err := svc.Login("LOCK_ACC", "admin", "wrong", DeviceInfo{Platform: "windows", IP: "10.0.0.1"})
require.ErrorIs(t, err, ErrInvalidCredentials)
}
// 第 4 次即便密码正确也被锁定拒绝。
_, _, err := svc.Login("LOCK_ACC", "admin", "password123", DeviceInfo{Platform: "windows", IP: "10.0.0.1"})
assert.ErrorIs(t, err, ErrTooManyAttempts)
}
// TestLogin_IPLockoutAcrossAccounts 单 IP 跨多个(不存在的)账号累计失败达 IP 阈值后锁该 IP。
// 每次用不同用户名,账号维度永不触发锁定,只有 IP 维度会锁——验证 per-IP 防撞库 + 防内存灌爆。
func TestLogin_IPLockoutAcrossAccounts(t *testing.T) {
db := testutil.SetupTestDB()
testutil.CreateTestShop(db, "LOCK_IP")
config.C.Session.MaxFailures = 5
config.C.Session.IPMaxFailures = 4
const attackIP = "203.0.113.9"
svc := NewAuthService(db)
defer loginLim.reset("ip|" + attackIP)
// 4 次不同用户名(invalid_user),账号 key 各不相同永不锁;IP key 累计到 4 → 锁 IP。
for i := 0; i < 4; i++ {
uname := "ghost" + strconv.Itoa(i)
_, _, err := svc.Login("LOCK_IP", uname, "whatever", DeviceInfo{Platform: "windows", IP: attackIP})
require.ErrorIs(t, err, ErrInvalidCredentials)
}
// 同 IP 再来一发(仍是新用户名,账号维度无锁)→ 被 IP 锁拦下。
_, _, err := svc.Login("LOCK_IP", "ghostX", "whatever", DeviceInfo{Platform: "windows", IP: attackIP})
assert.ErrorIs(t, err, ErrTooManyAttempts)
// 另一 IP 不受影响。
_, _, err = svc.Login("LOCK_IP", "ghostY", "whatever", DeviceInfo{Platform: "windows", IP: "198.51.100.7"})
assert.ErrorIs(t, err, ErrInvalidCredentials)
}
func TestHashPassword(t *testing.T) {
hash, err := HashPassword("mypassword")
require.NoError(t, err)
+145 -86
View File
@@ -1,18 +1,13 @@
package service
import (
"crypto/hmac"
"crypto/sha256"
"encoding/base32"
"errors"
"fmt"
"log"
"strings"
"time"
"github.com/google/uuid"
"gorm.io/gorm"
"github.com/wangjia/jiu/backend/config"
"github.com/wangjia/jiu/backend/internal/middleware"
"github.com/wangjia/jiu/backend/internal/model"
"github.com/wangjia/jiu/backend/internal/util"
@@ -22,7 +17,12 @@ var (
ErrLicenseNotFound = errors.New("license not found")
ErrLicenseInactive = errors.New("license is inactive")
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 {
@@ -33,69 +33,150 @@ func NewLicenseService(db *gorm.DB) *LicenseService {
return &LicenseService{db: db}
}
// GenerateKey 生成许可证激活码
// 格式:HMAC-SHA256(shopID+licenseType+expiry, secret) → base32, 每5字符加'-'
func GenerateKey(shopID uint64, licenseType string, expiresAt *time.Time) string {
payload := fmt.Sprintf("%d:%s", shopID, licenseType)
if expiresAt != nil {
payload += ":" + expiresAt.Format("20060102")
}
mac := hmac.New(sha256.New, []byte(config.C.License.HMACSecret))
mac.Write([]byte(payload))
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
// Redeem 兑换一张激活码(时长券):校验码有效且未使用 → 把时长叠加到门店当前授权的到期时间
// → 标记码已用 → 绑定本设备。整个过程在单事务内,check-then-act 用 FOR UPDATE 锁行,
// 保证一码只能被成功兑换一次(并发下恰一个成功)。
//
// 时长叠加规则:新到期 = max(今天, 当前到期) + duration_daysduration_days=0 视为永久(到期置 NULL)。
func (s *LicenseService) Redeem(shopID uint64, rawCode, deviceID, deviceName, platform string) (*model.License, error) {
code := util.NormalizeCode(rawCode)
if code == "" {
return nil, ErrCodeNotFound
}
// 激活成功即清除该店 phase 缓存:续费/换新授权码后写权限即时恢复,不必等 30s TTL。
defer middleware.InvalidateLicensePhase(shopID)
var existing model.LicenseDevice
err := s.db.Where("license_id = ? AND device_id = ?", lic.ID, deviceID).First(&existing).Error
if err == nil {
// Device already bound — just touch last_seen_at (handled by autoUpdateTime)
if err := s.db.Model(&existing).Update("device_name", deviceName).Error; err != nil {
return nil, err
var result model.License
err := s.db.Transaction(func(tx *gorm.DB) error {
// 1) 锁定并校验兑换码
var lc model.LicenseCode
if err := tx.Set("gorm:query_option", "FOR UPDATE").
Where("code = ?", code).First(&lc).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrCodeNotFound
}
return err
}
switch lc.Status {
case "redeemed":
return ErrCodeUsed
case "void":
return ErrCodeVoid
}
return &lic, nil
}
// New device — enforce max_devices
var count int64
if err := s.db.Model(&model.LicenseDevice{}).Where("license_id = ?", lic.ID).Count(&count).Error; err != nil {
now := time.Now()
// 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
}
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,
ShopID: shopID,
ShopID: lic.ShopID,
DeviceID: deviceID,
DeviceName: deviceName,
Platform: platform,
}
if err := s.db.Create(&dev).Error; err != nil {
return nil, err
}
return &lic, nil
}).Error
}
// Verify 验证设备许可证(客户端启动时调用)。
@@ -188,33 +269,15 @@ func (s *LicenseService) Deactivate(shopID uint64, deviceID string) error {
Delete(&model.LicenseDevice{}).Error
}
// issueTrialLicense 为门店签发并写入一条 30 天 trial license。
// 私钥未配置或签发/落库失败返回 error,由调用方决定如何处理(注册路径 Fatal、
// 登录路径仅记日志)。
// issueTrialLicense 为门店写入一条 30 天 trial license。直接建行(无需签名/私钥),
// license_key 用合成唯一值占位以满足 NOT NULL/UNIQUE。落库失败返回 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)
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{
ShopID: shopID,
LicenseKey: token,
LicenseKey: "TRIAL-" + uuid.New().String(),
Type: "trial",
Tier: "standard",
ExpiresAt: &expiresAt,
IsActive: true,
MaxDevices: 1,
@@ -228,12 +291,8 @@ func issueTrialLicense(db *gorm.DB, shopID uint64) error {
return nil
}
// createTrialLicense 在注册事务中为新门店签发 30 天 trial license。
// 私钥未配置时 Fatal,防止生产环境静默跳过导致新注册门店无 license。
// createTrialLicense 在注册事务中为新门店写入 30 天 trial license。失败仅记日志、不阻断注册。
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 {
log.Printf("[license] failed to create trial license for shop %d: %v", shopID, err)
}
+158 -142
View File
@@ -6,155 +6,185 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gorm.io/gorm"
"github.com/wangjia/jiu/backend/internal/model"
"github.com/wangjia/jiu/backend/internal/util"
"github.com/wangjia/jiu/backend/testutil"
)
func TestLicenseService_Activate_Success(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LIC001")
expiry := time.Now().Add(30 * 24 * time.Hour)
lic := &model.License{
ShopID: shop.ID,
LicenseKey: "AAAAA-BBBBB-CCCCC-DDDDD",
IsActive: true,
ExpiresAt: &expiry,
MaxDevices: 3,
}
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)
// createCode 在码池写入一张 unused 兑换码(默认 type=annual / tier=standard)。
func createCode(t *testing.T, db *gorm.DB, code string, durationDays, maxDevices int) {
t.Helper()
require.NoError(t, db.Create(&model.LicenseCode{
Code: util.NormalizeCode(code),
Type: "annual",
Tier: "standard",
DurationDays: durationDays,
MaxDevices: maxDevices,
Status: "unused",
}).Error)
}
func TestLicenseService_Activate_SameDeviceIdempotent(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LIC002")
lic := &model.License{
ShopID: shop.ID,
LicenseKey: "EEEEE-FFFFF-GGGGG-HHHHH",
IsActive: true,
MaxDevices: 3,
func daysFromNow(t *time.Time) float64 {
if t == nil {
return 0
}
require.NoError(t, db.Create(lic).Error)
// Pre-bind the device
require.NoError(t, db.Create(&model.LicenseDevice{
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "same-device",
return time.Until(*t).Hours() / 24
}
// 首次兑换(门店尚无授权行):新建一行,到期 = 今天 + 时长,码标记已用并绑定设备。
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)
svc := NewLicenseService(db)
// Re-activating same device should succeed (idempotent)
result, err := svc.Activate(shop.ID, "EEEEE-FFFFF-GGGGG-HHHHH", "same-device", "Updated Name", "windows")
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)
_, err := svc.Redeem(shop.ID, "JIUKU-VOID-CODE0", "dev-1", "", "")
assert.Equal(t, ErrCodeVoid, err)
}
func TestLicenseService_Activate_DeviceLimitExceeded(t *testing.T) {
// 设备超上限:兑换整体回滚——码保持 unused、到期不变。
func TestLicenseService_Redeem_DeviceLimitRollsBack(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LIC003")
shop := testutil.CreateTestShop(db, "RDM009")
expiry := time.Now().Add(10 * 24 * time.Hour)
lic := &model.License{
ShopID: shop.ID,
LicenseKey: "IIIII-JJJJJ-KKKKK-LLLLL",
IsActive: true,
MaxDevices: 2,
ShopID: shop.ID, LicenseKey: "TRIAL-RDM009", Type: "trial", Tier: "standard",
ExpiresAt: &expiry, IsActive: true, MaxDevices: 1,
}
require.NoError(t, db.Create(lic).Error)
// Fill up the device limit
require.NoError(t, db.Create(&model.LicenseDevice{LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "dev-1"}).Error)
require.NoError(t, db.Create(&model.LicenseDevice{LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "dev-2"}).Error)
// 已占满 1 个设备名额
require.NoError(t, db.Create(&model.LicenseDevice{
LicenseID: lic.ID, ShopID: shop.ID, DeviceID: "old-dev",
}).Error)
createCode(t, db, "JIUKU-DEVL-IMIT", 365, 0) // 不提升设备上限
svc := NewLicenseService(db)
result, err := svc.Activate(shop.ID, "IIIII-JJJJJ-KKKKK-LLLLL", "dev-3", "", "")
assert.Error(t, err)
_, err := svc.Redeem(shop.ID, "JIUKU-DEVL-IMIT", "new-dev", "New PC", "windows")
assert.Equal(t, ErrDeviceLimitExceed, err)
assert.Nil(t, result)
}
func TestLicenseService_Activate_NotFound(t *testing.T) {
db := testutil.SetupTestDB()
shop := testutil.CreateTestShop(db, "LIC004")
svc := NewLicenseService(db)
result, err := svc.Activate(shop.ID, "NONEX-ISTEN-TTTTT-LICCC", "device-001", "", "")
assert.Error(t, err)
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)
// 码仍未使用(回滚)
var lc model.LicenseCode
require.NoError(t, db.Where("code = ?", "JIUKUDEVLIMIT").First(&lc).Error)
assert.Equal(t, "unused", lc.Status)
// 到期不变
var after model.License
require.NoError(t, db.First(&after, lic.ID).Error)
assert.InDelta(t, 10, daysFromNow(after.ExpiresAt), 1)
}
func TestLicenseService_Verify_Success(t *testing.T) {
@@ -229,17 +259,3 @@ func TestLicenseService_Verify_NoExpiry(t *testing.T) {
require.NoError(t, err)
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]))
}
-106
View File
@@ -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)
}
-110
View File
@@ -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)
}
+40
View File
@@ -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-XXXX8 位随机段)。
// 使用 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()
}
+94
View File
@@ -0,0 +1,94 @@
package util
import (
"strings"
"testing"
)
func TestNormalizeCode(t *testing.T) {
cases := []struct {
name string
in string
want string
}{
{"展示格式带连字符", "JIUKU-YD9S-7CAP", "JIUKUYD9S7CAP"},
{"小写转大写", "jiuku-yd9s-7cap", "JIUKUYD9S7CAP"},
{"去除空格", "JIUKU YD9S 7CAP", "JIUKUYD9S7CAP"},
{"去除制表/换行/回车", "JIUKU\tYD9S\n7CAP\r", "JIUKUYD9S7CAP"},
{"已归一化幂等", "JIUKUYD9S7CAP", "JIUKUYD9S7CAP"},
{"空串", "", ""},
{"混合大小写与分隔", " jiUKu--yd9s 7caP\n", "JIUKUYD9S7CAP"},
}
for _, c := range cases {
t.Run(c.name, func(t *testing.T) {
if got := NormalizeCode(c.in); got != c.want {
t.Fatalf("NormalizeCode(%q) = %q, want %q", c.in, got, c.want)
}
})
}
}
func TestNormalizeCodeIdempotent(t *testing.T) {
// 二次归一化结果不变
once := NormalizeCode("jiuku-yd9s-7cap")
if twice := NormalizeCode(once); twice != once {
t.Fatalf("NormalizeCode 非幂等: %q -> %q", once, twice)
}
}
func TestGenerateRedeemCodeFormat(t *testing.T) {
for i := 0; i < 200; i++ {
code := GenerateRedeemCode()
// 展示格式:JIUKU-XXXX-XXXX
parts := strings.Split(code, "-")
if len(parts) != 3 {
t.Fatalf("码段数应为 3,得到 %d%q", len(parts), code)
}
if parts[0] != "JIUKU" {
t.Fatalf("前缀应为 JIUKU,得到 %q%q", parts[0], code)
}
if len(parts[1]) != 4 || len(parts[2]) != 4 {
t.Fatalf("随机段长度应各为 4,得到 %d/%d%q", len(parts[1]), len(parts[2]), code)
}
// 归一化后 = 前缀 + 8 位随机
norm := NormalizeCode(code)
if len(norm) != len("JIUKU")+8 {
t.Fatalf("归一化长度应为 %d,得到 %d%q", len("JIUKU")+8, len(norm), norm)
}
// 随机段只含无歧义字母表字符(不含 0/O/1/I/L)
random := parts[1] + parts[2]
for _, r := range random {
if !strings.ContainsRune(codeAlphabet, r) {
t.Fatalf("出现字母表外字符 %q%q", r, code)
}
}
}
}
func TestCodeAlphabetExcludesAmbiguous(t *testing.T) {
for _, bad := range []rune{'0', 'O', '1', 'I', 'L'} {
if strings.ContainsRune(codeAlphabet, bad) {
t.Fatalf("字母表不应包含易混字符 %q", bad)
}
}
}
func TestGenerateRedeemCodeUnique(t *testing.T) {
// 不要求绝对唯一,但大量生成应几乎无碰撞(31^8 空间)
const n = 1000
seen := make(map[string]struct{}, n)
dup := 0
for i := 0; i < n; i++ {
c := NormalizeCode(GenerateRedeemCode())
if _, ok := seen[c]; ok {
dup++
}
seen[c] = struct{}{}
}
if dup > 0 {
t.Fatalf("%d 次生成出现 %d 次重复,随机性可疑", n, dup)
}
}
+8 -4
View File
@@ -24,9 +24,6 @@ func main() {
if config.C.Server.CORSOrigin == "*" {
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")
}
}
// 初始化数据库
@@ -46,6 +43,13 @@ func main() {
r := gin.New()
r.Use(gin.Logger(), gin.Recovery())
// 真实客户端 IP:只信任本机 nginx127.0.0.1/::1)写的 X-Real-IP,丢弃客户端伪造的
// X-Forwarded-For。这样 c.ClientIP() 返回不可伪造的真实 IP,是所有按 IP 限流/审计的基础。
// 代理链:client → nginx(127.0.0.1:8445) → 后端(127.0.0.1:8080);将来若在 nginx 前再加
// 一层代理,需把其地址并入下列可信网段。
_ = r.SetTrustedProxies([]string{"127.0.0.1", "::1"})
r.RemoteIPHeaders = []string{"X-Real-IP"}
// CORS
corsOrigin := config.C.Server.CORSOrigin
r.Use(func(c *gin.Context) {
@@ -101,6 +105,7 @@ func autoMigrate(db *gorm.DB) {
&model.User{},
&model.License{},
&model.LicenseDevice{},
&model.LicenseCode{},
&model.UserSession{},
&model.LoginAttempt{},
&model.ProductCategory{},
@@ -133,4 +138,3 @@ func autoMigrate(db *gorm.DB) {
}
log.Println("AutoMigrate completed")
}
+23
View File
@@ -107,6 +107,7 @@ CREATE TABLE IF NOT EXISTS `licenses` (
`license_key` VARCHAR(768) NOT NULL COMMENT 'Ed25519 signed token',
`device_id` VARCHAR(255) DEFAULT NULL COMMENT 'deprecated: use license_devices',
`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=永久',
`is_active` TINYINT(1) NOT NULL DEFAULT 1,
`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`)
) 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='兑换券码池';
-- ------------------------------------------------------------
-- 商品分类
-- ------------------------------------------------------------
+25 -8
View File
@@ -27,16 +27,16 @@ func InitConfig() {
AccessExpireMin: 60,
RefreshExpireH: 168,
},
License: config.LicenseConfig{
HMACSecret: "test-license-hmac-secret",
},
Session: config.SessionConfig{
LimitDesktop: 2,
LimitMobile: 2,
LimitWeb: 2,
MaxFailures: 5,
LockMinutes: 15,
LimitDesktop: 2,
LimitMobile: 2,
LimitWeb: 2,
MaxFailures: 5,
IPMaxFailures: 20,
LockMinutes: 15,
},
// 测试默认关闭限流,避免压测式用例触发 429;限流逻辑由 ratelimit_test.go 显式开启验证。
RateLimit: config.RateLimitConfig{Enabled: false},
}
}
@@ -125,12 +125,29 @@ func SetupTestDB() *gorm.DB {
license_key TEXT UNIQUE,
device_id TEXT,
type TEXT DEFAULT 'trial',
tier TEXT DEFAULT 'standard',
expires_at DATETIME,
is_active INTEGER DEFAULT 1,
max_devices INTEGER DEFAULT 3,
features TEXT,
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 (
id INTEGER PRIMARY KEY AUTOINCREMENT,
license_id INTEGER NOT NULL,
Binary file not shown.

Before

Width:  |  Height:  |  Size: 544 B

After

Width:  |  Height:  |  Size: 2.4 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 442 B

After

Width:  |  Height:  |  Size: 1.7 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 721 B

After

Width:  |  Height:  |  Size: 3.0 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.0 KiB

After

Width:  |  Height:  |  Size: 4.6 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.4 KiB

After

Width:  |  Height:  |  Size: 5.8 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 16 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 11 KiB

After

Width:  |  Height:  |  Size: 15 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 295 B

After

Width:  |  Height:  |  Size: 495 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 406 B

After

Width:  |  Height:  |  Size: 895 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 450 B

After

Width:  |  Height:  |  Size: 1.3 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 282 B

After

Width:  |  Height:  |  Size: 686 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 462 B

After

Width:  |  Height:  |  Size: 1.3 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 704 B

After

Width:  |  Height:  |  Size: 1.7 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 406 B

After

Width:  |  Height:  |  Size: 895 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 586 B

After

Width:  |  Height:  |  Size: 1.6 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 862 B

After

Width:  |  Height:  |  Size: 2.1 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 862 B

After

Width:  |  Height:  |  Size: 2.1 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.6 KiB

After

Width:  |  Height:  |  Size: 2.9 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 762 B

After

Width:  |  Height:  |  Size: 1.5 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.2 KiB

After

Width:  |  Height:  |  Size: 2.7 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.4 KiB

After

Width:  |  Height:  |  Size: 2.8 KiB

+140 -39
View File
@@ -1,6 +1,7 @@
import 'dart:async';
import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import '../../core/responsive/responsive.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:go_router/go_router.dart';
@@ -34,6 +35,16 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
bool _obscure = true;
String? _errorMessage;
// 客服邮箱(授权续费 / 锁定协助)。与官网 web/_data/site.json 的 support.email 保持一致。
static const _supportEmail = 'yammy2023@163.com';
/// 当前错误是否为「授权锁定」——后端返回固定文案
/// `license locked, please renew or contact support`(见 service.ErrLicenseLocked)。
bool get _isLicenseLocked {
final m = _errorMessage?.toLowerCase() ?? '';
return m.contains('license') && (m.contains('lock') || m.contains('expire'));
}
List<String> _hotelCodeHistory = [];
List<String> _usernameHistory = [];
@@ -242,6 +253,94 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
}
}
Future<void> _openRenewSite() async {
await launchUrl(
Uri.parse(AppConfig.publicBaseUrl),
mode: LaunchMode.externalApplication,
);
}
Future<void> _copySupportEmail() async {
await Clipboard.setData(const ClipboardData(text: _supportEmail));
if (!mounted) return;
ScaffoldMessenger.of(context).showSnackBar(
const SnackBar(content: Text('已复制客服邮箱:$_supportEmail')),
);
}
/// 授权锁定时的可操作提示:告诉用户「去哪里解决」——前往官网续费 / 联系客服。
Widget _buildLicenseLockedNotice() {
return Container(
width: double.infinity,
padding: const EdgeInsets.all(14),
decoration: BoxDecoration(
color: AppTheme.danger.withAlpha(15),
borderRadius: BorderRadius.circular(6),
border: Border.all(color: AppTheme.danger.withAlpha(80)),
),
child: Column(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
const Row(
children: [
Icon(Icons.lock_clock_outlined,
color: AppTheme.danger, size: 18),
SizedBox(width: 8),
Expanded(
child: Text(
'授权已过期,账号已锁定',
style: TextStyle(
color: AppTheme.danger,
fontSize: 14,
fontWeight: FontWeight.w600),
),
),
],
),
const SizedBox(height: 6),
const Text(
'当前门店授权已到期,暂时无法登录。请续费授权后重试,或联系客服协助开通。',
style: TextStyle(
color: AppTheme.textSecondary, fontSize: 13, height: 1.4),
),
const SizedBox(height: 12),
Wrap(
spacing: 8,
runSpacing: 8,
children: [
ElevatedButton.icon(
onPressed: _openRenewSite,
icon: const Icon(Icons.open_in_new, size: 16),
label: const Text('前往官网续费'),
style: ElevatedButton.styleFrom(
backgroundColor: AppTheme.danger,
foregroundColor: Colors.white,
padding: const EdgeInsets.symmetric(
horizontal: 14, vertical: 10),
shape: RoundedRectangleBorder(
borderRadius: BorderRadius.circular(4)),
),
),
OutlinedButton.icon(
onPressed: _copySupportEmail,
icon: const Icon(Icons.support_agent_outlined, size: 16),
label: const Text('复制客服邮箱'),
style: OutlinedButton.styleFrom(
foregroundColor: AppTheme.danger,
side: BorderSide(color: AppTheme.danger.withAlpha(120)),
padding: const EdgeInsets.symmetric(
horizontal: 14, vertical: 10),
shape: RoundedRectangleBorder(
borderRadius: BorderRadius.circular(4)),
),
),
],
),
],
),
);
}
@override
Widget build(BuildContext context) {
// 被踢下线 / 会话失效:登出后跳回登录页,弹一次提示并清空
@@ -283,20 +382,12 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
child: Column(
mainAxisSize: MainAxisSize.min,
children: [
// Logo
// Logo(岩美酒库品牌 mark,与 App 图标一致)
Container(
width: 80,
height: 80,
decoration: BoxDecoration(
gradient: const LinearGradient(
colors: [
AppTheme.primaryLight,
AppTheme.primaryDark
],
begin: Alignment.topLeft,
end: Alignment.bottomRight,
),
borderRadius: BorderRadius.circular(12),
borderRadius: BorderRadius.circular(15),
boxShadow: [
BoxShadow(
color: AppTheme.primary.withAlpha(102),
@@ -305,12 +396,19 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
),
],
),
child: const Icon(Icons.wine_bar,
color: Colors.white, size: 44),
child: ClipRRect(
borderRadius: BorderRadius.circular(15),
child: Image.asset(
'assets/brand/logo_mark.png',
width: 80,
height: 80,
fit: BoxFit.cover,
),
),
),
const SizedBox(height: 20),
const Text(
'酒库管理系统',
'岩美酒库',
style: TextStyle(
fontSize: 24,
fontWeight: FontWeight.w700,
@@ -320,7 +418,7 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
),
const SizedBox(height: 6),
const Text(
'酒店仓库管理解决方案',
'专业酒水仓储 · 进销存一体化管理',
style: TextStyle(
fontSize: 13,
color: AppTheme.textSecondary),
@@ -448,32 +546,35 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
// Inline error
if (_errorMessage != null) ...[
const SizedBox(height: 12),
Container(
width: double.infinity,
padding: const EdgeInsets.symmetric(
horizontal: 12, vertical: 10),
decoration: BoxDecoration(
color: AppTheme.danger.withAlpha(15),
borderRadius: BorderRadius.circular(4),
border: Border.all(
color: AppTheme.danger.withAlpha(80)),
),
child: Row(
children: [
const Icon(Icons.error_outline,
color: AppTheme.danger, size: 16),
const SizedBox(width: 8),
Expanded(
child: Text(
_errorMessage!,
style: const TextStyle(
color: AppTheme.danger,
fontSize: 13),
if (_isLicenseLocked)
_buildLicenseLockedNotice()
else
Container(
width: double.infinity,
padding: const EdgeInsets.symmetric(
horizontal: 12, vertical: 10),
decoration: BoxDecoration(
color: AppTheme.danger.withAlpha(15),
borderRadius: BorderRadius.circular(4),
border: Border.all(
color: AppTheme.danger.withAlpha(80)),
),
child: Row(
children: [
const Icon(Icons.error_outline,
color: AppTheme.danger, size: 16),
const SizedBox(width: 8),
Expanded(
child: Text(
_errorMessage!,
style: const TextStyle(
color: AppTheme.danger,
fontSize: 13),
),
),
),
],
],
),
),
),
],
const SizedBox(height: 24),
@@ -530,7 +631,7 @@ class _LoginScreenState extends ConsumerState<LoginScreen> {
right: 0,
child: Center(
child: Text(
'© 2026 酒库管理系统 v1.0.0',
'© 2026 岩美酒库 v1.0.0',
style: TextStyle(
color: Colors.white.withAlpha(102), fontSize: 12),
),
@@ -455,11 +455,11 @@ class _SettingsScreenState extends ConsumerState<SettingsScreen> {
child: Column(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
const Text('激活授权',
const Text('兑换激活码',
style:
TextStyle(fontSize: 14, fontWeight: FontWeight.w600)),
const SizedBox(height: 8),
const Text('输入从官方渠道获得的授权码以激活或续期',
const Text('输入购买或活动获得的激活码,时长将叠加到当前授权之后',
style: TextStyle(
fontSize: 13, color: AppTheme.textSecondary)),
const SizedBox(height: 12),
@@ -469,7 +469,7 @@ class _SettingsScreenState extends ConsumerState<SettingsScreen> {
child: TextField(
controller: _licenseKeyCtrl,
decoration: const InputDecoration(
hintText: 'ey... 授权码',
hintText: 'JIUKU-XXXX-XXXX',
isDense: true,
border: OutlineInputBorder(),
contentPadding:
@@ -487,7 +487,7 @@ class _SettingsScreenState extends ConsumerState<SettingsScreen> {
width: 16,
height: 16,
child: CircularProgressIndicator(strokeWidth: 2))
: const Text('激活'),
: const Text('兑换'),
),
],
],
@@ -514,7 +514,7 @@ class _SettingsScreenState extends ConsumerState<SettingsScreen> {
if (mounted) {
ScaffoldMessenger.of(context).showSnackBar(
const SnackBar(
content: Text('授权激活成功'), backgroundColor: AppTheme.success),
content: Text('兑换成功,授权已更新'), backgroundColor: AppTheme.success),
);
}
} catch (e) {
Binary file not shown.

Before

Width:  |  Height:  |  Size: 101 KiB

After

Width:  |  Height:  |  Size: 34 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 5.5 KiB

After

Width:  |  Height:  |  Size: 4.0 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 520 B

After

Width:  |  Height:  |  Size: 669 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 14 KiB

After

Width:  |  Height:  |  Size: 8.0 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 1.0 KiB

After

Width:  |  Height:  |  Size: 1.3 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 36 KiB

After

Width:  |  Height:  |  Size: 16 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 2.2 KiB

After

Width:  |  Height:  |  Size: 2.1 KiB

+1
View File
@@ -44,3 +44,4 @@ flutter:
assets:
- assets/fonts/NotoSansSC-Regular.ttf
- assets/config/app_info.json
- assets/brand/logo_mark.png
Binary file not shown.

Before

Width:  |  Height:  |  Size: 917 B

After

Width:  |  Height:  |  Size: 669 B

Binary file not shown.

Before

Width:  |  Height:  |  Size: 5.2 KiB

After

Width:  |  Height:  |  Size: 5.8 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 8.1 KiB

After

Width:  |  Height:  |  Size: 16 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 5.5 KiB

After

Width:  |  Height:  |  Size: 3.1 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 20 KiB

After

Width:  |  Height:  |  Size: 7.8 KiB

+30
View File
@@ -0,0 +1,30 @@
#!/usr/bin/env bash
# jiu-gencode.sh — 平台方在 EC2 上批量生成兑换券(激活码)的便捷封装。
#
# 自动读取 /opt/jiu/config/production.env 的数据库连接,调用同目录的
# jiu-gencode 二进制写入码池 license_codes。随 server 发版部署到
# /opt/jiu/backend/,无需再本地交叉编译 + scp。
#
# 远程一行用法(在你的 Mac 上):
# ssh -i ~/.ssh/wangjia.pem ec2-user@18.136.60.128 \
# '/opt/jiu/backend/jiu-gencode.sh -type annual -days 365 -count 100 -batch 2026-summer'
# ssh -i ~/.ssh/wangjia.pem ec2-user@18.136.60.128 \
# '/opt/jiu/backend/jiu-gencode.sh -type trial -days 7 -count 1'
#
# 参数透传给 jiu-gencode-type trial|monthly|annual|lifetime -days N(0=永久)
# -devices N(0=不改门店现值) -count N -batch <批次> -note <备注>
set -euo pipefail
ENV_FILE="${JIU_ENV_FILE:-/opt/jiu/config/production.env}"
BIN="$(dirname "$0")/jiu-gencode"
[ -f "$ENV_FILE" ] || { echo "jiu-gencode: 找不到环境文件 $ENV_FILE" >&2; exit 1; }
[ -x "$BIN" ] || { echo "jiu-gencode: 找不到可执行文件 $BIN" >&2; exit 1; }
# 加载 production.env:跳过注释/空行,逐行 export(不回显,避免泄漏密钥)。
while IFS= read -r line; do
case "$line" in \#*|'') continue ;; esac
export "$line"
done < "$ENV_FILE"
exec "$BIN" "$@"
+18
View File
@@ -1,3 +1,9 @@
# 粗粒度限流区(按客户端 IP):仅作未鉴权公开接口的最外层泄洪,阈值远高于正常使用,
# 真正的多维限流在后端 Go 中间件。本文件以 conf.d 形式 include 进 http{} 上下文,
# 故 limit_req_zone 放在 server{} 之外。10m 约可容纳 16 万个 IP 的状态。
limit_req_zone $binary_remote_addr zone=jiu_pub:10m rate=10r/s;
limit_req_status 429;
server {
listen 127.0.0.1:8445 ssl;
server_name jiu.51yanmei.com;
@@ -24,6 +30,16 @@ server {
proxy_read_timeout 300s;
}
# 未鉴权公开/登录接口:加最外层 per-IP 限流(鉴权后的业务 API 不在此限,交后端按店限流)。
# 须定义在通用 /api 正则之前(nginx 正则 location 按书写顺序首个命中者生效)。
location ~ ^/api/v1/(public|auth)/ {
limit_req zone=jiu_pub burst=20 nodelay;
proxy_pass http://127.0.0.1:8080;
proxy_set_header Host $host;
proxy_set_header X-Real-IP $remote_addr;
proxy_read_timeout 30s;
}
# API 反向代理
location ~ ^/(api|health|version) {
proxy_pass http://127.0.0.1:8080;
@@ -49,6 +65,7 @@ server {
# 公开商品详情页(扫码跳转)→ 后端注入 OG 标签后返回 Flutter index.html
location ~ ^/product/ {
limit_req zone=jiu_pub burst=20 nodelay;
proxy_pass http://127.0.0.1:8080;
proxy_set_header Host $host;
proxy_set_header X-Real-IP $remote_addr;
@@ -58,6 +75,7 @@ server {
# Flutter 路由会把 /product/:id 重写为 /app/product/:idbase-href=/app/
# 分享此 URL 时微信爬虫也需要 OG 标签 → 去掉 /app 前缀后转发后端
location ~ ^/app/product/ {
limit_req zone=jiu_pub burst=20 nodelay;
rewrite ^/app(/product/.+)$ $1 break;
proxy_pass http://127.0.0.1:8080;
proxy_set_header Host $host;
-3
View File
@@ -5,9 +5,6 @@ DATABASE_DSN=root:CHANGE_ME_DB_PASS@tcp(127.0.0.1:3306)/jiu_db?charset=utf8mb4&p
JWT_SECRET=CHANGE_ME_RANDOM_32CHARS
JWT_ACCESS_EXPIRE_MIN=60
JWT_REFRESH_EXPIRE_H=168
LICENSE_HMAC_SECRET=CHANGE_ME_RANDOM_32CHARS
LICENSE_ED25519_PUBLIC_KEY=zFkRV0E+oMjHbixiBamf2ZAq+n7En2V7EJRIurgtVxo=
LICENSE_ED25519_PRIVATE_KEY=CHANGE_ME_ED25519_PRIVATE_FROM_BITWARDEN
STORAGE_UPLOAD_DIR=/opt/jiu/images
STORAGE_BASE_URL=https://jiu.51yanmei.com/images
STORAGE_PUBLIC_URL=https://jiu.51yanmei.com
+1 -1
View File
@@ -16,7 +16,7 @@ set -euo pipefail
# ---- 可配置项 ----
RBW="${RBW_BIN:-/opt/homebrew/bin/rbw}"
BW_ITEM="${JIU_BW_ITEM:-jiu db password}" # 存放 jiu 密钥字段的 Bitwarden 条目
SECRET_KEYS="DATABASE_DSN JWT_SECRET LICENSE_HMAC_SECRET LICENSE_ED25519_PRIVATE_KEY DB_PASSWORD"
SECRET_KEYS="DATABASE_DSN JWT_SECRET DB_PASSWORD"
EC2_HOST="${EC2_HOST:-18.136.60.128}"
EC2_USER="${EC2_USER:-ec2-user}"
+40 -1
View File
@@ -1,6 +1,45 @@
# 授权信息功能 — 完整方案设计
> 本文档覆盖「付费授权」功能的产品决策、数据模型、API 设计、安全方案、前端方案及分期实施计划。所有编码工作在本文档评审通过后按计划分批推进。
> ⚠️ **本文档第 111 节为 v1Ed25519 自包含签名 token 方案),已被 v2「时长兑换券」模型取代。**
> 实际线上实现以下方「## v2」为准;下方 v1 各节仅作历史保留,其中「激活时验签」「license_token 列」「AdminOnly 激活」等已不再适用。
---
## v2 实际实现(时长兑换券模型,2026-06 起)
### 为什么改
v1 把许可证做成「一个 Ed25519 签名 token = 一张完整授权」,但运行时从不验签(签名形同虚设),且 token 内 `expires_at` 是**绝对日期**、不可叠加,不符合「买一张码续 1 年、可买多张累加」的消费直觉。经产品方确认改为**时长兑换券**。
### 模型要点
- **时长券**:每张码代表一段**时长**(`duration_days`),兑换 = `新到期 = max(今天, 当前到期) + 时长`**可叠加**`duration_days=0` 表示永久(到期置 NULL)。
- **短码 + 服务端码表**:码形如 `JIUKU-7F3A-9K2D``util.GenerateRedeemCode`crypto/rand + 无歧义字母表),归一化(`util.NormalizeCode`)后存 `license_codes` 表。**一码一次**由服务端码表 `status` 强制(签名无法防重放,故 v2 退役 ed25519)。
- **档位钩子**`license_codes.tier` / `licenses.tier`(默认 `standard`),为将来 pro/max-like 分档预留;当前不据此做能力差异,分档消费模式后续设计。
- **签发归平台方**:码由 CLI `go run ./cmd/gencode -type annual -days 365 -count N` 批量生成落库,用户购买/活动获得后在 App「设置 → 授权」兑换。**不开放 App 内自助签发**(防店铺管理员自签终身授权)。
### 数据模型
- 新表 `license_codes`(码池):`code`(uniq) / `type` / `tier` / `duration_days` / `max_devices` / `status`(unused/redeemed/void) / `redeemed_shop_id` / `redeemed_at` / `redeemed_device_id` / `batch` / `note`。见 `model.LicenseCode` + `schema.sql` + `testutil`
- `licenses` 表新增 `tier` 列(additive)。`license_key` 列降级为「当前权益占位」:trial 用 `TRIAL-<uuid>`、无既有行兜底用 `REDEEM-<uuid>`,兑换扩展时不改原值。**v1 的 `license_token` 列未实现,已废弃。**
### 兑换流程(`LicenseService.Redeem`,路由仍 `POST /license/activate` 以兼容客户端)
单事务 + `FOR UPDATE`check-then-act 并发安全):① 锁码校验 unused(否则 `无效激活码/已被使用/已失效`)→ ② 锁本店最新 `is_active` 授权行(无则新建)→ ③ 叠加到期、写 `type/tier/max_devices=max(现值,码值)` → ④ 绑定本设备(幂等 + `max_devices` 上限,超限整笔回滚)→ ⑤ 标记码 `redeemed` → ⑥ 提交后 `InvalidateLicensePhase` 即时生效。
### 与 v1 的差异(务必注意)
| 维度 | v1(已废弃) | v2(现状) |
|------|------|------|
| 码形态 | Ed25519 246 字符 token | 短码 `JIUKU-XXXX-XXXX` + 服务端码表 |
| 到期语义 | token 内绝对日期 | 时长券叠加,DB `expires_at` 为准 |
| 验签 | 计划激活时验签(实际从未生效) | 退役 ed25519,改服务端码表查验 |
| 签发 | `cmd/issue` 打印 token(需手动 INSERT | `cmd/gencode` 批量生成并落库 |
| 激活权限 | 计划加 `AdminOnly` | 维持 `ReadOnly` 豁免、不加 AdminOnly(兑换是店主自助行为) |
| trial | Ed25519 签发 | 直接建行(无需私钥),`createTrialLicense` 去除 Fatal |
| 运行时降级 | 同 v2`LicenseGuard` 读 DB `expires_at` 实时算 phase30s 缓存) | **不变,沿用** |
> v1 第 7 节「授权状态机与拦截层」(`CalcLicensePhase` / `LicenseGuard` / grace/readonly/locked**已落地且 v2 沿用**,是两版共同的运行时基础。
---
+6 -1
View File
@@ -13,10 +13,14 @@ echo "==> compile-backend: tag=${TAG}"
# Build Go backend (linux/amd64 for EC2)
cd backend
GOOS=linux GOARCH=amd64 CGO_ENABLED=0 go build -o jiu-server .
# Platform code-minting CLI (gencode) ships alongside the server so codes can be
# minted on EC2 without cross-compiling/scp each time.
GOOS=linux GOARCH=amd64 CGO_ENABLED=0 go build -o jiu-gencode ./cmd/gencode
cd ..
mkdir -p dist
mv backend/jiu-server dist/jiu-server
mv backend/jiu-gencode dist/jiu-gencode
# Shared infrastructure (nginx/systemd/env/compose). version.yaml is NOT here —
# it belongs to the client pipeline.
@@ -26,7 +30,8 @@ tar -czf dist/configs.tar.gz \
deploy/production.env.template \
deploy/setup-ec2.sh \
deploy/docker-compose.yml \
deploy/docker-compose.jiu.yml
deploy/docker-compose.jiu.yml \
deploy/jiu-gencode.sh
echo "==> compile-backend: done — dist/ contents:"
ls -lh dist/
+18
View File
@@ -27,6 +27,13 @@ echo "==> deploy-server: uploading files to EC2"
${SCP} dist/jiu-server "${EC2_USER}@${EC2_HOST}:/tmp/jiu-server"
${SCP} /tmp/jiu-configs/deploy/nginx-jiu.conf "${EC2_USER}@${EC2_HOST}:/tmp/nginx-jiu.conf"
# Platform code-minting CLI + wrapper (absent when rolling back to a pre-gencode
# release — upload only when present).
if [ -f dist/jiu-gencode ]; then
${SCP} dist/jiu-gencode "${EC2_USER}@${EC2_HOST}:/tmp/jiu-gencode"
${SCP} /tmp/jiu-configs/deploy/jiu-gencode.sh "${EC2_USER}@${EC2_HOST}:/tmp/jiu-gencode.sh"
fi
${SSH} "${EC2_USER}@${EC2_HOST}" << 'ENDSSH'
set -e
@@ -35,6 +42,17 @@ sudo systemctl stop jiu
cp /tmp/jiu-server /opt/jiu/backend/jiu-server
chmod +x /opt/jiu/backend/jiu-server
# Refresh the code-minting CLI if shipped in this release (skip gracefully on
# rollback to a pre-gencode release). Not a service — just a host-side binary.
if [ -f /tmp/jiu-gencode ]; then
cp /tmp/jiu-gencode /opt/jiu/backend/jiu-gencode
chmod +x /opt/jiu/backend/jiu-gencode
if [ -f /tmp/jiu-gencode.sh ]; then
cp /tmp/jiu-gencode.sh /opt/jiu/backend/jiu-gencode.sh
chmod +x /opt/jiu/backend/jiu-gencode.sh
fi
fi
# Start and health check
sudo systemctl start jiu
echo "Waiting for health check..."
+2 -1
View File
@@ -1,6 +1,6 @@
#!/usr/bin/env bash
# release-server.sh <tag> — create the Forgejo Release for the backend pipeline
# and upload jiu-server + configs.tar.gz.
# and upload jiu-server + jiu-gencode + configs.tar.gz.
set -euo pipefail
# shellcheck source=scripts/ci/lib-forgejo.sh
@@ -20,6 +20,7 @@ PYEOF
create_release "$TAG" "$BODY"
upload_asset dist/jiu-server
upload_asset dist/jiu-gencode
upload_asset dist/configs.tar.gz
echo "==> release-server: done — Release ${TAG} created"
+82
View File
@@ -0,0 +1,82 @@
#!/usr/bin/env bash
# set-license-expiry.sh <门店编号> <带符号天数>
#
# 测试工具:把某门店「当前有效授权」(licenses.is_active=1) 的到期时间改成
# 「相对现在 N 天」,方便快速切换授权阶段验证客户端表现。
#
# 带符号天数 DAYS
# 30 -> 30 天后到期 → normal 正常(未过期)
# 0 -> 今天到期
# -6 -> 已过期 6 天 → grace 宽限期(仍可登录,有横幅)
# -10 -> 已过期 10 天 → readonly 只读(可登录,禁写)
# -20 -> 已过期 20 天 → locked 锁定(无法登录)
#
# 默认改「线上 prod」EC2 上 jiu_mysql 容器里的 jiu_db.licenses。
# 连接参数可用环境变量覆盖:
# EC2_HOST (默认 18.136.60.128)
# EC2_USER (默认 ec2-user)
# SSH_KEY (默认 ~/.ssh/wangjia.pem)
#
# 用法示例:
# sh scripts/set-license-expiry.sh S003 -6 # S003 → 已过期 6 天(grace
# sh scripts/set-license-expiry.sh S003 30 # S003 → 恢复,30 天后到期
set -euo pipefail
SHOP="${1:-}"
DAYS="${2:-}"
if [ -z "$SHOP" ] || [ -z "$DAYS" ]; then
echo "用法: $0 <门店编号> <带符号天数>" >&2
echo "示例: $0 S003 -6 # S003 授权改为已过期 6 天(grace 宽限期)" >&2
echo " $0 S003 30 # S003 授权恢复,30 天后到期" >&2
exit 1
fi
case "$DAYS" in
''|*[!0-9-]*) echo "错误: 天数必须是整数(可带负号),收到: '$DAYS'" >&2; exit 1;;
esac
EC2_HOST="${EC2_HOST:-18.136.60.128}"
EC2_USER="${EC2_USER:-ec2-user}"
SSH_KEY="${SSH_KEY:-$HOME/.ssh/wangjia.pem}"
echo "==> 目标(线上 prod): ${EC2_USER}@${EC2_HOST} 门店=${SHOP} 到期=现在 ${DAYS}"
# 全部在远端单次 ssh 内完成;DB 密码只在远端从 production.env 读取,绝不打印/外传。
ssh -o ConnectTimeout=10 -o StrictHostKeyChecking=accept-new -i "$SSH_KEY" \
"${EC2_USER}@${EC2_HOST}" bash -s -- "$SHOP" "$DAYS" <<'ENDSSH'
set -euo pipefail
SHOP="$1"; DAYS="$2"
sudo grep '^DB_PASSWORD=' /opt/jiu/config/production.env | cut -d= -f2- | {
read -r PW
MYSQL() { sudo docker exec -i jiu_mysql mysql --default-character-set=utf8mb4 -uroot -p"$PW" jiu_db "$@"; }
echo "--- 改前 ---"
MYSQL -e "SELECT s.code, s.name, l.id, l.type, l.is_active, l.expires_at,
TIMESTAMPDIFF(DAY, l.expires_at, NOW()) AS days_expired
FROM licenses l JOIN shops s ON s.id = l.shop_id
WHERE s.code = '${SHOP}' AND l.is_active = 1 ORDER BY l.id DESC;"
MYSQL -e "UPDATE licenses SET expires_at = DATE_ADD(NOW(), INTERVAL ${DAYS} DAY)
WHERE shop_id = (SELECT id FROM shops WHERE code = '${SHOP}' LIMIT 1)
AND is_active = 1;"
echo "--- 改后 ---"
MYSQL -e "SELECT s.code, l.id, l.expires_at,
TIMESTAMPDIFF(DAY, l.expires_at, NOW()) AS days_expired
FROM licenses l JOIN shops s ON s.id = l.shop_id
WHERE s.code = '${SHOP}' AND l.is_active = 1 ORDER BY l.id DESC;"
}
ENDSSH
# 阶段提示(与后端 middleware.CalcLicensePhase 边界一致:grace<=7天, readonly<=15天, locked>15天)
if [ "$DAYS" -ge 0 ]; then
PHASE="normal 正常(未过期)"
else
EXP=$(( -DAYS ))
if [ "$EXP" -le 7 ]; then PHASE="grace 宽限期(仍可登录,有横幅提醒)"
elif [ "$EXP" -le 15 ]; then PHASE="readonly 只读(可登录,所有写操作被禁)"
else PHASE="locked 锁定(无法登录,登录页给出续费入口)"
fi
fi
echo "==> 完成。预计阶段: ${PHASE}"
echo "==> 提示: 后端 phase 有约 30s 的每店缓存;已登录的客户端会话需等 ~30s 或重新登录后生效。"