rimuru-ai 1.19.0

package/src/cli/cmd/run/runtime.ts

@@ -0,0 +1,814 @@
+// Top-level orchestrator for `run --interactive`.
+//
+// Wires the boot sequence, lifecycle (renderer + footer), stream transport,
+// and prompt queue together into a single session loop. Two entry points:
+//
+//   runInteractiveMode     -- used when an SDK client already exists (attach mode)
+//   runInteractiveLocalMode -- used for local in-process mode (no server)
+//
+// Both delegate to runInteractiveRuntime, which:
+//   1. resolves TUI config, model info, and session history,
+//   2. creates the split-footer lifecycle (renderer + RunFooter),
+//   3. starts the stream transport (SDK event subscription), lazily for fresh
+//      local sessions,
+//   4. runs the prompt queue until the footer closes.
+import { createRimuruClient } from "@rimurucode-ai/sdk/v2"
+import { Flag } from "@rimurucode-ai/core/flag/flag"
+import { MessageID } from "@/session/schema"
+import { createRunDemo } from "./demo"
+import { resolveModelInfo, resolveRunTuiConfig, resolveSessionInfo } from "./runtime.boot"
+import { createRuntimeLifecycle } from "./runtime.lifecycle"
+import { trace } from "./trace"
+import { cycleVariant, formatModelLabel, resolveSavedVariant, resolveVariant, saveVariant } from "./variant.shared"
+import type { LocalReplayAnchor, LocalReplayRow, RunInput, RunPrompt, RunProvider, StreamCommit } from "./types"
+
+/** @internal Exported for testing */
+export { pickVariant, resolveVariant } from "./variant.shared"
+
+/** @internal Exported for testing */
+export { runPromptQueue } from "./runtime.queue"
+
+type BootContext = Pick<
+  RunInput,
+  "sdk" | "directory" | "sessionID" | "sessionTitle" | "resume" | "agent" | "model" | "variant"
+>
+
+type CreateSessionInput = {
+  agent: string | undefined
+  model: RunInput["model"]
+  variant: string | undefined
+}
+
+type CreateSession = (sdk: RunInput["sdk"], input: CreateSessionInput) => Promise<{ id: string; title?: string }>
+
+type RunRuntimeInput = {
+  boot: () => Promise<BootContext>
+  afterPaint?: (ctx: BootContext) => Promise<void> | void
+  resolveSession?: (
+    ctx: BootContext,
+  ) => Promise<{ sessionID: string; sessionTitle?: string; agent?: string | undefined }>
+  createSession?: (ctx: BootContext, input: CreateSessionInput) => Promise<ResolvedSession>
+  files: RunInput["files"]
+  initialInput?: string
+  thinking: boolean
+  backgroundSubagents: boolean
+  replay?: boolean
+  replayLimit?: number
+  demo?: RunInput["demo"]
+}
+
+type RunLocalInput = {
+  directory: string
+  fetch: typeof globalThis.fetch
+  resolveAgent: () => Promise<string | undefined>
+  session: (sdk: RunInput["sdk"]) => Promise<{ id: string; title?: string } | undefined>
+  share: (sdk: RunInput["sdk"], sessionID: string) => Promise<void>
+  createSession?: CreateSession
+  agent: RunInput["agent"]
+  model: RunInput["model"]
+  variant: RunInput["variant"]
+  files: RunInput["files"]
+  initialInput?: string
+  thinking: boolean
+  backgroundSubagents: boolean
+  replay?: boolean
+  replayLimit?: number
+  demo?: RunInput["demo"]
+}
+
+type StreamTransportModule = Pick<
+  Awaited<typeof import("./stream.transport")>,
+  "createSessionTransport" | "formatUnknownError"
+>
+
+export type RunRuntimeDeps = {
+  createRuntimeLifecycle?: typeof createRuntimeLifecycle
+  streamTransport?: Promise<StreamTransportModule>
+}
+
+type StreamState = {
+  mod: StreamTransportModule
+  handle: Awaited<ReturnType<StreamTransportModule["createSessionTransport"]>>
+}
+
+type ResolvedSession = {
+  sessionID: string
+  sessionTitle?: string
+  agent?: string | undefined
+}
+
+function createSessionResolver(fn?: CreateSession) {
+  if (!fn) {
+    return undefined
+  }
+
+  return async (ctx: BootContext, input: CreateSessionInput): Promise<ResolvedSession> => {
+    const created = await fn(ctx.sdk, input)
+    if (!created.id) {
+      throw new Error("Failed to create session")
+    }
+
+    return {
+      sessionID: created.id,
+      sessionTitle: created.title,
+      agent: input.agent,
+    }
+  }
+}
+
+type RuntimeState = {
+  shown: boolean
+  aborting: boolean
+  model: RunInput["model"]
+  providers: RunProvider[]
+  variants: string[]
+  limits: Record<string, number>
+  activeVariant: string | undefined
+  sessionID: string
+  history: RunPrompt[]
+  localRows: LocalReplayRow[]
+  sessionTitle?: string
+  agent: string | undefined
+  switching?: Promise<void>
+  demo?: ReturnType<typeof createRunDemo>
+  selectSubagent?: (sessionID: string | undefined) => void
+  session?: Promise<void>
+  stream?: Promise<StreamState>
+}
+
+function hasSession(input: RunRuntimeInput, state: RuntimeState) {
+  return !input.resolveSession || !!state.sessionID
+}
+
+function eagerStream(input: RunRuntimeInput, ctx: BootContext) {
+  return ctx.resume === true || !input.resolveSession || !!input.demo
+}
+
+function variantsFor(providers: RunProvider[], model: RunInput["model"]) {
+  if (!model) {
+    return []
+  }
+
+  return Object.keys(providers.find((item) => item.id === model.providerID)?.models?.[model.modelID]?.variants ?? {})
+}
+
+const RESIZE_DELAY = 250
+const LOCAL_REPLAY_ROW_LIMIT = 100
+
+async function resolveExitTitle(
+  ctx: BootContext,
+  input: RunRuntimeInput,
+  state: RuntimeState,
+): Promise<string | undefined> {
+  if (!state.shown || !hasSession(input, state)) {
+    return undefined
+  }
+
+  return ctx.sdk.session
+    .get({
+      sessionID: state.sessionID,
+    })
+    .then((x) => x.data?.title)
+    .catch(() => undefined)
+}
+
+// Core runtime loop. Boot resolves the SDK context, then we set up the
+// lifecycle (renderer + footer), wire the stream transport for SDK events,
+// and feed prompts through the queue until the user exits.
+//
+// Files only attach on the first prompt turn -- after that, includeFiles
+// flips to false so subsequent turns don't re-send attachments.
+async function runInteractiveRuntime(input: RunRuntimeInput, deps: RunRuntimeDeps = {}): Promise<void> {
+  const start = performance.now()
+  const log = trace()
+  const tuiConfigTask = resolveRunTuiConfig()
+  const ctx = await input.boot()
+  const modelTask = resolveModelInfo(ctx.sdk, ctx.directory, ctx.model)
+  const sessionTask =
+    ctx.resume === true
+      ? resolveSessionInfo(ctx.sdk, ctx.sessionID, ctx.model)
+      : Promise.resolve({
+          first: true,
+          history: [],
+          variant: undefined,
+        })
+  const savedTask = resolveSavedVariant(ctx.model)
+  const [tuiConfig, session, savedVariant] = await Promise.all([tuiConfigTask, sessionTask, savedTask])
+  const state: RuntimeState = {
+    shown: !session.first,
+    aborting: false,
+    model: ctx.model,
+    providers: [],
+    variants: [],
+    limits: {},
+    activeVariant: resolveVariant(ctx.variant, session.variant, savedVariant, []),
+    sessionID: ctx.sessionID,
+    history: [...session.history],
+    localRows: [],
+    sessionTitle: ctx.sessionTitle,
+    agent: ctx.agent,
+  }
+  const ensureSession = () => {
+    if (!input.resolveSession || state.sessionID) {
+      return Promise.resolve()
+    }
+
+    if (state.session) {
+      return state.session
+    }
+
+    state.session = input.resolveSession(ctx).then((next) => {
+      state.sessionID = next.sessionID
+      state.sessionTitle = next.sessionTitle ?? state.sessionTitle
+      state.agent = next.agent
+    })
+    return state.session
+  }
+
+  const shell = await (deps.createRuntimeLifecycle ?? createRuntimeLifecycle)({
+    directory: ctx.directory,
+    findFiles: (query) =>
+      ctx.sdk.find
+        .files({ query, directory: ctx.directory })
+        .then((x) => x.data ?? [])
+        .catch(() => []),
+    agents: [],
+    resources: [],
+    sessionID: state.sessionID,
+    sessionTitle: state.sessionTitle,
+    getSessionID: () => state.sessionID,
+    first: session.first,
+    history: session.history,
+    agent: state.agent,
+    model: state.model,
+    variant: state.activeVariant,
+    tuiConfig,
+    backgroundSubagents: input.backgroundSubagents,
+    onPermissionReply: async (next) => {
+      if (state.demo?.permission(next)) {
+        return
+      }
+
+      log?.write("send.permission.reply", next)
+      await ctx.sdk.permission.reply(next)
+    },
+    onQuestionReply: async (next) => {
+      if (state.demo?.questionReply(next)) {
+        return
+      }
+
+      await ctx.sdk.question.reply(next)
+    },
+    onQuestionReject: async (next) => {
+      if (state.demo?.questionReject(next)) {
+        return
+      }
+
+      await ctx.sdk.question.reject(next)
+    },
+    onCycleVariant: () => {
+      if (!state.model || state.variants.length === 0) {
+        return {
+          status: "no variants available",
+        }
+      }
+
+      state.activeVariant = cycleVariant(state.activeVariant, state.variants)
+      saveVariant(state.model, state.activeVariant)
+      return {
+        status: state.activeVariant ? `variant ${state.activeVariant}` : "variant default",
+        modelLabel: formatModelLabel(state.model, state.activeVariant, state.providers),
+        variant: state.activeVariant,
+      }
+    },
+    onModelSelect: async (model) => {
+      if (state.model?.providerID === model.providerID && state.model.modelID === model.modelID) {
+        return
+      }
+
+      state.model = model
+      state.activeVariant = undefined
+      state.variants = variantsFor(state.providers, model)
+      const switching = resolveSavedVariant(model).then((saved) => {
+        const current = state.model
+        if (!current || current.providerID !== model.providerID || current.modelID !== model.modelID) {
+          return
+        }
+
+        state.activeVariant = resolveVariant(ctx.variant, undefined, saved, state.variants)
+      })
+      state.switching = switching
+      await switching
+      if (state.switching === switching) {
+        state.switching = undefined
+      }
+
+      const current = state.model
+      if (!current || current.providerID !== model.providerID || current.modelID !== model.modelID) {
+        return
+      }
+
+      return {
+        modelLabel: formatModelLabel(model, state.activeVariant, state.providers),
+        status: `model ${model.modelID}`,
+        variant: state.activeVariant,
+        variants: state.variants,
+      }
+    },
+    onVariantSelect: async (variant) => {
+      if (!state.model || state.variants.length === 0) {
+        return {
+          status: "no variants available",
+        }
+      }
+
+      if (variant && !state.variants.includes(variant)) {
+        return {
+          status: `variant ${variant} unavailable`,
+        }
+      }
+
+      state.activeVariant = variant
+      saveVariant(state.model, state.activeVariant)
+      return {
+        status: state.activeVariant ? `variant ${state.activeVariant}` : "variant default",
+        modelLabel: formatModelLabel(state.model, state.activeVariant, state.providers),
+        variant: state.activeVariant,
+        variants: state.variants,
+      }
+    },
+    onInterrupt: () => {
+      if (!hasSession(input, state) || state.aborting) {
+        return
+      }
+
+      state.aborting = true
+      void ctx.sdk.session
+        .abort({
+          sessionID: state.sessionID,
+        })
+        .catch(() => {})
+        .finally(() => {
+          state.aborting = false
+        })
+    },
+    onBackground: () => {
+      if (!hasSession(input, state)) return
+      void ctx.sdk.experimental.session.background({ sessionID: state.sessionID }).catch(() => {})
+    },
+    onSubagentSelect: (sessionID) => {
+      state.selectSubagent?.(sessionID)
+      log?.write("subagent.select", {
+        sessionID,
+      })
+    },
+  })
+  const footer = shell.footer
+  const rememberLocal = (commit: StreamCommit, after?: LocalReplayAnchor) => {
+    state.localRows = [...state.localRows, { commit, after }].slice(-LOCAL_REPLAY_ROW_LIMIT)
+  }
+
+  const loadCatalog = async (): Promise<void> => {
+    if (footer.isClosed) {
+      return
+    }
+
+    const [agents, resources, commands] = await Promise.all([
+      ctx.sdk.app
+        .agents({ directory: ctx.directory })
+        .then((x) => x.data ?? [])
+        .catch(() => []),
+      ctx.sdk.experimental.resource
+        .list({ directory: ctx.directory })
+        .then((x) => Object.values(x.data ?? {}))
+        .catch(() => []),
+      ctx.sdk.command
+        .list({ directory: ctx.directory })
+        .then((x) => x.data ?? [])
+        .catch(() => []),
+    ])
+    if (footer.isClosed) {
+      return
+    }
+
+    footer.event({
+      type: "catalog",
+      agents,
+      resources,
+      commands,
+    })
+  }
+
+  void footer
+    .idle()
+    .then(loadCatalog)
+    .catch(() => {})
+
+  if (Flag.OPENCODE_SHOW_TTFD) {
+    footer.append({
+      kind: "system",
+      text: `startup ${Math.max(0, Math.round(performance.now() - start))}ms`,
+      phase: "final",
+      source: "system",
+    })
+  }
+
+  if (input.demo) {
+    await ensureSession()
+    state.demo = createRunDemo({
+      footer,
+      sessionID: state.sessionID,
+      thinking: input.thinking,
+      limits: () => state.limits,
+    })
+  }
+
+  if (input.afterPaint) {
+    void Promise.resolve(input.afterPaint(ctx)).catch(() => {})
+  }
+
+  void modelTask.then((info) => {
+    state.providers = info.providers
+    state.variants = variantsFor(state.providers, state.model)
+    state.limits = info.limits
+
+    const next = resolveVariant(ctx.variant, session.variant, savedVariant, state.variants)
+    if (next !== state.activeVariant) {
+      state.activeVariant = next
+    }
+
+    if (footer.isClosed) {
+      return
+    }
+
+    footer.event({ type: "models", providers: info.providers })
+    footer.event({ type: "variants", variants: state.variants, current: state.activeVariant })
+    if (!state.model) {
+      return
+    }
+
+    footer.event({
+      type: "model",
+      model: formatModelLabel(state.model, state.activeVariant, state.providers),
+    })
+  })
+
+  const streamTask = deps.streamTransport ?? import("./stream.transport")
+  const ensureStream = () => {
+    if (state.stream) {
+      return state.stream
+    }
+
+    // Share eager prewarm and first-turn boot through one in-flight promise,
+    // but clear it if transport creation fails so a later prompt can retry.
+    const next = (async () => {
+      await ensureSession()
+      if (footer.isClosed) {
+        throw new Error("runtime closed")
+      }
+
+      const mod = await streamTask
+      if (footer.isClosed) {
+        throw new Error("runtime closed")
+      }
+
+      const handle = await mod.createSessionTransport({
+        sdk: ctx.sdk,
+        directory: ctx.directory,
+        sessionID: state.sessionID,
+        thinking: input.thinking,
+        replay: input.replay,
+        replayLimit: input.replayLimit,
+        limits: () => state.limits,
+        providers: () => state.providers,
+        footer,
+        trace: log,
+      })
+      if (footer.isClosed) {
+        await handle.close()
+        throw new Error("runtime closed")
+      }
+
+      state.selectSubagent = (sessionID) => handle.selectSubagent(sessionID)
+      return { mod, handle }
+    })()
+    state.stream = next
+    void next.catch(() => {
+      if (state.stream === next) {
+        state.stream = undefined
+      }
+    })
+    return next
+  }
+
+  let resizeTimer: ReturnType<typeof setTimeout> | undefined
+  const offResize = shell.onResize(() => {
+    if (resizeTimer) {
+      clearTimeout(resizeTimer)
+    }
+
+    resizeTimer = setTimeout(() => {
+      resizeTimer = undefined
+      if (footer.isClosed) {
+        return
+      }
+
+      shell.refreshTheme()
+      if (!input.replay || !state.stream) {
+        return
+      }
+
+      void state.stream
+        .then((item) =>
+          item.handle.replayOnResize({
+            localRows: () => state.localRows,
+            reset: () =>
+              shell.resetForReplay({
+                sessionTitle: state.sessionTitle,
+                sessionID: state.sessionID,
+                history: state.history,
+              }),
+          }),
+        )
+        .catch(() => {})
+    }, RESIZE_DELAY)
+  })
+
+  const runQueue = async () => {
+    let includeFiles = true
+    if (state.demo) {
+      await state.demo.start()
+    }
+
+    const mod = await import("./runtime.queue")
+    const createSession = input.createSession
+    await mod.runPromptQueue({
+      footer,
+      initialInput: input.initialInput,
+      trace: log,
+      onSend: (prompt) => {
+        state.shown = true
+        state.history.push(prompt)
+        if (prompt.mode !== "shell") {
+          rememberLocal({
+            kind: "user",
+            text: prompt.text,
+            phase: "start",
+            source: "system",
+            messageID: prompt.messageID,
+          })
+        }
+      },
+      onNewSession: createSession
+        ? async () => {
+            try {
+              await state.switching?.catch(() => {})
+              const created = await createSession(ctx, {
+                agent: state.agent,
+                model: state.model,
+                variant: state.activeVariant,
+              })
+              await footer.idle().catch(() => {})
+              await state.stream?.then((item) => item.handle.close()).catch(() => {})
+              state.stream = undefined
+              state.session = undefined
+              state.selectSubagent = undefined
+              state.shown = false
+              state.sessionID = created.sessionID
+              state.sessionTitle = created.sessionTitle
+              state.agent = created.agent ?? state.agent
+              state.history = []
+              state.localRows = []
+              includeFiles = true
+              state.demo = input.demo
+                ? createRunDemo({
+                    footer,
+                    sessionID: state.sessionID,
+                    thinking: input.thinking,
+                    limits: () => state.limits,
+                  })
+                : undefined
+              log?.write("session.new", {
+                sessionID: state.sessionID,
+              })
+              footer.event({
+                type: "stream.subagent",
+                state: {
+                  tabs: [],
+                  details: {},
+                  permissions: [],
+                  questions: [],
+                },
+              })
+              footer.event({ type: "stream.view", view: { type: "prompt" } })
+              footer.event({
+                type: "stream.patch",
+                patch: {
+                  phase: "idle",
+                  duration: "",
+                  usage: "",
+                  first: true,
+                },
+              })
+              footer.append({
+                kind: "system",
+                text: `new session ${state.sessionID}`,
+                phase: "final",
+                source: "system",
+              })
+              await state.demo?.start()
+            } catch (error) {
+              footer.event({
+                type: "stream.patch",
+                patch: {
+                  phase: "idle",
+                  status: "failed to start new session",
+                },
+              })
+              const commit = {
+                kind: "error",
+                text: error instanceof Error ? error.message : String(error),
+                phase: "start",
+                source: "system",
+                messageID: MessageID.ascending(),
+              } as const
+              rememberLocal(commit)
+              footer.append(commit)
+            }
+          }
+        : undefined,
+      run: async (prompt, signal) => {
+        if (state.demo && (await state.demo.prompt(prompt, signal))) {
+          return
+        }
+
+        await state.switching?.catch(() => {})
+
+        let outputAnchor: LocalReplayAnchor | undefined
+        try {
+          const next = await ensureStream()
+          await next.handle.runPromptTurn({
+            agent: state.agent,
+            model: state.model,
+            variant: state.activeVariant,
+            prompt,
+            files: input.files,
+            includeFiles,
+            onVisibleOutput: (anchor) => {
+              outputAnchor = anchor
+            },
+            signal,
+          })
+          if (prompt.messageID) {
+            state.localRows = state.localRows.filter(
+              (row) => row.commit.kind !== "user" || row.commit.messageID !== prompt.messageID,
+            )
+          }
+          includeFiles = false
+        } catch (error) {
+          if (signal.aborted || footer.isClosed) {
+            return
+          }
+
+          const text =
+            (await state.stream?.then((item) => item.mod).catch(() => undefined))?.formatUnknownError(error) ??
+            (error instanceof Error ? error.message : String(error))
+          const commit = {
+            kind: "error",
+            text,
+            phase: "start",
+            source: "system",
+            messageID: prompt.messageID,
+          } as const
+          rememberLocal(commit, outputAnchor)
+          footer.append(commit)
+        }
+      },
+    })
+  }
+
+  try {
+    const eager = eagerStream(input, ctx)
+    if (eager) {
+      if (input.replay && state.shown) {
+        // Replay commits immutable scrollback rows, so wait for provider names
+        // before bootstrapping existing session history.
+        await modelTask
+      }
+
+      await ensureStream()
+    }
+
+    if (!eager && input.resolveSession) {
+      queueMicrotask(() => {
+        if (footer.isClosed) {
+          return
+        }
+
+        void ensureStream().catch(() => {})
+      })
+    }
+
+    try {
+      await runQueue()
+    } finally {
+      if (resizeTimer) {
+        clearTimeout(resizeTimer)
+      }
+      offResize()
+      await state.stream?.then((item) => item.handle.close()).catch(() => {})
+    }
+  } finally {
+    const title = await resolveExitTitle(ctx, input, state)
+
+    await shell.close({
+      showExit: state.shown && hasSession(input, state),
+      sessionTitle: title,
+      sessionID: state.sessionID,
+      history: state.history,
+    })
+  }
+}
+
+// Local in-process mode. Creates an SDK client backed by a direct fetch to
+// the in-process server, so no external HTTP server is needed.
+export async function runInteractiveLocalMode(input: RunLocalInput): Promise<void> {
+  const sdk = createRimuruClient({
+    baseUrl: "http://rimuru.internal",
+    fetch: input.fetch,
+    directory: input.directory,
+  })
+  let session: Promise<ResolvedSession> | undefined
+
+  return runInteractiveRuntime({
+    files: input.files,
+    initialInput: input.initialInput,
+    thinking: input.thinking,
+    backgroundSubagents: input.backgroundSubagents,
+    replay: input.replay,
+    replayLimit: input.replayLimit,
+    demo: input.demo,
+    resolveSession: () => {
+      if (session) {
+        return session
+      }
+
+      session = Promise.all([input.resolveAgent(), input.session(sdk)]).then(([agent, next]) => {
+        if (!next?.id) {
+          throw new Error("Session not found")
+        }
+
+        void input.share(sdk, next.id).catch(() => {})
+        return {
+          sessionID: next.id,
+          sessionTitle: next.title,
+          agent,
+        }
+      })
+      return session
+    },
+    createSession: createSessionResolver(input.createSession),
+    boot: async () => {
+      return {
+        sdk,
+        directory: input.directory,
+        sessionID: "",
+        sessionTitle: undefined,
+        resume: false,
+        agent: input.agent,
+        model: input.model,
+        variant: input.variant,
+      }
+    },
+  })
+}
+
+// Attach mode. Uses the caller-provided SDK client directly.
+export async function runInteractiveMode(
+  input: RunInput & { createSession?: CreateSession },
+  deps?: RunRuntimeDeps,
+): Promise<void> {
+  return runInteractiveRuntime(
+    {
+      files: input.files,
+      initialInput: input.initialInput,
+      thinking: input.thinking,
+      backgroundSubagents: input.backgroundSubagents,
+      replay: input.replay,
+      replayLimit: input.replayLimit,
+      demo: input.demo,
+      boot: async () => ({
+        sdk: input.sdk,
+        directory: input.directory,
+        sessionID: input.sessionID,
+        sessionTitle: input.sessionTitle,
+        resume: input.resume,
+        agent: input.agent,
+        model: input.model,
+        variant: input.variant,
+      }),
+      createSession: createSessionResolver(input.createSession),
+    },
+    deps,
+  )
+}