@posthog/agent 2.3.261 → 2.3.263

package/dist/server/bin.cjs

@@ -5683,7 +5683,7 @@ var import_hono = require("hono");
 // package.json
 var package_default = {
   name: "@posthog/agent",
-  version: "2.3.261",
+  version: "2.3.263",
   repository: "https://github.com/PostHog/code",
   description: "TypeScript agent framework wrapping Claude Agent SDK with Git-based task execution for PostHog",
   exports: {
@@ -5844,7 +5844,9 @@ var POSTHOG_NOTIFICATIONS = {
   /** Marks a boundary for log compaction */
   COMPACT_BOUNDARY: "_posthog/compact_boundary",
   /** Token usage update for a session turn */
-  USAGE_UPDATE: "_posthog/usage_update"
+  USAGE_UPDATE: "_posthog/usage_update",
+  /** Response to a relayed permission request (plan approval, question) */
+  PERMISSION_RESPONSE: "_posthog/permission_response"
 };
 function isNotification(method, notification) {
   if (!method) return false;
@@ -8281,6 +8283,11 @@ async function canUseTool(context) {
   if (planFileResult) {
     return planFileResult;
   }
+  if (session.permissionMode === "plan") {
+    const message = `This tool is not available in plan mode. Write your plan to a file in ${getClaudePlansDir()} and call ExitPlanMode when ready.`;
+    await emitToolDenial(context, message);
+    return { behavior: "deny", message, interrupt: false };
+  }
   return handleDefaultPermissionFlow(context);
 }
 
@@ -12219,13 +12226,27 @@ var userMessageParamsSchema = import_v4.z.object({
     import_v4.z.array(import_v4.z.record(import_v4.z.string(), import_v4.z.unknown())).min(1, "Content is required")
   ])
 });
+var permissionResponseParamsSchema = import_v4.z.object({
+  requestId: import_v4.z.string().min(1, "requestId is required"),
+  optionId: import_v4.z.string().min(1, "optionId is required"),
+  customInput: import_v4.z.string().optional(),
+  answers: import_v4.z.record(import_v4.z.string(), import_v4.z.string()).optional()
+});
+var setConfigOptionParamsSchema = import_v4.z.object({
+  configId: import_v4.z.string().min(1, "configId is required"),
+  value: import_v4.z.string().min(1, "value is required")
+});
 var commandParamsSchemas = {
   user_message: userMessageParamsSchema,
   "posthog/user_message": userMessageParamsSchema,
   cancel: import_v4.z.object({}).optional(),
   "posthog/cancel": import_v4.z.object({}).optional(),
   close: import_v4.z.object({}).optional(),
-  "posthog/close": import_v4.z.object({}).optional()
+  "posthog/close": import_v4.z.object({}).optional(),
+  permission_response: permissionResponseParamsSchema,
+  "posthog/permission_response": permissionResponseParamsSchema,
+  set_config_option: setConfigOptionParamsSchema,
+  "posthog/set_config_option": setConfigOptionParamsSchema
 };
 function validateCommandParams(method, params) {
   const schema = commandParamsSchemas[method] ?? commandParamsSchemas[method.replace("posthog/", "")];
@@ -12342,6 +12363,7 @@ var AgentServer = class _AgentServer {
   // causing a second session to be created and duplicate Slack messages to be sent.
   initializationPromise = null;
   pendingEvents = [];
+  pendingPermissions = /* @__PURE__ */ new Map();
   detachSseController(controller) {
     if (this.session?.sseController === controller) {
       this.session.sseController = null;
@@ -12379,6 +12401,9 @@ var AgentServer = class _AgentServer {
   getEffectiveMode(payload) {
     return payload.mode ?? this.config.mode;
   }
+  getSessionPermissionMode() {
+    return this.session?.permissionMode ?? "default";
+  }
   createApp() {
     const app = new import_hono.Hono();
     app.get("/health", (c) => {
@@ -12423,6 +12448,7 @@ var AgentServer = class _AgentServer {
             await this.initializeSession(payload, sseController);
           } else {
             this.session.sseController = sseController;
+            this.session.hasDesktopConnected = true;
             this.replayPendingEvents();
           }
           this.sendSseEvent(sseController, {
@@ -12662,6 +12688,43 @@ var AgentServer = class _AgentServer {
         await this.cleanupSession();
         return { closed: true };
       }
+      case "posthog/set_config_option":
+      case "set_config_option": {
+        const configId = params.configId;
+        const value = params.value;
+        this.logger.info("Set config option requested", { configId, value });
+        const result = await this.session.clientConnection.setSessionConfigOption({
+          sessionId: this.session.acpSessionId,
+          configId,
+          value
+        });
+        return {
+          configOptions: result.configOptions
+        };
+      }
+      case POSTHOG_NOTIFICATIONS.PERMISSION_RESPONSE:
+      case "permission_response": {
+        const requestId = params.requestId;
+        const optionId = params.optionId;
+        const customInput = params.customInput;
+        const answers = params.answers;
+        this.logger.info("Permission response received", {
+          requestId,
+          optionId
+        });
+        const resolved = this.resolvePermission(
+          requestId,
+          optionId,
+          customInput,
+          answers
+        );
+        if (!resolved) {
+          throw new Error(
+            `No pending permission request found for id: ${requestId}`
+          );
+        }
+        return { resolved: true };
+      }
       default:
         throw new Error(`Unknown method: ${method}`);
     }
@@ -12780,6 +12843,8 @@ var AgentServer = class _AgentServer {
     if (prUrl) {
       this.detectedPrUrl = prUrl;
     }
+    const runState = preTaskRun?.state;
+    const initialPermissionMode = typeof runState?.initial_permission_mode === "string" ? runState.initial_permission_mode : "bypassPermissions";
     const sessionResponse = await clientConnection.newSession({
       cwd: this.config.repositoryPath ?? "/tmp/workspace",
       mcpServers: this.config.mcpServers ?? [],
@@ -12789,6 +12854,7 @@ var AgentServer = class _AgentServer {
         systemPrompt: this.buildSessionSystemPrompt(prUrl),
         allowedDomains: this.config.allowedDomains,
         jsonSchema: preTask?.json_schema ?? null,
+        permissionMode: initialPermissionMode,
         ...this.config.claudeCode?.plugins?.length && {
           claudeCode: {
             options: {
@@ -12811,7 +12877,9 @@ var AgentServer = class _AgentServer {
       treeTracker,
       sseController,
       deviceInfo,
-      logWriter
+      logWriter,
+      permissionMode: initialPermissionMode,
+      hasDesktopConnected: sseController !== null
     };
     this.logger = new Logger({
       debug: true,
@@ -12825,6 +12893,7 @@ var AgentServer = class _AgentServer {
     this.logger.info(
       `Agent version: ${this.config.version ?? package_default.version}`
     );
+    this.logger.info(`Initial permission mode: ${initialPermissionMode}`);
     this.posthogAPI.updateTaskRun(payload.task_id, payload.run_id, {
       status: "in_progress"
     }).catch(
@@ -13315,14 +13384,16 @@ ${attributionInstructions}
         this.logger.debug("Permission request", {
           mode,
           interactionOrigin,
+          kind: params.toolCall?.kind,
           options: params.options
         });
         const allowOption = params.options.find(
           (o) => o.kind === "allow_once" || o.kind === "allow_always"
         );
         const selectedOptionId = allowOption?.optionId ?? params.options[0].optionId;
+        const codeToolKind = params.toolCall?._meta?.codeToolKind;
+        const isPlanApproval = params.toolCall?.kind === "switch_mode";
         if (interactionOrigin === "slack") {
-          const codeToolKind = params.toolCall?._meta?.codeToolKind;
           if (codeToolKind === "question") {
             return this.buildSlackQuestionRelayResponse(
               payload,
@@ -13330,6 +13401,19 @@ ${attributionInstructions}
             );
           }
         }
+        {
+          const isQuestion = codeToolKind === "question";
+          const sessionPermissionMode = this.getSessionPermissionMode();
+          const needsRelay = isQuestion || isPlanApproval || sessionPermissionMode === "default";
+          if (needsRelay && this.session?.hasDesktopConnected) {
+            this.logger.info("Relaying permission to connected client", {
+              kind: params.toolCall?.kind,
+              isQuestion,
+              sessionPermissionMode
+            });
+            return this.relayPermissionToClient(params);
+          }
+        }
         if (this.shouldBlockPublishPermission(params)) {
           return {
             outcome: { outcome: "cancelled" },
@@ -13349,6 +13433,12 @@ ${attributionInstructions}
         this.logger.debug("Extension notification", { method, params });
       },
       sessionUpdate: async (params) => {
+        if (params.update?.sessionUpdate === "current_mode_update" && typeof params.update?.currentModeId === "string" && this.session) {
+          this.session.permissionMode = params.update.currentModeId;
+          this.logger.info("Permission mode updated", {
+            mode: params.update.currentModeId
+          });
+        }
         if (params.update?.sessionUpdate === "tool_call_update") {
           const meta = params.update?._meta?.claudeCode;
           const toolName = meta?.toolName;
@@ -13527,6 +13617,13 @@ ${attributionInstructions}
     } catch (error) {
       this.logger.error("Failed to flush session logs", error);
     }
+    for (const [, pending] of this.pendingPermissions) {
+      pending.resolve({
+        outcome: { outcome: "selected", optionId: "reject" },
+        _meta: { customInput: "Session is shutting down." }
+      });
+    }
+    this.pendingPermissions.clear();
     try {
       await this.session.acpConnection.cleanup();
     } catch (error) {
@@ -13604,6 +13701,39 @@ ${attributionInstructions}
       this.detachSseController(controller);
     }
   }
+  /**
+   * Relay a permission request (e.g., plan approval) to the connected desktop
+   * app via SSE and wait for a response via the `/command` endpoint.
+   *
+   * The promise waits indefinitely — if SSE is disconnected, the event is
+   * buffered by broadcastEvent and replayed when the client reconnects. Session
+   * cleanup force-resolves all pending permissions, so there is no leak.
+   */
+  relayPermissionToClient(params) {
+    const requestId = crypto.randomUUID();
+    this.broadcastEvent({
+      type: "permission_request",
+      requestId,
+      options: params.options,
+      toolCall: params.toolCall
+    });
+    return new Promise((resolve4) => {
+      this.pendingPermissions.set(requestId, { resolve: resolve4 });
+    });
+  }
+  resolvePermission(requestId, optionId, customInput, answers) {
+    const pending = this.pendingPermissions.get(requestId);
+    if (!pending) return false;
+    this.pendingPermissions.delete(requestId);
+    const meta = {};
+    if (customInput) meta.customInput = customInput;
+    if (answers) meta.answers = answers;
+    pending.resolve({
+      outcome: { outcome: "selected", optionId },
+      ...Object.keys(meta).length > 0 ? { _meta: meta } : {}
+    });
+    return true;
+  }
 };
 
 // src/server/bin.ts