@aria-cli/tools 1.0.12 → 1.0.14

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 (233) hide show
  1. package/dist/index.js +378 -70
  2. package/dist/network-runtime/index.js +8 -12
  3. package/dist-cjs/index.js +400 -435
  4. package/dist-cjs/network-runtime/index.js +8 -172
  5. package/package.json +8 -6
  6. package/dist/.tsbuildinfo +0 -1
  7. package/dist/ask-user-interaction.js +0 -22
  8. package/dist/cache/web-cache.js +0 -66
  9. package/dist/definitions/arion.js +0 -104
  10. package/dist/definitions/browser/browser.js +0 -418
  11. package/dist/definitions/browser/index.js +0 -4
  12. package/dist/definitions/browser/pw-downloads.js +0 -114
  13. package/dist/definitions/browser/pw-interactions.js +0 -199
  14. package/dist/definitions/browser/pw-responses.js +0 -76
  15. package/dist/definitions/browser/pw-session.js +0 -310
  16. package/dist/definitions/browser/pw-shared.js +0 -66
  17. package/dist/definitions/browser/pw-snapshot.js +0 -301
  18. package/dist/definitions/browser/pw-state.js +0 -62
  19. package/dist/definitions/browser/types.js +0 -4
  20. package/dist/definitions/code-intelligence.js +0 -470
  21. package/dist/definitions/core.js +0 -109
  22. package/dist/definitions/delegation.js +0 -512
  23. package/dist/definitions/deploy.js +0 -65
  24. package/dist/definitions/filesystem.js +0 -196
  25. package/dist/definitions/frg.js +0 -63
  26. package/dist/definitions/index.js +0 -20
  27. package/dist/definitions/memory.js +0 -123
  28. package/dist/definitions/messaging.js +0 -625
  29. package/dist/definitions/meta.js +0 -349
  30. package/dist/definitions/network.js +0 -159
  31. package/dist/definitions/outlook.js +0 -277
  32. package/dist/definitions/patch/apply-patch.js +0 -184
  33. package/dist/definitions/patch/fuzzy-match.js +0 -166
  34. package/dist/definitions/patch/index.js +0 -1
  35. package/dist/definitions/patch/patch-parser.js +0 -207
  36. package/dist/definitions/patch/sandbox-paths.js +0 -105
  37. package/dist/definitions/process/index.js +0 -4
  38. package/dist/definitions/process/process-registry.js +0 -213
  39. package/dist/definitions/process/process.js +0 -386
  40. package/dist/definitions/process/pty-keys.js +0 -254
  41. package/dist/definitions/process/session-slug.js +0 -142
  42. package/dist/definitions/quip.js +0 -195
  43. package/dist/definitions/search.js +0 -60
  44. package/dist/definitions/session-history.js +0 -69
  45. package/dist/definitions/shell.js +0 -181
  46. package/dist/definitions/slack.js +0 -180
  47. package/dist/definitions/web.js +0 -109
  48. package/dist/executors/apply-patch.js +0 -901
  49. package/dist/executors/arion.js +0 -119
  50. package/dist/executors/code-intelligence.js +0 -882
  51. package/dist/executors/deploy.js +0 -848
  52. package/dist/executors/filesystem.js +0 -1122
  53. package/dist/executors/frg-freshness.js +0 -576
  54. package/dist/executors/frg.js +0 -298
  55. package/dist/executors/index.js +0 -46
  56. package/dist/executors/learning-meta.js +0 -1146
  57. package/dist/executors/lsp-client.js +0 -296
  58. package/dist/executors/memory.js +0 -750
  59. package/dist/executors/meta.js +0 -220
  60. package/dist/executors/process-registry.js +0 -465
  61. package/dist/executors/pty-session-store.js +0 -30
  62. package/dist/executors/pty.js +0 -271
  63. package/dist/executors/restart.js +0 -119
  64. package/dist/executors/search-freshness.js +0 -195
  65. package/dist/executors/search-types.js +0 -52
  66. package/dist/executors/search.js +0 -66
  67. package/dist/executors/self-diagnose.js +0 -398
  68. package/dist/executors/session-history.js +0 -283
  69. package/dist/executors/shell-safety.js +0 -473
  70. package/dist/executors/shell.js +0 -954
  71. package/dist/executors/utils.js +0 -33
  72. package/dist/executors/web.js +0 -542
  73. package/dist/extraction/content-extraction.js +0 -235
  74. package/dist/extraction/index.js +0 -4
  75. package/dist/headless-control-contract.js +0 -967
  76. package/dist/local-control-http-auth.js +0 -2
  77. package/dist/mcp/client.js +0 -181
  78. package/dist/mcp/connection.js +0 -480
  79. package/dist/mcp/index.js +0 -10
  80. package/dist/mcp/jsonrpc.js +0 -144
  81. package/dist/mcp/types.js +0 -7
  82. package/dist/network-control-adapter.js +0 -72
  83. package/dist/network-runtime/address-types.js +0 -165
  84. package/dist/network-runtime/db-owner-fencing.js +0 -69
  85. package/dist/network-runtime/delivery-receipts.js +0 -267
  86. package/dist/network-runtime/direct-endpoint-authority.js +0 -25
  87. package/dist/network-runtime/local-control-contract.js +0 -627
  88. package/dist/network-runtime/node-store-contract.js +0 -34
  89. package/dist/network-runtime/pair-route-contract.js +0 -77
  90. package/dist/network-runtime/peer-capabilities.js +0 -28
  91. package/dist/network-runtime/peer-principal-ref.js +0 -12
  92. package/dist/network-runtime/peer-state-machine.js +0 -121
  93. package/dist/network-runtime/protocol-schemas.js +0 -205
  94. package/dist/network-runtime/runtime-bootstrap-contract.js +0 -60
  95. package/dist/outlook/desktop-session.js +0 -279
  96. package/dist/policy.js +0 -149
  97. package/dist/providers/brave.js +0 -62
  98. package/dist/providers/duckduckgo.js +0 -176
  99. package/dist/providers/exa.js +0 -63
  100. package/dist/providers/firecrawl.js +0 -55
  101. package/dist/providers/index.js +0 -7
  102. package/dist/providers/jina.js +0 -49
  103. package/dist/providers/router.js +0 -96
  104. package/dist/providers/search-provider.js +0 -32
  105. package/dist/providers/tavily.js +0 -54
  106. package/dist/quip/desktop-session.js +0 -317
  107. package/dist/registry/index.js +0 -1
  108. package/dist/registry/registry.js +0 -756
  109. package/dist/runtime-socket-local-control-client.js +0 -330
  110. package/dist/security/dns-normalization.js +0 -19
  111. package/dist/security/dns-pinning.js +0 -123
  112. package/dist/security/external-content.js +0 -91
  113. package/dist/security/ssrf.js +0 -181
  114. package/dist/slack/desktop-session.js +0 -324
  115. package/dist/tool-factory.js +0 -47
  116. package/dist/types.js +0 -7
  117. package/dist/utils/retry.js +0 -132
  118. package/dist/utils/safe-parse-json.js +0 -160
  119. package/dist/utils/url.js +0 -19
  120. package/dist-cjs/.tsbuildinfo +0 -1
  121. package/dist-cjs/ask-user-interaction.js +0 -27
  122. package/dist-cjs/cache/web-cache.js +0 -70
  123. package/dist-cjs/definitions/arion.js +0 -107
  124. package/dist-cjs/definitions/browser/browser.js +0 -421
  125. package/dist-cjs/definitions/browser/index.js +0 -8
  126. package/dist-cjs/definitions/browser/pw-downloads.js +0 -117
  127. package/dist-cjs/definitions/browser/pw-interactions.js +0 -213
  128. package/dist-cjs/definitions/browser/pw-responses.js +0 -84
  129. package/dist-cjs/definitions/browser/pw-session.js +0 -326
  130. package/dist-cjs/definitions/browser/pw-shared.js +0 -72
  131. package/dist-cjs/definitions/browser/pw-snapshot.js +0 -307
  132. package/dist-cjs/definitions/browser/pw-state.js +0 -70
  133. package/dist-cjs/definitions/browser/types.js +0 -5
  134. package/dist-cjs/definitions/code-intelligence.js +0 -473
  135. package/dist-cjs/definitions/core.js +0 -133
  136. package/dist-cjs/definitions/delegation.js +0 -515
  137. package/dist-cjs/definitions/deploy.js +0 -68
  138. package/dist-cjs/definitions/filesystem.js +0 -199
  139. package/dist-cjs/definitions/frg.js +0 -66
  140. package/dist-cjs/definitions/index.js +0 -43
  141. package/dist-cjs/definitions/memory.js +0 -126
  142. package/dist-cjs/definitions/messaging.js +0 -631
  143. package/dist-cjs/definitions/meta.js +0 -352
  144. package/dist-cjs/definitions/network.js +0 -162
  145. package/dist-cjs/definitions/outlook.js +0 -280
  146. package/dist-cjs/definitions/patch/apply-patch.js +0 -191
  147. package/dist-cjs/definitions/patch/fuzzy-match.js +0 -172
  148. package/dist-cjs/definitions/patch/index.js +0 -5
  149. package/dist-cjs/definitions/patch/patch-parser.js +0 -215
  150. package/dist-cjs/definitions/patch/sandbox-paths.js +0 -113
  151. package/dist-cjs/definitions/process/index.js +0 -8
  152. package/dist-cjs/definitions/process/process-registry.js +0 -231
  153. package/dist-cjs/definitions/process/process.js +0 -389
  154. package/dist-cjs/definitions/process/pty-keys.js +0 -259
  155. package/dist-cjs/definitions/process/session-slug.js +0 -145
  156. package/dist-cjs/definitions/quip.js +0 -198
  157. package/dist-cjs/definitions/search.js +0 -63
  158. package/dist-cjs/definitions/session-history.js +0 -72
  159. package/dist-cjs/definitions/shell.js +0 -184
  160. package/dist-cjs/definitions/slack.js +0 -183
  161. package/dist-cjs/definitions/web.js +0 -112
  162. package/dist-cjs/executors/apply-patch.js +0 -938
  163. package/dist-cjs/executors/arion.js +0 -125
  164. package/dist-cjs/executors/code-intelligence.js +0 -925
  165. package/dist-cjs/executors/deploy.js +0 -869
  166. package/dist-cjs/executors/filesystem.js +0 -1167
  167. package/dist-cjs/executors/frg-freshness.js +0 -627
  168. package/dist-cjs/executors/frg.js +0 -334
  169. package/dist-cjs/executors/index.js +0 -143
  170. package/dist-cjs/executors/learning-meta.js +0 -1165
  171. package/dist-cjs/executors/lsp-client.js +0 -310
  172. package/dist-cjs/executors/memory.js +0 -796
  173. package/dist-cjs/executors/meta.js +0 -226
  174. package/dist-cjs/executors/process-registry.js +0 -469
  175. package/dist-cjs/executors/pty-session-store.js +0 -34
  176. package/dist-cjs/executors/pty.js +0 -312
  177. package/dist-cjs/executors/restart.js +0 -155
  178. package/dist-cjs/executors/search-freshness.js +0 -234
  179. package/dist-cjs/executors/search-types.js +0 -56
  180. package/dist-cjs/executors/search.js +0 -102
  181. package/dist-cjs/executors/self-diagnose.js +0 -434
  182. package/dist-cjs/executors/session-history.js +0 -320
  183. package/dist-cjs/executors/shell-safety.js +0 -478
  184. package/dist-cjs/executors/shell.js +0 -1001
  185. package/dist-cjs/executors/utils.js +0 -73
  186. package/dist-cjs/executors/web.js +0 -547
  187. package/dist-cjs/extraction/content-extraction.js +0 -243
  188. package/dist-cjs/extraction/index.js +0 -8
  189. package/dist-cjs/headless-control-contract.js +0 -972
  190. package/dist-cjs/local-control-http-auth.js +0 -5
  191. package/dist-cjs/mcp/client.js +0 -185
  192. package/dist-cjs/mcp/connection.js +0 -484
  193. package/dist-cjs/mcp/index.js +0 -30
  194. package/dist-cjs/mcp/jsonrpc.js +0 -148
  195. package/dist-cjs/mcp/types.js +0 -8
  196. package/dist-cjs/network-control-adapter.js +0 -77
  197. package/dist-cjs/network-runtime/address-types.js +0 -168
  198. package/dist-cjs/network-runtime/db-owner-fencing.js +0 -76
  199. package/dist-cjs/network-runtime/delivery-receipts.js +0 -276
  200. package/dist-cjs/network-runtime/direct-endpoint-authority.js +0 -29
  201. package/dist-cjs/network-runtime/local-control-contract.js +0 -633
  202. package/dist-cjs/network-runtime/node-store-contract.js +0 -38
  203. package/dist-cjs/network-runtime/pair-route-contract.js +0 -80
  204. package/dist-cjs/network-runtime/peer-capabilities.js +0 -37
  205. package/dist-cjs/network-runtime/peer-principal-ref.js +0 -15
  206. package/dist-cjs/network-runtime/peer-state-machine.js +0 -129
  207. package/dist-cjs/network-runtime/protocol-schemas.js +0 -212
  208. package/dist-cjs/network-runtime/runtime-bootstrap-contract.js +0 -63
  209. package/dist-cjs/outlook/desktop-session.js +0 -318
  210. package/dist-cjs/policy.js +0 -155
  211. package/dist-cjs/providers/brave.js +0 -66
  212. package/dist-cjs/providers/duckduckgo.js +0 -180
  213. package/dist-cjs/providers/exa.js +0 -67
  214. package/dist-cjs/providers/firecrawl.js +0 -59
  215. package/dist-cjs/providers/index.js +0 -17
  216. package/dist-cjs/providers/jina.js +0 -53
  217. package/dist-cjs/providers/router.js +0 -100
  218. package/dist-cjs/providers/search-provider.js +0 -36
  219. package/dist-cjs/providers/tavily.js +0 -58
  220. package/dist-cjs/quip/desktop-session.js +0 -353
  221. package/dist-cjs/registry/index.js +0 -6
  222. package/dist-cjs/registry/registry.js +0 -761
  223. package/dist-cjs/runtime-socket-local-control-client.js +0 -367
  224. package/dist-cjs/security/dns-normalization.js +0 -22
  225. package/dist-cjs/security/dns-pinning.js +0 -160
  226. package/dist-cjs/security/external-content.js +0 -95
  227. package/dist-cjs/security/ssrf.js +0 -221
  228. package/dist-cjs/slack/desktop-session.js +0 -366
  229. package/dist-cjs/tool-factory.js +0 -50
  230. package/dist-cjs/types.js +0 -8
  231. package/dist-cjs/utils/retry.js +0 -169
  232. package/dist-cjs/utils/safe-parse-json.js +0 -164
  233. package/dist-cjs/utils/url.js +0 -23
@@ -1,226 +0,0 @@
1
- "use strict";
2
- /**
3
- * @aria/tools - Meta tool executors
4
- *
5
- * Implementation of meta operations for ARIA tool system:
6
- * - ask_user: Interactive user prompting with rate limiting
7
- * - quest_update: Create/update quests persisted via QuestStore
8
- * - quest_list: List quests with status filtering
9
- */
10
- Object.defineProperty(exports, "__esModule", { value: true });
11
- exports.resetAskUserCounter = resetAskUserCounter;
12
- exports.executeAskUser = executeAskUser;
13
- exports.executeQuestUpdate = executeQuestUpdate;
14
- exports.executeQuestList = executeQuestList;
15
- const nanoid_1 = require("nanoid");
16
- const utils_js_1 = require("./utils.js");
17
- /**
18
- * Module-level ask_user call counter for rate limiting.
19
- * Reset via resetAskUserCounter() between turns (called by the runner).
20
- */
21
- let askUserCallCount = 0;
22
- const ASK_USER_MAX_CALLS = 3;
23
- function isAskUserAnswersExhaustedError(error) {
24
- if (!error || typeof error !== "object")
25
- return false;
26
- const candidate = error;
27
- return (candidate.code === "ASK_USER_ANSWERS_EXHAUSTED" ||
28
- candidate.name === "AskUserAnswersExhaustedError");
29
- }
30
- /**
31
- * Reset the ask_user rate limit counter.
32
- * Should be called by the runner at the start of each turn.
33
- */
34
- function resetAskUserCounter() {
35
- askUserCallCount = 0;
36
- }
37
- /**
38
- * Prompt the user with one or more questions.
39
- * Rate-limited to ASK_USER_MAX_CALLS per turn to prevent abuse loops.
40
- */
41
- async function executeAskUser(input, ctx) {
42
- if (ctx.abortSignal?.aborted)
43
- return (0, utils_js_1.fail)("Operation cancelled");
44
- // Check if user interaction handler is wired
45
- if (!ctx.userInteraction) {
46
- return (0, utils_js_1.fail)("ask_user not available in this context");
47
- }
48
- // Rate limit — prefer per-session counter from ToolContext, fall back to module-level global
49
- const counter = ctx.askUserCallCounter ?? { count: askUserCallCount };
50
- if (counter.count >= ASK_USER_MAX_CALLS) {
51
- return (0, utils_js_1.fail)(`Rate limit: max ${ASK_USER_MAX_CALLS} ask_user calls per turn`);
52
- }
53
- // Validate input
54
- if (!input.questions || !Array.isArray(input.questions) || input.questions.length === 0) {
55
- return (0, utils_js_1.fail)("questions array is required and must not be empty");
56
- }
57
- try {
58
- const answers = await ctx.userInteraction.ask(input.questions);
59
- // Only exhaustion retries are free. A completed prompt attempt, even if it
60
- // fails for another reason, still burns the turn-local ask_user budget.
61
- counter.count++;
62
- if (!ctx.askUserCallCounter)
63
- askUserCallCount = counter.count;
64
- return (0, utils_js_1.success)(`Received ${answers.length} answer${answers.length === 1 ? "" : "s"} from user`, { answers });
65
- }
66
- catch (err) {
67
- if (isAskUserAnswersExhaustedError(err)) {
68
- throw err;
69
- }
70
- counter.count++;
71
- if (!ctx.askUserCallCounter)
72
- askUserCallCount = counter.count;
73
- return (0, utils_js_1.fail)((0, utils_js_1.getErrorMessage)(err));
74
- }
75
- }
76
- // ============================================================================
77
- // Quest Update
78
- // ============================================================================
79
- /** Valid quest statuses — used for runtime validation of LLM-provided values */
80
- const VALID_STATUSES = new Set(["open", "active", "blocked", "done"]);
81
- /**
82
- * Create or update quests, persisted via QuestStore (structured SQL storage).
83
- *
84
- * New quests get a nanoid-generated ID. Existing quests (with `id`) are updated
85
- * in-place via QuestStore.updateQuest().
86
- *
87
- * Each quest in the batch is processed independently — a failure on one quest
88
- * does not prevent others from being processed.
89
- */
90
- async function executeQuestUpdate(input, ctx) {
91
- if (ctx.abortSignal?.aborted)
92
- return (0, utils_js_1.fail)("Operation cancelled");
93
- if (!ctx.questStore) {
94
- return (0, utils_js_1.fail)("QuestStore is not available — quest persistence requires a database");
95
- }
96
- if (!input.quests || !Array.isArray(input.quests) || input.quests.length === 0) {
97
- return (0, utils_js_1.fail)("quests array is required and must not be empty");
98
- }
99
- const results = [];
100
- let failures = 0;
101
- for (const quest of input.quests) {
102
- // Runtime status validation — LLM may hallucinate invalid values
103
- if (quest.status !== undefined && !VALID_STATUSES.has(quest.status)) {
104
- results.push({
105
- id: quest.id ?? "unknown",
106
- title: quest.title ?? "(untitled)",
107
- status: quest.status,
108
- action: "failed",
109
- error: `Invalid status "${quest.status}" — must be one of: open, active, blocked, done`,
110
- });
111
- failures++;
112
- continue;
113
- }
114
- const isUpdate = !!quest.id;
115
- const questId = quest.id ?? `quest_${(0, nanoid_1.nanoid)(12)}`;
116
- try {
117
- if (isUpdate) {
118
- // Update existing quest — only pass defined fields to avoid overwriting
119
- const updates = {};
120
- if (quest.title !== undefined)
121
- updates.title = quest.title;
122
- if (quest.status !== undefined)
123
- updates.status = quest.status;
124
- if (quest.notes !== undefined)
125
- updates.progress = quest.notes;
126
- if (quest.priority !== undefined)
127
- updates.priority = quest.priority;
128
- if (quest.blocked_by !== undefined)
129
- updates.blockedBy = quest.blocked_by;
130
- const updated = ctx.questStore.updateQuest(questId, updates);
131
- results.push({
132
- id: questId,
133
- title: updated.title,
134
- status: updated.status,
135
- action: "updated",
136
- });
137
- }
138
- else {
139
- // Create new quest — title required, status defaults to "open"
140
- if (!quest.title) {
141
- results.push({
142
- id: questId,
143
- title: "(untitled)",
144
- status: quest.status ?? "open",
145
- action: "failed",
146
- error: "title is required when creating a new quest",
147
- });
148
- failures++;
149
- continue;
150
- }
151
- const created = ctx.questStore.createQuest({
152
- id: questId,
153
- title: quest.title,
154
- status: quest.status ?? "open",
155
- priority: quest.priority ?? 2,
156
- progress: quest.notes,
157
- blockedBy: quest.blocked_by,
158
- });
159
- results.push({
160
- id: questId,
161
- title: created.title,
162
- status: created.status,
163
- action: "created",
164
- });
165
- }
166
- }
167
- catch (err) {
168
- results.push({
169
- id: questId,
170
- title: quest.title ?? "(untitled)",
171
- status: quest.status ?? "unknown",
172
- action: "failed",
173
- error: (0, utils_js_1.getErrorMessage)(err),
174
- });
175
- failures++;
176
- }
177
- }
178
- const created = results.filter((r) => r.action === "created").length;
179
- const updated = results.filter((r) => r.action === "updated").length;
180
- const parts = [];
181
- if (created > 0)
182
- parts.push(`${created} created`);
183
- if (updated > 0)
184
- parts.push(`${updated} updated`);
185
- if (failures > 0)
186
- parts.push(`${failures} failed`);
187
- const data = { quests: results };
188
- if (failures > 0 && failures === input.quests.length) {
189
- return (0, utils_js_1.fail)(`All quests failed: ${parts.join(", ")}`, data);
190
- }
191
- if (failures > 0) {
192
- return (0, utils_js_1.success)(`Quests: ${parts.join(", ")} (partial failure)`, data);
193
- }
194
- return (0, utils_js_1.success)(`Quests: ${parts.join(", ")}`, data);
195
- }
196
- /**
197
- * List quests from QuestStore with optional status filtering.
198
- */
199
- async function executeQuestList(input, ctx) {
200
- if (ctx.abortSignal?.aborted)
201
- return (0, utils_js_1.fail)("Operation cancelled");
202
- if (!ctx.questStore) {
203
- return (0, utils_js_1.fail)("QuestStore is not available — quest listing requires a database");
204
- }
205
- try {
206
- const statusFilter = input.status ?? "all";
207
- const filter = statusFilter !== "all" ? { status: statusFilter } : undefined;
208
- const rawQuests = ctx.questStore.listQuests(filter);
209
- const quests = rawQuests.map((q) => ({
210
- id: q.id,
211
- title: q.title,
212
- status: q.status,
213
- priority: q.priority,
214
- blocked_by: q.blockedBy ?? "",
215
- notes: q.progress ?? "",
216
- updatedAt: q.updatedAt,
217
- }));
218
- return (0, utils_js_1.success)(`Found ${quests.length} quest${quests.length === 1 ? "" : "s"}`, {
219
- quests,
220
- count: quests.length,
221
- });
222
- }
223
- catch (err) {
224
- return (0, utils_js_1.fail)((0, utils_js_1.getErrorMessage)(err));
225
- }
226
- }
@@ -1,469 +0,0 @@
1
- "use strict";
2
- /**
3
- * @aria/tools - Spawned Process Registry
4
- *
5
- * Tracks PIDs of child processes spawned by shell executors.
6
- * Allows bulk cleanup (SIGTERM then SIGKILL) when a RunSession closes,
7
- * preventing orphaned processes from lingering.
8
- */
9
- Object.defineProperty(exports, "__esModule", { value: true });
10
- exports.SpawnedProcessRegistry = void 0;
11
- /**
12
- * How long to wait after SIGTERM before escalating to SIGKILL (ms).
13
- */
14
- const SIGKILL_TIMEOUT_MS = 2000;
15
- const POST_KILL_POLL_INTERVAL_MS = 25;
16
- const PROCESS_REAP_POLL_INTERVAL_MS = 250;
17
- /**
18
- * Max number of exited-process snapshots to retain for observability.
19
- */
20
- const EXIT_HISTORY_LIMIT = 100;
21
- /**
22
- * Registry that tracks spawned child process PIDs.
23
- *
24
- * Usage:
25
- * - Shell executors call `add(pid)` after spawning a process
26
- * - Shell executors call `remove(pid)` when a process exits normally
27
- * - RunSession calls `killAll()` during close() to clean up orphans
28
- */
29
- class SpawnedProcessRegistry {
30
- active = new Map();
31
- exited = new Map();
32
- waiters = new Map();
33
- reapers = new Map();
34
- /**
35
- * Register a spawned process PID.
36
- */
37
- add(pid, metadata = {}) {
38
- const existing = this.active.get(pid);
39
- if (existing) {
40
- this.active.set(pid, {
41
- ...existing,
42
- command: metadata.command ?? existing.command,
43
- args: metadata.args ? [...metadata.args] : existing.args,
44
- cwd: metadata.cwd ?? existing.cwd,
45
- interactive: metadata.interactive ?? existing.interactive,
46
- });
47
- return;
48
- }
49
- const startedAtMs = Date.now();
50
- this.active.set(pid, {
51
- pid,
52
- command: metadata.command ?? null,
53
- args: metadata.args ? [...metadata.args] : [],
54
- cwd: metadata.cwd ?? null,
55
- interactive: metadata.interactive ?? false,
56
- reapOnExit: isTrackedProcessAlive(pid),
57
- deferredExitCode: null,
58
- deferredSignal: null,
59
- startedAtMs,
60
- startedAt: new Date(startedAtMs).toISOString(),
61
- });
62
- this.exited.delete(pid);
63
- this.ensureReaper(pid);
64
- }
65
- /**
66
- * Deregister a process PID (e.g., after normal exit).
67
- */
68
- remove(pid, exit = {}) {
69
- const record = this.active.get(pid);
70
- if (!record) {
71
- const exitedRecord = this.exited.get(pid);
72
- if (exitedRecord) {
73
- this.exited.set(pid, {
74
- ...exitedRecord,
75
- exitCode: exitedRecord.exitCode ?? normalizeExitCode(exit.exitCode),
76
- signal: exitedRecord.exitCode !== null
77
- ? exitedRecord.signal
78
- : (exitedRecord.signal ?? normalizeSignal(exit.signal)),
79
- });
80
- }
81
- return;
82
- }
83
- this.clearReaper(pid);
84
- this.active.delete(pid);
85
- const endedAtMs = Date.now();
86
- const recordedExitCode = normalizeExitCode(exit.exitCode) ?? record.deferredExitCode;
87
- const exitedRecord = {
88
- ...record,
89
- exitCode: recordedExitCode,
90
- signal: record.deferredSignal ?? normalizeSignal(exit.signal),
91
- endedAtMs,
92
- endedAt: new Date(endedAtMs).toISOString(),
93
- };
94
- this.exited.set(pid, exitedRecord);
95
- this.pruneExitedHistory();
96
- const waiters = this.waiters.get(pid);
97
- if (waiters && waiters.length > 0) {
98
- this.waiters.delete(pid);
99
- for (const waiter of waiters) {
100
- waiter(exitedRecord);
101
- }
102
- }
103
- }
104
- /**
105
- * Check whether a PID is tracked.
106
- */
107
- has(pid) {
108
- return this.active.has(pid);
109
- }
110
- recordExitMetadata(pid, exit = {}) {
111
- const record = this.active.get(pid);
112
- if (!record) {
113
- return;
114
- }
115
- record.deferredExitCode = normalizeExitCode(exit.exitCode);
116
- record.deferredSignal = normalizeSignal(exit.signal);
117
- }
118
- /**
119
- * Number of tracked PIDs.
120
- */
121
- get size() {
122
- return this.active.size;
123
- }
124
- /**
125
- * Return a snapshot of all tracked PIDs.
126
- */
127
- getAll() {
128
- return [...this.active.keys()];
129
- }
130
- /**
131
- * Return process snapshots for observability.
132
- */
133
- listProcesses(options = {}) {
134
- const now = Date.now();
135
- const snapshots = [];
136
- for (const pid of [...this.active.keys()]) {
137
- this.reapIfExited(pid);
138
- }
139
- for (const record of this.active.values()) {
140
- snapshots.push(this.toRunningSnapshot(record, now));
141
- }
142
- if (options.includeExited) {
143
- for (const record of this.exited.values()) {
144
- snapshots.push(this.toExitedSnapshot(record));
145
- }
146
- }
147
- return snapshots.sort((a, b) => a.pid - b.pid);
148
- }
149
- /**
150
- * Return a single process snapshot by PID.
151
- */
152
- getProcess(pid) {
153
- this.reapIfExited(pid);
154
- const activeRecord = this.active.get(pid);
155
- if (activeRecord) {
156
- return this.toRunningSnapshot(activeRecord);
157
- }
158
- const exitedRecord = this.exited.get(pid);
159
- if (exitedRecord) {
160
- return this.toExitedSnapshot(exitedRecord);
161
- }
162
- return undefined;
163
- }
164
- /**
165
- * Wait for a tracked process to exit, up to timeoutMs.
166
- */
167
- async waitForExit(pid, timeoutMs = 30_000) {
168
- this.reapIfExited(pid);
169
- const exitedRecord = this.exited.get(pid);
170
- if (exitedRecord) {
171
- return {
172
- pid,
173
- status: "exited",
174
- timedOut: false,
175
- process: this.toExitedSnapshot(exitedRecord),
176
- };
177
- }
178
- const activeRecord = this.active.get(pid);
179
- if (!activeRecord) {
180
- return {
181
- pid,
182
- status: "not_found",
183
- timedOut: false,
184
- };
185
- }
186
- if (timeoutMs <= 0) {
187
- return {
188
- pid,
189
- status: "running",
190
- timedOut: true,
191
- process: this.toRunningSnapshot(activeRecord),
192
- };
193
- }
194
- return new Promise((resolve) => {
195
- let settled = false;
196
- const settle = (result) => {
197
- if (settled)
198
- return;
199
- settled = true;
200
- resolve(result);
201
- };
202
- const removeWaiter = (target) => {
203
- const existing = this.waiters.get(pid);
204
- if (!existing)
205
- return;
206
- const next = existing.filter((entry) => entry !== target);
207
- if (next.length === 0) {
208
- this.waiters.delete(pid);
209
- }
210
- else {
211
- this.waiters.set(pid, next);
212
- }
213
- };
214
- const waiter = (record) => {
215
- clearTimeout(timeoutId);
216
- removeWaiter(waiter);
217
- settle({
218
- pid,
219
- status: "exited",
220
- timedOut: false,
221
- process: this.toExitedSnapshot(record),
222
- });
223
- };
224
- const timeoutId = setTimeout(() => {
225
- const current = this.active.get(pid);
226
- if (current) {
227
- removeWaiter(waiter);
228
- settle({
229
- pid,
230
- status: "running",
231
- timedOut: true,
232
- process: this.toRunningSnapshot(current),
233
- });
234
- return;
235
- }
236
- const exited = this.exited.get(pid);
237
- if (exited) {
238
- removeWaiter(waiter);
239
- settle({
240
- pid,
241
- status: "exited",
242
- timedOut: false,
243
- process: this.toExitedSnapshot(exited),
244
- });
245
- return;
246
- }
247
- removeWaiter(waiter);
248
- settle({
249
- pid,
250
- status: "not_found",
251
- timedOut: false,
252
- });
253
- }, timeoutMs);
254
- const existingWaiters = this.waiters.get(pid) ?? [];
255
- existingWaiters.push(waiter);
256
- this.waiters.set(pid, existingWaiters);
257
- // Close the race window where remove() can happen between the initial
258
- // active check above and waiter registration.
259
- const exitedAfterRegistration = this.exited.get(pid);
260
- if (exitedAfterRegistration) {
261
- clearTimeout(timeoutId);
262
- removeWaiter(waiter);
263
- settle({
264
- pid,
265
- status: "exited",
266
- timedOut: false,
267
- process: this.toExitedSnapshot(exitedAfterRegistration),
268
- });
269
- return;
270
- }
271
- if (!this.active.has(pid)) {
272
- clearTimeout(timeoutId);
273
- removeWaiter(waiter);
274
- settle({
275
- pid,
276
- status: "not_found",
277
- timedOut: false,
278
- });
279
- }
280
- });
281
- }
282
- /**
283
- * Send SIGTERM to all tracked PIDs, then SIGKILL after a timeout
284
- * for any that are still alive. Clears the registry afterwards.
285
- *
286
- * Returns the list of PIDs that were signalled.
287
- */
288
- async killAll() {
289
- const tracked = [...this.active.keys()];
290
- if (tracked.length === 0)
291
- return [];
292
- // Phase 1: SIGTERM all tracked processes
293
- const alive = [];
294
- const signaledTerm = new Set();
295
- for (const pid of tracked) {
296
- if (isTrackedProcessAlive(pid)) {
297
- if (sendSignalToTrackedProcess(pid, "SIGTERM")) {
298
- alive.push(pid);
299
- signaledTerm.add(pid);
300
- }
301
- }
302
- }
303
- if (alive.length === 0) {
304
- for (const pid of tracked) {
305
- this.remove(pid);
306
- }
307
- return tracked;
308
- }
309
- // Phase 2: Wait, then SIGKILL any survivors
310
- await new Promise((resolve) => setTimeout(resolve, SIGKILL_TIMEOUT_MS));
311
- const signaledKill = new Set();
312
- for (const pid of alive) {
313
- if (isTrackedProcessAlive(pid)) {
314
- if (sendSignalToTrackedProcess(pid, "SIGKILL")) {
315
- signaledKill.add(pid);
316
- }
317
- }
318
- }
319
- await waitForProcessesToExit([...signaledKill], SIGKILL_TIMEOUT_MS);
320
- for (const pid of tracked) {
321
- this.remove(pid, {
322
- signal: signaledKill.has(pid) ? "SIGKILL" : signaledTerm.has(pid) ? "SIGTERM" : null,
323
- });
324
- }
325
- return tracked;
326
- }
327
- toRunningSnapshot(record, now = Date.now()) {
328
- return {
329
- pid: record.pid,
330
- command: record.command,
331
- args: [...record.args],
332
- cwd: record.cwd,
333
- interactive: record.interactive,
334
- startedAt: record.startedAt,
335
- runtimeMs: Math.max(0, now - record.startedAtMs),
336
- status: "running",
337
- exitCode: null,
338
- signal: null,
339
- endedAt: null,
340
- };
341
- }
342
- toExitedSnapshot(record) {
343
- return {
344
- pid: record.pid,
345
- command: record.command,
346
- args: [...record.args],
347
- cwd: record.cwd,
348
- interactive: record.interactive,
349
- startedAt: record.startedAt,
350
- runtimeMs: Math.max(0, record.endedAtMs - record.startedAtMs),
351
- status: "exited",
352
- exitCode: record.exitCode,
353
- signal: record.signal,
354
- endedAt: record.endedAt,
355
- };
356
- }
357
- pruneExitedHistory() {
358
- while (this.exited.size > EXIT_HISTORY_LIMIT) {
359
- const oldestPid = this.exited.keys().next().value;
360
- if (oldestPid === undefined) {
361
- break;
362
- }
363
- this.exited.delete(oldestPid);
364
- }
365
- }
366
- ensureReaper(pid) {
367
- if (!this.active.get(pid)?.reapOnExit) {
368
- return;
369
- }
370
- if (this.reapers.has(pid)) {
371
- return;
372
- }
373
- const timer = setInterval(() => {
374
- this.reapIfExited(pid);
375
- }, PROCESS_REAP_POLL_INTERVAL_MS);
376
- timer.unref?.();
377
- this.reapers.set(pid, timer);
378
- }
379
- clearReaper(pid) {
380
- const timer = this.reapers.get(pid);
381
- if (!timer) {
382
- return;
383
- }
384
- clearInterval(timer);
385
- this.reapers.delete(pid);
386
- }
387
- reapIfExited(pid) {
388
- const record = this.active.get(pid);
389
- if (!record) {
390
- this.clearReaper(pid);
391
- return false;
392
- }
393
- if (!record.reapOnExit) {
394
- return false;
395
- }
396
- if (isTrackedProcessAlive(pid)) {
397
- return false;
398
- }
399
- this.remove(pid);
400
- return true;
401
- }
402
- }
403
- exports.SpawnedProcessRegistry = SpawnedProcessRegistry;
404
- /**
405
- * Check whether a process is still alive by sending signal 0.
406
- */
407
- function isProcessAlive(pid) {
408
- try {
409
- process.kill(pid, 0);
410
- return true;
411
- }
412
- catch {
413
- return false;
414
- }
415
- }
416
- function isTrackedProcessAlive(pid) {
417
- if (pid > 0) {
418
- try {
419
- process.kill(-pid, 0);
420
- return true;
421
- }
422
- catch {
423
- // Fall back to the tracked pid itself when the process group is gone or unsupported.
424
- }
425
- }
426
- return isProcessAlive(pid);
427
- }
428
- function sendSignalToTrackedProcess(pid, signal) {
429
- try {
430
- if (pid > 0) {
431
- process.kill(-pid, signal);
432
- return true;
433
- }
434
- }
435
- catch {
436
- // Fall back to the tracked pid itself when there is no process group to target.
437
- }
438
- try {
439
- process.kill(pid, signal);
440
- return true;
441
- }
442
- catch {
443
- return false;
444
- }
445
- }
446
- async function waitForProcessesToExit(pids, timeoutMs) {
447
- if (pids.length === 0) {
448
- return;
449
- }
450
- const deadline = Date.now() + timeoutMs;
451
- while (Date.now() < deadline) {
452
- if (pids.every((pid) => !isTrackedProcessAlive(pid))) {
453
- return;
454
- }
455
- await new Promise((resolve) => setTimeout(resolve, POST_KILL_POLL_INTERVAL_MS));
456
- }
457
- }
458
- function normalizeExitCode(value) {
459
- if (typeof value !== "number" || Number.isNaN(value)) {
460
- return null;
461
- }
462
- return value;
463
- }
464
- function normalizeSignal(value) {
465
- if (typeof value !== "string" || value.length === 0) {
466
- return null;
467
- }
468
- return value;
469
- }