535 lines
16 KiB
TypeScript
535 lines
16 KiB
TypeScript
import * as path from "node:path";
|
|
import type {
|
|
ExtensionAPI,
|
|
ExtensionContext,
|
|
} from "@earendil-works/pi-coding-agent";
|
|
import { Box, Text } from "@earendil-works/pi-tui";
|
|
import { parseTaskFile, updateTaskInFile } from "./src/parser";
|
|
import {
|
|
buildExecutionPlan,
|
|
buildSequentialPlan,
|
|
formatExecutionPlan,
|
|
getReadyTasks,
|
|
} from "./src/dag";
|
|
import { ProgressTracker } from "./src/progress";
|
|
import { buildPlanPrompt } from "./src/prompts";
|
|
import { formatReflections } from "./src/reflection";
|
|
import { executeBatch } from "./src/executor";
|
|
import {
|
|
loadConfig,
|
|
resolveTaskArg,
|
|
formatProgressStatus,
|
|
formatAllPRDsStatus,
|
|
findProgressFile,
|
|
} from "./src/utils";
|
|
|
|
const COMMANDS = ["status", "resume", "next", "reset"] as const;
|
|
|
|
/**
|
|
* Detect if a token looks like a file path rather than a subcommand.
|
|
* Matches: @path, /path, ./path, ../path, path/to/file, path.md, path.yaml
|
|
*/
|
|
function looksLikePath(token: string): boolean {
|
|
return (
|
|
token.startsWith("@") ||
|
|
token.startsWith("/") ||
|
|
token.startsWith("./") ||
|
|
token.startsWith("../") ||
|
|
token.includes("/") ||
|
|
token.endsWith(".md") ||
|
|
token.endsWith(".yaml") ||
|
|
token.endsWith(".yml")
|
|
);
|
|
}
|
|
|
|
// ─── Extension Entry ────────────────────────────────────────────────────────
|
|
|
|
export default function ralphLoopExtension(pi: ExtensionAPI): void {
|
|
// Register custom message renderer for ralph progress messages.
|
|
// Renders an expandable tool-call tree: collapsed shows last 3 + "N more",
|
|
// expanded (Ctrl+O) shows every tool call.
|
|
pi.registerMessageRenderer(
|
|
"ralph-progress",
|
|
(message, { expanded }, theme) => {
|
|
const details = message.details as
|
|
| {
|
|
phase?: string;
|
|
toolCalls?: Array<{ name: string; label: string }>;
|
|
}
|
|
| undefined;
|
|
|
|
const MAX_COLLAPSED = 3;
|
|
const lines: string[] = [];
|
|
|
|
// Header line — e.g. "✓ 05 · billing-subscriptions-trials (2m 14s)"
|
|
lines.push(String(message.content));
|
|
|
|
// Build tool-call tree
|
|
if (details?.toolCalls && details.toolCalls.length > 0) {
|
|
const all = details.toolCalls;
|
|
|
|
if (expanded) {
|
|
// Expanded: show ALL tool calls
|
|
for (let i = 0; i < all.length; i++) {
|
|
const entry = all[i];
|
|
const isLast = i === all.length - 1;
|
|
const branch = isLast ? " └── " : " ├── ";
|
|
const tag = theme.fg("accent", `[${entry.name}]`);
|
|
lines.push(`${branch}${tag} ${entry.label}`);
|
|
}
|
|
} else {
|
|
// Collapsed: last N + "X more"
|
|
const shown = all.slice(-MAX_COLLAPSED);
|
|
const remaining = all.length - shown.length;
|
|
|
|
if (remaining > 0) {
|
|
lines.push(theme.fg("dim", ` ├── ${remaining} more`));
|
|
}
|
|
|
|
for (let i = 0; i < shown.length; i++) {
|
|
const entry = shown[i];
|
|
const isLast = i === shown.length - 1;
|
|
const branch = isLast ? " └── " : " ├── ";
|
|
const tag = theme.fg("accent", `[${entry.name}]`);
|
|
lines.push(`${branch}${tag} ${entry.label}`);
|
|
}
|
|
}
|
|
}
|
|
|
|
const text = lines.join("\n");
|
|
const box = new Box(1, 1, (t) => theme.bg("customMessageBg", t));
|
|
box.addChild(new Text(text, 0, 0));
|
|
return box;
|
|
},
|
|
);
|
|
|
|
pi.registerCommand("ralph", {
|
|
description:
|
|
"Execute tasks from a task file using DAG-based dependency resolution",
|
|
handler: async (args: string, ctx: ExtensionContext) => {
|
|
const parts = (args || "").trim().split(/\s+/).filter(Boolean);
|
|
|
|
// Wraps pi.sendMessage() for posting status to the chat history.
|
|
// Uses "ralph-progress" customType with a "progress" phase so the
|
|
// renderer omits the label prefix entirely (no [INFO] etc.).
|
|
// Accepts an optional meta object with toolCalls for the expandable view.
|
|
const sendProgress = (
|
|
content: string,
|
|
meta?: { toolCalls?: Array<{ name: string; label: string }> },
|
|
) => {
|
|
pi.sendMessage({
|
|
customType: "ralph-progress",
|
|
content,
|
|
display: true,
|
|
details: { phase: "progress", toolCalls: meta?.toolCalls },
|
|
});
|
|
};
|
|
|
|
// If no args, show plan. If first token looks like a path (@path, /path, ./path),
|
|
// route to run so the execution mode prompt fires.
|
|
if (parts.length === 0) {
|
|
return handlePlan(ctx, parts);
|
|
}
|
|
if (looksLikePath(parts[0])) {
|
|
return handleRun(ctx, parts, sendProgress);
|
|
}
|
|
|
|
const command = parts[0];
|
|
switch (command) {
|
|
case "run":
|
|
return handleRun(ctx, parts.slice(1), sendProgress);
|
|
case "plan":
|
|
return handlePlan(ctx, parts.slice(1));
|
|
case "status":
|
|
return handleStatus(ctx, parts.slice(1));
|
|
case "resume":
|
|
return handleResume(ctx, parts.slice(1), sendProgress);
|
|
case "next":
|
|
return handleNext(ctx, parts.slice(1), sendProgress);
|
|
case "reset":
|
|
return handleReset(ctx, parts.slice(1));
|
|
default: {
|
|
// Auto-discover progress and offer resume
|
|
const found = findProgressFile(process.cwd());
|
|
if (found) {
|
|
ctx.ui.notify(
|
|
`Unknown command: ${command}\n\nFound existing progress in ${found.path}\nUse /ralph resume to continue.\n\nAvailable: ${COMMANDS.join(", ")}`,
|
|
"warning",
|
|
);
|
|
} else {
|
|
ctx.ui.notify(
|
|
`Unknown command: ${command}\nAvailable: ${COMMANDS.join(", ")}`,
|
|
"error",
|
|
);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
});
|
|
}
|
|
|
|
// ─── /ralph plan ─────────────────────────────────────────────────────────────
|
|
|
|
async function handlePlan(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
): Promise<void> {
|
|
const taskFile = resolveTaskArg(args[0] || "README.md", process.cwd());
|
|
const project = parseTaskFile(taskFile);
|
|
|
|
const planPrompt = buildPlanPrompt(project);
|
|
const plan = buildExecutionPlan(project, new Set());
|
|
const formatted = formatExecutionPlan(plan);
|
|
|
|
ctx.ui.notify(`${planPrompt}\n\n${formatted}`, "info");
|
|
}
|
|
|
|
// ─── /ralph run ──────────────────────────────────────────────────────────────
|
|
|
|
async function handleRun(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
sendChatMessage?: (content: string) => void,
|
|
): Promise<void> {
|
|
const taskFile = resolveTaskArg(args[0] || "README.md", process.cwd());
|
|
|
|
// If targeting a specific task file and there's existing progress for it,
|
|
// auto-resume instead of starting fresh
|
|
const existingProgress = findProgressFile(process.cwd(), taskFile);
|
|
if (existingProgress) {
|
|
return handleResume(ctx, [args[0]!], sendChatMessage);
|
|
}
|
|
|
|
// No existing progress for this task — check for any progress at all
|
|
const found = findProgressFile(process.cwd());
|
|
if (found && !args[0]) {
|
|
// Offer to resume instead of starting fresh
|
|
const shouldResume = await ctx.ui.select(
|
|
"Found existing ralph progress. Resume?",
|
|
["Yes, resume", "No, start fresh"],
|
|
);
|
|
|
|
if (shouldResume?.startsWith("Yes")) {
|
|
return handleResume(ctx, [], sendChatMessage);
|
|
}
|
|
}
|
|
|
|
const project = parseTaskFile(taskFile);
|
|
|
|
// Determine projectDir: prefer existing .ralph/ location, otherwise use cwd
|
|
const projectDir = found
|
|
? path.dirname(path.dirname(found.path))
|
|
: process.cwd();
|
|
|
|
const config = loadConfig(projectDir);
|
|
const progress = new ProgressTracker(projectDir, taskFile);
|
|
|
|
// Set initial status
|
|
ctx.ui.setStatus(
|
|
"ralph",
|
|
`Starting ${project.tasks.length} tasks from ${path.basename(taskFile)}`,
|
|
);
|
|
|
|
const completed = new Set(progress.getCompletedTaskIds());
|
|
|
|
// Ask user for execution mode
|
|
const mode = await ctx.ui.select("Execution mode for this run?", [
|
|
"Parallel (DAG-optimized)",
|
|
"Sequential (one at a time)",
|
|
]);
|
|
const useParallel = mode?.startsWith("Parallel");
|
|
|
|
// Sequential mode: use buildSequentialPlan to avoid 29-task mega-batches
|
|
const plan = useParallel
|
|
? buildExecutionPlan(project, completed)
|
|
: buildSequentialPlan(project, completed);
|
|
|
|
for (const batch of plan.batches) {
|
|
if (progress.getState().paused) {
|
|
ctx.ui.notify(
|
|
"Execution paused. Use /ralph resume to continue.",
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
|
|
await executeBatch(
|
|
batch.batchIndex,
|
|
batch.tasks,
|
|
project,
|
|
config,
|
|
progress,
|
|
ctx as any,
|
|
{ parallel: useParallel },
|
|
sendChatMessage,
|
|
projectDir,
|
|
);
|
|
|
|
for (const task of batch.tasks) {
|
|
const status = progress.getTaskStatus(task.id);
|
|
updateTaskInFile(taskFile, task.id, status);
|
|
}
|
|
}
|
|
|
|
const state = progress.getState();
|
|
const output = formatProgressStatus(state);
|
|
|
|
const reflections = progress.getAllReflections();
|
|
if (reflections.length > 0) {
|
|
ctx.ui.notify(`${output}\n\n${formatReflections(reflections)}`, "info");
|
|
return;
|
|
}
|
|
|
|
ctx.ui.notify(output, "info");
|
|
}
|
|
|
|
// ─── /ralph status ───────────────────────────────────────────────────────────
|
|
|
|
async function handleStatus(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
): Promise<void> {
|
|
if (args[0]) {
|
|
const taskFile = resolveTaskArg(args[0], process.cwd());
|
|
const existingProgress = findProgressFile(process.cwd(), taskFile);
|
|
if (existingProgress) {
|
|
const projectDir = path.dirname(path.dirname(existingProgress.path));
|
|
const progress = new ProgressTracker(
|
|
projectDir,
|
|
taskFile,
|
|
existingProgress.prdKey,
|
|
);
|
|
ctx.ui.notify(formatProgressStatus(progress.getState()), "info");
|
|
return;
|
|
}
|
|
// No progress yet for this task — parse and show plan instead
|
|
const project = parseTaskFile(taskFile);
|
|
ctx.ui.notify(
|
|
`No progress for ${path.basename(taskFile)}. ${project.tasks.length} tasks found.\nUse /ralph run ${args[0]} to start.`,
|
|
"info",
|
|
);
|
|
return;
|
|
}
|
|
|
|
const found = findProgressFile(process.cwd());
|
|
if (!found) {
|
|
ctx.ui.notify(
|
|
"No .ralph/progress.json found. Start with /ralph run [task-file]",
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
|
|
ctx.ui.notify(formatAllPRDsStatus(found.state), "info");
|
|
}
|
|
|
|
// ─── /ralph resume ───────────────────────────────────────────────────────────
|
|
|
|
async function handleResume(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
sendChatMessage?: (content: string) => void,
|
|
): Promise<void> {
|
|
// If a task file arg is provided, find progress for that specific PRD
|
|
let taskFile: string;
|
|
let projectDir: string;
|
|
let found: ReturnType<typeof findProgressFile>;
|
|
|
|
if (args[0]) {
|
|
taskFile = resolveTaskArg(args[0], process.cwd());
|
|
found = findProgressFile(process.cwd(), taskFile);
|
|
if (!found) {
|
|
ctx.ui.notify(
|
|
`No existing progress for ${args[0]}. Start with /ralph run ${args[0]}`,
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
projectDir = path.dirname(path.dirname(found.path));
|
|
} else {
|
|
found = findProgressFile(process.cwd());
|
|
if (!found) {
|
|
ctx.ui.notify(
|
|
"No .ralph/progress.json found. Start with /ralph run [task-file]",
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
projectDir = path.dirname(path.dirname(found.path));
|
|
// For no-arg resume, use the first PRD's source path or legacy sourcePath
|
|
taskFile = found.state.prds
|
|
? Object.values(found.state.prds)[0].sourcePath
|
|
: found.state.sourcePath;
|
|
}
|
|
|
|
const project = parseTaskFile(taskFile);
|
|
const config = loadConfig(projectDir);
|
|
const progress = new ProgressTracker(projectDir, taskFile, found.prdKey);
|
|
|
|
progress.setPaused(false);
|
|
|
|
// Set resume status
|
|
ctx.ui.setStatus("ralph", `Resuming from ${path.basename(taskFile)}`);
|
|
|
|
const completed = new Set(progress.getCompletedTaskIds());
|
|
|
|
// Ask user for execution mode
|
|
const mode = await ctx.ui.select("Execution mode for this resume?", [
|
|
"Parallel (DAG-optimized)",
|
|
"Sequential (one at a time)",
|
|
]);
|
|
const useParallel = mode?.startsWith("Parallel");
|
|
|
|
// Sequential mode: use buildSequentialPlan to avoid 29-task mega-batches
|
|
const plan = useParallel
|
|
? buildExecutionPlan(project, completed)
|
|
: buildSequentialPlan(project, completed);
|
|
|
|
for (const batch of plan.batches) {
|
|
await executeBatch(
|
|
batch.batchIndex,
|
|
batch.tasks,
|
|
project,
|
|
config,
|
|
progress,
|
|
ctx as any,
|
|
{ parallel: useParallel },
|
|
sendChatMessage,
|
|
projectDir,
|
|
);
|
|
|
|
for (const task of batch.tasks) {
|
|
const status = progress.getTaskStatus(task.id);
|
|
updateTaskInFile(taskFile, task.id, status);
|
|
}
|
|
}
|
|
|
|
ctx.ui.notify(formatProgressStatus(progress.getState()), "info");
|
|
}
|
|
|
|
// ─── /ralph next ─────────────────────────────────────────────────────────────
|
|
|
|
async function handleNext(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
sendChatMessage?: (content: string) => void,
|
|
): Promise<void> {
|
|
let taskFile: string;
|
|
let projectDir: string;
|
|
let found: ReturnType<typeof findProgressFile>;
|
|
|
|
if (args[0]) {
|
|
taskFile = resolveTaskArg(args[0], process.cwd());
|
|
found = findProgressFile(process.cwd(), taskFile);
|
|
if (found) {
|
|
projectDir = path.dirname(path.dirname(found.path));
|
|
} else {
|
|
projectDir = process.cwd();
|
|
}
|
|
} else {
|
|
found = findProgressFile(process.cwd());
|
|
if (!found) {
|
|
ctx.ui.notify(
|
|
"No .ralph/progress.json found. Start with /ralph run [task-file]",
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
taskFile = found.state.prds
|
|
? Object.values(found.state.prds)[0].sourcePath
|
|
: found.state.sourcePath;
|
|
projectDir = path.dirname(path.dirname(found.path));
|
|
}
|
|
|
|
const project = parseTaskFile(taskFile);
|
|
const config = loadConfig(projectDir);
|
|
const progress = new ProgressTracker(projectDir, taskFile, found?.prdKey);
|
|
|
|
const completed = new Set(progress.getCompletedTaskIds());
|
|
const ready = getReadyTasks(project, completed);
|
|
|
|
if (ready.length === 0) {
|
|
ctx.ui.notify(
|
|
"No tasks ready to execute. All tasks completed or blocked.",
|
|
"info",
|
|
);
|
|
return;
|
|
}
|
|
|
|
const nextBatch = ready.slice(
|
|
0,
|
|
config.execution.maxParallel || ready.length,
|
|
);
|
|
|
|
for (const task of nextBatch) {
|
|
await executeBatch(
|
|
0,
|
|
[task],
|
|
project,
|
|
config,
|
|
progress,
|
|
ctx as any,
|
|
undefined,
|
|
sendChatMessage,
|
|
projectDir,
|
|
);
|
|
updateTaskInFile(taskFile, task.id, progress.getTaskStatus(task.id));
|
|
}
|
|
|
|
ctx.ui.notify(
|
|
`Executed: ${nextBatch.map((t) => t.id).join(", ")}\n\n${formatProgressStatus(progress.getState())}`,
|
|
"info",
|
|
);
|
|
}
|
|
|
|
// ─── /ralph reset ────────────────────────────────────────────────────────────
|
|
|
|
async function handleReset(
|
|
ctx: ExtensionContext,
|
|
args: string[],
|
|
): Promise<void> {
|
|
let projectDir: string;
|
|
if (args[0]) {
|
|
const taskFile = resolveTaskArg(args[0], process.cwd());
|
|
const found = findProgressFile(process.cwd(), taskFile);
|
|
projectDir = found ? path.dirname(path.dirname(found.path)) : process.cwd();
|
|
const progress = new ProgressTracker(projectDir, taskFile, found?.prdKey);
|
|
progress.reset();
|
|
} else {
|
|
const found = findProgressFile(process.cwd());
|
|
if (!found) {
|
|
ctx.ui.notify(
|
|
"No .ralph/progress.json found. Start with /ralph run [task-file]",
|
|
"warning",
|
|
);
|
|
return;
|
|
}
|
|
const projectDir = path.dirname(path.dirname(found.path));
|
|
const progress = new ProgressTracker(
|
|
projectDir,
|
|
found.state.prds
|
|
? Object.values(found.state.prds)[0].sourcePath
|
|
: found.state.sourcePath,
|
|
);
|
|
progress.reset();
|
|
}
|
|
|
|
ctx.ui.notify("Progress reset. All task statuses cleared.", "info");
|
|
}
|
|
|
|
// ─── Helpers ─────────────────────────────────────────────────────────────────
|
|
|
|
function formatDuration(ms: number): string {
|
|
const seconds = Math.floor(ms / 1000);
|
|
const minutes = Math.floor(seconds / 60);
|
|
const hours = Math.floor(minutes / 60);
|
|
|
|
if (hours > 0) {
|
|
return `${hours}h ${minutes % 60}m`;
|
|
}
|
|
if (minutes > 0) {
|
|
return `${minutes}m ${seconds % 60}s`;
|
|
}
|
|
return `${seconds}s`;
|
|
}
|