@posthog/wizard 1.36.1 → 2.0.0

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

@@ -3,19 +3,61 @@
  * Shared agent interface for PostHog wizards
  * Uses Claude Agent SDK directly with PostHog LLM gateway
  */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentErrorType = exports.AgentSignals = void 0;
+exports.checkClaudeSettingsOverrides = checkClaudeSettingsOverrides;
+exports.backupAndFixClaudeSettings = backupAndFixClaudeSettings;
+exports.restoreClaudeSettings = restoreClaudeSettings;
+exports.createStopHook = createStopHook;
+exports.buildWizardMetadata = buildWizardMetadata;
 exports.wizardCanUseTool = wizardCanUseTool;
 exports.initializeAgent = initializeAgent;
 exports.runAgent = runAgent;
 const path_1 = __importDefault(require("path"));
-const clack_1 = __importDefault(require("../utils/clack"));
+const fs = __importStar(require("fs"));
+const ui_1 = require("../ui");
 const debug_1 = require("../utils/debug");
 const analytics_1 = require("../utils/analytics");
 const constants_1 = require("./constants");
+const wizard_session_1 = require("./wizard-session");
+const wizard_abort_1 = require("../utils/wizard-abort");
+const custom_headers_1 = require("../utils/custom-headers");
 const urls_1 = require("../utils/urls");
 const safe_tools_1 = require("./safe-tools");
 const wizard_tools_1 = require("./wizard-tools");
@@ -64,6 +106,145 @@ var AgentErrorType;
     /** Generic API error */
     AgentErrorType["API_ERROR"] = "WIZARD_API_ERROR";
 })(AgentErrorType || (exports.AgentErrorType = AgentErrorType = {}));
+const BLOCKING_ENV_KEYS = ['ANTHROPIC_BASE_URL', 'ANTHROPIC_AUTH_TOKEN'];
+/**
+ * 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.
+ */
+function checkClaudeSettingsOverrides(workingDirectory) {
+    const candidates = [
+        path_1.default.join(workingDirectory, '.claude', 'settings.json'),
+        path_1.default.join(workingDirectory, '.claude', 'settings'),
+    ];
+    for (const filePath of candidates) {
+        try {
+            const raw = fs.readFileSync(filePath, 'utf-8');
+            const parsed = JSON.parse(raw);
+            const envBlock = parsed?.env;
+            if (envBlock && typeof envBlock === 'object') {
+                return BLOCKING_ENV_KEYS.filter((key) => key in envBlock);
+            }
+        }
+        catch {
+            // File doesn't exist or isn't valid JSON — skip
+        }
+    }
+    return [];
+}
+/**
+ * Copy .claude/settings.json to .wizard-backup (overwriting if it exists),
+ * then remove the original so the SDK doesn't load the blocking overrides.
+ */
+function backupAndFixClaudeSettings(workingDirectory) {
+    for (const name of ['settings.json', 'settings']) {
+        const filePath = path_1.default.join(workingDirectory, '.claude', name);
+        const backupPath = `${filePath}.wizard-backup`;
+        analytics_1.analytics.wizardCapture('backedup-claude-settings');
+        try {
+            fs.copyFileSync(filePath, backupPath);
+            fs.unlinkSync(filePath);
+            (0, wizard_abort_1.registerCleanup)(() => {
+                try {
+                    restoreClaudeSettings(workingDirectory);
+                }
+                catch (error) {
+                    analytics_1.analytics.captureException(error);
+                }
+            });
+            return true;
+        }
+        catch {
+            // File doesn't exist — try next candidate
+        }
+    }
+    return false;
+}
+/**
+ * Restore .claude/settings.json from .wizard-backup.
+ * Copies (not moves) so the backup is preserved.
+ */
+function restoreClaudeSettings(workingDirectory) {
+    for (const name of ['settings.json', 'settings']) {
+        const backup = path_1.default.join(workingDirectory, '.claude', `${name}.wizard-backup`);
+        try {
+            fs.copyFileSync(backup, path_1.default.join(workingDirectory, '.claude', name));
+            analytics_1.analytics.wizardCapture('restored-claude-settings');
+            return;
+        }
+        catch (error) {
+            analytics_1.analytics.captureException(error);
+        }
+    }
+}
+/**
+ * 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 {}
+ */
+function createStopHook(featureQueue) {
+    let featureIndex = 0;
+    let remarkRequested = false;
+    return (input) => {
+        (0, debug_1.logToFile)('Stop hook triggered', {
+            stop_hook_active: input.stop_hook_active,
+            featureIndex,
+            remarkRequested,
+            queueLength: featureQueue.length,
+        });
+        // Phase 1: drain feature queue
+        if (featureIndex < featureQueue.length) {
+            const feature = featureQueue[featureIndex++];
+            const prompt = wizard_session_1.ADDITIONAL_FEATURE_PROMPTS[feature];
+            (0, debug_1.logToFile)(`Stop hook: injecting feature prompt for ${feature}`);
+            return { decision: 'block', reason: prompt };
+        }
+        // Phase 2: collect remark (once)
+        if (!remarkRequested) {
+            remarkRequested = true;
+            (0, debug_1.logToFile)('Stop hook: requesting reflection');
+            return {
+                decision: 'block',
+                reason: `Before concluding, provide a brief remark about what information or guidance would have been useful to have in the integration prompt or documentation for this run. Specifically cite anything that would have prevented tool failures, erroneous edits, or other wasted turns. Format your response exactly as: ${exports.AgentSignals.WIZARD_REMARK} Your remark here`,
+            };
+        }
+        // Phase 3: allow stop
+        (0, debug_1.logToFile)('Stop hook: allowing stop');
+        return {};
+    };
+}
+/**
+ * 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").
+ */
+function buildWizardMetadata(flags = {}) {
+    const variantKey = flags[constants_1.WIZARD_VARIANT_FLAG_KEY];
+    const variant = (variantKey && constants_1.WIZARD_VARIANTS[variantKey]) ?? constants_1.WIZARD_VARIANTS['base'];
+    return { ...variant };
+}
+/**
+ * Build env for the SDK subprocess: process.env plus ANTHROPIC_CUSTOM_HEADERS from wizard metadata/flags.
+ */
+function buildAgentEnv(wizardMetadata, wizardFlags) {
+    const headers = (0, custom_headers_1.createCustomHeaders)();
+    for (const [key, value] of Object.entries(wizardMetadata)) {
+        headers.add(key.startsWith(constants_1.POSTHOG_PROPERTY_HEADER_PREFIX)
+            ? key
+            : `${constants_1.POSTHOG_PROPERTY_HEADER_PREFIX}${key}`, value);
+    }
+    for (const [flagKey, variant] of Object.entries(wizardFlags)) {
+        if (!flagKey.toLowerCase().startsWith('wizard'))
+            continue;
+        headers.addFlag(flagKey, variant);
+    }
+    const encoded = headers.encode();
+    (0, debug_1.logToFile)('ANTHROPIC_CUSTOM_HEADERS', encoded);
+    return encoded;
+}
 /**
  * Package managers that can be used to run commands.
  */
@@ -197,8 +378,7 @@ function wizardCanUseTool(toolName, input) {
     if (DANGEROUS_OPERATORS.test(command)) {
         (0, debug_1.logToFile)(`Denying bash command with dangerous operators: ${command}`);
         (0, debug_1.debug)(`Denying bash command with dangerous operators: ${command}`);
-        analytics_1.analytics.capture(constants_1.WIZARD_INTERACTION_EVENT_NAME, {
-            action: 'bash command denied',
+        analytics_1.analytics.wizardCapture('bash denied', {
             reason: 'dangerous operators',
             command,
         });
@@ -217,8 +397,7 @@ function wizardCanUseTool(toolName, input) {
         if (/[|&]/.test(baseCommand)) {
             (0, debug_1.logToFile)(`Denying bash command with multiple pipes: ${command}`);
             (0, debug_1.debug)(`Denying bash command with multiple pipes: ${command}`);
-            analytics_1.analytics.capture(constants_1.WIZARD_INTERACTION_EVENT_NAME, {
-                action: 'bash command denied',
+            analytics_1.analytics.wizardCapture('bash denied', {
                 reason: 'multiple pipes',
                 command,
             });
@@ -237,8 +416,7 @@ function wizardCanUseTool(toolName, input) {
     if (/[|&]/.test(normalized)) {
         (0, debug_1.logToFile)(`Denying bash command with pipe/&: ${command}`);
         (0, debug_1.debug)(`Denying bash command with pipe/&: ${command}`);
-        analytics_1.analytics.capture(constants_1.WIZARD_INTERACTION_EVENT_NAME, {
-            action: 'bash command denied',
+        analytics_1.analytics.wizardCapture('bash denied', {
             reason: 'disallowed pipe',
             command,
         });
@@ -255,8 +433,7 @@ function wizardCanUseTool(toolName, input) {
     }
     (0, debug_1.logToFile)(`Denying bash command: ${command}`);
     (0, debug_1.debug)(`Denying bash command: ${command}`);
-    analytics_1.analytics.capture(constants_1.WIZARD_INTERACTION_EVENT_NAME, {
-        action: 'bash command denied',
+    analytics_1.analytics.wizardCapture('bash denied', {
         reason: 'not in allowlist',
         command,
     });
@@ -273,7 +450,7 @@ async function initializeAgent(config, options) {
     (0, debug_1.initLogFile)();
     (0, debug_1.logToFile)('Agent initialization starting');
     (0, debug_1.logToFile)('Install directory:', options.installDir);
-    clack_1.default.log.step('Initializing Claude agent...');
+    (0, ui_1.getUI)().log.step('Initializing Claude agent...');
     try {
         // Configure LLM gateway environment variables (inherited by SDK subprocess)
         const gatewayUrl = (0, urls_1.getLlmGatewayUrlFromHost)(config.posthogApiHost);
@@ -306,6 +483,8 @@ async function initializeAgent(config, options) {
             workingDirectory: config.workingDirectory,
             mcpServers,
             model: 'anthropic/claude-sonnet-4-6',
+            wizardFlags: config.wizardFlags,
+            wizardMetadata: config.wizardMetadata,
         };
         (0, debug_1.logToFile)('Agent config:', {
             workingDirectory: agentRunConfig.workingDirectory,
@@ -321,12 +500,12 @@ async function initializeAgent(config, options) {
                 apiKeyPresent: !!config.posthogApiKey,
             });
         }
-        clack_1.default.log.step(`Verbose logs: ${(0, debug_1.getLogFilePath)()}`);
-        clack_1.default.log.success("Agent initialized. Let's get cooking!");
+        (0, ui_1.getUI)().log.step(`Verbose logs: ${(0, debug_1.getLogFilePath)()}`);
+        (0, ui_1.getUI)().log.success("Agent initialized. Let's get cooking!");
         return agentRunConfig;
     }
     catch (error) {
-        clack_1.default.log.error(`Failed to initialize agent: ${error.message}`);
+        (0, ui_1.getUI)().log.error(`Failed to initialize agent: ${error.message}`);
         (0, debug_1.logToFile)('Agent initialization error:', error);
         (0, debug_1.debug)('Agent initialization error:', error);
         throw error;
@@ -339,9 +518,8 @@ async function initializeAgent(config, options) {
  * @returns An object containing any error detected in the agent's output
  */
 async function runAgent(agentConfig, prompt, options, spinner, config, middleware) {
-    const { estimatedDurationMinutes = 8, spinnerMessage = 'Customizing your PostHog setup...', successMessage = 'PostHog integration complete', errorMessage = 'Integration failed', } = config ?? {};
+    const { spinnerMessage = 'Customizing your PostHog setup...', successMessage = 'PostHog integration complete', errorMessage = 'Integration failed', } = config ?? {};
     const { query } = await getSDKModule();
-    clack_1.default.log.step(`This whole process should take about ${estimatedDurationMinutes} minutes including error checking and fixes.\n\nGrab some coffee!`);
     spinner.start(spinnerMessage);
     const cliPath = getClaudeCodeExecutablePath();
     (0, debug_1.logToFile)('Starting agent run');
@@ -391,8 +569,7 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
                 analytics_1.analytics.capture(constants_1.WIZARD_REMARK_EVENT_NAME, { remark });
             }
         }
-        analytics_1.analytics.capture(constants_1.WIZARD_INTERACTION_EVENT_NAME, {
-            action: 'agent integration completed',
+        analytics_1.analytics.wizardCapture('agent completed', {
             duration_ms: durationMs,
             duration_seconds: durationSeconds,
         });
@@ -405,6 +582,9 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
         spinner.stop(successMessage);
         return {};
     };
+    // Event plan file watcher — cleaned up in finally block
+    let eventPlanWatcher;
+    let eventPlanInterval;
     try {
         // Tools needed for the wizard:
         // - File operations: Read, Write, Edit
@@ -446,6 +626,7 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
                     ...process.env,
                     // Prevent user's Anthropic API key from overriding the wizard's OAuth token
                     ANTHROPIC_API_KEY: undefined,
+                    ANTHROPIC_CUSTOM_HEADERS: buildAgentEnv(agentConfig.wizardMetadata ?? {}, agentConfig.wizardFlags ?? {}),
                 },
                 canUseTool: (toolName, input) => {
                     (0, debug_1.logToFile)('canUseTool called:', { toolName, input });
@@ -461,33 +642,53 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
                         (0, debug_1.debug)('CLI stderr:', data);
                     }
                 },
-                // Stop hook to have the agent reflect on its run
+                // Stop hook: drain additional feature queue, then collect remark, then allow stop
                 hooks: {
                     Stop: [
                         {
-                            hooks: [
-                                (input) => {
-                                    (0, debug_1.logToFile)('Stop hook triggered', {
-                                        stop_hook_active: input.stop_hook_active,
-                                    });
-                                    // Only ask for reflection on first stop (not after reflection is provided)
-                                    if (input.stop_hook_active) {
-                                        (0, debug_1.logToFile)('Stop hook: allowing stop (already reflected)');
-                                        return {}; // Allow stopping
-                                    }
-                                    (0, debug_1.logToFile)('Stop hook: requesting reflection');
-                                    return {
-                                        decision: 'block',
-                                        reason: `Before concluding, provide a brief remark about what information or guidance would have been useful to have in the integration prompt or documentation for this run. Specifically cite anything that would have prevented tool failures, erroneous edits, or other wasted turns. Format your response exactly as: ${exports.AgentSignals.WIZARD_REMARK} Your remark here`,
-                                    };
-                                },
-                            ],
+                            hooks: [createStopHook(config?.additionalFeatureQueue ?? [])],
                             timeout: 30,
                         },
                     ],
                 },
             },
         });
+        // Watch for .posthog-events.json and feed into the store
+        const eventPlanPath = path_1.default.join(agentConfig.workingDirectory, '.posthog-events.json');
+        const readEventPlan = () => {
+            try {
+                const content = fs.readFileSync(eventPlanPath, 'utf-8');
+                const parsed = JSON.parse(content);
+                if (Array.isArray(parsed)) {
+                    (0, ui_1.getUI)().setEventPlan(parsed.map((e) => ({
+                        name: (e.name ?? e.event ?? ''),
+                        description: (e.description ?? ''),
+                    })));
+                }
+            }
+            catch {
+                // File doesn't exist or isn't valid JSON yet
+            }
+        };
+        try {
+            eventPlanWatcher = fs.watch(eventPlanPath, () => readEventPlan());
+            readEventPlan();
+        }
+        catch {
+            // File doesn't exist yet — poll until it appears
+            eventPlanInterval = setInterval(() => {
+                try {
+                    fs.accessSync(eventPlanPath);
+                    readEventPlan();
+                    clearInterval(eventPlanInterval);
+                    eventPlanInterval = undefined;
+                    eventPlanWatcher = fs.watch(eventPlanPath, () => readEventPlan());
+                }
+                catch {
+                    // Still waiting
+                }
+            }, 1000);
+        }
         // Process the async generator
         for await (const message of response) {
             // Pass receivedSuccessResult so handleSDKMessage can suppress user-facing error
@@ -569,11 +770,16 @@ async function runAgent(agentConfig, prompt, options, spinner, config, middlewar
         }
         // No API error found, re-throw the original exception
         spinner.stop(errorMessage);
-        clack_1.default.log.error(`Error: ${error.message}`);
+        (0, ui_1.getUI)().log.error(`Error: ${error.message}`);
         (0, debug_1.logToFile)('Agent run failed:', error);
         (0, debug_1.debug)('Full error:', error);
         throw error;
     }
+    finally {
+        eventPlanWatcher?.close();
+        if (eventPlanInterval)
+            clearInterval(eventPlanInterval);
+    }
 }
 /**
  * Handle SDK messages and provide user feedback
@@ -599,10 +805,18 @@ function handleSDKMessage(message, options, spinner, collectedText, receivedSucc
                         const statusRegex = new RegExp(`^.*${exports.AgentSignals.STATUS.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}\\s*(.+?)$`, 'm');
                         const statusMatch = block.text.match(statusRegex);
                         if (statusMatch) {
-                            spinner.stop(statusMatch[1].trim());
-                            spinner.start('Integrating PostHog...');
+                            const statusText = statusMatch[1].trim();
+                            (0, ui_1.getUI)().pushStatus(statusText);
+                            spinner.message(statusText);
                         }
                     }
+                    // Intercept TodoWrite tool_use blocks for task progression
+                    if (block.type === 'tool_use' &&
+                        block.name === 'TodoWrite' &&
+                        block.input?.todos &&
+                        Array.isArray(block.input.todos)) {
+                        (0, ui_1.getUI)().syncTodos(block.input.todos);
+                    }
                 }
             }
             break;
@@ -619,7 +833,7 @@ function handleSDKMessage(message, options, spinner, collectedText, receivedSucc
                 // mode race conditions). Full message already logged above via JSON dump.
                 if (message.errors && !receivedSuccessResult) {
                     for (const err of message.errors) {
-                        clack_1.default.log.error(`Error: ${err}`);
+                        (0, ui_1.getUI)().log.error(`Error: ${err}`);
                         (0, debug_1.logToFile)('ERROR:', err);
                     }
                 }
@@ -636,7 +850,7 @@ function handleSDKMessage(message, options, spinner, collectedText, receivedSucc
                 // Full message already logged above via JSON dump.
                 if (message.errors && !receivedSuccessResult) {
                     for (const err of message.errors) {
-                        clack_1.default.log.error(`Error: ${err}`);
+                        (0, ui_1.getUI)().log.error(`Error: ${err}`);
                         (0, debug_1.logToFile)('ERROR:', err);
                     }
                 }