@uniqbit/mate-core 0.18.3-canary.0 → 0.18.3

package/src/lib/orchestrator/session-prepare.ts

@@ -0,0 +1,425 @@
+import { getActiveDistribution } from "../../distribution";
+import { FRAMEWORK_NAME } from "../../framework";
+import { recordCompanionSyncUnfinished } from "../../runtime/companion-git-state";
+import { withCompanionLock, type CompanionLockOptions } from "../../runtime/companion-lock";
+import {
+  launchRecoveryStalenessLines,
+  syncCompanionUnattended,
+  type UnattendedSyncOutcome,
+} from "../../runtime/companion-sync";
+import type { WrapState } from "../../runtime/wrap-state";
+import { syncCompanionFiles } from "../../tools/setup";
+import type { CapabilityPlugin, LaunchPreflightContext } from "../../tools/setup/plugin";
+import { attendedLockOptions, syncCompanionGit } from "./companion-git-sync";
+import { readCompanionConfig } from "./config-store";
+import { entryFailureMessage, firstFailure, type ProjectionChangedPaths } from "./projection-types";
+import { LaunchPreflightError, type FrameworkConfig, type LinkedRepository } from "./types";
+import { project, projectWorkingRepositoryBestEffort } from "./working-repo-projection";
+import { runWrapPass } from "./wrap-pass";
+
+/**
+ * Session Prepare: Git sync, file sync, one projection pass, preflight. Never
+ * spawns a runtime.
+ * - `attended`: managed launch; waits for the lock, stops at first failure.
+ * - `local`: plugin's awaited phase; no remote, only tries the lock.
+ * - `sync`: plugin's detached fast-forward, then `local` if the companion moved.
+ */
+
+export type PrepareMode = "attended" | "local" | "sync";
+export type PrepareRuntime = "claude" | "opencode";
+
+export type PrepareStepName =
+  | "lock"
+  | "git-sync"
+  | "file-sync"
+  | "launch-projection"
+  | "working-target"
+  | "preflight";
+
+export interface PrepareStep {
+  step: PrepareStepName;
+  outcome: "ok" | "skipped" | "warning" | "failed";
+  detail?: string;
+  /** The runtimes a `working-target` or `preflight` step answered for. */
+  runtimes?: PrepareRuntime[];
+}
+
+export interface PrepareResult {
+  /** False when a step failed; warnings alone keep it true. */
+  ok: boolean;
+  mode: PrepareMode;
+  runtime: PrepareRuntime;
+  /** Every runtime this prepare reconciled and preflighted. */
+  runtimes: PrepareRuntime[];
+  repositoryId?: string;
+  companionPath: string;
+  steps: PrepareStep[];
+  warnings: string[];
+  /** The sync fast-forwarded the companion. */
+  companionChanged?: true;
+  /** The first failed step's own error, for an attended caller to raise as it was. */
+  failure?: unknown;
+  /** The wrap state the `working-target` step's pass left; a launch reads no manifest itself. */
+  wrapState?: WrapState;
+}
+
+export interface PrepareSessionInput {
+  runtime: PrepareRuntime;
+  /**
+   * `local`/`sync`: the runtimes the repository is wrapped for, reconciled in
+   * one wrap pass and preflighted each; `[runtime]` when absent.
+   */
+  runtimes?: readonly PrepareRuntime[];
+  companionPath: string;
+  /** Absent for a companion-scope launch: nothing is projected. */
+  repository?: LinkedRepository;
+  mode: PrepareMode;
+  /** The companion config the caller already holds; read once here otherwise. */
+  config?: FrameworkConfig;
+  /** `--no-git`: an attended prepare skips only the Git sync. */
+  skipGit?: boolean;
+}
+
+/** The detached sync may queue behind an operator answering a merge conflict. */
+const SYNC_LOCK_TIMEOUT_MS = 5 * 60_000;
+
+/**
+ * Every enabled Capability's preflight for one runtime, run together. Throws a
+ * {@link LaunchPreflightError} listing what failed.
+ */
+async function runCapabilityPreflight(context: LaunchPreflightContext): Promise<void> {
+  const { config, providerId } = context;
+  const preflights = getActiveDistribution()
+    .registry.getAll()
+    .flatMap((plugin) => {
+      if (plugin.kind !== "capability" || !plugin.isEnabled(config)) return [];
+      const capability = plugin as CapabilityPlugin;
+      const preflight = capability.forProvider?.[providerId]?.preflight;
+      return preflight ? [{ capability, preflight }] : [];
+    });
+  const results = await Promise.all(
+    preflights.map(async ({ capability, preflight }) => {
+      try {
+        return await preflight(context);
+      } catch (error) {
+        throw new LaunchPreflightError(
+          `Capability preflight failed for ${capability.id} on ${providerId}: ${message(error)}`,
+          { cause: error },
+        );
+      }
+    }),
+  );
+  const diagnostics = results.flat();
+
+  if (diagnostics.length > 0) {
+    throw new LaunchPreflightError(
+      ["Capability launch preflight failed.", ...diagnostics.map((item) => `- ${item}`)].join("\n"),
+    );
+  }
+}
+
+export const sessionPrepareDeps = {
+  readCompanionConfig,
+  syncCompanionGit,
+  syncCompanionUnattended,
+  recordCompanionSyncUnfinished,
+  syncCompanionFiles,
+  projectLaunch: (input: Parameters<typeof project>[1]) => project("launch", input),
+  /**
+   * The Projection Root a Claude launch guarantees for itself. Companion
+   * resolution also writes it, but only on the branch that resolved a companion
+   * from the Working Repository; asking here makes the guarantee independent of
+   * which branch answered. Best-effort — a launch carries its own configuration.
+   */
+  refreshProjectionRoot: projectWorkingRepositoryBestEffort,
+  runWrapPass,
+  runCapabilityPreflight,
+  withCompanionLock,
+  /** As the launch decides it: Git may prompt only with a terminal on both ends. */
+  isInteractiveTerminal: () => !!(process.stdin.isTTY && process.stdout.isTTY),
+};
+
+function message(error: unknown): string {
+  return error instanceof Error ? error.message : String(error);
+}
+
+/** One prepare's running record; `failed` stops an attended prepare. */
+class StepLog {
+  readonly steps: PrepareStep[] = [];
+  readonly warnings: string[] = [];
+  failure: unknown;
+  wrapState: WrapState | undefined;
+  stopped = false;
+  companionChanged = false;
+
+  constructor(private readonly attended: boolean) {}
+
+  push(step: PrepareStep): void {
+    this.steps.push(step);
+  }
+
+  fail(
+    step: PrepareStepName,
+    error: unknown,
+    overrides: Pick<PrepareStep, "detail" | "runtimes"> = {},
+  ): void {
+    this.steps.push({ step, outcome: "failed", detail: message(error), ...overrides });
+    if (this.failure === undefined) this.failure = error;
+    if (this.attended) this.stopped = true;
+  }
+}
+
+async function runFileSync(
+  input: PrepareSessionInput,
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  const deps = sessionPrepareDeps;
+  try {
+    if (input.repository) {
+      await deps.syncCompanionFiles(input.companionPath, config, input.repository.path);
+    } else {
+      await deps.syncCompanionFiles(input.companionPath, config);
+    }
+    run.push({ step: "file-sync", outcome: "ok" });
+  } catch (error) {
+    run.fail("file-sync", error);
+  }
+}
+
+function wrapDetail(wrapped: {
+  writtenDocuments: string[];
+  changed: ProjectionChangedPaths;
+}): string {
+  const wrote = [...wrapped.writtenDocuments, ...wrapped.changed.written];
+  const withdrew = wrapped.changed.removed;
+  const parts: string[] = [];
+  if (wrote.length > 0) parts.push(`wrote ${wrote.join(", ")}`);
+  if (withdrew.length > 0) parts.push(`withdrew ${withdrew.join(", ")}`);
+  return parts.length > 0 ? `updated (${parts.join("; ")})` : "updated";
+}
+
+function runtimesOf(input: PrepareSessionInput): PrepareRuntime[] {
+  return input.mode !== "attended" && input.runtimes?.length
+    ? [...input.runtimes]
+    : [input.runtime];
+}
+
+/**
+ * One wrap pass for `runtimes`, OpenCode's copies included. Every managed
+ * launch is this pass followed by the agent, for either runtime.
+ */
+async function reconcileWrap(
+  input: PrepareSessionInput,
+  repository: LinkedRepository,
+  runtimes: PrepareRuntime[],
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  const wrapped = await sessionPrepareDeps.runWrapPass(input.companionPath, repository, {
+    runtimes,
+    config,
+  });
+  if (wrapped.kind === "failed") {
+    run.fail("working-target", new LaunchPreflightError(wrapped.message), { runtimes });
+    return;
+  }
+  run.warnings.push(...wrapped.warnings);
+  run.wrapState = wrapped.wrapState;
+  run.push({
+    step: "working-target",
+    outcome: "ok",
+    detail: wrapped.kind === "written" ? wrapDetail(wrapped) : "current",
+    runtimes,
+  });
+}
+
+/**
+ * Claude: the launch scope, then the Projection Root. A non-degradable entry is
+ * a guardrail the session was promised, so it fails the prepare; a degradable
+ * one warns and the prepare proceeds.
+ */
+async function projectClaude(
+  input: PrepareSessionInput,
+  repository: LinkedRepository,
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  const deps = sessionPrepareDeps;
+  const result = await deps.projectLaunch({
+    repoPath: repository.path,
+    companionPath: input.companionPath,
+    config,
+  });
+  for (const outcome of result.outcomes) {
+    if (outcome.state === "failed" && outcome.degradable) {
+      run.warnings.push(entryFailureMessage(outcome, repository.id));
+    }
+  }
+  const failure = firstFailure(result.outcomes);
+  if (failure) {
+    const detail = entryFailureMessage(failure, repository.id);
+    run.fail("launch-projection", new LaunchPreflightError(`${FRAMEWORK_NAME}: ${detail}`), {
+      detail,
+    });
+    return;
+  }
+  run.push({ step: "launch-projection", outcome: "ok" });
+  await deps.refreshProjectionRoot(input.companionPath, repository);
+  /** A launch is a wrap: created when missing, reconciled when stale, so the launch can defer to it. */
+  await reconcileWrap(input, repository, ["claude"], config, run);
+}
+
+/** Attended: the launch's own runtime. Otherwise one wrap pass for every wrapped runtime. */
+async function projectRepository(
+  input: PrepareSessionInput,
+  repository: LinkedRepository,
+  runtimes: PrepareRuntime[],
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  if (input.mode === "attended" && input.runtime === "claude") {
+    await projectClaude(input, repository, config, run);
+    return;
+  }
+  await reconcileWrap(input, repository, runtimes, config, run);
+}
+
+/** File sync, one projection pass, and the capability preflight per runtime. */
+async function runLocal(
+  input: PrepareSessionInput,
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  await runFileSync(input, config, run);
+  if (run.stopped) return;
+
+  const runtimes = runtimesOf(input);
+  if (input.repository) {
+    await projectRepository(input, input.repository, runtimes, config, run);
+    if (run.stopped) return;
+  }
+
+  /** Attended prepares one runtime, so running these together never skips a stop. */
+  const outcomes = await Promise.allSettled(
+    runtimes.map((runtime) =>
+      sessionPrepareDeps.runCapabilityPreflight({
+        companionPath: input.companionPath,
+        config,
+        repository: input.repository,
+        providerId: runtime,
+      }),
+    ),
+  );
+  outcomes.forEach((outcome, index) => {
+    const answered = [runtimes[index]!];
+    if (outcome.status === "fulfilled") {
+      run.push({ step: "preflight", outcome: "ok", runtimes: answered });
+    } else {
+      run.fail("preflight", outcome.reason, { runtimes: answered });
+    }
+  });
+}
+
+async function runAttended(
+  input: PrepareSessionInput,
+  config: FrameworkConfig,
+  run: StepLog,
+): Promise<void> {
+  const deps = sessionPrepareDeps;
+  if (input.skipGit) {
+    run.push({ step: "git-sync", outcome: "skipped", detail: "--no-git" });
+  } else if (config.git !== "auto") {
+    run.push({ step: "git-sync", outcome: "skipped", detail: "companion Git mode is not auto" });
+  } else {
+    try {
+      const synced = await deps.syncCompanionGit(
+        input.companionPath,
+        input.repository?.path,
+        deps.isInteractiveTerminal(),
+      );
+      run.push({ step: "git-sync", outcome: "ok" });
+      /** A pull may have changed the companion's own selection. */
+      if (synced?.changed) config = await deps.readCompanionConfig(input.companionPath);
+    } catch (error) {
+      run.fail("git-sync", error);
+      return;
+    }
+  }
+  await runLocal(input, config, run);
+}
+
+/** The sync's own config read: only a fast-forward can have changed it. */
+async function runSync(input: PrepareSessionInput, run: StepLog): Promise<void> {
+  const deps = sessionPrepareDeps;
+  let outcome: UnattendedSyncOutcome;
+  try {
+    outcome = await deps.syncCompanionUnattended(input.companionPath, { ttlMs: 0 });
+  } catch (error) {
+    const reason = message(error);
+    /** Persisted like any unfinished sync, so the TUI's session lines carry it. */
+    deps.recordCompanionSyncUnfinished(input.companionPath, reason);
+    outcome = { status: "unfinished", changed: false, companionPath: input.companionPath, reason };
+  }
+
+  if (outcome.status === "unfinished") {
+    run.push({ step: "git-sync", outcome: "warning", detail: outcome.reason });
+    run.warnings.push(...launchRecoveryStalenessLines(outcome, runtimesOf(input)));
+    return;
+  }
+  if (outcome.status !== "complete") {
+    run.push({ step: "git-sync", outcome: "skipped", detail: outcome.status });
+    return;
+  }
+  if (!outcome.changed) {
+    run.push({ step: "git-sync", outcome: "ok", detail: "current" });
+    return;
+  }
+  run.push({ step: "git-sync", outcome: "ok", detail: "fast-forwarded" });
+  run.companionChanged = true;
+  await runLocal(input, await deps.readCompanionConfig(input.companionPath), run);
+}
+
+function lockOptions(input: PrepareSessionInput): CompanionLockOptions {
+  if (input.mode === "attended") return attendedLockOptions(input.companionPath);
+  if (input.mode === "local") return { wait: false };
+  return { timeoutMs: SYNC_LOCK_TIMEOUT_MS };
+}
+
+export async function prepareSession(input: PrepareSessionInput): Promise<PrepareResult> {
+  const deps = sessionPrepareDeps;
+  const run = new StepLog(input.mode === "attended");
+  const result = (): PrepareResult => ({
+    ok: !run.steps.some((step) => step.outcome === "failed"),
+    mode: input.mode,
+    runtime: input.runtime,
+    runtimes: runtimesOf(input),
+    ...(input.repository ? { repositoryId: input.repository.id } : {}),
+    companionPath: input.companionPath,
+    steps: run.steps,
+    warnings: run.warnings,
+    ...(run.companionChanged ? { companionChanged: true as const } : {}),
+    ...(run.failure === undefined ? {} : { failure: run.failure }),
+    ...(run.wrapState ? { wrapState: run.wrapState } : {}),
+  });
+
+  const locked = await deps.withCompanionLock(input.companionPath, lockOptions(input), async () => {
+    if (input.mode === "sync") return runSync(input, run);
+    const config = input.config ?? (await deps.readCompanionConfig(input.companionPath));
+    return input.mode === "attended"
+      ? runAttended(input, config, run)
+      : runLocal(input, config, run);
+  });
+
+  if (!locked.acquired) {
+    if (input.mode === "local") {
+      /** Another prepare is on it; the projection already in place serves the session. */
+      run.push({ step: "lock", outcome: "skipped", detail: "another prepare holds the companion" });
+    } else {
+      const detail = "another companion sync did not finish in time; skipped";
+      run.push({ step: "lock", outcome: "warning", detail });
+      run.warnings.push(detail);
+    }
+  }
+  return result();
+}

package/src/lib/orchestrator/setup-compatibilities.ts

@@ -173,7 +173,7 @@ export const BUILTIN_SETUP_COMPATIBILITIES: SetupCompatibility[] = [
     kind: "capability",
     capability: { name: "react-doctor" },
     label: "React Doctor",
-    description: "Install React Doctor skills and Claude post-edit scans.",
+    description: "Install the mate-react-doctor skill that verifies React changes during apply.",
     defaultSelected: true,
     templateRoot: "capabilities/react-doctor",
   },

package/src/lib/orchestrator/setup-preflight.ts

@@ -6,7 +6,7 @@ import { parse } from "yaml";
 import { FRAMEWORK_NAME } from "../../framework";
 import { fileExists } from "../fs-utils";
 import { GlobalConfigStore } from "./global-config-store";
-import { CompanionResolver, type CompanionMatch } from "./companion-resolver";
+import { CompanionResolver, type CompanionMatch } from "./repository-link";
 
 export type SetupPreflightResult =
   | { kind: "existing-companion" }
@@ -104,6 +104,7 @@ export async function looksLikeWorkingRepo(cwd: string): Promise<boolean> {
   const resolvedCwd = path.resolve(cwd);
 
   for (const marker of STRONG_FILE_MARKERS) {
+    // oxlint-disable-next-line no-await-in-loop -- one strong marker settles it; stop probing
     if (await fileExists(path.join(resolvedCwd, marker))) {
       return true;
     }
@@ -115,6 +116,7 @@ export async function looksLikeWorkingRepo(cwd: string): Promise<boolean> {
 
   let weakCount = 0;
   for (const marker of WEAK_FILE_MARKERS) {
+    // oxlint-disable-next-line no-await-in-loop -- stops probing at the second weak marker
     if (await fileExists(path.join(resolvedCwd, marker))) {
       weakCount++;
       if (weakCount >= 2) return true;

package/src/lib/orchestrator/types.ts

@@ -127,9 +127,12 @@ export interface StudioTerminalConfig {
 
 /**
  * A named slice of a companion's configuration, activated by `MATE_AUDIENCE`.
- * Plugins add to the base list; `studio.terminal.agent` overrides the base.
+ * Plugins and capabilities add to the base lists; `studio.terminal.agent`
+ * overrides the base. Capabilities here name what a declared plugin provides,
+ * never a built-in one, so what `install` writes does not depend on the audience.
  */
 export interface AudienceConfig {
+  capabilities?: CapabilityConfig[];
   plugins?: PluginDeclaration[];
   studio?: { terminal?: StudioTerminalConfig };
 }