@build-astron-co/nimbus 0.4.1 → 0.4.3

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.
Files changed (435) hide show
  1. package/CHANGELOG.md +268 -89
  2. package/README.md +26 -567
  3. package/dist/src/agent/compaction-agent.js +24 -12
  4. package/dist/src/agent/context-manager.js +2 -1
  5. package/dist/src/agent/expand-files.js +2 -1
  6. package/dist/src/agent/loop.js +71 -33
  7. package/dist/src/agent/permissions.js +4 -2
  8. package/dist/src/agent/system-prompt.js +34 -17
  9. package/dist/src/app.js +1 -1
  10. package/dist/src/auth/keychain.js +8 -4
  11. package/dist/src/auth/store.js +70 -107
  12. package/dist/src/cli/init.js +35 -19
  13. package/dist/src/cli/run.js +18 -10
  14. package/dist/src/cli/serve.js +4 -2
  15. package/dist/src/cli.js +52 -11
  16. package/dist/src/commands/alias.js +5 -3
  17. package/dist/src/commands/audit/index.js +2 -1
  18. package/dist/src/commands/aws-terraform.js +36 -18
  19. package/dist/src/commands/completions.js +1 -1
  20. package/dist/src/commands/config.js +3 -2
  21. package/dist/src/commands/connect-github.js +92 -0
  22. package/dist/src/commands/cost/index.js +3 -2
  23. package/dist/src/commands/deploy.js +15 -10
  24. package/dist/src/commands/doctor.js +9 -6
  25. package/dist/src/commands/drift/index.js +2 -1
  26. package/dist/src/commands/export.js +5 -3
  27. package/dist/src/commands/generate-terraform.js +110 -2
  28. package/dist/src/commands/import.js +3 -3
  29. package/dist/src/commands/incident.js +10 -5
  30. package/dist/src/commands/login.js +8 -93
  31. package/dist/src/commands/logs.js +16 -8
  32. package/dist/src/commands/onboarding.js +6 -4
  33. package/dist/src/commands/pipeline.js +6 -3
  34. package/dist/src/commands/plugin.js +3 -2
  35. package/dist/src/commands/profile.js +27 -14
  36. package/dist/src/commands/questionnaire.js +1 -1
  37. package/dist/src/commands/rollback.js +3 -2
  38. package/dist/src/commands/rollout.js +5 -3
  39. package/dist/src/commands/runbook.js +17 -10
  40. package/dist/src/commands/schedule.js +10 -5
  41. package/dist/src/commands/status.js +2 -1
  42. package/dist/src/commands/team-context.js +12 -7
  43. package/dist/src/commands/template.js +1 -1
  44. package/dist/src/commands/tf/index.js +6 -3
  45. package/dist/src/commands/upgrade.js +5 -3
  46. package/dist/src/commands/version.js +6 -3
  47. package/dist/src/commands/watch.js +6 -3
  48. package/dist/src/compat/sqlite.js +5 -3
  49. package/dist/src/config/mode-store.js +2 -1
  50. package/dist/src/config/profiles.js +4 -2
  51. package/dist/src/config/types.js +2 -1
  52. package/dist/src/engine/executor.js +8 -4
  53. package/dist/src/engine/planner.js +9 -5
  54. package/dist/src/llm/providers/anthropic.js +6 -3
  55. package/dist/src/llm/providers/ollama.js +1 -1
  56. package/dist/src/llm/router.js +22 -7
  57. package/dist/src/nimbus.js +1 -0
  58. package/dist/src/sessions/manager.js +6 -3
  59. package/dist/src/sharing/viewer.js +2 -1
  60. package/dist/src/tools/file-ops.js +1 -2
  61. package/dist/src/tools/schemas/devops.js +197 -108
  62. package/dist/src/tools/schemas/standard.js +1 -1
  63. package/dist/src/ui/App.js +25 -13
  64. package/dist/src/ui/FileDiffModal.js +22 -11
  65. package/dist/src/ui/HelpModal.js +2 -1
  66. package/dist/src/ui/InputBox.js +6 -3
  67. package/dist/src/ui/MessageList.js +40 -20
  68. package/dist/src/ui/TerminalPane.js +2 -1
  69. package/dist/src/ui/ToolCallDisplay.js +12 -6
  70. package/dist/src/ui/TreePane.js +2 -1
  71. package/dist/src/ui/ink/index.js +37 -21
  72. package/dist/src/version.js +1 -1
  73. package/dist/src/watcher/index.js +8 -4
  74. package/package.json +3 -5
  75. package/src/__tests__/alias.test.ts +0 -133
  76. package/src/__tests__/app.test.ts +0 -76
  77. package/src/__tests__/audit.test.ts +0 -877
  78. package/src/__tests__/circuit-breaker.test.ts +0 -116
  79. package/src/__tests__/cli-run.test.ts +0 -351
  80. package/src/__tests__/compat-sqlite.test.ts +0 -68
  81. package/src/__tests__/context-manager.test.ts +0 -632
  82. package/src/__tests__/context.test.ts +0 -242
  83. package/src/__tests__/devops-terminal-gaps.test.ts +0 -718
  84. package/src/__tests__/doctor.test.ts +0 -48
  85. package/src/__tests__/enterprise.test.ts +0 -401
  86. package/src/__tests__/export.test.ts +0 -236
  87. package/src/__tests__/gap-11-18-20.test.ts +0 -958
  88. package/src/__tests__/generator.test.ts +0 -433
  89. package/src/__tests__/helm-streaming.test.ts +0 -127
  90. package/src/__tests__/hooks.test.ts +0 -582
  91. package/src/__tests__/incident.test.ts +0 -179
  92. package/src/__tests__/init.test.ts +0 -487
  93. package/src/__tests__/intent-parser.test.ts +0 -229
  94. package/src/__tests__/llm-router.test.ts +0 -209
  95. package/src/__tests__/logs.test.ts +0 -107
  96. package/src/__tests__/loop-errors.test.ts +0 -244
  97. package/src/__tests__/lsp.test.ts +0 -293
  98. package/src/__tests__/modes.test.ts +0 -336
  99. package/src/__tests__/perf-optimizations.test.ts +0 -847
  100. package/src/__tests__/permissions.test.ts +0 -338
  101. package/src/__tests__/pipeline.test.ts +0 -50
  102. package/src/__tests__/polish-phase3.test.ts +0 -340
  103. package/src/__tests__/profile.test.ts +0 -237
  104. package/src/__tests__/rollback.test.ts +0 -83
  105. package/src/__tests__/runbook.test.ts +0 -219
  106. package/src/__tests__/schedule.test.ts +0 -206
  107. package/src/__tests__/serve.test.ts +0 -275
  108. package/src/__tests__/sessions.test.ts +0 -322
  109. package/src/__tests__/sharing.test.ts +0 -340
  110. package/src/__tests__/snapshots.test.ts +0 -581
  111. package/src/__tests__/standalone-migration.test.ts +0 -199
  112. package/src/__tests__/state-db.test.ts +0 -334
  113. package/src/__tests__/status.test.ts +0 -158
  114. package/src/__tests__/stream-with-tools.test.ts +0 -778
  115. package/src/__tests__/subagents.test.ts +0 -176
  116. package/src/__tests__/system-prompt.test.ts +0 -248
  117. package/src/__tests__/terminal-gap-v2.test.ts +0 -395
  118. package/src/__tests__/terminal-parity.test.ts +0 -393
  119. package/src/__tests__/tf-apply.test.ts +0 -187
  120. package/src/__tests__/tool-converter.test.ts +0 -256
  121. package/src/__tests__/tool-schemas.test.ts +0 -602
  122. package/src/__tests__/tools.test.ts +0 -144
  123. package/src/__tests__/version-json.test.ts +0 -184
  124. package/src/__tests__/version.test.ts +0 -49
  125. package/src/__tests__/watch.test.ts +0 -129
  126. package/src/agent/compaction-agent.ts +0 -266
  127. package/src/agent/context-manager.ts +0 -499
  128. package/src/agent/context.ts +0 -427
  129. package/src/agent/deploy-preview.ts +0 -487
  130. package/src/agent/expand-files.ts +0 -108
  131. package/src/agent/index.ts +0 -68
  132. package/src/agent/loop.ts +0 -1998
  133. package/src/agent/modes.ts +0 -429
  134. package/src/agent/permissions.ts +0 -513
  135. package/src/agent/subagents/base.ts +0 -116
  136. package/src/agent/subagents/cost.ts +0 -51
  137. package/src/agent/subagents/explore.ts +0 -42
  138. package/src/agent/subagents/general.ts +0 -54
  139. package/src/agent/subagents/index.ts +0 -102
  140. package/src/agent/subagents/infra.ts +0 -59
  141. package/src/agent/subagents/security.ts +0 -69
  142. package/src/agent/system-prompt.ts +0 -990
  143. package/src/app.ts +0 -180
  144. package/src/audit/activity-log.ts +0 -290
  145. package/src/audit/compliance-checker.ts +0 -540
  146. package/src/audit/cost-tracker.ts +0 -318
  147. package/src/audit/index.ts +0 -23
  148. package/src/audit/security-scanner.ts +0 -641
  149. package/src/auth/guard.ts +0 -75
  150. package/src/auth/index.ts +0 -56
  151. package/src/auth/keychain.ts +0 -82
  152. package/src/auth/oauth.ts +0 -465
  153. package/src/auth/providers.ts +0 -470
  154. package/src/auth/sso.ts +0 -113
  155. package/src/auth/store.ts +0 -505
  156. package/src/auth/types.ts +0 -187
  157. package/src/build.ts +0 -141
  158. package/src/cli/index.ts +0 -16
  159. package/src/cli/init.ts +0 -1227
  160. package/src/cli/openapi-spec.ts +0 -356
  161. package/src/cli/run.ts +0 -628
  162. package/src/cli/serve-auth.ts +0 -80
  163. package/src/cli/serve.ts +0 -539
  164. package/src/cli/web.ts +0 -71
  165. package/src/cli.ts +0 -1728
  166. package/src/clients/core-engine-client.ts +0 -227
  167. package/src/clients/enterprise-client.ts +0 -334
  168. package/src/clients/generator-client.ts +0 -351
  169. package/src/clients/git-client.ts +0 -627
  170. package/src/clients/github-client.ts +0 -410
  171. package/src/clients/helm-client.ts +0 -504
  172. package/src/clients/index.ts +0 -80
  173. package/src/clients/k8s-client.ts +0 -497
  174. package/src/clients/llm-client.ts +0 -161
  175. package/src/clients/rest-client.ts +0 -130
  176. package/src/clients/service-discovery.ts +0 -38
  177. package/src/clients/terraform-client.ts +0 -482
  178. package/src/clients/tools-client.ts +0 -1843
  179. package/src/clients/ws-client.ts +0 -115
  180. package/src/commands/alias.ts +0 -100
  181. package/src/commands/analyze/index.ts +0 -352
  182. package/src/commands/apply/helm.ts +0 -473
  183. package/src/commands/apply/index.ts +0 -213
  184. package/src/commands/apply/k8s.ts +0 -454
  185. package/src/commands/apply/terraform.ts +0 -582
  186. package/src/commands/ask.ts +0 -167
  187. package/src/commands/audit/index.ts +0 -357
  188. package/src/commands/auth-cloud.ts +0 -407
  189. package/src/commands/auth-list.ts +0 -134
  190. package/src/commands/auth-profile.ts +0 -121
  191. package/src/commands/auth-refresh.ts +0 -187
  192. package/src/commands/auth-status.ts +0 -141
  193. package/src/commands/aws/ec2.ts +0 -501
  194. package/src/commands/aws/iam.ts +0 -397
  195. package/src/commands/aws/index.ts +0 -133
  196. package/src/commands/aws/lambda.ts +0 -396
  197. package/src/commands/aws/rds.ts +0 -439
  198. package/src/commands/aws/s3.ts +0 -439
  199. package/src/commands/aws/vpc.ts +0 -393
  200. package/src/commands/aws-discover.ts +0 -542
  201. package/src/commands/aws-terraform.ts +0 -755
  202. package/src/commands/azure/aks.ts +0 -376
  203. package/src/commands/azure/functions.ts +0 -253
  204. package/src/commands/azure/index.ts +0 -116
  205. package/src/commands/azure/storage.ts +0 -478
  206. package/src/commands/azure/vm.ts +0 -355
  207. package/src/commands/billing/index.ts +0 -256
  208. package/src/commands/chat.ts +0 -320
  209. package/src/commands/completions.ts +0 -268
  210. package/src/commands/config.ts +0 -372
  211. package/src/commands/cost/cloud-cost-estimator.ts +0 -266
  212. package/src/commands/cost/estimator.ts +0 -79
  213. package/src/commands/cost/index.ts +0 -810
  214. package/src/commands/cost/parsers/terraform.ts +0 -273
  215. package/src/commands/cost/parsers/types.ts +0 -25
  216. package/src/commands/cost/pricing/aws.ts +0 -544
  217. package/src/commands/cost/pricing/azure.ts +0 -499
  218. package/src/commands/cost/pricing/gcp.ts +0 -396
  219. package/src/commands/cost/pricing/index.ts +0 -40
  220. package/src/commands/demo.ts +0 -250
  221. package/src/commands/deploy.ts +0 -260
  222. package/src/commands/doctor.ts +0 -1386
  223. package/src/commands/drift/index.ts +0 -787
  224. package/src/commands/explain.ts +0 -277
  225. package/src/commands/export.ts +0 -146
  226. package/src/commands/feedback.ts +0 -389
  227. package/src/commands/fix.ts +0 -324
  228. package/src/commands/fs/index.ts +0 -402
  229. package/src/commands/gcp/compute.ts +0 -325
  230. package/src/commands/gcp/functions.ts +0 -271
  231. package/src/commands/gcp/gke.ts +0 -438
  232. package/src/commands/gcp/iam.ts +0 -344
  233. package/src/commands/gcp/index.ts +0 -129
  234. package/src/commands/gcp/storage.ts +0 -284
  235. package/src/commands/generate-helm.ts +0 -1249
  236. package/src/commands/generate-k8s.ts +0 -1508
  237. package/src/commands/generate-terraform.ts +0 -1202
  238. package/src/commands/gh/index.ts +0 -863
  239. package/src/commands/git/index.ts +0 -1343
  240. package/src/commands/helm/index.ts +0 -1126
  241. package/src/commands/help.ts +0 -715
  242. package/src/commands/history.ts +0 -149
  243. package/src/commands/import.ts +0 -868
  244. package/src/commands/incident.ts +0 -166
  245. package/src/commands/index.ts +0 -367
  246. package/src/commands/init.ts +0 -1051
  247. package/src/commands/k8s/index.ts +0 -1137
  248. package/src/commands/login.ts +0 -716
  249. package/src/commands/logout.ts +0 -83
  250. package/src/commands/logs.ts +0 -167
  251. package/src/commands/onboarding.ts +0 -405
  252. package/src/commands/pipeline.ts +0 -186
  253. package/src/commands/plan/display.ts +0 -279
  254. package/src/commands/plan/index.ts +0 -599
  255. package/src/commands/plugin.ts +0 -398
  256. package/src/commands/preview.ts +0 -452
  257. package/src/commands/profile.ts +0 -342
  258. package/src/commands/questionnaire.ts +0 -1172
  259. package/src/commands/resume.ts +0 -47
  260. package/src/commands/rollback.ts +0 -315
  261. package/src/commands/rollout.ts +0 -88
  262. package/src/commands/runbook.ts +0 -346
  263. package/src/commands/schedule.ts +0 -236
  264. package/src/commands/status.ts +0 -252
  265. package/src/commands/team/index.ts +0 -346
  266. package/src/commands/team-context.ts +0 -220
  267. package/src/commands/template.ts +0 -233
  268. package/src/commands/tf/index.ts +0 -1093
  269. package/src/commands/upgrade.ts +0 -607
  270. package/src/commands/usage/index.ts +0 -134
  271. package/src/commands/version.ts +0 -174
  272. package/src/commands/watch.ts +0 -153
  273. package/src/compat/index.ts +0 -2
  274. package/src/compat/runtime.ts +0 -12
  275. package/src/compat/sqlite.ts +0 -177
  276. package/src/config/index.ts +0 -17
  277. package/src/config/manager.ts +0 -530
  278. package/src/config/mode-store.ts +0 -62
  279. package/src/config/profiles.ts +0 -84
  280. package/src/config/safety-policy.ts +0 -358
  281. package/src/config/schema.ts +0 -125
  282. package/src/config/types.ts +0 -609
  283. package/src/config/workspace-state.ts +0 -53
  284. package/src/context/context-db.ts +0 -199
  285. package/src/demo/index.ts +0 -349
  286. package/src/demo/scenarios/full-journey.ts +0 -229
  287. package/src/demo/scenarios/getting-started.ts +0 -127
  288. package/src/demo/scenarios/helm-release.ts +0 -341
  289. package/src/demo/scenarios/k8s-deployment.ts +0 -194
  290. package/src/demo/scenarios/terraform-vpc.ts +0 -170
  291. package/src/demo/types.ts +0 -92
  292. package/src/engine/cost-estimator.ts +0 -480
  293. package/src/engine/diagram-generator.ts +0 -256
  294. package/src/engine/drift-detector.ts +0 -902
  295. package/src/engine/executor.ts +0 -1066
  296. package/src/engine/index.ts +0 -76
  297. package/src/engine/orchestrator.ts +0 -636
  298. package/src/engine/planner.ts +0 -787
  299. package/src/engine/safety.ts +0 -743
  300. package/src/engine/verifier.ts +0 -770
  301. package/src/enterprise/audit.ts +0 -348
  302. package/src/enterprise/auth.ts +0 -270
  303. package/src/enterprise/billing.ts +0 -822
  304. package/src/enterprise/index.ts +0 -17
  305. package/src/enterprise/teams.ts +0 -443
  306. package/src/generator/best-practices.ts +0 -1608
  307. package/src/generator/helm.ts +0 -630
  308. package/src/generator/index.ts +0 -37
  309. package/src/generator/intent-parser.ts +0 -514
  310. package/src/generator/kubernetes.ts +0 -976
  311. package/src/generator/terraform.ts +0 -1875
  312. package/src/history/index.ts +0 -8
  313. package/src/history/manager.ts +0 -250
  314. package/src/history/types.ts +0 -34
  315. package/src/hooks/config.ts +0 -432
  316. package/src/hooks/engine.ts +0 -392
  317. package/src/hooks/index.ts +0 -4
  318. package/src/llm/auth-bridge.ts +0 -198
  319. package/src/llm/circuit-breaker.ts +0 -140
  320. package/src/llm/config-loader.ts +0 -201
  321. package/src/llm/cost-calculator.ts +0 -171
  322. package/src/llm/index.ts +0 -8
  323. package/src/llm/model-aliases.ts +0 -115
  324. package/src/llm/provider-registry.ts +0 -63
  325. package/src/llm/providers/anthropic.ts +0 -462
  326. package/src/llm/providers/bedrock.ts +0 -477
  327. package/src/llm/providers/google.ts +0 -405
  328. package/src/llm/providers/ollama.ts +0 -767
  329. package/src/llm/providers/openai-compatible.ts +0 -340
  330. package/src/llm/providers/openai.ts +0 -328
  331. package/src/llm/providers/openrouter.ts +0 -338
  332. package/src/llm/router.ts +0 -1104
  333. package/src/llm/types.ts +0 -232
  334. package/src/lsp/client.ts +0 -298
  335. package/src/lsp/languages.ts +0 -119
  336. package/src/lsp/manager.ts +0 -294
  337. package/src/mcp/client.ts +0 -402
  338. package/src/mcp/index.ts +0 -5
  339. package/src/mcp/manager.ts +0 -133
  340. package/src/nimbus.ts +0 -233
  341. package/src/plugins/index.ts +0 -27
  342. package/src/plugins/loader.ts +0 -334
  343. package/src/plugins/manager.ts +0 -376
  344. package/src/plugins/types.ts +0 -284
  345. package/src/scanners/cicd-scanner.ts +0 -258
  346. package/src/scanners/cloud-scanner.ts +0 -466
  347. package/src/scanners/framework-scanner.ts +0 -469
  348. package/src/scanners/iac-scanner.ts +0 -388
  349. package/src/scanners/index.ts +0 -539
  350. package/src/scanners/language-scanner.ts +0 -276
  351. package/src/scanners/package-manager-scanner.ts +0 -277
  352. package/src/scanners/types.ts +0 -172
  353. package/src/sessions/manager.ts +0 -472
  354. package/src/sessions/types.ts +0 -44
  355. package/src/sharing/sync.ts +0 -300
  356. package/src/sharing/viewer.ts +0 -163
  357. package/src/snapshots/index.ts +0 -2
  358. package/src/snapshots/manager.ts +0 -530
  359. package/src/state/artifacts.ts +0 -147
  360. package/src/state/audit.ts +0 -137
  361. package/src/state/billing.ts +0 -240
  362. package/src/state/checkpoints.ts +0 -117
  363. package/src/state/config.ts +0 -67
  364. package/src/state/conversations.ts +0 -14
  365. package/src/state/credentials.ts +0 -154
  366. package/src/state/db.ts +0 -58
  367. package/src/state/index.ts +0 -26
  368. package/src/state/messages.ts +0 -115
  369. package/src/state/projects.ts +0 -123
  370. package/src/state/schema.ts +0 -236
  371. package/src/state/sessions.ts +0 -147
  372. package/src/state/teams.ts +0 -200
  373. package/src/telemetry.ts +0 -108
  374. package/src/tools/aws-ops.ts +0 -952
  375. package/src/tools/azure-ops.ts +0 -579
  376. package/src/tools/file-ops.ts +0 -615
  377. package/src/tools/gcp-ops.ts +0 -625
  378. package/src/tools/git-ops.ts +0 -773
  379. package/src/tools/github-ops.ts +0 -799
  380. package/src/tools/helm-ops.ts +0 -943
  381. package/src/tools/index.ts +0 -17
  382. package/src/tools/k8s-ops.ts +0 -819
  383. package/src/tools/schemas/converter.ts +0 -184
  384. package/src/tools/schemas/devops.ts +0 -3502
  385. package/src/tools/schemas/index.ts +0 -73
  386. package/src/tools/schemas/standard.ts +0 -1148
  387. package/src/tools/schemas/types.ts +0 -735
  388. package/src/tools/spawn-exec.ts +0 -148
  389. package/src/tools/terraform-ops.ts +0 -862
  390. package/src/types/ambient.d.ts +0 -193
  391. package/src/types/config.ts +0 -83
  392. package/src/types/drift.ts +0 -116
  393. package/src/types/enterprise.ts +0 -335
  394. package/src/types/index.ts +0 -20
  395. package/src/types/plan.ts +0 -44
  396. package/src/types/request.ts +0 -65
  397. package/src/types/response.ts +0 -54
  398. package/src/types/service.ts +0 -51
  399. package/src/ui/App.tsx +0 -2114
  400. package/src/ui/DeployPreview.tsx +0 -174
  401. package/src/ui/FileDiffModal.tsx +0 -162
  402. package/src/ui/Header.tsx +0 -131
  403. package/src/ui/HelpModal.tsx +0 -57
  404. package/src/ui/InputBox.tsx +0 -503
  405. package/src/ui/MessageList.tsx +0 -1032
  406. package/src/ui/PermissionPrompt.tsx +0 -163
  407. package/src/ui/StatusBar.tsx +0 -277
  408. package/src/ui/TerminalPane.tsx +0 -84
  409. package/src/ui/ToolCallDisplay.tsx +0 -643
  410. package/src/ui/TreePane.tsx +0 -132
  411. package/src/ui/chat-ui.ts +0 -850
  412. package/src/ui/index.ts +0 -33
  413. package/src/ui/ink/index.ts +0 -1444
  414. package/src/ui/streaming.ts +0 -176
  415. package/src/ui/theme.ts +0 -104
  416. package/src/ui/types.ts +0 -75
  417. package/src/utils/analytics.ts +0 -72
  418. package/src/utils/cost-warning.ts +0 -27
  419. package/src/utils/env.ts +0 -46
  420. package/src/utils/errors.ts +0 -69
  421. package/src/utils/event-bus.ts +0 -38
  422. package/src/utils/index.ts +0 -24
  423. package/src/utils/logger.ts +0 -171
  424. package/src/utils/rate-limiter.ts +0 -121
  425. package/src/utils/service-auth.ts +0 -49
  426. package/src/utils/validation.ts +0 -53
  427. package/src/version.ts +0 -4
  428. package/src/watcher/index.ts +0 -214
  429. package/src/wizard/approval.ts +0 -383
  430. package/src/wizard/index.ts +0 -25
  431. package/src/wizard/prompts.ts +0 -338
  432. package/src/wizard/types.ts +0 -172
  433. package/src/wizard/ui.ts +0 -556
  434. package/src/wizard/wizard.ts +0 -304
  435. package/tsconfig.json +0 -24
@@ -1,822 +0,0 @@
1
- /**
2
- * Enterprise Billing - Subscription management and usage tracking.
3
- *
4
- * Embedded replacement for services/billing-service.
5
- * All business logic is preserved verbatim from:
6
- * - services/billing-service/src/routes/subscriptions.ts
7
- * - services/billing-service/src/routes/usage.ts
8
- *
9
- * HTTP handlers, routes, and per-service SQLite are stripped.
10
- * State is read/written through the unified database via ../state/billing.
11
- */
12
-
13
- import {
14
- createSubscription as stateCreateSubscription,
15
- getSubscription as stateGetSubscription,
16
- updateSubscription as stateUpdateSubscription,
17
- recordUsage as stateRecordUsage,
18
- getUsage as stateGetUsage,
19
- getUsageSummary as stateGetUsageSummary,
20
- type SubscriptionRecord,
21
- type UsageRecord,
22
- type UsageSummary as StateUsageSummary,
23
- } from '../state/billing';
24
-
25
- // ---------------------------------------------------------------------------
26
- // Plan catalog (preserved verbatim from billing-service/src/routes/subscriptions.ts)
27
- // ---------------------------------------------------------------------------
28
-
29
- interface PlanDetails {
30
- name: string;
31
- stripe_price_id: string;
32
- amount_cents: number;
33
- currency: string;
34
- interval: 'month' | 'year';
35
- seats_included: number;
36
- features: string[];
37
- }
38
-
39
- const PLAN_CATALOG: Record<string, PlanDetails> = {
40
- free: {
41
- name: 'Nimbus Free',
42
- stripe_price_id: 'price_demo_free_monthly',
43
- amount_cents: 0,
44
- currency: 'usd',
45
- interval: 'month',
46
- seats_included: 5,
47
- features: [
48
- '5 team members',
49
- '100K tokens/month',
50
- 'Community support',
51
- 'Basic Terraform generation',
52
- ],
53
- },
54
- pro: {
55
- name: 'Nimbus Pro',
56
- stripe_price_id: 'price_demo_pro_monthly',
57
- amount_cents: 4900,
58
- currency: 'usd',
59
- interval: 'month',
60
- seats_included: 25,
61
- features: [
62
- '25 team members',
63
- '5M tokens/month',
64
- 'Priority support',
65
- 'Multi-cloud generation',
66
- 'Drift detection',
67
- 'Cost optimization',
68
- 'Helm & K8s generation',
69
- ],
70
- },
71
- enterprise: {
72
- name: 'Nimbus Enterprise',
73
- stripe_price_id: 'price_demo_enterprise_monthly',
74
- amount_cents: 19900,
75
- currency: 'usd',
76
- interval: 'month',
77
- seats_included: 100,
78
- features: [
79
- 'Unlimited team members',
80
- 'Unlimited tokens',
81
- 'Dedicated support & SLA',
82
- 'SSO / SAML integration',
83
- 'Audit log export',
84
- 'Custom policy engine',
85
- 'Private deployment option',
86
- 'Advanced RBAC',
87
- ],
88
- },
89
- };
90
-
91
- const VALID_PLANS = ['free', 'pro', 'enterprise'] as const;
92
- type ValidPlan = (typeof VALID_PLANS)[number];
93
-
94
- // ---------------------------------------------------------------------------
95
- // Plan quota definitions (preserved verbatim from billing-service/src/routes/usage.ts)
96
- // ---------------------------------------------------------------------------
97
-
98
- interface PlanQuota {
99
- tokensPerMonth: number;
100
- operationsPerMonth: number;
101
- costCapUsd: number;
102
- }
103
-
104
- const PLAN_QUOTAS: Record<string, PlanQuota> = {
105
- free: {
106
- tokensPerMonth: 100_000,
107
- operationsPerMonth: 500,
108
- costCapUsd: 0,
109
- },
110
- pro: {
111
- tokensPerMonth: 5_000_000,
112
- operationsPerMonth: 25_000,
113
- costCapUsd: 100,
114
- },
115
- enterprise: {
116
- tokensPerMonth: -1, // unlimited
117
- operationsPerMonth: -1,
118
- costCapUsd: -1,
119
- },
120
- };
121
-
122
- // ---------------------------------------------------------------------------
123
- // Response type definitions (mirrors @nimbus/shared-types shapes)
124
- // ---------------------------------------------------------------------------
125
-
126
- /** Alias for the billing-specific plan type, re-exported from teams.ts */
127
- type TeamPlan = ValidPlan;
128
-
129
- export interface BillingStatus {
130
- plan: TeamPlan;
131
- status: 'active' | 'canceled' | 'past_due' | 'trialing';
132
- currentPeriodStart: string;
133
- currentPeriodEnd: string;
134
- cancelAtPeriodEnd: boolean;
135
- seats: {
136
- used: number;
137
- total: number;
138
- };
139
- }
140
-
141
- export interface StripeSubscriptionResponse {
142
- billing: BillingStatus;
143
- stripe: {
144
- id: string;
145
- object: 'subscription';
146
- customer: string;
147
- status: string;
148
- current_period_start: number;
149
- current_period_end: number;
150
- created: number;
151
- cancel_at_period_end: boolean;
152
- canceled_at: number | null;
153
- plan: {
154
- id: string;
155
- object: 'plan';
156
- product: string;
157
- nickname: string;
158
- amount: number;
159
- currency: string;
160
- interval: string;
161
- interval_count: number;
162
- active: boolean;
163
- };
164
- items: {
165
- object: 'list';
166
- data: Array<{
167
- id: string;
168
- object: 'subscription_item';
169
- price: {
170
- id: string;
171
- object: 'price';
172
- unit_amount: number;
173
- currency: string;
174
- recurring: { interval: string; interval_count: number };
175
- product: string;
176
- };
177
- quantity: number;
178
- }>;
179
- total_count: number;
180
- };
181
- latest_invoice: {
182
- id: string;
183
- object: 'invoice';
184
- number: string;
185
- status: 'draft' | 'open' | 'paid' | 'void';
186
- amount_due: number;
187
- amount_paid: number;
188
- currency: string;
189
- customer_email: string;
190
- period_start: number;
191
- period_end: number;
192
- subtotal: number;
193
- tax: number;
194
- total: number;
195
- lines: {
196
- object: 'list';
197
- data: Array<{
198
- id: string;
199
- object: 'line_item';
200
- description: string;
201
- amount: number;
202
- currency: string;
203
- quantity: number;
204
- period: { start: number; end: number };
205
- }>;
206
- total_count: number;
207
- };
208
- payment_intent: {
209
- id: string;
210
- object: 'payment_intent';
211
- status: 'succeeded' | 'requires_payment_method';
212
- amount: number;
213
- currency: string;
214
- };
215
- hosted_invoice_url: string;
216
- invoice_pdf: string;
217
- created: number;
218
- };
219
- metadata: { team_id: string; plan_name: string; provisioned_by: string };
220
- default_payment_method: {
221
- id: string;
222
- object: 'payment_method';
223
- type: 'card';
224
- card: { brand: string; last4: string; exp_month: number; exp_year: number; funding: string };
225
- };
226
- };
227
- }
228
-
229
- export interface StripeCancellationResponse {
230
- billing: BillingStatus;
231
- cancellation: {
232
- id: string;
233
- object: 'subscription';
234
- status: 'canceled';
235
- canceled_at: number;
236
- cancel_at_period_end: boolean;
237
- current_period_end: number;
238
- ended_at: number | null;
239
- cancellation_details: {
240
- comment: string | null;
241
- feedback: string | null;
242
- reason: 'cancellation_requested';
243
- };
244
- };
245
- }
246
-
247
- export interface SubscribeRequest {
248
- teamId: string;
249
- plan: string;
250
- paymentMethodId?: string;
251
- seats?: number;
252
- }
253
-
254
- export interface RecordUsageRequest {
255
- teamId: string;
256
- userId?: string;
257
- operationType: string;
258
- tokensUsed: number;
259
- costUsd: number;
260
- }
261
-
262
- export interface UsageRecordConfirmation {
263
- recorded: true;
264
- id: string;
265
- timestamp: string;
266
- teamId: string;
267
- operationType: string;
268
- tokensUsed: number;
269
- costUsd: number;
270
- }
271
-
272
- export interface EnhancedUsageSummary {
273
- period: { start: string; end: string };
274
- totals: { operations: number; tokensUsed: number; costUsd: number };
275
- byOperationType: Record<string, { count: number; tokensUsed: number; costUsd: number }>;
276
- byUser?: Record<string, { count: number; tokensUsed: number; costUsd: number }>;
277
- dailyBreakdown: Array<{ date: string; operations: number; tokensUsed: number; costUsd: number }>;
278
- quota: {
279
- plan: string;
280
- tokens: {
281
- used: number;
282
- limit: number;
283
- remaining: number;
284
- percentUsed: number;
285
- unlimited: boolean;
286
- };
287
- operations: {
288
- used: number;
289
- limit: number;
290
- remaining: number;
291
- percentUsed: number;
292
- unlimited: boolean;
293
- };
294
- cost: {
295
- accrued: number;
296
- cap: number;
297
- remaining: number;
298
- percentUsed: number;
299
- unlimited: boolean;
300
- };
301
- };
302
- rateLimit: {
303
- requestsPerMinute: number;
304
- tokensPerMinute: number;
305
- currentMinuteRequests: number;
306
- currentMinuteTokens: number;
307
- };
308
- }
309
-
310
- // ---------------------------------------------------------------------------
311
- // Private helpers
312
- // ---------------------------------------------------------------------------
313
-
314
- /**
315
- * Generate a random alphanumeric demo ID of the given length.
316
- * Preserved verbatim from billing-service/src/db/adapter.ts.
317
- */
318
- function generateDemoId(length: number): string {
319
- const chars = 'abcdefghijklmnopqrstuvwxyz0123456789';
320
- let result = '';
321
- for (let i = 0; i < length; i++) {
322
- result += chars.charAt(Math.floor(Math.random() * chars.length));
323
- }
324
- return result;
325
- }
326
-
327
- /**
328
- * Build a realistic Stripe-format subscription object for demo presentations.
329
- * Preserved verbatim from billing-service/src/routes/subscriptions.ts.
330
- */
331
- function buildStripeSubscription(
332
- teamId: string,
333
- plan: string,
334
- stripeSubId: string,
335
- stripeCustomerId: string,
336
- periodStart: Date,
337
- periodEnd: Date,
338
- seats: number,
339
- canceled: boolean = false,
340
- canceledAt: Date | null = null
341
- ): StripeSubscriptionResponse['stripe'] {
342
- const planDetails = PLAN_CATALOG[plan] || PLAN_CATALOG.pro;
343
- const createdTimestamp = Math.floor((periodStart.getTime() - 5000) / 1000);
344
- const periodStartUnix = Math.floor(periodStart.getTime() / 1000);
345
- const periodEndUnix = Math.floor(periodEnd.getTime() / 1000);
346
- const invoiceId = `in_demo_${generateDemoId(24)}`;
347
- const invoiceNumber = `NIM-${periodStart.getFullYear()}${String(periodStart.getMonth() + 1).padStart(2, '0')}-${String(Math.floor(Math.random() * 9999) + 1).padStart(4, '0')}`;
348
- const seatAmount = planDetails.amount_cents * seats;
349
-
350
- return {
351
- id: stripeSubId,
352
- object: 'subscription',
353
- customer: stripeCustomerId,
354
- status: canceled ? 'canceled' : 'active',
355
- current_period_start: periodStartUnix,
356
- current_period_end: periodEndUnix,
357
- created: createdTimestamp,
358
- cancel_at_period_end: canceled,
359
- canceled_at: canceledAt ? Math.floor(canceledAt.getTime() / 1000) : null,
360
- plan: {
361
- id: planDetails.stripe_price_id,
362
- object: 'plan',
363
- product: `prod_demo_nimbus_${plan}`,
364
- nickname: planDetails.name,
365
- amount: planDetails.amount_cents,
366
- currency: planDetails.currency,
367
- interval: planDetails.interval,
368
- interval_count: 1,
369
- active: true,
370
- },
371
- items: {
372
- object: 'list',
373
- data: [
374
- {
375
- id: `si_demo_${generateDemoId(14)}`,
376
- object: 'subscription_item',
377
- price: {
378
- id: planDetails.stripe_price_id,
379
- object: 'price',
380
- unit_amount: planDetails.amount_cents,
381
- currency: planDetails.currency,
382
- recurring: { interval: planDetails.interval, interval_count: 1 },
383
- product: `prod_demo_nimbus_${plan}`,
384
- },
385
- quantity: seats,
386
- },
387
- ],
388
- total_count: 1,
389
- },
390
- latest_invoice: {
391
- id: invoiceId,
392
- object: 'invoice',
393
- number: invoiceNumber,
394
- status: 'paid',
395
- amount_due: seatAmount,
396
- amount_paid: seatAmount,
397
- currency: planDetails.currency,
398
- customer_email: `billing+${teamId}@nimbus.dev`,
399
- period_start: periodStartUnix,
400
- period_end: periodEndUnix,
401
- subtotal: seatAmount,
402
- tax: 0,
403
- total: seatAmount,
404
- lines: {
405
- object: 'list',
406
- data: [
407
- {
408
- id: `il_demo_${generateDemoId(14)}`,
409
- object: 'line_item',
410
- description: `${planDetails.name} (${seats} seat${seats !== 1 ? 's' : ''} x $${(planDetails.amount_cents / 100).toFixed(2)}/mo)`,
411
- amount: seatAmount,
412
- currency: planDetails.currency,
413
- quantity: seats,
414
- period: { start: periodStartUnix, end: periodEndUnix },
415
- },
416
- ],
417
- total_count: 1,
418
- },
419
- payment_intent: {
420
- id: `pi_demo_${generateDemoId(24)}`,
421
- object: 'payment_intent',
422
- status: 'succeeded',
423
- amount: seatAmount,
424
- currency: planDetails.currency,
425
- },
426
- hosted_invoice_url: `https://invoice.stripe.com/i/demo/${invoiceId}`,
427
- invoice_pdf: `https://pay.stripe.com/invoice/${invoiceId}/pdf`,
428
- created: createdTimestamp,
429
- },
430
- metadata: {
431
- team_id: teamId,
432
- plan_name: planDetails.name,
433
- provisioned_by: 'nimbus-billing-service',
434
- },
435
- default_payment_method: {
436
- id: `pm_demo_${generateDemoId(14)}`,
437
- object: 'payment_method',
438
- type: 'card',
439
- card: {
440
- brand: 'visa',
441
- last4: '4242',
442
- exp_month: 12,
443
- exp_year: new Date().getFullYear() + 2,
444
- funding: 'credit',
445
- },
446
- },
447
- };
448
- }
449
-
450
- /**
451
- * Build quota status relative to the team's plan.
452
- * Preserved verbatim from billing-service/src/routes/usage.ts.
453
- */
454
- function buildQuota(
455
- plan: string,
456
- totalTokens: number,
457
- totalOperations: number,
458
- totalCost: number
459
- ): EnhancedUsageSummary['quota'] {
460
- const quota = PLAN_QUOTAS[plan] || PLAN_QUOTAS.free;
461
- const unlimited = plan === 'enterprise';
462
-
463
- const tokensLimit = quota.tokensPerMonth;
464
- const opsLimit = quota.operationsPerMonth;
465
- const costCap = quota.costCapUsd;
466
-
467
- return {
468
- plan,
469
- tokens: {
470
- used: totalTokens,
471
- limit: unlimited ? -1 : tokensLimit,
472
- remaining: unlimited ? -1 : Math.max(0, tokensLimit - totalTokens),
473
- percentUsed: unlimited
474
- ? 0
475
- : tokensLimit > 0
476
- ? Math.min(100, Math.round((totalTokens / tokensLimit) * 100 * 100) / 100)
477
- : 0,
478
- unlimited,
479
- },
480
- operations: {
481
- used: totalOperations,
482
- limit: unlimited ? -1 : opsLimit,
483
- remaining: unlimited ? -1 : Math.max(0, opsLimit - totalOperations),
484
- percentUsed: unlimited
485
- ? 0
486
- : opsLimit > 0
487
- ? Math.min(100, Math.round((totalOperations / opsLimit) * 100 * 100) / 100)
488
- : 0,
489
- unlimited,
490
- },
491
- cost: {
492
- accrued: Math.round(totalCost * 100) / 100,
493
- cap: unlimited ? -1 : costCap,
494
- remaining: unlimited ? -1 : Math.max(0, Math.round((costCap - totalCost) * 100) / 100),
495
- percentUsed: unlimited
496
- ? 0
497
- : costCap > 0
498
- ? Math.min(100, Math.round((totalCost / costCap) * 100 * 100) / 100)
499
- : 0,
500
- unlimited,
501
- },
502
- };
503
- }
504
-
505
- /**
506
- * Build rate limit metadata for the current billing window.
507
- * Preserved verbatim from billing-service/src/routes/usage.ts.
508
- */
509
- function buildRateLimit(plan: string): EnhancedUsageSummary['rateLimit'] {
510
- const limits: Record<string, { rpm: number; tpm: number }> = {
511
- free: { rpm: 30, tpm: 10_000 },
512
- pro: { rpm: 120, tpm: 100_000 },
513
- enterprise: { rpm: 600, tpm: 1_000_000 },
514
- };
515
-
516
- const planLimits = limits[plan] || limits.free;
517
-
518
- return {
519
- requestsPerMinute: planLimits.rpm,
520
- tokensPerMinute: planLimits.tpm,
521
- // Simulated current-minute counters (low values for demo)
522
- currentMinuteRequests: Math.floor(Math.random() * 5),
523
- currentMinuteTokens: Math.floor(Math.random() * 2000),
524
- };
525
- }
526
-
527
- // ---------------------------------------------------------------------------
528
- // Public API - Subscriptions
529
- // ---------------------------------------------------------------------------
530
-
531
- /**
532
- * Get the current billing status for a team.
533
- *
534
- * Returns free-plan defaults when no subscription record exists.
535
- */
536
- export async function getBillingStatus(teamId: string): Promise<BillingStatus> {
537
- const subscription: SubscriptionRecord | null = stateGetSubscription(teamId);
538
-
539
- if (!subscription) {
540
- return {
541
- plan: 'free',
542
- status: 'active',
543
- currentPeriodStart: new Date().toISOString(),
544
- currentPeriodEnd: new Date(Date.now() + 30 * 24 * 60 * 60 * 1000).toISOString(),
545
- cancelAtPeriodEnd: false,
546
- seats: {
547
- used: 1,
548
- total: 5,
549
- },
550
- };
551
- }
552
-
553
- // The unified billing schema (src/state/billing.ts) stores status directly
554
- // and does not have cancel_at_period_end or seats_used/seats_total columns.
555
- // We derive cancelAtPeriodEnd from status === 'canceled' for backward
556
- // compatibility with the original service response shape.
557
- return {
558
- plan: subscription.plan as TeamPlan,
559
- status: subscription.status as BillingStatus['status'],
560
- currentPeriodStart: subscription.currentPeriodStart || new Date().toISOString(),
561
- currentPeriodEnd:
562
- subscription.currentPeriodEnd ||
563
- new Date(Date.now() + 30 * 24 * 60 * 60 * 1000).toISOString(),
564
- cancelAtPeriodEnd: subscription.status === 'canceled',
565
- seats: {
566
- used: 1, // seat tracking not available in the unified schema; caller can augment
567
- total: PLAN_CATALOG[subscription.plan]?.seats_included ?? 5,
568
- },
569
- };
570
- }
571
-
572
- /**
573
- * Subscribe a team to a plan.
574
- *
575
- * Creates or updates the subscription record in the unified database and
576
- * returns a rich Stripe-like subscription object for demo presentations.
577
- */
578
- export async function subscribe(request: SubscribeRequest): Promise<StripeSubscriptionResponse> {
579
- const { teamId, plan, seats } = request;
580
-
581
- if (!teamId) {
582
- throw new Error('Team ID is required');
583
- }
584
-
585
- if (!VALID_PLANS.includes(plan as ValidPlan)) {
586
- throw new Error(`Invalid plan: ${plan}. Must be one of: ${VALID_PLANS.join(', ')}`);
587
- }
588
-
589
- const stripeSubscriptionId = `sub_demo_${generateDemoId(24)}`;
590
- const periodStart = new Date();
591
- const periodEnd = new Date(Date.now() + 30 * 24 * 60 * 60 * 1000);
592
- const seatsTotal = seats || (plan === 'enterprise' ? 100 : plan === 'pro' ? 25 : 5);
593
- const stripeCustomerId = `cus_demo_${generateDemoId(14)}`;
594
-
595
- const existing = stateGetSubscription(teamId);
596
-
597
- if (existing) {
598
- stateUpdateSubscription(teamId, {
599
- plan,
600
- status: 'active',
601
- currentPeriodStart: periodStart.toISOString(),
602
- currentPeriodEnd: periodEnd.toISOString(),
603
- });
604
- } else {
605
- stateCreateSubscription(
606
- crypto.randomUUID(),
607
- teamId,
608
- plan,
609
- 'active',
610
- periodStart.toISOString(),
611
- periodEnd.toISOString()
612
- );
613
- }
614
-
615
- const billingStatus = await getBillingStatus(teamId);
616
-
617
- const stripeObject = buildStripeSubscription(
618
- teamId,
619
- plan,
620
- stripeSubscriptionId,
621
- stripeCustomerId,
622
- periodStart,
623
- periodEnd,
624
- seatsTotal
625
- );
626
-
627
- return {
628
- billing: billingStatus,
629
- stripe: stripeObject,
630
- };
631
- }
632
-
633
- /**
634
- * Cancel a team's active subscription.
635
- *
636
- * Marks the subscription as canceled in the unified database and returns a
637
- * Stripe-like cancellation confirmation object.
638
- */
639
- export async function cancelSubscription(teamId: string): Promise<StripeCancellationResponse> {
640
- if (!teamId) {
641
- throw new Error('Team ID is required');
642
- }
643
-
644
- const subscription = stateGetSubscription(teamId);
645
- if (!subscription) {
646
- throw new Error('No subscription found');
647
- }
648
-
649
- if (subscription.plan === 'free') {
650
- throw new Error('Cannot cancel free plan');
651
- }
652
-
653
- const canceledAt = new Date();
654
-
655
- stateUpdateSubscription(teamId, { status: 'canceled' });
656
-
657
- const billingStatus = await getBillingStatus(teamId);
658
-
659
- const periodEnd = subscription.currentPeriodEnd
660
- ? new Date(subscription.currentPeriodEnd)
661
- : new Date(Date.now() + 30 * 24 * 60 * 60 * 1000);
662
-
663
- return {
664
- billing: billingStatus,
665
- cancellation: {
666
- id: `sub_demo_${generateDemoId(24)}`,
667
- object: 'subscription',
668
- status: 'canceled',
669
- canceled_at: Math.floor(canceledAt.getTime() / 1000),
670
- cancel_at_period_end: true,
671
- current_period_end: Math.floor(periodEnd.getTime() / 1000),
672
- ended_at: null,
673
- cancellation_details: {
674
- comment: null,
675
- feedback: null,
676
- reason: 'cancellation_requested',
677
- },
678
- },
679
- };
680
- }
681
-
682
- // ---------------------------------------------------------------------------
683
- // Public API - Usage tracking
684
- // ---------------------------------------------------------------------------
685
-
686
- /**
687
- * Record a usage event for a team.
688
- *
689
- * Validates numeric inputs and returns a confirmation receipt with the
690
- * generated record ID.
691
- */
692
- export async function recordUsage(request: RecordUsageRequest): Promise<UsageRecordConfirmation> {
693
- const { teamId, userId, operationType, tokensUsed, costUsd } = request;
694
-
695
- if (!teamId || !operationType) {
696
- throw new Error('Team ID and operation type are required');
697
- }
698
-
699
- if (!Number.isFinite(tokensUsed) || tokensUsed < 0) {
700
- throw new Error('tokensUsed must be a non-negative number');
701
- }
702
- if (!Number.isFinite(costUsd) || costUsd < 0) {
703
- throw new Error('costUsd must be a non-negative number');
704
- }
705
-
706
- const id = crypto.randomUUID();
707
-
708
- // stateRecordUsage(id, type, quantity, unit, costUsd, teamId, userId, metadata?)
709
- stateRecordUsage(id, operationType, tokensUsed, 'tokens', costUsd, teamId, userId);
710
-
711
- return {
712
- recorded: true,
713
- id,
714
- timestamp: new Date().toISOString(),
715
- teamId,
716
- operationType,
717
- tokensUsed,
718
- costUsd,
719
- };
720
- }
721
-
722
- /**
723
- * Get a usage summary for a team over a given period.
724
- *
725
- * Returns an enhanced summary with daily breakdown, quota status relative to
726
- * the team's current plan, and rate limit metadata.
727
- */
728
- export async function getUsage(
729
- teamId: string,
730
- period: 'day' | 'week' | 'month' = 'month'
731
- ): Promise<EnhancedUsageSummary> {
732
- const now = new Date();
733
- let since: Date;
734
-
735
- switch (period) {
736
- case 'day':
737
- since = new Date(now.getTime() - 24 * 60 * 60 * 1000);
738
- break;
739
- case 'week':
740
- since = new Date(now.getTime() - 7 * 24 * 60 * 60 * 1000);
741
- break;
742
- case 'month':
743
- default:
744
- since = new Date(now.getTime() - 30 * 24 * 60 * 60 * 1000);
745
- break;
746
- }
747
-
748
- // The unified billing state module provides getUsageSummary() aggregated by
749
- // type, and getUsage() for raw records. We use getUsageSummary() for
750
- // byOperationType totals and getUsage() for per-user and daily breakdowns.
751
- const summaryRows: StateUsageSummary[] = stateGetUsageSummary(teamId, since, now);
752
- const rawRecords: UsageRecord[] = stateGetUsage(teamId, since, now, 10000, 0);
753
-
754
- // Aggregate totals and byOperationType from summary rows
755
- let totalOperations = 0;
756
- let totalTokens = 0;
757
- let totalCost = 0;
758
- const byOperationType: Record<string, { count: number; tokensUsed: number; costUsd: number }> =
759
- {};
760
-
761
- for (const row of summaryRows) {
762
- totalOperations += row.count;
763
- totalTokens += row.totalQuantity;
764
- totalCost += row.totalCost;
765
-
766
- byOperationType[row.type] = {
767
- count: row.count,
768
- tokensUsed: row.totalQuantity,
769
- costUsd: row.totalCost,
770
- };
771
- }
772
-
773
- // Aggregate byUser from raw records
774
- const byUser: Record<string, { count: number; tokensUsed: number; costUsd: number }> = {};
775
- for (const rec of rawRecords) {
776
- if (rec.userId) {
777
- const existing = byUser[rec.userId] ?? { count: 0, tokensUsed: 0, costUsd: 0 };
778
- byUser[rec.userId] = {
779
- count: existing.count + 1,
780
- tokensUsed: existing.tokensUsed + rec.quantity,
781
- costUsd: existing.costUsd + (rec.costUsd ?? 0),
782
- };
783
- }
784
- }
785
-
786
- // Build daily breakdown from raw records (group by calendar date)
787
- const dailyMap: Record<string, { operations: number; tokensUsed: number; costUsd: number }> = {};
788
- for (const rec of rawRecords) {
789
- const date = rec.createdAt.slice(0, 10); // "YYYY-MM-DD"
790
- const existing = dailyMap[date] ?? { operations: 0, tokensUsed: 0, costUsd: 0 };
791
- dailyMap[date] = {
792
- operations: existing.operations + 1,
793
- tokensUsed: existing.tokensUsed + rec.quantity,
794
- costUsd: existing.costUsd + (rec.costUsd ?? 0),
795
- };
796
- }
797
-
798
- const dailyBreakdown = Object.entries(dailyMap)
799
- .sort(([a], [b]) => a.localeCompare(b))
800
- .map(([date, agg]) => ({ date, ...agg }));
801
-
802
- // Resolve the team's current plan for quota calculation
803
- const subscription = stateGetSubscription(teamId);
804
- const plan = subscription?.plan || 'free';
805
-
806
- return {
807
- period: {
808
- start: since.toISOString(),
809
- end: now.toISOString(),
810
- },
811
- totals: {
812
- operations: totalOperations,
813
- tokensUsed: totalTokens,
814
- costUsd: Math.round(totalCost * 100) / 100,
815
- },
816
- byOperationType,
817
- byUser: Object.keys(byUser).length > 0 ? byUser : undefined,
818
- dailyBreakdown,
819
- quota: buildQuota(plan, totalTokens, totalOperations, totalCost),
820
- rateLimit: buildRateLimit(plan),
821
- };
822
- }