aiblueprint-cli 1.4.13 → 1.4.14

package/claude-code-config/scripts/statusline/__tests__/statusline.test.ts

@@ -0,0 +1,309 @@
+import { describe, expect, it } from "bun:test";
+import { renderStatusline, type StatuslineData } from "../src/index";
+import { defaultConfig, type StatuslineConfig } from "../src/lib/config";
+
+function createMockData(
+	overrides: Partial<StatuslineData> = {},
+): StatuslineData {
+	return {
+		branch: "main",
+		dirPath: "~/project",
+		modelName: "Sonnet 4.5",
+		sessionCost: "0.5",
+		sessionDuration: "10m",
+		contextTokens: 50000,
+		contextPercentage: 25,
+		usageLimits: {
+			five_hour: { utilization: 30, resets_at: "2025-01-01T15:00:00Z" },
+			seven_day: { utilization: 10, resets_at: "2025-01-07T00:00:00Z" },
+		},
+		periodCost: 1.5,
+		todayCost: 2.0,
+		...overrides,
+	};
+}
+
+function createConfig(
+	overrides: Partial<StatuslineConfig> = {},
+): StatuslineConfig {
+	return { ...defaultConfig, ...overrides };
+}
+
+describe("renderStatusline", () => {
+	describe("basic rendering", () => {
+		it("should render branch and directory", () => {
+			const data = createMockData();
+			const config = createConfig();
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("main");
+			expect(output).toContain("~/project");
+		});
+
+		it("should hide Sonnet model name when showSonnetModel is false", () => {
+			const data = createMockData({ modelName: "Sonnet 4.5" });
+			const config = createConfig({ showSonnetModel: false });
+			const output = renderStatusline(data, config);
+
+			expect(output).not.toContain("Sonnet");
+		});
+
+		it("should show Sonnet model name when showSonnetModel is true", () => {
+			const data = createMockData({ modelName: "Sonnet 4.5" });
+			const config = createConfig({ showSonnetModel: true });
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("Sonnet");
+		});
+
+		it("should always show non-Sonnet models", () => {
+			const data = createMockData({ modelName: "Opus 4.5" });
+			const config = createConfig({ showSonnetModel: false });
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("Opus");
+		});
+	});
+
+	describe("session info", () => {
+		it("should show cost when enabled", () => {
+			const data = createMockData({ sessionCost: "1.5" });
+			const config = createConfig({
+				session: {
+					...defaultConfig.session,
+					cost: { enabled: true, format: "decimal1" },
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("$");
+			expect(output).toContain("1.5");
+		});
+
+		it("should hide cost when disabled", () => {
+			const data = createMockData({ sessionCost: "1.5" });
+			const config = createConfig({
+				session: {
+					...defaultConfig.session,
+					cost: { enabled: false, format: "decimal1" },
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("S:");
+		});
+
+		it("should show duration when enabled", () => {
+			const data = createMockData({ sessionDuration: "15m" });
+			const config = createConfig({
+				session: {
+					...defaultConfig.session,
+					duration: { enabled: true },
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("15m");
+		});
+
+		it("should show percentage when enabled", () => {
+			const data = createMockData({ contextPercentage: 45 });
+			const config = createConfig({
+				session: {
+					...defaultConfig.session,
+					percentage: {
+						enabled: true,
+						showValue: true,
+						progressBar: {
+							...defaultConfig.session.percentage.progressBar,
+							enabled: false,
+						},
+					},
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("45");
+			expect(output).toContain("%");
+		});
+	});
+
+	describe("limits section", () => {
+		it("should show limits when enabled", () => {
+			const data = createMockData({
+				usageLimits: {
+					five_hour: { utilization: 50, resets_at: "2025-01-01T15:00:00Z" },
+					seven_day: null,
+				},
+			});
+			const config = createConfig({
+				limits: {
+					...defaultConfig.limits,
+					enabled: true,
+					percentage: {
+						enabled: true,
+						showValue: true,
+						progressBar: {
+							...defaultConfig.limits.percentage.progressBar,
+							enabled: false,
+						},
+					},
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("L:");
+			expect(output).toContain("50");
+		});
+
+		it("should hide limits when disabled", () => {
+			const data = createMockData();
+			const config = createConfig({
+				limits: { ...defaultConfig.limits, enabled: false },
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).not.toContain("L:");
+		});
+
+		it("should show reset time when enabled", () => {
+			const futureTime = new Date(Date.now() + 3600000).toISOString();
+			const data = createMockData({
+				usageLimits: {
+					five_hour: { utilization: 50, resets_at: futureTime },
+					seven_day: null,
+				},
+			});
+			const config = createConfig({
+				limits: {
+					...defaultConfig.limits,
+					enabled: true,
+					showTimeLeft: true,
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toMatch(/\(\d+[hm]/);
+		});
+	});
+
+	describe("weekly usage", () => {
+		it("should show weekly when enabled=true", () => {
+			const data = createMockData({
+				usageLimits: {
+					five_hour: { utilization: 30, resets_at: null },
+					seven_day: { utilization: 20, resets_at: null },
+				},
+			});
+			const config = createConfig({
+				weeklyUsage: {
+					...defaultConfig.weeklyUsage,
+					enabled: true,
+					percentage: {
+						enabled: true,
+						showValue: true,
+						progressBar: {
+							...defaultConfig.weeklyUsage.percentage.progressBar,
+							enabled: false,
+						},
+					},
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("W:");
+		});
+
+		it("should show weekly when enabled=90% and five_hour >= 90", () => {
+			const data = createMockData({
+				usageLimits: {
+					five_hour: { utilization: 95, resets_at: null },
+					seven_day: { utilization: 20, resets_at: null },
+				},
+			});
+			const config = createConfig({
+				weeklyUsage: {
+					...defaultConfig.weeklyUsage,
+					enabled: "90%",
+					percentage: {
+						enabled: true,
+						showValue: true,
+						progressBar: {
+							...defaultConfig.weeklyUsage.percentage.progressBar,
+							enabled: false,
+						},
+					},
+				},
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("W:");
+		});
+
+		it("should hide weekly when enabled=90% and five_hour < 90", () => {
+			const data = createMockData({
+				usageLimits: {
+					five_hour: { utilization: 50, resets_at: null },
+					seven_day: { utilization: 20, resets_at: null },
+				},
+			});
+			const config = createConfig({
+				weeklyUsage: { ...defaultConfig.weeklyUsage, enabled: "90%" },
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).not.toContain("W:");
+		});
+	});
+
+	describe("daily spend", () => {
+		it("should show daily cost when enabled and > 0", () => {
+			const data = createMockData({ todayCost: 5.5 });
+			const config = createConfig({
+				dailySpend: { cost: { enabled: true, format: "decimal1" } },
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("D:");
+			expect(output).toContain("5.5");
+		});
+
+		it("should hide daily cost when todayCost is 0", () => {
+			const data = createMockData({ todayCost: 0 });
+			const config = createConfig({
+				dailySpend: { cost: { enabled: true, format: "decimal1" } },
+			});
+			const output = renderStatusline(data, config);
+
+			expect(output).not.toContain("D:");
+		});
+	});
+
+	describe("separators", () => {
+		it("should use configured separator", () => {
+			const data = createMockData();
+			const config = createConfig({ separator: "|" });
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("|");
+		});
+	});
+
+	describe("one line vs two lines", () => {
+		it("should render on one line when oneLine is true", () => {
+			const data = createMockData();
+			const config = createConfig({ oneLine: true });
+			const output = renderStatusline(data, config);
+
+			expect(output.split("\n").length).toBe(1);
+		});
+
+		it("should render on two lines when oneLine is false", () => {
+			const data = createMockData();
+			const config = createConfig({ oneLine: false });
+			const output = renderStatusline(data, config);
+
+			expect(output).toContain("\n");
+		});
+	});
+});

package/claude-code-config/scripts/statusline/data/.gitignore

@@ -0,0 +1,8 @@
+# Ignore all data files
+*.db
+*.json
+
+# But keep this .gitignore
+!.gitignore
+
+last_payload.txt

package/claude-code-config/scripts/statusline/defaults.json

@@ -1,4 +1,8 @@
 {
+	"features": {
+		"usageLimits": true,
+		"spendTracking": true
+	},
 	"oneLine": true,
 	"showSonnetModel": false,
 	"pathDisplayMode": "truncated",

package/claude-code-config/scripts/statusline/docs/ARCHITECTURE.md

@@ -0,0 +1,166 @@
+# Statusline Architecture - Session & Period Tracking
+
+## Concepts Fondamentaux
+
+### Session Claude Code
+
+Une **session** est créée chaque fois qu'on ouvre un chat avec Claude Code.
+
+Caractéristiques:
+- **ID unique**: `session_id` (UUID)
+- **Coût cumulatif**: Le coût total de la session qui AUGMENTE au fil du temps
+- **Persistance**: Une session peut durer des heures/jours
+- **Commande /clear**: Efface la conversation mais GARDE la même session
+- **Le coût ne reset jamais**: Si une session a coûté $10, puis on fait /clear, le coût reste $10 et continue d'augmenter
+
+```
+Session "abc-123"
+├── Début: $0
+├── Après 1h: $5
+├── /clear (conversation effacée)
+├── Après 2h: $12 (coût continue d'augmenter)
+├── Fermeture terminal
+├── Réouverture (même session)
+└── Après 3h: $18 (toujours cumulatif)
+```
+
+### Période (5 heures)
+
+Une **période** est une fenêtre de 5 heures pour le rate limiting d'Anthropic.
+
+Caractéristiques:
+- **resets_at**: Timestamp de fin de période (ex: "2025-12-09T10:00:00.000Z")
+- **Durée fixe**: 5 heures
+- **Indépendant des sessions**: Les sessions peuvent traverser plusieurs périodes
+
+```
+Période A (05:00-10:00)     Période B (10:00-15:00)
+├────────────────────┤     ├────────────────────┤
+   Session 1: +$5              Session 1: +$3 (continue)
+   Session 2: +$8              Session 3: +$10 (nouvelle)
+   ─────────────               ─────────────
+   Total: $13                  Total: $13
+```
+
+## Le Problème Actuel
+
+### Bug: Double Comptage
+
+Quand on calcule le coût d'une période en sommant les coûts totaux des sessions:
+
+```
+Session X: coût total = $24
+Session X: last_resets_at = période actuelle
+
+getCurrentPeriodCost() retourne $24 ❌ FAUX!
+```
+
+Mais si la session X avait déjà $10 AVANT cette période:
+- Coût réel de cette période = $24 - $10 = $14
+- On affiche $24 au lieu de $14 = **double comptage**
+
+### Scénario Réel
+
+```
+1. Période A: Session X coûte $10, on compte $10 ✓
+2. Période B commence
+3. Session X continue, coût monte à $24
+4. On calcule période B: on voit session X = $24
+5. On affiche $24 pour période B ❌
+6. Mais on avait déjà compté $10 en période A!
+7. Total affiché: $10 + $24 = $34
+8. Total réel: $24
+```
+
+## Solution: Tracking des Deltas par Session
+
+### Nouvelle Structure de Données
+
+On doit tracker pour chaque session:
+- Combien on a DÉJÀ compté
+- Dans quelle période on l'a compté
+
+```sql
+-- Table: sessions
+session_id    TEXT PRIMARY KEY
+total_cost    REAL        -- Coût total actuel de la session
+cwd           TEXT
+date          TEXT
+duration_ms   INTEGER
+lines_added   INTEGER
+lines_removed INTEGER
+
+-- Table: session_period_tracking
+session_id    TEXT
+period_id     TEXT        -- resets_at normalisé
+counted_cost  REAL        -- Combien on a compté pour cette période
+last_update   INTEGER     -- Timestamp
+PRIMARY KEY (session_id, period_id)
+
+-- Table: periods
+period_id     TEXT PRIMARY KEY  -- resets_at normalisé
+total_cost    REAL              -- Somme des deltas de toutes les sessions
+utilization   INTEGER
+date          TEXT
+```
+
+### Algorithme Correct
+
+```
+Quand saveSession(session_id, new_cost, current_period):
+  1. Chercher session dans DB
+  2. Si existe:
+       old_cost = session.total_cost
+       delta = new_cost - old_cost
+     Sinon:
+       delta = new_cost
+
+  3. Chercher tracking pour (session_id, current_period)
+  4. Si existe:
+       // Déjà compté dans cette période, calculer le vrai delta
+       already_counted = tracking.counted_cost
+       period_delta = delta  // Le delta depuis la dernière update
+     Sinon:
+       // Nouvelle session dans cette période
+       period_delta = delta
+
+  5. Mettre à jour:
+     - session.total_cost = new_cost
+     - tracking.counted_cost += period_delta
+     - period.total_cost += period_delta
+```
+
+### Exemple Concret
+
+```
+Session X traverse 2 périodes:
+
+Période A:
+  - Session X: $0 → $10
+  - tracking(X, A).counted_cost = $10
+  - period(A).total_cost = $10
+
+Période B:
+  - Session X: $10 → $24
+  - delta = $24 - $10 = $14
+  - tracking(X, B).counted_cost = $14
+  - period(B).total_cost = $14
+
+Calcul période B:
+  - On lit period(B).total_cost = $14 ✓ CORRECT!
+```
+
+## Avantages SQLite
+
+1. **Transactions ACID**: Pas de corruption de données
+2. **Requêtes complexes**: Agrégations, joins faciles
+3. **Index**: Performance pour les lookups par session_id
+4. **Single file**: Toujours portable
+5. **Intégrité référentielle**: FK entre tables
+
+## Migration
+
+1. Créer nouveau fichier `statusline.db`
+2. Migrer données existantes de spend.json et daily-usage.json
+3. Recalculer les period_costs correctement
+4. Supprimer les anciens fichiers JSON

package/claude-code-config/scripts/statusline/fixtures/mock-transcript.jsonl

@@ -0,0 +1,4 @@
+{"parentUuid":null,"isSidechain":false,"userType":"external","cwd":"/Users/melvynx/.claude/scripts/statusline","sessionId":"demo-session","version":"2.0.31","gitBranch":"main","timestamp":"2025-11-11T10:00:00.000Z","message":{"model":"claude-sonnet-4-5-20250929","id":"msg_01","type":"message","role":"user","content":[{"type":"text","text":"Hello"}],"usage":{"input_tokens":1000,"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"output_tokens":100}},"uuid":"uuid-1"}
+{"parentUuid":"uuid-1","isSidechain":false,"userType":"external","cwd":"/Users/melvynx/.claude/scripts/statusline","sessionId":"demo-session","version":"2.0.31","gitBranch":"main","timestamp":"2025-11-11T10:05:00.000Z","message":{"model":"claude-sonnet-4-5-20250929","id":"msg_02","type":"message","role":"assistant","content":[{"type":"text","text":"Hi there!"}],"usage":{"input_tokens":50000,"cache_creation_input_tokens":30000,"cache_read_input_tokens":0,"output_tokens":200}},"uuid":"uuid-2"}
+{"parentUuid":"uuid-2","isSidechain":false,"userType":"external","cwd":"/Users/melvynx/.claude/scripts/statusline","sessionId":"demo-session","version":"2.0.31","gitBranch":"main","timestamp":"2025-11-11T10:10:00.000Z","message":{"model":"claude-sonnet-4-5-20250929","id":"msg_03","type":"message","role":"user","content":[{"type":"text","text":"Can you help me?"}],"usage":{"input_tokens":20000,"cache_creation_input_tokens":0,"cache_read_input_tokens":40000,"output_tokens":150}},"uuid":"uuid-3"}
+{"parentUuid":"uuid-3","isSidechain":false,"userType":"external","cwd":"/Users/melvynx/.claude/scripts/statusline","sessionId":"demo-session","version":"2.0.31","gitBranch":"main","timestamp":"2025-11-11T10:15:00.000Z","message":{"model":"claude-sonnet-4-5-20250929","id":"msg_04","type":"message","role":"assistant","content":[{"type":"text","text":"Of course! What do you need?"}],"usage":{"input_tokens":30000,"cache_creation_input_tokens":0,"cache_read_input_tokens":45000,"output_tokens":300}},"uuid":"uuid-4"}

package/claude-code-config/scripts/statusline/fixtures/test-input.json

@@ -0,0 +1,35 @@
+{
+	"session_id": "06a7b019-03f8-4083-a9db-410d95cb01e6",
+	"transcript_path": "/Users/melvynx/.claude/scripts/statusline/fixtures/mock-transcript.jsonl",
+	"cwd": "/Users/melvynx/.claude",
+	"model": {
+		"id": "claude-sonnet-4-5-20250929",
+		"display_name": "Sonnet 4.5"
+	},
+	"workspace": {
+		"current_dir": "/Users/melvynx/.claude",
+		"project_dir": "/Users/melvynx/.claude"
+	},
+	"version": "2.0.68",
+	"output_style": {
+		"name": "default"
+	},
+	"cost": {
+		"total_cost_usd": 0.17468000000000003,
+		"total_duration_ms": 385160,
+		"total_api_duration_ms": 252694,
+		"total_lines_added": 185,
+		"total_lines_removed": 75
+	},
+	"context_window": {
+		"total_input_tokens": 136349,
+		"total_output_tokens": 21612,
+		"context_window_size": 200000,
+		"current_usage": {
+			"input_tokens": 62500,
+			"cache_creation_input_tokens": 0,
+			"cache_read_input_tokens": 0
+		}
+	},
+	"exceeds_200k_tokens": false
+}

package/claude-code-config/scripts/statusline/src/index.ts

@@ -15,13 +15,46 @@ import { getGitStatus } from "./lib/git";
 import {
 	renderStatusline,
 	type StatuslineData,
+	type UsageLimit,
 } from "./lib/render-pure";
 import type { HookInput } from "./lib/types";
 
+// Optional feature imports - just delete the folder to disable!
+let getUsageLimits: any = null;
+let normalizeResetsAt: any = null;
+let getPeriodCost: any = null;
+let getTodayCostV2: any = null;
+let saveSessionV2: any = null;
+
+try {
+	const limitsModule = await import("./lib/features/limits");
+	getUsageLimits = limitsModule.getUsageLimits;
+} catch {
+	// Limits feature not available - that's OK!
+}
+
+try {
+	const utilsModule = await import("./lib/utils");
+	normalizeResetsAt = utilsModule.normalizeResetsAt;
+} catch {
+	// Fallback normalizeResetsAt
+	normalizeResetsAt = (resetsAt: string) => resetsAt;
+}
+
+try {
+	const spendModule = await import("./lib/features/spend");
+	getPeriodCost = spendModule.getPeriodCost;
+	getTodayCostV2 = spendModule.getTodayCostV2;
+	saveSessionV2 = spendModule.saveSessionV2;
+} catch {
+	// Spend tracking feature not available - that's OK!
+}
+
 // Re-export from render-pure for backwards compatibility
 export {
 	renderStatusline,
 	type StatuslineData,
+	type UsageLimit,
 } from "./lib/render-pure";
 
 const CONFIG_FILE_PATH = join(import.meta.dir, "..", "statusline.config.json");
@@ -49,6 +82,18 @@ async function main() {
 		await writeFile(LAST_PAYLOAD_PATH, JSON.stringify(input, null, 2));
 
 		const config = await loadConfig();
+
+		// Get usage limits (if feature exists)
+		const usageLimits = getUsageLimits
+			? await getUsageLimits()
+			: { five_hour: null, seven_day: null };
+		const currentResetsAt = usageLimits.five_hour?.resets_at ?? undefined;
+
+		// Save session with current period context (if feature exists)
+		if (saveSessionV2) {
+			await saveSessionV2(input, currentResetsAt);
+		}
+
 		const git = await getGitStatus();
 
 		let contextTokens: number | null;
@@ -88,6 +133,18 @@ async function main() {
 			contextPercentage = contextData.percentage;
 		}
 
+		// Get period cost from SQLite (if feature exists)
+		let periodCost: number | undefined;
+		let todayCost: number | undefined;
+
+		if (getPeriodCost && getTodayCostV2 && normalizeResetsAt) {
+			const normalizedPeriodId = currentResetsAt
+				? normalizeResetsAt(currentResetsAt)
+				: null;
+			periodCost = normalizedPeriodId ? getPeriodCost(normalizedPeriodId) : 0;
+			todayCost = getTodayCostV2();
+		}
+
 		const data: StatuslineData = {
 			branch: formatBranch(git, config.git),
 			dirPath: formatPath(input.workspace.current_dir, config.pathDisplayMode),
@@ -99,6 +156,23 @@ async function main() {
 			sessionDuration: formatDuration(input.cost.total_duration_ms),
 			contextTokens,
 			contextPercentage,
+			...(getUsageLimits && {
+				usageLimits: {
+					five_hour: usageLimits.five_hour
+						? {
+								utilization: usageLimits.five_hour.utilization,
+								resets_at: usageLimits.five_hour.resets_at,
+							}
+						: null,
+					seven_day: usageLimits.seven_day
+						? {
+								utilization: usageLimits.seven_day.utilization,
+								resets_at: usageLimits.seven_day.resets_at,
+							}
+						: null,
+				},
+			}),
+			...((getPeriodCost || getTodayCostV2) && { periodCost, todayCost }),
 		};
 
 		const output = renderStatusline(data, config);

package/claude-code-config/scripts/statusline/src/lib/config-types.ts

@@ -50,6 +50,10 @@ export interface PercentageConfig {
 }
 
 export interface StatuslineConfig {
+	features?: {
+		usageLimits?: boolean;
+		spendTracking?: boolean;
+	};
 	oneLine: boolean;
 	showSonnetModel: boolean;
 	pathDisplayMode: "full" | "truncated" | "basename";