dsh-cursor-dev 0.12.0 → 0.12.2

package/lib/tool.js

@@ -22,6 +22,7 @@ import { delegationDepthOf, resolveChildDepth } from './mirror.js';
 import { sameModel } from './models.js';
 import { buildPrompt } from './prompt.js';
 import { storeKeyOf } from './session-pool.js';
+import { cursorQuotaFailure, quotaBlockedMessage } from './quota-guard.js';
 
 /** Default wall-clock budget for one Cursor turn. */
 const DEFAULT_TIMEOUT_SECONDS = 1800;
@@ -89,7 +90,7 @@ function clean(value, seen) {
  * @param deps - plugin services and shared state.
  */
 export function cursorCodeTool(deps) {
-    const { pool, journal, index, logger, unavailable, mirrors } = deps;
+    const { pool, journal, index, logger, unavailable, mirrors, quotaGuard } = deps;
     // The recursion budget this tool enforces: the child it starts is a real
     // DSH child session, so a subagent calling this tool would create a
     // GRANDCHILD. `1` is the platform's own default (only a top-level agent
@@ -291,6 +292,15 @@ export function cursorCodeTool(deps) {
                 }
             }
 
+            const quotaStatus = await quotaGuard?.check?.();
+            if (quotaStatus?.blocked === true) {
+                return toLosslessJson({
+                    ok: false, sessionKey: 'quota-exhausted', conversationKey,
+                    stopReason: 'error', durationMs: 0,
+                    error: quotaBlockedMessage(quotaStatus), steps: [], changes: [],
+                });
+            }
+
             if (input.resetSession) {
                 pool.dropConversation(conversationKey);
                 journal.clear(conversationKey);
@@ -377,8 +387,8 @@ export function cursorCodeTool(deps) {
             let reused = false;
             let reloaded = false;
             const lifecycle = exec.cursorLifecycle;
-            // External native runs live in the subagents service as a row without a
-            // local transcript; only a lifecycle that asks for the mirror creates one.
+            // The service's external lifecycle id is separate from this child:
+            // the panel follows the mirror, whose transcript is written below.
             const mirrorEnabled = lifecycle?.mirror !== false;
             const openChild = async () => {
                 const entry = await mirrors?.open({
@@ -436,6 +446,13 @@ export function cursorCodeTool(deps) {
                         if (pickedMode !== undefined && session.modes?.currentModeId !== pickedMode)
                             await session.setMode(pickedMode);
                         turn = await session.prompt(input.prompt, abort.signal);
+                        if (cursorQuotaFailure(turn.text)) {
+                            const status = quotaGuard?.observeFailure?.(turn.text);
+                            if (status?.blocked === true) {
+                                const partialTurn = { ...turn, stopReason: 'error' };
+                                throw Object.assign(new Error(quotaBlockedMessage(status)), { partialTurn });
+                            }
+                        }
                         pool.recordTurn(sessionKey, turn);
                         const after = await fingerprint(workspaceRoot, deps.maxScanFiles);
                         const changes = mergeChanges(turn.diffs, compareFingerprints(before, after), workspaceRoot);
@@ -484,7 +501,10 @@ export function cursorCodeTool(deps) {
                 });
             }
             catch (error) {
-                const message = error instanceof Error ? error.message : String(error);
+                const quotaStatus = cursorQuotaFailure(error) ? quotaGuard?.observeFailure?.(error) : undefined;
+                const message = quotaStatus?.blocked === true
+                    ? quotaBlockedMessage(quotaStatus)
+                    : error instanceof Error ? error.message : String(error);
                 logger?.warn?.(`cursor_code failed: ${message}`);
                 return toLosslessJson({
                     ok: false, sessionKey, conversationKey, generation: picked.generation,