@xiaobuyu/nodesign 0.0.1 → 0.0.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +156 -106
- package/bin/nodesign.js +0 -0
- package/package.json +1 -1
- package/server/api/README.md +45 -45
- package/server/api/_guard.js +75 -75
- package/server/api/admin.js +187 -187
- package/server/api/assets/docx-page.js +103 -103
- package/server/api/assets/helpers.js +22 -22
- package/server/api/assets/notes.js +144 -144
- package/server/api/assets.js +896 -896
- package/server/api/board.js +57 -57
- package/server/api/browse.js +138 -138
- package/server/api/canvas.js +444 -444
- package/server/api/chatai.js +324 -324
- package/server/api/exports/build-standalone.js +980 -980
- package/server/api/exports/cards.js +118 -118
- package/server/api/exports/docx-pdf.js +46 -46
- package/server/api/exports/export-page.js +142 -142
- package/server/api/exports/handoff.js +147 -147
- package/server/api/exports.js +690 -690
- package/server/api/instruction.js +75 -75
- package/server/api/me.js +116 -116
- package/server/api/memory.js +185 -185
- package/server/api/pending-changes.js +366 -366
- package/server/api/plugins.js +164 -164
- package/server/api/projects.js +220 -220
- package/server/api/publish.js +64 -64
- package/server/api/recent.js +73 -73
- package/server/api/sessions.js +589 -589
- package/server/api/skills.js +77 -77
- package/server/api/standalone-fit.js +168 -168
- package/server/api/turn-compose.js +180 -180
- package/server/api/turn-inflight.js +44 -44
- package/server/auth/README.md +17 -17
- package/server/auth/middleware.js +138 -138
- package/server/auth/origin-guard.js +71 -71
- package/server/auth/session.js +108 -108
- package/server/auth/tier.js +113 -113
- package/server/auth/users-store.js +348 -348
- package/server/edit/README.md +29 -29
- package/server/engine/README.md +113 -113
- package/server/engine/agent/agent-shared.js +550 -550
- package/server/engine/agent/auto-mode-default-hard-deny.txt +4 -4
- package/server/engine/agent/auto-mode-rules.js +78 -78
- package/server/engine/agent/context.js +418 -418
- package/server/engine/agent/events.js +464 -464
- package/server/engine/agent/hooks/canvas-validate.js +199 -199
- package/server/engine/agent/hooks/failure.js +120 -120
- package/server/engine/agent/hooks/file-events.js +75 -75
- package/server/engine/agent/hooks/lifecycle.js +233 -233
- package/server/engine/agent/hooks/post-canvas-focus.js +80 -80
- package/server/engine/agent/hooks/post-guidance.js +206 -206
- package/server/engine/agent/hooks/post-subagent-report.js +104 -104
- package/server/engine/agent/hooks/post-trim.js +58 -58
- package/server/engine/agent/hooks/pre-board-neighborhood.js +34 -34
- package/server/engine/agent/hooks/pre-defaults.js +84 -84
- package/server/engine/agent/hooks/pre-injectors.js +277 -277
- package/server/engine/agent/hooks/pre-performance-log-guard.js +78 -78
- package/server/engine/agent/hooks/pre-starter-files.js +61 -61
- package/server/engine/agent/hooks/pre-workspace-scope-guard.js +78 -78
- package/server/engine/agent/hooks/site-validate.js +94 -94
- package/server/engine/agent/hooks/tool-prompts.js +31 -31
- package/server/engine/agent/hooks/turn-state-memory.js +56 -56
- package/server/engine/agent/hooks/user-prompt-submit.js +215 -215
- package/server/engine/agent/hooks.js +354 -354
- package/server/engine/agent/init-contract.js +69 -69
- package/server/engine/agent/isolation.js +159 -159
- package/server/engine/agent/model-table.js +292 -292
- package/server/engine/agent/plugin-loader.js +266 -266
- package/server/engine/agent/prompts/nodesign-prelude.md +336 -336
- package/server/engine/agent/prompts/tools/ask-user-question-protocol.md +71 -71
- package/server/engine/agent/prompts/tools/direct-edit-protocol.md +98 -98
- package/server/engine/agent/prompts/tools/generate-image-cookbook.gemini-gateway.md +583 -583
- package/server/engine/agent/prompts/tools/generate-image-cookbook.md +483 -483
- package/server/engine/agent/prompts/tools/hybrid-reference.md +100 -100
- package/server/engine/agent/prompts/tools/paint-still-cookbook.md +270 -270
- package/server/engine/agent/prompts/tools/roll-film-cookbook.md +70 -70
- package/server/engine/agent/prompts/tools/site-reference.md +106 -106
- package/server/engine/agent/prompts/tools/tweaks-syntax.md +125 -125
- package/server/engine/agent/prompts/tools/vision-checker-dispatch.md +48 -48
- package/server/engine/agent/session-loop.js +1024 -1024
- package/server/engine/agent/session-model.js +164 -164
- package/server/engine/agent/skill.js +204 -204
- package/server/engine/agent/task-events.js +78 -78
- package/server/engine/agents/ds-extractor.md +132 -132
- package/server/engine/agents/explorer.md +190 -190
- package/server/engine/agents/index.js +230 -230
- package/server/engine/agents/schemas/design-system.json +156 -156
- package/server/engine/agents/schemas/tweak-schema.json +113 -113
- package/server/engine/agents/tweak-proposer.md +127 -127
- package/server/engine/agents/vision-checker.md +254 -254
- package/server/engine/browse/capture.js +422 -422
- package/server/engine/browse/card.js +137 -137
- package/server/engine/browse/handover.js +59 -59
- package/server/engine/browse/page-digest.js +119 -119
- package/server/engine/browse/refs.js +196 -196
- package/server/engine/browse/registry.js +307 -307
- package/server/engine/browse/screencast.js +170 -170
- package/server/engine/browse/state.js +121 -121
- package/server/engine/chatai/chat-log.js +79 -79
- package/server/engine/chatai/index.js +193 -193
- package/server/engine/chatai/openai-compat.js +152 -152
- package/server/engine/chatai/orchestrate.js +309 -309
- package/server/engine/chatai/perform.js +40 -40
- package/server/engine/chatai/summarize.js +79 -79
- package/server/engine/mcp/capability-gate.js +52 -52
- package/server/engine/mcp/index.js +339 -339
- package/server/engine/mcp/param-sanitizer.js +142 -142
- package/server/engine/mcp/tools/arrange-on-board.js +124 -124
- package/server/engine/mcp/tools/artifact-session.js +278 -278
- package/server/engine/mcp/tools/browse-computer.js +421 -421
- package/server/engine/mcp/tools/browse-find-batch.js +183 -183
- package/server/engine/mcp/tools/browse-screenshot.js +157 -157
- package/server/engine/mcp/tools/browse.js +480 -480
- package/server/engine/mcp/tools/build-docx.js +101 -101
- package/server/engine/mcp/tools/clear-pending-changes.js +99 -99
- package/server/engine/mcp/tools/create-on-board.js +155 -155
- package/server/engine/mcp/tools/crystallize-skill.js +168 -168
- package/server/engine/mcp/tools/explain-style.js +237 -237
- package/server/engine/mcp/tools/export-handoff.js +133 -133
- package/server/engine/mcp/tools/expose-tweaks.js +149 -149
- package/server/engine/mcp/tools/generate-image.js +842 -842
- package/server/engine/mcp/tools/get-computed-styles.js +164 -164
- package/server/engine/mcp/tools/get-pending-changes.js +205 -205
- package/server/engine/mcp/tools/h3box-ssh.js +84 -84
- package/server/engine/mcp/tools/helpers/acquire-page.js +82 -82
- package/server/engine/mcp/tools/helpers/motion-lab.js +554 -554
- package/server/engine/mcp/tools/helpers/motion-scroll.js +112 -112
- package/server/engine/mcp/tools/helpers/perception-page.js +215 -215
- package/server/engine/mcp/tools/helpers/reference-download.js +103 -103
- package/server/engine/mcp/tools/helpers/rembg-bridge.py +80 -80
- package/server/engine/mcp/tools/helpers/shot-pipeline.js +280 -280
- package/server/engine/mcp/tools/highlight.js +68 -68
- package/server/engine/mcp/tools/list-pages.js +222 -222
- package/server/engine/mcp/tools/lookup-tags.js +59 -59
- package/server/engine/mcp/tools/navigate-to-page.js +60 -60
- package/server/engine/mcp/tools/organize-board.js +64 -64
- package/server/engine/mcp/tools/paint-still.js +434 -434
- package/server/engine/mcp/tools/pin-to-board.js +130 -130
- package/server/engine/mcp/tools/preview-deck.js +130 -130
- package/server/engine/mcp/tools/profile-scroll.js +305 -305
- package/server/engine/mcp/tools/publish-site.js +102 -102
- package/server/engine/mcp/tools/query-elements.js +195 -195
- package/server/engine/mcp/tools/read-board.js +103 -103
- package/server/engine/mcp/tools/read-document.js +93 -93
- package/server/engine/mcp/tools/read-page.js +264 -264
- package/server/engine/mcp/tools/read-tavern-json.js +143 -143
- package/server/engine/mcp/tools/record-decision.js +140 -140
- package/server/engine/mcp/tools/relate-on-board.js +110 -110
- package/server/engine/mcp/tools/report-issue.js +115 -115
- package/server/engine/mcp/tools/roll-film.js +266 -266
- package/server/engine/mcp/tools/screenshot-docx.js +138 -138
- package/server/engine/mcp/tools/screenshot-url.js +195 -195
- package/server/engine/mcp/tools/screenshot.js +583 -583
- package/server/engine/mcp/tools/tier-gate.js +96 -96
- package/server/engine/mcp/tools/trace-motion.js +215 -215
- package/server/engine/mcp/tools/web-search.js +548 -548
- package/server/engine/motion/inventory.js +349 -349
- package/server/engine/perception/session.js +235 -235
- package/server/engine/plugins/nodesign/.claude-plugin/plugin.json +5 -5
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/SKILL.md +192 -192
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/canvas.template.html +667 -667
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/hybrid-grid.md +22 -22
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/image-led-cover.md +25 -25
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/portrait.md +23 -23
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/quote-backdrop.md +21 -21
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/section-divider.md +23 -23
- package/server/engine/plugins/nodesign/skills/deskskill-engine-mini/patterns/text-led.md +21 -21
- package/server/engine/plugins/nodesign/skills/docx-craft/SKILL.md +162 -162
- package/server/engine/plugins/nodesign/skills/docx-craft/references/token-schema.md +407 -407
- package/server/engine/plugins/nodesign/skills/docx-craft//346/226/207/346/241/243.template.json +80 -80
- package/server/engine/plugins/nodesign/skills/rp-craft/SKILL.md +228 -228
- package/server/engine/plugins/nodesign/skills/rp-craft/patterns//346/274/224/345/207/272/351/241/265-/346/234/200/345/260/217/345/256/236/347/216/260.html +78 -78
- package/server/engine/plugins/nodesign/skills/rp-craft//346/274/224/345/207/272.template.js +268 -268
- package/server/engine/plugins/nodesign/skills/site-craft/SKILL.md +323 -323
- package/server/engine/plugins/nodesign/skills/site-craft/patterns/build-lane.md +134 -134
- package/server/engine/plugins/nodesign/skills/site-craft/patterns/cutout-collage.md +61 -61
- package/server/engine/plugins/nodesign/skills/site-craft/patterns/mock-app.md +163 -163
- package/server/engine/plugins/nodesign/skills/site-craft/patterns/page-transitions.md +86 -86
- package/server/engine/runs/active-runs.js +698 -698
- package/server/engine/runs/live-turn.js +252 -252
- package/server/engine/runs/store.js +364 -364
- package/server/engine/runs/turn-relay.js +217 -217
- package/server/engine/runtime/workspace.js +164 -164
- package/server/lib/artifact-file-path.js +80 -80
- package/server/lib/artifact-target.js +284 -284
- package/server/lib/asset-refs.js +239 -239
- package/server/lib/async-queue.js +104 -104
- package/server/lib/auto-relations.js +130 -130
- package/server/lib/binding-types.js +60 -60
- package/server/lib/board-kind-sizes.js +62 -62
- package/server/lib/board-relations.js +133 -133
- package/server/lib/browse-proxy.js +180 -180
- package/server/lib/canvas-id.js +36 -36
- package/server/lib/cover.js +227 -227
- package/server/lib/danbooru-tags.js +291 -291
- package/server/lib/doc-extract.js +156 -156
- package/server/lib/docx/build-from-source.js +260 -260
- package/server/lib/docx/build.js +490 -490
- package/server/lib/docx/dump-styles.js +287 -287
- package/server/lib/docx/fonts/nodesign-cjk.conf +164 -164
- package/server/lib/docx/merge-runs.js +129 -129
- package/server/lib/docx/numbering.js +218 -218
- package/server/lib/docx/order.js +152 -152
- package/server/lib/docx/rawzip.js +172 -172
- package/server/lib/docx/render.js +92 -92
- package/server/lib/docx/text-lint.js +225 -225
- package/server/lib/docx/tokens.js +335 -335
- package/server/lib/docx/units.js +29 -29
- package/server/lib/docx/xml.js +291 -291
- package/server/lib/docx-pages.js +187 -187
- package/server/lib/export-collect.js +289 -289
- package/server/lib/export-package.js +210 -210
- package/server/lib/html-srcset.js +145 -145
- package/server/lib/image-variant.js +460 -460
- package/server/lib/ingress/forward-openai-chat.js +305 -305
- package/server/lib/ingress/openai-chat.js +491 -491
- package/server/lib/ingress/session-notice.js +66 -66
- package/server/lib/ingress/session-routes.js +68 -68
- package/server/lib/ingress/slot-probe.js +130 -130
- package/server/lib/ingress/upstream-billing.js +52 -52
- package/server/lib/ingress/upstream-fail-streak.js +72 -72
- package/server/lib/ingress/upstream-truncation.js +48 -48
- package/server/lib/issues-store.js +182 -182
- package/server/lib/kinds/deck.js +72 -72
- package/server/lib/kinds/docx.js +283 -283
- package/server/lib/kinds/file-kinds.js +113 -113
- package/server/lib/kinds/index.js +202 -202
- package/server/lib/kinds/site.js +161 -161
- package/server/lib/model-ingress.js +543 -543
- package/server/lib/moderation.js +255 -255
- package/server/lib/notice-store.js +103 -103
- package/server/lib/plugin-install.js +104 -104
- package/server/lib/plugin-validator.js +721 -721
- package/server/lib/publish-store.js +115 -115
- package/server/lib/quick-summary.js +52 -52
- package/server/lib/quota.js +270 -270
- package/server/lib/rate-window.js +29 -29
- package/server/lib/reference-assets.js +92 -92
- package/server/lib/region-shot.js +135 -135
- package/server/lib/safe-path.js +73 -73
- package/server/lib/sdk-session.js +33 -33
- package/server/lib/showcase-store.js +92 -92
- package/server/lib/site-publish.js +465 -465
- package/server/lib/ssrf-guard.js +432 -432
- package/server/lib/task-scan.js +158 -158
- package/server/lib/tavern-json.js +154 -154
- package/server/lib/video-variant.js +237 -237
- package/server/lib/workspace-path.js +33 -33
- package/server/ops/fix-sdk-musl.mjs +40 -40
- package/server/ops/install-macos-fonts.sh +114 -114
- package/server/ops/macos-fonts.conf +336 -336
- package/server/ops/sandbox-shim/bwrap +57 -57
- package/server/projects/assets-summary.js +119 -119
- package/server/projects/auto-name.js +53 -53
- package/server/projects/board-store.js +640 -640
- package/server/projects/move-entry.js +103 -103
- package/server/projects/store.js +270 -270
- package/server/projects/ui-config.js +54 -54
- package/server/projects/workspace-templates.js +53 -53
- package/server/projects/workspace.js +1025 -1025
- package/server/runtime/local-config.js +177 -177
- package/server/runtime/profile.js +72 -72
- package/server/services/rembg-service.py +334 -334
- package/server/shared/README.md +25 -25
- package/server/shared/deck.js +54 -54
- package/server/shared/time.js +49 -49
- package/server/style-pipeline/README.md +53 -53
- package/server/ws/broker.js +36 -36
- package/server/ws/browse-channel.js +177 -177
- package/server/ws/index.js +386 -386
- package/web/README.md +89 -89
- package/web/dist/assets/BrowserWindow-BYCgizHw.js +11 -0
- package/web/dist/assets/{DeckWindow-4G0xX7P3.js → DeckWindow-pNydEM4L.js} +1 -1
- package/web/dist/assets/{DocxWindow-BXQsoPkh.js → DocxWindow-PPgzh3Gq.js} +1 -1
- package/web/dist/assets/{SiteWindow-D6eS-2gM.js → SiteWindow-DM0LQPCO.js} +1 -1
- package/web/dist/assets/{index-Cbl3ATvp.js → index-BHTBJlpY.js} +32 -32
- package/web/dist/assets/{pending-edit-apply-DTjo2geW.js → pending-edit-apply-C0I13CbD.js} +1 -1
- package/web/dist/assets/{rotate-cw-LMT3x1Zj.js → rotate-cw-B1UQ8Hld.js} +1 -1
- package/web/dist/assets/{square-dashed-mouse-pointer-CBoAkfhi.js → square-dashed-mouse-pointer-8lrgqmfv.js} +1 -1
- package/web/dist/index.html +17 -17
- package/web/dist/assets/BrowserWindow-Dc7J7CYc.js +0 -11
package/server/lib/quota.js
CHANGED
|
@@ -1,270 +1,270 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* server/lib/quota.js — 每用户日用量限额(2026-07-30 内测;07-31 口径改成金额)
|
|
3
|
-
*
|
|
4
|
-
* # 为什么闸门数的是钱不是 token
|
|
5
|
-
*
|
|
6
|
-
* 原口径是 sum(input_tokens + output_tokens),缓存不计。实测这个口径量错了主项:
|
|
7
|
-
* 07-31 当天全站 input+output 是 317k,cache_read 是 83M,差 260 倍。也就是说
|
|
8
|
-
* 配额条显示"才用了 4%"的人,实际推给账号的负载可能已经是一整天的大头。
|
|
9
|
-
*
|
|
10
|
-
* 更要命的是**缓存过期**。同一轮对话,缓存全命中和全过期的 token 计数几乎一样,
|
|
11
|
-
* 但过期后那批 token 从 $0.30/M 的 cache_read 变成 $3/M 的 input,还要再花
|
|
12
|
-
* $6.00/M 重写一次缓存,金额能差十倍。token 口径对这件事完全瞎。
|
|
13
|
-
*
|
|
14
|
-
* SDK 的 `ModelUsage.costUSD` 是原生字段,CLI 内部按分模型价目表算好给出来的。
|
|
15
|
-
* 逆推验证精确到小数点后 7 位:
|
|
16
|
-
* 9703×$3/M + 39032×$15/M + 6032461×$0.30/M + 414101×$6.00/M = $4.9089333
|
|
17
|
-
* (SDK 报 4.908933300000001;haiku 行 532×$1/M + 17×$5/M 也精确命中)
|
|
18
|
-
* 缓存过期的代价自动体现在里面 —— 失效后 API 逐请求如实上报 input/cache_read
|
|
19
|
-
* 的实际拆分,costUSD 照单算,不需要我们建模 TTL 或命中率。
|
|
20
|
-
*
|
|
21
|
-
* # 数据源必须是 run_model_usage.cost_usd,不能是 runs.total_cost_usd
|
|
22
|
-
*
|
|
23
|
-
* runs 那列有历史污染:07-31 计量重做之前,写进去的是 SDK 的**会话累计**
|
|
24
|
-
* total_cost_usd(同一会话每轮都存一次当时的累计值),admin 的存量行因此虚高
|
|
25
|
-
* 三倍($165 vs 实际 $53.6)。run_model_usage 是重做后新建的表,只由差分路径
|
|
26
|
-
* 写入,两个真人用户的行都跟公式精确对齐。
|
|
27
|
-
*
|
|
28
|
-
* # 口径覆盖范围
|
|
29
|
-
*
|
|
30
|
-
* 包含:该会话所有模型的 input / output / cache_read / cache_create,含子代理
|
|
31
|
-
* sidechain(Miel 从没手选过 haiku,她的 haiku 行是子代理自己用的,照样进账)。
|
|
32
|
-
* 不含:生图(codex 骑订阅)、web_search(Tavily/百度/Exa/智谱各自的账单)、
|
|
33
|
-
* 机器本身。所以这个数字的准确定义是「该用户对 Anthropic 账号造成的负载」,
|
|
34
|
-
* 恰好就是我们要配给的东西。
|
|
35
|
-
*
|
|
36
|
-
* 限额来源:users.daily_cost_limit_usd 优先,NULL 走 env NODESIGN_USER_DAILY_USD
|
|
37
|
-
* (默认 $5)。admin 不限。按 Asia/Shanghai 日界滚动现算,没有定时任务。
|
|
38
|
-
*/
|
|
39
|
-
|
|
40
|
-
import { readFileSync } from 'node:fs';
|
|
41
|
-
import db, { getRun } from '../engine/runs/store.js';
|
|
42
|
-
import { countRunningTurns, listRunningTurnRunIds } from '../engine/runs/active-runs.js';
|
|
43
|
-
|
|
44
|
-
const DAY_MS = 24 * 60 * 60 * 1000;
|
|
45
|
-
const TZ_OFFSET_MS = 8 * 60 * 60 * 1000; // Asia/Shanghai,内测口径写死
|
|
46
|
-
|
|
47
|
-
/** 当天(+08:00 日界)的 UTC 起点,格式对齐 SQLite datetime('now') */
|
|
48
|
-
export function dayStartUtcSql(now = Date.now()) {
|
|
49
|
-
const startLocal = Math.floor((now + TZ_OFFSET_MS) / DAY_MS) * DAY_MS - TZ_OFFSET_MS;
|
|
50
|
-
return new Date(startLocal).toISOString().slice(0, 19).replace('T', ' ');
|
|
51
|
-
}
|
|
52
|
-
|
|
53
|
-
/** 当天已花的美元(闸门真口径)。数据源见文件头:run_model_usage.cost_usd */
|
|
54
|
-
export function usedCostToday(userId, now = Date.now()) {
|
|
55
|
-
const row = db.prepare(
|
|
56
|
-
`SELECT COALESCE(SUM(m.cost_usd), 0) AS used
|
|
57
|
-
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
58
|
-
WHERE r.user_id = ? AND r.created_at >= ?`,
|
|
59
|
-
).get(userId, dayStartUtcSql(now));
|
|
60
|
-
return row.used;
|
|
61
|
-
}
|
|
62
|
-
|
|
63
|
-
/** 全史花费(终身额度口径)。同一数据源同一公式,只是不设日界 */
|
|
64
|
-
export function usedCostTotal(userId) {
|
|
65
|
-
const row = db.prepare(
|
|
66
|
-
`SELECT COALESCE(SUM(m.cost_usd), 0) AS used
|
|
67
|
-
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
68
|
-
WHERE r.user_id = ?`,
|
|
69
|
-
).get(userId);
|
|
70
|
-
return row.used;
|
|
71
|
-
}
|
|
72
|
-
|
|
73
|
-
/**
|
|
74
|
-
* 当天 input+output token 数。**不再是闸门口径**,只留给 admin 视图和
|
|
75
|
-
* 前端的参考行 —— 用户看得懂 token,看不懂 $0.87 意味着多少对话。
|
|
76
|
-
*/
|
|
77
|
-
export function usedTokensToday(userId, now = Date.now()) {
|
|
78
|
-
const row = db.prepare(
|
|
79
|
-
`SELECT COALESCE(SUM(COALESCE(input_tokens,0) + COALESCE(output_tokens,0)), 0) AS used
|
|
80
|
-
FROM runs WHERE user_id = ? AND created_at >= ?`,
|
|
81
|
-
).get(userId, dayStartUtcSql(now));
|
|
82
|
-
return row.used;
|
|
83
|
-
}
|
|
84
|
-
|
|
85
|
-
/**
|
|
86
|
-
* 默认日额度(美元)。
|
|
87
|
-
*
|
|
88
|
-
* $15 的依据是 07-31 晚上第一个真用户的实际曲线:Miel 一小时零四分跑了 32 轮,
|
|
89
|
-
* 烧掉 $6.60,约 $6/小时(速率随会话变长而涨 —— 上下文越长每轮读的缓存越多)。
|
|
90
|
-
* 一晚上三小时的热情使用就是 $18 量级。
|
|
91
|
-
*
|
|
92
|
-
* 先按 $5 配过,实测下来那个数会在她玩到一半时把人挡在外面 —— 而她正是唯一
|
|
93
|
-
* 越过冷启动的人。日限额的职责是拦住失控,不是给正常使用配给;卡到最活跃的
|
|
94
|
-
* 那个人身上,说明数字定错了,不是人用多了。
|
|
95
|
-
*
|
|
96
|
-
* 参照:admin 重度开发日 $1.03/轮 × 52 轮 = $53.6(自己不限额)。
|
|
97
|
-
*/
|
|
98
|
-
export function defaultDailyLimit() {
|
|
99
|
-
const v = Number(process.env.NODESIGN_USER_DAILY_USD);
|
|
100
|
-
return Number.isFinite(v) && v > 0 ? v : 15;
|
|
101
|
-
}
|
|
102
|
-
|
|
103
|
-
/** 生图按张计价(08-21 深夜,basic 开放生图):每张记 $0.20 进本回合账,跟模型花费同一本、同一把日限尺。env NODESIGN_IMAGE_PRICE_USD */
|
|
104
|
-
export function imageChargeUsd(env = process.env) {
|
|
105
|
-
const v = Number(env.NODESIGN_IMAGE_PRICE_USD);
|
|
106
|
-
return Number.isFinite(v) && v >= 0 ? v : 0.2;
|
|
107
|
-
}
|
|
108
|
-
|
|
109
|
-
/** @returns {number|null} 美元;null = 不限(admin) */
|
|
110
|
-
export function limitFor(user) {
|
|
111
|
-
if (!user || user.role === 'admin') return null;
|
|
112
|
-
return user.dailyCostLimitUsd ?? defaultDailyLimit();
|
|
113
|
-
}
|
|
114
|
-
|
|
115
|
-
/** 金额展示:$1.36。小额也留两位小数,别把 $0.06 显示成 $0 让人以为没扣 */
|
|
116
|
-
export function fmtUsd(n) {
|
|
117
|
-
return `$${(Number(n) || 0).toFixed(2)}`;
|
|
118
|
-
}
|
|
119
|
-
|
|
120
|
-
/**
|
|
121
|
-
* 额度检查。两种口径(08-02 起):
|
|
122
|
-
* daily — 默认。+08:00 日界滚动,隔天刷新(内测熟人号)。
|
|
123
|
-
* lifetime — users.lifetime_cost_limit_usd 非空即生效:全史花费封顶,
|
|
124
|
-
* 永不刷新。给简历上的通用邀请码用(注册时从
|
|
125
|
-
* invites.grant_lifetime_usd 复制到用户身上,与码解耦)。
|
|
126
|
-
* 终身口径**取代**日限不叠加 —— HR 试用要的是当晚完整体验,
|
|
127
|
-
* 细水长流反而是坏体验;单号封顶 $15 本身就是敞口上限。
|
|
128
|
-
*
|
|
129
|
-
* `used` 是与 limit 同口径的比较数(终身号 = 全史);`usedToday` 恒为当日数,
|
|
130
|
-
* 展示层要"今天花了多少"时不用管口径。
|
|
131
|
-
*
|
|
132
|
-
* @returns {{ ok, kind: 'unlimited'|'daily'|'lifetime', used, limit, usedToday }} 单位美元
|
|
133
|
-
*/
|
|
134
|
-
export function checkQuota(user, now = Date.now()) {
|
|
135
|
-
const usedToday = usedCostToday(user?.id, now);
|
|
136
|
-
if (!user || user.role === 'admin') {
|
|
137
|
-
return { ok: true, kind: 'unlimited', used: usedToday, limit: null, usedToday };
|
|
138
|
-
}
|
|
139
|
-
if (user.lifetimeCostLimitUsd != null) {
|
|
140
|
-
const used = usedCostTotal(user.id);
|
|
141
|
-
return { ok: used < user.lifetimeCostLimitUsd, kind: 'lifetime', used, limit: user.lifetimeCostLimitUsd, usedToday };
|
|
142
|
-
}
|
|
143
|
-
const limit = user.dailyCostLimitUsd ?? defaultDailyLimit();
|
|
144
|
-
return { ok: usedToday < limit, kind: 'daily', used: usedToday, limit, usedToday };
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
/**
|
|
148
|
-
* 免费模型的闸(08-21):金额对 $0 的模型没有意义,免费档按**当日轮次**封顶 + 并发闸
|
|
149
|
-
* (checkConcurrency 不变)。数据源 runs.created_at(同 usedCostToday 的日界)。
|
|
150
|
-
* 口径:数的是该账号今天**所有**轮次,不按模型分 —— 公开注册号只能跑免费模型,
|
|
151
|
-
* 邀请码号跑免费模型时也照这个数(订阅额度是另一把尺,互不抵扣)。
|
|
152
|
-
* 上限 .env NODESIGN_FREE_DAILY_TURNS,默认 300;admin 不限。
|
|
153
|
-
*/
|
|
154
|
-
export function freeDailyTurnLimit() {
|
|
155
|
-
const v = Number(process.env.NODESIGN_FREE_DAILY_TURNS);
|
|
156
|
-
return Number.isFinite(v) && v > 0 ? v : 300;
|
|
157
|
-
}
|
|
158
|
-
export function usedTurnsToday(userId, now = Date.now()) {
|
|
159
|
-
const row = db.prepare('SELECT COUNT(*) AS n FROM runs WHERE user_id = ? AND created_at >= ?').get(userId, dayStartUtcSql(now));
|
|
160
|
-
return row.n;
|
|
161
|
-
}
|
|
162
|
-
export function checkFreeQuota(user, now = Date.now()) {
|
|
163
|
-
const used = usedTurnsToday(user?.id, now);
|
|
164
|
-
if (!user || user.role === 'admin') return { ok: true, kind: 'unlimited', used, limit: null };
|
|
165
|
-
const limit = freeDailyTurnLimit();
|
|
166
|
-
return { ok: used < limit, kind: 'free-turns', used, limit };
|
|
167
|
-
}
|
|
168
|
-
|
|
169
|
-
// ── 分模型明细(2026-07-31)──
|
|
170
|
-
//
|
|
171
|
-
// **只用于展示,不再是独立闸门。** 改成金额口径之后,分模型限额失去了理由:
|
|
172
|
-
// 原来配 sonnet 300k / opus 100k 是为了让 opus 更难挥霍,而金额天然做到这件事
|
|
173
|
-
// —— 同样一轮对话 opus 就是烧掉五倍预算,不需要第二个数字去表达同一个意图。
|
|
174
|
-
// 代价是没了"这个模型满了可以换那个继续"的退路,撞线就是撞线;换来的是用户
|
|
175
|
-
// 只需要理解一个数字,以及不会出现"opus 额度还剩很多但钱其实早花光了"的错位。
|
|
176
|
-
//
|
|
177
|
-
// 家族归并按模型串关键字:表里存的是 SDK 上报的原始模型名('claude-sonnet-5'),
|
|
178
|
-
// 用户视角只有 Sonnet / Opus 两个选项,版本号是实现细节。
|
|
179
|
-
// 注意:如果哪天重开 kimi gateway,SDK 上报的是 spoof 后的 opus 名,
|
|
180
|
-
// 这里会把 kimi 记成 opus —— 见 model-context.js APP_TO_SDK_MODEL。
|
|
181
|
-
|
|
182
|
-
export function modelFamily(model) {
|
|
183
|
-
const s = String(model || '').toLowerCase();
|
|
184
|
-
if (s.includes('sonnet')) return 'sonnet';
|
|
185
|
-
if (s.includes('opus')) return 'opus';
|
|
186
|
-
if (s.includes('haiku')) return 'haiku';
|
|
187
|
-
return 'other';
|
|
188
|
-
}
|
|
189
|
-
|
|
190
|
-
/** 家族显示名(429 文案 / 前端徽标共用) */
|
|
191
|
-
export function familyLabel(family) {
|
|
192
|
-
return { sonnet: 'Sonnet', opus: 'Opus', haiku: 'Haiku' }[family] || family;
|
|
193
|
-
}
|
|
194
|
-
|
|
195
|
-
/**
|
|
196
|
-
* 当天该用户按家族聚合的花费与 token,如
|
|
197
|
-
* `{ sonnet: { costUsd: 1.36, tokens: 18831 }, haiku: { costUsd: 0, tokens: 549 } }`
|
|
198
|
-
* 只有真出现过的家族才有键(没用过 opus 就没有 opus 键)。
|
|
199
|
-
*/
|
|
200
|
-
export function usedTodayByFamily(userId, now = Date.now()) {
|
|
201
|
-
const rows = db.prepare(
|
|
202
|
-
`SELECT m.model AS model,
|
|
203
|
-
SUM(m.cost_usd) AS cost,
|
|
204
|
-
SUM(m.input_tokens + m.output_tokens) AS tokens
|
|
205
|
-
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
206
|
-
WHERE r.user_id = ? AND r.created_at >= ?
|
|
207
|
-
GROUP BY m.model`,
|
|
208
|
-
).all(userId, dayStartUtcSql(now));
|
|
209
|
-
const byFamily = {};
|
|
210
|
-
for (const row of rows) {
|
|
211
|
-
const f = modelFamily(row.model);
|
|
212
|
-
const acc = byFamily[f] || (byFamily[f] = { costUsd: 0, tokens: 0 });
|
|
213
|
-
acc.costUsd += row.cost || 0;
|
|
214
|
-
acc.tokens += row.tokens || 0;
|
|
215
|
-
}
|
|
216
|
-
return byFamily;
|
|
217
|
-
}
|
|
218
|
-
|
|
219
|
-
// ── 并发闸门 ──
|
|
220
|
-
// 语义是"拒绝返 429"而不是"排队 await":turn.js 是 202 fire-and-forget,
|
|
221
|
-
// 闸门必须在 202 之前同步判。同 session 追加消息走既有排队语义不经这里。
|
|
222
|
-
//
|
|
223
|
-
// 08-21 经营态转向后分两档:
|
|
224
|
-
// - 订阅行(花站主的订阅):全局固定数 NODESIGN_MAX_CONCURRENT_RUNS(默认 3)
|
|
225
|
-
// - 免费行(Ox 这类零价模型):不受全局固定数限制。它的约束只剩机器本身 ——
|
|
226
|
-
// 每个 claude CLI 进程 350MB~1GB、机器 1 vCPU/8G 无 swap,所以改成**内存闸**:
|
|
227
|
-
// MemAvailable 低于 NODESIGN_MIN_FREE_MEM_MB(默认 700)就拒,外加一个防失控的
|
|
228
|
-
// 上限 NODESIGN_FREE_MAX_CONCURRENT_RUNS(默认 12)。内存闸对两档都生效(订阅行
|
|
229
|
-
// 的固定数 3 本来就是按内存拍的,内存先见底时固定数也救不了)
|
|
230
|
-
// - 每用户同时 1 个(admin 免)两档都一样:这是公平性不是资源
|
|
231
|
-
|
|
232
|
-
export function memAvailableMb() {
|
|
233
|
-
try {
|
|
234
|
-
const m = /MemAvailable:\s+(\d+) kB/.exec(readFileSync('/proc/meminfo', 'utf8'));
|
|
235
|
-
return m ? Math.floor(Number(m[1]) / 1024) : null;
|
|
236
|
-
} catch { return null; } // 非 Linux / 读不到 → 不以内存拒
|
|
237
|
-
}
|
|
238
|
-
|
|
239
|
-
/** 纯决策:输入全是数,方便测 */
|
|
240
|
-
export function decideConcurrency({ running, mine, isAdmin, free, memMb, env = process.env }) {
|
|
241
|
-
const minMem = Number(env.NODESIGN_MIN_FREE_MEM_MB) || 700;
|
|
242
|
-
if (memMb != null && memMb < minMem) {
|
|
243
|
-
return { ok: false, code: 'BUSY', message: `机器内存快满了(${running} 个任务在跑),稍等一会儿再发` };
|
|
244
|
-
}
|
|
245
|
-
const globalMax = free
|
|
246
|
-
? (Number(env.NODESIGN_FREE_MAX_CONCURRENT_RUNS) || 12)
|
|
247
|
-
: (Number(env.NODESIGN_MAX_CONCURRENT_RUNS) || 3);
|
|
248
|
-
if (running >= globalMax) {
|
|
249
|
-
return { ok: false, code: 'BUSY', message: `现在有点挤(${running} 个任务在跑),稍等一会儿再发` };
|
|
250
|
-
}
|
|
251
|
-
if (!isAdmin) {
|
|
252
|
-
const perUser = Number(env.NODESIGN_USER_CONCURRENT_RUNS) || 1;
|
|
253
|
-
if (mine >= perUser) {
|
|
254
|
-
return { ok: false, code: 'BUSY', message: '你有任务正在跑,等它完成再开下一个(同一对话里追加消息不受限)' };
|
|
255
|
-
}
|
|
256
|
-
}
|
|
257
|
-
return { ok: true };
|
|
258
|
-
}
|
|
259
|
-
|
|
260
|
-
export function checkConcurrency(user, { free = false } = {}) {
|
|
261
|
-
// running turn → user 归属:runId 查 runs.user_id(不给 session 注册表加
|
|
262
|
-
// userId 字段 —— 正在跑的 turn 就几个,查表成本可忽略)
|
|
263
|
-
let mine = 0;
|
|
264
|
-
if (user?.role !== 'admin') {
|
|
265
|
-
for (const rid of listRunningTurnRunIds()) {
|
|
266
|
-
if (getRun(rid)?.userId === user?.id) mine += 1;
|
|
267
|
-
}
|
|
268
|
-
}
|
|
269
|
-
return decideConcurrency({ running: countRunningTurns(), mine, isAdmin: user?.role === 'admin', free, memMb: memAvailableMb() });
|
|
270
|
-
}
|
|
1
|
+
/**
|
|
2
|
+
* server/lib/quota.js — 每用户日用量限额(2026-07-30 内测;07-31 口径改成金额)
|
|
3
|
+
*
|
|
4
|
+
* # 为什么闸门数的是钱不是 token
|
|
5
|
+
*
|
|
6
|
+
* 原口径是 sum(input_tokens + output_tokens),缓存不计。实测这个口径量错了主项:
|
|
7
|
+
* 07-31 当天全站 input+output 是 317k,cache_read 是 83M,差 260 倍。也就是说
|
|
8
|
+
* 配额条显示"才用了 4%"的人,实际推给账号的负载可能已经是一整天的大头。
|
|
9
|
+
*
|
|
10
|
+
* 更要命的是**缓存过期**。同一轮对话,缓存全命中和全过期的 token 计数几乎一样,
|
|
11
|
+
* 但过期后那批 token 从 $0.30/M 的 cache_read 变成 $3/M 的 input,还要再花
|
|
12
|
+
* $6.00/M 重写一次缓存,金额能差十倍。token 口径对这件事完全瞎。
|
|
13
|
+
*
|
|
14
|
+
* SDK 的 `ModelUsage.costUSD` 是原生字段,CLI 内部按分模型价目表算好给出来的。
|
|
15
|
+
* 逆推验证精确到小数点后 7 位:
|
|
16
|
+
* 9703×$3/M + 39032×$15/M + 6032461×$0.30/M + 414101×$6.00/M = $4.9089333
|
|
17
|
+
* (SDK 报 4.908933300000001;haiku 行 532×$1/M + 17×$5/M 也精确命中)
|
|
18
|
+
* 缓存过期的代价自动体现在里面 —— 失效后 API 逐请求如实上报 input/cache_read
|
|
19
|
+
* 的实际拆分,costUSD 照单算,不需要我们建模 TTL 或命中率。
|
|
20
|
+
*
|
|
21
|
+
* # 数据源必须是 run_model_usage.cost_usd,不能是 runs.total_cost_usd
|
|
22
|
+
*
|
|
23
|
+
* runs 那列有历史污染:07-31 计量重做之前,写进去的是 SDK 的**会话累计**
|
|
24
|
+
* total_cost_usd(同一会话每轮都存一次当时的累计值),admin 的存量行因此虚高
|
|
25
|
+
* 三倍($165 vs 实际 $53.6)。run_model_usage 是重做后新建的表,只由差分路径
|
|
26
|
+
* 写入,两个真人用户的行都跟公式精确对齐。
|
|
27
|
+
*
|
|
28
|
+
* # 口径覆盖范围
|
|
29
|
+
*
|
|
30
|
+
* 包含:该会话所有模型的 input / output / cache_read / cache_create,含子代理
|
|
31
|
+
* sidechain(Miel 从没手选过 haiku,她的 haiku 行是子代理自己用的,照样进账)。
|
|
32
|
+
* 不含:生图(codex 骑订阅)、web_search(Tavily/百度/Exa/智谱各自的账单)、
|
|
33
|
+
* 机器本身。所以这个数字的准确定义是「该用户对 Anthropic 账号造成的负载」,
|
|
34
|
+
* 恰好就是我们要配给的东西。
|
|
35
|
+
*
|
|
36
|
+
* 限额来源:users.daily_cost_limit_usd 优先,NULL 走 env NODESIGN_USER_DAILY_USD
|
|
37
|
+
* (默认 $5)。admin 不限。按 Asia/Shanghai 日界滚动现算,没有定时任务。
|
|
38
|
+
*/
|
|
39
|
+
|
|
40
|
+
import { readFileSync } from 'node:fs';
|
|
41
|
+
import db, { getRun } from '../engine/runs/store.js';
|
|
42
|
+
import { countRunningTurns, listRunningTurnRunIds } from '../engine/runs/active-runs.js';
|
|
43
|
+
|
|
44
|
+
const DAY_MS = 24 * 60 * 60 * 1000;
|
|
45
|
+
const TZ_OFFSET_MS = 8 * 60 * 60 * 1000; // Asia/Shanghai,内测口径写死
|
|
46
|
+
|
|
47
|
+
/** 当天(+08:00 日界)的 UTC 起点,格式对齐 SQLite datetime('now') */
|
|
48
|
+
export function dayStartUtcSql(now = Date.now()) {
|
|
49
|
+
const startLocal = Math.floor((now + TZ_OFFSET_MS) / DAY_MS) * DAY_MS - TZ_OFFSET_MS;
|
|
50
|
+
return new Date(startLocal).toISOString().slice(0, 19).replace('T', ' ');
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
/** 当天已花的美元(闸门真口径)。数据源见文件头:run_model_usage.cost_usd */
|
|
54
|
+
export function usedCostToday(userId, now = Date.now()) {
|
|
55
|
+
const row = db.prepare(
|
|
56
|
+
`SELECT COALESCE(SUM(m.cost_usd), 0) AS used
|
|
57
|
+
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
58
|
+
WHERE r.user_id = ? AND r.created_at >= ?`,
|
|
59
|
+
).get(userId, dayStartUtcSql(now));
|
|
60
|
+
return row.used;
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
/** 全史花费(终身额度口径)。同一数据源同一公式,只是不设日界 */
|
|
64
|
+
export function usedCostTotal(userId) {
|
|
65
|
+
const row = db.prepare(
|
|
66
|
+
`SELECT COALESCE(SUM(m.cost_usd), 0) AS used
|
|
67
|
+
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
68
|
+
WHERE r.user_id = ?`,
|
|
69
|
+
).get(userId);
|
|
70
|
+
return row.used;
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
/**
|
|
74
|
+
* 当天 input+output token 数。**不再是闸门口径**,只留给 admin 视图和
|
|
75
|
+
* 前端的参考行 —— 用户看得懂 token,看不懂 $0.87 意味着多少对话。
|
|
76
|
+
*/
|
|
77
|
+
export function usedTokensToday(userId, now = Date.now()) {
|
|
78
|
+
const row = db.prepare(
|
|
79
|
+
`SELECT COALESCE(SUM(COALESCE(input_tokens,0) + COALESCE(output_tokens,0)), 0) AS used
|
|
80
|
+
FROM runs WHERE user_id = ? AND created_at >= ?`,
|
|
81
|
+
).get(userId, dayStartUtcSql(now));
|
|
82
|
+
return row.used;
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
/**
|
|
86
|
+
* 默认日额度(美元)。
|
|
87
|
+
*
|
|
88
|
+
* $15 的依据是 07-31 晚上第一个真用户的实际曲线:Miel 一小时零四分跑了 32 轮,
|
|
89
|
+
* 烧掉 $6.60,约 $6/小时(速率随会话变长而涨 —— 上下文越长每轮读的缓存越多)。
|
|
90
|
+
* 一晚上三小时的热情使用就是 $18 量级。
|
|
91
|
+
*
|
|
92
|
+
* 先按 $5 配过,实测下来那个数会在她玩到一半时把人挡在外面 —— 而她正是唯一
|
|
93
|
+
* 越过冷启动的人。日限额的职责是拦住失控,不是给正常使用配给;卡到最活跃的
|
|
94
|
+
* 那个人身上,说明数字定错了,不是人用多了。
|
|
95
|
+
*
|
|
96
|
+
* 参照:admin 重度开发日 $1.03/轮 × 52 轮 = $53.6(自己不限额)。
|
|
97
|
+
*/
|
|
98
|
+
export function defaultDailyLimit() {
|
|
99
|
+
const v = Number(process.env.NODESIGN_USER_DAILY_USD);
|
|
100
|
+
return Number.isFinite(v) && v > 0 ? v : 15;
|
|
101
|
+
}
|
|
102
|
+
|
|
103
|
+
/** 生图按张计价(08-21 深夜,basic 开放生图):每张记 $0.20 进本回合账,跟模型花费同一本、同一把日限尺。env NODESIGN_IMAGE_PRICE_USD */
|
|
104
|
+
export function imageChargeUsd(env = process.env) {
|
|
105
|
+
const v = Number(env.NODESIGN_IMAGE_PRICE_USD);
|
|
106
|
+
return Number.isFinite(v) && v >= 0 ? v : 0.2;
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
/** @returns {number|null} 美元;null = 不限(admin) */
|
|
110
|
+
export function limitFor(user) {
|
|
111
|
+
if (!user || user.role === 'admin') return null;
|
|
112
|
+
return user.dailyCostLimitUsd ?? defaultDailyLimit();
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
/** 金额展示:$1.36。小额也留两位小数,别把 $0.06 显示成 $0 让人以为没扣 */
|
|
116
|
+
export function fmtUsd(n) {
|
|
117
|
+
return `$${(Number(n) || 0).toFixed(2)}`;
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
/**
|
|
121
|
+
* 额度检查。两种口径(08-02 起):
|
|
122
|
+
* daily — 默认。+08:00 日界滚动,隔天刷新(内测熟人号)。
|
|
123
|
+
* lifetime — users.lifetime_cost_limit_usd 非空即生效:全史花费封顶,
|
|
124
|
+
* 永不刷新。给简历上的通用邀请码用(注册时从
|
|
125
|
+
* invites.grant_lifetime_usd 复制到用户身上,与码解耦)。
|
|
126
|
+
* 终身口径**取代**日限不叠加 —— HR 试用要的是当晚完整体验,
|
|
127
|
+
* 细水长流反而是坏体验;单号封顶 $15 本身就是敞口上限。
|
|
128
|
+
*
|
|
129
|
+
* `used` 是与 limit 同口径的比较数(终身号 = 全史);`usedToday` 恒为当日数,
|
|
130
|
+
* 展示层要"今天花了多少"时不用管口径。
|
|
131
|
+
*
|
|
132
|
+
* @returns {{ ok, kind: 'unlimited'|'daily'|'lifetime', used, limit, usedToday }} 单位美元
|
|
133
|
+
*/
|
|
134
|
+
export function checkQuota(user, now = Date.now()) {
|
|
135
|
+
const usedToday = usedCostToday(user?.id, now);
|
|
136
|
+
if (!user || user.role === 'admin') {
|
|
137
|
+
return { ok: true, kind: 'unlimited', used: usedToday, limit: null, usedToday };
|
|
138
|
+
}
|
|
139
|
+
if (user.lifetimeCostLimitUsd != null) {
|
|
140
|
+
const used = usedCostTotal(user.id);
|
|
141
|
+
return { ok: used < user.lifetimeCostLimitUsd, kind: 'lifetime', used, limit: user.lifetimeCostLimitUsd, usedToday };
|
|
142
|
+
}
|
|
143
|
+
const limit = user.dailyCostLimitUsd ?? defaultDailyLimit();
|
|
144
|
+
return { ok: usedToday < limit, kind: 'daily', used: usedToday, limit, usedToday };
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
/**
|
|
148
|
+
* 免费模型的闸(08-21):金额对 $0 的模型没有意义,免费档按**当日轮次**封顶 + 并发闸
|
|
149
|
+
* (checkConcurrency 不变)。数据源 runs.created_at(同 usedCostToday 的日界)。
|
|
150
|
+
* 口径:数的是该账号今天**所有**轮次,不按模型分 —— 公开注册号只能跑免费模型,
|
|
151
|
+
* 邀请码号跑免费模型时也照这个数(订阅额度是另一把尺,互不抵扣)。
|
|
152
|
+
* 上限 .env NODESIGN_FREE_DAILY_TURNS,默认 300;admin 不限。
|
|
153
|
+
*/
|
|
154
|
+
export function freeDailyTurnLimit() {
|
|
155
|
+
const v = Number(process.env.NODESIGN_FREE_DAILY_TURNS);
|
|
156
|
+
return Number.isFinite(v) && v > 0 ? v : 300;
|
|
157
|
+
}
|
|
158
|
+
export function usedTurnsToday(userId, now = Date.now()) {
|
|
159
|
+
const row = db.prepare('SELECT COUNT(*) AS n FROM runs WHERE user_id = ? AND created_at >= ?').get(userId, dayStartUtcSql(now));
|
|
160
|
+
return row.n;
|
|
161
|
+
}
|
|
162
|
+
export function checkFreeQuota(user, now = Date.now()) {
|
|
163
|
+
const used = usedTurnsToday(user?.id, now);
|
|
164
|
+
if (!user || user.role === 'admin') return { ok: true, kind: 'unlimited', used, limit: null };
|
|
165
|
+
const limit = freeDailyTurnLimit();
|
|
166
|
+
return { ok: used < limit, kind: 'free-turns', used, limit };
|
|
167
|
+
}
|
|
168
|
+
|
|
169
|
+
// ── 分模型明细(2026-07-31)──
|
|
170
|
+
//
|
|
171
|
+
// **只用于展示,不再是独立闸门。** 改成金额口径之后,分模型限额失去了理由:
|
|
172
|
+
// 原来配 sonnet 300k / opus 100k 是为了让 opus 更难挥霍,而金额天然做到这件事
|
|
173
|
+
// —— 同样一轮对话 opus 就是烧掉五倍预算,不需要第二个数字去表达同一个意图。
|
|
174
|
+
// 代价是没了"这个模型满了可以换那个继续"的退路,撞线就是撞线;换来的是用户
|
|
175
|
+
// 只需要理解一个数字,以及不会出现"opus 额度还剩很多但钱其实早花光了"的错位。
|
|
176
|
+
//
|
|
177
|
+
// 家族归并按模型串关键字:表里存的是 SDK 上报的原始模型名('claude-sonnet-5'),
|
|
178
|
+
// 用户视角只有 Sonnet / Opus 两个选项,版本号是实现细节。
|
|
179
|
+
// 注意:如果哪天重开 kimi gateway,SDK 上报的是 spoof 后的 opus 名,
|
|
180
|
+
// 这里会把 kimi 记成 opus —— 见 model-context.js APP_TO_SDK_MODEL。
|
|
181
|
+
|
|
182
|
+
export function modelFamily(model) {
|
|
183
|
+
const s = String(model || '').toLowerCase();
|
|
184
|
+
if (s.includes('sonnet')) return 'sonnet';
|
|
185
|
+
if (s.includes('opus')) return 'opus';
|
|
186
|
+
if (s.includes('haiku')) return 'haiku';
|
|
187
|
+
return 'other';
|
|
188
|
+
}
|
|
189
|
+
|
|
190
|
+
/** 家族显示名(429 文案 / 前端徽标共用) */
|
|
191
|
+
export function familyLabel(family) {
|
|
192
|
+
return { sonnet: 'Sonnet', opus: 'Opus', haiku: 'Haiku' }[family] || family;
|
|
193
|
+
}
|
|
194
|
+
|
|
195
|
+
/**
|
|
196
|
+
* 当天该用户按家族聚合的花费与 token,如
|
|
197
|
+
* `{ sonnet: { costUsd: 1.36, tokens: 18831 }, haiku: { costUsd: 0, tokens: 549 } }`
|
|
198
|
+
* 只有真出现过的家族才有键(没用过 opus 就没有 opus 键)。
|
|
199
|
+
*/
|
|
200
|
+
export function usedTodayByFamily(userId, now = Date.now()) {
|
|
201
|
+
const rows = db.prepare(
|
|
202
|
+
`SELECT m.model AS model,
|
|
203
|
+
SUM(m.cost_usd) AS cost,
|
|
204
|
+
SUM(m.input_tokens + m.output_tokens) AS tokens
|
|
205
|
+
FROM run_model_usage m JOIN runs r ON r.id = m.run_id
|
|
206
|
+
WHERE r.user_id = ? AND r.created_at >= ?
|
|
207
|
+
GROUP BY m.model`,
|
|
208
|
+
).all(userId, dayStartUtcSql(now));
|
|
209
|
+
const byFamily = {};
|
|
210
|
+
for (const row of rows) {
|
|
211
|
+
const f = modelFamily(row.model);
|
|
212
|
+
const acc = byFamily[f] || (byFamily[f] = { costUsd: 0, tokens: 0 });
|
|
213
|
+
acc.costUsd += row.cost || 0;
|
|
214
|
+
acc.tokens += row.tokens || 0;
|
|
215
|
+
}
|
|
216
|
+
return byFamily;
|
|
217
|
+
}
|
|
218
|
+
|
|
219
|
+
// ── 并发闸门 ──
|
|
220
|
+
// 语义是"拒绝返 429"而不是"排队 await":turn.js 是 202 fire-and-forget,
|
|
221
|
+
// 闸门必须在 202 之前同步判。同 session 追加消息走既有排队语义不经这里。
|
|
222
|
+
//
|
|
223
|
+
// 08-21 经营态转向后分两档:
|
|
224
|
+
// - 订阅行(花站主的订阅):全局固定数 NODESIGN_MAX_CONCURRENT_RUNS(默认 3)
|
|
225
|
+
// - 免费行(Ox 这类零价模型):不受全局固定数限制。它的约束只剩机器本身 ——
|
|
226
|
+
// 每个 claude CLI 进程 350MB~1GB、机器 1 vCPU/8G 无 swap,所以改成**内存闸**:
|
|
227
|
+
// MemAvailable 低于 NODESIGN_MIN_FREE_MEM_MB(默认 700)就拒,外加一个防失控的
|
|
228
|
+
// 上限 NODESIGN_FREE_MAX_CONCURRENT_RUNS(默认 12)。内存闸对两档都生效(订阅行
|
|
229
|
+
// 的固定数 3 本来就是按内存拍的,内存先见底时固定数也救不了)
|
|
230
|
+
// - 每用户同时 1 个(admin 免)两档都一样:这是公平性不是资源
|
|
231
|
+
|
|
232
|
+
export function memAvailableMb() {
|
|
233
|
+
try {
|
|
234
|
+
const m = /MemAvailable:\s+(\d+) kB/.exec(readFileSync('/proc/meminfo', 'utf8'));
|
|
235
|
+
return m ? Math.floor(Number(m[1]) / 1024) : null;
|
|
236
|
+
} catch { return null; } // 非 Linux / 读不到 → 不以内存拒
|
|
237
|
+
}
|
|
238
|
+
|
|
239
|
+
/** 纯决策:输入全是数,方便测 */
|
|
240
|
+
export function decideConcurrency({ running, mine, isAdmin, free, memMb, env = process.env }) {
|
|
241
|
+
const minMem = Number(env.NODESIGN_MIN_FREE_MEM_MB) || 700;
|
|
242
|
+
if (memMb != null && memMb < minMem) {
|
|
243
|
+
return { ok: false, code: 'BUSY', message: `机器内存快满了(${running} 个任务在跑),稍等一会儿再发` };
|
|
244
|
+
}
|
|
245
|
+
const globalMax = free
|
|
246
|
+
? (Number(env.NODESIGN_FREE_MAX_CONCURRENT_RUNS) || 12)
|
|
247
|
+
: (Number(env.NODESIGN_MAX_CONCURRENT_RUNS) || 3);
|
|
248
|
+
if (running >= globalMax) {
|
|
249
|
+
return { ok: false, code: 'BUSY', message: `现在有点挤(${running} 个任务在跑),稍等一会儿再发` };
|
|
250
|
+
}
|
|
251
|
+
if (!isAdmin) {
|
|
252
|
+
const perUser = Number(env.NODESIGN_USER_CONCURRENT_RUNS) || 1;
|
|
253
|
+
if (mine >= perUser) {
|
|
254
|
+
return { ok: false, code: 'BUSY', message: '你有任务正在跑,等它完成再开下一个(同一对话里追加消息不受限)' };
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
return { ok: true };
|
|
258
|
+
}
|
|
259
|
+
|
|
260
|
+
export function checkConcurrency(user, { free = false } = {}) {
|
|
261
|
+
// running turn → user 归属:runId 查 runs.user_id(不给 session 注册表加
|
|
262
|
+
// userId 字段 —— 正在跑的 turn 就几个,查表成本可忽略)
|
|
263
|
+
let mine = 0;
|
|
264
|
+
if (user?.role !== 'admin') {
|
|
265
|
+
for (const rid of listRunningTurnRunIds()) {
|
|
266
|
+
if (getRun(rid)?.userId === user?.id) mine += 1;
|
|
267
|
+
}
|
|
268
|
+
}
|
|
269
|
+
return decideConcurrency({ running: countRunningTurns(), mine, isAdmin: user?.role === 'admin', free, memMb: memAvailableMb() });
|
|
270
|
+
}
|
|
@@ -1,29 +1,29 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* rate-window.js —— 内存滑动窗口限流。
|
|
3
|
-
*
|
|
4
|
-
* 给演出端点用的频率闸门:金额日限(quota.js)拦不住单日尖峰——iframe 同源
|
|
5
|
-
* 带 cookie,agent 写的页面里一个 setInterval 就是烧钱机,必须有按分钟的闸。
|
|
6
|
-
* 内存实现就够:限流是保护性关卡不是记账,重启清零无所谓。
|
|
7
|
-
*/
|
|
8
|
-
|
|
9
|
-
export function makeRateWindow({ limit, windowMs }) {
|
|
10
|
-
const hits = new Map(); // key → 命中时间戳数组(只留窗口内的)
|
|
11
|
-
return {
|
|
12
|
-
/** 窗口内已命中几次(只看不扣;注册这类"先判再做、做成了才扣"的场景用) */
|
|
13
|
-
count(key, now = Date.now()) {
|
|
14
|
-
return (hits.get(key) || []).filter(t => now - t < windowMs).length;
|
|
15
|
-
},
|
|
16
|
-
/** @returns {{ok: true} | {ok: false, retryAfterMs: number}} */
|
|
17
|
-
take(key, now = Date.now()) {
|
|
18
|
-
const arr = (hits.get(key) || []).filter(t => now - t < windowMs);
|
|
19
|
-
if (arr.length >= limit) {
|
|
20
|
-
hits.set(key, arr);
|
|
21
|
-
// limit=0(闸门关死)时窗口里没有命中记录,arr[0] 是 undefined → NaN;按整窗算
|
|
22
|
-
return { ok: false, retryAfterMs: arr.length ? windowMs - (now - arr[0]) : windowMs };
|
|
23
|
-
}
|
|
24
|
-
arr.push(now);
|
|
25
|
-
hits.set(key, arr);
|
|
26
|
-
return { ok: true };
|
|
27
|
-
},
|
|
28
|
-
};
|
|
29
|
-
}
|
|
1
|
+
/**
|
|
2
|
+
* rate-window.js —— 内存滑动窗口限流。
|
|
3
|
+
*
|
|
4
|
+
* 给演出端点用的频率闸门:金额日限(quota.js)拦不住单日尖峰——iframe 同源
|
|
5
|
+
* 带 cookie,agent 写的页面里一个 setInterval 就是烧钱机,必须有按分钟的闸。
|
|
6
|
+
* 内存实现就够:限流是保护性关卡不是记账,重启清零无所谓。
|
|
7
|
+
*/
|
|
8
|
+
|
|
9
|
+
export function makeRateWindow({ limit, windowMs }) {
|
|
10
|
+
const hits = new Map(); // key → 命中时间戳数组(只留窗口内的)
|
|
11
|
+
return {
|
|
12
|
+
/** 窗口内已命中几次(只看不扣;注册这类"先判再做、做成了才扣"的场景用) */
|
|
13
|
+
count(key, now = Date.now()) {
|
|
14
|
+
return (hits.get(key) || []).filter(t => now - t < windowMs).length;
|
|
15
|
+
},
|
|
16
|
+
/** @returns {{ok: true} | {ok: false, retryAfterMs: number}} */
|
|
17
|
+
take(key, now = Date.now()) {
|
|
18
|
+
const arr = (hits.get(key) || []).filter(t => now - t < windowMs);
|
|
19
|
+
if (arr.length >= limit) {
|
|
20
|
+
hits.set(key, arr);
|
|
21
|
+
// limit=0(闸门关死)时窗口里没有命中记录,arr[0] 是 undefined → NaN;按整窗算
|
|
22
|
+
return { ok: false, retryAfterMs: arr.length ? windowMs - (now - arr[0]) : windowMs };
|
|
23
|
+
}
|
|
24
|
+
arr.push(now);
|
|
25
|
+
hits.set(key, arr);
|
|
26
|
+
return { ok: true };
|
|
27
|
+
},
|
|
28
|
+
};
|
|
29
|
+
}
|