yzcode-cli 1.0.2 → 1.0.4

package/skills/mcpSkillBuilders.ts

@@ -0,0 +1,44 @@
+import type {
+  createSkillCommand,
+  parseSkillFrontmatterFields,
+} from './loadSkillsDir.ts'
+
+/**
+ * Write-once registry for the two loadSkillsDir functions that MCP skill
+ * discovery needs. This module is a dependency-graph leaf: it imports nothing
+ * but types, so both mcpSkills.ts and loadSkillsDir.ts can depend on it
+ * without forming a cycle (client.ts → mcpSkills.ts → loadSkillsDir.ts → …
+ * → client.ts).
+ *
+ * The non-literal dynamic-import approach ("await import(variable)") fails at
+ * runtime in Bun-bundled binaries — the specifier is resolved against the
+ * chunk's /$bunfs/root/… path, not the original source tree, yielding "Cannot
+ * find module './loadSkillsDir.js'". A literal dynamic import works in bunfs
+ * but dependency-cruiser tracks it, and because loadSkillsDir transitively
+ * reaches almost everything, the single new edge fans out into many new cycle
+ * violations in the diff check.
+ *
+ * Registration happens at loadSkillsDir.ts module init, which is eagerly
+ * evaluated at startup via the static import from commands.ts — long before
+ * any MCP server connects.
+ */
+
+export type MCPSkillBuilders = {
+  createSkillCommand: typeof createSkillCommand
+  parseSkillFrontmatterFields: typeof parseSkillFrontmatterFields
+}
+
+let builders: MCPSkillBuilders | null = null
+
+export function registerMCPSkillBuilders(b: MCPSkillBuilders): void {
+  builders = b
+}
+
+export function getMCPSkillBuilders(): MCPSkillBuilders {
+  if (!builders) {
+    throw new Error(
+      'MCP skill builders not registered — loadSkillsDir.ts has not been evaluated yet',
+    )
+  }
+  return builders
+}

package/state/AppState.tsx

@@ -1,13 +1,13 @@
 import { c as _c } from "react/compiler-runtime";
 import { feature } from 'bun:bundle';
 import React, { useContext, useEffect, useEffectEvent, useState, useSyncExternalStore } from 'react';
-import { MailboxProvider } from '../context/mailbox.js';
-import { useSettingsChange } from '../hooks/useSettingsChange.js';
-import { logForDebugging } from '../utils/debug.js';
-import { createDisabledBypassPermissionsContext, isBypassPermissionsModeDisabled } from '../utils/permissions/permissionSetup.js';
-import { applySettingsChange } from '../utils/settings/applySettingsChange.js';
-import type { SettingSource } from '../utils/settings/constants.js';
-import { createStore } from './store.js';
+import { MailboxProvider } from '../context/mailbox.ts';
+import { useSettingsChange } from '../hooks/useSettingsChange.ts';
+import { logForDebugging } from '../utils/debug.ts';
+import { createDisabledBypassPermissionsContext, isBypassPermissionsModeDisabled } from '../utils/permissions/permissionSetup.ts';
+import { applySettingsChange } from '../utils/settings/applySettingsChange.ts';
+import type { SettingSource } from '../utils/settings/constants.ts';
+import { createStore } from './store.ts';
 
 // DCE: voice context is ant-only. External builds get a passthrough.
 /* eslint-disable @typescript-eslint/no-require-imports */
@@ -18,12 +18,12 @@ const VoiceProvider: (props: {
 }) => children;
 
 /* eslint-enable @typescript-eslint/no-require-imports */
-import { type AppState, type AppStateStore, getDefaultAppState } from './AppStateStore.js';
+import { type AppState, type AppStateStore, getDefaultAppState } from './AppStateStore.ts';
 
 // TODO: Remove these re-exports once all callers import directly from
 // ./AppStateStore.js. Kept for back-compat during migration so .ts callers
 // can incrementally move off the .tsx import and stop pulling React.
-export { type AppState, type AppStateStore, type CompletionBoundary, getDefaultAppState, IDLE_SPECULATION_STATE, type SpeculationResult, type SpeculationState } from './AppStateStore.js';
+export { type AppState, type AppStateStore, type CompletionBoundary, getDefaultAppState, IDLE_SPECULATION_STATE, type SpeculationResult, type SpeculationState } from './AppStateStore.ts';
 export const AppStoreContext = React.createContext<AppStateStore | null>(null);
 type Props = {
   children: React.ReactNode;

package/state/AppStateStore.ts

@@ -1,42 +1,42 @@
-import type { Notification } from 'src/context/notifications.js'
-import type { TodoList } from 'src/utils/todo/types.js'
-import type { BridgePermissionCallbacks } from '../bridge/bridgePermissionCallbacks.js'
-import type { Command } from '../commands.js'
-import type { ChannelPermissionCallbacks } from '../services/mcp/channelPermissions.js'
-import type { ElicitationRequestEvent } from '../services/mcp/elicitationHandler.js'
+import type { Notification } from '../context/notifications.ts'
+import type { TodoList } from '../utils/todo/types.ts'
+import type { BridgePermissionCallbacks } from '../bridge/bridgePermissionCallbacks.ts'
+import type { Command } from '../commands.ts'
+import type { ChannelPermissionCallbacks } from '../services/mcp/channelPermissions.ts'
+import type { ElicitationRequestEvent } from '../services/mcp/elicitationHandler.ts'
 import type {
   MCPServerConnection,
   ServerResource,
-} from '../services/mcp/types.js'
-import { shouldEnablePromptSuggestion } from '../services/PromptSuggestion/promptSuggestion.js'
+} from '../services/mcp/types.ts'
+import { shouldEnablePromptSuggestion } from '../services/PromptSuggestion/promptSuggestion.ts'
 import {
   getEmptyToolPermissionContext,
   type Tool,
   type ToolPermissionContext,
-} from '../Tool.js'
-import type { TaskState } from '../tasks/types.js'
-import type { AgentColorName } from '../tools/AgentTool/agentColorManager.js'
-import type { AgentDefinitionsResult } from '../tools/AgentTool/loadAgentsDir.js'
-import type { AllowedPrompt } from '../tools/ExitPlanModeTool/ExitPlanModeV2Tool.js'
-import type { AgentId } from '../types/ids.js'
-import type { Message, UserMessage } from '../types/message.js'
-import type { LoadedPlugin, PluginError } from '../types/plugin.js'
-import type { DeepImmutable } from '../types/utils.js'
+} from '../Tool.ts'
+import type { TaskState } from '../tasks/types.ts'
+import type { AgentColorName } from '../tools/AgentTool/agentColorManager.ts'
+import type { AgentDefinitionsResult } from '../tools/AgentTool/loadAgentsDir.ts'
+import type { AllowedPrompt } from '../tools/ExitPlanModeTool/ExitPlanModeV2Tool.ts'
+import type { AgentId } from '../types/ids.ts'
+import type { Message, UserMessage } from '../types/message.ts'
+import type { LoadedPlugin, PluginError } from '../types/plugin.ts'
+import type { DeepImmutable } from '../types/utils.ts'
 import {
   type AttributionState,
   createEmptyAttributionState,
-} from '../utils/commitAttribution.js'
-import type { EffortValue } from '../utils/effort.js'
-import type { FileHistoryState } from '../utils/fileHistory.js'
-import type { REPLHookContext } from '../utils/hooks/postSamplingHooks.js'
-import type { SessionHooksState } from '../utils/hooks/sessionHooks.js'
-import type { ModelSetting } from '../utils/model/model.js'
-import type { DenialTrackingState } from '../utils/permissions/denialTracking.js'
-import type { PermissionMode } from '../utils/permissions/PermissionMode.js'
-import { getInitialSettings } from '../utils/settings/settings.js'
-import type { SettingsJson } from '../utils/settings/types.js'
-import { shouldEnableThinkingByDefault } from '../utils/thinking.js'
-import type { Store } from './store.js'
+} from '../utils/commitAttribution.ts'
+import type { EffortValue } from '../utils/effort.ts'
+import type { FileHistoryState } from '../utils/fileHistory.ts'
+import type { REPLHookContext } from '../utils/hooks/postSamplingHooks.ts'
+import type { SessionHooksState } from '../utils/hooks/sessionHooks.ts'
+import type { ModelSetting } from '../utils/model/model.ts'
+import type { DenialTrackingState } from '../utils/permissions/denialTracking.ts'
+import type { PermissionMode } from '../utils/permissions/PermissionMode.ts'
+import { getInitialSettings } from '../utils/settings/settings.ts'
+import type { SettingsJson } from '../utils/settings/types.ts'
+import { shouldEnableThinkingByDefault } from '../utils/thinking.ts'
+import type { Store } from './store.ts'
 
 export type CompletionBoundary =
   | { type: 'complete'; completedAt: number; outputTokens: number }

package/state/onChangeAppState.ts

@@ -1,24 +1,24 @@
-import { setMainLoopModelOverride } from '../bootstrap/state.js'
+import { setMainLoopModelOverride } from '../bootstrap/state.ts'
 import {
   clearApiKeyHelperCache,
   clearAwsCredentialsCache,
   clearGcpCredentialsCache,
-} from '../utils/auth.js'
-import { getGlobalConfig, saveGlobalConfig } from '../utils/config.js'
-import { toError } from '../utils/errors.js'
-import { logError } from '../utils/log.js'
-import { applyConfigEnvironmentVariables } from '../utils/managedEnv.js'
+} from '../utils/auth.ts'
+import { getGlobalConfig, saveGlobalConfig } from '../utils/config.ts'
+import { toError } from '../utils/errors.ts'
+import { logError } from '../utils/log.ts'
+import { applyConfigEnvironmentVariables } from '../utils/managedEnv.ts'
 import {
   permissionModeFromString,
   toExternalPermissionMode,
-} from '../utils/permissions/PermissionMode.js'
+} from '../utils/permissions/PermissionMode.ts'
 import {
   notifyPermissionModeChanged,
   notifySessionMetadataChanged,
   type SessionExternalMetadata,
-} from '../utils/sessionState.js'
-import { updateSettingsForSource } from '../utils/settings/settings.js'
-import type { AppState } from './AppStateStore.js'
+} from '../utils/sessionState.ts'
+import { updateSettingsForSource } from '../utils/settings/settings.ts'
+import type { AppState } from './AppStateStore.ts'
 
 // Inverse of the push below — restore on worker restart.
 export function externalMetadataToAppState(

package/state/selectors.ts

@@ -3,10 +3,10 @@
  * Keep selectors pure and simple - just data extraction, no side effects.
  */
 
-import type { InProcessTeammateTaskState } from '../tasks/InProcessTeammateTask/types.js'
-import { isInProcessTeammateTask } from '../tasks/InProcessTeammateTask/types.js'
-import type { LocalAgentTaskState } from '../tasks/LocalAgentTask/LocalAgentTask.js'
-import type { AppState } from './AppStateStore.js'
+import type { InProcessTeammateTaskState } from '../tasks/InProcessTeammateTask/types.ts'
+import { isInProcessTeammateTask } from '../tasks/InProcessTeammateTask/types.ts'
+import type { LocalAgentTaskState } from '../tasks/LocalAgentTask/LocalAgentTask.ts'
+import type { AppState } from './AppStateStore.ts'
 
 /**
  * Get the currently viewed teammate task, if any.

package/state/teammateViewHelpers.ts

@@ -1,12 +1,12 @@
-import { logEvent } from '../services/analytics/index.js'
-import { isTerminalTaskStatus } from '../Task.js'
-import type { LocalAgentTaskState } from '../tasks/LocalAgentTask/LocalAgentTask.js'
+import { logEvent } from '../services/analytics/index.ts'
+import { isTerminalTaskStatus } from '../Task.ts'
+import type { LocalAgentTaskState } from '../tasks/LocalAgentTask/LocalAgentTask.ts'
 
 // Inlined from framework.ts — importing creates a cycle through
 // BackgroundTasksDialog. Keep in sync with PANEL_GRACE_MS there.
 const PANEL_GRACE_MS = 30_000
 
-import type { AppState } from './AppState.js'
+import type { AppState } from './AppState.ts'
 
 // Inline type check instead of importing isLocalAgentTask — breaks the
 // teammateViewHelpers → LocalAgentTask runtime edge that creates a cycle

package/tasks/DreamTask/DreamTask.ts

@@ -0,0 +1,157 @@
+// Background task entry for auto-dream (memory consolidation subagent).
+// Makes the otherwise-invisible forked agent visible in the footer pill and
+// Shift+Down dialog. The dream agent itself is unchanged — this is pure UI
+// surfacing via the existing task registry.
+
+import { rollbackConsolidationLock } from '../../services/autoDream/consolidationLock.ts'
+import type { SetAppState, Task, TaskStateBase } from '../../Task.ts'
+import { createTaskStateBase, generateTaskId } from '../../Task.ts'
+import { registerTask, updateTaskState } from '../../utils/task/framework.ts'
+
+// Keep only the N most recent turns for live display.
+const MAX_TURNS = 30
+
+// A single assistant turn from the dream agent, tool uses collapsed to a count.
+export type DreamTurn = {
+  text: string
+  toolUseCount: number
+}
+
+// No phase detection — the dream prompt has a 4-stage structure
+// (orient/gather/consolidate/prune) but we don't parse it. Just flip from
+// 'starting' to 'updating' when the first Edit/Write tool_use lands.
+export type DreamPhase = 'starting' | 'updating'
+
+export type DreamTaskState = TaskStateBase & {
+  type: 'dream'
+  phase: DreamPhase
+  sessionsReviewing: number
+  /**
+   * Paths observed in Edit/Write tool_use blocks via onMessage. This is an
+   * INCOMPLETE reflection of what the dream agent actually changed — it misses
+   * any bash-mediated writes and only captures the tool calls we pattern-match.
+   * Treat as "at least these were touched", not "only these were touched".
+   */
+  filesTouched: string[]
+  /** Assistant text responses, tool uses collapsed. Prompt is NOT included. */
+  turns: DreamTurn[]
+  abortController?: AbortController
+  /** Stashed so kill can rewind the lock mtime (same path as fork-failure). */
+  priorMtime: number
+}
+
+export function isDreamTask(task: unknown): task is DreamTaskState {
+  return (
+    typeof task === 'object' &&
+    task !== null &&
+    'type' in task &&
+    task.type === 'dream'
+  )
+}
+
+export function registerDreamTask(
+  setAppState: SetAppState,
+  opts: {
+    sessionsReviewing: number
+    priorMtime: number
+    abortController: AbortController
+  },
+): string {
+  const id = generateTaskId('dream')
+  const task: DreamTaskState = {
+    ...createTaskStateBase(id, 'dream', 'dreaming'),
+    type: 'dream',
+    status: 'running',
+    phase: 'starting',
+    sessionsReviewing: opts.sessionsReviewing,
+    filesTouched: [],
+    turns: [],
+    abortController: opts.abortController,
+    priorMtime: opts.priorMtime,
+  }
+  registerTask(task, setAppState)
+  return id
+}
+
+export function addDreamTurn(
+  taskId: string,
+  turn: DreamTurn,
+  touchedPaths: string[],
+  setAppState: SetAppState,
+): void {
+  updateTaskState<DreamTaskState>(taskId, setAppState, task => {
+    const seen = new Set(task.filesTouched)
+    const newTouched = touchedPaths.filter(p => !seen.has(p) && seen.add(p))
+    // Skip the update entirely if the turn is empty AND nothing new was
+    // touched. Avoids re-rendering on pure no-ops.
+    if (
+      turn.text === '' &&
+      turn.toolUseCount === 0 &&
+      newTouched.length === 0
+    ) {
+      return task
+    }
+    return {
+      ...task,
+      phase: newTouched.length > 0 ? 'updating' : task.phase,
+      filesTouched:
+        newTouched.length > 0
+          ? [...task.filesTouched, ...newTouched]
+          : task.filesTouched,
+      turns: task.turns.slice(-(MAX_TURNS - 1)).concat(turn),
+    }
+  })
+}
+
+export function completeDreamTask(
+  taskId: string,
+  setAppState: SetAppState,
+): void {
+  // notified: true immediately — dream has no model-facing notification path
+  // (it's UI-only), and eviction requires terminal + notified. The inline
+  // appendSystemMessage completion note IS the user surface.
+  updateTaskState<DreamTaskState>(taskId, setAppState, task => ({
+    ...task,
+    status: 'completed',
+    endTime: Date.now(),
+    notified: true,
+    abortController: undefined,
+  }))
+}
+
+export function failDreamTask(taskId: string, setAppState: SetAppState): void {
+  updateTaskState<DreamTaskState>(taskId, setAppState, task => ({
+    ...task,
+    status: 'failed',
+    endTime: Date.now(),
+    notified: true,
+    abortController: undefined,
+  }))
+}
+
+export const DreamTask: Task = {
+  name: 'DreamTask',
+  type: 'dream',
+
+  async kill(taskId, setAppState) {
+    let priorMtime: number | undefined
+    updateTaskState<DreamTaskState>(taskId, setAppState, task => {
+      if (task.status !== 'running') return task
+      task.abortController?.abort()
+      priorMtime = task.priorMtime
+      return {
+        ...task,
+        status: 'killed',
+        endTime: Date.now(),
+        notified: true,
+        abortController: undefined,
+      }
+    })
+    // Rewind the lock mtime so the next session can retry. Same path as the
+    // fork-failure catch in autoDream.ts. If updateTaskState was a no-op
+    // (already terminal), priorMtime stays undefined and we skip.
+    if (priorMtime !== undefined) {
+      await rollbackConsolidationLock(priorMtime)
+    }
+  },
+}

package/tasks/InProcessTeammateTask/InProcessTeammateTask.tsx

@@ -0,0 +1,126 @@
+/**
+ * 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.ts';
+import type { Message } from '../../types/message.ts';
+import { logForDebugging } from '../../utils/debug.ts';
+import { createUserMessage } from '../../utils/messages.ts';
+import { killInProcessTeammate } from '../../utils/swarm/spawnInProcess.ts';
+import { updateTaskState } from '../../utils/task/framework.ts';
+import type { InProcessTeammateTaskState } from './types.ts';
+import { appendCappedMessage, isInProcessTeammateTask } from './types.ts';
+
+/**
+ * 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+IHtcbiAgICBpZiAodGFzay5zdGF0dXMgIT09ICdydW5uaW5nJykge1xuICAgICAgcmV0dXJuIHRhc2tcbiAgICB9XG5cbiAgICByZXR1cm4ge1xuICAgICAgLi4udGFzayxcbiAgICAgIG1lc3NhZ2VzOiBhcHBlbmRDYXBwZWRNZXNzYWdlKHRhc2subWVzc2FnZXMsIG1lc3NhZ2UpLFxuICAgIH1cbiAgfSlcbn1cblxuLyoqXG4gKiBJbmplY3QgYSB1c2VyIG1lc3NhZ2UgdG8gYSB0ZWFtbWF0ZSdzIHBlbmRpbmcgcXVldWUuXG4gKiBVc2VkIHdoZW4gdmlld2luZyBhIHRlYW1tYXRlJ3MgdHJhbnNjcmlwdCB0byBzZW5kIHR5cGVkIG1lc3NhZ2VzIHRvIHRoZW0uXG4gKiBBbHNvIGFkZHMgdGhlIG1lc3NhZ2UgdG8gdGFzay5tZXNzYWdlcyBzbyBpdCBhcHBlYXJzIGltbWVkaWF0ZWx5IGluIHRoZSB0cmFuc2NyaXB0LlxuICovXG5leHBvcnQgZnVuY3Rpb24gaW5qZWN0VXNlck1lc3NhZ2VUb1RlYW1tYXRlKFxuICB0YXNrSWQ6IHN0cmluZyxcbiAgbWVzc2FnZTogc3RyaW5nLFxuICBzZXRBcHBTdGF0ZTogU2V0QXBwU3RhdGUsXG4pOiB2b2lkIHtcbiAgdXBkYXRlVGFza1N0YXRlPEluUHJvY2Vzc1RlYW1tYXRlVGFza1N0YXRlPih0YXNrSWQsIHNldEFwcFN0YXRlLCB0YXNrID0+IHtcbiAgICAvLyBBbGxvdyBtZXNzYWdlIGluamVjdGlvbiB3aGVuIHRlYW1tYXRlIGlzIHJ1bm5pbmcgb3IgaWRsZSAod2FpdGluZyBmb3IgaW5wdXQpXG4gICAgLy8gT25seSByZWplY3QgaWYgdGVhbW1hdGUgaXMgaW4gYSB0ZXJtaW5hbCBzdGF0ZVxuICAgIGlmIChpc1Rlcm1pbmFsVGFza1N0YXR1cyh0YXNrLnN0YXR1cykpIHtcbiAgICAgIGxvZ0ZvckRlYnVnZ2luZyhcbiAgICAgICAgYERyb3BwaW5nIG1lc3NhZ2UgZm9yIHRlYW1tYXRlIHRhc2sgJHt0YXNrSWR9OiB0YXNrIHN0YXR1cyBpcyBcIiR7dGFzay5zdGF0dXN9XCJgLFxuICAgICAgKVxuICAgICAgcmV0dXJuIHRhc2tcbiAgICB9XG5cbiAgICByZXR1cm4ge1xuICAgICAgLi4udGFzayxcbiAgICAgIHBlbmRpbmdVc2VyTWVzc2FnZXM6IFsuLi50YXNrLnBlbmRpbmdVc2VyTWVzc2FnZXMsIG1lc3NhZ2VdLFxuICAgICAgbWVzc2FnZXM6IGFwcGVuZENhcHBlZE1lc3NhZ2UoXG4gICAgICAgIHRhc2subWVzc2FnZXMsXG4gICAgICAgIGNyZWF0ZVVzZXJNZXNzYWdlKHsgY29udGVudDogbWVzc2FnZSB9KSxcbiAgICAgICksXG4gICAgfVxuICB9KVxufVxuXG4vKipcbiAqIEdldCB0ZWFtbWF0ZSB0YXNrIGJ5IGFnZW50IElEIGZyb20gQXBwU3RhdGUuXG4gKiBQcmVmZXJzIHJ1bm5pbmcgdGFza3Mgb3ZlciBraWxsZWQvY29tcGxldGVkIG9uZXMgaW4gY2FzZSBtdWx0aXBsZSB0YXNrc1xuICogd2l0aCB0aGUgc2FtZSBhZ2VudElkIGV4aXN0LlxuICogUmV0dXJucyB1bmRlZmluZWQgaWYgbm90IGZvdW5kLlxuICovXG5leHBvcnQgZnVuY3Rpb24gZmluZFRlYW1tYXRlVGFza0J5QWdlbnRJZChcbiAgYWdlbnRJZDogc3RyaW5nLFxuICB0YXNrczogUmVjb3JkPHN0cmluZywgVGFza1N0YXRlQmFzZT4sXG4pOiBJblByb2Nlc3NUZWFtbWF0ZVRhc2tTdGF0ZSB8IHVuZGVmaW5lZCB7XG4gIGxldCBmYWxsYmFjazogSW5Qcm9jZXNzVGVhbW1hdGVUYXNrU3RhdGUgfCB1bmRlZmluZWRcbiAgZm9yIChjb25zdCB0YXNrIG9mIE9iamVjdC52YWx1ZXModGFza3MpKSB7XG4gICAgaWYgKGlzSW5Qcm9jZXNzVGVhbW1hdGVUYXNrKHRhc2spICYmIHRhc2suaWRlbnRpdHkuYWdlbnRJZCA9PT0gYWdlbnRJZCkge1xuICAgICAgLy8gUHJlZmVyIHJ1bm5pbmcgdGFza3MgaW4gY2FzZSBvbGQga2lsbGVkIHRhc2tzIHN0aWxsIGV4aXN0IGluIEFwcFN0YXRlXG4gICAgICAvLyBhbG9uZ3NpZGUgbmV3IHJ1bm5pbmcgb25lcyB3aXRoIHRoZSBzYW1lIGFnZW50SWRcbiAgICAgIGlmICh0YXNrLnN0YXR1cyA9PT0gJ3J1bm5pbmcnKSB7XG4gICAgICAgIHJldHVybiB0YXNrXG4gICAgICB9XG4gICAgICAvLyBLZWVwIGZpcnN0IG1hdGNoIGFzIGZhbGxiYWNrIGluIGNhc2Ugbm8gcnVubmluZyB0YXNrIGV4aXN0c1xuICAgICAgaWYgKCFmYWxsYmFjaykge1xuICAgICAgICBmYWxsYmFjayA9IHRhc2tcbiAgICAgIH1cbiAgICB9XG4gIH1cbiAgcmV0dXJuIGZhbGxiYWNrXG59XG5cbi8qKlxuICogR2V0IGFsbCBpbi1wcm9jZXNzIHRlYW1tYXRlIHRhc2tzIGZyb20gQXBwU3RhdGUuXG4gKi9cbmV4cG9ydCBmdW5jdGlvbiBnZXRBbGxJblByb2Nlc3NUZWFtbWF0ZVRhc2tzKFxuICB0YXNrczogUmVjb3JkPHN0cmluZywgVGFza1N0YXRlQmFzZT4sXG4pOiBJblByb2Nlc3NUZWFtbWF0ZVRhc2tTdGF0ZVtdIHtcbiAgcmV0dXJuIE9iamVjdC52YWx1ZXModGFza3MpLmZpbHRlcihpc0luUHJvY2Vzc1RlYW1tYXRlVGFzaylcbn1cblxuLyoqXG4gKiBHZXQgcnVubmluZyBpbi1wcm9jZXNzIHRlYW1tYXRlcyBzb3J0ZWQgYWxwaGFiZXRpY2FsbHkgYnkgYWdlbnROYW1lLlxuICogU2hhcmVkIGJldHdlZW4gVGVhbW1hdGVTcGlubmVyVHJlZSBkaXNwbGF5LCBQcm9tcHRJbnB1dCBmb290ZXIgc2VsZWN0b3IsXG4gKiBhbmQgdXNlQmFja2dyb3VuZFRhc2tOYXZpZ2F0aW9uIOKAlCBzZWxlY3RlZElQQWdlbnRJbmRleCBtYXBzIGludG8gdGhpc1xuICogYXJyYXksIHNvIGFsbCB0aHJlZSBtdXN0IGFncmVlIG9uIHNvcnQgb3JkZXIuXG4gKi9cbmV4cG9ydCBmdW5jdGlvbiBnZXRSdW5uaW5nVGVhbW1hdGVzU29ydGVkKFxuICB0YXNrczogUmVjb3JkPHN0cmluZywgVGFza1N0YXRlQmFzZT4sXG4pOiBJblByb2Nlc3NUZWFtbWF0ZVRhc2tTdGF0ZVtdIHtcbiAgcmV0dXJuIGdldEFsbEluUHJvY2Vzc1RlYW1tYXRlVGFza3ModGFza3MpXG4gICAgLmZpbHRlcih0ID0+IHQuc3RhdHVzID09PSAncnVubmluZycpXG4gICAgLnNvcnQoKGEsIGIpID0+IGEuaWRlbnRpdHkuYWdlbnROYW1lLmxvY2FsZUNvbXBhcmUoYi5pZGVudGl0eS5hZ2VudE5hbWUpKVxufVxuIl0sIm1hcHBpbmdzIjoiQUFBQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTs7QUFFQSxTQUNFQSxvQkFBb0IsRUFDcEIsS0FBS0MsV0FBVyxFQUNoQixLQUFLQyxJQUFJLEVBQ1QsS0FBS0MsYUFBYSxRQUNiLGVBQWU7QUFDdEIsY0FBY0MsT0FBTyxRQUFRLHdCQUF3QjtBQUNyRCxTQUFTQyxlQUFlLFFBQVEsc0JBQXNCO0FBQ3RELFNBQVNDLGlCQUFpQixRQUFRLHlCQUF5QjtBQUMzRCxTQUFTQyxxQkFBcUIsUUFBUSxxQ0FBcUM7QUFDM0UsU0FBU0MsZUFBZSxRQUFRLCtCQUErQjtBQUMvRCxjQUFjQywwQkFBMEIsUUFBUSxZQUFZO0FBQzVELFNBQVNDLG1CQUFtQixFQUFFQyx1QkFBdUIsUUFBUSxZQUFZOztBQUV6RTtBQUNBO0FBQ0E7QUFDQSxPQUFPLE1BQU1DLHFCQUFxQixFQUFFVixJQUFJLEdBQUc7RUFDekNXLElBQUksRUFBRSx1QkFBdUI7RUFDN0JDLElBQUksRUFBRSxxQkFBcUI7RUFDM0IsTUFBTUMsSUFBSUEsQ0FBQ0MsTUFBTSxFQUFFQyxXQUFXLEVBQUU7SUFDOUJWLHFCQUFxQixDQUFDUyxNQUFNLEVBQUVDLFdBQVcsQ0FBQztFQUM1QztBQUNGLENBQUM7O0FBRUQ7QUFDQTtBQUNBO0FBQ0EsT0FBTyxTQUFTQyx1QkFBdUJBLENBQ3JDRixNQUFNLEVBQUUsTUFBTSxFQUNkQyxXQUFXLEVBQUVoQixXQUFXLENBQ3pCLEVBQUUsSUFBSSxDQUFDO0VBQ05PLGVBQWUsQ0FBQ0MsMEJBQTBCLENBQUMsQ0FBQ08sTUFBTSxFQUFFQyxXQUFXLEVBQUVFLElBQUksSUFBSTtJQUN2RSxJQUFJQSxJQUFJLENBQUNDLE1BQU0sS0FBSyxTQUFTLElBQUlELElBQUksQ0FBQ0UsaUJBQWlCLEVBQUU7TUFDdkQsT0FBT0YsSUFBSTtJQUNiO0lBRUEsT0FBTztNQUNMLEdBQUdBLElBQUk7TUFDUEUsaUJBQWlCLEVBQUU7SUFDckIsQ0FBQztFQUNILENBQUMsQ0FBQztBQUNKOztBQUVBO0FBQ0E7QUFDQTtBQUNBO0FBQ0EsT0FBTyxTQUFTQyxxQkFBcUJBLENBQ25DTixNQUFNLEVBQUUsTUFBTSxFQUNkTyxPQUFPLEVBQUVuQixPQUFPLEVBQ2hCYSxXQUFXLEVBQUVoQixXQUFXLENBQ3pCLEVBQUUsSUFBSSxDQUFDO0VBQ05PLGVBQWUsQ0FBQ0MsMEJBQTBCLENBQUMsQ0FBQ08sTUFBTSxFQUFFQyxXQUFXLEVBQUVFLElBQUksSUFBSTtJQUN2RSxJQUFJQSxJQUFJLENBQUNDLE1BQU0sS0FBSyxTQUFTLEVBQUU7TUFDN0IsT0FBT0QsSUFBSTtJQUNiO0lBRUEsT0FBTztNQUNMLEdBQUdBLElBQUk7TUFDUEssUUFBUSxFQUFFZCxtQkFBbUIsQ0FBQ1MsSUFBSSxDQUFDSyxRQUFRLEVBQUVELE9BQU87SUFDdEQsQ0FBQztFQUNILENBQUMsQ0FBQztBQUNKOztBQUVBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQSxPQUFPLFNBQVNFLDJCQUEyQkEsQ0FDekNULE1BQU0sRUFBRSxNQUFNLEVBQ2RPLE9BQU8sRUFBRSxNQUFNLEVBQ2ZOLFdBQVcsRUFBRWhCLFdBQVcsQ0FDekIsRUFBRSxJQUFJLENBQUM7RUFDTk8sZUFBZSxDQUFDQywwQkFBMEIsQ0FBQyxDQUFDTyxNQUFNLEVBQUVDLFdBQVcsRUFBRUUsSUFBSSxJQUFJO0lBQ3ZFO0lBQ0E7SUFDQSxJQUFJbkIsb0JBQW9CLENBQUNtQixJQUFJLENBQUNDLE1BQU0sQ0FBQyxFQUFFO01BQ3JDZixlQUFlLENBQ2Isc0NBQXNDVyxNQUFNLHFCQUFxQkcsSUFBSSxDQUFDQyxNQUFNLEdBQzlFLENBQUM7TUFDRCxPQUFPRCxJQUFJO0lBQ2I7SUFFQSxPQUFPO01BQ0wsR0FBR0EsSUFBSTtNQUNQTyxtQkFBbUIsRUFBRSxDQUFDLEdBQUdQLElBQUksQ0FBQ08sbUJBQW1CLEVBQUVILE9BQU8sQ0FBQztNQUMzREMsUUFBUSxFQUFFZCxtQkFBbUIsQ0FDM0JTLElBQUksQ0FBQ0ssUUFBUSxFQUNibEIsaUJBQWlCLENBQUM7UUFBRXFCLE9BQU8sRUFBRUo7TUFBUSxDQUFDLENBQ3hDO0lBQ0YsQ0FBQztFQUNILENBQUMsQ0FBQztBQUNKOztBQUVBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBLE9BQU8sU0FBU0sseUJBQXlCQSxDQUN2Q0MsT0FBTyxFQUFFLE1BQU0sRUFDZkMsS0FBSyxFQUFFQyxNQUFNLENBQUMsTUFBTSxFQUFFNUIsYUFBYSxDQUFDLENBQ3JDLEVBQUVNLDBCQUEwQixHQUFHLFNBQVMsQ0FBQztFQUN4QyxJQUFJdUIsUUFBUSxFQUFFdkIsMEJBQTBCLEdBQUcsU0FBUztFQUNwRCxLQUFLLE1BQU1VLElBQUksSUFBSWMsTUFBTSxDQUFDQyxNQUFNLENBQUNKLEtBQUssQ0FBQyxFQUFFO0lBQ3ZDLElBQUluQix1QkFBdUIsQ0FBQ1EsSUFBSSxDQUFDLElBQUlBLElBQUksQ0FBQ2dCLFFBQVEsQ0FBQ04sT0FBTyxLQUFLQSxPQUFPLEVBQUU7TUFDdEU7TUFDQTtNQUNBLElBQUlWLElBQUksQ0FBQ0MsTUFBTSxLQUFLLFNBQVMsRUFBRTtRQUM3QixPQUFPRCxJQUFJO01BQ2I7TUFDQTtNQUNBLElBQUksQ0FBQ2EsUUFBUSxFQUFFO1FBQ2JBLFFBQVEsR0FBR2IsSUFBSTtNQUNqQjtJQUNGO0VBQ0Y7RUFDQSxPQUFPYSxRQUFRO0FBQ2pCOztBQUVBO0FBQ0E7QUFDQTtBQUNBLE9BQU8sU0FBU0ksNEJBQTRCQSxDQUMxQ04sS0FBSyxFQUFFQyxNQUFNLENBQUMsTUFBTSxFQUFFNUIsYUFBYSxDQUFDLENBQ3JDLEVBQUVNLDBCQUEwQixFQUFFLENBQUM7RUFDOUIsT0FBT3dCLE1BQU0sQ0FBQ0MsTUFBTSxDQUFDSixLQUFLLENBQUMsQ0FBQ08sTUFBTSxDQUFDMUIsdUJBQXVCLENBQUM7QUFDN0Q7O0FBRUE7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0EsT0FBTyxTQUFTMkIseUJBQXlCQSxDQUN2Q1IsS0FBSyxFQUFFQyxNQUFNLENBQUMsTUFBTSxFQUFFNUIsYUFBYSxDQUFDLENBQ3JDLEVBQUVNLDBCQUEwQixFQUFFLENBQUM7RUFDOUIsT0FBTzJCLDRCQUE0QixDQUFDTixLQUFLLENBQUMsQ0FDdkNPLE1BQU0sQ0FBQ0UsQ0FBQyxJQUFJQSxDQUFDLENBQUNuQixNQUFNLEtBQUssU0FBUyxDQUFDLENBQ25Db0IsSUFBSSxDQUFDLENBQUNDLENBQUMsRUFBRUMsQ0FBQyxLQUFLRCxDQUFDLENBQUNOLFFBQVEsQ0FBQ1EsU0FBUyxDQUFDQyxhQUFhLENBQUNGLENBQUMsQ0FBQ1AsUUFBUSxDQUFDUSxTQUFTLENBQUMsQ0FBQztBQUM3RSIsImlnbm9yZUxpc3QiOltdfQ==

package/tasks/InProcessTeammateTask/types.ts

@@ -0,0 +1,121 @@
+import type { TaskStateBase } from '../../Task.ts'
+import type { AgentToolResult } from '../../tools/AgentTool/agentToolUtils.ts'
+import type { AgentDefinition } from '../../tools/AgentTool/loadAgentsDir.ts'
+import type { Message } from '../../types/message.ts'
+import type { PermissionMode } from '../../utils/permissions/PermissionMode.ts'
+import type { AgentProgress } from '../LocalAgentTask/LocalAgentTask.ts'
+
+/**
+ * Teammate identity stored in task state.
+ * Same shape as TeammateContext (runtime) but stored as plain data.
+ * TeammateContext is for AsyncLocalStorage; this is for AppState persistence.
+ */
+export type TeammateIdentity = {
+  agentId: string // e.g., "researcher@my-team"
+  agentName: string // e.g., "researcher"
+  teamName: string
+  color?: string
+  planModeRequired: boolean
+  parentSessionId: string // Leader's session ID
+}
+
+export type InProcessTeammateTaskState = TaskStateBase & {
+  type: 'in_process_teammate'
+
+  // Identity as sub-object (matches TeammateContext shape for consistency)
+  // Stored as plain data in AppState, NOT a reference to AsyncLocalStorage
+  identity: TeammateIdentity
+
+  // Execution
+  prompt: string
+  // Optional model override for this teammate
+  model?: string
+  // Optional: Only set if teammate uses a specific agent definition
+  // Many teammates run as general-purpose agents without a predefined definition
+  selectedAgent?: AgentDefinition
+  abortController?: AbortController // Runtime only, not serialized to disk - kills WHOLE teammate
+  currentWorkAbortController?: AbortController // Runtime only - aborts current turn without killing teammate
+  unregisterCleanup?: () => void // Runtime only
+
+  // Plan mode approval tracking (planModeRequired is in identity)
+  awaitingPlanApproval: boolean
+
+  // Permission mode for this teammate (cycled independently via Shift+Tab when viewing)
+  permissionMode: PermissionMode
+
+  // State
+  error?: string
+  result?: AgentToolResult // Reuse existing type since teammates run via runAgent()
+  progress?: AgentProgress
+
+  // Conversation history for zoomed view (NOT mailbox messages)
+  // Mailbox messages are stored separately in teamContext.inProcessMailboxes
+  messages?: Message[]
+
+  // Tool use IDs currently being executed (for animation in transcript view)
+  inProgressToolUseIDs?: Set<string>
+
+  // Queue of user messages to deliver when viewing teammate transcript
+  pendingUserMessages: string[]
+
+  // UI: random spinner verbs (stable across re-renders, shared between components)
+  spinnerVerb?: string
+  pastTenseVerb?: string
+
+  // Lifecycle
+  isIdle: boolean
+  shutdownRequested: boolean
+
+  // Callbacks to notify when teammate becomes idle (runtime only)
+  // Used by leader to efficiently wait without polling
+  onIdleCallbacks?: Array<() => void>
+
+  // Progress tracking (for computing deltas in notifications)
+  lastReportedToolCount: number
+  lastReportedTokenCount: number
+}
+
+export function isInProcessTeammateTask(
+  task: unknown,
+): task is InProcessTeammateTaskState {
+  return (
+    typeof task === 'object' &&
+    task !== null &&
+    'type' in task &&
+    task.type === 'in_process_teammate'
+  )
+}
+
+/**
+ * Cap on the number of messages kept in task.messages (the AppState UI mirror).
+ *
+ * task.messages exists purely for the zoomed transcript dialog, which only
+ * needs recent context. The full conversation lives in the local allMessages
+ * array (inProcessRunner) and on disk at the agent transcript path.
+ *
+ * BQ analysis (round 9, 2026-03-20) showed ~20MB RSS per agent at 500+ turn
+ * sessions and ~125MB per concurrent agent in swarm bursts. Whale session
+ * 9a990de8 launched 292 agents in 2 minutes and reached 36.8GB. The dominant
+ * cost is this array holding a second full copy of every message.
+ */
+export const TEAMMATE_MESSAGES_UI_CAP = 50
+
+/**
+ * Append an item to a message array, capping the result at
+ * TEAMMATE_MESSAGES_UI_CAP entries by dropping the oldest. Always returns
+ * a new array (AppState immutability).
+ */
+export function appendCappedMessage<T>(
+  prev: readonly T[] | undefined,
+  item: T,
+): T[] {
+  if (prev === undefined || prev.length === 0) {
+    return [item]
+  }
+  if (prev.length >= TEAMMATE_MESSAGES_UI_CAP) {
+    const next = prev.slice(-(TEAMMATE_MESSAGES_UI_CAP - 1))
+    next.push(item)
+    return next
+  }
+  return [...prev, item]
+}