@fitlab-ai/agent-infra 0.9.1 → 0.9.3

package/dist/lib/sandbox/commands/create.js

@@ -6,14 +6,15 @@ import { execFileSync } from 'node:child_process';
 import { parseArgs } from 'node:util';
 import * as p from '@clack/prompts';
 import pc from 'picocolors';
-import * as toml from 'smol-toml';
+import { listAgentClientAdapters } from '../../agent-clients/registry.js';
 import { loadConfig } from '../config.js';
-import { assertValidBranchName, containerName, containerNameCandidates, parsePositiveIntegerOption, sandboxBranchLabel, sandboxImageConfigLabel, sandboxImageRefreshLabel, sandboxLabel, shareBranchDir, shareCommonDir, shellConfigDir, worktreeDirCandidates } from '../constants.js';
+import { assertValidBranchName, containerName, containerNameCandidates, parsePositiveIntegerOption, sandboxBranchLabel, sandboxImageConfigLabel, sandboxImageRefreshLabel, sandboxLabel, sandboxRuntimeCapabilityLabel, shareBranchDir, shareCommonDir, shellConfigDir, worktreeDirCandidates } from '../constants.js';
 import { prepareDockerfile } from '../dockerfile.js';
 import { assertBuildProxyCompatibility, buildProxyFailureHint, prepareBuildProxy, redactBuildProxyValues } from '../build-proxy.js';
 import { detectEngine, ensureDocker } from '../engine.js';
 import { commandForEngine, execEngine, run, runEngine, runOk, runOkEngine, runSafe, runSafeEngine, runVerboseEngine } from '../shell.js';
 import { resolveTaskBranch } from '../task-resolver.js';
+import { createSandboxCapabilityPlan, runSandboxHooks, runBoundedSandboxHookCommand } from '../agent-client-reconciler.js';
 import { resolveTools, tmpfsSeedStagingPath, tmpfsSeedTargetPath, toolConfigDirCandidates } from '../tools.js';
 import { assertFreshSandboxReady, hydrateTmpfsSeedEntries, prepareTmpfsMounts } from '../recovery.js';
 import { hostJoin, toEnginePath, volumeArg } from '../engines/wsl2-paths.js';
@@ -22,32 +23,13 @@ import { clipboardHostDir, CONTAINER_CLIPBOARD_MOUNT } from '../clipboard/paths.
 import { validateSelinuxDisableEnv } from '../engines/selinux.js';
 import { dotfilesCacheDir, materializeDotfiles } from '../dotfiles.js';
 import { ensureSandboxDiscoveryReadmes } from '../readme-scaffold.js';
+import { removeDirRecursive } from '../../remove-dir.js';
 import { prepareClaudeCredentials, redactCommandError, validateClaudeCredentialsEnvOverride } from '../credentials.js';
 import { detectHostTimezone } from '../host-timezone.js';
 import { buildImageSignature, buildSandboxImageArgs, isRefreshDisabled, isRefreshDue, parseImageLabels, parseRefreshTimestamp } from '../image-build.js';
-const OPENCODE_YOLO_PERMISSION = '{"*":"allow","read":"allow","bash":"allow","edit":"allow","webfetch":"allow","external_directory":"allow","doom_loop":"allow"}';
+export { ensureCodexModelInheritance, ensureCodexWorkspaceTrust } from '../../agent-clients/adapters/codex-sandbox.js';
 const SANDBOX_ALIAS_BLOCK_BEGIN = '# >>> agent-infra managed aliases >>>';
 const SANDBOX_ALIAS_BLOCK_END = '# <<< agent-infra managed aliases <<<';
-const SANDBOX_ALIAS_NAMES = [
-    'claude-yolo',
-    'opencode-yolo',
-    'codex-yolo',
-    'gemini-yolo',
-    'cy',
-    'oy',
-    'xy',
-    'gy'
-];
-const DEFAULT_SANDBOX_ALIASES = `alias claude-yolo='claude --dangerously-skip-permissions; tput ed'
-alias opencode-yolo='OPENCODE_PERMISSION='\\''${OPENCODE_YOLO_PERMISSION}'\\'' opencode; tput ed'
-alias codex-yolo='codex --yolo; tput ed'
-alias gemini-yolo='gemini --yolo; tput ed'
-
-alias cy='claude --dangerously-skip-permissions; tput ed'
-alias oy='OPENCODE_PERMISSION='\\''${OPENCODE_YOLO_PERMISSION}'\\'' opencode; tput ed'
-alias xy='codex --yolo; tput ed'
-alias gy='gemini --yolo; tput ed'
-`;
 const CONTAINER_HOME = '/home/devuser';
 const CONTAINER_SHELL_CONFIG_MOUNT = `${CONTAINER_HOME}/.host-shell-config`;
 const USAGE = `Usage: ai sandbox create <branch> [base] [--cpu <n>] [--memory <n>] [--no-refresh] [--inherit-proxy|-P] [--inherit-build-proxy|-B]
@@ -268,7 +250,7 @@ export function ensureShellConfigSymlinks(engine, container, execFn = execEngine
 }
 export function prepareHostShellConfig({ home, project, branch, repoRoot }) {
     const hostDir = hostShellConfigDir(home, project, branch);
-    fs.rmSync(hostDir, { recursive: true, force: true });
+    removeDirRecursive(hostDir);
     fs.mkdirSync(hostDir, { recursive: true });
     writeSanitizedGitconfig({
         home,
@@ -306,11 +288,19 @@ function normalizeWorktreePath(worktreePath) {
     if (!worktreePath) {
         return '';
     }
+    const wslMount = process.platform === 'win32'
+        ? worktreePath.match(/^\/mnt\/([A-Za-z])(?:\/(.*))?$/)
+        : null;
+    const hostPath = wslMount
+        ? `${wslMount[1].toUpperCase()}:\\${(wslMount[2] ?? '').replace(/\//g, '\\')}`
+        : worktreePath;
     try {
-        return fs.existsSync(worktreePath) ? fs.realpathSync(worktreePath) : path.resolve(worktreePath);
+        const normalized = fs.existsSync(hostPath) ? fs.realpathSync(hostPath) : path.resolve(hostPath);
+        return process.platform === 'win32' ? normalized.toLowerCase() : normalized;
     }
     catch {
-        return path.resolve(worktreePath);
+        const normalized = path.resolve(hostPath);
+        return process.platform === 'win32' ? normalized.toLowerCase() : normalized;
     }
 }
 export function getGitSigningKey({ home, repoPath = null, execFn = execFileSync } = {}) {
@@ -481,7 +471,7 @@ export function collectHostProxyEntries(env) {
     });
 }
 export function buildContainerEnvFile(resolvedTools, engine, runSafeEngineFn = runSafeEngine, options = {}) {
-    const { additionalEntries = [], mkdtempFn = fs.mkdtempSync, writeFileFn = fs.writeFileSync, chmodFn = fs.chmodSync, rmFn = fs.rmSync, tmpDir = os.tmpdir(), runSafeFn = runSafe } = options;
+    const { additionalEntries = [], mkdtempFn = fs.mkdtempSync, writeFileFn = fs.writeFileSync, chmodFn = fs.chmodSync, rmFn = removeDirRecursive, tmpDir = os.tmpdir(), runSafeFn = runSafe } = options;
     const entries = resolvedTools.flatMap(({ tool }) => Object.entries(tool.envVars ?? {}));
     entries.push(...additionalEntries);
     let ghToken = runSafeEngineFn(engine, 'gh', ['auth', 'token']);
@@ -705,141 +695,6 @@ export function ensureClaudeSettings(toolDir, hostHomeDir) {
         fs.writeFileSync(settingsPath, JSON.stringify(data, null, 4), 'utf8');
     }
 }
-function resolveHostCatalogPath(value, hostHomeDir) {
-    if (typeof value !== 'string' || value === '') {
-        return null;
-    }
-    let resolved;
-    if (value === '~' || value.startsWith('~/') || value.startsWith('~\\')) {
-        resolved = path.join(hostHomeDir, value.slice(1).replace(/^[/\\]+/, ''));
-    }
-    else if (path.isAbsolute(value)) {
-        resolved = value;
-    }
-    else {
-        resolved = path.join(hostHomeDir, '.codex', value);
-    }
-    try {
-        if (!fs.statSync(resolved).isFile()) {
-            return null;
-        }
-        fs.accessSync(resolved, fs.constants.R_OK);
-        return resolved;
-    }
-    catch {
-        return null;
-    }
-}
-const CODEX_DISABLED_FEATURE_FLAGS = ['apps', 'enable_mcp_apps'];
-function removeCodexMcpServers(sandboxParsed) {
-    if (!Object.hasOwn(sandboxParsed, 'mcp_servers')) {
-        return false;
-    }
-    delete sandboxParsed.mcp_servers;
-    return true;
-}
-function inheritDisabledCodexFeatureFlags(sandboxParsed, hostParsed) {
-    if (!isJsonObjectRecord(hostParsed.features)) {
-        return false;
-    }
-    if (Object.hasOwn(sandboxParsed, 'features') && !isJsonObjectRecord(sandboxParsed.features)) {
-        return false;
-    }
-    let sandboxFeatures = sandboxParsed.features;
-    let changed = false;
-    for (const key of CODEX_DISABLED_FEATURE_FLAGS) {
-        if (hostParsed.features[key] !== false) {
-            continue;
-        }
-        if (!sandboxFeatures) {
-            sandboxFeatures = {};
-            sandboxParsed.features = sandboxFeatures;
-        }
-        if (sandboxFeatures[key] !== false) {
-            sandboxFeatures[key] = false;
-            changed = true;
-        }
-    }
-    return changed;
-}
-export function ensureCodexModelInheritance(toolDir, hostHomeDir, containerCodexDir = '/home/devuser/.codex') {
-    if (!hostHomeDir) {
-        return;
-    }
-    const hostConfigPath = path.join(hostHomeDir, '.codex', 'config.toml');
-    if (!fs.existsSync(hostConfigPath)) {
-        return;
-    }
-    let hostParsed;
-    try {
-        hostParsed = toml.parse(fs.readFileSync(hostConfigPath, 'utf8'));
-    }
-    catch {
-        return;
-    }
-    const sandboxConfigPath = path.join(toolDir, 'config.toml');
-    // This rewrites sandbox-side TOML and drops comments; the host config stays untouched.
-    let sandboxParsed = {};
-    if (fs.existsSync(sandboxConfigPath)) {
-        try {
-            sandboxParsed = toml.parse(fs.readFileSync(sandboxConfigPath, 'utf8'));
-        }
-        catch {
-            return;
-        }
-    }
-    let changed = removeCodexMcpServers(sandboxParsed);
-    changed = inheritDisabledCodexFeatureFlags(sandboxParsed, hostParsed) || changed;
-    const inheritSpecs = [
-        ['model', 'string'],
-        ['model_reasoning_effort', 'string'],
-        ['model_auto_compact_token_limit', 'number']
-    ];
-    for (const [key, type] of inheritSpecs) {
-        if (Object.hasOwn(sandboxParsed, key)) {
-            continue;
-        }
-        const value = hostParsed[key];
-        if (type === 'string' && (typeof value !== 'string' || value === '')) {
-            continue;
-        }
-        if (type === 'number' && (typeof value !== 'number' || !Number.isFinite(value) || value <= 0)) {
-            continue;
-        }
-        sandboxParsed[key] = value;
-        changed = true;
-    }
-    if (!Object.hasOwn(sandboxParsed, 'model_catalog_json')) {
-        const hostCatalogPath = resolveHostCatalogPath(hostParsed['model_catalog_json'], hostHomeDir);
-        if (hostCatalogPath) {
-            try {
-                const basename = path.basename(hostCatalogPath);
-                const destDir = path.join(toolDir, 'model-catalogs');
-                fs.mkdirSync(destDir, { recursive: true });
-                fs.copyFileSync(hostCatalogPath, path.join(destDir, basename));
-                sandboxParsed['model_catalog_json'] = path.posix.join(containerCodexDir, 'model-catalogs', basename);
-                changed = true;
-            }
-            catch {
-                // Copy failed (e.g. permissions): skip catalog, keep scalar inheritance intact.
-            }
-        }
-    }
-    if (changed) {
-        fs.writeFileSync(sandboxConfigPath, `${toml.stringify(sandboxParsed)}\n`, 'utf8');
-    }
-}
-export function ensureCodexWorkspaceTrust(toolDir) {
-    const configPath = path.join(toolDir, 'config.toml');
-    let content = '';
-    if (fs.existsSync(configPath)) {
-        content = fs.readFileSync(configPath, 'utf8');
-    }
-    if (!content.includes('[projects."/workspace"]')) {
-        const entry = '\n[projects."/workspace"]\ntrust_level = "trusted"\n';
-        fs.writeFileSync(configPath, content + entry, 'utf8');
-    }
-}
 export function ensureOpenCodeModelInheritance(toolDir, hostHomeDir) {
     if (!hostHomeDir) {
         return;
@@ -874,22 +729,6 @@ export function ensureOpenCodeModelInheritance(toolDir, hostHomeDir) {
         fs.writeFileSync(sandboxConfigPath, JSON.stringify(sandboxJson, null, 2), 'utf8');
     }
 }
-export function ensureGeminiWorkspaceTrust(toolDir) {
-    const trustPath = path.join(toolDir, 'trustedFolders.json');
-    let data = {};
-    if (fs.existsSync(trustPath)) {
-        try {
-            data = JSON.parse(fs.readFileSync(trustPath, 'utf8'));
-        }
-        catch {
-            // malformed JSON, start fresh
-        }
-    }
-    if (data['/workspace'] !== 'TRUST_FOLDER') {
-        data['/workspace'] = 'TRUST_FOLDER';
-        fs.writeFileSync(trustPath, JSON.stringify(data, null, 2), 'utf8');
-    }
-}
 export function sandboxAliasesPath(home) {
     return hostJoin(home, '.agent-infra', 'aliases', 'sandbox.sh');
 }
@@ -900,7 +739,7 @@ function stripManagedSandboxAliasBlocks(content) {
     const blockPattern = new RegExp(`${escapeRegExp(SANDBOX_ALIAS_BLOCK_BEGIN)}[\\s\\S]*?${escapeRegExp(SANDBOX_ALIAS_BLOCK_END)}\\n?`, 'g');
     return content.replace(blockPattern, '').trimEnd();
 }
-function isLegacyManagedSandboxAliasFile(content) {
+function isLegacyManagedSandboxAliasFile(content, aliases) {
     const lines = content
         .split(/\r?\n/)
         .map((line) => line.trim())
@@ -908,19 +747,25 @@ function isLegacyManagedSandboxAliasFile(content) {
     if (lines.length === 0) {
         return false;
     }
-    const aliasPattern = new RegExp(`^alias (${SANDBOX_ALIAS_NAMES.map(escapeRegExp).join('|')})=`);
+    const aliasPattern = new RegExp(`^alias (${aliases.map((alias) => escapeRegExp(alias.name)).join('|')})=`);
     return lines.every((line) => aliasPattern.test(line));
 }
-export function ensureSandboxAliasesFile(home) {
+function registrySandboxAliases() {
+    return listAgentClientAdapters().flatMap((adapter) => adapter.sandbox.aliases);
+}
+function renderSandboxAliases(aliases) {
+    return `${aliases.map((alias) => `alias ${alias.name}='${alias.command.replaceAll("'", "'\\''")}'`).join('\n')}\n`;
+}
+export function ensureSandboxAliasesFile(home, aliases = registrySandboxAliases()) {
     const aliasesPath = sandboxAliasesPath(home);
-    const managedBlock = `${SANDBOX_ALIAS_BLOCK_BEGIN}\n${DEFAULT_SANDBOX_ALIASES}${SANDBOX_ALIAS_BLOCK_END}\n`;
+    const managedBlock = `${SANDBOX_ALIAS_BLOCK_BEGIN}\n${renderSandboxAliases(aliases)}${SANDBOX_ALIAS_BLOCK_END}\n`;
     fs.mkdirSync(path.dirname(aliasesPath), { recursive: true });
     const created = !fs.existsSync(aliasesPath);
     let existing = '';
     if (!created) {
         existing = fs.readFileSync(aliasesPath, 'utf8');
     }
-    const userContent = isLegacyManagedSandboxAliasFile(existing)
+    const userContent = isLegacyManagedSandboxAliasFile(existing, aliases)
         ? ''
         : stripManagedSandboxAliasBlocks(existing);
     const nextContent = userContent
@@ -1018,15 +863,14 @@ export async function create(args) {
     };
     const worktreeCandidates = worktreeDirCandidates(effectiveConfig, branch);
     assertBranchAvailable(config.repoRoot, branch, { allowedWorktrees: worktreeCandidates });
-    const tools = resolveTools(effectiveConfig);
+    const capabilityPlan = createSandboxCapabilityPlan(effectiveConfig);
+    const tools = [...capabilityPlan.tools];
     const resolvedTools = resolveToolDirs(effectiveConfig, tools, branch);
     // Fatal credential states still fail before filesystem/docker side effects.
-    // A genuinely missing Claude Code credential only removes Claude Code's
-    // sandbox config and credential mounts for this create run.
+    // Missing credentials leave the selected Claude Code capability installed
+    // and mounted; only the absent live credential file is omitted.
     const credentialOutcome = prepareClaudeCredentials(effectiveConfig.home, effectiveConfig.project, resolvedTools);
-    const effectiveResolvedTools = credentialOutcome.status === 'SKIPPED'
-        ? resolvedTools.filter(({ tool }) => tool.id !== 'claude-code')
-        : resolvedTools;
+    const effectiveResolvedTools = resolvedTools;
     const container = containerName(effectiveConfig, branch);
     const worktree = worktreeCandidates.find((candidate) => fs.existsSync(candidate)) ?? worktreeCandidates[0] ?? '';
     const shareCommon = shareCommonDir(effectiveConfig);
@@ -1045,6 +889,16 @@ export async function create(args) {
     }
     try {
         p.log.step('Checking container engine...');
+        const prepareHookResults = await runSandboxHooks({
+            hooks: capabilityPlan.hooksByPhase.prepare,
+            phase: 'prepare',
+            context: { config: effectiveConfig, plan: capabilityPlan },
+            runCommand: runBoundedSandboxHookCommand
+        });
+        const prepareFailure = prepareHookResults.find((result) => result.status === 'fatal');
+        if (prepareFailure) {
+            throw new Error(prepareFailure.message ?? `Sandbox hook '${prepareFailure.hookId}' failed.`);
+        }
         await ensureDocker(effectiveConfig, (detail) => {
             p.log.info(`  ${detail}`);
         });
@@ -1111,7 +965,7 @@ export async function create(args) {
                         if (fs.readdirSync(worktree).length > 0) {
                             return `Worktree exists at ${worktree}`;
                         }
-                        fs.rmSync(worktree, { recursive: true, force: true });
+                        removeDirRecursive(worktree);
                     }
                     const branchExists = runOk('git', [
                         '-C',
@@ -1200,7 +1054,7 @@ export async function create(args) {
                             runSafeEngine(engine, 'docker', ['rm', name]);
                         }
                     }
-                    const aliasesFile = ensureSandboxAliasesFile(effectiveConfig.home);
+                    const aliasesFile = ensureSandboxAliasesFile(effectiveConfig.home, capabilityPlan.aliases);
                     if (aliasesFile.created) {
                         message(`Created default sandbox aliases at ${aliasesFile.path}`);
                     }
@@ -1229,6 +1083,24 @@ export async function create(args) {
                     let hostShellConfig;
                     let tmpfsSeedPlan = [];
                     try {
+                        const beforeCreateResults = await runSandboxHooks({
+                            hooks: capabilityPlan.hooksByPhase['before-container-create'],
+                            phase: 'before-container-create',
+                            context: {
+                                config: effectiveConfig,
+                                plan: capabilityPlan,
+                                create: {
+                                    hostHome: effectiveConfig.home,
+                                    resolvedTools: effectiveResolvedTools
+                                }
+                            },
+                            runCommand: runBoundedSandboxHookCommand
+                        });
+                        const beforeCreateFailure = beforeCreateResults.find((result) => result.status === 'fatal');
+                        if (beforeCreateFailure) {
+                            throw new Error(beforeCreateFailure.message
+                                ?? `Sandbox hook '${beforeCreateFailure.hookId}' failed.`);
+                        }
                         const claudeCodeEntry = effectiveResolvedTools.find(({ tool }) => tool.id === 'claude-code');
                         if (claudeCodeEntry) {
                             ensureClaudeOnboarding(claudeCodeEntry.dir, effectiveConfig.home);
@@ -1237,15 +1109,6 @@ export async function create(args) {
                             // provider/API settings are enough for Claude Code. If no usable
                             // auth was present, the claude-code entry was removed above.
                         }
-                        const codexEntry = effectiveResolvedTools.find(({ tool }) => tool.id === 'codex');
-                        if (codexEntry) {
-                            ensureCodexModelInheritance(codexEntry.dir, effectiveConfig.home, codexEntry.tool.containerMount);
-                            ensureCodexWorkspaceTrust(codexEntry.dir);
-                        }
-                        const geminiEntry = effectiveResolvedTools.find(({ tool }) => tool.id === 'gemini-cli');
-                        if (geminiEntry) {
-                            ensureGeminiWorkspaceTrust(geminiEntry.dir);
-                        }
                         const opencodeEntry = effectiveResolvedTools.find(({ tool }) => tool.id === 'opencode');
                         if (opencodeEntry) {
                             // The TUI reads <toolDir>/opencode.json via OPENCODE_CONFIG pinned in tools.js.
@@ -1315,6 +1178,8 @@ export async function create(args) {
                             sandboxLabel(effectiveConfig),
                             '--label',
                             `${sandboxBranchLabel(effectiveConfig)}=${branch}`,
+                            '--label',
+                            `${sandboxRuntimeCapabilityLabel(effectiveConfig)}=${capabilityPlan.runtimeSignature}`,
                             ...coreVolumes,
                             ...buildClipboardVolumeArgs(engine, effectiveConfig.home),
                             '-v',
@@ -1392,6 +1257,17 @@ export async function create(args) {
                             runSafeEngine(engine, 'docker', ['exec', container, 'bash', '-lc', command]);
                         }
                     }
+                    const afterStartResults = await runSandboxHooks({
+                        hooks: capabilityPlan.hooksByPhase['after-container-start'],
+                        phase: 'after-container-start',
+                        context: { config: effectiveConfig, plan: capabilityPlan },
+                        runCommand: runBoundedSandboxHookCommand
+                    });
+                    const afterStartFailure = afterStartResults.find((result) => result.status === 'fatal');
+                    if (afterStartFailure) {
+                        throw new Error(afterStartFailure.message
+                            ?? `Sandbox hook '${afterStartFailure.hookId}' failed.`);
+                    }
                     return 'Container started';
                 }
             }

package/dist/lib/sandbox/commands/enter.js

@@ -2,6 +2,7 @@ import { loadConfig } from '../config.js';
 import { assertValidBranchName, containerNameCandidates, sandboxBranchLabel, sandboxLabel } from '../constants.js';
 import { detectEngine } from '../engine.js';
 import { ensureSandboxReady } from '../recovery.js';
+import { createSandboxCapabilityPlan, runBoundedSandboxHookCommand, runSandboxHooks } from '../agent-client-reconciler.js';
 import { formatCredentialWarnings, formatRemaining, hasClaudeProviderAuth, reconcileClaudeCredentials, redactCommandError, validateClaudeCredentialsEnvOverride } from '../credentials.js';
 import { runInteractiveEngine } from '../shell.js';
 import { dotfilesCacheDir, materializeDotfiles } from '../dotfiles.js';
@@ -123,7 +124,22 @@ export async function enter(args) {
     });
     const container = ready.container;
     const cmd = parsed.command;
-    if (config.tools.includes('claude-code')) {
+    const capabilityPlan = createSandboxCapabilityPlan(config);
+    const enterHookResults = await runSandboxHooks({
+        hooks: capabilityPlan.hooksByPhase['before-enter'],
+        phase: 'before-enter',
+        context: { config, plan: capabilityPlan },
+        runCommand: runBoundedSandboxHookCommand
+    });
+    for (const result of enterHookResults) {
+        if (result.status === 'fatal') {
+            throw new Error(result.message ?? `Sandbox hook '${result.hookId}' failed.`);
+        }
+        if (result.status === 'warning') {
+            process.stderr.write(`Warning: ${result.message ?? `Sandbox hook '${result.hookId}' did not complete.`}\n`);
+        }
+    }
+    if (capabilityPlan.selectedAgentClients.some((adapter) => adapter.id === 'claude-code')) {
         try {
             // Scan all projects so a refresh from a neighbouring sandbox can still flow back to the host.
             const providerAuthAvailable = hasClaudeProviderAuth(config.home);

package/dist/lib/sandbox/commands/prune.js

@@ -10,7 +10,7 @@ import { hostJoin } from '../engines/wsl2-paths.js';
 import { removeManagedDir, removeWorktreeDir } from '../managed-fs.js';
 import { parseLabels } from './ls.js';
 import { runEngine, runSafe } from '../shell.js';
-import { resolveTools } from '../tools.js';
+import { createSandboxCapabilityPlan } from '../agent-client-reconciler.js';
 const USAGE = `Usage: ai sandbox prune [--dry-run]`;
 function listChildDirs(base) {
     if (!fs.existsSync(base)) {
@@ -129,7 +129,7 @@ export async function prune(args) {
         return;
     }
     const config = loadConfig();
-    const tools = resolveTools(config);
+    const tools = [...createSandboxCapabilityPlan(config).cleanupInventory];
     const engine = detectEngine(config);
     const psArgs = [
         'ps',

package/dist/lib/sandbox/commands/rm.js

@@ -10,7 +10,8 @@ import { pruneSandboxDanglingImages } from '../image-prune.js';
 import { removeManagedDir, removeWorktreeDir } from '../managed-fs.js';
 import { runOk, runSafe, runSafeEngine } from '../shell.js';
 import { resolveTaskBranch } from '../task-resolver.js';
-import { resolveTools, toolConfigDirCandidates, toolProjectDirCandidates } from '../tools.js';
+import { toolConfigDirCandidates, toolProjectDirCandidates } from '../tools.js';
+import { createSandboxCapabilityPlan } from '../agent-client-reconciler.js';
 import { fetchSandboxRows } from './list-running.js';
 import { lookupShortIdByBranch } from '../../task/short-id.js';
 const USAGE = `Usage:
@@ -284,7 +285,7 @@ export async function rm(args) {
         throw new Error(USAGE);
     }
     const config = loadConfig();
-    const tools = resolveTools(config);
+    const tools = [...createSandboxCapabilityPlan(config).cleanupInventory];
     if (values.purge) {
         await rmPurge(config, tools);
         return;

package/dist/lib/sandbox/commands/show.js

@@ -8,7 +8,7 @@ import { resolveTools, toolConfigDirCandidates } from '../tools.js';
 const USAGE = `Usage: ai sandbox show <branch | TASK-id | N>
 
 Shows one sandbox's worktree path and per-tool state paths (Claude Code,
-Codex, Gemini CLI, OpenCode). The argument follows the same contract as
+Codex, Antigravity CLI, OpenCode). The argument follows the same contract as
 'ai sandbox exec' and 'ai sandbox start': N (bare) is the recommended form
 for task short ids (e.g. 'ai sandbox show 11'); a TASK-id or a plain
 branch name are also accepted. Use 'ai sandbox ls' for the container list.`;

package/dist/lib/sandbox/config.js

@@ -3,6 +3,8 @@ import path from 'node:path';
 import { homedir, platform } from 'node:os';
 import { execFileSync } from 'node:child_process';
 import pc from 'picocolors';
+import { normalizeAgentClients } from '../agent-clients/config.js';
+import { isAgentClientId } from '../agent-clients/types.js';
 import { validateSandboxEngine } from './engine.js';
 import { hostJoin } from './engines/wsl2-paths.js';
 import { findRuntimeEngineMismatches } from './runtime-engines.js';
@@ -10,7 +12,7 @@ import { parseCustomTools } from './tools.js';
 const DEFAULTS = Object.freeze({
     engine: null,
     runtimes: ['node22'],
-    tools: ['agent-infra', 'claude-code', 'codex', 'gemini-cli', 'opencode'],
+    tools: ['agent-infra'],
     refreshIntervalDays: 7,
     dockerfile: null,
     vm: {
@@ -46,6 +48,18 @@ function cloneDefaults() {
         vm: { ...DEFAULTS.vm }
     };
 }
+function resolveSandboxToolIds(sandbox, agentClients, defaults) {
+    if (Array.isArray(sandbox.tools)) {
+        const tools = sandbox.tools.filter((tool) => typeof tool === 'string');
+        return agentClients.source === 'canonical'
+            ? tools.filter((tool) => !isAgentClientId(tool))
+            : tools;
+    }
+    if (agentClients.remainingSandboxTools !== undefined) {
+        return [...agentClients.remainingSandboxTools];
+    }
+    return [...defaults];
+}
 export function loadConfig({ platformFn = platform, writeStderr = (chunk) => process.stderr.write(chunk) } = {}) {
     const repoRoot = detectRepoRoot();
     const home = homedir();
@@ -57,6 +71,8 @@ export function loadConfig({ platformFn = platform, writeStderr = (chunk) => pro
         throw new Error('No .agents/.airc.json found. Run "ai init" first.');
     }
     const airc = JSON.parse(fs.readFileSync(configPath, 'utf8'));
+    const sandboxInput = airc.sandbox;
+    const agentClients = normalizeAgentClients(airc);
     const defaults = cloneDefaults();
     const sandbox = airc.sandbox ?? {};
     const engine = validateSandboxEngine(sandbox.engine ?? defaults.engine, { platformFn });
@@ -83,6 +99,7 @@ export function loadConfig({ platformFn = platform, writeStderr = (chunk) => pro
         }
     }
     const customTools = parseCustomTools(sandbox.customTools, { home });
+    const tools = resolveSandboxToolIds(sandbox, agentClients, defaults.tools);
     return {
         repoRoot,
         configPath,
@@ -97,10 +114,10 @@ export function loadConfig({ platformFn = platform, writeStderr = (chunk) => pro
         dotfilesDir: hostJoin(home, '.agent-infra', 'dotfiles'),
         engine,
         runtimes,
-        tools: Array.isArray(sandbox.tools) && sandbox.tools.length > 0
-            ? [...sandbox.tools]
-            : defaults.tools,
+        tools,
         customTools,
+        agentClientState: agentClients.state,
+        agentClientSource: agentClients.source,
         refreshIntervalDays: asNonNegativeIntegerOrDefault(sandbox.refreshIntervalDays, defaults.refreshIntervalDays),
         dockerfile,
         vm: {

package/dist/lib/sandbox/constants.js

@@ -75,6 +75,9 @@ export function sandboxImageConfigLabel(config) {
 export function sandboxImageRefreshLabel(config) {
     return `${sandboxLabel(config)}.last-refresh`;
 }
+export function sandboxRuntimeCapabilityLabel(config) {
+    return `agent-infra.sandbox.runtime-capability.${config.project}`;
+}
 export function parsePositiveIntegerOption(value, optionName) {
     if (value === undefined || value === null) {
         return undefined;

package/dist/lib/sandbox/dockerfile.js

@@ -3,6 +3,7 @@ import os from 'node:os';
 import path from 'node:path';
 import { createHash } from 'node:crypto';
 import { fileURLToPath } from 'node:url';
+import { removeDirRecursive } from '../remove-dir.js';
 const RUNTIMES_DIR = path.join(path.dirname(fileURLToPath(import.meta.url)), 'runtimes');
 function listRuntimeFragments() {
     return fs.readdirSync(RUNTIMES_DIR)
@@ -63,7 +64,7 @@ export function prepareDockerfile(config) {
         path: tempPath,
         signature: dockerfileSignature(config),
         cleanup() {
-            fs.rmSync(tempDir, { recursive: true, force: true });
+            removeDirRecursive(tempDir);
         }
     };
 }

package/dist/lib/sandbox/managed-fs.js

@@ -1,6 +1,6 @@
-import fs from 'node:fs';
 import path from 'node:path';
-import { run } from './shell.js';
+import { run, runSafe } from './shell.js';
+import { removeDirRecursive } from '../remove-dir.js';
 export function assertManagedPath(root, target) {
     const resolvedRoot = path.resolve(root);
     const resolvedTarget = path.resolve(target);
@@ -12,14 +12,19 @@ export function assertManagedPath(root, target) {
 }
 export function removeManagedDir(root, dir) {
     assertManagedPath(root, dir);
-    fs.rmSync(dir, { recursive: true, force: true });
+    removeDirRecursive(dir);
 }
-export function removeWorktreeDir(repoRoot, worktreeBase, dir) {
+export function removeWorktreeDir(repoRoot, worktreeBase, dir, { runFn = run, runSafeFn = runSafe } = {}) {
     try {
-        run('git', ['-C', repoRoot, 'worktree', 'remove', dir, '--force']);
+        runFn('git', ['-C', repoRoot, 'worktree', 'remove', dir, '--force']);
     }
     catch {
+        // On WSL2 the worktree is registered under its `/mnt/<drive>/...` path, so
+        // `git worktree remove <windows-path>` cannot match it. Delete the managed
+        // directory directly, then prune the now-dangling worktree metadata so the
+        // branch is no longer reported as checked out.
         removeManagedDir(worktreeBase, dir);
+        runSafeFn('git', ['-C', repoRoot, 'worktree', 'prune']);
     }
 }
 //# sourceMappingURL=managed-fs.js.map