@atolis-hq/wake 0.1.7 → 0.1.8

package/dist/src/core/policy-engine.js

@@ -1,4 +1,5 @@
 import { awaitingApprovalRunnerSentinel, failedRunnerSentinel } from '../domain/stages.js';
+import { resolveCustomCommand } from '../domain/custom-commands.js';
 import { builtInDefaultWorkflowDefinition, chooseAction as chooseWorkflowAction, } from '../domain/workflows.js';
 function isAwaitingApproval(issue) {
     const context = issue.context;
@@ -10,12 +11,10 @@ function isAwaitingApproval(issue) {
 const approvedCommandPattern = /^\/approved\b/i;
 const changesCommandPattern = /^\/changes\b/i;
 const questionCommandPattern = /^\/question\b/i;
-const codeReviewCommandPattern = /^\/codereview\b/i;
 // The action Wake runs when a correlated PR gets new reviewer feedback while
 // the work item is awaiting approval. Not configurable per workflow: it's a
 // lateral response to a PR surface, not a workflow stage.
 const reviewFeedbackAction = 'revise';
-const codeReviewAction = 'codereview';
 function matchesCommand(body, pattern) {
     return body.split(/\r?\n/).some((line) => pattern.test(line.trim()));
 }
@@ -173,13 +172,8 @@ export function createPolicyEngine() {
             }
             return reviewFeedbackAction;
         },
-        resolveCodeReviewRequest(issue) {
-            const latestHumanComment = latestUnhandledHumanComment(issue);
-            if (latestHumanComment === undefined ||
-                !matchesCommand(latestHumanComment.body, codeReviewCommandPattern)) {
-                return null;
-            }
-            return codeReviewAction;
+        resolveCustomCommandRequest(issue, config) {
+            return resolveCustomCommand(issue, config);
         },
         qualifiesForMint(unresolved, config) {
             const resourceUri = unresolved.sourceRefs.resourceUri;

package/dist/src/core/projection-updater.js

@@ -1,6 +1,7 @@
 import { CORRELATION_PRIMARY_CONFLICT_EVENT, CORRELATION_REGISTERED_EVENT, CORRELATION_RETRACTED_EVENT, UNRESOLVED_WORK_ITEM_KEY, parseIssueStateRecord, } from '../domain/schema.js';
 import { doneRunnerSentinel, stageFromLabels } from '../domain/stages.js';
 import { builtInDefaultWorkflowDefinition, defaultWorkflowName, workflowStageVocabulary, } from '../domain/workflows.js';
+import { isCustomCommandAction } from '../domain/custom-commands.js';
 import { createEventEnvelope } from '../lib/event-log.js';
 function stringArrayFromPayload(value) {
     return Array.isArray(value)
@@ -150,7 +151,10 @@ async function applyEvent(current, event, ctx, config) {
         const isForwardProgression = payload.nextStage !== undefined && payload.nextStage !== current.wake.stage;
         const stageChanged = payload.nextStage !== undefined && payload.nextStage !== current.wake.stage;
         const isFailed = payload.sentinel === 'FAILED';
-        const isCompletedCodeReview = payload.action === 'codereview' && payload.sentinel === doneRunnerSentinel;
+        const isCompletedCustomCommand = payload.action !== undefined &&
+            payload.sentinel === doneRunnerSentinel &&
+            config !== undefined &&
+            isCustomCommandAction(payload.action, config);
         const shouldClearSession = isForwardProgression || isFailed;
         return parseIssueStateRecord({
             ...current,
@@ -160,15 +164,15 @@ async function applyEvent(current, event, ctx, config) {
                 ...(payload.handledCommentId === undefined
                     ? {}
                     : { lastHandledCommentId: payload.handledCommentId }),
-                ...(payload.sentinel === undefined || isCompletedCodeReview
+                ...(payload.sentinel === undefined || isCompletedCustomCommand
                     ? {}
                     : { lastRunSentinel: payload.sentinel }),
-                ...(payload.action === undefined || isCompletedCodeReview
+                ...(payload.action === undefined || isCompletedCustomCommand
                     ? {}
                     : { lastRunAction: payload.action }),
                 ...(payload.sentinel === doneRunnerSentinel &&
                     payload.action !== undefined &&
-                    !isCompletedCodeReview
+                    !isCompletedCustomCommand
                     ? { lastCompletedAction: payload.action }
                     : {}),
                 // Remembered so the approval path knows which action to resume or

package/dist/src/core/tick-runner.js

@@ -12,6 +12,7 @@ import { awaitingApprovalRunnerSentinel, stageLabelForStage } from '../domain/st
 import { chooseAction as chooseWorkflowAction, isKnownWorkflowStage, workflowChangedBlockReason, workflowForProjection, workflowNameForProjection, } from '../domain/workflows.js';
 import { createEventEnvelope } from '../lib/event-log.js';
 import { branchNameForIssue } from '../domain/branch-naming.js';
+import { customCommandWorkspace, isCustomCommandAction } from '../domain/custom-commands.js';
 import { resolveQuotaPauseUntil } from './quota-backoff.js';
 function latestHumanCommentId(candidate) {
     const human = candidate.comments.filter((c) => !c.isBotAuthored);
@@ -36,8 +37,11 @@ function latestHumanCommentId(candidate) {
 function isReviewThreadResourceUri(resourceUri) {
     return resourceUri.split(':')[1] === 'pr-review-thread';
 }
-function isLateralReadOnlyAction(action) {
-    return action === 'codereview';
+function isLateralReadOnlyAction(action, config) {
+    return isCustomCommandAction(action, config);
+}
+function workspaceModeForCustomCommand(action, config) {
+    return customCommandWorkspace(action, config);
 }
 function isFreshTriggeringComment(candidate) {
     const context = candidate.context;
@@ -212,11 +216,11 @@ export function createTickRunner(deps) {
             return false;
         }
         if (isAwaitingApproval(projection)) {
-            return (policy.resolveCodeReviewRequest(projection) !== null ||
+            return (policy.resolveCustomCommandRequest(projection, deps.config) !== null ||
                 policy.resolveApprovalTransition(projection) !== null ||
                 policy.resolvePendingReviewFeedback(projection) !== null);
         }
-        const nextAction = policy.resolveCodeReviewRequest(projection) ??
+        const nextAction = policy.resolveCustomCommandRequest(projection, deps.config)?.action ??
             policy.chooseAction(projection, workflow) ??
             policy.chooseRetryActionAfterHumanReply(projection);
         return nextAction !== null && policy.needsWakeAction(projection, workflow);
@@ -952,7 +956,7 @@ export function createTickRunner(deps) {
                 if (isAwaitingApproval(issue)) {
                     return policy.needsWakeAction(issue, workflow);
                 }
-                const nextAction = policy.resolveCodeReviewRequest(issue) ??
+                const nextAction = policy.resolveCustomCommandRequest(issue, deps.config)?.action ??
                     policy.chooseAction(issue, workflow) ??
                     policy.chooseRetryActionAfterHumanReply(issue);
                 return nextAction !== null && policy.needsWakeAction(issue, workflow);
@@ -966,14 +970,16 @@ export function createTickRunner(deps) {
             }
             const workflowName = workflowNameForProjection(candidate, deps.config);
             let action;
+            let command;
             let claimedStage = candidate.wake.stage;
             let workspaceMode = 'none';
             if (isAwaitingApproval(candidate)) {
-                const codeReviewRequest = policy.resolveCodeReviewRequest(candidate);
-                if (codeReviewRequest !== null) {
-                    action = codeReviewRequest;
+                const customCommandRequest = policy.resolveCustomCommandRequest(candidate, deps.config);
+                if (customCommandRequest !== null) {
+                    action = customCommandRequest.action;
+                    command = customCommandRequest.command;
                     claimedStage = candidate.wake.stage;
-                    workspaceMode = 'read-only';
+                    workspaceMode = customCommandRequest.workspace;
                 }
                 else {
                     const approvalResolution = policy.resolveApprovalTransition(candidate);
@@ -1054,7 +1060,7 @@ export function createTickRunner(deps) {
                 // scratch (#258 follow-up incident: a FAILED `revise` run fell
                 // back to a full fresh `implement` run and lost the PR-feedback
                 // context).
-                const nextAction = policy.resolveCodeReviewRequest(candidate) ??
+                const nextAction = policy.resolveCustomCommandRequest(candidate, deps.config)?.action ??
                     policy.chooseRetryActionAfterHumanReply(candidate) ??
                     workflowAction?.action ??
                     null;
@@ -1062,10 +1068,10 @@ export function createTickRunner(deps) {
                     return { status: 'idle' };
                 }
                 action = nextAction;
+                command = policy.resolveCustomCommandRequest(candidate, deps.config)?.command;
                 claimedStage = workflowAction?.stage ?? candidate.wake.stage;
-                workspaceMode = isLateralReadOnlyAction(action)
-                    ? 'read-only'
-                    : (workflowAction?.workspace ?? 'none');
+                workspaceMode =
+                    workspaceModeForCustomCommand(action, deps.config) ?? workflowAction?.workspace ?? 'none';
             }
             // Resolve routing (with sideways fallback across quota-paused runners,
             // #67) before claiming a run, so a fully-paused tier costs nothing more
@@ -1076,6 +1082,7 @@ export function createTickRunner(deps) {
                 stage: claimedStage,
                 action,
                 workflowName,
+                ...(command === undefined ? {} : { command }),
                 now: tickStartedAt,
                 ...(ledgerAtStart === null ? {} : { ledger: ledgerAtStart }),
             });
@@ -1162,7 +1169,7 @@ export function createTickRunner(deps) {
                 // the protocol; treat it as AWAITING_APPROVAL so the gate is enforced.
                 const skipApproval = runnerResult.metadata?.skipApproval;
                 const sentinel = rawSentinel === 'DONE' && skipApproval === false ? 'AWAITING_APPROVAL' : rawSentinel;
-                const nextStage = isLateralReadOnlyAction(action) && sentinel === 'DONE'
+                const nextStage = isLateralReadOnlyAction(action, deps.config) && sentinel === 'DONE'
                     ? null
                     : lifecycle.nextStageFromSentinel(claimedStage, sentinel, workflow);
                 const finishedAt = deps.clock.now().toISOString();

package/dist/src/domain/custom-commands.js

@@ -0,0 +1,58 @@
+export const reservedCommandNames = ['approved', 'changes', 'question'];
+function latestUnhandledHumanComment(issue) {
+    const context = issue.context;
+    const handledCommentId = typeof context.lastHandledCommentId === 'string' ? context.lastHandledCommentId : undefined;
+    const lastBotIndex = issue.comments.reduce((acc, c, i) => (c.isBotAuthored ? i : acc), -1);
+    const humanCommentsAfterBot = issue.comments
+        .slice(lastBotIndex + 1)
+        .filter((c) => !c.isBotAuthored);
+    const latestHumanComment = humanCommentsAfterBot.at(-1);
+    if (latestHumanComment === undefined || latestHumanComment.id === handledCommentId) {
+        return undefined;
+    }
+    return latestHumanComment;
+}
+function commandNameFromBody(body) {
+    for (const line of body.split(/\r?\n/)) {
+        const match = /^\/([A-Za-z0-9_.-]+)\b/.exec(line.trim());
+        if (match?.[1] !== undefined) {
+            return match[1].toLowerCase();
+        }
+    }
+    return null;
+}
+export function customCommandAction(command, config) {
+    const entry = Object.entries(config.commands).find(([name]) => name.toLowerCase() === command.toLowerCase());
+    if (entry === undefined) {
+        return undefined;
+    }
+    const [name, definition] = entry;
+    return definition.action ?? name;
+}
+export function isCustomCommandAction(action, config) {
+    return Object.entries(config.commands).some(([name, definition]) => (definition.action ?? name) === action);
+}
+export function customCommandWorkspace(action, config) {
+    return Object.entries(config.commands).find(([name, definition]) => (definition.action ?? name) === action)?.[1].workspace;
+}
+export function resolveCustomCommand(issue, config) {
+    const latestHumanComment = latestUnhandledHumanComment(issue);
+    if (latestHumanComment === undefined) {
+        return null;
+    }
+    const command = commandNameFromBody(latestHumanComment.body);
+    if (command === null) {
+        return null;
+    }
+    const definitionEntry = Object.entries(config.commands).find(([name]) => name.toLowerCase() === command);
+    if (definitionEntry === undefined) {
+        return null;
+    }
+    const [configuredName, definition] = definitionEntry;
+    return {
+        action: definition.action ?? configuredName,
+        command: configuredName,
+        comment: latestHumanComment,
+        workspace: definition.workspace,
+    };
+}

package/dist/src/domain/runner-routing.js

@@ -13,6 +13,11 @@ export function maxConfiguredRunnerTimeoutMs(config) {
             activeRunnerNames.add(stageRoute.runner);
         }
     }
+    for (const commandRoute of Object.values(config.commands)) {
+        if (commandRoute.runner !== undefined) {
+            activeRunnerNames.add(commandRoute.runner);
+        }
+    }
     const registryTimeouts = [...activeRunnerNames]
         .map((name) => config.runners[name])
         .map((entry) => (entry === undefined || entry.kind === 'fake' ? undefined : entry.timeoutMs))
@@ -57,6 +62,23 @@ export function resolveRunnerRouting(input) {
     if (workflow === undefined) {
         throw new Error(`Unknown workflow "${workflowName}".`);
     }
+    const commandRouteEntry = input.command === undefined
+        ? Object.entries(input.config.commands).find(([name, route]) => (route.action ?? name) === input.action)
+        : Object.entries(input.config.commands).find(([name]) => name.toLowerCase() === input.command?.toLowerCase());
+    const commandRoute = commandRouteEntry?.[1];
+    if (commandRoute?.runner !== undefined) {
+        const runnerName = commandRoute.runner;
+        const entry = input.config.runners[runnerName];
+        if (entry === undefined) {
+            throw new Error(`Command ${input.action} pins unknown runner "${runnerName}".`);
+        }
+        return {
+            runnerName,
+            runnerKind: entry.kind,
+            ...(commandRoute.tier === undefined ? {} : { tier: commandRoute.tier }),
+            reason: `command ${input.action} pins runner ${runnerName}`,
+        };
+    }
     const stageRoute = workflow.stages[input.stage];
     if (stageRoute?.runner !== undefined) {
         const entry = input.config.runners[stageRoute.runner];
@@ -70,7 +92,7 @@ export function resolveRunnerRouting(input) {
             reason: `stage ${input.stage} pins runner ${stageRoute.runner}`,
         };
     }
-    const tier = stageRoute?.tier ?? input.config.defaultTier;
+    const tier = commandRoute?.tier ?? stageRoute?.tier ?? input.config.defaultTier;
     const candidates = input.config.tiers[tier];
     if (candidates === undefined || candidates.length === 0) {
         throw new Error(`Stage ${input.stage} routes to unknown or empty tier "${tier}".`);
@@ -101,8 +123,10 @@ export function resolveRunnerRouting(input) {
             ? `tier ${tier} recovery probe on ${runnerName} (estimated pause not yet elapsed)`
             : isFallback
                 ? `tier ${tier} fell back to ${runnerName} (higher-priority candidates paused)`
-                : stageRoute?.tier === undefined
-                    ? `defaultTier ${tier} selected runner ${runnerName}`
-                    : `stage ${input.stage} tier ${tier} selected runner ${runnerName}`,
+                : commandRoute?.tier !== undefined
+                    ? `command ${input.action} tier ${tier} selected runner ${runnerName}`
+                    : stageRoute?.tier === undefined
+                        ? `defaultTier ${tier} selected runner ${runnerName}`
+                        : `stage ${input.stage} tier ${tier} selected runner ${runnerName}`,
     };
 }

package/dist/src/domain/schema.js

@@ -2,6 +2,7 @@ import { existsSync } from 'node:fs';
 import { dirname, join } from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { z } from 'zod';
+import { reservedCommandNames } from './custom-commands.js';
 import { runnerSentinelValues } from './stages.js';
 import { correlationProvenanceSchema, correlationRelationSchema, correlationRoleSchema, resourceUriSchema, } from './resource-uri.js';
 const isoTimestampSchema = z.string().datetime({ offset: true });
@@ -308,6 +309,9 @@ const workflowStageSchema = stageRouteSchema.extend({
     workspace: workflowWorkspaceSchema,
     onDone: identifierSchema,
 });
+const customCommandSchema = stageRouteSchema.extend({
+    workspace: workflowWorkspaceSchema.default('read-only'),
+});
 const workflowDefinitionSchema = z.object({
     entryStage: identifierSchema.optional(),
     stages: z.record(identifierSchema, workflowStageSchema),
@@ -511,6 +515,13 @@ export const wakeConfigSchema = z
             },
         },
     }),
+    commands: z.record(identifierSchema, customCommandSchema).default({
+        codereview: {
+            action: 'codereview',
+            workspace: 'read-only',
+            tier: 'standard',
+        },
+    }),
     stages: z.record(z.string(), stageRouteSchema).default({
         queue: { action: 'refine', tier: 'light' },
         implement: { action: 'implement', tier: 'standard' },
@@ -615,6 +626,7 @@ export const wakeConfigSchema = z
     .superRefine((config, ctx) => {
     const promptsRoot = config.paths.promptsRoot ?? defaultPromptsRoot();
     const workflowEntries = Object.entries(config.workflows);
+    const commandEntries = Object.entries(config.commands);
     if (workflowEntries.length === 0) {
         ctx.addIssue({
             code: z.ZodIssueCode.custom,
@@ -690,6 +702,22 @@ export const wakeConfigSchema = z
             });
         }
     }
+    for (const [commandName, command] of commandEntries) {
+        if (reservedCommandNames.includes(commandName.toLowerCase())) {
+            ctx.addIssue({
+                code: z.ZodIssueCode.custom,
+                path: ['commands', commandName],
+                message: `Command "/${commandName}" is reserved for Wake approval control.`,
+            });
+        }
+        if (command.action === undefined && !promptExists(promptsRoot, commandName)) {
+            ctx.addIssue({
+                code: z.ZodIssueCode.custom,
+                path: ['commands', commandName, 'action'],
+                message: `Command "/${commandName}" omits action but no prompts/${commandName}.md template exists.`,
+            });
+        }
+    }
 });
 export const claudePrintResultSchema = z
     .object({

package/dist/src/version.js

@@ -1 +1 @@
-export const wakeVersion = "0.1.7+gaa68b14";
+export const wakeVersion = "0.1.8+g8f32cc9";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atolis-hq/wake",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Local autonomous agent control plane for software development",
   "license": "Apache-2.0",
   "repository": {