claude-code-infinite 1.0.26 → 1.0.28

package/dist/proxy.js

@@ -56,6 +56,7 @@ import { COMPRESSED_NOTICE, compressedTotalsText, DEGRADED_NOTICE, NOT_COMPRESSE
 import { NoticeDeliveryQueue, MEMTREE_COMPACT_COMMAND, MEMTREE_HELP_COMMAND, TRAILER_LABEL, LINK_LABEL, linkLines, urlLine, terminalSupportsHyperlinks, isMemtreeViewCommand, sessionCommandArgs, parseNoticeHookInput, } from "./hooks.js";
 import { USAGE_LABEL, USAGE_SIGN_IN_NOTE } from "./usage-link.js";
 import { hyperlink } from "./payment-gate.js";
+import { keyInvalidNoticeText, keyInvalidRefusalMessage } from "./key-status.js";
 import { PaymentNoticeThrottle, pausedCompactionText, pausedRefusalMessage, paymentNoticeText, subscribeUrlFromDetail } from "./payment-notice.js";
 import { MEMTREE_LINKS_MAX_SESSIONS } from "./memtree-links.js";
 import { describeClaudeCodeRequest, inspectMonitorTranscript, knownSideRequestKind, transcriptWarning, isClaudeCodeSideRequest, memtreeClientMeta, sessionTag, } from "./cc-request.js";
@@ -304,11 +305,15 @@ export function startProxy(opts) {
     const shutdownAbort = new AbortController();
     const state = {
         paymentNotice: new PaymentNoticeThrottle(opts.paymentNoticeIntervalMs),
+        keyNotice: new PaymentNoticeThrottle(opts.paymentNoticeIntervalMs),
+        keyInvalid: () => opts.memtree.keyInvalid,
+        keyPageUrl: opts.memtree.keyPageUrl,
         pausedNearWindowWarned: new Set(),
         unpaidDetail: () => opts.memtree.paymentRequiredDetail,
         pausedDetail: () => opts.memtree.pausedDetail,
         usageUrl: opts.memtree.usageDashboardUrl,
         sessionStartNotice: opts.sessionStartNotice,
+        projectTrees: opts.projectTrees,
         diagnostics: opts.diagnostics,
         notices: new NoticeDeliveryQueue(),
         mainNoticePending: false,
@@ -726,17 +731,20 @@ onMainStop) {
             recordSessionStart(state, parsed.source);
             noteMainSessionStart(state, parsed.source, parsed.session_id);
         }
+        // Earlier trees in this project: asked now, so its wait overlaps the resume's.
+        const contextPromise = projectTreesContext(state.projectTrees, parsed);
         // A resume waits briefly for the newest tree; a fresh session has none to wait for.
         if (parsed.agent_id === undefined && parsed.source === "resume") {
             await state.newestTrees?.settle(parsed.session_id);
         }
         const line = joinNoticeLines(resumeLinkLine(state, parsed), resumePaymentNotice(state, parsed), extraSessionStartLine(state.sessionStartNotice, parsed));
-        if (!line) {
+        const context = await contextPromise;
+        if (!line && !context) {
             res.writeHead(204);
             res.end();
             return;
         }
-        const body = Buffer.from(JSON.stringify({ systemMessage: line }), "utf-8");
+        const body = Buffer.from(JSON.stringify(sessionStartOutput(line, context)), "utf-8");
         res.writeHead(200, {
             "content-type": "application/json",
             "content-length": String(body.length),
@@ -1697,6 +1705,13 @@ async function handleMessages(req, res, opts, upstream, state) {
     // it pending for the next turn.
     if (result && sessionId !== undefined)
         state.compactNow.delete(sessionId);
+    // A first 402 starts fetching the short checkout link (memtree.ts): give the
+    // paused notice below a moment to show it instead of the long one. Only a turn
+    // that can show the notice waits, and only while that fetch is in flight.
+    if (!result && displayForThisTurn && state.mainPromptGeneration === noticePromptGeneration &&
+        state.paymentNotice.due() && opts.memtree.paymentRequiredDetail !== null) {
+        await opts.memtree.paymentLinkSettled?.(PAYMENT_LINK_NOTICE_WAIT_MS);
+    }
     if (downstreamClosedDuringCompression ||
         (res.destroyed && !res.writableFinished)) {
         releaseUncommittedRouteDecision();
@@ -1724,7 +1739,15 @@ async function handleMessages(req, res, opts, upstream, state) {
         const paymentDetail = opts.memtree.paymentRequiredDetail;
         const mayQueueNotice = displayForThisTurn && state.mainPromptGeneration === noticePromptGeneration;
         const overBudget = reachedBudget(rec.compaction);
-        if (mayQueueNotice && paymentDetail !== null &&
+        if (state.keyInvalid()) {
+            // A refused key (401/403) says why MemTree is off and how to fix it;
+            // throttled like the payment notice, and quiet in between.
+            if (mayQueueNotice && state.keyNotice.due()) {
+                // A refusal of this very request says the same and withdraws it.
+                pausedNotices.set(rec, state.notices.queueSuffix(keyInvalidNoticeText(state.keyPageUrl), () => state.keyNotice.markShown(), noticePromptId));
+            }
+        }
+        else if (mayQueueNotice && paymentDetail !== null &&
             (state.paymentNotice.due() || (overBudget && !nearWindowWarned(state, sessionId)))) {
             // A paused refusal of this very request withdraws it (refuseWholeRequest),
             // and any later one withdraws its near-window line.
@@ -1984,11 +2007,39 @@ function extraSessionStartLine(provider, input) {
         return undefined;
     }
 }
+/**
+ * Earlier sessions in this project (project-trees.ts) for a main session's
+ * start: fresh, resumed, cleared or forked, never a compaction. Never throws.
+ */
+async function projectTreesContext(hint, input) {
+    if (!hint || input.agent_id !== undefined || input.source === "compact")
+        return undefined;
+    try {
+        return await hint.contextFor(input.session_id);
+    }
+    catch {
+        return undefined;
+    }
+}
+/**
+ * The SessionStart hook's answer: the banner line for the user
+ * (`systemMessage`), model context (`additionalContext`), or both.
+ */
+export function sessionStartOutput(line, context) {
+    return {
+        ...(line ? { systemMessage: line } : {}),
+        ...(context
+            ? { hookSpecificOutput: { hookEventName: "SessionStart", additionalContext: context } }
+            : {}),
+    };
+}
 function joinNoticeLines(...parts) {
     const present = parts.filter((part) => !!part);
     return present.length ? present.join("\n") : undefined;
 }
 export const MEMTREE_COMPACT_MIN_TOKENS = 20_000;
+/** How long a paused turn waits for the short checkout link before its notice (once). */
+export const PAYMENT_LINK_NOTICE_WAIT_MS = 1_000;
 /** The reply to a ccc slash command, or undefined for an ordinary prompt. */
 function sessionCommandReply(state, sessionId, prompt) {
     const unpaid = state.unpaidDetail();
@@ -3554,7 +3605,9 @@ async function recoverToolRouteMiss(args) {
         // a later turn. An unpaid key (402, possibly set by this very compress)
         // or a shutting-down proxy gets no retry.
         if (!state.shutdownSignal.aborted &&
-            opts.memtree.paymentRequiredDetail === null) {
+            opts.memtree.paymentRequiredDetail === null &&
+            !opts.memtree.keyInvalid // a refused key refuses the index too
+        ) {
             opts.memtree.indexInBackground(hash, msgsForMemtree, modelContextLimit, requestSessionId(req), args.clientMeta, transcriptTimesFor(opts, requestSessionId(req), agentAttributionId(req)));
         }
         return forwardOriginal();
@@ -3962,9 +4015,12 @@ async function wholeRequestRefused(sizing, candidate, limits) {
 }
 /** Write the bounded fallback refusal (route-fallback.ts); always false. */
 function refuseWholeRequest(args, inputTokens, budgetTokens) {
-    const unpaid = args.state.pausedDetail();
-    const failure = unpaid !== null ? pausedRefusal(pausedRefusalMessage(unpaid))
-        : args.state.fallbackFailures.fail(args.routeKey, { awaitingIndex: args.rec.indexBehind === true });
+    // A refused key or a paused one: a retry cannot fix either, so it is final at once.
+    const keyInvalid = args.state.keyInvalid();
+    const unpaid = keyInvalid ? null : args.state.pausedDetail();
+    const failure = keyInvalid ? pausedRefusal(keyInvalidRefusalMessage(args.state.keyPageUrl))
+        : unpaid !== null ? pausedRefusal(pausedRefusalMessage(unpaid))
+            : args.state.fallbackFailures.fail(args.routeKey, { awaitingIndex: args.rec.indexBehind === true });
     args.rec.forwardedBytes = 0;
     args.rec.approxInputTokens = 0;
     args.state.diagnostics?.record("refusal", {
@@ -3977,8 +4033,9 @@ function refuseWholeRequest(args, inputTokens, budgetTokens) {
         inputTokens, outputTokens: args.body.max_tokens ?? 0,
         outputReservation: bodyOutputReservation(args.body), indexBehind: args.rec.indexBehind === true,
         budgetTokens, modelContextLimit: args.modelContextLimit, ...(unpaid !== null ? { paused: true } : {}),
+        ...(keyInvalid ? { keyInvalid: true } : {}),
     };
-    if (unpaid !== null)
+    if (unpaid !== null || keyInvalid)
         cancelPausedNotice(args.state, args.rec);
     if (args.res.headersSent || args.res.destroyed)
         return false;
@@ -3987,6 +4044,9 @@ function refuseWholeRequest(args, inputTokens, budgetTokens) {
     if (unpaid !== null) {
         notePausedRefusal(args.state, args.rec, args.routeKey, requestSessionId(args.req));
     }
+    // The refusal named the key: it counts as the key notice on the user's screen.
+    if (keyInvalid && args.rec.routeLane === "main")
+        args.state.keyNotice.markShown();
     return false;
 }
 /**
@@ -4025,6 +4085,8 @@ function cancelPausedNotice(state, rec) {
  * already queued the paused notice carries the line itself.
  */
 function warnPausedNearWindow(state, rec, sessionId) {
+    if (state.keyInvalid())
+        return; // a refused key, not a pause: the key notice says why
     const detail = state.unpaidDetail();
     // Paid again: a later pause (next month's limit) warns afresh.
     if (detail === null)