@octelium/console-agent 0.1.0-b-ca02.1.1
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/LICENSE +267 -0
- package/README.md +207 -0
- package/apis/octelium.binpb +0 -0
- package/dist/agent/pi/auth.d.ts +32 -0
- package/dist/agent/pi/auth.js +279 -0
- package/dist/agent/pi/auth.js.map +1 -0
- package/dist/agent/pi/backend.d.ts +77 -0
- package/dist/agent/pi/backend.js +518 -0
- package/dist/agent/pi/backend.js.map +1 -0
- package/dist/agent/pi/mapper.d.ts +33 -0
- package/dist/agent/pi/mapper.js +386 -0
- package/dist/agent/pi/mapper.js.map +1 -0
- package/dist/agent/pi/model.d.ts +47 -0
- package/dist/agent/pi/model.js +199 -0
- package/dist/agent/pi/model.js.map +1 -0
- package/dist/agent/prompt.d.ts +22 -0
- package/dist/agent/prompt.js +75 -0
- package/dist/agent/prompt.js.map +1 -0
- package/dist/agent/resources.d.ts +13 -0
- package/dist/agent/resources.js +78 -0
- package/dist/agent/resources.js.map +1 -0
- package/dist/agent/tools/builtin.d.ts +13 -0
- package/dist/agent/tools/builtin.js +80 -0
- package/dist/agent/tools/builtin.js.map +1 -0
- package/dist/agent/tools/octelium.d.ts +28 -0
- package/dist/agent/tools/octelium.js +236 -0
- package/dist/agent/tools/octelium.js.map +1 -0
- package/dist/agent/tools/present.d.ts +61 -0
- package/dist/agent/tools/present.js +273 -0
- package/dist/agent/tools/present.js.map +1 -0
- package/dist/agent/tools/util.d.ts +19 -0
- package/dist/agent/tools/util.js +103 -0
- package/dist/agent/tools/util.js.map +1 -0
- package/dist/agent/tools/web.d.ts +24 -0
- package/dist/agent/tools/web.js +237 -0
- package/dist/agent/tools/web.js.map +1 -0
- package/dist/agent/types.d.ts +60 -0
- package/dist/agent/types.js +2 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/app.d.ts +36 -0
- package/dist/app.js +126 -0
- package/dist/app.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +249 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +159 -0
- package/dist/config.js +251 -0
- package/dist/config.js.map +1 -0
- package/dist/errors.d.ts +13 -0
- package/dist/errors.js +18 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.d.ts +6 -0
- package/dist/index.js +6 -0
- package/dist/index.js.map +1 -0
- package/dist/log.d.ts +11 -0
- package/dist/log.js +41 -0
- package/dist/log.js.map +1 -0
- package/dist/octelium/catalog.d.ts +61 -0
- package/dist/octelium/catalog.js +595 -0
- package/dist/octelium/catalog.js.map +1 -0
- package/dist/octelium/client.d.ts +53 -0
- package/dist/octelium/client.js +201 -0
- package/dist/octelium/client.js.map +1 -0
- package/dist/octelium/search.d.ts +24 -0
- package/dist/octelium/search.js +211 -0
- package/dist/octelium/search.js.map +1 -0
- package/dist/protocol/index.d.ts +432 -0
- package/dist/protocol/index.js +44 -0
- package/dist/protocol/index.js.map +1 -0
- package/dist/runs/builder.d.ts +17 -0
- package/dist/runs/builder.js +81 -0
- package/dist/runs/builder.js.map +1 -0
- package/dist/runs/manager.d.ts +52 -0
- package/dist/runs/manager.js +536 -0
- package/dist/runs/manager.js.map +1 -0
- package/dist/server/http.d.ts +9 -0
- package/dist/server/http.js +100 -0
- package/dist/server/http.js.map +1 -0
- package/dist/server/server.d.ts +48 -0
- package/dist/server/server.js +483 -0
- package/dist/server/server.js.map +1 -0
- package/dist/skills.d.ts +4 -0
- package/dist/skills.js +69 -0
- package/dist/skills.js.map +1 -0
- package/dist/store/conversations.d.ts +26 -0
- package/dist/store/conversations.js +183 -0
- package/dist/store/conversations.js.map +1 -0
- package/dist/store/files.d.ts +27 -0
- package/dist/store/files.js +161 -0
- package/dist/store/files.js.map +1 -0
- package/dist/store/settings.d.ts +14 -0
- package/dist/store/settings.js +48 -0
- package/dist/store/settings.js.map +1 -0
- package/dist/version.d.ts +1 -0
- package/dist/version.js +11 -0
- package/dist/version.js.map +1 -0
- package/package.json +57 -0
|
@@ -0,0 +1,536 @@
|
|
|
1
|
+
import { ResourceCache } from "../agent/resources.js";
|
|
2
|
+
import { badRequest, conflict, notFound, tooManyRequests } from "../errors.js";
|
|
3
|
+
import { newID } from "../store/conversations.js";
|
|
4
|
+
import { MessageBuilder } from "./builder.js";
|
|
5
|
+
export const DEFAULT_CONVERSATION_TITLE = "New conversation";
|
|
6
|
+
const maxInputChars = 200_000;
|
|
7
|
+
export const truncateTitle = (text, max = 60) => {
|
|
8
|
+
const flat = text.replace(/\s+/g, " ").trim();
|
|
9
|
+
return flat.length <= max ? flat : `${flat.slice(0, max - 1).trimEnd()}…`;
|
|
10
|
+
};
|
|
11
|
+
export class RunManager {
|
|
12
|
+
opts;
|
|
13
|
+
runs = new Map();
|
|
14
|
+
activeByConversation = new Map();
|
|
15
|
+
resourceCaches = new Map();
|
|
16
|
+
retentionMs;
|
|
17
|
+
constructor(opts) {
|
|
18
|
+
this.opts = opts;
|
|
19
|
+
this.retentionMs = opts.retentionMs ?? 10 * 60 * 1000;
|
|
20
|
+
}
|
|
21
|
+
get activeRunCount() {
|
|
22
|
+
return this.activeByConversation.size;
|
|
23
|
+
}
|
|
24
|
+
startRun(conversationId, input) {
|
|
25
|
+
const { store, files } = this.opts;
|
|
26
|
+
const conversation = store.get(conversationId);
|
|
27
|
+
if (!conversation) {
|
|
28
|
+
throw notFound(`Conversation not found: ${conversationId}`);
|
|
29
|
+
}
|
|
30
|
+
if (this.activeByConversation.has(conversationId)) {
|
|
31
|
+
throw conflict("The conversation already has an active run");
|
|
32
|
+
}
|
|
33
|
+
if (this.activeByConversation.size >= this.opts.maxConcurrentRuns) {
|
|
34
|
+
throw tooManyRequests("Too many concurrent runs");
|
|
35
|
+
}
|
|
36
|
+
const text = typeof input?.text === "string" ? input.text.trim() : "";
|
|
37
|
+
const attachmentIDs = Array.isArray(input?.attachments)
|
|
38
|
+
? input.attachments
|
|
39
|
+
: [];
|
|
40
|
+
if (text.length > maxInputChars) {
|
|
41
|
+
throw badRequest(`The input exceeds ${maxInputChars} characters`);
|
|
42
|
+
}
|
|
43
|
+
const attachments = [];
|
|
44
|
+
for (const id of attachmentIDs) {
|
|
45
|
+
const file = typeof id === "string" ? files.getUpload(id) : undefined;
|
|
46
|
+
if (!file) {
|
|
47
|
+
throw badRequest(`Unknown attachment: ${String(id)}`);
|
|
48
|
+
}
|
|
49
|
+
attachments.push(file);
|
|
50
|
+
}
|
|
51
|
+
if (text === "" && attachments.length === 0) {
|
|
52
|
+
throw badRequest("The input text must not be empty");
|
|
53
|
+
}
|
|
54
|
+
const now = new Date().toISOString();
|
|
55
|
+
const runId = newID();
|
|
56
|
+
const userMessage = {
|
|
57
|
+
id: newID(),
|
|
58
|
+
conversationId,
|
|
59
|
+
role: "user",
|
|
60
|
+
status: "completed",
|
|
61
|
+
createdAt: now,
|
|
62
|
+
completedAt: now,
|
|
63
|
+
runId,
|
|
64
|
+
blocks: text === ""
|
|
65
|
+
? []
|
|
66
|
+
: [{ id: "b1", type: "markdown", text, createdAt: now }],
|
|
67
|
+
attachments: attachments.length > 0 ? attachments : undefined,
|
|
68
|
+
};
|
|
69
|
+
store.appendMessage(conversationId, userMessage);
|
|
70
|
+
const message = {
|
|
71
|
+
id: newID(),
|
|
72
|
+
conversationId,
|
|
73
|
+
role: "assistant",
|
|
74
|
+
status: "streaming",
|
|
75
|
+
createdAt: now,
|
|
76
|
+
runId,
|
|
77
|
+
blocks: [],
|
|
78
|
+
model: this.opts.backend.model,
|
|
79
|
+
};
|
|
80
|
+
const run = {
|
|
81
|
+
id: runId,
|
|
82
|
+
conversationId,
|
|
83
|
+
status: "running",
|
|
84
|
+
createdAt: now,
|
|
85
|
+
userMessageId: userMessage.id,
|
|
86
|
+
assistantMessageId: message.id,
|
|
87
|
+
lastSeq: 0,
|
|
88
|
+
};
|
|
89
|
+
let resolveDone = () => { };
|
|
90
|
+
const donePromise = new Promise((resolve) => {
|
|
91
|
+
resolveDone = resolve;
|
|
92
|
+
});
|
|
93
|
+
const ar = {
|
|
94
|
+
run,
|
|
95
|
+
message,
|
|
96
|
+
builder: undefined,
|
|
97
|
+
events: [],
|
|
98
|
+
listeners: new Set(),
|
|
99
|
+
controller: new AbortController(),
|
|
100
|
+
approvals: new Map(),
|
|
101
|
+
pendingDeltas: new Map(),
|
|
102
|
+
done: false,
|
|
103
|
+
donePromise,
|
|
104
|
+
resolveDone,
|
|
105
|
+
};
|
|
106
|
+
ar.builder = new MessageBuilder(message, (payload) => this.emit(ar, payload), () => this.schedulePersist(ar));
|
|
107
|
+
this.runs.set(runId, ar);
|
|
108
|
+
this.activeByConversation.set(conversationId, ar);
|
|
109
|
+
let current = store.update(conversationId, { activeRunId: runId });
|
|
110
|
+
this.push(ar, { type: "run.started", run: structuredClone(run) });
|
|
111
|
+
this.push(ar, {
|
|
112
|
+
type: "message.created",
|
|
113
|
+
message: structuredClone(userMessage),
|
|
114
|
+
});
|
|
115
|
+
this.push(ar, {
|
|
116
|
+
type: "message.created",
|
|
117
|
+
message: structuredClone(message),
|
|
118
|
+
});
|
|
119
|
+
const isFirst = conversation.title === DEFAULT_CONVERSATION_TITLE;
|
|
120
|
+
if (isFirst && text !== "") {
|
|
121
|
+
current = store.update(conversationId, { title: truncateTitle(text) });
|
|
122
|
+
this.push(ar, {
|
|
123
|
+
type: "conversation.updated",
|
|
124
|
+
conversation: structuredClone(current),
|
|
125
|
+
});
|
|
126
|
+
if (this.opts.generateTitles) {
|
|
127
|
+
void this.generateTitle(conversationId, text, current.title);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
void this.execute(ar, current, text, attachments);
|
|
131
|
+
return structuredClone(run);
|
|
132
|
+
}
|
|
133
|
+
async execute(ar, conversation, text, attachments) {
|
|
134
|
+
let result;
|
|
135
|
+
try {
|
|
136
|
+
result = await this.opts.backend.run({ text, attachments }, {
|
|
137
|
+
runId: ar.run.id,
|
|
138
|
+
conversation,
|
|
139
|
+
builder: ar.builder,
|
|
140
|
+
signal: ar.controller.signal,
|
|
141
|
+
setActivity: (activity) => this.setActivity(ar, activity),
|
|
142
|
+
});
|
|
143
|
+
}
|
|
144
|
+
catch (err) {
|
|
145
|
+
this.opts.logger.warn("Run failed", {
|
|
146
|
+
runId: ar.run.id,
|
|
147
|
+
error: err,
|
|
148
|
+
});
|
|
149
|
+
result = {
|
|
150
|
+
status: "failed",
|
|
151
|
+
error: { message: err.message ?? String(err) },
|
|
152
|
+
};
|
|
153
|
+
}
|
|
154
|
+
try {
|
|
155
|
+
this.finalize(ar, result);
|
|
156
|
+
}
|
|
157
|
+
catch (err) {
|
|
158
|
+
this.opts.logger.error("Could not finalize run", {
|
|
159
|
+
runId: ar.run.id,
|
|
160
|
+
error: err,
|
|
161
|
+
});
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
finalize(ar, result) {
|
|
165
|
+
const { store } = this.opts;
|
|
166
|
+
const { run, message, builder } = ar;
|
|
167
|
+
const cancelled = ar.controller.signal.aborted;
|
|
168
|
+
const status = cancelled ? "cancelled" : result.status;
|
|
169
|
+
const now = new Date().toISOString();
|
|
170
|
+
for (const approval of [...ar.approvals.values()]) {
|
|
171
|
+
approval.finish({ approved: false, reason: "cancelled" }, "cancelled");
|
|
172
|
+
}
|
|
173
|
+
for (const block of message.blocks) {
|
|
174
|
+
if (block.type === "tool" &&
|
|
175
|
+
(block.status === "pending" ||
|
|
176
|
+
block.status === "running" ||
|
|
177
|
+
block.status === "awaiting_approval")) {
|
|
178
|
+
builder.updateBlock(block.id, {
|
|
179
|
+
status: cancelled ? "cancelled" : "failed",
|
|
180
|
+
completedAt: now,
|
|
181
|
+
});
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
if (status === "failed" &&
|
|
185
|
+
result.error &&
|
|
186
|
+
!message.blocks.some((b) => b.type === "error")) {
|
|
187
|
+
builder.addBlock({
|
|
188
|
+
type: "error",
|
|
189
|
+
message: result.error.message,
|
|
190
|
+
code: result.error.code,
|
|
191
|
+
});
|
|
192
|
+
}
|
|
193
|
+
message.status =
|
|
194
|
+
status === "completed"
|
|
195
|
+
? "completed"
|
|
196
|
+
: status === "cancelled"
|
|
197
|
+
? "cancelled"
|
|
198
|
+
: "failed";
|
|
199
|
+
message.completedAt = now;
|
|
200
|
+
message.model ??= this.opts.backend.model;
|
|
201
|
+
if (result.usage) {
|
|
202
|
+
message.usage = result.usage;
|
|
203
|
+
}
|
|
204
|
+
if (result.error && status === "failed") {
|
|
205
|
+
message.error = result.error;
|
|
206
|
+
}
|
|
207
|
+
run.status = status;
|
|
208
|
+
run.completedAt = now;
|
|
209
|
+
delete run.activity;
|
|
210
|
+
if (result.usage) {
|
|
211
|
+
run.usage = result.usage;
|
|
212
|
+
}
|
|
213
|
+
if (result.error && status === "failed") {
|
|
214
|
+
run.error = result.error;
|
|
215
|
+
}
|
|
216
|
+
this.flushDeltas(ar);
|
|
217
|
+
if (ar.persistTimer) {
|
|
218
|
+
clearTimeout(ar.persistTimer);
|
|
219
|
+
ar.persistTimer = undefined;
|
|
220
|
+
}
|
|
221
|
+
store.appendMessage(run.conversationId, message);
|
|
222
|
+
store.clearPending(run.conversationId);
|
|
223
|
+
store.update(run.conversationId, { activeRunId: undefined });
|
|
224
|
+
ar.done = true;
|
|
225
|
+
this.activeByConversation.delete(run.conversationId);
|
|
226
|
+
this.push(ar, {
|
|
227
|
+
type: "message.completed",
|
|
228
|
+
message: structuredClone(message),
|
|
229
|
+
});
|
|
230
|
+
this.push(ar, {
|
|
231
|
+
type: status === "completed"
|
|
232
|
+
? "run.completed"
|
|
233
|
+
: status === "cancelled"
|
|
234
|
+
? "run.cancelled"
|
|
235
|
+
: "run.failed",
|
|
236
|
+
run: structuredClone(run),
|
|
237
|
+
});
|
|
238
|
+
ar.listeners.clear();
|
|
239
|
+
ar.resolveDone();
|
|
240
|
+
setTimeout(() => {
|
|
241
|
+
if (this.runs.get(run.id) === ar) {
|
|
242
|
+
this.runs.delete(run.id);
|
|
243
|
+
}
|
|
244
|
+
}, this.retentionMs).unref();
|
|
245
|
+
}
|
|
246
|
+
emit(ar, payload) {
|
|
247
|
+
if (payload.type === "block.delta") {
|
|
248
|
+
const pending = ar.pendingDeltas.get(payload.blockId);
|
|
249
|
+
if (pending) {
|
|
250
|
+
pending.text += payload.text;
|
|
251
|
+
}
|
|
252
|
+
else {
|
|
253
|
+
ar.pendingDeltas.set(payload.blockId, {
|
|
254
|
+
messageId: payload.messageId,
|
|
255
|
+
text: payload.text,
|
|
256
|
+
});
|
|
257
|
+
}
|
|
258
|
+
if (!ar.flushTimer) {
|
|
259
|
+
ar.flushTimer = setTimeout(() => this.flushDeltas(ar), 30);
|
|
260
|
+
}
|
|
261
|
+
return;
|
|
262
|
+
}
|
|
263
|
+
this.flushDeltas(ar);
|
|
264
|
+
this.push(ar, payload);
|
|
265
|
+
}
|
|
266
|
+
flushDeltas(ar) {
|
|
267
|
+
if (ar.flushTimer) {
|
|
268
|
+
clearTimeout(ar.flushTimer);
|
|
269
|
+
ar.flushTimer = undefined;
|
|
270
|
+
}
|
|
271
|
+
if (ar.pendingDeltas.size === 0) {
|
|
272
|
+
return;
|
|
273
|
+
}
|
|
274
|
+
const pending = [...ar.pendingDeltas.entries()];
|
|
275
|
+
ar.pendingDeltas.clear();
|
|
276
|
+
for (const [blockId, { messageId, text }] of pending) {
|
|
277
|
+
this.push(ar, { type: "block.delta", messageId, blockId, text });
|
|
278
|
+
}
|
|
279
|
+
}
|
|
280
|
+
push(ar, payload) {
|
|
281
|
+
const event = {
|
|
282
|
+
seq: ++ar.run.lastSeq,
|
|
283
|
+
ts: new Date().toISOString(),
|
|
284
|
+
conversationId: ar.run.conversationId,
|
|
285
|
+
runId: ar.run.id,
|
|
286
|
+
...payload,
|
|
287
|
+
};
|
|
288
|
+
ar.events.push(event);
|
|
289
|
+
for (const listener of ar.listeners) {
|
|
290
|
+
try {
|
|
291
|
+
listener(event);
|
|
292
|
+
}
|
|
293
|
+
catch (err) {
|
|
294
|
+
this.opts.logger.debug("Event listener failed", {
|
|
295
|
+
error: err,
|
|
296
|
+
});
|
|
297
|
+
}
|
|
298
|
+
}
|
|
299
|
+
}
|
|
300
|
+
schedulePersist(ar) {
|
|
301
|
+
if (ar.persistTimer || ar.done) {
|
|
302
|
+
return;
|
|
303
|
+
}
|
|
304
|
+
ar.persistTimer = setTimeout(() => {
|
|
305
|
+
ar.persistTimer = undefined;
|
|
306
|
+
if (!ar.done) {
|
|
307
|
+
try {
|
|
308
|
+
this.opts.store.savePending(ar.run.conversationId, ar.message);
|
|
309
|
+
}
|
|
310
|
+
catch (err) {
|
|
311
|
+
this.opts.logger.warn("Could not persist the pending message", {
|
|
312
|
+
error: err,
|
|
313
|
+
});
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
}, 1000);
|
|
317
|
+
ar.persistTimer.unref();
|
|
318
|
+
}
|
|
319
|
+
setStatus(ar, status) {
|
|
320
|
+
if (ar.done || ar.run.status === status) {
|
|
321
|
+
return;
|
|
322
|
+
}
|
|
323
|
+
ar.run.status = status;
|
|
324
|
+
this.emit(ar, { type: "run.updated", run: structuredClone(ar.run) });
|
|
325
|
+
}
|
|
326
|
+
setActivity(ar, activity) {
|
|
327
|
+
if (ar.done) {
|
|
328
|
+
return;
|
|
329
|
+
}
|
|
330
|
+
if (activity) {
|
|
331
|
+
ar.run.activity = activity;
|
|
332
|
+
}
|
|
333
|
+
else if (ar.run.activity) {
|
|
334
|
+
delete ar.run.activity;
|
|
335
|
+
}
|
|
336
|
+
else {
|
|
337
|
+
return;
|
|
338
|
+
}
|
|
339
|
+
this.emit(ar, { type: "run.updated", run: structuredClone(ar.run) });
|
|
340
|
+
}
|
|
341
|
+
async generateTitle(conversationId, text, fallbackTitle) {
|
|
342
|
+
try {
|
|
343
|
+
const title = await this.opts.backend.generateTitle(text);
|
|
344
|
+
if (!title) {
|
|
345
|
+
return;
|
|
346
|
+
}
|
|
347
|
+
const conversation = this.opts.store.get(conversationId);
|
|
348
|
+
if (!conversation || conversation.title !== fallbackTitle) {
|
|
349
|
+
return;
|
|
350
|
+
}
|
|
351
|
+
const updated = this.opts.store.update(conversationId, { title });
|
|
352
|
+
const ar = this.activeByConversation.get(conversationId);
|
|
353
|
+
if (updated && ar && !ar.done) {
|
|
354
|
+
this.emit(ar, {
|
|
355
|
+
type: "conversation.updated",
|
|
356
|
+
conversation: structuredClone(updated),
|
|
357
|
+
});
|
|
358
|
+
}
|
|
359
|
+
}
|
|
360
|
+
catch (err) {
|
|
361
|
+
this.opts.logger.debug("Could not generate a title", {
|
|
362
|
+
error: err,
|
|
363
|
+
});
|
|
364
|
+
}
|
|
365
|
+
}
|
|
366
|
+
getRun(runId) {
|
|
367
|
+
const ar = this.runs.get(runId);
|
|
368
|
+
return ar ? structuredClone(ar.run) : undefined;
|
|
369
|
+
}
|
|
370
|
+
getSnapshot(runId) {
|
|
371
|
+
const ar = this.runs.get(runId);
|
|
372
|
+
if (!ar) {
|
|
373
|
+
return undefined;
|
|
374
|
+
}
|
|
375
|
+
this.flushDeltas(ar);
|
|
376
|
+
return {
|
|
377
|
+
run: structuredClone(ar.run),
|
|
378
|
+
message: structuredClone(ar.message),
|
|
379
|
+
};
|
|
380
|
+
}
|
|
381
|
+
getActiveSnapshot(conversationId) {
|
|
382
|
+
const ar = this.activeByConversation.get(conversationId);
|
|
383
|
+
return ar ? this.getSnapshot(ar.run.id) : undefined;
|
|
384
|
+
}
|
|
385
|
+
subscribe(runId, afterSeq, listener) {
|
|
386
|
+
const ar = this.runs.get(runId);
|
|
387
|
+
if (!ar) {
|
|
388
|
+
return undefined;
|
|
389
|
+
}
|
|
390
|
+
this.flushDeltas(ar);
|
|
391
|
+
const replay = ar.events.filter((e) => e.seq > afterSeq);
|
|
392
|
+
if (ar.done) {
|
|
393
|
+
return { replay, finished: true, unsubscribe: () => { } };
|
|
394
|
+
}
|
|
395
|
+
ar.listeners.add(listener);
|
|
396
|
+
return {
|
|
397
|
+
replay,
|
|
398
|
+
finished: false,
|
|
399
|
+
unsubscribe: () => {
|
|
400
|
+
ar.listeners.delete(listener);
|
|
401
|
+
},
|
|
402
|
+
};
|
|
403
|
+
}
|
|
404
|
+
cancel(runId) {
|
|
405
|
+
const ar = this.runs.get(runId);
|
|
406
|
+
if (!ar) {
|
|
407
|
+
throw notFound(`Run not found: ${runId}`);
|
|
408
|
+
}
|
|
409
|
+
if (!ar.done && !ar.controller.signal.aborted) {
|
|
410
|
+
ar.controller.abort();
|
|
411
|
+
for (const approval of [...ar.approvals.values()]) {
|
|
412
|
+
approval.finish({ approved: false, reason: "cancelled" }, "cancelled");
|
|
413
|
+
}
|
|
414
|
+
}
|
|
415
|
+
return structuredClone(ar.run);
|
|
416
|
+
}
|
|
417
|
+
async cancelConversation(conversationId) {
|
|
418
|
+
const ar = this.activeByConversation.get(conversationId);
|
|
419
|
+
if (!ar) {
|
|
420
|
+
return;
|
|
421
|
+
}
|
|
422
|
+
this.cancel(ar.run.id);
|
|
423
|
+
await Promise.race([
|
|
424
|
+
ar.donePromise,
|
|
425
|
+
new Promise((resolve) => setTimeout(resolve, 10000).unref()),
|
|
426
|
+
]);
|
|
427
|
+
}
|
|
428
|
+
forgetConversation(conversationId) {
|
|
429
|
+
this.resourceCaches.delete(conversationId);
|
|
430
|
+
}
|
|
431
|
+
decideApproval(runId, approvalId, decision, reason) {
|
|
432
|
+
const ar = this.runs.get(runId);
|
|
433
|
+
if (!ar) {
|
|
434
|
+
throw notFound(`Run not found: ${runId}`);
|
|
435
|
+
}
|
|
436
|
+
const approval = ar.approvals.get(approvalId);
|
|
437
|
+
if (!approval) {
|
|
438
|
+
throw notFound(`Pending approval not found: ${approvalId}`);
|
|
439
|
+
}
|
|
440
|
+
if (decision !== "approve" && decision !== "reject") {
|
|
441
|
+
throw badRequest('The decision must be either "approve" or "reject"');
|
|
442
|
+
}
|
|
443
|
+
approval.finish({ approved: decision === "approve", reason }, decision === "approve" ? "approved" : "rejected");
|
|
444
|
+
return structuredClone(ar.builder.getBlock(approval.blockId));
|
|
445
|
+
}
|
|
446
|
+
requestApproval(ar, req, signal) {
|
|
447
|
+
if (ar.done || ar.controller.signal.aborted) {
|
|
448
|
+
return Promise.resolve({ approved: false, reason: "cancelled" });
|
|
449
|
+
}
|
|
450
|
+
const { builder } = ar;
|
|
451
|
+
const approvalId = newID();
|
|
452
|
+
const toolBlock = builder.getToolBlock(req.toolCallId);
|
|
453
|
+
if (toolBlock) {
|
|
454
|
+
builder.updateBlock(toolBlock.id, {
|
|
455
|
+
status: "awaiting_approval",
|
|
456
|
+
});
|
|
457
|
+
}
|
|
458
|
+
const block = builder.addBlock({
|
|
459
|
+
type: "approval",
|
|
460
|
+
approvalId,
|
|
461
|
+
toolCallId: req.toolCallId,
|
|
462
|
+
title: req.title,
|
|
463
|
+
description: req.description,
|
|
464
|
+
risk: req.risk,
|
|
465
|
+
status: "pending",
|
|
466
|
+
preview: req.preview,
|
|
467
|
+
});
|
|
468
|
+
this.setStatus(ar, "awaiting_approval");
|
|
469
|
+
return new Promise((resolve) => {
|
|
470
|
+
let settled = false;
|
|
471
|
+
const onAbort = () => finish({ approved: false, reason: "cancelled" }, "cancelled");
|
|
472
|
+
const finish = (result, status) => {
|
|
473
|
+
if (settled) {
|
|
474
|
+
return;
|
|
475
|
+
}
|
|
476
|
+
settled = true;
|
|
477
|
+
signal?.removeEventListener("abort", onAbort);
|
|
478
|
+
ar.approvals.delete(approvalId);
|
|
479
|
+
builder.updateBlock(block.id, {
|
|
480
|
+
status,
|
|
481
|
+
reason: result.reason,
|
|
482
|
+
decidedAt: new Date().toISOString(),
|
|
483
|
+
});
|
|
484
|
+
const tool = builder.getToolBlock(req.toolCallId);
|
|
485
|
+
if (tool && status !== "cancelled") {
|
|
486
|
+
builder.updateBlock(tool.id, {
|
|
487
|
+
status: result.approved ? "running" : "rejected",
|
|
488
|
+
});
|
|
489
|
+
}
|
|
490
|
+
if (ar.approvals.size === 0 && !ar.done) {
|
|
491
|
+
this.setStatus(ar, "running");
|
|
492
|
+
}
|
|
493
|
+
resolve(result);
|
|
494
|
+
};
|
|
495
|
+
ar.approvals.set(approvalId, {
|
|
496
|
+
blockId: block.id,
|
|
497
|
+
toolCallId: req.toolCallId,
|
|
498
|
+
finish,
|
|
499
|
+
});
|
|
500
|
+
signal?.addEventListener("abort", onAbort, { once: true });
|
|
501
|
+
});
|
|
502
|
+
}
|
|
503
|
+
resourceCache(conversationId) {
|
|
504
|
+
let cache = this.resourceCaches.get(conversationId);
|
|
505
|
+
if (!cache) {
|
|
506
|
+
cache = new ResourceCache();
|
|
507
|
+
this.resourceCaches.set(conversationId, cache);
|
|
508
|
+
}
|
|
509
|
+
return cache;
|
|
510
|
+
}
|
|
511
|
+
current(conversationId) {
|
|
512
|
+
const ar = this.activeByConversation.get(conversationId);
|
|
513
|
+
if (!ar || ar.done) {
|
|
514
|
+
return undefined;
|
|
515
|
+
}
|
|
516
|
+
return {
|
|
517
|
+
runId: ar.run.id,
|
|
518
|
+
conversationId,
|
|
519
|
+
builder: ar.builder,
|
|
520
|
+
resources: this.resourceCache(conversationId),
|
|
521
|
+
resultsDir: this.opts.store.resultsDir(conversationId),
|
|
522
|
+
requestApproval: (req, signal) => this.requestApproval(ar, req, signal),
|
|
523
|
+
};
|
|
524
|
+
}
|
|
525
|
+
async shutdown(timeoutMs = 10000) {
|
|
526
|
+
const pending = [...this.activeByConversation.values()];
|
|
527
|
+
for (const ar of pending) {
|
|
528
|
+
this.cancel(ar.run.id);
|
|
529
|
+
}
|
|
530
|
+
await Promise.race([
|
|
531
|
+
Promise.all(pending.map((ar) => ar.donePromise)),
|
|
532
|
+
new Promise((resolve) => setTimeout(resolve, timeoutMs).unref()),
|
|
533
|
+
]);
|
|
534
|
+
}
|
|
535
|
+
}
|
|
536
|
+
//# sourceMappingURL=manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/runs/manager.ts"],"names":[],"mappings":"AAuBA,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/E,OAAO,EAAE,KAAK,EAA0B,MAAM,2BAA2B,CAAC;AAE1E,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,MAAM,CAAC,MAAM,0BAA0B,GAAG,kBAAkB,CAAC;AAE7D,MAAM,aAAa,GAAG,OAAO,CAAC;AAwC9B,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,GAAG,GAAG,EAAE,EAAU,EAAE;IAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9C,OAAO,IAAI,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC;AAC5E,CAAC,CAAC;AAEF,MAAM,OAAO,UAAU;IACb,IAAI,CAAoB;IACxB,IAAI,GAAG,IAAI,GAAG,EAAqB,CAAC;IACpC,oBAAoB,GAAG,IAAI,GAAG,EAAqB,CAAC;IACpD,cAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;IAClD,WAAW,CAAS;IAE5B,YAAY,IAAuB;QACjC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IACxD,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC;IACxC,CAAC;IAED,QAAQ,CAAC,cAAsB,EAAE,KAAe;QAC9C,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;QACnC,MAAM,YAAY,GAAG,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,QAAQ,CAAC,2BAA2B,cAAc,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;YAClD,MAAM,QAAQ,CAAC,4CAA4C,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAClE,MAAM,eAAe,CAAC,0BAA0B,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,IAAI,GAAG,OAAO,KAAK,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC;YACrD,CAAC,CAAC,KAAK,CAAC,WAAW;YACnB,CAAC,CAAC,EAAE,CAAC;QACP,IAAI,IAAI,CAAC,MAAM,GAAG,aAAa,EAAE,CAAC;YAChC,MAAM,UAAU,CAAC,qBAAqB,aAAa,aAAa,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,WAAW,GAAe,EAAE,CAAC;QACnC,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,UAAU,CAAC,uBAAuB,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACxD,CAAC;YACD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;QAED,IAAI,IAAI,KAAK,EAAE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,MAAM,UAAU,CAAC,kCAAkC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,KAAK,EAAE,CAAC;QAEtB,MAAM,WAAW,GAAY;YAC3B,EAAE,EAAE,KAAK,EAAE;YACX,cAAc;YACd,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,GAAG;YACd,WAAW,EAAE,GAAG;YAChB,KAAK;YACL,MAAM,EACJ,IAAI,KAAK,EAAE;gBACT,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC;YAC5D,WAAW,EAAE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC9D,CAAC;QACF,KAAK,CAAC,aAAa,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAY;YACvB,EAAE,EAAE,KAAK,EAAE;YACX,cAAc;YACd,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,GAAG;YACd,KAAK;YACL,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK;SAC/B,CAAC;QAEF,MAAM,GAAG,GAAQ;YACf,EAAE,EAAE,KAAK;YACT,cAAc;YACd,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,GAAG;YACd,aAAa,EAAE,WAAW,CAAC,EAAE;YAC7B,kBAAkB,EAAE,OAAO,CAAC,EAAE;YAC9B,OAAO,EAAE,CAAC;SACX,CAAC;QAEF,IAAI,WAAW,GAAG,GAAG,EAAE,GAAE,CAAC,CAAC;QAC3B,MAAM,WAAW,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAChD,WAAW,GAAG,OAAO,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,GAAc;YACpB,GAAG;YACH,OAAO;YACP,OAAO,EAAE,SAAsC;YAC/C,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,GAAG,EAAE;YACpB,UAAU,EAAE,IAAI,eAAe,EAAE;YACjC,SAAS,EAAE,IAAI,GAAG,EAAE;YACpB,aAAa,EAAE,IAAI,GAAG,EAAE;YACxB,IAAI,EAAE,KAAK;YACX,WAAW;YACX,WAAW;SACZ,CAAC;QACF,EAAE,CAAC,OAAO,GAAG,IAAI,cAAc,CAC7B,OAAO,EACP,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,EACnC,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAC/B,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACzB,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAClD,IAAI,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,CAAE,CAAC;QAEpE,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,eAAe,CAAC,WAAW,CAAC;SACtC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,eAAe,CAAC,OAAO,CAAC;SAClC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,KAAK,0BAA0B,CAAC;QAClE,IAAI,OAAO,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YAC3B,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC;YACxE,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;gBACZ,IAAI,EAAE,sBAAsB;gBAC5B,YAAY,EAAE,eAAe,CAAC,OAAO,CAAC;aACvC,CAAC,CAAC;YACH,IAAI,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC7B,KAAK,IAAI,CAAC,aAAa,CAAC,cAAc,EAAE,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,KAAK,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAElD,OAAO,eAAe,CAAC,GAAG,CAAC,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,EAAa,EACb,YAA0B,EAC1B,IAAY,EACZ,WAAuB;QAEvB,IAAI,MAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAClC,EAAE,IAAI,EAAE,WAAW,EAAE,EACrB;gBACE,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE;gBAChB,YAAY;gBACZ,OAAO,EAAE,EAAE,CAAC,OAAO;gBACnB,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM;gBAC5B,WAAW,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,QAAQ,CAAC;aAC1D,CACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE;gBAClC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE;gBAChB,KAAK,EAAE,GAAY;aACpB,CAAC,CAAC;YACH,MAAM,GAAG;gBACP,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,EAAE,OAAO,EAAG,GAAa,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE;aAC1D,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;gBAC/C,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE;gBAChB,KAAK,EAAE,GAAY;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,QAAQ,CAAC,EAAa,EAAE,MAAwB;QACtD,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;QAC5B,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC;QAC/C,MAAM,MAAM,GAAc,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QAClE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,KAAK,MAAM,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAClD,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,WAAW,CAAC,CAAC;QACzE,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,IACE,KAAK,CAAC,IAAI,KAAK,MAAM;gBACrB,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS;oBACzB,KAAK,CAAC,MAAM,KAAK,SAAS;oBAC1B,KAAK,CAAC,MAAM,KAAK,mBAAmB,CAAC,EACvC,CAAC;gBACD,OAAO,CAAC,WAAW,CAAY,KAAK,CAAC,EAAE,EAAE;oBACvC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;oBAC1C,WAAW,EAAE,GAAG;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IACE,MAAM,KAAK,QAAQ;YACnB,MAAM,CAAC,KAAK;YACZ,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,EAC/C,CAAC;YACD,OAAO,CAAC,QAAQ,CAAC;gBACf,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;gBAC7B,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;aACxB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,MAAM;YACZ,MAAM,KAAK,WAAW;gBACpB,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,MAAM,KAAK,WAAW;oBACtB,CAAC,CAAC,WAAW;oBACb,CAAC,CAAC,QAAQ,CAAC;QACjB,OAAO,CAAC,WAAW,GAAG,GAAG,CAAC;QAC1B,OAAO,CAAC,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QAC1C,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC/B,CAAC;QAED,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;QACpB,GAAG,CAAC,WAAW,GAAG,GAAG,CAAC;QACtB,OAAO,GAAG,CAAC,QAAQ,CAAC;QACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC3B,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxC,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACrB,IAAI,EAAE,CAAC,YAAY,EAAE,CAAC;YACpB,YAAY,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC;YAC9B,EAAE,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACjD,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACvC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;QAE7D,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC;QACf,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAErD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,IAAI,EAAE,mBAAmB;YACzB,OAAO,EAAE,eAAe,CAAC,OAAO,CAAC;SAClC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,IAAI,EACF,MAAM,KAAK,WAAW;gBACpB,CAAC,CAAC,eAAe;gBACjB,CAAC,CAAC,MAAM,KAAK,WAAW;oBACtB,CAAC,CAAC,eAAe;oBACjB,CAAC,CAAC,YAAY;YACpB,GAAG,EAAE,eAAe,CAAC,GAAG,CAAC;SAC1B,CAAC,CAAC;QAEH,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACrB,EAAE,CAAC,WAAW,EAAE,CAAC;QAEjB,UAAU,CAAC,GAAG,EAAE;YACd,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;gBACjC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAEO,IAAI,CAAC,EAAa,EAAE,OAA0B;QACpD,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE;oBACpC,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;YACD,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC;gBACnB,EAAE,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC;IAEO,WAAW,CAAC,EAAa;QAC/B,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;YAClB,YAAY,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;YAC5B,EAAE,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,CAAC;QACD,IAAI,EAAE,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAChD,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QACzB,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,IAAI,OAAO,EAAE,CAAC;YACrD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAEO,IAAI,CAAC,EAAa,EAAE,OAA0B;QACpD,MAAM,KAAK,GAAG;YACZ,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO;YACrB,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC5B,cAAc,EAAE,EAAE,CAAC,GAAG,CAAC,cAAc;YACrC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE;YAChB,GAAG,OAAO;SACG,CAAC;QAChB,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtB,KAAK,MAAM,QAAQ,IAAI,EAAE,CAAC,SAAS,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE;oBAC9C,KAAK,EAAE,GAAY;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,EAAa;QACnC,IAAI,EAAE,CAAC,YAAY,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,EAAE,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,EAAE,CAAC,YAAY,GAAG,SAAS,CAAC;YAC5B,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBACb,IAAI,CAAC;oBACH,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC;gBACjE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;wBAC7D,KAAK,EAAE,GAAY;qBACpB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,CAAC;QACT,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAEO,SAAS,CAAC,EAAa,EAAE,MAAiB;QAChD,IAAI,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,EAAE,CAAC,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;QACvB,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,eAAe,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,WAAW,CAAC,EAAa,EAAE,QAAiC;QAClE,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,EAAE,CAAC;YACb,EAAE,CAAC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC7B,CAAC;aAAM,IAAI,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC3B,OAAO,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC;QACzB,CAAC;aAAM,CAAC;YACN,OAAO;QACT,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,eAAe,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,cAAsB,EACtB,IAAY,EACZ,aAAqB;QAErB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACzD,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;gBAC1D,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAClE,MAAM,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACzD,IAAI,OAAO,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC9B,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;oBACZ,IAAI,EAAE,sBAAsB;oBAC5B,YAAY,EAAE,eAAe,CAAC,OAAO,CAAC;iBACvC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE;gBACnD,KAAK,EAAE,GAAY;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChC,OAAO,EAAE,CAAC,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClD,CAAC;IAED,WAAW,CAAC,KAAa;QACvB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACrB,OAAO;YACL,GAAG,EAAE,eAAe,CAAC,EAAE,CAAC,GAAG,CAAC;YAC5B,OAAO,EAAE,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC;SACrC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,cAAsB;QACtC,MAAM,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACzD,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,SAAS,CACP,KAAa,EACb,QAAgB,EAChB,QAAqC;QAErC,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACrB,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC;QACzD,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC;QAC3D,CAAC;QACD,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3B,OAAO;YACL,MAAM;YACN,QAAQ,EAAE,KAAK;YACf,WAAW,EAAE,GAAG,EAAE;gBAChB,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChC,CAAC;SACF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,MAAM,QAAQ,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC9C,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACtB,KAAK,MAAM,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;gBAClD,QAAQ,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,WAAW,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;QACD,OAAO,eAAe,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,cAAsB;QAC7C,MAAM,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACzD,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvB,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,EAAE,CAAC,WAAW;YACd,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC;SAC7D,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,cAAsB;QACvC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IAC7C,CAAC;IAED,cAAc,CACZ,KAAa,EACb,UAAkB,EAClB,QAA0B,EAC1B,MAAe;QAEf,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,MAAM,QAAQ,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,QAAQ,GAAG,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,QAAQ,CAAC,+BAA+B,UAAU,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACpD,MAAM,UAAU,CAAC,mDAAmD,CAAC,CAAC;QACxE,CAAC;QACD,QAAQ,CAAC,MAAM,CACb,EAAE,QAAQ,EAAE,QAAQ,KAAK,SAAS,EAAE,MAAM,EAAE,EAC5C,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CACjD,CAAC;QACF,OAAO,eAAe,CACpB,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAgB,QAAQ,CAAC,OAAO,CAAE,CACtD,CAAC;IACJ,CAAC;IAEO,eAAe,CACrB,EAAa,EACb,GAAoB,EACpB,MAAoB;QAEpB,IAAI,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC5C,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,KAAK,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACvD,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,CAAC,WAAW,CAAY,SAAS,CAAC,EAAE,EAAE;gBAC3C,MAAM,EAAE,mBAAmB;aAC5B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAgB;YAC5C,IAAI,EAAE,UAAU;YAChB,UAAU;YACV,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,GAAG,CAAC,OAAO;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC;QAExC,OAAO,IAAI,OAAO,CAAiB,CAAC,OAAO,EAAE,EAAE;YAC7C,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,MAAM,OAAO,GAAG,GAAG,EAAE,CACnB,MAAM,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,WAAW,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,CACb,MAAsB,EACtB,MAA+B,EAC/B,EAAE;gBACF,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO;gBACT,CAAC;gBACD,OAAO,GAAG,IAAI,CAAC;gBACf,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC9C,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAEhC,OAAO,CAAC,WAAW,CAAgB,KAAK,CAAC,EAAE,EAAE;oBAC3C,MAAM;oBACN,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC,CAAC;gBAEH,MAAM,IAAI,GAAG,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAClD,IAAI,IAAI,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;oBACnC,OAAO,CAAC,WAAW,CAAY,IAAI,CAAC,EAAE,EAAE;wBACtC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;qBACjD,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;oBACxC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;gBAChC,CAAC;gBACD,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC,CAAC;YAEF,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;gBAC3B,OAAO,EAAE,KAAK,CAAC,EAAE;gBACjB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,MAAM;aACP,CAAC,CAAC;YACH,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,cAAsB;QAC1C,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,IAAI,aAAa,EAAE,CAAC;YAC5B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CAAC,cAAsB;QAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACzD,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE;YAChB,cAAc;YACd,OAAO,EAAE,EAAE,CAAC,OAAO;YACnB,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;YAC7C,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,CAAC;YACtD,eAAe,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC;SACxE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAAS,GAAG,KAAK;QAC9B,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC;QACxD,KAAK,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;QACD,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;YAChD,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,KAAK,EAAE,CAAC;SACjE,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { IncomingMessage, ServerResponse } from "node:http";
|
|
2
|
+
export declare const sendJSON: (res: ServerResponse, status: number, body: unknown) => void;
|
|
3
|
+
export declare const sendNoContent: (res: ServerResponse) => void;
|
|
4
|
+
export declare const sendError: (res: ServerResponse, err: unknown) => void;
|
|
5
|
+
export declare const readBody: (req: IncomingMessage, maxBytes: number) => Promise<Buffer>;
|
|
6
|
+
export declare const readJSON: <T>(req: IncomingMessage, maxBytes?: number) => Promise<T>;
|
|
7
|
+
export declare const originMatches: (origin: string, pattern: string) => boolean;
|
|
8
|
+
export declare const requestOrigin: (req: IncomingMessage) => string | undefined;
|
|
9
|
+
export declare const contentDisposition: (type: "inline" | "attachment", name: string) => string;
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { badRequest, HTTPError, payloadTooLarge } from "../errors.js";
|
|
2
|
+
export const sendJSON = (res, status, body) => {
|
|
3
|
+
const data = JSON.stringify(body);
|
|
4
|
+
res.writeHead(status, {
|
|
5
|
+
"content-type": "application/json; charset=utf-8",
|
|
6
|
+
"content-length": Buffer.byteLength(data),
|
|
7
|
+
"cache-control": "no-store",
|
|
8
|
+
});
|
|
9
|
+
res.end(data);
|
|
10
|
+
};
|
|
11
|
+
export const sendNoContent = (res) => {
|
|
12
|
+
res.writeHead(204, { "cache-control": "no-store" });
|
|
13
|
+
res.end();
|
|
14
|
+
};
|
|
15
|
+
export const sendError = (res, err) => {
|
|
16
|
+
const httpErr = err instanceof HTTPError
|
|
17
|
+
? err
|
|
18
|
+
: new HTTPError(500, "internal", "Internal server error");
|
|
19
|
+
if (res.headersSent) {
|
|
20
|
+
res.destroy();
|
|
21
|
+
return;
|
|
22
|
+
}
|
|
23
|
+
const body = {
|
|
24
|
+
error: { code: httpErr.code, message: httpErr.message },
|
|
25
|
+
};
|
|
26
|
+
sendJSON(res, httpErr.status, body);
|
|
27
|
+
};
|
|
28
|
+
export const readBody = async (req, maxBytes) => {
|
|
29
|
+
const contentLength = Number(req.headers["content-length"] ?? "0");
|
|
30
|
+
if (contentLength > maxBytes) {
|
|
31
|
+
throw payloadTooLarge(`The request body exceeds ${maxBytes} bytes`);
|
|
32
|
+
}
|
|
33
|
+
const chunks = [];
|
|
34
|
+
let size = 0;
|
|
35
|
+
for await (const chunk of req) {
|
|
36
|
+
size += chunk.length;
|
|
37
|
+
if (size > maxBytes) {
|
|
38
|
+
throw payloadTooLarge(`The request body exceeds ${maxBytes} bytes`);
|
|
39
|
+
}
|
|
40
|
+
chunks.push(chunk);
|
|
41
|
+
}
|
|
42
|
+
return Buffer.concat(chunks);
|
|
43
|
+
};
|
|
44
|
+
export const readJSON = async (req, maxBytes = 1024 * 1024) => {
|
|
45
|
+
const body = await readBody(req, maxBytes);
|
|
46
|
+
if (body.length === 0) {
|
|
47
|
+
return {};
|
|
48
|
+
}
|
|
49
|
+
const contentType = (req.headers["content-type"] ?? "").toLowerCase();
|
|
50
|
+
if (!contentType.startsWith("application/json")) {
|
|
51
|
+
throw badRequest("The request body must be application/json");
|
|
52
|
+
}
|
|
53
|
+
try {
|
|
54
|
+
const parsed = JSON.parse(body.toString("utf8"));
|
|
55
|
+
if (typeof parsed !== "object" ||
|
|
56
|
+
parsed === null ||
|
|
57
|
+
Array.isArray(parsed)) {
|
|
58
|
+
throw new Error("not an object");
|
|
59
|
+
}
|
|
60
|
+
return parsed;
|
|
61
|
+
}
|
|
62
|
+
catch {
|
|
63
|
+
throw badRequest("The request body must be a JSON object");
|
|
64
|
+
}
|
|
65
|
+
};
|
|
66
|
+
export const originMatches = (origin, pattern) => {
|
|
67
|
+
if (pattern === "*") {
|
|
68
|
+
return true;
|
|
69
|
+
}
|
|
70
|
+
if (!pattern.includes("*")) {
|
|
71
|
+
return origin === pattern;
|
|
72
|
+
}
|
|
73
|
+
const idx = pattern.indexOf("*.");
|
|
74
|
+
if (idx < 0) {
|
|
75
|
+
return false;
|
|
76
|
+
}
|
|
77
|
+
const prefix = pattern.slice(0, idx);
|
|
78
|
+
const suffix = pattern.slice(idx + 1);
|
|
79
|
+
return (origin.startsWith(prefix) &&
|
|
80
|
+
origin.endsWith(suffix) &&
|
|
81
|
+
origin.length > prefix.length + suffix.length &&
|
|
82
|
+
!origin.slice(prefix.length, origin.length - suffix.length).includes("/"));
|
|
83
|
+
};
|
|
84
|
+
export const requestOrigin = (req) => {
|
|
85
|
+
const host = req.headers["x-forwarded-host"]
|
|
86
|
+
?.split(",")[0]
|
|
87
|
+
?.trim() ?? req.headers.host;
|
|
88
|
+
if (!host) {
|
|
89
|
+
return undefined;
|
|
90
|
+
}
|
|
91
|
+
const proto = req.headers["x-forwarded-proto"]
|
|
92
|
+
?.split(",")[0]
|
|
93
|
+
?.trim() ?? "http";
|
|
94
|
+
return `${proto}://${host}`;
|
|
95
|
+
};
|
|
96
|
+
export const contentDisposition = (type, name) => {
|
|
97
|
+
const fallback = name.replace(/[^\x20-\x7e]|["\\]/g, "_");
|
|
98
|
+
return `${type}; filename="${fallback}"; filename*=UTF-8''${encodeURIComponent(name)}`;
|
|
99
|
+
};
|
|
100
|
+
//# sourceMappingURL=http.js.map
|