claude-code/tasks/InProcessTeammateTask/InProcessTeammateTask.tsx

/**
 * InProcessTeammateTask - Manages in-process teammate lifecycle
 *
 * This component implements the Task interface for in-process teammates.
 * Unlike LocalAgentTask (background agents), in-process teammates:
 * 1. Run in the same Node.js process using AsyncLocalStorage for isolation
 * 2. Have team-aware identity (agentName@teamName)
 * 3. Support plan mode approval flow
 * 4. Can be idle (waiting for work) or active (processing)
 */

import { isTerminalTaskStatus, type SetAppState, type Task, type TaskStateBase } from '../../Task.js';
import type { Message } from '../../types/message.js';
import { logForDebugging } from '../../utils/debug.js';
import { createUserMessage } from '../../utils/messages.js';
import { killInProcessTeammate } from '../../utils/swarm/spawnInProcess.js';
import { updateTaskState } from '../../utils/task/framework.js';
import type { InProcessTeammateTaskState } from './types.js';
import { appendCappedMessage, isInProcessTeammateTask } from './types.js';

/**
 * InProcessTeammateTask - Handles in-process teammate execution.
 */
export const InProcessTeammateTask: Task = {
  name: 'InProcessTeammateTask',
  type: 'in_process_teammate',
  async kill(taskId, setAppState) {
    killInProcessTeammate(taskId, setAppState);
  }
};

/**
 * Request shutdown for a teammate.
 */
export function requestTeammateShutdown(taskId: string, setAppState: SetAppState): void {
  updateTaskState<InProcessTeammateTaskState>(taskId, setAppState, task => {
    if (task.status !== 'running' || task.shutdownRequested) {
      return task;
    }
    return {
      ...task,
      shutdownRequested: true
    };
  });
}

/**
 * Append a message to a teammate's conversation history.
 * Used for zoomed view to show the teammate's conversation.
 */
export function appendTeammateMessage(taskId: string, message: Message, setAppState: SetAppState): void {
  updateTaskState<InProcessTeammateTaskState>(taskId, setAppState, task => {
    if (task.status !== 'running') {
      return task;
    }
    return {
      ...task,
      messages: appendCappedMessage(task.messages, message)
    };
  });
}

/**
 * Inject a user message to a teammate's pending queue.
 * Used when viewing a teammate's transcript to send typed messages to them.
 * Also adds the message to task.messages so it appears immediately in the transcript.
 */
export function injectUserMessageToTeammate(taskId: string, message: string, setAppState: SetAppState): void {
  updateTaskState<InProcessTeammateTaskState>(taskId, setAppState, task => {
    // Allow message injection when teammate is running or idle (waiting for input)
    // Only reject if teammate is in a terminal state
    if (isTerminalTaskStatus(task.status)) {
      logForDebugging(`Dropping message for teammate task ${taskId}: task status is "${task.status}"`);
      return task;
    }
    return {
      ...task,
      pendingUserMessages: [...task.pendingUserMessages, message],
      messages: appendCappedMessage(task.messages, createUserMessage({
        content: message
      }))
    };
  });
}

/**
 * Get teammate task by agent ID from AppState.
 * Prefers running tasks over killed/completed ones in case multiple tasks
 * with the same agentId exist.
 * Returns undefined if not found.
 */
export function findTeammateTaskByAgentId(agentId: string, tasks: Record<string, TaskStateBase>): InProcessTeammateTaskState | undefined {
  let fallback: InProcessTeammateTaskState | undefined;
  for (const task of Object.values(tasks)) {
    if (isInProcessTeammateTask(task) && task.identity.agentId === agentId) {
      // Prefer running tasks in case old killed tasks still exist in AppState
      // alongside new running ones with the same agentId
      if (task.status === 'running') {
        return task;
      }
      // Keep first match as fallback in case no running task exists
      if (!fallback) {
        fallback = task;
      }
    }
  }
  return fallback;
}

/**
 * Get all in-process teammate tasks from AppState.
 */
export function getAllInProcessTeammateTasks(tasks: Record<string, TaskStateBase>): InProcessTeammateTaskState[] {
  return Object.values(tasks).filter(isInProcessTeammateTask);
}

/**
 * Get running in-process teammates sorted alphabetically by agentName.
 * Shared between TeammateSpinnerTree display, PromptInput footer selector,
 * and useBackgroundTaskNavigation — selectedIPAgentIndex maps into this
 * array, so all three must agree on sort order.
 */
export function getRunningTeammatesSorted(tasks: Record<string, TaskStateBase>): InProcessTeammateTaskState[] {
  return getAllInProcessTeammateTasks(tasks).filter(t => t.status === 'running').sort((a, b) => a.identity.agentName.localeCompare(b.identity.agentName));
}
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJpc1Rlcm1pbmFsVGFza1N0YXR1cyIsIlNldEFwcFN0YXRlIiwiVGFzayIsIlRhc2tTdGF0ZUJhc2UiLCJNZXNzYWdlIiwibG9nRm9yRGVidWdnaW5nIiwiY3JlYXRlVXNlck1lc3NhZ2UiLCJraWxsSW5Qcm9jZXNzVGVhbW1hdGUiLCJ1cGRhdGVUYXNrU3RhdGUiLCJJblByb2Nlc3NUZWFtbWF0ZVRhc2tTdGF0ZSIsImFwcGVuZENhcHBlZE1lc3NhZ2UiLCJpc0luUHJvY2Vzc1RlYW1tYXRlVGFzayIsIkluUHJvY2Vzc1RlYW1tYXRlVGFzayIsIm5hbWUiLCJ0eXBlIiwia2lsbCIsInRhc2tJZCIsInNldEFwcFN0YXRlIiwicmVxdWVzdFRlYW1tYXRlU2h1dGRvd24iLCJ0YXNrIiwic3RhdHVzIiwic2h1dGRvd25SZXF1ZXN0ZWQiLCJhcHBlbmRUZWFtbWF0ZU1lc3NhZ2UiLCJtZXNzYWdlIiwibWVzc2FnZXMiLCJpbmplY3RVc2VyTWVzc2FnZVRvVGVhbW1hdGUiLCJwZW5kaW5nVXNlck1lc3NhZ2VzIiwiY29udGVudCIsImZpbmRUZWFtbWF0ZVRhc2tCeUFnZW50SWQiLCJhZ2VudElkIiwidGFza3MiLCJSZWNvcmQiLCJmYWxsYmFjayIsIk9iamVjdCIsInZhbHVlcyIsImlkZW50aXR5IiwiZ2V0QWxsSW5Qcm9jZXNzVGVhbW1hdGVUYXNrcyIsImZpbHRlciIsImdldFJ1bm5pbmdUZWFtbWF0ZXNTb3J0ZWQiLCJ0Iiwic29ydCIsImEiLCJiIiwiYWdlbnROYW1lIiwibG9jYWxlQ29tcGFyZSJdLCJzb3VyY2VzIjpbIkluUHJvY2Vzc1RlYW1tYXRlVGFzay50c3giXSwic291cmNlc0NvbnRlbnQiOlsiLyoqXG4gKiBJblByb2Nlc3NUZWFtbWF0ZVRhc2sgLSBNYW5hZ2VzIGluLXByb2Nlc3MgdGVhbW1hdGUgbGlmZWN5Y2xlXG4gKlxuICogVGhpcyBjb21wb25lbnQgaW1wbGVtZW50cyB0aGUgVGFzayBpbnRlcmZhY2UgZm9yIGluLXByb2Nlc3MgdGVhbW1hdGVzLlxuICogVW5saWtlIExvY2FsQWdlbnRUYXNrIChiYWNrZ3JvdW5kIGFnZW50cyksIGluLXByb2Nlc3MgdGVhbW1hdGVzOlxuICogMS4gUnVuIGluIHRoZSBzYW1lIE5vZGUuanMgcHJvY2VzcyB1c2luZyBBc3luY0xvY2FsU3RvcmFnZSBmb3IgaXNvbGF0aW9uXG4gKiAyLiBIYXZlIHRlYW0tYXdhcmUgaWRlbnRpdHkgKGFnZW50TmFtZUB0ZWFtTmFtZSlcbiAqIDMuIFN1cHBvcnQgcGxhbiBtb2RlIGFwcHJvdmFsIGZsb3dcbiAqIDQuIENhbiBiZSBpZGxlICh3YWl0aW5nIGZvciB3b3JrKSBvciBhY3RpdmUgKHByb2Nlc3NpbmcpXG4gKi9cblxuaW1wb3J0IHtcbiAgaXNUZXJtaW5hbFRhc2tTdGF0dXMsXG4gIHR5cGUgU2V0QXBwU3RhdGUsXG4gIHR5cGUgVGFzayxcbiAgdHlwZSBUYXNrU3RhdGVCYXNlLFxufSBmcm9tICcuLi8uLi9UYXNrLmpzJ1xuaW1wb3J0IHR5cGUgeyBNZXNzYWdlIH0gZnJvbSAnLi4vLi4vdHlwZXMvbWVzc2FnZS5qcydcbmltcG9ydCB7IGxvZ0ZvckRlYnVnZ2luZyB9IGZyb20gJy4uLy4uL3V0aWxzL2RlYnVnLmpzJ1xuaW1wb3J0IHsgY3JlYXRlVXNlck1lc3NhZ2UgfSBmcm9tICcuLi8uLi91dGlscy9tZXNzYWdlcy5qcydcbmltcG9ydCB7IGtpbGxJblByb2Nlc3NUZWFtbWF0ZSB9IGZyb20gJy4uLy4uL3V0aWxzL3N3YXJtL3NwYXduSW5Qcm9jZXNzLmpzJ1xuaW1wb3J0IHsgdXBkYXRlVGFza1N0YXRlIH0gZnJvbSAnLi4vLi4vdXRpbHMvdGFzay9mcmFtZXdvcmsuanMnXG5pbXBvcnQgdHlwZSB7IEluUHJvY2Vzc1RlYW1tYXRlVGFza1N0YXRlIH0gZnJvbSAnLi90eXBlcy5qcydcbmltcG9ydCB7IGFwcGVuZENhcHBlZE1lc3NhZ2UsIGlzSW5Qcm9jZXNzVGVhbW1hdGVUYXNrIH0gZnJvbSAnLi90eXBlcy5qcydcblxuLyoqXG4gKiBJblByb2Nlc3NUZWFtbWF0ZVRhc2sgLSBIYW5kbGVzIGluLXByb2Nlc3MgdGVhbW1hdGUgZXhlY3V0aW9uLlxuICovXG5leHBvcnQgY29uc3QgSW5Qcm9jZXNzVGVhbW1hdGVUYXNrOiBUYXNrID0ge1xuICBuYW1lOiAnSW5Qcm9jZXNzVGVhbW1hdGVUYXNrJyxcbiAgdHlwZTogJ2luX3Byb2Nlc3NfdGVhbW1hdGUnLFxuICBhc3luYyBraWxsKHRhc2tJZCwgc2V0QXBwU3RhdGUpIHtcbiAgICBraWxsSW5Qcm9jZXNzVGVhbW1hdGUodGFza0lkLCBzZXRBcHBTdGF0ZSlcbiAgfSxcbn1cblxuLyoqXG4gKiBSZXF1ZXN0IHNodXRkb3duIGZvciBhIHRlYW1tYXRlLlxuICovXG5leHBvcnQgZnVuY3Rpb24gcmVxdWVzdFRlYW1tYXRlU2h1dGRvd24oXG4gIHRhc2tJZDogc3RyaW5nLFxuICBzZXRBcHBTdGF0ZTogU2V0QXBwU3RhdGUsXG4pOiB2b2lkIHtcbiAgdXBkYXRlVGFza1N0YXRlPEluUHJvY2Vzc1RlYW1tYXRlVGFza1N0YXRlPih0YXNrSWQsIHNldEFwcFN0YXRlLCB0YXNrID0+IHtcbiAgICBpZiAodGFzay5zdGF0dXMgIT09ICdydW5uaW5nJyB8fCB0YXNrLnNodXRkb3duUmVxdWVzdGVkKSB7XG4gICAgICByZXR1cm4gdGFza1xuICAgIH1cblxuICAgIHJldHVybiB7XG4gICAgICAuLi50YXNrLFxuICAgICAgc2h1dGRvd25SZXF1ZXN0ZWQ6IHRydWUsXG4gICAgfVxuICB9KVxufVxuXG4vKipcbiAqIEFwcGVuZCBhIG1lc3NhZ2UgdG8gYSB0ZWFtbWF0ZSdzIGNvbnZlcnNhdGlvbiBoaXN0b3J5LlxuICogVXNlZCBmb3Igem9vbWVkIHZpZXcgdG8gc2hvdyB0aGUgdGVhbW1hdGUncyBjb252ZXJzYXRpb24uXG4gKi9cbmV4cG9ydCBmdW5jdGlvbiBhcHBlbmRUZWFtbWF0ZU1lc3NhZ2UoXG4gIHRhc2tJZDogc3RyaW5nLFxuICBtZXNzYWdlOiBNZXNzYWdlLFxuICBzZXRBcHBTdGF0ZTogU2V0QXBwU3RhdGUsXG4pOiB2b2lkIHtcbiAgdXBkYXRlVGFza1N0YXRlPEluUHJvY2Vzc1RlYW1tYXRlVGFza1N0YXRlPih0YXNrSWQsIHNldEFwcFN0YXRlLCB0YXNrID0+IHtcbiAgICBpZiAodGFzay5zdGF0dXMgIT09ICdydW5uaW5nJykge1xuICAgICAgcmV0dXJuIHRhc2tcbiAgICB9XG5cbiAgICByZXR1cm4ge1xuICAgICAgLi4udGFzayxcbiAgICAgIG1lc3NhZ2VzOiBhcHBlbmRDYXBwZWRNZXNzYWdlKHRhc2subWVzc2FnZXMsIG1lc3NhZ2UpLFxuICAgIH1cbiAgfSlcbn1cblxuLyoqXG4gKiBJbmplY3QgYSB1c2VyIG1lc3NhZ2UgdG8gYSB0ZWFtbWF0ZSdzIHBlbmRpbmc