@rallycry/conveyor-agent 7.1.9 → 7.2.1

package/dist/{chunk-5CGROG7V.js → chunk-2JONMCGY.js}

@@ -598,12 +598,14 @@ var ModeController = class {
 // src/runner/lifecycle.ts
 var DEFAULT_LIFECYCLE_CONFIG = {
   idleTimeoutMs: 30 * 60 * 1e3,
-  heartbeatIntervalMs: 3e4
+  heartbeatIntervalMs: 3e4,
+  tokenRefreshIntervalMs: 45 * 60 * 1e3
 };
 var Lifecycle = class {
   config;
   callbacks;
   heartbeatTimer = null;
+  tokenRefreshTimer = null;
   idleTimer = null;
   idleCheckInterval = null;
   constructor(config, callbacks) {
@@ -623,6 +625,19 @@ var Lifecycle = class {
       this.heartbeatTimer = null;
     }
   }
+  // ── Token refresh ─────────────────────────────────────────────────
+  startTokenRefresh() {
+    this.stopTokenRefresh();
+    this.tokenRefreshTimer = setInterval(() => {
+      this.callbacks.onTokenRefresh();
+    }, this.config.tokenRefreshIntervalMs);
+  }
+  stopTokenRefresh() {
+    if (this.tokenRefreshTimer) {
+      clearInterval(this.tokenRefreshTimer);
+      this.tokenRefreshTimer = null;
+    }
+  }
   // ── Idle timer ─────────────────────────────────────────────────────
   startIdleTimer() {
     this.clearIdleTimers();
@@ -636,6 +651,7 @@ var Lifecycle = class {
   // ── Cleanup ────────────────────────────────────────────────────────
   destroy() {
     this.stopHeartbeat();
+    this.stopTokenRefresh();
     this.clearIdleTimers();
   }
   // ── Private ────────────────────────────────────────────────────────
@@ -3025,16 +3041,18 @@ function buildDiscoveryTools(connection) {
         title: z3.string().optional().describe("The new task title"),
         storyPointValue: z3.number().optional().describe(SP_DESCRIPTION2),
         tagIds: z3.array(z3.string()).optional().describe("Array of tag IDs to assign"),
-        githubPRUrl: z3.string().url().optional().describe("GitHub pull request URL to link to this task")
+        githubPRUrl: z3.string().url().optional().describe("GitHub pull request URL to link to this task"),
+        githubBranch: z3.string().optional().describe("Set the GitHub branch name for this task (e.g. 'conveyor/my-feature-abc123')")
       },
-      async ({ title, storyPointValue, tagIds, githubPRUrl }) => {
+      async ({ title, storyPointValue, tagIds, githubPRUrl, githubBranch }) => {
         try {
           await connection.call("updateTaskProperties", {
             sessionId: connection.sessionId,
             title,
             storyPointValue,
             tagIds,
-            githubPRUrl
+            githubPRUrl,
+            githubBranch
           });
           const updatedFields = [];
           if (title !== void 0) updatedFields.push(`title to "${title}"`);
@@ -3042,6 +3060,7 @@ function buildDiscoveryTools(connection) {
             updatedFields.push(`story points to ${storyPointValue}`);
           if (tagIds !== void 0) updatedFields.push(`tags (${tagIds.length} tag(s))`);
           if (githubPRUrl !== void 0) updatedFields.push(`PR link to "${githubPRUrl}"`);
+          if (githubBranch !== void 0) updatedFields.push(`branch to "${githubBranch}"`);
           return textResult(`Task properties updated: ${updatedFields.join(", ")}`);
         } catch (error) {
           return textResult(
@@ -4916,7 +4935,8 @@ async function handleExitPlanMode(host, input) {
     }
     if (host.agentMode === "discovery") {
       host.hasExitedPlanMode = true;
-      host.pendingModeRestart = true;
+      host.discoveryCompleted = true;
+      host.requestStop();
       void host.connection.triggerIdentification();
       host.connection.postChatMessage(
         "Plan complete. Running identification \u2014 icon and agent assignment will be set automatically."
@@ -5471,6 +5491,7 @@ var QueryBridge = class {
   activeQuery = null;
   pendingToolOutputs = [];
   _stopped = false;
+  _discoveryCompleted = false;
   _isParentTask = false;
   _wasRateLimited = false;
   _abortController = null;
@@ -5481,6 +5502,9 @@ var QueryBridge = class {
   get isStopped() {
     return this._stopped;
   }
+  get isDiscoveryCompleted() {
+    return this._discoveryCompleted;
+  }
   get isParentTask() {
     return this._isParentTask;
   }
@@ -5554,6 +5578,12 @@ var QueryBridge = class {
       set pendingModeRestart(val) {
         bridge.mode.pendingModeRestart = val;
       },
+      get discoveryCompleted() {
+        return bridge._discoveryCompleted;
+      },
+      set discoveryCompleted(val) {
+        bridge._discoveryCompleted = val;
+      },
       get wasRateLimited() {
         return bridge._wasRateLimited;
       },
@@ -5641,6 +5671,9 @@ var SessionRunner = class _SessionRunner {
   _state = "connecting";
   stopped = false;
   interrupted = false;
+  /** Defense-in-depth: set when the agent emits a "completed" event.
+   *  Prevents the core loop from processing any further messages. */
+  completedThisTurn = false;
   taskContext = null;
   fullContext = null;
   queryBridge = null;
@@ -5664,7 +5697,8 @@ var SessionRunner = class _SessionRunner {
           this.inputResolver = null;
           resolver(null);
         }
-      }
+      },
+      onTokenRefresh: () => void this.refreshGithubToken()
     });
   }
   get state() {
@@ -5689,6 +5723,7 @@ var SessionRunner = class _SessionRunner {
     this.connection.sendEvent({ type: "connected", sessionId: this.sessionId });
     this.wireConnectionCallbacks();
     this.lifecycle.startHeartbeat();
+    this.lifecycle.startTokenRefresh();
     const { pendingMessages: serverMessages } = await this.connection.call("connectAgent", {
       sessionId: this.sessionId
     });
@@ -5771,6 +5806,13 @@ var SessionRunner = class _SessionRunner {
   // ── Core loop ──────────────────────────────────────────────────────
   async coreLoop() {
     while (!this.stopped) {
+      if (this.completedThisTurn) {
+        process.stderr.write(
+          "[conveyor-agent] Completed \u2014 ignoring further messages (agent-side guard)\n"
+        );
+        this.stopped = true;
+        break;
+      }
       if (this._state === "idle") {
         this.lifecycle.startIdleTimer();
         const msg = await this.waitForMessage();
@@ -5790,6 +5832,13 @@ var SessionRunner = class _SessionRunner {
           userId: msg.userId
         });
         await this.executeQuery(msg.content);
+        if (this.queryBridge?.isDiscoveryCompleted) {
+          process.stderr.write(
+            "[conveyor-agent] Discovery completed via ExitPlanMode, shutting down\n"
+          );
+          this.stopped = true;
+          break;
+        }
         if (this.stopped) break;
         if (this.interrupted) {
           this.interrupted = false;
@@ -6016,7 +6065,12 @@ var SessionRunner = class _SessionRunner {
       runnerConfig,
       {
         onStatusChange: (status) => this.callbacks.onStatusChange(status),
-        onEvent: (event) => this.callbacks.onEvent(event)
+        onEvent: (event) => {
+          if (event.type === "completed") {
+            this.completedThisTurn = true;
+          }
+          return this.callbacks.onEvent(event);
+        }
       },
       this.config.workspaceDir
     );
@@ -6067,6 +6121,22 @@ var SessionRunner = class _SessionRunner {
       }
     });
   }
+  /** Proactively refresh the GitHub token before the 1-hour expiry. */
+  async refreshGithubToken() {
+    try {
+      const res = await this.connection.call("refreshGithubToken", {
+        sessionId: this.connection.sessionId
+      });
+      updateRemoteToken(this.config.workspaceDir, res.token);
+      process.env.GITHUB_TOKEN = res.token;
+      process.env.GH_TOKEN = res.token;
+      process.stderr.write("[conveyor-agent] Proactively refreshed GitHub token\n");
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      process.stderr.write(`[conveyor-agent] Warning: proactive token refresh failed: ${msg}
+`);
+    }
+  }
   /** Re-fetch task context to pick up a newly created branch and check it out. */
   async refreshBranchForBuilding() {
     try {
@@ -7145,7 +7215,7 @@ var ProjectRunner = class {
   async handleAuditTags(request) {
     this.connection.emitStatus("busy");
     try {
-      const { handleTagAudit } = await import("./tag-audit-handler-WPGSBNXO.js");
+      const { handleTagAudit } = await import("./tag-audit-handler-ZJYIOGCJ.js");
       await handleTagAudit(request, this.connection, this.projectDir);
     } catch (error) {
       const msg = error instanceof Error ? error.message : String(error);
@@ -7168,7 +7238,7 @@ var ProjectRunner = class {
   async handleAuditTasks(request) {
     this.connection.emitStatus("busy");
     try {
-      const { handleTaskAudit } = await import("./task-audit-handler-AOYSFO5B.js");
+      const { handleTaskAudit } = await import("./task-audit-handler-UL4HXCE4.js");
       await handleTaskAudit(request, this.connection, this.projectDir);
     } catch (error) {
       const msg = error instanceof Error ? error.message : String(error);
@@ -7201,6 +7271,11 @@ var ProjectRunner = class {
         } catch {
         }
       }
+      await this.connection.call("reportTaskAuditBatchComplete", {
+        projectId: this.connection.projectId,
+        requestId: request.requestId
+      }).catch(() => {
+      });
     } finally {
       this.connection.emitStatus("idle");
     }
@@ -7608,4 +7683,4 @@ export {
   loadForwardPorts,
   loadConveyorConfig
 };
-//# sourceMappingURL=chunk-5CGROG7V.js.map
+//# sourceMappingURL=chunk-2JONMCGY.js.map