@posthog/wizard 1.36.0 → 2.0.0

package/dist/src/{tanstack-start → frameworks/tanstack-start}/tanstack-start-wizard-agent.js

@@ -1,10 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TANSTACK_START_AGENT_CONFIG = void 0;
-const package_manager_detection_1 = require("../lib/package-manager-detection");
-const constants_1 = require("../lib/constants");
-const package_json_1 = require("../utils/package-json");
-const clack_utils_1 = require("../utils/clack-utils");
+const package_manager_detection_1 = require("../../lib/package-manager-detection");
+const constants_1 = require("../../lib/constants");
+const package_json_1 = require("../../utils/package-json");
+const setup_utils_1 = require("../../utils/setup-utils");
 const utils_1 = require("./utils");
 exports.TANSTACK_START_AGENT_CONFIG = {
     metadata: {
@@ -19,11 +19,13 @@ exports.TANSTACK_START_AGENT_CONFIG = {
         getVersionBucket: utils_1.getTanStackStartVersionBucket,
         minimumVersion: '1.0.0',
         getInstalledVersion: async (options) => {
-            const packageJson = await (0, clack_utils_1.getPackageDotJson)(options);
-            return (0, package_json_1.getPackageVersion)('@tanstack/react-start', packageJson);
+            const packageJson = await (0, setup_utils_1.tryGetPackageJson)(options);
+            return packageJson
+                ? (0, package_json_1.getPackageVersion)('@tanstack/react-start', packageJson)
+                : undefined;
         },
         detect: async (options) => {
-            const packageJson = await (0, clack_utils_1.tryGetPackageJson)(options);
+            const packageJson = await (0, setup_utils_1.tryGetPackageJson)(options);
             return packageJson
                 ? (0, package_json_1.hasPackageInstalled)('@tanstack/react-start', packageJson)
                 : false;

package/dist/src/frameworks/tanstack-start/tanstack-start-wizard-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tanstack-start-wizard-agent.js","sourceRoot":"","sources":["../../../../src/frameworks/tanstack-start/tanstack-start-wizard-agent.ts"],"names":[],"mappings":";;;AAGA,mFAAgF;AAChF,mDAAkD;AAClD,2DAIkC;AAClC,yDAA4D;AAC5D,mCAAwD;AAI3C,QAAA,2BAA2B,GACtC;IACE,QAAQ,EAAE;QACR,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,uBAAW,CAAC,aAAa;QACtC,OAAO,EAAE,0CAA0C;KACpD;IAED,SAAS,EAAE;QACT,WAAW,EAAE,uBAAuB;QACpC,kBAAkB,EAAE,gBAAgB;QACpC,UAAU,EAAE,CAAC,WAAoB,EAAE,EAAE,CACnC,IAAA,gCAAiB,EACf,uBAAuB,EACvB,WAA6B,CAC9B;QACH,gBAAgB,EAAE,qCAA6B;QAC/C,cAAc,EAAE,OAAO;QACvB,mBAAmB,EAAE,KAAK,EAAE,OAAsB,EAAE,EAAE;YACpD,MAAM,WAAW,GAAG,MAAM,IAAA,+BAAiB,EAAC,OAAO,CAAC,CAAC;YACrD,OAAO,WAAW;gBAChB,CAAC,CAAC,IAAA,gCAAiB,EAAC,uBAAuB,EAAE,WAAW,CAAC;gBACzD,CAAC,CAAC,SAAS,CAAC;QAChB,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,WAAW,GAAG,MAAM,IAAA,+BAAiB,EAAC,OAAO,CAAC,CAAC;YACrD,OAAO,WAAW;gBAChB,CAAC,CAAC,IAAA,kCAAmB,EAAC,uBAAuB,EAAE,WAAW,CAAC;gBAC3D,CAAC,CAAC,KAAK,CAAC;QACZ,CAAC;QACD,oBAAoB,EAAE,qDAAyB;KAChD;IAED,WAAW,EAAE;QACX,eAAe,EAAE,KAAK;QACtB,UAAU,EAAE,CAAC,MAAc,EAAE,IAAY,EAAE,EAAE,CAAC,CAAC;YAC7C,uBAAuB,EAAE,MAAM;YAC/B,wBAAwB,EAAE,IAAI;SAC/B,CAAC;KACH;IAED,SAAS,EAAE;QACT,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;KACpB;IAED,OAAO,EAAE;QACP,oBAAoB,EAClB,oJAAoJ;QACtJ,yBAAyB,EAAE,GAAG,EAAE;YAC9B,uGAAuG;YACvG,MAAM,WAAW,GAAG,sBAAsB,CAAC;YAE3C,OAAO;gBACL,sBAAsB,WAAW,mCAAmC,WAAW,qBAAqB;aACrG,CAAC;QACJ,CAAC;KACF;IAED,EAAE,EAAE;QACF,cAAc,EAAE,8BAA8B;QAC9C,wBAAwB,EAAE,CAAC;QAC3B,eAAe,EAAE,GAAG,EAAE,CAAC;YACrB,gDAAgD;YAChD,6CAA6C;YAC7C,0CAA0C;SAC3C;QACD,iBAAiB,EAAE,GAAG,EAAE,CAAC;YACvB,wDAAwD;YACxD,qDAAqD;SACtD;KACF;CACF,CAAC","sourcesContent":["/* TanStack Start wizard using posthog-agent with PostHog MCP */\nimport type { WizardOptions } from '../../utils/types';\nimport type { FrameworkConfig } from '../../lib/framework-config';\nimport { detectNodePackageManagers } from '../../lib/package-manager-detection';\nimport { Integration } from '../../lib/constants';\nimport {\n  getPackageVersion,\n  hasPackageInstalled,\n  type PackageDotJson,\n} from '../../utils/package-json';\nimport { tryGetPackageJson } from '../../utils/setup-utils';\nimport { getTanStackStartVersionBucket } from './utils';\n\ntype TanStackStartContext = Record<string, unknown>;\n\nexport const TANSTACK_START_AGENT_CONFIG: FrameworkConfig<TanStackStartContext> =\n  {\n    metadata: {\n      name: 'TanStack Start',\n      integration: Integration.tanstackStart,\n      docsUrl: 'https://posthog.com/docs/libraries/react',\n    },\n\n    detection: {\n      packageName: '@tanstack/react-start',\n      packageDisplayName: 'TanStack Start',\n      getVersion: (packageJson: unknown) =>\n        getPackageVersion(\n          '@tanstack/react-start',\n          packageJson as PackageDotJson,\n        ),\n      getVersionBucket: getTanStackStartVersionBucket,\n      minimumVersion: '1.0.0',\n      getInstalledVersion: async (options: WizardOptions) => {\n        const packageJson = await tryGetPackageJson(options);\n        return packageJson\n          ? getPackageVersion('@tanstack/react-start', packageJson)\n          : undefined;\n      },\n      detect: async (options) => {\n        const packageJson = await tryGetPackageJson(options);\n        return packageJson\n          ? hasPackageInstalled('@tanstack/react-start', packageJson)\n          : false;\n      },\n      detectPackageManager: detectNodePackageManagers,\n    },\n\n    environment: {\n      uploadToHosting: false,\n      getEnvVars: (apiKey: string, host: string) => ({\n        VITE_PUBLIC_POSTHOG_KEY: apiKey,\n        VITE_PUBLIC_POSTHOG_HOST: host,\n      }),\n    },\n\n    analytics: {\n      getTags: () => ({}),\n    },\n\n    prompts: {\n      projectTypeDetection:\n        'This is a JavaScript/TypeScript project. Look for package.json and lockfiles (package-lock.json, yarn.lock, pnpm-lock.yaml, bun.lockb) to confirm.',\n      getAdditionalContextLines: () => {\n        // TanStack Start always uses file-based routing (it's a full-stack framework built on TanStack Router)\n        const frameworkId = 'react-tanstack-start';\n\n        return [\n          `Framework docs ID: ${frameworkId} (use posthog://docs/frameworks/${frameworkId} for documentation)`,\n        ];\n      },\n    },\n\n    ui: {\n      successMessage: 'PostHog integration complete',\n      estimatedDurationMinutes: 8,\n      getOutroChanges: () => [\n        `Analyzed your TanStack Start project structure`,\n        `Created and configured PostHog initializers`,\n        `Integrated PostHog into your application`,\n      ],\n      getOutroNextSteps: () => [\n        'Start your development server to see PostHog in action',\n        'Visit your PostHog dashboard to see incoming events',\n      ],\n    },\n  };\n"]}

package/dist/src/{tanstack-start → frameworks/tanstack-start}/utils.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getTanStackStartVersionBucket = void 0;
-const semver_1 = require("../utils/semver");
+const semver_1 = require("../../utils/semver");
 /**
  * Get TanStack Start version bucket for analytics
  */

package/dist/src/frameworks/tanstack-start/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../src/frameworks/tanstack-start/utils.ts"],"names":[],"mappings":";;;AAAA,+CAAyD;AAEzD;;GAEG;AACU,QAAA,6BAA6B,GAAG,IAAA,4BAAmB,GAAE,CAAC","sourcesContent":["import { createVersionBucket } from '../../utils/semver';\n\n/**\n * Get TanStack Start version bucket for analytics\n */\nexport const getTanStackStartVersionBucket = createVersionBucket();\n"]}

package/dist/src/{vue → frameworks/vue}/vue-wizard-agent.d.ts

@@ -1,4 +1,4 @@
-import type { FrameworkConfig } from '../lib/framework-config';
+import type { FrameworkConfig } from '../../lib/framework-config';
 type VueContext = Record<string, unknown>;
 export declare const VUE_AGENT_CONFIG: FrameworkConfig<VueContext>;
 export {};

package/dist/src/{vue → frameworks/vue}/vue-wizard-agent.js

@@ -1,11 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VUE_AGENT_CONFIG = void 0;
-const package_manager_detection_1 = require("../lib/package-manager-detection");
-const constants_1 = require("../lib/constants");
-const package_json_1 = require("../utils/package-json");
-const clack_utils_1 = require("../utils/clack-utils");
-const semver_1 = require("../utils/semver");
+const package_manager_detection_1 = require("../../lib/package-manager-detection");
+const constants_1 = require("../../lib/constants");
+const package_json_1 = require("../../utils/package-json");
+const setup_utils_1 = require("../../utils/setup-utils");
+const semver_1 = require("../../utils/semver");
 const getVueVersionBucket = (0, semver_1.createVersionBucket)();
 exports.VUE_AGENT_CONFIG = {
     metadata: {
@@ -20,11 +20,11 @@ exports.VUE_AGENT_CONFIG = {
         getVersion: (packageJson) => (0, package_json_1.getPackageVersion)('vue', packageJson),
         getVersionBucket: getVueVersionBucket,
         getInstalledVersion: async (options) => {
-            const packageJson = await (0, clack_utils_1.getPackageDotJson)(options);
-            return (0, package_json_1.getPackageVersion)('vue', packageJson);
+            const packageJson = await (0, setup_utils_1.tryGetPackageJson)(options);
+            return packageJson ? (0, package_json_1.getPackageVersion)('vue', packageJson) : undefined;
         },
         detect: async (options) => {
-            const packageJson = await (0, clack_utils_1.tryGetPackageJson)(options);
+            const packageJson = await (0, setup_utils_1.tryGetPackageJson)(options);
             return packageJson ? (0, package_json_1.hasPackageInstalled)('vue', packageJson) : false;
         },
         detectPackageManager: package_manager_detection_1.detectNodePackageManagers,

package/dist/src/frameworks/vue/vue-wizard-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vue-wizard-agent.js","sourceRoot":"","sources":["../../../../src/frameworks/vue/vue-wizard-agent.ts"],"names":[],"mappings":";;;AAEA,mFAAgF;AAChF,mDAAkD;AAClD,2DAIkC;AAClC,yDAA4D;AAC5D,+CAAyD;AAEzD,MAAM,mBAAmB,GAAG,IAAA,4BAAmB,GAAE,CAAC;AAIrC,QAAA,gBAAgB,GAAgC;IAC3D,QAAQ,EAAE;QACR,IAAI,EAAE,KAAK;QACX,WAAW,EAAE,uBAAW,CAAC,GAAG;QAC5B,OAAO,EAAE,wCAAwC;QACjD,IAAI,EAAE,IAAI;KACX;IAED,SAAS,EAAE;QACT,WAAW,EAAE,KAAK;QAClB,kBAAkB,EAAE,KAAK;QACzB,UAAU,EAAE,CAAC,WAAoB,EAAE,EAAE,CACnC,IAAA,gCAAiB,EAAC,KAAK,EAAE,WAA6B,CAAC;QACzD,gBAAgB,EAAE,mBAAmB;QACrC,mBAAmB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACrC,MAAM,WAAW,GAAG,MAAM,IAAA,+BAAiB,EAAC,OAAO,CAAC,CAAC;YACrD,OAAO,WAAW,CAAC,CAAC,CAAC,IAAA,gCAAiB,EAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACzE,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,WAAW,GAAG,MAAM,IAAA,+BAAiB,EAAC,OAAO,CAAC,CAAC;YACrD,OAAO,WAAW,CAAC,CAAC,CAAC,IAAA,kCAAmB,EAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACvE,CAAC;QACD,oBAAoB,EAAE,qDAAyB;KAChD;IAED,WAAW,EAAE;QACX,eAAe,EAAE,IAAI;QACrB,UAAU,EAAE,CAAC,MAAc,EAAE,IAAY,EAAE,EAAE,CAAC,CAAC;YAC7C,gBAAgB,EAAE,MAAM;YACxB,iBAAiB,EAAE,IAAI;SACxB,CAAC;KACH;IAED,SAAS,EAAE;QACT,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;KACpB;IAED,OAAO,EAAE;QACP,oBAAoB,EAClB,oJAAoJ;QACtJ,yBAAyB,EAAE,GAAG,EAAE;YAC9B,MAAM,WAAW,GAAG,KAAK,CAAC;YAC1B,OAAO;gBACL,sBAAsB,WAAW,mCAAmC,WAAW,qBAAqB;aACrG,CAAC;QACJ,CAAC;KACF;IAED,EAAE,EAAE;QACF,cAAc,EAAE,8BAA8B;QAC9C,wBAAwB,EAAE,CAAC;QAC3B,eAAe,EAAE,GAAG,EAAE,CAAC;YACrB,qCAAqC;YACrC,6CAA6C;YAC7C,0CAA0C;SAC3C;QACD,iBAAiB,EAAE,GAAG,EAAE,CAAC;YACvB,wDAAwD;YACxD,qDAAqD;SACtD;KACF;CACF,CAAC","sourcesContent":["/* Vue wizard using posthog-agent with PostHog MCP */\nimport type { FrameworkConfig } from '../../lib/framework-config';\nimport { detectNodePackageManagers } from '../../lib/package-manager-detection';\nimport { Integration } from '../../lib/constants';\nimport {\n  getPackageVersion,\n  hasPackageInstalled,\n  type PackageDotJson,\n} from '../../utils/package-json';\nimport { tryGetPackageJson } from '../../utils/setup-utils';\nimport { createVersionBucket } from '../../utils/semver';\n\nconst getVueVersionBucket = createVersionBucket();\n\ntype VueContext = Record<string, unknown>;\n\nexport const VUE_AGENT_CONFIG: FrameworkConfig<VueContext> = {\n  metadata: {\n    name: 'Vue',\n    integration: Integration.vue,\n    docsUrl: 'https://posthog.com/docs/libraries/vue',\n    beta: true,\n  },\n\n  detection: {\n    packageName: 'vue',\n    packageDisplayName: 'Vue',\n    getVersion: (packageJson: unknown) =>\n      getPackageVersion('vue', packageJson as PackageDotJson),\n    getVersionBucket: getVueVersionBucket,\n    getInstalledVersion: async (options) => {\n      const packageJson = await tryGetPackageJson(options);\n      return packageJson ? getPackageVersion('vue', packageJson) : undefined;\n    },\n    detect: async (options) => {\n      const packageJson = await tryGetPackageJson(options);\n      return packageJson ? hasPackageInstalled('vue', packageJson) : false;\n    },\n    detectPackageManager: detectNodePackageManagers,\n  },\n\n  environment: {\n    uploadToHosting: true,\n    getEnvVars: (apiKey: string, host: string) => ({\n      VITE_POSTHOG_KEY: apiKey,\n      VITE_POSTHOG_HOST: host,\n    }),\n  },\n\n  analytics: {\n    getTags: () => ({}),\n  },\n\n  prompts: {\n    projectTypeDetection:\n      'This is a JavaScript/TypeScript project. Look for package.json and lockfiles (package-lock.json, yarn.lock, pnpm-lock.yaml, bun.lockb) to confirm.',\n    getAdditionalContextLines: () => {\n      const frameworkId = 'vue';\n      return [\n        `Framework docs ID: ${frameworkId} (use posthog://docs/frameworks/${frameworkId} for documentation)`,\n      ];\n    },\n  },\n\n  ui: {\n    successMessage: 'PostHog integration complete',\n    estimatedDurationMinutes: 5,\n    getOutroChanges: () => [\n      'Analyzed your Vue project structure',\n      'Created and configured PostHog initializers',\n      'Integrated PostHog into your application',\n    ],\n    getOutroNextSteps: () => [\n      'Start your development server to see PostHog in action',\n      'Visit your PostHog dashboard to see incoming events',\n    ],\n  },\n};\n"]}

package/dist/src/lib/__tests__/agent-interface.test.js

@@ -1,11 +1,8 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 const agent_interface_1 = require("../agent-interface");
+const wizard_session_1 = require("../wizard-session");
 // Mock dependencies
-jest.mock('../../utils/clack');
 jest.mock('../../utils/analytics');
 jest.mock('../../utils/debug');
 // Mock the SDK module
@@ -13,9 +10,38 @@ const mockQuery = jest.fn();
 jest.mock('@anthropic-ai/claude-agent-sdk', () => ({
     query: (...args) => mockQuery(...args),
 }));
-// Get mocked clack for spinner
-const clack_1 = __importDefault(require("../../utils/clack"));
-const mockClack = clack_1.default;
+// Mock the UI layer
+const mockUIInstance = {
+    log: {
+        step: jest.fn(),
+        success: jest.fn(),
+        error: jest.fn(),
+        warn: jest.fn(),
+        info: jest.fn(),
+    },
+    spinner: jest.fn(),
+    select: jest.fn(),
+    confirm: jest.fn(),
+    text: jest.fn(),
+    intro: jest.fn(),
+    outro: jest.fn(),
+    cancel: jest.fn(),
+    note: jest.fn(),
+    isCancel: jest.fn(),
+    setDetectedFramework: jest.fn(),
+    setCredentials: jest.fn(),
+    pushStatus: jest.fn(),
+    setLoginUrl: jest.fn(),
+    showServiceStatus: jest.fn(),
+    showSettingsOverride: jest.fn(),
+    startRun: jest.fn(),
+    syncTodos: jest.fn(),
+    groupMultiselect: jest.fn(),
+    multiselect: jest.fn(),
+};
+jest.mock('../../ui', () => ({
+    getUI: () => mockUIInstance,
+}));
 describe('runAgent', () => {
     let mockSpinner;
     const defaultOptions = {
@@ -39,18 +65,11 @@ describe('runAgent', () => {
         mockSpinner = {
             start: jest.fn(),
             stop: jest.fn(),
-            message: '',
-        };
-        mockClack.spinner = jest.fn().mockReturnValue(mockSpinner);
-        mockClack.log = {
-            step: jest.fn(),
-            success: jest.fn(),
-            error: jest.fn(),
-            warn: jest.fn(),
-            warning: jest.fn(),
-            info: jest.fn(),
             message: jest.fn(),
         };
+        mockUIInstance.spinner.mockReturnValue(mockSpinner);
+        // Reset log mocks
+        Object.values(mockUIInstance.log).forEach((fn) => fn.mockReset());
     });
     describe('race condition handling', () => {
         it('should return success when agent completes successfully then SDK cleanup fails', async () => {
@@ -185,9 +204,62 @@ describe('runAgent', () => {
             // Should return success (empty object), not error
             expect(result).toEqual({});
             expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');
-            // clack.log.error should NOT have been called (errors suppressed for user)
-            expect(mockClack.log.error).not.toHaveBeenCalled();
+            // ui.log.error should NOT have been called (errors suppressed for user)
+            expect(mockUIInstance.log.error).not.toHaveBeenCalled();
         });
     });
 });
+describe('createStopHook', () => {
+    const hookInput = { stop_hook_active: false };
+    it('empty queue: first call blocks for remark, second allows stop', () => {
+        const hook = (0, agent_interface_1.createStopHook)([]);
+        // First call → remark prompt
+        const first = hook(hookInput);
+        expect(first).toHaveProperty('decision', 'block');
+        expect(first.reason).toContain('WIZARD-REMARK');
+        // Second call → allow stop
+        const second = hook(hookInput);
+        expect(second).toEqual({});
+    });
+    it('single feature: feature prompt, then remark, then allow stop', () => {
+        const hook = (0, agent_interface_1.createStopHook)([wizard_session_1.AdditionalFeature.LLM]);
+        // First call → LLM feature prompt
+        const first = hook(hookInput);
+        expect(first).toHaveProperty('decision', 'block');
+        expect(first.reason).toBe(wizard_session_1.ADDITIONAL_FEATURE_PROMPTS[wizard_session_1.AdditionalFeature.LLM]);
+        // Second call → remark prompt
+        const second = hook(hookInput);
+        expect(second).toHaveProperty('decision', 'block');
+        expect(second.reason).toContain('WIZARD-REMARK');
+        // Third call → allow stop
+        const third = hook(hookInput);
+        expect(third).toEqual({});
+    });
+    it('multiple queue entries: drains all, then remark, then allow stop', () => {
+        // Queue the same feature twice to exercise multi-item draining
+        const hook = (0, agent_interface_1.createStopHook)([wizard_session_1.AdditionalFeature.LLM, wizard_session_1.AdditionalFeature.LLM]);
+        // First call → LLM prompt
+        const first = hook(hookInput);
+        expect(first).toHaveProperty('decision', 'block');
+        expect(first.reason).toBe(wizard_session_1.ADDITIONAL_FEATURE_PROMPTS[wizard_session_1.AdditionalFeature.LLM]);
+        // Second call → LLM prompt again
+        const second = hook(hookInput);
+        expect(second).toHaveProperty('decision', 'block');
+        expect(second.reason).toBe(wizard_session_1.ADDITIONAL_FEATURE_PROMPTS[wizard_session_1.AdditionalFeature.LLM]);
+        // Third call → remark prompt
+        const third = hook(hookInput);
+        expect(third).toHaveProperty('decision', 'block');
+        expect(third.reason).toContain('WIZARD-REMARK');
+        // Fourth call → allow stop
+        const fourth = hook(hookInput);
+        expect(fourth).toEqual({});
+    });
+    it('allow stop is idempotent after all phases complete', () => {
+        const hook = (0, agent_interface_1.createStopHook)([]);
+        hook(hookInput); // remark
+        hook(hookInput); // allow
+        const extra = hook(hookInput); // still allow
+        expect(extra).toEqual({});
+    });
+});
 //# sourceMappingURL=agent-interface.test.js.map

package/dist/src/lib/__tests__/agent-interface.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-interface.test.js","sourceRoot":"","sources":["../../../../src/lib/__tests__/agent-interface.test.ts"],"names":[],"mappings":";;;;;AAAA,wDAA8C;AAG9C,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAC/B,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACnC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAE/B,sBAAsB;AACtB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,+BAA+B;AAC/B,8DAAsC;AACtC,MAAM,SAAS,GAAG,eAAkC,CAAC;AAErD,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,WAIH,CAAC;IAEF,MAAM,cAAc,GAAkB;QACpC,KAAK,EAAE,KAAK;QACZ,UAAU,EAAE,WAAW;QACvB,YAAY,EAAE,KAAK;QACnB,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,KAAK;QACb,QAAQ,EAAE,KAAK;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,KAAK;QACX,SAAS,EAAE,KAAK;KACjB,CAAC;IAEF,MAAM,kBAAkB,GAAG;QACzB,gBAAgB,EAAE,WAAW;QAC7B,UAAU,EAAE,EAAE;QACd,KAAK,EAAE,0BAA0B;KAClC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,WAAW,GAAG;YACZ,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QAC3D,SAAS,CAAC,GAAG,GAAG;YACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;YAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;YAClB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;SACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,qCAAqC;YACrC,yCAAyC;YACzC,6DAA6D;YAC7D,iEAAiE;YACjE,yBAAyB;YACzB,8EAA8E;YAE9E,QAAQ,CAAC,CAAC,6BAA6B;gBACrC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,8BAA8B;iBACvC,CAAC;gBAEF,2DAA2D;gBAC3D,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC1E,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAA0D,EAC1D;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,qEAAqE;YAErE,QAAQ,CAAC,CAAC,0BAA0B;gBAClC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,mCAAmC;gBACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,0BAA0B,EAAE,CAAC,CAAC;YAExD,MAAM,MAAM,CACV,IAAA,0BAAQ,EACN,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAA0D,EAC1D;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAEtC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,2DAA2D;YAC3D,mEAAmE;YAEnE,QAAQ,CAAC,CAAC,4BAA4B;gBACpC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS,EAAE,2CAA2C;oBAC/D,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,sCAAsC;iBAC/C,CAAC;gBAEF,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,4BAA4B,EAAE,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAA0D,EAC1D;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,gFAAgF;YAChF,iEAAiE;YACjE,+DAA+D;YAC/D,0CAA0C;YAC1C,mEAAmE;YACnE,EAAE;YACF,0EAA0E;YAC1E,4DAA4D;YAE5D,QAAQ,CAAC,CAAC,yCAAyC;gBACjD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,sDAAsD;gBACtD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,SAAS,EAAE,GAAG;oBACd,MAAM,EAAE,oDAAoD;oBAC5D,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,IAAI;iBACrB,CAAC;gBAEF,iEAAiE;gBACjE,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,CAAC;oBACjB,MAAM,EAAE;wBACN,sDAAsD;wBACtD,qDAAqD;wBACrD,qDAAqD;wBACrD,mCAAmC;qBACpC;iBACF,CAAC;YACJ,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,yCAAyC,EAAE,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAA0D,EAC1D;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAE9D,2EAA2E;YAC3E,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { runAgent } from '../agent-interface';\nimport type { WizardOptions } from '../../utils/types';\n\n// Mock dependencies\njest.mock('../../utils/clack');\njest.mock('../../utils/analytics');\njest.mock('../../utils/debug');\n\n// Mock the SDK module\nconst mockQuery = jest.fn();\njest.mock('@anthropic-ai/claude-agent-sdk', () => ({\n  query: (...args: unknown[]) => mockQuery(...args),\n}));\n\n// Get mocked clack for spinner\nimport clack from '../../utils/clack';\nconst mockClack = clack as jest.Mocked<typeof clack>;\n\ndescribe('runAgent', () => {\n  let mockSpinner: {\n    start: jest.Mock;\n    stop: jest.Mock;\n    message: string;\n  };\n\n  const defaultOptions: WizardOptions = {\n    debug: false,\n    installDir: '/test/dir',\n    forceInstall: false,\n    default: false,\n    signup: false,\n    localMcp: false,\n    ci: false,\n    menu: false,\n    benchmark: false,\n  };\n\n  const defaultAgentConfig = {\n    workingDirectory: '/test/dir',\n    mcpServers: {},\n    model: 'claude-opus-4-5-20251101',\n  };\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    mockSpinner = {\n      start: jest.fn(),\n      stop: jest.fn(),\n      message: '',\n    };\n\n    mockClack.spinner = jest.fn().mockReturnValue(mockSpinner);\n    mockClack.log = {\n      step: jest.fn(),\n      success: jest.fn(),\n      error: jest.fn(),\n      warn: jest.fn(),\n      warning: jest.fn(),\n      info: jest.fn(),\n      message: jest.fn(),\n    };\n  });\n\n  describe('race condition handling', () => {\n    it('should return success when agent completes successfully then SDK cleanup fails', async () => {\n      // This simulates the race condition:\n      // 1. Agent completes with success result\n      // 2. signalDone() is called, completing the prompt generator\n      // 3. SDK tries to send cleanup command while streaming is active\n      // 4. SDK throws an error\n      // The fix should recognize we already got a success and return success anyway\n\n      function* mockGeneratorWithCleanupError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          result: 'Agent completed successfully',\n        };\n\n        // Simulate the SDK cleanup error that occurs after success\n        throw new Error('only prompt commands are supported in streaming mode');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithCleanupError());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as ReturnType<typeof clack.spinner>,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not throw\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n    });\n\n    it('should still throw when no success result was received before error', async () => {\n      // If we never got a success result, errors should propagate normally\n\n      function* mockGeneratorWithOnlyError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // No success result, just an error\n        throw new Error('Actual SDK error');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithOnlyError());\n\n      await expect(\n        runAgent(\n          defaultAgentConfig,\n          'test prompt',\n          defaultOptions,\n          mockSpinner as unknown as ReturnType<typeof clack.spinner>,\n          {\n            successMessage: 'Test success',\n            errorMessage: 'Test error',\n          },\n        ),\n      ).rejects.toThrow('Actual SDK error');\n\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test error');\n    });\n\n    it('should not treat error results as success', async () => {\n      // A result with is_error: true should not count as success\n      // Even if subtype is 'success', the is_error flag takes precedence\n\n      function* mockGeneratorWithErrorResult() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success', // subtype can be success but is_error true\n          is_error: true,\n          result: 'API Error: 500 Internal Server Error',\n        };\n\n        throw new Error('Process exited with code 1');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithErrorResult());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as ReturnType<typeof clack.spinner>,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return API error, not success\n      expect(result.error).toBe('WIZARD_API_ERROR');\n      expect(result.message).toContain('API Error');\n    });\n\n    it('should suppress user-facing errors when SDK yields error result after success', async () => {\n      // This test models actual SDK behavior where the SDK emits TWO result messages:\n      // 1. SDK yields success result (num_turns: 105, is_error: false)\n      // 2. SDK yields a SECOND result with is_error: true containing\n      //    accumulated cleanup/telemetry errors\n      // 3. The errors should be logged to file but NOT shown to the user\n      //\n      // This differs from the thrown exception test above - here the SDK YIELDS\n      // an error result message instead of THROWING an exception.\n\n      function* mockGeneratorWithYieldedErrorAfterSuccess() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // First result: success (this is the real completion)\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          num_turns: 105,\n          result: '[WIZARD-REMARK] Integration completed successfully',\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 5.83,\n        };\n\n        // Second result: error (SDK cleanup noise - yielded, not thrown)\n        yield {\n          type: 'result',\n          subtype: 'error_during_execution',\n          is_error: true,\n          num_turns: 0,\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 0,\n          errors: [\n            'only prompt commands are supported in streaming mode',\n            'Error: 1P event logging: 14 events failed to export',\n            'Error: 1P event logging: 13 events failed to export',\n            'Error: Failed to export 14 events',\n          ],\n        };\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithYieldedErrorAfterSuccess());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as ReturnType<typeof clack.spinner>,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not error\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n\n      // clack.log.error should NOT have been called (errors suppressed for user)\n      expect(mockClack.log.error).not.toHaveBeenCalled();\n    });\n  });\n});\n"]}
+{"version":3,"file":"agent-interface.test.js","sourceRoot":"","sources":["../../../../src/lib/__tests__/agent-interface.test.ts"],"names":[],"mappings":";;AAAA,wDAA8D;AAG9D,sDAG2B;AAE3B,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACnC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAE/B,sBAAsB;AACtB,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC5B,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IACjD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;CAClD,CAAC,CAAC,CAAC;AAEJ,oBAAoB;AACpB,MAAM,cAAc,GAAG;IACrB,GAAG,EAAE;QACH,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;KAChB;IACD,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;IAClB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;IACf,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;IACrB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;IACtB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC5B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;IACpB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;CACvB,CAAC;AACF,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3B,KAAK,EAAE,GAAG,EAAE,CAAC,cAAc;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,WAIH,CAAC;IAEF,MAAM,cAAc,GAAkB;QACpC,KAAK,EAAE,KAAK;QACZ,UAAU,EAAE,WAAW;QACvB,YAAY,EAAE,KAAK;QACnB,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,KAAK;QACb,QAAQ,EAAE,KAAK;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,KAAK;QACX,SAAS,EAAE,KAAK;KACjB,CAAC;IAEF,MAAM,kBAAkB,GAAG;QACzB,gBAAgB,EAAE,WAAW;QAC7B,UAAU,EAAE,EAAE;QACd,KAAK,EAAE,0BAA0B;KAClC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,WAAW,GAAG;YACZ,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;SACnB,CAAC;QAEF,cAAc,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACpD,kBAAkB;QAClB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,qCAAqC;YACrC,yCAAyC;YACzC,6DAA6D;YAC7D,iEAAiE;YACjE,yBAAyB;YACzB,8EAA8E;YAE9E,QAAQ,CAAC,CAAC,6BAA6B;gBACrC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,8BAA8B;iBACvC,CAAC;gBAEF,2DAA2D;gBAC3D,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC1E,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,6BAA6B,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,qEAAqE;YAErE,QAAQ,CAAC,CAAC,0BAA0B;gBAClC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,mCAAmC;gBACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,0BAA0B,EAAE,CAAC,CAAC;YAExD,MAAM,MAAM,CACV,IAAA,0BAAQ,EACN,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAEtC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,2DAA2D;YAC3D,mEAAmE;YAEnE,QAAQ,CAAC,CAAC,4BAA4B;gBACpC,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS,EAAE,2CAA2C;oBAC/D,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,sCAAsC;iBAC/C,CAAC;gBAEF,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,4BAA4B,EAAE,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,gFAAgF;YAChF,iEAAiE;YACjE,+DAA+D;YAC/D,0CAA0C;YAC1C,mEAAmE;YACnE,EAAE;YACF,0EAA0E;YAC1E,4DAA4D;YAE5D,QAAQ,CAAC,CAAC,yCAAyC;gBACjD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,MAAM;oBACf,KAAK,EAAE,0BAA0B;oBACjC,KAAK,EAAE,EAAE;oBACT,WAAW,EAAE,EAAE;iBAChB,CAAC;gBAEF,sDAAsD;gBACtD,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,KAAK;oBACf,SAAS,EAAE,GAAG;oBACd,MAAM,EAAE,oDAAoD;oBAC5D,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,IAAI;iBACrB,CAAC;gBAEF,iEAAiE;gBACjE,MAAM;oBACJ,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,sCAAsC;oBAClD,cAAc,EAAE,CAAC;oBACjB,MAAM,EAAE;wBACN,sDAAsD;wBACtD,qDAAqD;wBACrD,qDAAqD;wBACrD,mCAAmC;qBACpC;iBACF,CAAC;YACJ,CAAC;YAED,SAAS,CAAC,eAAe,CAAC,yCAAyC,EAAE,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAQ,EAC3B,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,WAAuC,EACvC;gBACE,cAAc,EAAE,cAAc;gBAC9B,YAAY,EAAE,YAAY;aAC3B,CACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;YAE9D,wEAAwE;YACxE,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,SAAS,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;IAE9C,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErD,kCAAkC;QAClC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,8BAA8B;QAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAEzE,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,+DAA+D;QAC/D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,CAAC,kCAAiB,CAAC,GAAG,EAAE,kCAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAE5E,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,IAAI,CAC/C,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,iCAAiC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,CAAE,MAA6B,CAAC,MAAM,CAAC,CAAC,IAAI,CAChD,2CAA0B,CAAC,kCAAiB,CAAC,GAAG,CAAC,CAClD,CAAC;QAEF,6BAA6B;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAE,KAA4B,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAExE,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,IAAI,GAAG,IAAA,gCAAc,EAAC,EAAE,CAAC,CAAC;QAEhC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QAC1B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { runAgent, createStopHook } from '../agent-interface';\nimport type { WizardOptions } from '../../utils/types';\nimport type { SpinnerHandle } from '../../ui';\nimport {\n  AdditionalFeature,\n  ADDITIONAL_FEATURE_PROMPTS,\n} from '../wizard-session';\n\n// Mock dependencies\njest.mock('../../utils/analytics');\njest.mock('../../utils/debug');\n\n// Mock the SDK module\nconst mockQuery = jest.fn();\njest.mock('@anthropic-ai/claude-agent-sdk', () => ({\n  query: (...args: unknown[]) => mockQuery(...args),\n}));\n\n// Mock the UI layer\nconst mockUIInstance = {\n  log: {\n    step: jest.fn(),\n    success: jest.fn(),\n    error: jest.fn(),\n    warn: jest.fn(),\n    info: jest.fn(),\n  },\n  spinner: jest.fn(),\n  select: jest.fn(),\n  confirm: jest.fn(),\n  text: jest.fn(),\n  intro: jest.fn(),\n  outro: jest.fn(),\n  cancel: jest.fn(),\n  note: jest.fn(),\n  isCancel: jest.fn(),\n  setDetectedFramework: jest.fn(),\n  setCredentials: jest.fn(),\n  pushStatus: jest.fn(),\n  setLoginUrl: jest.fn(),\n  showServiceStatus: jest.fn(),\n  showSettingsOverride: jest.fn(),\n  startRun: jest.fn(),\n  syncTodos: jest.fn(),\n  groupMultiselect: jest.fn(),\n  multiselect: jest.fn(),\n};\njest.mock('../../ui', () => ({\n  getUI: () => mockUIInstance,\n}));\n\ndescribe('runAgent', () => {\n  let mockSpinner: {\n    start: jest.Mock;\n    stop: jest.Mock;\n    message: jest.Mock;\n  };\n\n  const defaultOptions: WizardOptions = {\n    debug: false,\n    installDir: '/test/dir',\n    forceInstall: false,\n    default: false,\n    signup: false,\n    localMcp: false,\n    ci: false,\n    menu: false,\n    benchmark: false,\n  };\n\n  const defaultAgentConfig = {\n    workingDirectory: '/test/dir',\n    mcpServers: {},\n    model: 'claude-opus-4-5-20251101',\n  };\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    mockSpinner = {\n      start: jest.fn(),\n      stop: jest.fn(),\n      message: jest.fn(),\n    };\n\n    mockUIInstance.spinner.mockReturnValue(mockSpinner);\n    // Reset log mocks\n    Object.values(mockUIInstance.log).forEach((fn) => fn.mockReset());\n  });\n\n  describe('race condition handling', () => {\n    it('should return success when agent completes successfully then SDK cleanup fails', async () => {\n      // This simulates the race condition:\n      // 1. Agent completes with success result\n      // 2. signalDone() is called, completing the prompt generator\n      // 3. SDK tries to send cleanup command while streaming is active\n      // 4. SDK throws an error\n      // The fix should recognize we already got a success and return success anyway\n\n      function* mockGeneratorWithCleanupError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          result: 'Agent completed successfully',\n        };\n\n        // Simulate the SDK cleanup error that occurs after success\n        throw new Error('only prompt commands are supported in streaming mode');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithCleanupError());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not throw\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n    });\n\n    it('should still throw when no success result was received before error', async () => {\n      // If we never got a success result, errors should propagate normally\n\n      function* mockGeneratorWithOnlyError() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // No success result, just an error\n        throw new Error('Actual SDK error');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithOnlyError());\n\n      await expect(\n        runAgent(\n          defaultAgentConfig,\n          'test prompt',\n          defaultOptions,\n          mockSpinner as unknown as SpinnerHandle,\n          {\n            successMessage: 'Test success',\n            errorMessage: 'Test error',\n          },\n        ),\n      ).rejects.toThrow('Actual SDK error');\n\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test error');\n    });\n\n    it('should not treat error results as success', async () => {\n      // A result with is_error: true should not count as success\n      // Even if subtype is 'success', the is_error flag takes precedence\n\n      function* mockGeneratorWithErrorResult() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        yield {\n          type: 'result',\n          subtype: 'success', // subtype can be success but is_error true\n          is_error: true,\n          result: 'API Error: 500 Internal Server Error',\n        };\n\n        throw new Error('Process exited with code 1');\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithErrorResult());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return API error, not success\n      expect(result.error).toBe('WIZARD_API_ERROR');\n      expect(result.message).toContain('API Error');\n    });\n\n    it('should suppress user-facing errors when SDK yields error result after success', async () => {\n      // This test models actual SDK behavior where the SDK emits TWO result messages:\n      // 1. SDK yields success result (num_turns: 105, is_error: false)\n      // 2. SDK yields a SECOND result with is_error: true containing\n      //    accumulated cleanup/telemetry errors\n      // 3. The errors should be logged to file but NOT shown to the user\n      //\n      // This differs from the thrown exception test above - here the SDK YIELDS\n      // an error result message instead of THROWING an exception.\n\n      function* mockGeneratorWithYieldedErrorAfterSuccess() {\n        yield {\n          type: 'system',\n          subtype: 'init',\n          model: 'claude-opus-4-5-20251101',\n          tools: [],\n          mcp_servers: [],\n        };\n\n        // First result: success (this is the real completion)\n        yield {\n          type: 'result',\n          subtype: 'success',\n          is_error: false,\n          num_turns: 105,\n          result: '[WIZARD-REMARK] Integration completed successfully',\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 5.83,\n        };\n\n        // Second result: error (SDK cleanup noise - yielded, not thrown)\n        yield {\n          type: 'result',\n          subtype: 'error_during_execution',\n          is_error: true,\n          num_turns: 0,\n          session_id: '2ce14bda-6d86-4220-b5bb-ab24f7004290',\n          total_cost_usd: 0,\n          errors: [\n            'only prompt commands are supported in streaming mode',\n            'Error: 1P event logging: 14 events failed to export',\n            'Error: 1P event logging: 13 events failed to export',\n            'Error: Failed to export 14 events',\n          ],\n        };\n      }\n\n      mockQuery.mockReturnValue(mockGeneratorWithYieldedErrorAfterSuccess());\n\n      const result = await runAgent(\n        defaultAgentConfig,\n        'test prompt',\n        defaultOptions,\n        mockSpinner as unknown as SpinnerHandle,\n        {\n          successMessage: 'Test success',\n          errorMessage: 'Test error',\n        },\n      );\n\n      // Should return success (empty object), not error\n      expect(result).toEqual({});\n      expect(mockSpinner.stop).toHaveBeenCalledWith('Test success');\n\n      // ui.log.error should NOT have been called (errors suppressed for user)\n      expect(mockUIInstance.log.error).not.toHaveBeenCalled();\n    });\n  });\n});\n\ndescribe('createStopHook', () => {\n  const hookInput = { stop_hook_active: false };\n\n  it('empty queue: first call blocks for remark, second allows stop', () => {\n    const hook = createStopHook([]);\n\n    // First call → remark prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Second call → allow stop\n    const second = hook(hookInput);\n    expect(second).toEqual({});\n  });\n\n  it('single feature: feature prompt, then remark, then allow stop', () => {\n    const hook = createStopHook([AdditionalFeature.LLM]);\n\n    // First call → LLM feature prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → remark prompt\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Third call → allow stop\n    const third = hook(hookInput);\n    expect(third).toEqual({});\n  });\n\n  it('multiple queue entries: drains all, then remark, then allow stop', () => {\n    // Queue the same feature twice to exercise multi-item draining\n    const hook = createStopHook([AdditionalFeature.LLM, AdditionalFeature.LLM]);\n\n    // First call → LLM prompt\n    const first = hook(hookInput);\n    expect(first).toHaveProperty('decision', 'block');\n    expect((first as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Second call → LLM prompt again\n    const second = hook(hookInput);\n    expect(second).toHaveProperty('decision', 'block');\n    expect((second as { reason: string }).reason).toBe(\n      ADDITIONAL_FEATURE_PROMPTS[AdditionalFeature.LLM],\n    );\n\n    // Third call → remark prompt\n    const third = hook(hookInput);\n    expect(third).toHaveProperty('decision', 'block');\n    expect((third as { reason: string }).reason).toContain('WIZARD-REMARK');\n\n    // Fourth call → allow stop\n    const fourth = hook(hookInput);\n    expect(fourth).toEqual({});\n  });\n\n  it('allow stop is idempotent after all phases complete', () => {\n    const hook = createStopHook([]);\n\n    hook(hookInput); // remark\n    hook(hookInput); // allow\n    const extra = hook(hookInput); // still allow\n    expect(extra).toEqual({});\n  });\n});\n"]}

package/dist/src/lib/agent-interface.d.ts

@@ -2,8 +2,9 @@
  * Shared agent interface for PostHog wizards
  * Uses Claude Agent SDK directly with PostHog LLM gateway
  */
-import clack from '../utils/clack';
+import { type SpinnerHandle } from '../ui';
 import type { WizardOptions } from '../utils/types';
+import { type AdditionalFeature } from './wizard-session';
 import type { PackageManagerDetector } from './package-manager-detection';
 type McpServersConfig = any;
 export declare const AgentSignals: {
@@ -33,6 +34,22 @@ export declare enum AgentErrorType {
     /** Generic API error */
     API_ERROR = "WIZARD_API_ERROR"
 }
+/**
+ * Check if .claude/settings.json in the project directory contains env
+ * overrides for blocking keys that block the Wizard from accessing the PostHog LLM Gateway.
+ * Returns the list of matched key names, or an empty array if none found.
+ */
+export declare function checkClaudeSettingsOverrides(workingDirectory: string): string[];
+/**
+ * Copy .claude/settings.json to .wizard-backup (overwriting if it exists),
+ * then remove the original so the SDK doesn't load the blocking overrides.
+ */
+export declare function backupAndFixClaudeSettings(workingDirectory: string): boolean;
+/**
+ * Restore .claude/settings.json from .wizard-backup.
+ * Copies (not moves) so the backup is preserved.
+ */
+export declare function restoreClaudeSettings(workingDirectory: string): void;
 export type AgentConfig = {
     workingDirectory: string;
     posthogMcpUrl: string;
@@ -42,7 +59,29 @@ export type AgentConfig = {
         url: string;
     }>;
     detectPackageManager: PackageManagerDetector;
+    /** Feature flag key -> variant (evaluated at start of run). */
+    wizardFlags?: Record<string, string>;
+    wizardMetadata?: Record<string, string>;
+};
+/**
+ * Stop hook return type: either allow stop or block with a reason.
+ */
+export type StopHookResult = Record<string, never> | {
+    decision: 'block';
+    reason: string;
 };
+/**
+ * Create a stop hook callback that drains the additional feature queue,
+ * then collects a remark, then allows stop.
+ *
+ * Three-phase logic using closure state:
+ *   Phase 1 — drain queue: block with each feature prompt in order
+ *   Phase 2 — collect remark (once): block with remark prompt
+ *   Phase 3 — allow stop: return {}
+ */
+export declare function createStopHook(featureQueue: readonly AdditionalFeature[]): (input: {
+    stop_hook_active: boolean;
+}) => StopHookResult;
 /**
  * Internal configuration object returned by initializeAgent
  */
@@ -50,7 +89,14 @@ type AgentRunConfig = {
     workingDirectory: string;
     mcpServers: McpServersConfig;
     model: string;
+    wizardFlags?: Record<string, string>;
+    wizardMetadata?: Record<string, string>;
 };
+/**
+ * Select wizard metadata from WIZARD_VARIANTS using the variant feature flag.
+ * If the flag is missing or the value is not in config, returns the "base" variant (VARIANT: "base").
+ */
+export declare function buildWizardMetadata(flags?: Record<string, string>): Record<string, string>;
 /**
  * Permission hook that allows only safe commands.
  * - Package manager install commands
@@ -76,11 +122,12 @@ export declare function initializeAgent(config: AgentConfig, options: WizardOpti
  *
  * @returns An object containing any error detected in the agent's output
  */
-export declare function runAgent(agentConfig: AgentRunConfig, prompt: string, options: WizardOptions, spinner: ReturnType<typeof clack.spinner>, config?: {
+export declare function runAgent(agentConfig: AgentRunConfig, prompt: string, options: WizardOptions, spinner: SpinnerHandle, config?: {
     estimatedDurationMinutes?: number;
     spinnerMessage?: string;
     successMessage?: string;
     errorMessage?: string;
+    additionalFeatureQueue?: readonly AdditionalFeature[];
 }, middleware?: {
     onMessage(message: any): void;
     finalize(resultMessage: any, totalDurationMs: number): any;