mirror of
https://github.com/AgentSeal/codeburn.git
synced 2026-04-28 06:59:37 +00:00
553cf2d706
Adds `codeburn plan set <id>` to configure a subscription plan (Claude Pro, Claude Max, Cursor Pro, or custom). When set, the Overview panel renders an API-equivalent progress bar against subscription price with a projected month-end cost. Closes the loudest demand signal on the repo: issue #11 ("Subscription vs API Use") from two independent voices, plus the routing-decision use case raised in #12. - src/config.ts: extends CodeburnConfig with Plan, adds readPlan/savePlan/clearPlan - src/plans.ts: presets (claude-pro $20, claude-max $200, cursor-pro $20) - src/plan-usage.ts: getPlanUsage, resetDay-aware period math (1-28), median-of-7-day-trailing projection - src/cli.ts: `codeburn plan [show|set|reset]` subcommand, plan wired into JSON outputs for report/today/month/status (only when active) - src/dashboard.tsx: Plan row in Overview, color-coded (green under 80%, orange near, red over), with days-until-reset - README.md: Plans section with honest framing (API-equivalent vs subscription price, not token allowance) - tests/plan-usage.test.ts, tests/plans.test.ts, tests/cli-plan.test.ts: period math, presets, CLI round-trip Resets respect resetDay across month boundaries. Uses median daily spend (not mean) so one huge day doesn't distort the month-end projection. Fixes #11 Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
63 lines
2 KiB
TypeScript
63 lines
2 KiB
TypeScript
import { mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
|
|
import { describe, it, expect } from 'vitest'
|
|
|
|
import { clearPlan, readPlan, savePlan } from '../src/config.js'
|
|
import { getPresetPlan, isPlanId, isPlanProvider } from '../src/plans.js'
|
|
|
|
describe('plan presets', () => {
|
|
it('resolves builtin presets', () => {
|
|
expect(getPresetPlan('claude-pro')).toMatchObject({ id: 'claude-pro', monthlyUsd: 20, provider: 'claude' })
|
|
expect(getPresetPlan('claude-max')).toMatchObject({ id: 'claude-max', monthlyUsd: 200, provider: 'claude' })
|
|
expect(getPresetPlan('cursor-pro')).toMatchObject({ id: 'cursor-pro', monthlyUsd: 20, provider: 'cursor' })
|
|
expect(getPresetPlan('custom')).toBeNull()
|
|
})
|
|
|
|
it('validates ids and providers', () => {
|
|
expect(isPlanId('claude-pro')).toBe(true)
|
|
expect(isPlanId('none')).toBe(true)
|
|
expect(isPlanId('bad-plan')).toBe(false)
|
|
|
|
expect(isPlanProvider('all')).toBe(true)
|
|
expect(isPlanProvider('claude')).toBe(true)
|
|
expect(isPlanProvider('invalid')).toBe(false)
|
|
})
|
|
})
|
|
|
|
describe('plan config persistence', () => {
|
|
it('round-trips savePlan/readPlan and clearPlan', async () => {
|
|
const dir = await mkdtemp(join(tmpdir(), 'codeburn-plan-test-'))
|
|
const previousHome = process.env['HOME']
|
|
process.env['HOME'] = dir
|
|
|
|
try {
|
|
await savePlan({
|
|
id: 'claude-max',
|
|
monthlyUsd: 200,
|
|
provider: 'claude',
|
|
resetDay: 12,
|
|
setAt: '2026-04-17T12:00:00.000Z',
|
|
})
|
|
|
|
const plan = await readPlan()
|
|
expect(plan).toMatchObject({
|
|
id: 'claude-max',
|
|
monthlyUsd: 200,
|
|
provider: 'claude',
|
|
resetDay: 12,
|
|
})
|
|
|
|
await clearPlan()
|
|
expect(await readPlan()).toBeUndefined()
|
|
} finally {
|
|
if (previousHome === undefined) {
|
|
delete process.env['HOME']
|
|
} else {
|
|
process.env['HOME'] = previousHome
|
|
}
|
|
await rm(dir, { recursive: true, force: true })
|
|
}
|
|
})
|
|
})
|