@fitlab-ai/agent-infra 0.9.13 → 0.9.14

package/lib/internal/platform-pr.ts

@@ -14,6 +14,7 @@ import {
 import type { PullRequestResult } from '../platform/pull-requests.ts';
 import { reportWrite, summaryContext, syncPullRequestSummary } from '../platform/pr-summary.ts';
 import type { PlatformResult } from '../platform/types.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
 
 const USAGE = `Usage: agent-infra-internal platform-pr inspect <task-ref> [--cwd <path>]
        agent-infra-internal platform-pr resolve-external <task-ref> --agent <agent> [--pr <token>] [--dry-run] [--cwd <path>]
@@ -67,10 +68,22 @@ function readFile(value: string, cwd: string): string {
 }
 
 async function platformPr(args: string[] = []): Promise<void> {
+  if (!ensureInternalHandlerRoute('platform-pr', args)) return;
   if (args[0] === '--help' || args[0] === '-h') { process.stdout.write(USAGE); return; }
   const operation = args[0];
-  if (!operation || !['inspect', 'resolve-external', 'create', 'bind', 'skip', 'sync', 'sync-in-labels', 'summary-context', 'change-report', 'summary-sync'].includes(operation)) { fail('a valid operation is required'); return; }
-  if (operation === 'sync-in-labels') {
+  if (!operation || ![
+    internalHandlerRoute('platform-pr', 'inspect', operation),
+    internalHandlerRoute('platform-pr', 'resolve-external', operation),
+    internalHandlerRoute('platform-pr', 'create', operation),
+    internalHandlerRoute('platform-pr', 'bind', operation),
+    internalHandlerRoute('platform-pr', 'skip', operation),
+    internalHandlerRoute('platform-pr', 'sync', operation),
+    internalHandlerRoute('platform-pr', 'sync-in-labels', operation),
+    internalHandlerRoute('platform-pr', 'summary-context', operation),
+    internalHandlerRoute('platform-pr', 'change-report', operation),
+    internalHandlerRoute('platform-pr', 'summary-sync', operation)
+  ].some(Boolean)) { fail('a valid operation is required'); return; }
+  if (internalHandlerRoute('platform-pr', 'sync-in-labels', operation)) {
     const parsed = parse(args, 1);
     if (parsed.error) { fail(parsed.error); return; }
     const values = parsed.values;
@@ -102,20 +115,20 @@ async function platformPr(args: string[] = []): Promise<void> {
   };
   const unexpected = Object.keys(values).find((name) => !allowed[operation]!.includes(name));
   if (unexpected) { fail(`${operation} does not accept --${unexpected}`); return; }
-  if (operation === 'inspect') { finish(await inspectPlatformPullRequest(taskRef, { cwd })); return; }
-  if (operation === 'summary-context') { finish(await summaryContext(taskRef, { cwd })); return; }
+  if (internalHandlerRoute('platform-pr', 'inspect', operation)) { finish(await inspectPlatformPullRequest(taskRef, { cwd })); return; }
+  if (internalHandlerRoute('platform-pr', 'summary-context', operation)) { finish(await summaryContext(taskRef, { cwd })); return; }
   if (typeof values.agent !== 'string' || !values.agent) { fail(`${operation} requires --agent`); return; }
   const agent = normalizeAgentToken(values.agent);
   if (!agent) { fail(`invalid --agent '${values.agent}': ${AGENT_USAGE_HINT}`); return; }
   values.agent = agent;
-  if (operation === 'change-report') {
+  if (internalHandlerRoute('platform-pr', 'change-report', operation)) {
     if (typeof values.mechanicalFile !== 'string' || typeof values.precheckFile !== 'string') { fail('change-report requires --mechanical-file and --precheck-file'); return; }
     finish(await reportWrite(taskRef, {
       cwd, agent: values.agent, mechanicalFile: values.mechanicalFile, precheckFile: values.precheckFile, dryRun: values.dryRun === true
     }));
     return;
   }
-  if (operation === 'skip') {
+  if (internalHandlerRoute('platform-pr', 'skip', operation)) {
     finish(await skipPlatformPullRequestFact(taskRef, { cwd, agent, dryRun: values.dryRun === true }));
     return;
   }
@@ -124,25 +137,25 @@ async function platformPr(args: string[] = []): Promise<void> {
     fail('--result must be pr_created, pr_reused, or no_op');
     return;
   }
-  if (operation === 'resolve-external') {
+  if (internalHandlerRoute('platform-pr', 'resolve-external', operation)) {
     const pr = values.pr === undefined ? undefined : values.pr;
     if (pr !== undefined && (typeof pr !== 'string' || !pr)) { fail('resolve-external requires --pr <token>'); return; }
     finish(await resolveExternalPullRequest(taskRef, { cwd, agent: values.agent, pr, dryRun: values.dryRun === true }));
     return;
   }
-  if (operation === 'bind') {
+  if (internalHandlerRoute('platform-pr', 'bind', operation)) {
     const pr = values.pr;
     if (typeof pr !== 'string' || !pr) { fail('bind requires --pr <token>'); return; }
     finish(await bindPlatformPullRequest(taskRef, { cwd, agent: values.agent, pr, dryRun: values.dryRun === true }));
     return;
   }
-  if (operation === 'sync') {
+  if (internalHandlerRoute('platform-pr', 'sync', operation)) {
     if (values.metadata !== true && values.closingIssue !== true) { fail('sync requires --metadata or --closing-issue'); return; }
     if (!primaryResult) { fail('sync requires --result pr_created, pr_reused, or no_op'); return; }
     finish(await syncPlatformPullRequest(taskRef, { cwd, agent: values.agent, metadata: values.metadata === true, closingIssue: values.closingIssue === true, primaryResult: primaryResult as 'pr_created' | 'pr_reused' | 'no_op', dryRun: values.dryRun === true }));
     return;
   }
-  if (operation === 'summary-sync') {
+  if (internalHandlerRoute('platform-pr', 'summary-sync', operation)) {
     if (typeof values.bodyFile !== 'string') { fail('summary-sync requires --body-file'); return; }
     if (typeof values.changeReportFile !== 'string') { fail('summary-sync requires --change-report-file'); return; }
     if (!primaryResult) { fail('summary-sync requires --result pr_created, pr_reused, or no_op'); return; }

package/lib/internal/platform-release-notes.ts

@@ -2,6 +2,7 @@ import fs from 'node:fs';
 import path from 'node:path';
 
 import { publishReleaseNotes, releaseNoteContext, stageReleaseNotes } from '../platform/release-notes.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
 
 type ParsedOptions = Readonly<{ values: ReadonlyMap<string, string>; switches: ReadonlySet<string> }>;
 
@@ -37,8 +38,9 @@ function finish(result: { status: string; [key: string]: unknown }): void {
 }
 
 async function platformReleaseNotes(args: string[] = []): Promise<void> {
+  if (!ensureInternalHandlerRoute('platform-release-notes', args)) return;
   const [action, ...rest] = args;
-  if (action === 'context') {
+  if (internalHandlerRoute('platform-release-notes', 'context', action ?? '')) {
     const parsed = parseOptions(rest, ['--cwd', '--history-limit', '--from-tag', '--to-tag', '--branch']);
     if (!parsed) return finish(invalidInput('invalid context options'));
     const cwd = path.resolve(parsed.values.get('--cwd') || process.cwd());
@@ -51,7 +53,7 @@ async function platformReleaseNotes(args: string[] = []): Promise<void> {
     }, { cwd }));
     return;
   }
-  if (action === 'stage') {
+  if (internalHandlerRoute('platform-release-notes', 'stage', action ?? '')) {
     const parsed = parseOptions(rest, ['--cwd', '--notes-file']);
     if (!parsed) return finish(invalidInput('invalid stage options'));
     const notesFile = parsed.values.get('--notes-file');
@@ -62,7 +64,7 @@ async function platformReleaseNotes(args: string[] = []): Promise<void> {
     ));
     return;
   }
-  if (action === 'publish') {
+  if (internalHandlerRoute('platform-release-notes', 'publish', action ?? '')) {
     const parsed = parseOptions(
       rest,
       ['--cwd', '--notes-file', '--tag', '--title', '--expected-sha256'],

package/lib/internal/platform-security.ts

@@ -0,0 +1,83 @@
+import path from 'node:path';
+
+import { dismissSecurityAlert, readCommentFile, readSecurityAlert } from '../platform/security-alerts.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
+
+const USAGE = `Usage: agent-infra-internal platform-security read --kind <dependabot|code-scanning> --number <number> [--cwd <path>]
+       agent-infra-internal platform-security dismiss --kind <dependabot|code-scanning> --number <number> --reason <reason> --comment-file <path> [--cwd <path>]
+`;
+
+function fail(message: string): void {
+  process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, error: { code: 'SECURITY_INPUT_INVALID', message, retryable: false } })}\n`);
+  process.stderr.write(USAGE);
+  process.exitCode = 1;
+}
+
+function finish(result: { status: string; [key: string]: unknown }): void {
+  process.stdout.write(`${JSON.stringify(result)}\n`);
+  process.exitCode = result.status === 'blocked' ? 2 : result.status === 'failed' ? 1 : 0;
+}
+
+function parseOptions(args: string[]): { values: Map<string, string>; switches: Set<string> } | null {
+  const values = new Map<string, string>();
+  const switches = new Set<string>();
+  const valueFlags = new Set(['--cwd', '--kind', '--number', '--reason', '--comment-file']);
+  for (let index = 0; index < args.length; index += 1) {
+    const flag = args[index]!;
+    if (flag === '--help' || flag === '-h') return null;
+    if (!valueFlags.has(flag) || values.has(flag) || switches.has(flag)) return null;
+    const value = args[++index];
+    if (!value || value.startsWith('--')) return null;
+    values.set(flag, value);
+  }
+  return { values, switches };
+}
+
+function required(values: Map<string, string>, flag: string): string | null {
+  const value = values.get(flag);
+  return value && !value.startsWith('--') ? value : null;
+}
+
+async function platformSecurity(args: string[] = []): Promise<void> {
+  if (!ensureInternalHandlerRoute('platform-security', args)) return;
+  if (args[0] === '--help' || args[0] === '-h') { process.stdout.write(USAGE); return; }
+  const action = args[0];
+  if (!internalHandlerRoute('platform-security', 'read', action ?? '')
+    && !internalHandlerRoute('platform-security', 'dismiss', action ?? '')) {
+    fail('a valid operation is required');
+    return;
+  }
+  if (action !== 'read' && action !== 'dismiss') { fail('a valid operation is required'); return; }
+  const parsed = parseOptions(args.slice(1));
+  if (!parsed) { fail('invalid security alert options'); return; }
+  const kind = required(parsed.values, '--kind');
+  const numberRaw = required(parsed.values, '--number');
+  const cwd = path.resolve(required(parsed.values, '--cwd') || process.cwd());
+  if (!kind || !numberRaw) { fail(`${action} requires --kind and --number`); return; }
+  const number = Number(numberRaw);
+  if (!Number.isSafeInteger(number) || number <= 0) { fail('--number must be a positive integer'); return; }
+
+  if (action === 'read') {
+    if ([...parsed.values.keys()].some((flag) => !['--kind', '--number', '--cwd'].includes(flag))) {
+      fail('read does not accept dismissal options');
+      return;
+    }
+    finish(await readSecurityAlert({ kind: kind as 'dependabot' | 'code-scanning', number }, { cwd }));
+    return;
+  }
+
+  const reason = required(parsed.values, '--reason');
+  const commentFile = required(parsed.values, '--comment-file');
+  if (!reason || !commentFile) { fail('dismiss requires --reason and --comment-file'); return; }
+  const comment = readCommentFile(commentFile, cwd);
+  if (!comment.ok) {
+    finish({ status: 'failed', changed: false, error: comment.error });
+    return;
+  }
+  finish(await dismissSecurityAlert(
+    { kind: kind as 'dependabot' | 'code-scanning', number, reason, comment: comment.value },
+    { cwd }
+  ));
+}
+
+export { platformSecurity };

package/lib/internal/pr-review-grade.ts

@@ -11,6 +11,7 @@ import {
 import type { DecisionRecord, HostResolution } from '../pr-review/evidence-grading.ts';
 import { inspectPlatformPullRequestByNumber } from '../platform/pull-requests.ts';
 import { verifyInProcess } from '../task/verification-engine.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
 
 const USAGE = `Usage: agent-infra-internal pr-review-grade decide --input-file <path|-> [--cwd <path>]
        agent-infra-internal pr-review-grade resolve-host --pr <token> [--cwd <path>]
@@ -50,9 +51,14 @@ function readInput(value: string, cwd: string): string {
 }
 
 async function prReviewGrade(args: string[] = []): Promise<void> {
+  if (!ensureInternalHandlerRoute('pr-review-grade', args)) return;
   if (args[0] === '--help' || args[0] === '-h') { process.stdout.write(USAGE); return; }
   const operation = args[0];
-  if (!operation || !['decide', 'resolve-host', 'verify-artifact'].includes(operation)) { fail('a valid operation is required'); return; }
+  if (!operation || ![
+    internalHandlerRoute('pr-review-grade', 'decide', operation),
+    internalHandlerRoute('pr-review-grade', 'resolve-host', operation),
+    internalHandlerRoute('pr-review-grade', 'verify-artifact', operation)
+  ].some(Boolean)) { fail('a valid operation is required'); return; }
   const parsed = parseFlags(args, 1);
   if (parsed.error) { fail(parsed.error); return; }
   const values = parsed.values;
@@ -65,7 +71,7 @@ async function prReviewGrade(args: string[] = []): Promise<void> {
   const unexpected = Object.keys(values).find((key) => !allowed[operation]!.includes(key));
   if (unexpected) { fail(`${operation} does not accept '--${unexpected}'`); return; }
 
-  if (operation === 'decide') {
+  if (internalHandlerRoute('pr-review-grade', 'decide', operation)) {
     if (typeof values.inputFile !== 'string') { fail('decide requires --input-file'); return; }
     let input: unknown;
     try {
@@ -84,7 +90,7 @@ async function prReviewGrade(args: string[] = []): Promise<void> {
     return;
   }
 
-  if (operation === 'resolve-host') {
+  if (internalHandlerRoute('pr-review-grade', 'resolve-host', operation)) {
     const pr = typeof values.pr === 'string' ? values.pr : '';
     if (!pr) { fail('resolve-host requires --pr <token>'); return; }
     const inspected = await inspectPlatformPullRequestByNumber(pr, { cwd });
@@ -109,6 +115,7 @@ async function prReviewGrade(args: string[] = []): Promise<void> {
     return;
   }
 
+  if (!internalHandlerRoute('pr-review-grade', 'verify-artifact', operation)) { fail('operation is not registered'); return; }
   if (typeof values.artifactFile !== 'string') { fail('verify-artifact requires --artifact-file'); return; }
   const artifactPath = path.resolve(cwd, values.artifactFile);
   let result: Record<string, unknown>;

package/lib/internal/release-workflow.ts

@@ -9,6 +9,7 @@ import { inspectPlatformRelease, reconcileReleaseMilestones } from '../platform/
 import { inspectHomebrewChannel, inspectNpmChannel } from '../release/channels.ts';
 import { releaseSnapshot } from '../release/workflow.ts';
 import type { PostReleaseFacts, ReleaseFacts } from '../release/workflow.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
 
 function command(cwd: string, executable: string, args: string[]) {
   return spawnSync(executable, args, { cwd, encoding: 'utf8', stdio: ['ignore', 'pipe', 'pipe'] });
@@ -312,17 +313,24 @@ async function releaseWorkflow(args: string[] = []): Promise<void> {
   const version = String(rawVersion || '').replace(/^v/, '');
   const cwdIndex = rest.indexOf('--cwd');
   const cwd = path.resolve(cwdIndex >= 0 && rest[cwdIndex + 1] ? rest[cwdIndex + 1]! : process.cwd());
-  if (!['inspect', 'prepare', 'publish', 'post-prepare', 'post-publish'].includes(action || '') || !semver.valid(version)) {
+  if (![
+    internalHandlerRoute('release-workflow', 'inspect', action || ''),
+    internalHandlerRoute('release-workflow', 'prepare', action || ''),
+    internalHandlerRoute('release-workflow', 'publish', action || ''),
+    internalHandlerRoute('release-workflow', 'post-prepare', action || ''),
+    internalHandlerRoute('release-workflow', 'post-publish', action || '')
+  ].some(Boolean) || !semver.valid(version)) {
     process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, error: { code: 'RELEASE_INPUT_INVALID', message: 'Usage: release-workflow inspect|prepare|publish|post-prepare|post-publish <version>' } })}\n`);
     process.exitCode = 1; return;
   }
+  if (!ensureInternalHandlerRoute('release-workflow', args)) return;
   const before = releaseSnapshot(version, await inspectFacts(cwd, version));
-  if (action === 'inspect') { process.stdout.write(`${JSON.stringify({ status: 'no-op', changed: false, snapshot: before, error: null })}\n`); return; }
+  if (internalHandlerRoute('release-workflow', 'inspect', action || '')) { process.stdout.write(`${JSON.stringify({ status: 'no-op', changed: false, snapshot: before, error: null })}\n`); return; }
   if (before.facts.localTagConflict) {
     process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, snapshot: before, error: { code: 'GIT_TAG_CONFLICT', message: `Tag v${version} is not reachable from HEAD` } })}\n`);
     process.exitCode = 1; return;
   }
-  if (action === 'prepare') {
+  if (internalHandlerRoute('release-workflow', 'prepare', action || '')) {
     if (before.phase !== 'unprepared') {
       const milestones = await reconcileReleaseMilestones(version, { cwd });
       process.stdout.write(`${JSON.stringify({ ...milestones, snapshot: before })}\n`);
@@ -352,14 +360,14 @@ async function releaseWorkflow(args: string[] = []): Promise<void> {
     process.stdout.write(`${JSON.stringify({ status, changed: true, snapshot, operations: milestones?.operations ?? [], error: tagged.status !== 0 ? { code: 'GIT_TAG_FAILED', message: String(tagged.stderr) } : milestones?.error ?? null })}\n`);
     process.exitCode = failed ? 1 : blocked ? 2 : 0; return;
   }
-  if (action === 'publish') {
+  if (internalHandlerRoute('release-workflow', 'publish', action || '')) {
     if (!before.facts.localTag || !['prepared', 'partially-published'].includes(before.phase)) { process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, snapshot: before, error: { code: 'RELEASE_PHASE_INVALID', message: `Cannot publish from ${before.phase}` } })}\n`); process.exitCode = 1; return; }
     const branch = git(cwd, ['branch', '--show-current']) || '';
     const refs = [...(!before.facts.remoteBranch ? [branch] : []), ...(!before.facts.remoteTag ? [`refs/tags/v${version}`] : [])];
     const result = pushGitRefs({ cwd, remote: 'origin', refs });
     process.stdout.write(`${JSON.stringify({ ...result, snapshot: releaseSnapshot(version, await inspectFacts(cwd, version)) })}\n`); process.exitCode = result.status === 'failed' ? 1 : result.status === 'degraded' ? 2 : 0; return;
   }
-  if (action === 'post-publish') {
+  if (internalHandlerRoute('release-workflow', 'post-publish', action || '')) {
     const expectedValues = optionValues(rest, '--expected-sha256');
     if (expectedValues.length !== 1 || !/^sha256:[0-9a-f]{64}$/.test(expectedValues[0]!)) {
       process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, snapshot: before, error: { code: 'RELEASE_INPUT_INVALID', message: 'post-publish requires one --expected-sha256 sha256:<64 hex>' } })}\n`);
@@ -394,6 +402,10 @@ async function releaseWorkflow(args: string[] = []): Promise<void> {
     process.exitCode = blocked ? 2 : 1; return;
   }
 
+  if (!internalHandlerRoute('release-workflow', 'post-prepare', action || '')) {
+    process.stdout.write(`${JSON.stringify({ status: 'failed', changed: false, snapshot: before, error: { code: 'RELEASE_INPUT_INVALID', message: 'release-workflow action is not registered' } })}\n`);
+    process.exitCode = 1; return;
+  }
   if (before.phase === 'complete') {
     process.stdout.write(`${JSON.stringify({ status: 'no-op', changed: false, snapshot: before, error: null })}\n`);
     return;

package/lib/internal/sandbox-control.ts

@@ -12,6 +12,8 @@ import {
   controllerProofFromContext,
   verifyCodexSandboxControllerContextWithWarnings
 } from '../agent-clients/adapters/codex-lifecycle/sandbox-controller.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
+import { parseTaskCreateResult, taskCreateExitCode } from '../task/create-service.ts';
 
 function isCanonicalCodexPrepare(args: readonly string[]): boolean {
   if (args[1] !== 'prepare') return false;
@@ -45,6 +47,17 @@ function writeClientError(error: SandboxControlClientError): void {
   if (error.requestId) process.stderr.write(`SANDBOX_CONTROL_REQUEST_ID: ${error.requestId}\n`);
 }
 
+function recoveredExitCode(response: Awaited<ReturnType<typeof recoverSandboxControl>>): number {
+  if (response.phase !== 'completed') return response.exitCode ?? 1;
+  try {
+    const result = parseTaskCreateResult(JSON.parse(response.stdout));
+    if (result.control?.requestId === response.id) return taskCreateExitCode(result);
+  } catch {
+    // Other control families use their own result envelopes.
+  }
+  return response.exitCode ?? 1;
+}
+
 function sandboxFinalizationClient(args: string[]): void {
   if (args.length !== 4 || !args[0] || args[1] !== 'complete' || args[2] !== '--agent' || !args[3]) {
     const error = { code: 'TASK_FINALIZATION_PAYLOAD_INVALID', message: 'task ref, complete intent, and --agent are required', retryable: false };
@@ -87,8 +100,9 @@ function sandboxFinalizationClient(args: string[]): void {
 }
 
 async function sandboxControl(args: string[]): Promise<void> {
+  if (!ensureInternalHandlerRoute('sandbox-control', args)) return;
   const [operation, ...rest] = args;
-  if (operation === 'serve') {
+  if (internalHandlerRoute('sandbox-control', 'serve', operation ?? '')) {
     const manifestIndex = rest.indexOf('--manifest');
     const manifest = manifestIndex >= 0 ? rest[manifestIndex + 1] : undefined;
     if (!manifest) throw new Error('sandbox-control serve requires --manifest');
@@ -104,7 +118,7 @@ async function sandboxControl(args: string[]): Promise<void> {
     }
     return;
   }
-  if (operation === 'execute') {
+  if (internalHandlerRoute('sandbox-control', 'execute', operation ?? '')) {
     const requestIndex = rest.indexOf('--request');
     const nonceIndex = rest.indexOf('--nonce');
     const request = requestIndex >= 0 ? rest[requestIndex + 1] : undefined;
@@ -113,7 +127,7 @@ async function sandboxControl(args: string[]): Promise<void> {
     await runSandboxControlExecutor(request, nonce);
     return;
   }
-  if (operation === 'recover') {
+  if (internalHandlerRoute('sandbox-control', 'recover', operation ?? '')) {
     if (rest.length !== 1 || !rest[0]) throw new Error('sandbox-control recover requires <request-id>');
     let response;
     try {
@@ -128,10 +142,10 @@ async function sandboxControl(args: string[]): Promise<void> {
     process.stderr.write(response.stderr);
     process.exitCode = response.phase === 'rejected'
       ? response.error?.retryable ? 75 : 1
-      : response.exitCode ?? 1;
+      : recoveredExitCode(response);
     return;
   }
-  if (operation === 'client') {
+  if (internalHandlerRoute('sandbox-control', 'client', operation ?? '')) {
     const [family = '', ...commandArgs] = rest;
     if (family === 'task-finalization') {
       sandboxFinalizationClient(commandArgs);

package/lib/internal/task-activity.ts

@@ -9,6 +9,7 @@ import type {
 import { consumeHumanOverride, failureId, overrideDryRunConflict } from '../task/human-override.ts';
 import { resolveTaskRef } from '../task/resolve-ref.ts';
 import { TaskExecutionLockError, withTaskExecutionLock } from '../task/task-execution-lock.ts';
+import { ensureInternalHandlerRoute, internalHandlerRoute } from './cli-route-inventory.ts';
 
 const USAGE = `Usage: agent-infra-internal task-activity <task-ref> pr-review-inspect
        agent-infra-internal task-activity <task-ref> pr-review-start --agent <agent> --artifact <canonical.md> --head <40hex> [--dry-run]
@@ -44,9 +45,16 @@ function usageFailure(message: string): void {
 }
 
 async function taskActivity(args: string[] = []): Promise<void> {
+  if (!ensureInternalHandlerRoute('task-activity', args)) return;
   if (args[0] === '--help' || args[0] === '-h') { process.stdout.write(USAGE); return; }
   const [taskRef, kind] = args;
-  if (!taskRef || taskRef.startsWith('--') || !kind || !OPERATIONS.has(kind)) {
+  if (!taskRef || taskRef.startsWith('--') || !kind || !OPERATIONS.has(kind)
+    || ![
+      internalHandlerRoute('task-activity', 'pr-review-inspect', kind),
+      internalHandlerRoute('task-activity', 'pr-review-start', kind),
+      internalHandlerRoute('task-activity', 'pr-review-complete', kind),
+      internalHandlerRoute('task-activity', 'pr-review-terminate', kind)
+    ].some(Boolean)) {
     usageFailure('task ref and a supported PR review intent are required');
     return;
   }
@@ -72,7 +80,7 @@ async function taskActivity(args: string[] = []): Promise<void> {
   const allowed = ALLOWED[kind]!;
   const unexpected = Object.keys(values).find((key) => !allowed.has(key));
   if (unexpected) { usageFailure(`${kind} does not accept '${unexpected}'`); return; }
-  if (kind === 'pr-review-inspect') {
+  if (internalHandlerRoute('task-activity', 'pr-review-inspect', kind)) {
     const result = inspectPrReviewActivity(values as PrReviewInspectIntent);
     process.stdout.write(`${JSON.stringify(result)}\n`);
     if (result.status === 'failed') process.exitCode = 1;
@@ -81,9 +89,9 @@ async function taskActivity(args: string[] = []): Promise<void> {
   for (const key of COMMON) {
     if (values[key] === undefined) { usageFailure(`${kind} requires '${key}'`); return; }
   }
-  const operationRequired = kind === 'pr-review-complete'
+  const operationRequired = internalHandlerRoute('task-activity', 'pr-review-complete', kind)
     ? ['verdict', 'blockers', 'major', 'minor']
-    : kind === 'pr-review-terminate' ? ['outcome', 'reason'] : [];
+    : internalHandlerRoute('task-activity', 'pr-review-terminate', kind) ? ['outcome', 'reason'] : [];
   for (const key of operationRequired) {
     if (values[key] === undefined) { usageFailure(`${kind} requires '${key}'`); return; }
   }