pi-crew 0.1.51 → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +56 -1
- package/README.md +176 -781
- package/agents/analyst.md +11 -11
- package/agents/critic.md +11 -11
- package/agents/executor.md +11 -11
- package/agents/explorer.md +11 -11
- package/agents/planner.md +11 -11
- package/agents/reviewer.md +11 -11
- package/agents/security-reviewer.md +11 -11
- package/agents/test-engineer.md +11 -11
- package/agents/verifier.md +70 -11
- package/agents/writer.md +11 -11
- package/docs/actions-reference.md +595 -0
- package/docs/commands-reference.md +347 -0
- package/docs/runtime-flow.md +148 -148
- package/index.ts +6 -6
- package/package.json +99 -99
- package/skills/async-worker-recovery/SKILL.md +42 -42
- package/skills/context-artifact-hygiene/SKILL.md +52 -52
- package/skills/delegation-patterns/SKILL.md +54 -54
- package/skills/mailbox-interactive/SKILL.md +40 -40
- package/skills/model-routing-context/SKILL.md +39 -39
- package/skills/multi-perspective-review/SKILL.md +58 -58
- package/skills/observability-reliability/SKILL.md +41 -41
- package/skills/orchestration/SKILL.md +157 -157
- package/skills/ownership-session-security/SKILL.md +41 -41
- package/skills/pi-extension-lifecycle/SKILL.md +39 -39
- package/skills/requirements-to-task-packet/SKILL.md +63 -63
- package/skills/resource-discovery-config/SKILL.md +41 -41
- package/skills/runtime-state-reader/SKILL.md +44 -44
- package/skills/secure-agent-orchestration-review/SKILL.md +45 -45
- package/skills/state-mutation-locking/SKILL.md +42 -42
- package/skills/systematic-debugging/SKILL.md +67 -67
- package/skills/ui-render-performance/SKILL.md +39 -39
- package/skills/verification-before-done/SKILL.md +57 -57
- package/skills/worktree-isolation/SKILL.md +39 -39
- package/src/adapters/claude-adapter.ts +25 -0
- package/src/adapters/codex-adapter.ts +21 -0
- package/src/adapters/cursor-adapter.ts +17 -0
- package/src/adapters/export-util.ts +137 -0
- package/src/adapters/index.ts +15 -0
- package/src/adapters/registry.ts +18 -0
- package/src/adapters/types.ts +23 -0
- package/src/agents/agent-config.ts +2 -0
- package/src/agents/agent-search.ts +98 -98
- package/src/agents/discover-agents.ts +2 -1
- package/src/config/config.ts +13 -1
- package/src/config/drift-detector.ts +211 -0
- package/src/config/markers.ts +327 -0
- package/src/config/resilient-parser.ts +108 -0
- package/src/config/suggestions.ts +74 -0
- package/src/extension/cross-extension-rpc.ts +103 -94
- package/src/extension/project-init.ts +21 -1
- package/src/extension/register.ts +45 -14
- package/src/extension/registration/commands.ts +77 -8
- package/src/extension/registration/subagent-tools.ts +10 -1
- package/src/extension/registration/team-tool.ts +10 -1
- package/src/extension/registration/viewers.ts +48 -34
- package/src/extension/run-bundle-schema.ts +89 -89
- package/src/extension/run-import.ts +25 -1
- package/src/extension/run-index.ts +5 -1
- package/src/extension/run-maintenance.ts +142 -68
- package/src/extension/team-manager-command.ts +10 -1
- package/src/extension/team-tool/doctor.ts +28 -3
- package/src/extension/team-tool/handle-settings.ts +195 -188
- package/src/extension/team-tool/inspect.ts +41 -41
- package/src/extension/team-tool/intent-policy.ts +42 -42
- package/src/extension/team-tool/lifecycle-actions.ts +27 -8
- package/src/extension/team-tool/plan.ts +19 -19
- package/src/extension/team-tool/run.ts +12 -1
- package/src/extension/team-tool.ts +11 -1
- package/src/i18n.ts +184 -184
- package/src/observability/exporters/otlp-exporter.ts +92 -77
- package/src/prompt/prompt-runtime.ts +72 -72
- package/src/runtime/agent-memory.ts +72 -72
- package/src/runtime/agent-observability.ts +114 -114
- package/src/runtime/async-marker.ts +26 -26
- package/src/runtime/attention-events.ts +28 -28
- package/src/runtime/auto-resume.ts +100 -0
- package/src/runtime/background-runner.ts +11 -1
- package/src/runtime/cancellation-token.ts +89 -89
- package/src/runtime/cancellation.ts +61 -61
- package/src/runtime/capability-inventory.ts +116 -116
- package/src/runtime/child-pi.ts +7 -2
- package/src/runtime/compaction-summary.ts +271 -0
- package/src/runtime/completion-guard.ts +190 -190
- package/src/runtime/crash-recovery.ts +33 -0
- package/src/runtime/delta-conflict.ts +360 -0
- package/src/runtime/direct-run.ts +35 -35
- package/src/runtime/foreground-control.ts +82 -82
- package/src/runtime/green-contract.ts +46 -46
- package/src/runtime/group-join.ts +106 -106
- package/src/runtime/heartbeat-gradient.ts +28 -28
- package/src/runtime/heartbeat-watcher.ts +124 -124
- package/src/runtime/iteration-hooks.ts +262 -0
- package/src/runtime/live-agent-control.ts +88 -88
- package/src/runtime/live-control-realtime.ts +36 -36
- package/src/runtime/live-extension-bridge.ts +150 -150
- package/src/runtime/live-irc.ts +92 -92
- package/src/runtime/live-session-health.ts +100 -100
- package/src/runtime/loop-gates.ts +129 -0
- package/src/runtime/metric-parser.ts +40 -0
- package/src/runtime/notebook-helpers.ts +90 -90
- package/src/runtime/orphan-sentinel.ts +7 -7
- package/src/runtime/parallel-research.ts +44 -44
- package/src/runtime/phase-progress.ts +217 -0
- package/src/runtime/pi-args.ts +38 -11
- package/src/runtime/pi-json-output.ts +111 -111
- package/src/runtime/pi-spawn.ts +57 -7
- package/src/runtime/policy-engine.ts +79 -79
- package/src/runtime/post-checks.ts +122 -0
- package/src/runtime/progress-event-coalescer.ts +43 -43
- package/src/runtime/prose-compressor.ts +164 -164
- package/src/runtime/recovery-recipes.ts +74 -74
- package/src/runtime/result-extractor.ts +121 -121
- package/src/runtime/role-permission.ts +39 -39
- package/src/runtime/sensitive-paths.ts +2 -2
- package/src/runtime/session-resources.ts +25 -25
- package/src/runtime/session-snapshot.ts +59 -59
- package/src/runtime/session-usage.ts +79 -79
- package/src/runtime/sidechain-output.ts +29 -29
- package/src/runtime/stream-preview.ts +177 -177
- package/src/runtime/supervisor-contact.ts +59 -59
- package/src/runtime/task-display.ts +38 -38
- package/src/runtime/task-graph.ts +207 -0
- package/src/runtime/task-quality.ts +207 -0
- package/src/runtime/task-runner/capabilities.ts +78 -78
- package/src/runtime/task-runner/live-executor.ts +7 -1
- package/src/runtime/task-runner/progress.ts +119 -119
- package/src/runtime/task-runner/prompt-pipeline.ts +64 -64
- package/src/runtime/task-runner/result-utils.ts +14 -14
- package/src/runtime/task-runner/run-projection.ts +103 -103
- package/src/runtime/task-runner/state-helpers.ts +22 -22
- package/src/runtime/team-runner.ts +117 -7
- package/src/runtime/worker-heartbeat.ts +21 -21
- package/src/runtime/worker-startup.ts +57 -57
- package/src/runtime/workflow-state.ts +187 -0
- package/src/runtime/workspace-tree.ts +298 -298
- package/src/schema/config-schema.ts +11 -0
- package/src/schema/validation-types.ts +148 -0
- package/src/skills/skill-templates.ts +374 -0
- package/src/state/active-run-registry.ts +35 -11
- package/src/state/atomic-write.ts +33 -26
- package/src/state/contracts.ts +1 -0
- package/src/state/event-reconstructor.ts +217 -0
- package/src/state/locks.ts +2 -13
- package/src/state/mailbox.ts +4 -3
- package/src/state/state-store.ts +32 -14
- package/src/state/task-claims.ts +44 -44
- package/src/state/types.ts +9 -0
- package/src/state/usage.ts +29 -29
- package/src/subagents/async-entry.ts +1 -1
- package/src/subagents/index.ts +3 -3
- package/src/subagents/live/control.ts +1 -1
- package/src/subagents/live/manager.ts +1 -1
- package/src/subagents/live/realtime.ts +1 -1
- package/src/subagents/live/session-runtime.ts +1 -1
- package/src/subagents/manager.ts +1 -1
- package/src/subagents/spawn.ts +1 -1
- package/src/teams/team-serializer.ts +38 -38
- package/src/types/diff.d.ts +18 -18
- package/src/ui/crew-footer.ts +101 -101
- package/src/ui/crew-select-list.ts +111 -111
- package/src/ui/crew-widget.ts +5 -2
- package/src/ui/dashboard-panes/cancellation-pane.ts +42 -42
- package/src/ui/dashboard-panes/capability-pane.ts +59 -59
- package/src/ui/dashboard-panes/mailbox-pane.ts +35 -35
- package/src/ui/dashboard-panes/metrics-pane.ts +34 -34
- package/src/ui/dashboard-panes/progress-pane.ts +11 -0
- package/src/ui/dynamic-border.ts +25 -25
- package/src/ui/layout-primitives.ts +106 -106
- package/src/ui/loaders.ts +158 -158
- package/src/ui/render-coalescer.ts +51 -51
- package/src/ui/render-diff.ts +119 -119
- package/src/ui/render-scheduler.ts +143 -143
- package/src/ui/run-action-dispatcher.ts +10 -1
- package/src/ui/spinner.ts +17 -17
- package/src/ui/status-colors.ts +58 -58
- package/src/ui/syntax-highlight.ts +116 -116
- package/src/ui/transcript-entries.ts +258 -258
- package/src/utils/completion-dedupe.ts +63 -63
- package/src/utils/frontmatter.ts +68 -68
- package/src/utils/git.ts +262 -262
- package/src/utils/ids.ts +17 -17
- package/src/utils/incremental-reader.ts +104 -104
- package/src/utils/names.ts +27 -27
- package/src/utils/redaction.ts +44 -44
- package/src/utils/safe-paths.ts +47 -47
- package/src/utils/scan-cache.ts +136 -136
- package/src/utils/sleep.ts +40 -26
- package/src/utils/task-name-generator.ts +337 -337
- package/src/workflows/validate-workflow.ts +40 -40
- package/src/worktree/branch-freshness.ts +45 -45
- package/teams/default.team.md +12 -12
- package/teams/fast-fix.team.md +11 -11
- package/teams/implementation.team.md +18 -18
- package/teams/parallel-research.team.md +14 -14
- package/teams/research.team.md +11 -11
- package/teams/review.team.md +12 -12
- package/workflows/default.workflow.md +30 -29
- package/workflows/fast-fix.workflow.md +23 -22
- package/workflows/implementation.workflow.md +43 -43
- package/workflows/parallel-research.workflow.md +46 -46
- package/workflows/research.workflow.md +22 -22
- package/workflows/review.workflow.md +30 -30
- package/docs/refactor-tasks-phase3.md +0 -394
- package/docs/refactor-tasks-phase4.md +0 -564
- package/docs/refactor-tasks-phase5.md +0 -402
- package/docs/refactor-tasks-phase6.md +0 -662
- package/docs/refactor-tasks.md +0 -1484
- package/docs/research/AGENT-EXECUTION-ARCHITECTURE.md +0 -261
- package/docs/research/AGENT-LIFECYCLE-COMPARISON.md +0 -111
- package/docs/research/AUDIT_OH_MY_PI.md +0 -261
- package/docs/research/AUDIT_PI_CREW.md +0 -457
- package/docs/research/CAVEMAN-DEEP-RESEARCH.md +0 -281
- package/docs/research/COMPARISON_OH_MY_PI_VS_PI_CREW.md +0 -264
- package/docs/research/DEEP-RESEARCH-PI-POWERBAR.md +0 -343
- package/docs/research/DEEP_RESEARCH_SUBAGENT_ARCHITECTURE.md +0 -480
- package/docs/research/GAP_CLOSURE_IMPLEMENTATION_PLAN.md +0 -354
- package/docs/research/IMPLEMENTATION_PLAN.md +0 -385
- package/docs/research/LIVE-SESSION-PRODUCTION-READY-PLAN.md +0 -502
- package/docs/research/OH-MY-PI-DEEP-RESEARCH-v14.7.6.md +0 -266
- package/docs/research/REMAINING-GAPS-PLAN.md +0 -363
- package/docs/research/SESSION-SUMMARY-2026-05-08.md +0 -146
- package/docs/research/UI-RESPONSIVENESS-AUDIT.md +0 -173
- package/docs/research-awesome-agent-skills-distillation.md +0 -100
- package/docs/research-extension-examples.md +0 -297
- package/docs/research-extension-system.md +0 -324
- package/docs/research-oh-my-pi-distillation.md +0 -369
- package/docs/research-optimization-plan.md +0 -548
- package/docs/research-phase10-distillation.md +0 -199
- package/docs/research-phase11-distillation.md +0 -201
- package/docs/research-phase8-operator-experience-plan.md +0 -819
- package/docs/research-phase9-observability-reliability-plan.md +0 -1190
- package/docs/research-pi-coding-agent.md +0 -357
- package/docs/research-source-pi-crew-reference.md +0 -174
- package/docs/research-ui-optimization-plan.md +0 -480
- package/docs/source-runtime-refactor-map.md +0 -107
- package/src/utils/atomic-write.ts +0 -33
|
@@ -1,114 +1,114 @@
|
|
|
1
|
-
import * as fs from "node:fs";
|
|
2
|
-
import type { TeamRunManifest } from "../state/types.ts";
|
|
3
|
-
import { agentOutputPath, readCrewAgents } from "./crew-agent-records.ts";
|
|
4
|
-
import type { CrewAgentRecord } from "./crew-agent-runtime.ts";
|
|
5
|
-
|
|
6
|
-
const TOOL_LABELS: Record<string, string> = {
|
|
7
|
-
read: "reading",
|
|
8
|
-
bash: "running command",
|
|
9
|
-
edit: "editing",
|
|
10
|
-
write: "writing",
|
|
11
|
-
grep: "searching",
|
|
12
|
-
find: "finding files",
|
|
13
|
-
ls: "listing",
|
|
14
|
-
};
|
|
15
|
-
|
|
16
|
-
export interface TextTailResult {
|
|
17
|
-
path: string;
|
|
18
|
-
text: string;
|
|
19
|
-
bytes: number;
|
|
20
|
-
truncated: boolean;
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
export function readTextTail(filePath: string, maxBytes = 64_000): TextTailResult {
|
|
24
|
-
if (!fs.existsSync(filePath)) return { path: filePath, text: "", bytes: 0, truncated: false };
|
|
25
|
-
const stat = fs.statSync(filePath);
|
|
26
|
-
const bytesToRead = Math.min(stat.size, Math.max(0, maxBytes));
|
|
27
|
-
const fd = fs.openSync(filePath, "r");
|
|
28
|
-
try {
|
|
29
|
-
const buffer = Buffer.alloc(bytesToRead);
|
|
30
|
-
fs.readSync(fd, buffer, 0, bytesToRead, stat.size - bytesToRead);
|
|
31
|
-
return { path: filePath, text: buffer.toString("utf-8"), bytes: stat.size, truncated: stat.size > bytesToRead };
|
|
32
|
-
} finally {
|
|
33
|
-
fs.closeSync(fd);
|
|
34
|
-
}
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
function compactDuration(ms: number | undefined): string | undefined {
|
|
38
|
-
if (ms === undefined || !Number.isFinite(ms)) return undefined;
|
|
39
|
-
if (ms < 1000) return `${Math.round(ms)}ms`;
|
|
40
|
-
if (ms < 60_000) return `${(ms / 1000).toFixed(1)}s`;
|
|
41
|
-
return `${Math.floor(ms / 60_000)}m${Math.floor((ms % 60_000) / 1000)}s`;
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
function ageBetween(start: string | undefined, end: string | undefined): string | undefined {
|
|
45
|
-
if (!start) return undefined;
|
|
46
|
-
const stop = end ? new Date(end).getTime() : Date.now();
|
|
47
|
-
const ms = Math.max(0, stop - new Date(start).getTime());
|
|
48
|
-
return compactDuration(ms);
|
|
49
|
-
}
|
|
50
|
-
|
|
51
|
-
function activityText(agent: CrewAgentRecord): string {
|
|
52
|
-
const parts: string[] = [];
|
|
53
|
-
if (agent.progress?.activityState) parts.push(agent.progress.activityState);
|
|
54
|
-
if (agent.progress?.currentTool) parts.push(TOOL_LABELS[agent.progress.currentTool] ?? `tool=${agent.progress.currentTool}`);
|
|
55
|
-
if (agent.toolUses !== undefined) parts.push(`tools=${agent.toolUses}`);
|
|
56
|
-
if (agent.progress?.tokens !== undefined) parts.push(`tokens=${agent.progress.tokens}`);
|
|
57
|
-
if (agent.progress?.turns !== undefined) parts.push(`turns=${agent.progress.turns}`);
|
|
58
|
-
const duration = compactDuration(agent.progress?.durationMs) ?? ageBetween(agent.startedAt, agent.completedAt);
|
|
59
|
-
if (duration) parts.push(duration);
|
|
60
|
-
if (agent.progress?.failedTool) parts.push(`failedTool=${agent.progress.failedTool}`);
|
|
61
|
-
if (agent.progress?.recentOutput?.length) parts.push(`last=${agent.progress.recentOutput.at(-1)}`);
|
|
62
|
-
return parts.join(" ") || "idle";
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
function statusGlyph(status: CrewAgentRecord["status"]): string {
|
|
66
|
-
if (status === "completed") return "✓";
|
|
67
|
-
if (status === "failed") return "✗";
|
|
68
|
-
if (status === "running") return "▶";
|
|
69
|
-
if (status === "cancelled" || status === "stopped") return "■";
|
|
70
|
-
return "·";
|
|
71
|
-
}
|
|
72
|
-
|
|
73
|
-
function outputWarning(manifest: TeamRunManifest, agent: CrewAgentRecord): string {
|
|
74
|
-
if (agent.status !== "completed") return "";
|
|
75
|
-
try {
|
|
76
|
-
const outputPath = agentOutputPath(manifest, agent.taskId);
|
|
77
|
-
if (!fs.existsSync(outputPath)) return " no-output";
|
|
78
|
-
return fs.statSync(outputPath).size === 0 ? " no-output" : "";
|
|
79
|
-
} catch {
|
|
80
|
-
return " no-output";
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
|
|
84
|
-
function agentLine(manifest: TeamRunManifest, agent: CrewAgentRecord): string {
|
|
85
|
-
return `- ${statusGlyph(agent.status)} ${agent.taskId} ${agent.role} → ${agent.agent} · ${agent.status} · ${agent.runtime} · ${activityText(agent)}${outputWarning(manifest, agent)}${agent.error ? ` · error=${agent.error}` : ""}`;
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
export function buildAgentDashboard(manifest: TeamRunManifest): { text: string; groups: Record<string, CrewAgentRecord[]> } {
|
|
89
|
-
const agents = readCrewAgents(manifest);
|
|
90
|
-
const groups: Record<string, CrewAgentRecord[]> = {
|
|
91
|
-
running: agents.filter((agent) => agent.status === "running"),
|
|
92
|
-
queued: agents.filter((agent) => agent.status === "queued"),
|
|
93
|
-
recent: agents.filter((agent) => agent.status !== "running" && agent.status !== "queued"),
|
|
94
|
-
};
|
|
95
|
-
const lines = [
|
|
96
|
-
`Crew agents for ${manifest.runId}`,
|
|
97
|
-
`Run: ${manifest.status} · ${manifest.team}/${manifest.workflow ?? "none"} · agents=${agents.length}`,
|
|
98
|
-
`Counts: running=${groups.running.length}, queued=${groups.queued.length}, recent=${groups.recent.length}`,
|
|
99
|
-
"",
|
|
100
|
-
"## Running",
|
|
101
|
-
...(groups.running.length ? groups.running.map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
102
|
-
"",
|
|
103
|
-
"## Queued",
|
|
104
|
-
...(groups.queued.length ? groups.queued.map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
105
|
-
"",
|
|
106
|
-
"## Recent",
|
|
107
|
-
...(groups.recent.length ? groups.recent.slice(-10).map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
108
|
-
];
|
|
109
|
-
return { text: lines.join("\n"), groups };
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
export function readAgentOutput(manifest: TeamRunManifest, taskId: string, maxBytes?: number): TextTailResult {
|
|
113
|
-
return readTextTail(agentOutputPath(manifest, taskId), maxBytes);
|
|
114
|
-
}
|
|
1
|
+
import * as fs from "node:fs";
|
|
2
|
+
import type { TeamRunManifest } from "../state/types.ts";
|
|
3
|
+
import { agentOutputPath, readCrewAgents } from "./crew-agent-records.ts";
|
|
4
|
+
import type { CrewAgentRecord } from "./crew-agent-runtime.ts";
|
|
5
|
+
|
|
6
|
+
const TOOL_LABELS: Record<string, string> = {
|
|
7
|
+
read: "reading",
|
|
8
|
+
bash: "running command",
|
|
9
|
+
edit: "editing",
|
|
10
|
+
write: "writing",
|
|
11
|
+
grep: "searching",
|
|
12
|
+
find: "finding files",
|
|
13
|
+
ls: "listing",
|
|
14
|
+
};
|
|
15
|
+
|
|
16
|
+
export interface TextTailResult {
|
|
17
|
+
path: string;
|
|
18
|
+
text: string;
|
|
19
|
+
bytes: number;
|
|
20
|
+
truncated: boolean;
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
export function readTextTail(filePath: string, maxBytes = 64_000): TextTailResult {
|
|
24
|
+
if (!fs.existsSync(filePath)) return { path: filePath, text: "", bytes: 0, truncated: false };
|
|
25
|
+
const stat = fs.statSync(filePath);
|
|
26
|
+
const bytesToRead = Math.min(stat.size, Math.max(0, maxBytes));
|
|
27
|
+
const fd = fs.openSync(filePath, "r");
|
|
28
|
+
try {
|
|
29
|
+
const buffer = Buffer.alloc(bytesToRead);
|
|
30
|
+
fs.readSync(fd, buffer, 0, bytesToRead, stat.size - bytesToRead);
|
|
31
|
+
return { path: filePath, text: buffer.toString("utf-8"), bytes: stat.size, truncated: stat.size > bytesToRead };
|
|
32
|
+
} finally {
|
|
33
|
+
fs.closeSync(fd);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
function compactDuration(ms: number | undefined): string | undefined {
|
|
38
|
+
if (ms === undefined || !Number.isFinite(ms)) return undefined;
|
|
39
|
+
if (ms < 1000) return `${Math.round(ms)}ms`;
|
|
40
|
+
if (ms < 60_000) return `${(ms / 1000).toFixed(1)}s`;
|
|
41
|
+
return `${Math.floor(ms / 60_000)}m${Math.floor((ms % 60_000) / 1000)}s`;
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
function ageBetween(start: string | undefined, end: string | undefined): string | undefined {
|
|
45
|
+
if (!start) return undefined;
|
|
46
|
+
const stop = end ? new Date(end).getTime() : Date.now();
|
|
47
|
+
const ms = Math.max(0, stop - new Date(start).getTime());
|
|
48
|
+
return compactDuration(ms);
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
function activityText(agent: CrewAgentRecord): string {
|
|
52
|
+
const parts: string[] = [];
|
|
53
|
+
if (agent.progress?.activityState) parts.push(agent.progress.activityState);
|
|
54
|
+
if (agent.progress?.currentTool) parts.push(TOOL_LABELS[agent.progress.currentTool] ?? `tool=${agent.progress.currentTool}`);
|
|
55
|
+
if (agent.toolUses !== undefined) parts.push(`tools=${agent.toolUses}`);
|
|
56
|
+
if (agent.progress?.tokens !== undefined) parts.push(`tokens=${agent.progress.tokens}`);
|
|
57
|
+
if (agent.progress?.turns !== undefined) parts.push(`turns=${agent.progress.turns}`);
|
|
58
|
+
const duration = compactDuration(agent.progress?.durationMs) ?? ageBetween(agent.startedAt, agent.completedAt);
|
|
59
|
+
if (duration) parts.push(duration);
|
|
60
|
+
if (agent.progress?.failedTool) parts.push(`failedTool=${agent.progress.failedTool}`);
|
|
61
|
+
if (agent.progress?.recentOutput?.length) parts.push(`last=${agent.progress.recentOutput.at(-1)}`);
|
|
62
|
+
return parts.join(" ") || "idle";
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
function statusGlyph(status: CrewAgentRecord["status"]): string {
|
|
66
|
+
if (status === "completed") return "✓";
|
|
67
|
+
if (status === "failed") return "✗";
|
|
68
|
+
if (status === "running") return "▶";
|
|
69
|
+
if (status === "cancelled" || status === "stopped") return "■";
|
|
70
|
+
return "·";
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
function outputWarning(manifest: TeamRunManifest, agent: CrewAgentRecord): string {
|
|
74
|
+
if (agent.status !== "completed") return "";
|
|
75
|
+
try {
|
|
76
|
+
const outputPath = agentOutputPath(manifest, agent.taskId);
|
|
77
|
+
if (!fs.existsSync(outputPath)) return " no-output";
|
|
78
|
+
return fs.statSync(outputPath).size === 0 ? " no-output" : "";
|
|
79
|
+
} catch {
|
|
80
|
+
return " no-output";
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
|
|
84
|
+
function agentLine(manifest: TeamRunManifest, agent: CrewAgentRecord): string {
|
|
85
|
+
return `- ${statusGlyph(agent.status)} ${agent.taskId} ${agent.role} → ${agent.agent} · ${agent.status} · ${agent.runtime} · ${activityText(agent)}${outputWarning(manifest, agent)}${agent.error ? ` · error=${agent.error}` : ""}`;
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
export function buildAgentDashboard(manifest: TeamRunManifest): { text: string; groups: Record<string, CrewAgentRecord[]> } {
|
|
89
|
+
const agents = readCrewAgents(manifest);
|
|
90
|
+
const groups: Record<string, CrewAgentRecord[]> = {
|
|
91
|
+
running: agents.filter((agent) => agent.status === "running"),
|
|
92
|
+
queued: agents.filter((agent) => agent.status === "queued"),
|
|
93
|
+
recent: agents.filter((agent) => agent.status !== "running" && agent.status !== "queued"),
|
|
94
|
+
};
|
|
95
|
+
const lines = [
|
|
96
|
+
`Crew agents for ${manifest.runId}`,
|
|
97
|
+
`Run: ${manifest.status} · ${manifest.team}/${manifest.workflow ?? "none"} · agents=${agents.length}`,
|
|
98
|
+
`Counts: running=${groups.running.length}, queued=${groups.queued.length}, recent=${groups.recent.length}`,
|
|
99
|
+
"",
|
|
100
|
+
"## Running",
|
|
101
|
+
...(groups.running.length ? groups.running.map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
102
|
+
"",
|
|
103
|
+
"## Queued",
|
|
104
|
+
...(groups.queued.length ? groups.queued.map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
105
|
+
"",
|
|
106
|
+
"## Recent",
|
|
107
|
+
...(groups.recent.length ? groups.recent.slice(-10).map((agent) => agentLine(manifest, agent)) : ["- (none)"]),
|
|
108
|
+
];
|
|
109
|
+
return { text: lines.join("\n"), groups };
|
|
110
|
+
}
|
|
111
|
+
|
|
112
|
+
export function readAgentOutput(manifest: TeamRunManifest, taskId: string, maxBytes?: number): TextTailResult {
|
|
113
|
+
return readTextTail(agentOutputPath(manifest, taskId), maxBytes);
|
|
114
|
+
}
|
|
@@ -1,26 +1,26 @@
|
|
|
1
|
-
import * as fs from "node:fs";
|
|
2
|
-
import * as path from "node:path";
|
|
3
|
-
import { atomicWriteJson } from "../state/atomic-write.ts";
|
|
4
|
-
import type { TeamRunManifest } from "../state/types.ts";
|
|
5
|
-
|
|
6
|
-
export interface AsyncStartMarker {
|
|
7
|
-
pid: number;
|
|
8
|
-
startedAt: string;
|
|
9
|
-
}
|
|
10
|
-
|
|
11
|
-
export function asyncStartMarkerPath(manifest: Pick<TeamRunManifest, "stateRoot">): string {
|
|
12
|
-
return path.join(manifest.stateRoot, "async.pid");
|
|
13
|
-
}
|
|
14
|
-
|
|
15
|
-
export function writeAsyncStartMarker(manifest: Pick<TeamRunManifest, "stateRoot">, marker: AsyncStartMarker): void {
|
|
16
|
-
atomicWriteJson(asyncStartMarkerPath(manifest), marker);
|
|
17
|
-
}
|
|
18
|
-
|
|
19
|
-
export function hasAsyncStartMarker(manifest: Pick<TeamRunManifest, "stateRoot">): boolean {
|
|
20
|
-
try {
|
|
21
|
-
const raw = JSON.parse(fs.readFileSync(asyncStartMarkerPath(manifest), "utf-8")) as Partial<AsyncStartMarker>;
|
|
22
|
-
return typeof raw.pid === "number" && Number.isInteger(raw.pid) && raw.pid > 0 && typeof raw.startedAt === "string" && raw.startedAt.length > 0;
|
|
23
|
-
} catch {
|
|
24
|
-
return false;
|
|
25
|
-
}
|
|
26
|
-
}
|
|
1
|
+
import * as fs from "node:fs";
|
|
2
|
+
import * as path from "node:path";
|
|
3
|
+
import { atomicWriteJson } from "../state/atomic-write.ts";
|
|
4
|
+
import type { TeamRunManifest } from "../state/types.ts";
|
|
5
|
+
|
|
6
|
+
export interface AsyncStartMarker {
|
|
7
|
+
pid: number;
|
|
8
|
+
startedAt: string;
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
export function asyncStartMarkerPath(manifest: Pick<TeamRunManifest, "stateRoot">): string {
|
|
12
|
+
return path.join(manifest.stateRoot, "async.pid");
|
|
13
|
+
}
|
|
14
|
+
|
|
15
|
+
export function writeAsyncStartMarker(manifest: Pick<TeamRunManifest, "stateRoot">, marker: AsyncStartMarker): void {
|
|
16
|
+
atomicWriteJson(asyncStartMarkerPath(manifest), marker);
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
export function hasAsyncStartMarker(manifest: Pick<TeamRunManifest, "stateRoot">): boolean {
|
|
20
|
+
try {
|
|
21
|
+
const raw = JSON.parse(fs.readFileSync(asyncStartMarkerPath(manifest), "utf-8")) as Partial<AsyncStartMarker>;
|
|
22
|
+
return typeof raw.pid === "number" && Number.isInteger(raw.pid) && raw.pid > 0 && typeof raw.startedAt === "string" && raw.startedAt.length > 0;
|
|
23
|
+
} catch {
|
|
24
|
+
return false;
|
|
25
|
+
}
|
|
26
|
+
}
|
|
@@ -1,28 +1,28 @@
|
|
|
1
|
-
import { appendEvent, readEvents } from "../state/event-log.ts";
|
|
2
|
-
import type { CrewAttentionEventData, TeamRunManifest } from "../state/types.ts";
|
|
3
|
-
|
|
4
|
-
export interface AppendTaskAttentionInput {
|
|
5
|
-
manifest: TeamRunManifest;
|
|
6
|
-
taskId?: string;
|
|
7
|
-
message: string;
|
|
8
|
-
data: CrewAttentionEventData;
|
|
9
|
-
}
|
|
10
|
-
|
|
11
|
-
export function appendTaskAttentionEvent(input: AppendTaskAttentionInput): boolean {
|
|
12
|
-
const recent = readEvents(input.manifest.eventsPath).slice(-200);
|
|
13
|
-
const dedupKey = `${input.taskId ?? ""}:${input.data.reason}:${input.data.activityState}`;
|
|
14
|
-
const duplicate = recent.some(
|
|
15
|
-
(event) =>
|
|
16
|
-
event.type === "task.attention" &&
|
|
17
|
-
`${event.taskId ?? ""}:${event.data?.reason ?? ""}:${event.data?.activityState ?? ""}` === dedupKey,
|
|
18
|
-
);
|
|
19
|
-
if (duplicate) return false;
|
|
20
|
-
appendEvent(input.manifest.eventsPath, {
|
|
21
|
-
type: "task.attention",
|
|
22
|
-
runId: input.manifest.runId,
|
|
23
|
-
taskId: input.taskId,
|
|
24
|
-
message: input.message,
|
|
25
|
-
data: { ...input.data },
|
|
26
|
-
});
|
|
27
|
-
return true;
|
|
28
|
-
}
|
|
1
|
+
import { appendEvent, readEvents } from "../state/event-log.ts";
|
|
2
|
+
import type { CrewAttentionEventData, TeamRunManifest } from "../state/types.ts";
|
|
3
|
+
|
|
4
|
+
export interface AppendTaskAttentionInput {
|
|
5
|
+
manifest: TeamRunManifest;
|
|
6
|
+
taskId?: string;
|
|
7
|
+
message: string;
|
|
8
|
+
data: CrewAttentionEventData;
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
export function appendTaskAttentionEvent(input: AppendTaskAttentionInput): boolean {
|
|
12
|
+
const recent = readEvents(input.manifest.eventsPath).slice(-200);
|
|
13
|
+
const dedupKey = `${input.taskId ?? ""}:${input.data.reason}:${input.data.activityState}`;
|
|
14
|
+
const duplicate = recent.some(
|
|
15
|
+
(event) =>
|
|
16
|
+
event.type === "task.attention" &&
|
|
17
|
+
`${event.taskId ?? ""}:${event.data?.reason ?? ""}:${event.data?.activityState ?? ""}` === dedupKey,
|
|
18
|
+
);
|
|
19
|
+
if (duplicate) return false;
|
|
20
|
+
appendEvent(input.manifest.eventsPath, {
|
|
21
|
+
type: "task.attention",
|
|
22
|
+
runId: input.manifest.runId,
|
|
23
|
+
taskId: input.taskId,
|
|
24
|
+
message: input.message,
|
|
25
|
+
data: { ...input.data },
|
|
26
|
+
});
|
|
27
|
+
return true;
|
|
28
|
+
}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Auto-resume controller with settle window — prevents premature context injection
|
|
3
|
+
* by waiting for Pi's setTimeout(0) + setTimeout(100) to drain before resuming.
|
|
4
|
+
*
|
|
5
|
+
* Distilled from pi-autoresearch's auto-resume / settle-window pattern.
|
|
6
|
+
*/
|
|
7
|
+
|
|
8
|
+
/** Time to wait before auto-resuming (ms). Outlasts Pi's setTimeout(0) + setTimeout(100). */
|
|
9
|
+
export const SETTLE_WINDOW_MS = 800;
|
|
10
|
+
|
|
11
|
+
/** Maximum number of auto-resume turns before forcing a stop. */
|
|
12
|
+
export const MAX_AUTORESUME_TURNS = 20;
|
|
13
|
+
|
|
14
|
+
/**
|
|
15
|
+
* Controller for scheduling auto-resume actions with a settle window.
|
|
16
|
+
*
|
|
17
|
+
* The settle window ensures that pending async operations (e.g., Pi's
|
|
18
|
+
* internal setTimeout(0) and setTimeout(100) callbacks) have completed
|
|
19
|
+
* before the resume callback fires.
|
|
20
|
+
*/
|
|
21
|
+
export class AutoResumeController {
|
|
22
|
+
private timerId: ReturnType<typeof setTimeout> | null = null;
|
|
23
|
+
private turnCount = 0;
|
|
24
|
+
private cancelled = false;
|
|
25
|
+
|
|
26
|
+
/**
|
|
27
|
+
* Get the current auto-resume turn count.
|
|
28
|
+
*/
|
|
29
|
+
get currentTurnCount(): number {
|
|
30
|
+
return this.turnCount;
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
/**
|
|
34
|
+
* Get the maximum allowed auto-resume turns.
|
|
35
|
+
*/
|
|
36
|
+
get maxTurns(): number {
|
|
37
|
+
return MAX_AUTORESUME_TURNS;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
/**
|
|
41
|
+
* Schedule an auto-resume callback after the settle window elapses.
|
|
42
|
+
*
|
|
43
|
+
* If a resume is already pending, it is cancelled first (debounce behavior).
|
|
44
|
+
* If the turn limit has been reached, the callback is not scheduled.
|
|
45
|
+
*
|
|
46
|
+
* @param message - Description of why the resume is being scheduled
|
|
47
|
+
* @param callback - Function to call after the settle window
|
|
48
|
+
*/
|
|
49
|
+
scheduleResume(message: string, callback: () => void): void {
|
|
50
|
+
// Cancel any existing pending resume
|
|
51
|
+
this.cancelResume();
|
|
52
|
+
|
|
53
|
+
// Enforce turn limit
|
|
54
|
+
if (this.turnCount >= MAX_AUTORESUME_TURNS) {
|
|
55
|
+
return;
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
this.cancelled = false;
|
|
59
|
+
this.turnCount++;
|
|
60
|
+
|
|
61
|
+
this.timerId = setTimeout(() => {
|
|
62
|
+
if (!this.cancelled) {
|
|
63
|
+
this.timerId = null;
|
|
64
|
+
callback();
|
|
65
|
+
}
|
|
66
|
+
}, SETTLE_WINDOW_MS);
|
|
67
|
+
|
|
68
|
+
// Prevent the timer from keeping the process alive
|
|
69
|
+
if (this.timerId && typeof this.timerId === "object" && "unref" in this.timerId) {
|
|
70
|
+
this.timerId.unref();
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
/**
|
|
75
|
+
* Cancel any pending auto-resume.
|
|
76
|
+
*/
|
|
77
|
+
cancelResume(): void {
|
|
78
|
+
if (this.timerId !== null) {
|
|
79
|
+
clearTimeout(this.timerId);
|
|
80
|
+
this.timerId = null;
|
|
81
|
+
}
|
|
82
|
+
this.cancelled = true;
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
/**
|
|
86
|
+
* Check whether an auto-resume is currently pending.
|
|
87
|
+
*/
|
|
88
|
+
hasPendingResume(): boolean {
|
|
89
|
+
return this.timerId !== null;
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
/**
|
|
93
|
+
* Reset the turn counter. Called when a new agent starts processing
|
|
94
|
+
* to allow a fresh set of auto-resume turns.
|
|
95
|
+
*/
|
|
96
|
+
resetTurnCount(): void {
|
|
97
|
+
this.turnCount = 0;
|
|
98
|
+
this.cancelled = false;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
@@ -4,7 +4,17 @@ import { appendEvent } from "../state/event-log.ts";
|
|
|
4
4
|
import { loadRunManifestById, saveRunManifest, updateRunStatus } from "../state/state-store.ts";
|
|
5
5
|
import { allWorkflows, discoverWorkflows } from "../workflows/discover-workflows.ts";
|
|
6
6
|
import { loadConfig } from "../config/config.ts";
|
|
7
|
-
|
|
7
|
+
// Heavy runtime — lazy-loaded to avoid pulling team-runner into background-runner
|
|
8
|
+
// at module load time. Only needed when a background run actually starts.
|
|
9
|
+
import type { executeTeamRun as ExecuteTeamRunFn } from "./team-runner.ts";
|
|
10
|
+
let _cachedExecuteTeamRun: typeof ExecuteTeamRunFn | undefined;
|
|
11
|
+
async function executeTeamRun(...args: Parameters<typeof ExecuteTeamRunFn>): Promise<Awaited<ReturnType<typeof ExecuteTeamRunFn>>> {
|
|
12
|
+
if (!_cachedExecuteTeamRun) {
|
|
13
|
+
const mod = await import("./team-runner.ts");
|
|
14
|
+
_cachedExecuteTeamRun = mod.executeTeamRun;
|
|
15
|
+
}
|
|
16
|
+
return _cachedExecuteTeamRun(...args);
|
|
17
|
+
}
|
|
8
18
|
import { resolveCrewRuntime, runtimeResolutionState } from "./runtime-resolver.ts";
|
|
9
19
|
import { directTeamAndWorkflowFromRun } from "./direct-run.ts";
|
|
10
20
|
import { expandParallelResearchWorkflow } from "./parallel-research.ts";
|
|
@@ -1,89 +1,89 @@
|
|
|
1
|
-
import { CrewCancellationError, type CancellationReason, cancellationReasonFromUnknown } from "./cancellation.ts";
|
|
2
|
-
|
|
3
|
-
export interface CancellationTokenState {
|
|
4
|
-
aborted: boolean;
|
|
5
|
-
reason?: CancellationReason;
|
|
6
|
-
lastHeartbeatAt?: string;
|
|
7
|
-
lastHeartbeatStage?: string;
|
|
8
|
-
}
|
|
9
|
-
|
|
10
|
-
export interface CancellationTokenOptions {
|
|
11
|
-
signal?: AbortSignal;
|
|
12
|
-
onHeartbeat?: (state: CancellationTokenState) => void;
|
|
13
|
-
now?: () => Date;
|
|
14
|
-
}
|
|
15
|
-
|
|
16
|
-
export class CancellationToken {
|
|
17
|
-
readonly #controller = new AbortController();
|
|
18
|
-
readonly #onHeartbeat?: (state: CancellationTokenState) => void;
|
|
19
|
-
readonly #now: () => Date;
|
|
20
|
-
#reason?: CancellationReason;
|
|
21
|
-
#lastHeartbeatAt?: string;
|
|
22
|
-
#lastHeartbeatStage?: string;
|
|
23
|
-
|
|
24
|
-
constructor(options: CancellationTokenOptions = {}) {
|
|
25
|
-
this.#onHeartbeat = options.onHeartbeat;
|
|
26
|
-
this.#now = options.now ?? (() => new Date());
|
|
27
|
-
if (options.signal?.aborted) this.abort(options.signal.reason);
|
|
28
|
-
else if (options.signal) options.signal.addEventListener("abort", () => this.abort(options.signal?.reason), { once: true });
|
|
29
|
-
}
|
|
30
|
-
|
|
31
|
-
get signal(): AbortSignal { return this.#controller.signal; }
|
|
32
|
-
get aborted(): boolean { return this.#controller.signal.aborted; }
|
|
33
|
-
get reason(): CancellationReason | undefined { return this.#reason; }
|
|
34
|
-
get lastHeartbeatAt(): string | undefined { return this.#lastHeartbeatAt; }
|
|
35
|
-
get lastHeartbeatStage(): string | undefined { return this.#lastHeartbeatStage; }
|
|
36
|
-
|
|
37
|
-
heartbeat(stage?: string): CancellationTokenState {
|
|
38
|
-
this.throwIfCancelled();
|
|
39
|
-
this.#lastHeartbeatAt = this.#now().toISOString();
|
|
40
|
-
this.#lastHeartbeatStage = stage;
|
|
41
|
-
const state = this.state();
|
|
42
|
-
this.#onHeartbeat?.(state);
|
|
43
|
-
return state;
|
|
44
|
-
}
|
|
45
|
-
|
|
46
|
-
throwIfCancelled(): void {
|
|
47
|
-
if (this.aborted) throw new CrewCancellationError(this.#reason ?? cancellationReasonFromUnknown(this.#controller.signal.reason));
|
|
48
|
-
}
|
|
49
|
-
|
|
50
|
-
abort(reason?: unknown): void {
|
|
51
|
-
if (this.aborted) return;
|
|
52
|
-
this.#reason = cancellationReasonFromUnknown(reason);
|
|
53
|
-
this.#controller.abort(this.#reason);
|
|
54
|
-
}
|
|
55
|
-
|
|
56
|
-
wait(ms: number): Promise<void> {
|
|
57
|
-
this.throwIfCancelled();
|
|
58
|
-
if (ms <= 0) return Promise.resolve();
|
|
59
|
-
return new Promise((resolve, reject) => {
|
|
60
|
-
let timeout: NodeJS.Timeout | undefined;
|
|
61
|
-
const cleanup = (): void => {
|
|
62
|
-
if (timeout) clearTimeout(timeout);
|
|
63
|
-
this.signal.removeEventListener("abort", onAbort);
|
|
64
|
-
};
|
|
65
|
-
const onAbort = (): void => {
|
|
66
|
-
cleanup();
|
|
67
|
-
reject(new CrewCancellationError(this.#reason ?? cancellationReasonFromUnknown(this.signal.reason)));
|
|
68
|
-
};
|
|
69
|
-
timeout = setTimeout(() => {
|
|
70
|
-
cleanup();
|
|
71
|
-
resolve();
|
|
72
|
-
}, ms);
|
|
73
|
-
this.signal.addEventListener("abort", onAbort, { once: true });
|
|
74
|
-
});
|
|
75
|
-
}
|
|
76
|
-
|
|
77
|
-
state(): CancellationTokenState {
|
|
78
|
-
return {
|
|
79
|
-
aborted: this.aborted,
|
|
80
|
-
...(this.#reason ? { reason: this.#reason } : {}),
|
|
81
|
-
...(this.#lastHeartbeatAt ? { lastHeartbeatAt: this.#lastHeartbeatAt } : {}),
|
|
82
|
-
...(this.#lastHeartbeatStage ? { lastHeartbeatStage: this.#lastHeartbeatStage } : {}),
|
|
83
|
-
};
|
|
84
|
-
}
|
|
85
|
-
}
|
|
86
|
-
|
|
87
|
-
export function createCancellationToken(options: CancellationTokenOptions = {}): CancellationToken {
|
|
88
|
-
return new CancellationToken(options);
|
|
89
|
-
}
|
|
1
|
+
import { CrewCancellationError, type CancellationReason, cancellationReasonFromUnknown } from "./cancellation.ts";
|
|
2
|
+
|
|
3
|
+
export interface CancellationTokenState {
|
|
4
|
+
aborted: boolean;
|
|
5
|
+
reason?: CancellationReason;
|
|
6
|
+
lastHeartbeatAt?: string;
|
|
7
|
+
lastHeartbeatStage?: string;
|
|
8
|
+
}
|
|
9
|
+
|
|
10
|
+
export interface CancellationTokenOptions {
|
|
11
|
+
signal?: AbortSignal;
|
|
12
|
+
onHeartbeat?: (state: CancellationTokenState) => void;
|
|
13
|
+
now?: () => Date;
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
export class CancellationToken {
|
|
17
|
+
readonly #controller = new AbortController();
|
|
18
|
+
readonly #onHeartbeat?: (state: CancellationTokenState) => void;
|
|
19
|
+
readonly #now: () => Date;
|
|
20
|
+
#reason?: CancellationReason;
|
|
21
|
+
#lastHeartbeatAt?: string;
|
|
22
|
+
#lastHeartbeatStage?: string;
|
|
23
|
+
|
|
24
|
+
constructor(options: CancellationTokenOptions = {}) {
|
|
25
|
+
this.#onHeartbeat = options.onHeartbeat;
|
|
26
|
+
this.#now = options.now ?? (() => new Date());
|
|
27
|
+
if (options.signal?.aborted) this.abort(options.signal.reason);
|
|
28
|
+
else if (options.signal) options.signal.addEventListener("abort", () => this.abort(options.signal?.reason), { once: true });
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
get signal(): AbortSignal { return this.#controller.signal; }
|
|
32
|
+
get aborted(): boolean { return this.#controller.signal.aborted; }
|
|
33
|
+
get reason(): CancellationReason | undefined { return this.#reason; }
|
|
34
|
+
get lastHeartbeatAt(): string | undefined { return this.#lastHeartbeatAt; }
|
|
35
|
+
get lastHeartbeatStage(): string | undefined { return this.#lastHeartbeatStage; }
|
|
36
|
+
|
|
37
|
+
heartbeat(stage?: string): CancellationTokenState {
|
|
38
|
+
this.throwIfCancelled();
|
|
39
|
+
this.#lastHeartbeatAt = this.#now().toISOString();
|
|
40
|
+
this.#lastHeartbeatStage = stage;
|
|
41
|
+
const state = this.state();
|
|
42
|
+
this.#onHeartbeat?.(state);
|
|
43
|
+
return state;
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
throwIfCancelled(): void {
|
|
47
|
+
if (this.aborted) throw new CrewCancellationError(this.#reason ?? cancellationReasonFromUnknown(this.#controller.signal.reason));
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
abort(reason?: unknown): void {
|
|
51
|
+
if (this.aborted) return;
|
|
52
|
+
this.#reason = cancellationReasonFromUnknown(reason);
|
|
53
|
+
this.#controller.abort(this.#reason);
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
wait(ms: number): Promise<void> {
|
|
57
|
+
this.throwIfCancelled();
|
|
58
|
+
if (ms <= 0) return Promise.resolve();
|
|
59
|
+
return new Promise((resolve, reject) => {
|
|
60
|
+
let timeout: NodeJS.Timeout | undefined;
|
|
61
|
+
const cleanup = (): void => {
|
|
62
|
+
if (timeout) clearTimeout(timeout);
|
|
63
|
+
this.signal.removeEventListener("abort", onAbort);
|
|
64
|
+
};
|
|
65
|
+
const onAbort = (): void => {
|
|
66
|
+
cleanup();
|
|
67
|
+
reject(new CrewCancellationError(this.#reason ?? cancellationReasonFromUnknown(this.signal.reason)));
|
|
68
|
+
};
|
|
69
|
+
timeout = setTimeout(() => {
|
|
70
|
+
cleanup();
|
|
71
|
+
resolve();
|
|
72
|
+
}, ms);
|
|
73
|
+
this.signal.addEventListener("abort", onAbort, { once: true });
|
|
74
|
+
});
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
state(): CancellationTokenState {
|
|
78
|
+
return {
|
|
79
|
+
aborted: this.aborted,
|
|
80
|
+
...(this.#reason ? { reason: this.#reason } : {}),
|
|
81
|
+
...(this.#lastHeartbeatAt ? { lastHeartbeatAt: this.#lastHeartbeatAt } : {}),
|
|
82
|
+
...(this.#lastHeartbeatStage ? { lastHeartbeatStage: this.#lastHeartbeatStage } : {}),
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
|
|
87
|
+
export function createCancellationToken(options: CancellationTokenOptions = {}): CancellationToken {
|
|
88
|
+
return new CancellationToken(options);
|
|
89
|
+
}
|