pi-crew 0.1.44 → 0.1.45

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.
Files changed (142) hide show
  1. package/CHANGELOG.md +27 -0
  2. package/agents/analyst.md +11 -11
  3. package/agents/critic.md +11 -11
  4. package/agents/executor.md +11 -11
  5. package/agents/explorer.md +11 -11
  6. package/agents/planner.md +11 -11
  7. package/agents/reviewer.md +11 -11
  8. package/agents/security-reviewer.md +11 -11
  9. package/agents/test-engineer.md +11 -11
  10. package/agents/verifier.md +11 -11
  11. package/agents/writer.md +11 -11
  12. package/docs/refactor-tasks-phase3.md +394 -394
  13. package/docs/refactor-tasks-phase4.md +564 -564
  14. package/docs/refactor-tasks-phase5.md +402 -402
  15. package/docs/refactor-tasks-phase6.md +662 -662
  16. package/docs/research-extension-examples.md +297 -297
  17. package/docs/research-extension-system.md +324 -324
  18. package/docs/research-optimization-plan.md +548 -548
  19. package/docs/research-phase10-distillation.md +198 -198
  20. package/docs/research-phase11-distillation.md +201 -201
  21. package/docs/research-pi-coding-agent.md +357 -357
  22. package/docs/research-source-pi-crew-reference.md +174 -174
  23. package/docs/runtime-flow.md +148 -148
  24. package/docs/source-runtime-refactor-map.md +83 -83
  25. package/index.ts +6 -6
  26. package/package.json +1 -1
  27. package/src/agents/agent-serializer.ts +34 -34
  28. package/src/extension/cross-extension-rpc.ts +82 -82
  29. package/src/extension/register.ts +8 -1
  30. package/src/extension/registration/commands.ts +18 -2
  31. package/src/extension/registration/compaction-guard.ts +125 -125
  32. package/src/extension/registration/subagent-tools.ts +148 -148
  33. package/src/extension/registration/team-tool.ts +26 -8
  34. package/src/extension/run-bundle-schema.ts +89 -89
  35. package/src/extension/run-maintenance.ts +43 -43
  36. package/src/extension/team-tool/cancel.ts +105 -102
  37. package/src/extension/team-tool/context.ts +1 -0
  38. package/src/extension/team-tool/handle-settings.ts +188 -188
  39. package/src/extension/team-tool/inspect.ts +41 -41
  40. package/src/extension/team-tool/lifecycle-actions.ts +79 -79
  41. package/src/extension/team-tool/plan.ts +19 -19
  42. package/src/extension/team-tool/respond.ts +83 -66
  43. package/src/extension/team-tool/run.ts +1 -0
  44. package/src/i18n.ts +184 -184
  45. package/src/observability/exporters/otlp-exporter.ts +77 -77
  46. package/src/prompt/prompt-runtime.ts +72 -72
  47. package/src/runtime/agent-control.ts +63 -63
  48. package/src/runtime/agent-memory.ts +72 -72
  49. package/src/runtime/agent-observability.ts +114 -114
  50. package/src/runtime/async-marker.ts +26 -26
  51. package/src/runtime/attention-events.ts +28 -28
  52. package/src/runtime/background-runner.ts +53 -53
  53. package/src/runtime/child-pi.ts +444 -444
  54. package/src/runtime/completion-guard.ts +190 -190
  55. package/src/runtime/crew-agent-records.ts +8 -0
  56. package/src/runtime/delivery-coordinator.ts +153 -142
  57. package/src/runtime/direct-run.ts +35 -35
  58. package/src/runtime/foreground-control.ts +82 -82
  59. package/src/runtime/green-contract.ts +46 -46
  60. package/src/runtime/group-join.ts +106 -106
  61. package/src/runtime/heartbeat-gradient.ts +28 -28
  62. package/src/runtime/heartbeat-watcher.ts +124 -124
  63. package/src/runtime/live-agent-control.ts +87 -87
  64. package/src/runtime/live-agent-manager.ts +85 -85
  65. package/src/runtime/live-control-realtime.ts +36 -36
  66. package/src/runtime/live-session-runtime.ts +305 -305
  67. package/src/runtime/overflow-recovery.ts +175 -156
  68. package/src/runtime/parallel-research.ts +44 -44
  69. package/src/runtime/pi-json-output.ts +111 -111
  70. package/src/runtime/policy-engine.ts +79 -79
  71. package/src/runtime/progress-event-coalescer.ts +43 -43
  72. package/src/runtime/recovery-recipes.ts +74 -74
  73. package/src/runtime/retry-executor.ts +64 -64
  74. package/src/runtime/role-permission.ts +39 -39
  75. package/src/runtime/session-resources.ts +25 -25
  76. package/src/runtime/session-snapshot.ts +59 -59
  77. package/src/runtime/session-usage.ts +79 -79
  78. package/src/runtime/sidechain-output.ts +29 -29
  79. package/src/runtime/stale-reconciler.ts +199 -179
  80. package/src/runtime/supervisor-contact.ts +59 -59
  81. package/src/runtime/task-display.ts +38 -38
  82. package/src/runtime/task-output-context.ts +127 -127
  83. package/src/runtime/task-runner/live-executor.ts +101 -101
  84. package/src/runtime/task-runner/progress.ts +119 -119
  85. package/src/runtime/task-runner/result-utils.ts +14 -14
  86. package/src/runtime/task-runner/state-helpers.ts +22 -22
  87. package/src/runtime/team-runner.ts +13 -4
  88. package/src/runtime/worker-heartbeat.ts +21 -21
  89. package/src/runtime/worker-startup.ts +57 -57
  90. package/src/state/state-store.ts +43 -0
  91. package/src/state/task-claims.ts +44 -44
  92. package/src/state/types.ts +2 -0
  93. package/src/state/usage.ts +29 -29
  94. package/src/subagents/async-entry.ts +1 -1
  95. package/src/subagents/index.ts +3 -3
  96. package/src/subagents/live/control.ts +1 -1
  97. package/src/subagents/live/manager.ts +1 -1
  98. package/src/subagents/live/realtime.ts +1 -1
  99. package/src/subagents/live/session-runtime.ts +1 -1
  100. package/src/subagents/manager.ts +1 -1
  101. package/src/subagents/spawn.ts +1 -1
  102. package/src/teams/team-serializer.ts +38 -38
  103. package/src/types/diff.d.ts +18 -18
  104. package/src/ui/crew-footer.ts +101 -101
  105. package/src/ui/crew-select-list.ts +111 -111
  106. package/src/ui/crew-widget.ts +5 -1
  107. package/src/ui/dashboard-panes/mailbox-pane.ts +2 -1
  108. package/src/ui/dashboard-panes/metrics-pane.ts +34 -34
  109. package/src/ui/dynamic-border.ts +25 -25
  110. package/src/ui/layout-primitives.ts +106 -106
  111. package/src/ui/loaders.ts +158 -158
  112. package/src/ui/powerbar-publisher.ts +1 -1
  113. package/src/ui/render-diff.ts +119 -119
  114. package/src/ui/render-scheduler.ts +143 -143
  115. package/src/ui/run-snapshot-cache.ts +56 -37
  116. package/src/ui/snapshot-types.ts +5 -0
  117. package/src/ui/spinner.ts +17 -17
  118. package/src/ui/status-colors.ts +58 -58
  119. package/src/ui/syntax-highlight.ts +116 -116
  120. package/src/utils/atomic-write.ts +33 -33
  121. package/src/utils/completion-dedupe.ts +63 -63
  122. package/src/utils/frontmatter.ts +68 -68
  123. package/src/utils/git.ts +262 -262
  124. package/src/utils/ids.ts +12 -12
  125. package/src/utils/names.ts +27 -27
  126. package/src/utils/redaction.ts +44 -44
  127. package/src/utils/safe-paths.ts +47 -47
  128. package/src/utils/sleep.ts +32 -32
  129. package/src/workflows/validate-workflow.ts +40 -40
  130. package/src/worktree/branch-freshness.ts +45 -45
  131. package/teams/default.team.md +12 -12
  132. package/teams/fast-fix.team.md +11 -11
  133. package/teams/implementation.team.md +18 -18
  134. package/teams/parallel-research.team.md +14 -14
  135. package/teams/research.team.md +11 -11
  136. package/teams/review.team.md +12 -12
  137. package/workflows/default.workflow.md +29 -29
  138. package/workflows/fast-fix.workflow.md +22 -22
  139. package/workflows/implementation.workflow.md +38 -38
  140. package/workflows/parallel-research.workflow.md +46 -46
  141. package/workflows/research.workflow.md +22 -22
  142. package/workflows/review.workflow.md +30 -30
package/src/ui/loaders.ts CHANGED
@@ -1,158 +1,158 @@
1
- import { pad, truncate } from "../utils/visual.ts";
2
- import type { CrewTheme } from "./theme-adapter.ts";
3
- import { asCrewTheme } from "./theme-adapter.ts";
4
- import { DynamicCrewBorder } from "./dynamic-border.ts";
5
-
6
- export interface BorderedLoaderOptions {
7
- message: string;
8
- cancellable?: boolean;
9
- frames?: string[];
10
- intervalMs?: number;
11
- minWidth?: number;
12
- onAbort?: () => void;
13
- }
14
-
15
- const DEFAULT_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
16
-
17
- export class CrewBorderedLoader {
18
- private readonly abortController = new AbortController();
19
- private readonly frameOptions: string[];
20
- private readonly intervalMs: number;
21
- private readonly minWidth: number;
22
- private readonly onAbort?: () => void;
23
- private theme: CrewTheme;
24
- private message: string;
25
- private lineCache = "";
26
- private width = 0;
27
- private startedAt = Date.now();
28
-
29
- constructor(_ui: unknown, themeLike: unknown, options: BorderedLoaderOptions) {
30
- const theme = asCrewTheme(themeLike);
31
- this.theme = theme;
32
- this.message = options.message;
33
- this.minWidth = Math.max(12, options.minWidth ?? 24);
34
- this.onAbort = options.onAbort;
35
- this.frameOptions = options.frames ?? DEFAULT_FRAMES;
36
- this.intervalMs = Math.max(40, options.intervalMs ?? 120);
37
- }
38
-
39
- private spinnerFrame(): string {
40
- if (this.frameOptions.length === 0) return "•";
41
- const elapsed = Date.now() - this.startedAt;
42
- const index = Math.floor(elapsed / this.intervalMs) % this.frameOptions.length;
43
- return this.frameOptions[Math.max(0, index)];
44
- }
45
-
46
- setMessage(message: string): void {
47
- this.message = message;
48
- }
49
-
50
- get signal(): AbortSignal {
51
- return this.abortController.signal;
52
- }
53
-
54
- handleInput(data: string): void {
55
- if (!this.onAbort || this.abortController.signal.aborted) return;
56
- if (data === "c" || data === "q" || data === "\u001b" || data === "\u0003") {
57
- this.abortController.abort();
58
- this.onAbort();
59
- }
60
- }
61
-
62
- render(width: number): string[] {
63
- if (width === this.width && this.lineCache) {
64
- return this.lineCache.split("\n");
65
- }
66
- const innerWidth = Math.max(this.minWidth - 4, 1);
67
- const contentWidth = Math.max(1, Math.min(width - 4, innerWidth));
68
- const frame = this.spinnerFrame();
69
- const loaderLine = ` ${frame} ${truncate(this.message, Math.max(1, contentWidth - 4))} `;
70
- const body = ` ${truncate(loaderLine, contentWidth - 2)} `;
71
- const inner = ` ${pad(body, contentWidth - 1)} `;
72
- const padWidth = Math.max(0, width - (contentWidth + 4));
73
- const leftRightPad = " ".repeat(Math.floor(padWidth / 2));
74
- const widthAwareInner = contentWidth + padWidth;
75
- const border = new DynamicCrewBorder(this.theme).render(widthAwareInner + 2)[0];
76
- const top = `${leftRightPad}${this.theme.fg("border", "┌")}${border}${this.theme.fg("border", "┐")}`;
77
- const line = `${leftRightPad}${this.theme.fg("border", "│")} ${truncate(inner, widthAwareInner)} ${this.theme.fg("border", "│")}`;
78
- const hint = `${leftRightPad}${this.theme.fg("border", "│")}${" ".repeat(widthAwareInner + 2)}${this.theme.fg("border", "│")}`;
79
- const bottom = `${leftRightPad}${this.theme.fg("border", "└")}${border}${this.theme.fg("border", "┘")}`;
80
- const lineWithHint = optionsHint(this.theme, this.message, widthAwareInner);
81
- this.width = width;
82
- const lines = [
83
- top,
84
- line,
85
- `${leftRightPad}│ ${pad(lineWithHint, widthAwareInner)} │`,
86
- hint,
87
- bottom,
88
- ];
89
- this.lineCache = lines.join("\n");
90
- return lines;
91
- }
92
-
93
- invalidate(): void {
94
- this.lineCache = "";
95
- this.width = 0;
96
- }
97
-
98
- dispose(): void {
99
- this.abortController.abort();
100
- }
101
- }
102
-
103
- export interface CountdownTimerOptions {
104
- timeoutMs: number;
105
- onTick: (seconds: number) => void;
106
- onExpire: () => void;
107
- }
108
-
109
- export class CountdownTimer {
110
- private readonly onExpire: () => void;
111
- private readonly onTick: (seconds: number) => void;
112
- private readonly startedAt: number;
113
- private readonly timeoutMs: number;
114
- private timer: ReturnType<typeof setTimeout> | undefined;
115
- private expired = false;
116
-
117
- constructor(options: CountdownTimerOptions) {
118
- this.timeoutMs = Math.max(0, options.timeoutMs);
119
- this.onTick = options.onTick;
120
- this.onExpire = options.onExpire;
121
- this.startedAt = Date.now();
122
- this.onTick(this.secondsLeft());
123
- if (this.timeoutMs === 0) {
124
- this.emitExpire();
125
- return;
126
- }
127
- this.timer = setInterval(() => {
128
- const seconds = this.secondsLeft();
129
- this.onTick(seconds);
130
- if (seconds <= 0) {
131
- this.emitExpire();
132
- }
133
- }, 1000);
134
- }
135
-
136
- private emitExpire(): void {
137
- if (this.expired) return;
138
- this.expired = true;
139
- this.dispose();
140
- this.onExpire();
141
- }
142
-
143
- private secondsLeft(): number {
144
- const remainingMs = this.startedAt + this.timeoutMs - Date.now();
145
- return Math.max(0, Math.ceil(remainingMs / 1000));
146
- }
147
-
148
- dispose(): void {
149
- if (this.timer === undefined) return;
150
- clearInterval(this.timer);
151
- this.timer = undefined;
152
- }
153
- }
154
-
155
- function optionsHint(theme: CrewTheme, message: string, width: number): string {
156
- if (!message) return "";
157
- return truncate(theme.fg("muted", message), width);
158
- }
1
+ import { pad, truncate } from "../utils/visual.ts";
2
+ import type { CrewTheme } from "./theme-adapter.ts";
3
+ import { asCrewTheme } from "./theme-adapter.ts";
4
+ import { DynamicCrewBorder } from "./dynamic-border.ts";
5
+
6
+ export interface BorderedLoaderOptions {
7
+ message: string;
8
+ cancellable?: boolean;
9
+ frames?: string[];
10
+ intervalMs?: number;
11
+ minWidth?: number;
12
+ onAbort?: () => void;
13
+ }
14
+
15
+ const DEFAULT_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
16
+
17
+ export class CrewBorderedLoader {
18
+ private readonly abortController = new AbortController();
19
+ private readonly frameOptions: string[];
20
+ private readonly intervalMs: number;
21
+ private readonly minWidth: number;
22
+ private readonly onAbort?: () => void;
23
+ private theme: CrewTheme;
24
+ private message: string;
25
+ private lineCache = "";
26
+ private width = 0;
27
+ private startedAt = Date.now();
28
+
29
+ constructor(_ui: unknown, themeLike: unknown, options: BorderedLoaderOptions) {
30
+ const theme = asCrewTheme(themeLike);
31
+ this.theme = theme;
32
+ this.message = options.message;
33
+ this.minWidth = Math.max(12, options.minWidth ?? 24);
34
+ this.onAbort = options.onAbort;
35
+ this.frameOptions = options.frames ?? DEFAULT_FRAMES;
36
+ this.intervalMs = Math.max(40, options.intervalMs ?? 120);
37
+ }
38
+
39
+ private spinnerFrame(): string {
40
+ if (this.frameOptions.length === 0) return "•";
41
+ const elapsed = Date.now() - this.startedAt;
42
+ const index = Math.floor(elapsed / this.intervalMs) % this.frameOptions.length;
43
+ return this.frameOptions[Math.max(0, index)];
44
+ }
45
+
46
+ setMessage(message: string): void {
47
+ this.message = message;
48
+ }
49
+
50
+ get signal(): AbortSignal {
51
+ return this.abortController.signal;
52
+ }
53
+
54
+ handleInput(data: string): void {
55
+ if (!this.onAbort || this.abortController.signal.aborted) return;
56
+ if (data === "c" || data === "q" || data === "\u001b" || data === "\u0003") {
57
+ this.abortController.abort();
58
+ this.onAbort();
59
+ }
60
+ }
61
+
62
+ render(width: number): string[] {
63
+ if (width === this.width && this.lineCache) {
64
+ return this.lineCache.split("\n");
65
+ }
66
+ const innerWidth = Math.max(this.minWidth - 4, 1);
67
+ const contentWidth = Math.max(1, Math.min(width - 4, innerWidth));
68
+ const frame = this.spinnerFrame();
69
+ const loaderLine = ` ${frame} ${truncate(this.message, Math.max(1, contentWidth - 4))} `;
70
+ const body = ` ${truncate(loaderLine, contentWidth - 2)} `;
71
+ const inner = ` ${pad(body, contentWidth - 1)} `;
72
+ const padWidth = Math.max(0, width - (contentWidth + 4));
73
+ const leftRightPad = " ".repeat(Math.floor(padWidth / 2));
74
+ const widthAwareInner = contentWidth + padWidth;
75
+ const border = new DynamicCrewBorder(this.theme).render(widthAwareInner + 2)[0];
76
+ const top = `${leftRightPad}${this.theme.fg("border", "┌")}${border}${this.theme.fg("border", "┐")}`;
77
+ const line = `${leftRightPad}${this.theme.fg("border", "│")} ${truncate(inner, widthAwareInner)} ${this.theme.fg("border", "│")}`;
78
+ const hint = `${leftRightPad}${this.theme.fg("border", "│")}${" ".repeat(widthAwareInner + 2)}${this.theme.fg("border", "│")}`;
79
+ const bottom = `${leftRightPad}${this.theme.fg("border", "└")}${border}${this.theme.fg("border", "┘")}`;
80
+ const lineWithHint = optionsHint(this.theme, this.message, widthAwareInner);
81
+ this.width = width;
82
+ const lines = [
83
+ top,
84
+ line,
85
+ `${leftRightPad}│ ${pad(lineWithHint, widthAwareInner)} │`,
86
+ hint,
87
+ bottom,
88
+ ];
89
+ this.lineCache = lines.join("\n");
90
+ return lines;
91
+ }
92
+
93
+ invalidate(): void {
94
+ this.lineCache = "";
95
+ this.width = 0;
96
+ }
97
+
98
+ dispose(): void {
99
+ this.abortController.abort();
100
+ }
101
+ }
102
+
103
+ export interface CountdownTimerOptions {
104
+ timeoutMs: number;
105
+ onTick: (seconds: number) => void;
106
+ onExpire: () => void;
107
+ }
108
+
109
+ export class CountdownTimer {
110
+ private readonly onExpire: () => void;
111
+ private readonly onTick: (seconds: number) => void;
112
+ private readonly startedAt: number;
113
+ private readonly timeoutMs: number;
114
+ private timer: ReturnType<typeof setTimeout> | undefined;
115
+ private expired = false;
116
+
117
+ constructor(options: CountdownTimerOptions) {
118
+ this.timeoutMs = Math.max(0, options.timeoutMs);
119
+ this.onTick = options.onTick;
120
+ this.onExpire = options.onExpire;
121
+ this.startedAt = Date.now();
122
+ this.onTick(this.secondsLeft());
123
+ if (this.timeoutMs === 0) {
124
+ this.emitExpire();
125
+ return;
126
+ }
127
+ this.timer = setInterval(() => {
128
+ const seconds = this.secondsLeft();
129
+ this.onTick(seconds);
130
+ if (seconds <= 0) {
131
+ this.emitExpire();
132
+ }
133
+ }, 1000);
134
+ }
135
+
136
+ private emitExpire(): void {
137
+ if (this.expired) return;
138
+ this.expired = true;
139
+ this.dispose();
140
+ this.onExpire();
141
+ }
142
+
143
+ private secondsLeft(): number {
144
+ const remainingMs = this.startedAt + this.timeoutMs - Date.now();
145
+ return Math.max(0, Math.ceil(remainingMs / 1000));
146
+ }
147
+
148
+ dispose(): void {
149
+ if (this.timer === undefined) return;
150
+ clearInterval(this.timer);
151
+ this.timer = undefined;
152
+ }
153
+ }
154
+
155
+ function optionsHint(theme: CrewTheme, message: string, width: number): string {
156
+ if (!message) return "";
157
+ return truncate(theme.fg("muted", message), width);
158
+ }
@@ -92,7 +92,7 @@ export function updatePiCrewPowerbar(events: EventBus, cwd: string, config?: Cre
92
92
  const agents = active.flatMap((item) => item.agents);
93
93
  const tasks = active.flatMap((item) => item.tasks);
94
94
  const running = agents.filter((agent) => agent.status === "running").length;
95
- const waiting = active.reduce((sum, item) => sum + (item.snapshot?.progress.queued ?? item.tasks.reduce((s, t) => s + (t.status === "queued" ? 1 : 0), 0)), 0);
95
+ const waiting = active.reduce((sum, item) => sum + (item.snapshot ? item.snapshot.progress.queued + (item.snapshot.progress.waiting ?? 0) : item.tasks.reduce((s, t) => s + (t.status === "queued" || t.status === "waiting" ? 1 : 0), 0)), 0);
96
96
  const completed = active.reduce((sum, item) => sum + (item.snapshot?.progress.completed ?? item.tasks.reduce((s, t) => s + (t.status === "completed" ? 1 : 0), 0)), 0);
97
97
  const total = Math.max(1, active.reduce((sum, item) => sum + (item.snapshot?.progress.total ?? item.tasks.length), 0) || agents.length);
98
98
  const usage = aggregateUsage(tasks);
@@ -1,119 +1,119 @@
1
- import * as Diff from "diff";
2
- import type { CrewTheme } from "./theme-adapter.ts";
3
- import { asCrewTheme } from "./theme-adapter.ts";
4
-
5
- interface ParsedDiffLine {
6
- prefix: string;
7
- lineNum: string; content: string;
8
- }
9
-
10
- interface DiffLineContent {
11
- lineNum: string;
12
- content: string;
13
- }
14
-
15
- function parseDiffLine(line: string): ParsedDiffLine | null {
16
- const match = line.match(/^([+-\s])(\s*\d*)\s(.*)$/);
17
- if (!match) return null;
18
- return { prefix: match[1], lineNum: match[2], content: match[3] };
19
- }
20
-
21
- function replaceTabs(text: string): string {
22
- return text.replace(/\t/g, " ");
23
- }
24
-
25
- function renderIntraLineDiff(theme: CrewTheme, oldContent: string, newContent: string): { removedLine: string; addedLine: string } {
26
- const wordDiff = Diff.diffWords(oldContent, newContent);
27
- let removedLine = "";
28
- let addedLine = "";
29
- let isFirstRemoved = true;
30
- let isFirstAdded = true;
31
-
32
- for (const part of wordDiff) {
33
- if (part.removed) {
34
- let value = part.value;
35
- if (isFirstRemoved) {
36
- const leadingWs = value.match(/^(\s*)/)?.[1] ?? "";
37
- value = value.slice(leadingWs.length);
38
- removedLine += leadingWs;
39
- isFirstRemoved = false;
40
- }
41
- if (value) removedLine += theme.inverse?.(value) ?? value;
42
- } else if (part.added) {
43
- let value = part.value;
44
- if (isFirstAdded) {
45
- const leadingWs = value.match(/^(\s*)/)?.[1] ?? "";
46
- value = value.slice(leadingWs.length);
47
- addedLine += leadingWs;
48
- isFirstAdded = false;
49
- }
50
- if (value) addedLine += theme.inverse?.(value) ?? value;
51
- } else {
52
- removedLine += part.value;
53
- addedLine += part.value;
54
- }
55
- }
56
-
57
- return { removedLine, addedLine };
58
- }
59
-
60
- export interface RenderDiffOptions {
61
- filePath?: string;
62
- theme?: unknown;
63
- }
64
-
65
- export function renderDiff(diffText: string, options: RenderDiffOptions = {}): string {
66
- const theme = asCrewTheme(options.theme);
67
- const lines = diffText.split("\n");
68
- const result: string[] = [];
69
- let i = 0;
70
-
71
- while (i < lines.length) {
72
- const line = lines[i] ?? "";
73
- const parsed = parseDiffLine(line);
74
- if (!parsed) {
75
- result.push(theme.fg("toolDiffContext", line));
76
- i++;
77
- continue;
78
- }
79
-
80
- if (parsed.prefix === "-") {
81
- const removedLines: DiffLineContent[] = [];
82
- while (i < lines.length) {
83
- const nextParsed = parseDiffLine(lines[i] ?? "");
84
- if (!nextParsed || nextParsed.prefix !== "-") break;
85
- removedLines.push({ lineNum: nextParsed.lineNum, content: nextParsed.content });
86
- i++;
87
- }
88
-
89
- const addedLines: DiffLineContent[] = [];
90
- while (i < lines.length) {
91
- const nextParsed = parseDiffLine(lines[i] ?? "");
92
- if (!nextParsed || nextParsed.prefix !== "+") break;
93
- addedLines.push({ lineNum: nextParsed.lineNum, content: nextParsed.content });
94
- i++;
95
- }
96
-
97
- if (removedLines.length === 1 && addedLines.length === 1) {
98
- const { removedLine, addedLine } = renderIntraLineDiff(theme, replaceTabs(removedLines[0]!.content), replaceTabs(addedLines[0]!.content));
99
- result.push(theme.fg("toolDiffRemoved", `-${removedLines[0]!.lineNum} ${removedLine}`));
100
- result.push(theme.fg("toolDiffAdded", `+${addedLines[0]!.lineNum} ${addedLine}`));
101
- } else {
102
- for (const removed of removedLines) {
103
- result.push(theme.fg("toolDiffRemoved", `-${removed.lineNum} ${replaceTabs(removed.content)}`));
104
- }
105
- for (const added of addedLines) {
106
- result.push(theme.fg("toolDiffAdded", `+${added.lineNum} ${replaceTabs(added.content)}`));
107
- }
108
- }
109
- } else if (parsed.prefix === "+") {
110
- result.push(theme.fg("toolDiffAdded", `+${parsed.lineNum} ${replaceTabs(parsed.content)}`));
111
- i++;
112
- } else {
113
- result.push(theme.fg("toolDiffContext", ` ${parsed.lineNum} ${replaceTabs(parsed.content)}`));
114
- i++;
115
- }
116
- }
117
-
118
- return result.join("\n");
119
- }
1
+ import * as Diff from "diff";
2
+ import type { CrewTheme } from "./theme-adapter.ts";
3
+ import { asCrewTheme } from "./theme-adapter.ts";
4
+
5
+ interface ParsedDiffLine {
6
+ prefix: string;
7
+ lineNum: string; content: string;
8
+ }
9
+
10
+ interface DiffLineContent {
11
+ lineNum: string;
12
+ content: string;
13
+ }
14
+
15
+ function parseDiffLine(line: string): ParsedDiffLine | null {
16
+ const match = line.match(/^([+-\s])(\s*\d*)\s(.*)$/);
17
+ if (!match) return null;
18
+ return { prefix: match[1], lineNum: match[2], content: match[3] };
19
+ }
20
+
21
+ function replaceTabs(text: string): string {
22
+ return text.replace(/\t/g, " ");
23
+ }
24
+
25
+ function renderIntraLineDiff(theme: CrewTheme, oldContent: string, newContent: string): { removedLine: string; addedLine: string } {
26
+ const wordDiff = Diff.diffWords(oldContent, newContent);
27
+ let removedLine = "";
28
+ let addedLine = "";
29
+ let isFirstRemoved = true;
30
+ let isFirstAdded = true;
31
+
32
+ for (const part of wordDiff) {
33
+ if (part.removed) {
34
+ let value = part.value;
35
+ if (isFirstRemoved) {
36
+ const leadingWs = value.match(/^(\s*)/)?.[1] ?? "";
37
+ value = value.slice(leadingWs.length);
38
+ removedLine += leadingWs;
39
+ isFirstRemoved = false;
40
+ }
41
+ if (value) removedLine += theme.inverse?.(value) ?? value;
42
+ } else if (part.added) {
43
+ let value = part.value;
44
+ if (isFirstAdded) {
45
+ const leadingWs = value.match(/^(\s*)/)?.[1] ?? "";
46
+ value = value.slice(leadingWs.length);
47
+ addedLine += leadingWs;
48
+ isFirstAdded = false;
49
+ }
50
+ if (value) addedLine += theme.inverse?.(value) ?? value;
51
+ } else {
52
+ removedLine += part.value;
53
+ addedLine += part.value;
54
+ }
55
+ }
56
+
57
+ return { removedLine, addedLine };
58
+ }
59
+
60
+ export interface RenderDiffOptions {
61
+ filePath?: string;
62
+ theme?: unknown;
63
+ }
64
+
65
+ export function renderDiff(diffText: string, options: RenderDiffOptions = {}): string {
66
+ const theme = asCrewTheme(options.theme);
67
+ const lines = diffText.split("\n");
68
+ const result: string[] = [];
69
+ let i = 0;
70
+
71
+ while (i < lines.length) {
72
+ const line = lines[i] ?? "";
73
+ const parsed = parseDiffLine(line);
74
+ if (!parsed) {
75
+ result.push(theme.fg("toolDiffContext", line));
76
+ i++;
77
+ continue;
78
+ }
79
+
80
+ if (parsed.prefix === "-") {
81
+ const removedLines: DiffLineContent[] = [];
82
+ while (i < lines.length) {
83
+ const nextParsed = parseDiffLine(lines[i] ?? "");
84
+ if (!nextParsed || nextParsed.prefix !== "-") break;
85
+ removedLines.push({ lineNum: nextParsed.lineNum, content: nextParsed.content });
86
+ i++;
87
+ }
88
+
89
+ const addedLines: DiffLineContent[] = [];
90
+ while (i < lines.length) {
91
+ const nextParsed = parseDiffLine(lines[i] ?? "");
92
+ if (!nextParsed || nextParsed.prefix !== "+") break;
93
+ addedLines.push({ lineNum: nextParsed.lineNum, content: nextParsed.content });
94
+ i++;
95
+ }
96
+
97
+ if (removedLines.length === 1 && addedLines.length === 1) {
98
+ const { removedLine, addedLine } = renderIntraLineDiff(theme, replaceTabs(removedLines[0]!.content), replaceTabs(addedLines[0]!.content));
99
+ result.push(theme.fg("toolDiffRemoved", `-${removedLines[0]!.lineNum} ${removedLine}`));
100
+ result.push(theme.fg("toolDiffAdded", `+${addedLines[0]!.lineNum} ${addedLine}`));
101
+ } else {
102
+ for (const removed of removedLines) {
103
+ result.push(theme.fg("toolDiffRemoved", `-${removed.lineNum} ${replaceTabs(removed.content)}`));
104
+ }
105
+ for (const added of addedLines) {
106
+ result.push(theme.fg("toolDiffAdded", `+${added.lineNum} ${replaceTabs(added.content)}`));
107
+ }
108
+ }
109
+ } else if (parsed.prefix === "+") {
110
+ result.push(theme.fg("toolDiffAdded", `+${parsed.lineNum} ${replaceTabs(parsed.content)}`));
111
+ i++;
112
+ } else {
113
+ result.push(theme.fg("toolDiffContext", ` ${parsed.lineNum} ${replaceTabs(parsed.content)}`));
114
+ i++;
115
+ }
116
+ }
117
+
118
+ return result.join("\n");
119
+ }