oh-my-copilot 4.13.1 → 4.13.21

package/dist/__tests__/hud/render-enterprise.test.js

@@ -0,0 +1,232 @@
+/**
+ * Tests for enterprise cost rendering in render.ts
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { render } from '../../hud/render.js';
+import { DEFAULT_HUD_CONFIG } from '../../hud/types.js';
+// Mock git elements
+vi.mock('../../hud/elements/git.js', () => ({
+    renderGitRepo: vi.fn(() => null),
+    renderGitBranch: vi.fn(() => null),
+    renderGitStatus: vi.fn(() => null),
+}));
+vi.mock('../../hud/elements/cwd.js', () => ({
+    renderCwd: vi.fn(() => null),
+}));
+// Strip ANSI codes for readable assertions
+function strip(s) {
+    return s.replace(/\x1b\[[0-9;]*[a-zA-Z]/g, '');
+}
+function createContext(overrides = {}) {
+    return {
+        contextPercent: 30,
+        modelName: 'claude-sonnet-4-5',
+        ralph: null,
+        ultrawork: null,
+        prd: null,
+        autopilot: null,
+        activeAgents: [],
+        todos: [],
+        backgroundTasks: [],
+        cwd: '/home/user/project',
+        lastSkill: null,
+        rateLimitsResult: null,
+        customBuckets: null,
+        pendingPermission: null,
+        thinkingState: null,
+        sessionHealth: null,
+        omcVersion: null,
+        updateAvailable: null,
+        toolCallCount: 0,
+        agentCallCount: 0,
+        skillCallCount: 0,
+        promptTime: null,
+        apiKeySource: null,
+        profileName: null,
+        sessionSummary: null,
+        ...overrides,
+    };
+}
+function createConfig(overrides = {}) {
+    return {
+        ...DEFAULT_HUD_CONFIG,
+        elements: {
+            ...DEFAULT_HUD_CONFIG.elements,
+            omcLabel: false,
+            rateLimits: false,
+            ralph: false,
+            autopilot: false,
+            prdStory: false,
+            activeSkills: false,
+            contextBar: false,
+            agents: false,
+            backgroundTasks: false,
+            todos: false,
+            promptTime: false,
+            sessionHealth: false,
+            showCallCounts: false,
+            thinking: false,
+            ...overrides,
+        },
+    };
+}
+describe('render - enterprise cost branch', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    it('renders spent:$ when subscriptionType is enterprise and enterpriseSpentUsd is set', async () => {
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    enterpriseSpentUsd: 3323.93,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+                stale: false,
+            },
+        });
+        const config = createConfig({ showEnterpriseCost: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).toContain('spent:$3,323.93');
+    });
+    it('does NOT render tok: when enterprise cost renders successfully', async () => {
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    enterpriseSpentUsd: 100,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+            lastRequestTokenUsage: { inputTokens: 1200, outputTokens: 340 },
+        });
+        const config = createConfig({ showTokens: true, showEnterpriseCost: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).not.toContain('tok:');
+        expect(plain).toContain('spent:');
+    });
+    it('detects enterprise via rateLimitTier containing claude_zero', async () => {
+        const context = createContext({
+            rateLimitTier: 'default_claude_zero',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    enterpriseSpentUsd: 50,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const config = createConfig({ showEnterpriseCost: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).toContain('spent:$50.00');
+    });
+    it('falls back to token rendering when enterprise but no cost data (API error)', async () => {
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: null,
+                error: 'network',
+            },
+            lastRequestTokenUsage: { inputTokens: 1200, outputTokens: 340 },
+        });
+        const config = createConfig({ showTokens: true, showEnterpriseCost: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        // No cost data available → fall back to tokens
+        expect(plain).toContain('tok:');
+    });
+    it('does not render enterprise cost when showEnterpriseCost is false', async () => {
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    enterpriseSpentUsd: 100,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const config = createConfig({ showEnterpriseCost: false });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).not.toContain('spent:');
+    });
+    it('does not render enterprise cost when enterpriseMode is forced false', async () => {
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    enterpriseSpentUsd: 100,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const config = createConfig({ enterpriseMode: false });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).not.toContain('spent:');
+    });
+    it('suppresses 5h/wk rate-limit display when enterprise cost data is present', async () => {
+        // Regression: enterprise API returns five_hour: null, so clamp(null) produced
+        // fiveHourPercent: 0 which rendered a misleading "5h:0% wk:0%" alongside the cost.
+        const context = createContext({
+            subscriptionType: 'enterprise',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 0,
+                    weeklyPercent: 0,
+                    enterpriseSpentUsd: 3323.93,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const config = createConfig({
+            rateLimits: true,
+            showEnterpriseCost: true,
+        });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).toContain('spent:$3,323.93');
+        expect(plain).not.toMatch(/5h:\s*0%/);
+        expect(plain).not.toMatch(/wk:\s*0%/);
+    });
+    it('still renders 5h/wk for non-enterprise users (no regression on Pro/Max)', async () => {
+        const context = createContext({
+            subscriptionType: 'pro',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 45,
+                    weeklyPercent: 12,
+                },
+            },
+        });
+        const config = createConfig({ rateLimits: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).toMatch(/5h:\s*45%/);
+    });
+    it('renders token usage for non-enterprise user even with showEnterpriseCost: true', async () => {
+        const context = createContext({
+            subscriptionType: 'pro',
+            lastRequestTokenUsage: { inputTokens: 1200, outputTokens: 340 },
+        });
+        const config = createConfig({ showTokens: true, showEnterpriseCost: true });
+        const output = await render(context, config);
+        const plain = strip(output);
+        expect(plain).toContain('tok:');
+        expect(plain).not.toContain('spent:');
+    });
+});
+//# sourceMappingURL=render-enterprise.test.js.map

package/dist/__tests__/hud/render-enterprise.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"render-enterprise.test.js","sourceRoot":"","sources":["../../../src/__tests__/hud/render-enterprise.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAyC,MAAM,oBAAoB,CAAC;AAE/F,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;IAChC,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;IAClC,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;CACnC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;CAC7B,CAAC,CAAC,CAAC;AAEJ,2CAA2C;AAC3C,SAAS,KAAK,CAAC,CAAS;IACtB,OAAO,CAAC,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,aAAa,CAAC,YAAuC,EAAE;IAC9D,OAAO;QACL,cAAc,EAAE,EAAE;QAClB,SAAS,EAAE,mBAAmB;QAC9B,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,IAAI;QACf,GAAG,EAAE,IAAI;QACT,SAAS,EAAE,IAAI;QACf,YAAY,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE;QACT,eAAe,EAAE,EAAE;QACnB,GAAG,EAAE,oBAAoB;QACzB,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,IAAI;QACtB,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,IAAI;QACnB,UAAU,EAAE,IAAI;QAChB,eAAe,EAAE,IAAI;QACrB,aAAa,EAAE,CAAC;QAChB,cAAc,EAAE,CAAC;QACjB,cAAc,EAAE,CAAC;QACjB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,IAAI;QACpB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,YAA4C,EAAE;IAClE,OAAO;QACL,GAAG,kBAAkB;QACrB,QAAQ,EAAE;YACR,GAAG,kBAAkB,CAAC,QAAQ;YAC9B,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,KAAK;YACjB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;YACnB,UAAU,EAAE,KAAK;YACjB,MAAM,EAAE,KAAK;YACb,eAAe,EAAE,KAAK;YACtB,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,KAAK;YACjB,aAAa,EAAE,KAAK;YACpB,cAAc,EAAE,KAAK;YACrB,QAAQ,EAAE,KAAK;YACf,GAAG,SAAS;SACb;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,kBAAkB,EAAE,OAAO;oBAC3B,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;gBACD,KAAK,EAAE,KAAK;aACb;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,kBAAkB,EAAE,GAAG;oBACvB,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;YACD,qBAAqB,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE;SAChE,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,aAAa,EAAE,qBAAqB;YACpC,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,kBAAkB,EAAE,EAAE;oBACtB,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,SAAS;aACjB;YACD,qBAAqB,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE;SAChE,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,+CAA+C;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,kBAAkB,EAAE,GAAG;oBACvB,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,kBAAkB,EAAE,GAAG;oBACvB,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,8EAA8E;QAC9E,mFAAmF;QACnF,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,aAAa,EAAE,CAAC;oBAChB,kBAAkB,EAAE,OAAO;oBAC3B,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC;YAC1B,UAAU,EAAE,IAAI;YAChB,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,KAAK;YACvB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,EAAE;iBAClB;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,OAAO,GAAG,aAAa,CAAC;YAC5B,gBAAgB,EAAE,KAAK;YACvB,qBAAqB,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE;SAChE,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/hud/render-rate-limits-priority.test.js

@@ -45,6 +45,9 @@ function makeContext(overrides = {}) {
         ...overrides,
     };
 }
+function stripAnsi(value) {
+    return value.replace(/\x1b\[[0-9;]*[a-zA-Z]/g, '');
+}
 function makeConfig(overrides = {}) {
     return {
         ...DEFAULT_HUD_CONFIG,
@@ -78,6 +81,188 @@ describe('render: rate limits display priority', () => {
         expect(output).toContain('45%');
         expect(output).not.toContain('[API 429]');
     });
+    it('renders 5h/wk/sn rate limits when subscription info is unavailable', async () => {
+        const context = makeContext({
+            subscriptionType: null,
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 36,
+                    weeklyPercent: 32,
+                    sonnetWeeklyPercent: 8,
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        expect(output).toContain('5h:');
+        expect(output).toContain('36%');
+        expect(output).toContain('wk:');
+        expect(output).toContain('32%');
+        expect(output).toContain('sn:');
+        expect(output).toContain('8%');
+    });
+    it('renders exact Max 20x cache-shaped rate limits when legacy enterprise spend fields are present', async () => {
+        const context = makeContext({
+            subscriptionType: 'max',
+            rateLimitTier: 'default_claude_max_20x',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 4,
+                    weeklyPercent: 6,
+                    enterpriseSpentUsd: 200.5,
+                    enterpriseLimitUsd: 200,
+                    enterpriseUtilization: 100,
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        const plain = stripAnsi(output);
+        expect(plain.trim()).not.toBe('');
+        expect(plain).toContain('5h:');
+        expect(plain).toContain('4%');
+        expect(plain).toContain('wk:');
+        expect(plain).toContain('6%');
+        expect(plain).not.toContain('spent:');
+    });
+    it('renders Max 20x 5h/wk/sn limits when enterprise spent exists but enterprise limit is null', async () => {
+        const context = makeContext({
+            subscriptionType: 'max',
+            rateLimitTier: 'default_claude_max_20x',
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 36,
+                    weeklyPercent: 32,
+                    sonnetWeeklyPercent: 8,
+                    enterpriseSpentUsd: 12.34,
+                    enterpriseLimitUsd: null,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        const plain = stripAnsi(output);
+        expect(plain.trim()).not.toBe('');
+        expect(plain).toContain('5h:');
+        expect(plain).toContain('36%');
+        expect(plain).toContain('wk:');
+        expect(plain).toContain('32%');
+        expect(plain).toContain('sn:');
+        expect(plain).toContain('8%');
+        expect(plain).not.toContain('spent:');
+    });
+    it.each([
+        ['zero', 0],
+        ['negative', -1],
+        ['null', null],
+        ['undefined', undefined],
+    ])('renders normal rate limits for Max-like responses with %s enterprise limit', async (_caseName, enterpriseLimitUsd) => {
+        const context = makeContext({
+            subscriptionType: 'max',
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 36,
+                    weeklyPercent: 32,
+                    sonnetWeeklyPercent: 8,
+                    enterpriseSpentUsd: 12.34,
+                    enterpriseLimitUsd,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        expect(output).toContain('5h:');
+        expect(output).toContain('36%');
+        expect(output).toContain('wk:');
+        expect(output).toContain('32%');
+        expect(output).toContain('sn:');
+        expect(output).toContain('8%');
+        expect(output).not.toContain('spent:');
+    });
+    it.each(['pro', 'max'])('renders normal 5h/wk limits for non-enterprise %s when enterprise spend is nonzero', async (subscriptionType) => {
+        const context = makeContext({
+            subscriptionType,
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 36,
+                    weeklyPercent: 32,
+                    enterpriseSpentUsd: 12.34,
+                    enterpriseLimitUsd: 50,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        expect(output).toContain('5h:');
+        expect(output).toContain('36%');
+        expect(output).toContain('wk:');
+        expect(output).toContain('32%');
+        expect(output).not.toContain('spent:');
+    });
+    it.each(['pro', 'max'])('renders normal 5h/wk limits for non-enterprise %s when enterprise spend is zero', async (subscriptionType) => {
+        const context = makeContext({
+            subscriptionType,
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 10,
+                    weeklyPercent: 2,
+                    enterpriseSpentUsd: 0,
+                    enterpriseLimitUsd: 50,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        expect(output).toContain('5h:');
+        expect(output).toContain('10%');
+        expect(output).toContain('wk:');
+        expect(output).toContain('2%');
+        expect(output).not.toContain('spent:');
+    });
+    it('renders enterprise cost only for actual enterprise with the same billing fields', async () => {
+        const context = makeContext({
+            subscriptionType: 'enterprise',
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 4,
+                    weeklyPercent: 6,
+                    enterpriseSpentUsd: 200.5,
+                    enterpriseLimitUsd: 200,
+                    enterpriseUtilization: 100,
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        const plain = stripAnsi(output);
+        expect(plain).toContain('spent:');
+        expect(plain).toContain('$200.50/$200.00');
+        expect(plain).toContain('(100%)');
+        expect(plain).not.toContain('5h:');
+        expect(plain).not.toContain('wk:');
+    });
+    it('uses enterprise cost instead of double-rendering 5h/wk for actual enterprise zero spend', async () => {
+        const context = makeContext({
+            subscriptionType: 'enterprise',
+            rateLimitTier: null,
+            rateLimitsResult: {
+                rateLimits: {
+                    fiveHourPercent: 10,
+                    weeklyPercent: 2,
+                    enterpriseSpentUsd: 0,
+                    enterpriseLimitUsd: 50,
+                    enterpriseCurrency: 'USD',
+                },
+            },
+        });
+        const output = await render(context, makeConfig());
+        expect(output).toContain('spent:');
+        expect(output).toContain('$0.00/$50.00');
+        expect(output).not.toContain('5h:');
+        expect(output).not.toContain('wk:');
+    });
     it('shows [API 429] when error=rate_limited and rateLimits is null', async () => {
         const context = makeContext({
             rateLimitsResult: {

package/dist/__tests__/hud/render-rate-limits-priority.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"render-rate-limits-priority.test.js","sourceRoot":"","sources":["../../../src/__tests__/hud/render-rate-limits-priority.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,oEAAoE;AACpE,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;IACzB,eAAe,EAAE,GAAG,EAAE,CAAC,IAAI;CAC5B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;CACtB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,SAAS,WAAW,CAAC,YAAuC,EAAE;IAC5D,OAAO;QACL,cAAc,EAAE,EAAE;QAClB,SAAS,EAAE,MAAM;QACjB,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,IAAI;QACf,GAAG,EAAE,IAAI;QACT,SAAS,EAAE,IAAI;QACf,YAAY,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE;QACT,eAAe,EAAE,EAAE;QACnB,GAAG,EAAE,WAAW;QAChB,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,IAAI;QACtB,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,IAAI;QACnB,UAAU,EAAE,OAAO;QACnB,eAAe,EAAE,IAAI;QACrB,aAAa,EAAE,CAAC;QAChB,cAAc,EAAE,CAAC;QACjB,cAAc,EAAE,CAAC;QACjB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,IAAI;QACpB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,YAAgC,EAAE;IACpD,OAAO;QACL,GAAG,kBAAkB;QACrB,QAAQ,EAAE;YACR,GAAG,kBAAkB,CAAC,QAAQ;YAC9B,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,KAAK;YACjB,MAAM,EAAE,KAAK;YACb,eAAe,EAAE,KAAK;YACtB,KAAK,EAAE,KAAK;YACZ,YAAY,EAAE,KAAK;YACnB,SAAS,EAAE,KAAK;YAChB,aAAa,EAAE,KAAK;YACpB,UAAU,EAAE,KAAK;YACjB,cAAc,EAAE,KAAK;SACtB;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBACtD,KAAK,EAAE,cAAc;aACtB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,6CAA6C;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,cAAc;aACtB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,SAAS;aACjB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,MAAM;aACd;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;aACvD;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,gBAAgB;aACxB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"render-rate-limits-priority.test.js","sourceRoot":"","sources":["../../../src/__tests__/hud/render-rate-limits-priority.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,oEAAoE;AACpE,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;IACzB,eAAe,EAAE,GAAG,EAAE,CAAC,IAAI;CAC5B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;CACtB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,SAAS,WAAW,CAAC,YAAuC,EAAE;IAC5D,OAAO;QACL,cAAc,EAAE,EAAE;QAClB,SAAS,EAAE,MAAM;QACjB,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,IAAI;QACf,GAAG,EAAE,IAAI;QACT,SAAS,EAAE,IAAI;QACf,YAAY,EAAE,EAAE;QAChB,KAAK,EAAE,EAAE;QACT,eAAe,EAAE,EAAE;QACnB,GAAG,EAAE,WAAW;QAChB,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,IAAI;QACtB,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,IAAI;QACnB,UAAU,EAAE,OAAO;QACnB,eAAe,EAAE,IAAI;QACrB,aAAa,EAAE,CAAC;QAChB,cAAc,EAAE,CAAC;QACjB,cAAc,EAAE,CAAC;QACjB,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;QAClB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,IAAI;QACpB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,KAAa;IAC9B,OAAO,KAAK,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,UAAU,CAAC,YAAgC,EAAE;IACpD,OAAO;QACL,GAAG,kBAAkB;QACrB,QAAQ,EAAE;YACR,GAAG,kBAAkB,CAAC,QAAQ;YAC9B,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,KAAK;YACjB,MAAM,EAAE,KAAK;YACb,eAAe,EAAE,KAAK;YACtB,KAAK,EAAE,KAAK;YACZ,YAAY,EAAE,KAAK;YACnB,SAAS,EAAE,KAAK;YAChB,aAAa,EAAE,KAAK;YACpB,UAAU,EAAE,KAAK;YACjB,cAAc,EAAE,KAAK;SACtB;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBACtD,KAAK,EAAE,cAAc;aACtB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,6CAA6C;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,IAAI;YACtB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,EAAE;oBACjB,mBAAmB,EAAE,CAAC;iBACvB;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gGAAgG,EAAE,KAAK,IAAI,EAAE;QAC9G,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,KAAK;YACvB,aAAa,EAAE,wBAAwB;YACvC,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,aAAa,EAAE,CAAC;oBAChB,kBAAkB,EAAE,KAAK;oBACzB,kBAAkB,EAAE,GAAG;oBACvB,qBAAqB,EAAE,GAAG;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAEhC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,KAAK;YACvB,aAAa,EAAE,wBAAwB;YACvC,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,EAAE;oBACjB,mBAAmB,EAAE,CAAC;oBACtB,kBAAkB,EAAE,KAAK;oBACzB,kBAAkB,EAAE,IAAI;oBACxB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAEhC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;QACN,CAAC,MAAM,EAAE,CAAC,CAAC;QACX,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QAChB,CAAC,MAAM,EAAE,IAAI,CAAC;QACd,CAAC,WAAW,EAAE,SAAS,CAAC;KAChB,CAAC,CACT,4EAA4E,EAC5E,KAAK,EAAE,SAAS,EAAE,kBAAkB,EAAE,EAAE;QACtC,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,KAAK;YACvB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,EAAE;oBACjB,mBAAmB,EAAE,CAAC;oBACtB,kBAAkB,EAAE,KAAK;oBACzB,kBAAkB;oBAClB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC,CACF,CAAC;IAEF,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAU,CAAC,CAC9B,oFAAoF,EACpF,KAAK,EAAE,gBAAgB,EAAE,EAAE;QACzB,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB;YAChB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,EAAE;oBACjB,kBAAkB,EAAE,KAAK;oBACzB,kBAAkB,EAAE,EAAE;oBACtB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC,CACF,CAAC;IAEF,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAU,CAAC,CAC9B,iFAAiF,EACjF,KAAK,EAAE,gBAAgB,EAAE,EAAE;QACzB,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB;YAChB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,CAAC;oBAChB,kBAAkB,EAAE,CAAC;oBACrB,kBAAkB,EAAE,EAAE;oBACtB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC,CACF,CAAC;IAEF,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,YAAY;YAC9B,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,CAAC;oBAClB,aAAa,EAAE,CAAC;oBAChB,kBAAkB,EAAE,KAAK;oBACzB,kBAAkB,EAAE,GAAG;oBACvB,qBAAqB,EAAE,GAAG;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAEhC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE,YAAY;YAC9B,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE;gBAChB,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE;oBACnB,aAAa,EAAE,CAAC;oBAChB,kBAAkB,EAAE,CAAC;oBACrB,kBAAkB,EAAE,EAAE;oBACtB,kBAAkB,EAAE,KAAK;iBAC1B;aACF;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,cAAc;aACtB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,SAAS;aACjB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,MAAM;aACd;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;aACvD;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,OAAO,GAAG,WAAW,CAAC;YAC1B,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,gBAAgB;aACxB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/hud/stdin-session-scope.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=stdin-session-scope.test.d.ts.map

package/dist/__tests__/hud/stdin-session-scope.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdin-session-scope.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/hud/stdin-session-scope.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/hud/stdin-session-scope.test.js

@@ -0,0 +1,169 @@
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
+import { existsSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from 'fs';
+import { tmpdir } from 'os';
+import { join } from 'path';
+import { execSync } from 'child_process';
+import { readStdinCache, writeStdinCache } from '../../hud/stdin.js';
+function makeStdin(overrides = {}) {
+    return {
+        model: { display_name: 'Sonnet' },
+        workspace: { current_dir: '/tmp' },
+        cwd: '/tmp',
+        transcript_path: '/tmp/session.jsonl',
+        ...overrides,
+    };
+}
+describe('HUD stdin cache path is session-scoped', () => {
+    let tmpRoot;
+    let originalCwd;
+    const envKeys = ['CLAUDE_SESSION_ID', 'CLAUDECODE_SESSION_ID'];
+    const savedEnv = {};
+    beforeEach(() => {
+        tmpRoot = mkdtempSync(join(tmpdir(), 'omc-hud-stdin-cache-'));
+        // Make a real git repo so getWorktreeRoot() (which shells out to git
+        // rev-parse) deterministically returns tmpRoot instead of leaking into
+        // the surrounding workspace.
+        execSync('git init --quiet', { cwd: tmpRoot });
+        originalCwd = process.cwd();
+        process.chdir(tmpRoot);
+        for (const key of envKeys) {
+            savedEnv[key] = process.env[key];
+            delete process.env[key];
+        }
+    });
+    afterEach(() => {
+        process.chdir(originalCwd);
+        for (const key of envKeys) {
+            if (savedEnv[key] === undefined) {
+                delete process.env[key];
+            }
+            else {
+                process.env[key] = savedEnv[key];
+            }
+        }
+        rmSync(tmpRoot, { recursive: true, force: true });
+    });
+    it('writes to a session-scoped path when CLAUDE_SESSION_ID is set', () => {
+        process.env.CLAUDE_SESSION_ID = 'test-session-aaa';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const expected = join(tmpRoot, '.omcp', 'state', 'sessions', 'test-session-aaa', 'hud-stdin-cache.json');
+        expect(existsSync(expected)).toBe(true);
+        const loaded = JSON.parse(readFileSync(expected, 'utf-8'));
+        expect(loaded.cwd).toBe(tmpRoot);
+    });
+    it('falls back to the legacy flat path when no session env var is set', () => {
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const expected = join(tmpRoot, '.omcp', 'state', 'hud-stdin-cache.json');
+        expect(existsSync(expected)).toBe(true);
+        const sessionScoped = join(tmpRoot, '.omcp', 'state', 'sessions');
+        expect(existsSync(sessionScoped)).toBe(false);
+    });
+    it('accepts CLAUDECODE_SESSION_ID as the session id source', () => {
+        process.env.CLAUDECODE_SESSION_ID = 'test-session-bbb';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const expected = join(tmpRoot, '.omcp', 'state', 'sessions', 'test-session-bbb', 'hud-stdin-cache.json');
+        expect(existsSync(expected)).toBe(true);
+    });
+    it('prevents two concurrent sessions from clobbering each other', () => {
+        process.env.CLAUDE_SESSION_ID = 'session-alpha';
+        const alpha = makeStdin({ cwd: tmpRoot, transcript_path: `${tmpRoot}/alpha.jsonl` });
+        writeStdinCache(alpha);
+        process.env.CLAUDE_SESSION_ID = 'session-beta';
+        const beta = makeStdin({ cwd: tmpRoot, transcript_path: `${tmpRoot}/beta.jsonl` });
+        writeStdinCache(beta);
+        // Reading back from each session must return its own snapshot.
+        process.env.CLAUDE_SESSION_ID = 'session-alpha';
+        expect(readStdinCache()?.transcript_path).toBe(`${tmpRoot}/alpha.jsonl`);
+        process.env.CLAUDE_SESSION_ID = 'session-beta';
+        expect(readStdinCache()?.transcript_path).toBe(`${tmpRoot}/beta.jsonl`);
+    });
+    it('readStdinCache ignores a legacy flat file when a session id is set', () => {
+        const stateDir = join(tmpRoot, '.omcp', 'state');
+        mkdirSync(stateDir, { recursive: true });
+        // Simulate a stale legacy cache written by an older build.
+        const legacy = makeStdin({ cwd: '/legacy/cwd' });
+        writeFileSync(join(stateDir, 'hud-stdin-cache.json'), JSON.stringify(legacy));
+        process.env.CLAUDE_SESSION_ID = 'fresh-session';
+        // Without a session file yet, read should miss rather than return the
+        // legacy (cross-session) value.
+        expect(readStdinCache()).toBeNull();
+    });
+    // Unsafe / malformed session ids must NOT escape the session-scoped directory.
+    // `getStdinCachePath` delegates validation to the shared `getSessionStateDir`
+    // helper (which calls `validateSessionId`), so any id that fails the repo-wide
+    // contract should fall back to the legacy flat path rather than being
+    // interpolated into a filesystem path.
+    it.each([
+        ['path traversal with ..', '../../../etc/passwd'],
+        ['path traversal with parent only', '..'],
+        ['forward slash', 'foo/bar'],
+        ['backslash (Windows traversal)', 'foo\\bar'],
+        ['leading underscore (regex first-char violation)', '_foo'],
+        ['overlong id (>256 chars)', 'a'.repeat(300)],
+    ])('rejects unsafe CLAUDE_SESSION_ID (%s) and falls back to the legacy path', (_label, unsafeId) => {
+        process.env.CLAUDE_SESSION_ID = unsafeId;
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        // Nothing may be written to the session-scoped tree at all.
+        const sessionsDir = join(tmpRoot, '.omcp', 'state', 'sessions');
+        expect(existsSync(sessionsDir)).toBe(false);
+        // And in particular, nothing outside the intended state dir.
+        const etcProbe = join(tmpRoot, 'etc', 'passwd');
+        expect(existsSync(etcProbe)).toBe(false);
+        // Legacy flat fallback should be populated instead.
+        const legacy = join(tmpRoot, '.omcp', 'state', 'hud-stdin-cache.json');
+        expect(existsSync(legacy)).toBe(true);
+    });
+    it('treats whitespace-only CLAUDE_SESSION_ID as unset and falls back', () => {
+        process.env.CLAUDE_SESSION_ID = '   ';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const sessionsDir = join(tmpRoot, '.omcp', 'state', 'sessions');
+        expect(existsSync(sessionsDir)).toBe(false);
+        const legacy = join(tmpRoot, '.omcp', 'state', 'hud-stdin-cache.json');
+        expect(existsSync(legacy)).toBe(true);
+    });
+    it('falls through to CLAUDECODE_SESSION_ID when CLAUDE_SESSION_ID is empty', () => {
+        // Regression: `??` alone would accept "" as defined and never consult
+        // the secondary variable.
+        process.env.CLAUDE_SESSION_ID = '';
+        process.env.CLAUDECODE_SESSION_ID = 'secondary-session';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const expected = join(tmpRoot, '.omcp', 'state', 'sessions', 'secondary-session', 'hud-stdin-cache.json');
+        expect(existsSync(expected)).toBe(true);
+    });
+    it('falls through to CLAUDECODE_SESSION_ID when CLAUDE_SESSION_ID is present but invalid', () => {
+        // Regression: a non-empty-but-invalid primary must not silently bypass
+        // a valid secondary. The previous implementation resolved the primary
+        // first, then fell straight to the legacy path when validation threw,
+        // never giving the secondary a chance.
+        process.env.CLAUDE_SESSION_ID = '../../../etc/passwd';
+        process.env.CLAUDECODE_SESSION_ID = 'valid-secondary';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const expectedSecondary = join(tmpRoot, '.omcp', 'state', 'sessions', 'valid-secondary', 'hud-stdin-cache.json');
+        expect(existsSync(expectedSecondary)).toBe(true);
+        // And in particular, the legacy flat path must NOT have been used —
+        // otherwise concurrent sessions could still clobber each other.
+        const legacy = join(tmpRoot, '.omcp', 'state', 'hud-stdin-cache.json');
+        expect(existsSync(legacy)).toBe(false);
+        // Safety probe: traversal from primary must not have escaped.
+        const etcProbe = join(tmpRoot, 'etc', 'passwd');
+        expect(existsSync(etcProbe)).toBe(false);
+    });
+    it('falls back to the legacy path only when every candidate is invalid', () => {
+        process.env.CLAUDE_SESSION_ID = '../traverse';
+        process.env.CLAUDECODE_SESSION_ID = 'foo/bar';
+        const stdin = makeStdin({ cwd: tmpRoot });
+        writeStdinCache(stdin);
+        const legacy = join(tmpRoot, '.omcp', 'state', 'hud-stdin-cache.json');
+        expect(existsSync(legacy)).toBe(true);
+        const sessionsDir = join(tmpRoot, '.omcp', 'state', 'sessions');
+        expect(existsSync(sessionsDir)).toBe(false);
+    });
+});
+//# sourceMappingURL=stdin-session-scope.test.js.map