@roodriigoooo/pi-docket 0.4.0

package/extensions/docket-config.ts

@@ -0,0 +1,88 @@
+import { existsSync } from "node:fs";
+import fs from "node:fs/promises";
+import path from "node:path";
+import { getAgentDir } from "@mariozechner/pi-coding-agent";
+import type { CheckpointSummarizerConfig } from "./checkpoint-summarizer.js";
+
+export type DocketWorkerConfig = {
+	guardrailsPath?: string;
+	/** Hide ended workers from the dock once they go untouched for this many minutes. Set 0 to keep them visible. */
+	dockIdleHideMinutes?: number;
+	/** Auto-prune ended worker dirs once they go untouched for this many hours. Set 0 to disable. */
+	pruneAfterHours?: number;
+	/** Max simultaneously active workers (across the whole tmux session). Excess /docket spawn calls are rejected. */
+	maxActive?: number;
+	/** Max child-spawn depth. Top-level parent is depth 0; its children are depth 1; etc. */
+	maxSpawnDepth?: number;
+	/** Project-default kind picked when /docket spawn is invoked without --as. */
+	defaultKind?: string;
+	/** When true, dock writes a compact worker line to tmux status-right so attached panes still see fleet state. */
+	tmuxStatusLine?: boolean;
+	/** When true, every spawned worker also runs tmux pipe-pane to <worker-dir>/pane.log for post-hoc debug. */
+	captureTerminal?: boolean;
+	/** When true, /docket offers to re-window orphan workers when the shared tmux session is gone but their dirs are still live. */
+	autoRespawn?: boolean;
+	/** When true (default), a short summary message is appended to the parent session when a worker reaches ready. Set false to keep the parent JSONL fully manual. */
+	autoEmbedSummary?: boolean;
+};
+
+export type DocketConfig = {
+	maxArtifacts: number;
+	maxBodyChars: number;
+	checkpointArtifacts: number;
+	consumedRetentionDays: number;
+	summarizer: CheckpointSummarizerConfig;
+	worker?: DocketWorkerConfig;
+};
+
+export const DEFAULT_CONFIG: DocketConfig = {
+	maxArtifacts: 300,
+	maxBodyChars: 6000,
+	checkpointArtifacts: 24,
+	consumedRetentionDays: 7,
+	summarizer: {
+		enabled: true,
+		maxOutputTokens: 1200,
+		maxInputChars: 36000,
+		timeoutMs: 120000,
+	},
+	worker: {
+		dockIdleHideMinutes: 30,
+		pruneAfterHours: 24,
+		maxActive: 8,
+		maxSpawnDepth: 2,
+		tmuxStatusLine: false,
+		captureTerminal: false,
+		autoRespawn: false,
+		autoEmbedSummary: true,
+	},
+};
+
+async function readJsonFile<T>(file: string, fallback: T): Promise<T> {
+	try {
+		if (!existsSync(file)) return fallback;
+		return JSON.parse(await fs.readFile(file, "utf8")) as T;
+	} catch {
+		return fallback;
+	}
+}
+
+export async function loadConfig(cwd: string): Promise<DocketConfig> {
+	const globalConfig = await readJsonFile<Partial<DocketConfig>>(path.join(getAgentDir(), "docket.json"), {});
+	const projectConfig = await readJsonFile<Partial<DocketConfig>>(path.join(cwd, ".pi", "docket.json"), {});
+	return {
+		...DEFAULT_CONFIG,
+		...globalConfig,
+		...projectConfig,
+		summarizer: {
+			...DEFAULT_CONFIG.summarizer,
+			...(globalConfig.summarizer ?? {}),
+			...(projectConfig.summarizer ?? {}),
+		},
+		worker: {
+			...DEFAULT_CONFIG.worker,
+			...(globalConfig.worker ?? {}),
+			...(projectConfig.worker ?? {}),
+		},
+	};
+}

package/extensions/docket-extension-surface.ts

@@ -0,0 +1,43 @@
+import type { WorkerEvent } from "./worker-events.js";
+import type { WorkerKind, WorkerKindRegistry } from "./worker-kinds.js";
+
+export type DocketExtensionSurface = {
+	registerWorkerKind(kind: Omit<WorkerKind, "source"> & { source?: WorkerKind["source"] }): () => void;
+	listWorkerKinds(): WorkerKind[];
+	onWorkerEvent(handler: (event: { workerId: string; event: WorkerEvent }) => void): () => void;
+};
+
+const SURFACE_KEY = "__docket";
+
+type EventHandler = (event: { workerId: string; event: WorkerEvent }) => void;
+
+export type DocketExtensionSurfaceInternals = DocketExtensionSurface & {
+	emitWorkerEvent(workerId: string, event: WorkerEvent): void;
+};
+
+export function installDocketExtensionSurface(registry: WorkerKindRegistry): DocketExtensionSurfaceInternals {
+	const handlers = new Set<EventHandler>();
+	const surface: DocketExtensionSurfaceInternals = {
+		registerWorkerKind(kind) {
+			return registry.register({ ...kind, source: kind.source ?? "runtime" });
+		},
+		listWorkerKinds() {
+			return registry.list();
+		},
+		onWorkerEvent(handler) {
+			handlers.add(handler);
+			return () => handlers.delete(handler);
+		},
+		emitWorkerEvent(workerId, event) {
+			for (const handler of handlers) {
+				try { handler({ workerId, event }); } catch { /* never let a subscriber break docket */ }
+			}
+		},
+	};
+	(globalThis as Record<string, unknown>)[SURFACE_KEY] = surface;
+	return surface;
+}
+
+export function getDocketExtensionSurface(): DocketExtensionSurfaceInternals | undefined {
+	return (globalThis as Record<string, unknown>)[SURFACE_KEY] as DocketExtensionSurfaceInternals | undefined;
+}