@relaycast/engine 3.1.1 → 4.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/adapters/node/event-queue.d.ts +10 -0
- package/dist/adapters/node/event-queue.d.ts.map +1 -1
- package/dist/adapters/node/event-queue.js +17 -0
- package/dist/adapters/node/event-queue.js.map +1 -1
- package/dist/adapters/node/index.d.ts.map +1 -1
- package/dist/adapters/node/index.js +10 -0
- package/dist/adapters/node/index.js.map +1 -1
- package/dist/adapters/node/realtime.d.ts +24 -2
- package/dist/adapters/node/realtime.d.ts.map +1 -1
- package/dist/adapters/node/realtime.js +178 -1
- package/dist/adapters/node/realtime.js.map +1 -1
- package/dist/bin/serve.js +18 -1
- package/dist/bin/serve.js.map +1 -1
- package/dist/db/migrations/0016_fleet_nodes.sql +189 -0
- package/dist/db/migrations/0017_spawn_reservation_and_retry_state.sql +5 -0
- package/dist/db/migrations/0018_spawn_reserved_at.sql +2 -0
- package/dist/db/migrations/0019_fleet_mailbox.sql +71 -0
- package/dist/db/migrations/0020_workspace_retention.sql +3 -0
- package/dist/db/schema.d.ts +1771 -911
- package/dist/db/schema.d.ts.map +1 -1
- package/dist/db/schema.js +77 -7
- package/dist/db/schema.js.map +1 -1
- package/dist/engine/action.d.ts +84 -10
- package/dist/engine/action.d.ts.map +1 -1
- package/dist/engine/action.js +523 -85
- package/dist/engine/action.js.map +1 -1
- package/dist/engine/agent.js +2 -2
- package/dist/engine/agent.js.map +1 -1
- package/dist/engine/delivery.d.ts +41 -6
- package/dist/engine/delivery.d.ts.map +1 -1
- package/dist/engine/delivery.js +316 -25
- package/dist/engine/delivery.js.map +1 -1
- package/dist/engine/deliveryWire.d.ts +34 -0
- package/dist/engine/deliveryWire.d.ts.map +1 -0
- package/dist/engine/deliveryWire.js +70 -0
- package/dist/engine/deliveryWire.js.map +1 -0
- package/dist/engine/deliveryWrites.d.ts +47 -0
- package/dist/engine/deliveryWrites.d.ts.map +1 -1
- package/dist/engine/deliveryWrites.js +155 -5
- package/dist/engine/deliveryWrites.js.map +1 -1
- package/dist/engine/dm.d.ts +4 -4
- package/dist/engine/dm.d.ts.map +1 -1
- package/dist/engine/dm.js +17 -7
- package/dist/engine/dm.js.map +1 -1
- package/dist/engine/eventQueue.d.ts.map +1 -1
- package/dist/engine/eventQueue.js +5 -0
- package/dist/engine/eventQueue.js.map +1 -1
- package/dist/engine/eventSubscription.d.ts +6 -0
- package/dist/engine/eventSubscription.d.ts.map +1 -1
- package/dist/engine/eventSubscription.js +13 -0
- package/dist/engine/eventSubscription.js.map +1 -1
- package/dist/engine/groupDm.d.ts +4 -0
- package/dist/engine/groupDm.d.ts.map +1 -1
- package/dist/engine/groupDm.js +16 -4
- package/dist/engine/groupDm.js.map +1 -1
- package/dist/engine/invocationCompletion.d.ts +13 -0
- package/dist/engine/invocationCompletion.d.ts.map +1 -0
- package/dist/engine/invocationCompletion.js +65 -0
- package/dist/engine/invocationCompletion.js.map +1 -0
- package/dist/engine/mailboxConfig.d.ts +9 -0
- package/dist/engine/mailboxConfig.d.ts.map +1 -0
- package/dist/engine/mailboxConfig.js +16 -0
- package/dist/engine/mailboxConfig.js.map +1 -0
- package/dist/engine/message.d.ts +4 -0
- package/dist/engine/message.d.ts.map +1 -1
- package/dist/engine/message.js +16 -4
- package/dist/engine/message.js.map +1 -1
- package/dist/engine/node.d.ts +201 -0
- package/dist/engine/node.d.ts.map +1 -0
- package/dist/engine/node.js +533 -0
- package/dist/engine/node.js.map +1 -0
- package/dist/engine/placement.d.ts +26 -0
- package/dist/engine/placement.d.ts.map +1 -0
- package/dist/engine/placement.js +242 -0
- package/dist/engine/placement.js.map +1 -0
- package/dist/engine/receipt.d.ts.map +1 -1
- package/dist/engine/receipt.js +5 -6
- package/dist/engine/receipt.js.map +1 -1
- package/dist/engine/retention.d.ts +65 -0
- package/dist/engine/retention.d.ts.map +1 -0
- package/dist/engine/retention.js +173 -0
- package/dist/engine/retention.js.map +1 -0
- package/dist/engine/snowflake.d.ts +11 -0
- package/dist/engine/snowflake.d.ts.map +1 -1
- package/dist/engine/snowflake.js +14 -0
- package/dist/engine/snowflake.js.map +1 -1
- package/dist/engine/thread.d.ts +4 -0
- package/dist/engine/thread.d.ts.map +1 -1
- package/dist/engine/thread.js +16 -4
- package/dist/engine/thread.js.map +1 -1
- package/dist/engine/trigger.d.ts +79 -0
- package/dist/engine/trigger.d.ts.map +1 -0
- package/dist/engine/trigger.js +151 -0
- package/dist/engine/trigger.js.map +1 -0
- package/dist/engine.d.ts.map +1 -1
- package/dist/engine.js +61 -2
- package/dist/engine.js.map +1 -1
- package/dist/entrypoints/node.d.ts.map +1 -1
- package/dist/entrypoints/node.js +36 -2
- package/dist/entrypoints/node.js.map +1 -1
- package/dist/env.d.ts +6 -0
- package/dist/env.d.ts.map +1 -1
- package/dist/index.d.ts +4 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +6 -1
- package/dist/index.js.map +1 -1
- package/dist/lib/fleetNodes.d.ts +9 -0
- package/dist/lib/fleetNodes.d.ts.map +1 -0
- package/dist/lib/fleetNodes.js +66 -0
- package/dist/lib/fleetNodes.js.map +1 -0
- package/dist/lib/origin.d.ts.map +1 -1
- package/dist/lib/origin.js +0 -5
- package/dist/lib/origin.js.map +1 -1
- package/dist/lib/serverTelemetry.d.ts.map +1 -1
- package/dist/lib/serverTelemetry.js +0 -1
- package/dist/lib/serverTelemetry.js.map +1 -1
- package/dist/middleware/fleetNodes.d.ts +10 -0
- package/dist/middleware/fleetNodes.d.ts.map +1 -0
- package/dist/middleware/fleetNodes.js +18 -0
- package/dist/middleware/fleetNodes.js.map +1 -0
- package/dist/ports/index.d.ts +24 -2
- package/dist/ports/index.d.ts.map +1 -1
- package/dist/ports/realtime.d.ts +34 -2
- package/dist/ports/realtime.d.ts.map +1 -1
- package/dist/ports/realtime.js +1 -1
- package/dist/routes/action.d.ts.map +1 -1
- package/dist/routes/action.js +21 -21
- package/dist/routes/action.js.map +1 -1
- package/dist/routes/delivery.d.ts.map +1 -1
- package/dist/routes/delivery.js +7 -2
- package/dist/routes/delivery.js.map +1 -1
- package/dist/routes/deliveryRouting.d.ts +10 -0
- package/dist/routes/deliveryRouting.d.ts.map +1 -0
- package/dist/routes/deliveryRouting.js +91 -0
- package/dist/routes/deliveryRouting.js.map +1 -0
- package/dist/routes/dm.d.ts.map +1 -1
- package/dist/routes/dm.js +14 -13
- package/dist/routes/dm.js.map +1 -1
- package/dist/routes/groupDm.d.ts.map +1 -1
- package/dist/routes/groupDm.js +14 -15
- package/dist/routes/groupDm.js.map +1 -1
- package/dist/routes/inbox.d.ts.map +1 -1
- package/dist/routes/inbox.js +7 -0
- package/dist/routes/inbox.js.map +1 -1
- package/dist/routes/message.d.ts.map +1 -1
- package/dist/routes/message.js +45 -23
- package/dist/routes/message.js.map +1 -1
- package/dist/routes/node.d.ts +4 -0
- package/dist/routes/node.d.ts.map +1 -0
- package/dist/routes/node.js +57 -0
- package/dist/routes/node.js.map +1 -0
- package/dist/routes/thread.d.ts.map +1 -1
- package/dist/routes/thread.js +15 -16
- package/dist/routes/thread.js.map +1 -1
- package/dist/routes/trigger.d.ts +4 -0
- package/dist/routes/trigger.d.ts.map +1 -0
- package/dist/routes/trigger.js +83 -0
- package/dist/routes/trigger.js.map +1 -0
- package/dist/routes/webhookOutbox.d.ts +13 -6
- package/dist/routes/webhookOutbox.d.ts.map +1 -1
- package/dist/routes/webhookOutbox.js +29 -6
- package/dist/routes/webhookOutbox.js.map +1 -1
- package/dist/routes/workspace.d.ts.map +1 -1
- package/dist/routes/workspace.js +90 -0
- package/dist/routes/workspace.js.map +1 -1
- package/package.json +3 -3
|
@@ -0,0 +1,533 @@
|
|
|
1
|
+
import { and, eq, inArray, ne, or, sql } from 'drizzle-orm';
|
|
2
|
+
import { parseFleetBrokerToRelaycastMessage } from '@relaycast/types';
|
|
3
|
+
import { actionInvocations, actions, agents, channelMembers, channels, nodes } from '../db/schema.js';
|
|
4
|
+
import { randomHex, sha256Hex } from '../lib/crypto.js';
|
|
5
|
+
import { codedError } from '../lib/httpError.js';
|
|
6
|
+
import { runAtomic } from '../ports/database.js';
|
|
7
|
+
import { generateId } from './snowflake.js';
|
|
8
|
+
import { isNodeLive, nodeHasCapability } from './placement.js';
|
|
9
|
+
import { completeNodeInvocation, rescheduleInvocationsForLostNode, rescheduleNodeInvocation } from './action.js';
|
|
10
|
+
import { emitInvocationCompletionEffects } from './invocationCompletion.js';
|
|
11
|
+
import { ackDeliveriesUpToSeq, deliverPendingToNode } from './delivery.js';
|
|
12
|
+
function capabilityName(capability) {
|
|
13
|
+
if (typeof capability === 'string')
|
|
14
|
+
return capability;
|
|
15
|
+
return capability?.name ?? null;
|
|
16
|
+
}
|
|
17
|
+
function normalizeCapabilities(capabilities) {
|
|
18
|
+
return capabilities.map((capability) => (typeof capability === 'string' ? { name: capability } : capability));
|
|
19
|
+
}
|
|
20
|
+
function requestId(message) {
|
|
21
|
+
return message.id ?? generateId();
|
|
22
|
+
}
|
|
23
|
+
function publicNode(row) {
|
|
24
|
+
const live = isNodeLive(row);
|
|
25
|
+
return {
|
|
26
|
+
id: row.id,
|
|
27
|
+
name: row.name,
|
|
28
|
+
capabilities: row.capabilities,
|
|
29
|
+
tags: row.tags,
|
|
30
|
+
version: row.version,
|
|
31
|
+
status: live ? 'online' : 'offline',
|
|
32
|
+
live,
|
|
33
|
+
handlers_live: live && row.handlersLive,
|
|
34
|
+
load: row.load,
|
|
35
|
+
active_agents: row.activeAgents,
|
|
36
|
+
max_agents: row.maxAgents,
|
|
37
|
+
last_heartbeat_at: row.lastHeartbeatAt?.toISOString() ?? null,
|
|
38
|
+
created_at: row.createdAt.toISOString(),
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
async function ensureCapabilityActions(db, workspaceId, nodeId, capabilities) {
|
|
42
|
+
for (const capability of capabilities) {
|
|
43
|
+
const name = capabilityName(capability);
|
|
44
|
+
if (!name || name.startsWith('spawn:'))
|
|
45
|
+
continue;
|
|
46
|
+
const [existing] = await db
|
|
47
|
+
.select()
|
|
48
|
+
.from(actions)
|
|
49
|
+
.where(and(eq(actions.workspaceId, workspaceId), eq(actions.name, name)));
|
|
50
|
+
if (!existing) {
|
|
51
|
+
await db.insert(actions).values({
|
|
52
|
+
id: `act_${generateId()}`,
|
|
53
|
+
workspaceId,
|
|
54
|
+
name,
|
|
55
|
+
description: `Node handler ${name}`,
|
|
56
|
+
handlerAgentId: null,
|
|
57
|
+
handlerNodeId: nodeId,
|
|
58
|
+
inputSchema: {},
|
|
59
|
+
outputSchema: {},
|
|
60
|
+
availableTo: null,
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
else if (!existing.handlerAgentId && (!existing.handlerNodeId || existing.handlerNodeId === nodeId)) {
|
|
64
|
+
await db
|
|
65
|
+
.update(actions)
|
|
66
|
+
.set({ handlerNodeId: nodeId, isActive: true })
|
|
67
|
+
.where(eq(actions.id, existing.id));
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
export async function createNodeToken(db, workspaceId, data) {
|
|
72
|
+
const token = `nt_live_${randomHex(24)}`;
|
|
73
|
+
const tokenHash = await sha256Hex(token);
|
|
74
|
+
const existing = await getNodeByName(db, workspaceId, data.name);
|
|
75
|
+
const now = new Date();
|
|
76
|
+
if (existing) {
|
|
77
|
+
const [updated] = await db
|
|
78
|
+
.update(nodes)
|
|
79
|
+
.set({
|
|
80
|
+
tokenHash,
|
|
81
|
+
capabilities: normalizeCapabilities(data.capabilities ?? existing.capabilities),
|
|
82
|
+
maxAgents: data.max_agents ?? existing.maxAgents,
|
|
83
|
+
tags: data.tags ?? existing.tags,
|
|
84
|
+
version: data.version ?? existing.version,
|
|
85
|
+
status: 'offline',
|
|
86
|
+
handlersLive: false,
|
|
87
|
+
})
|
|
88
|
+
.where(eq(nodes.id, existing.id))
|
|
89
|
+
.returning();
|
|
90
|
+
return { ...publicNode(updated), token };
|
|
91
|
+
}
|
|
92
|
+
const [created] = await db
|
|
93
|
+
.insert(nodes)
|
|
94
|
+
.values({
|
|
95
|
+
id: data.node_id ?? `node_${generateId()}`,
|
|
96
|
+
workspaceId,
|
|
97
|
+
name: data.name,
|
|
98
|
+
tokenHash,
|
|
99
|
+
capabilities: normalizeCapabilities(data.capabilities ?? []),
|
|
100
|
+
maxAgents: data.max_agents ?? 0,
|
|
101
|
+
tags: data.tags ?? [],
|
|
102
|
+
version: data.version ?? 'unknown',
|
|
103
|
+
status: 'offline',
|
|
104
|
+
handlersLive: false,
|
|
105
|
+
load: 0,
|
|
106
|
+
activeAgents: 0,
|
|
107
|
+
createdAt: now,
|
|
108
|
+
})
|
|
109
|
+
.returning();
|
|
110
|
+
return { ...publicNode(created), token };
|
|
111
|
+
}
|
|
112
|
+
export async function getNodeByTokenHash(db, tokenHash) {
|
|
113
|
+
const [node] = await db.select().from(nodes).where(eq(nodes.tokenHash, tokenHash));
|
|
114
|
+
return node ?? null;
|
|
115
|
+
}
|
|
116
|
+
export async function getNodeByName(db, workspaceId, name) {
|
|
117
|
+
const normalized = name.startsWith('#') ? name.slice(1) : name;
|
|
118
|
+
const [node] = await db
|
|
119
|
+
.select()
|
|
120
|
+
.from(nodes)
|
|
121
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.name, normalized)));
|
|
122
|
+
return node ?? null;
|
|
123
|
+
}
|
|
124
|
+
export async function registerNode(db, workspaceId, authenticatedNodeId, message) {
|
|
125
|
+
if (message.node_id !== authenticatedNodeId) {
|
|
126
|
+
throw codedError('node_id does not match the authenticated node token', 'node_id_mismatch', 403);
|
|
127
|
+
}
|
|
128
|
+
const now = new Date();
|
|
129
|
+
const [existingByName] = await db
|
|
130
|
+
.select()
|
|
131
|
+
.from(nodes)
|
|
132
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.name, message.name)));
|
|
133
|
+
if (existingByName && existingByName.id !== authenticatedNodeId) {
|
|
134
|
+
throw codedError(`Node name "${message.name}" is already enrolled`, 'node_name_conflict', 409);
|
|
135
|
+
}
|
|
136
|
+
const [updated] = await db
|
|
137
|
+
.update(nodes)
|
|
138
|
+
.set({
|
|
139
|
+
name: message.name,
|
|
140
|
+
capabilities: message.capabilities,
|
|
141
|
+
maxAgents: message.max_agents,
|
|
142
|
+
tags: message.tags,
|
|
143
|
+
version: message.version,
|
|
144
|
+
status: 'online',
|
|
145
|
+
handlersLive: message.capabilities.length > 0,
|
|
146
|
+
lastHeartbeatAt: now,
|
|
147
|
+
})
|
|
148
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.id, authenticatedNodeId)))
|
|
149
|
+
.returning();
|
|
150
|
+
if (!updated) {
|
|
151
|
+
throw codedError('Node token is not enrolled in this workspace', 'node_not_found', 404);
|
|
152
|
+
}
|
|
153
|
+
await ensureCapabilityActions(db, workspaceId, updated.id, message.capabilities);
|
|
154
|
+
return publicNode(updated);
|
|
155
|
+
}
|
|
156
|
+
export async function heartbeatNode(db, workspaceId, nodeId, message) {
|
|
157
|
+
const [updated] = await db
|
|
158
|
+
.update(nodes)
|
|
159
|
+
.set({
|
|
160
|
+
status: 'online',
|
|
161
|
+
load: message.load,
|
|
162
|
+
activeAgents: message.active_agents,
|
|
163
|
+
handlersLive: message.handlers_live,
|
|
164
|
+
lastHeartbeatAt: new Date(),
|
|
165
|
+
})
|
|
166
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.id, nodeId)))
|
|
167
|
+
.returning();
|
|
168
|
+
return updated ? publicNode(updated) : null;
|
|
169
|
+
}
|
|
170
|
+
export async function markNodeOffline(db, registry, workspaceId, nodeId) {
|
|
171
|
+
await db
|
|
172
|
+
.update(nodes)
|
|
173
|
+
.set({
|
|
174
|
+
status: 'offline',
|
|
175
|
+
handlersLive: false,
|
|
176
|
+
load: 0,
|
|
177
|
+
activeAgents: 0,
|
|
178
|
+
lastHeartbeatAt: new Date(),
|
|
179
|
+
})
|
|
180
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.id, nodeId)));
|
|
181
|
+
await db
|
|
182
|
+
.update(agents)
|
|
183
|
+
.set({ status: 'offline', lastSeen: new Date() })
|
|
184
|
+
.where(and(eq(agents.workspaceId, workspaceId), eq(agents.locationType, 'via_node'), eq(agents.locationNodeId, nodeId)));
|
|
185
|
+
await rescheduleInvocationsForLostNode(db, registry, workspaceId, nodeId);
|
|
186
|
+
}
|
|
187
|
+
export async function deregisterNode(db, registry, workspaceId, nodeId) {
|
|
188
|
+
await markNodeOffline(db, registry, workspaceId, nodeId);
|
|
189
|
+
}
|
|
190
|
+
export async function sweepOfflineNodes(db, registry) {
|
|
191
|
+
const rows = await db.select().from(nodes).where(eq(nodes.status, 'online'));
|
|
192
|
+
const stale = rows.filter((node) => !isNodeLive(node));
|
|
193
|
+
for (const node of stale) {
|
|
194
|
+
await markNodeOffline(db, registry, node.workspaceId, node.id);
|
|
195
|
+
}
|
|
196
|
+
return stale.length;
|
|
197
|
+
}
|
|
198
|
+
async function autoJoinGeneral(db, workspaceId, agentId) {
|
|
199
|
+
const [generalChannel] = await db
|
|
200
|
+
.select()
|
|
201
|
+
.from(channels)
|
|
202
|
+
.where(and(eq(channels.workspaceId, workspaceId), eq(channels.name, 'general')));
|
|
203
|
+
if (generalChannel) {
|
|
204
|
+
await db.insert(channelMembers).values({
|
|
205
|
+
channelId: generalChannel.id,
|
|
206
|
+
agentId,
|
|
207
|
+
role: 'member',
|
|
208
|
+
}).onConflictDoNothing();
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
export async function registerAgentViaNode(db, workspaceId, nodeId, message) {
|
|
212
|
+
return runAtomic(db, async (tx) => {
|
|
213
|
+
const token = `at_live_${randomHex(16)}`;
|
|
214
|
+
const tokenHash = await sha256Hex(token);
|
|
215
|
+
const now = new Date().toISOString();
|
|
216
|
+
const fleetMetadata = sql `json_object(
|
|
217
|
+
'node_id', ${nodeId},
|
|
218
|
+
'invocation_id', ${message.invocation_id ?? null},
|
|
219
|
+
'registered_at', ${now}
|
|
220
|
+
)`;
|
|
221
|
+
const [result] = await tx
|
|
222
|
+
.insert(agents)
|
|
223
|
+
.values({
|
|
224
|
+
id: generateId(),
|
|
225
|
+
workspaceId,
|
|
226
|
+
name: message.name,
|
|
227
|
+
handle: `@${message.name}`,
|
|
228
|
+
type: 'agent',
|
|
229
|
+
tokenHash,
|
|
230
|
+
status: 'active',
|
|
231
|
+
persona: null,
|
|
232
|
+
metadata: { fleet: { node_id: nodeId, invocation_id: message.invocation_id ?? null, registered_at: now } },
|
|
233
|
+
locationType: 'via_node',
|
|
234
|
+
locationNodeId: nodeId,
|
|
235
|
+
originNodeId: nodeId,
|
|
236
|
+
resumable: message.resumable ?? false,
|
|
237
|
+
sessionRef: message.session_ref ?? null,
|
|
238
|
+
})
|
|
239
|
+
.onConflictDoUpdate({
|
|
240
|
+
target: [agents.workspaceId, agents.name],
|
|
241
|
+
set: {
|
|
242
|
+
tokenHash,
|
|
243
|
+
status: 'active',
|
|
244
|
+
lastSeen: new Date(),
|
|
245
|
+
metadata: sql `json_patch(COALESCE(${agents.metadata}, '{}'), ${fleetMetadata})`,
|
|
246
|
+
locationType: 'via_node',
|
|
247
|
+
locationNodeId: nodeId,
|
|
248
|
+
originNodeId: sql `COALESCE(${agents.originNodeId}, ${nodeId})`,
|
|
249
|
+
resumable: message.resumable ?? false,
|
|
250
|
+
sessionRef: message.session_ref ?? null,
|
|
251
|
+
},
|
|
252
|
+
setWhere: or(ne(agents.status, 'active'), and(eq(agents.locationType, 'via_node'), eq(agents.locationNodeId, nodeId))),
|
|
253
|
+
})
|
|
254
|
+
.returning();
|
|
255
|
+
if (!result) {
|
|
256
|
+
throw codedError(`Agent "${message.name}" already has an active location`, 'agent_location_conflict', 409);
|
|
257
|
+
}
|
|
258
|
+
await autoJoinGeneral(tx, workspaceId, result.id);
|
|
259
|
+
return {
|
|
260
|
+
agent_id: result.id,
|
|
261
|
+
name: result.name,
|
|
262
|
+
token,
|
|
263
|
+
};
|
|
264
|
+
});
|
|
265
|
+
}
|
|
266
|
+
export async function deregisterAgentViaNode(db, workspaceId, nodeId, message) {
|
|
267
|
+
const conditions = [
|
|
268
|
+
eq(agents.workspaceId, workspaceId),
|
|
269
|
+
eq(agents.locationType, 'via_node'),
|
|
270
|
+
eq(agents.locationNodeId, nodeId),
|
|
271
|
+
];
|
|
272
|
+
if (message.agent_id) {
|
|
273
|
+
conditions.push(eq(agents.id, message.agent_id));
|
|
274
|
+
}
|
|
275
|
+
else if (message.name) {
|
|
276
|
+
conditions.push(eq(agents.name, message.name));
|
|
277
|
+
}
|
|
278
|
+
else {
|
|
279
|
+
return null;
|
|
280
|
+
}
|
|
281
|
+
const [updated] = await db
|
|
282
|
+
.update(agents)
|
|
283
|
+
.set({ status: 'offline', lastSeen: new Date() })
|
|
284
|
+
.where(and(...conditions))
|
|
285
|
+
.returning();
|
|
286
|
+
return updated ?? null;
|
|
287
|
+
}
|
|
288
|
+
export async function reconcileInventory(db, registry, workspaceId, nodeId, inventoryAgents, completionDeps) {
|
|
289
|
+
const names = new Set(inventoryAgents.map((agent) => agent.name));
|
|
290
|
+
const liveInvocationIds = new Set(inventoryAgents.flatMap((agent) => (agent.invocation_id ? [agent.invocation_id] : [])));
|
|
291
|
+
let completedInvocations = 0;
|
|
292
|
+
// Pre-validate every item against the current state BEFORE mutating anything,
|
|
293
|
+
// so a conflict on a later item can't leave earlier items partially
|
|
294
|
+
// reconciled (the control handler turns a throw into an error reply). Existing
|
|
295
|
+
// rows are cached for reuse in the apply pass below.
|
|
296
|
+
const existingByName = new Map();
|
|
297
|
+
for (const item of inventoryAgents) {
|
|
298
|
+
const [existing] = await db
|
|
299
|
+
.select()
|
|
300
|
+
.from(agents)
|
|
301
|
+
.where(and(eq(agents.workspaceId, workspaceId), eq(agents.name, item.name)));
|
|
302
|
+
if (!existing)
|
|
303
|
+
continue;
|
|
304
|
+
existingByName.set(item.name, existing);
|
|
305
|
+
if (existing.status !== 'active')
|
|
306
|
+
continue;
|
|
307
|
+
const [boundNode] = await db
|
|
308
|
+
.select()
|
|
309
|
+
.from(nodes)
|
|
310
|
+
.where(and(eq(nodes.workspaceId, workspaceId), eq(nodes.id, existing.locationNodeId ?? '')));
|
|
311
|
+
const boundNodeLive = !!boundNode && isNodeLive(boundNode);
|
|
312
|
+
const conflict = existing.locationType !== 'via_node' || !existing.locationNodeId || (existing.locationNodeId !== nodeId && boundNodeLive);
|
|
313
|
+
if (conflict) {
|
|
314
|
+
console.warn('[node.inventory] rejected active-name claim', {
|
|
315
|
+
workspace_id: workspaceId,
|
|
316
|
+
node_id: nodeId,
|
|
317
|
+
agent_id: existing.id,
|
|
318
|
+
agent_name: existing.name,
|
|
319
|
+
existing_location_type: existing.locationType,
|
|
320
|
+
existing_location_node_id: existing.locationNodeId,
|
|
321
|
+
});
|
|
322
|
+
throw codedError(`Agent "${item.name}" is already active on another live location`, 'agent_location_conflict', 409);
|
|
323
|
+
}
|
|
324
|
+
}
|
|
325
|
+
for (const item of inventoryAgents) {
|
|
326
|
+
const existing = existingByName.get(item.name);
|
|
327
|
+
if (existing) {
|
|
328
|
+
await db
|
|
329
|
+
.update(agents)
|
|
330
|
+
.set({
|
|
331
|
+
status: 'active',
|
|
332
|
+
lastSeen: new Date(),
|
|
333
|
+
locationType: 'via_node',
|
|
334
|
+
locationNodeId: nodeId,
|
|
335
|
+
originNodeId: existing.originNodeId ?? nodeId,
|
|
336
|
+
sessionRef: item.session_ref ?? existing.sessionRef,
|
|
337
|
+
})
|
|
338
|
+
.where(eq(agents.id, existing.id));
|
|
339
|
+
}
|
|
340
|
+
if (!item.invocation_id)
|
|
341
|
+
continue;
|
|
342
|
+
const completed = await completeNodeInvocation(db, registry, workspaceId, nodeId, item.invocation_id, {
|
|
343
|
+
output: {
|
|
344
|
+
agent_id: item.agent_id,
|
|
345
|
+
name: item.name,
|
|
346
|
+
invocation_id: item.invocation_id ?? null,
|
|
347
|
+
session_ref: item.session_ref ?? null,
|
|
348
|
+
},
|
|
349
|
+
});
|
|
350
|
+
if (completed) {
|
|
351
|
+
completedInvocations++;
|
|
352
|
+
if (completionDeps) {
|
|
353
|
+
await emitInvocationCompletionEffects(completionDeps, workspaceId, completed);
|
|
354
|
+
}
|
|
355
|
+
}
|
|
356
|
+
}
|
|
357
|
+
const nodeAgents = await db
|
|
358
|
+
.select({ id: agents.id, name: agents.name })
|
|
359
|
+
.from(agents)
|
|
360
|
+
.where(and(eq(agents.workspaceId, workspaceId), eq(agents.locationType, 'via_node'), eq(agents.locationNodeId, nodeId), eq(agents.status, 'active')));
|
|
361
|
+
const missing = nodeAgents.filter((agent) => !names.has(agent.name)).map((agent) => agent.id);
|
|
362
|
+
if (missing.length > 0) {
|
|
363
|
+
await db
|
|
364
|
+
.update(agents)
|
|
365
|
+
.set({ status: 'offline', lastSeen: new Date() })
|
|
366
|
+
.where(inArray(agents.id, missing));
|
|
367
|
+
}
|
|
368
|
+
const openInvocations = await db
|
|
369
|
+
.select({
|
|
370
|
+
id: actionInvocations.id,
|
|
371
|
+
workspaceId: actionInvocations.workspaceId,
|
|
372
|
+
actionName: actionInvocations.actionName,
|
|
373
|
+
callerId: actionInvocations.callerId,
|
|
374
|
+
input: actionInvocations.input,
|
|
375
|
+
status: actionInvocations.status,
|
|
376
|
+
dispatchedNodeId: actionInvocations.dispatchedNodeId,
|
|
377
|
+
attemptedNodeIds: actionInvocations.attemptedNodeIds,
|
|
378
|
+
dispatchAttempts: actionInvocations.dispatchAttempts,
|
|
379
|
+
})
|
|
380
|
+
.from(actionInvocations)
|
|
381
|
+
.where(and(eq(actionInvocations.workspaceId, workspaceId), eq(actionInvocations.dispatchedNodeId, nodeId), inArray(actionInvocations.status, ['pending', 'dispatched'])));
|
|
382
|
+
let rescheduledInvocations = 0;
|
|
383
|
+
for (const invocation of openInvocations) {
|
|
384
|
+
if (liveInvocationIds.has(invocation.id))
|
|
385
|
+
continue;
|
|
386
|
+
try {
|
|
387
|
+
if (await rescheduleNodeInvocation(db, registry, invocation)) {
|
|
388
|
+
rescheduledInvocations++;
|
|
389
|
+
}
|
|
390
|
+
}
|
|
391
|
+
catch {
|
|
392
|
+
await db
|
|
393
|
+
.update(actionInvocations)
|
|
394
|
+
.set({
|
|
395
|
+
status: 'pending',
|
|
396
|
+
dispatchedNodeId: null,
|
|
397
|
+
dispatchedAt: null,
|
|
398
|
+
retryAfterAt: new Date(Date.now() + 5_000),
|
|
399
|
+
})
|
|
400
|
+
.where(eq(actionInvocations.id, invocation.id));
|
|
401
|
+
}
|
|
402
|
+
}
|
|
403
|
+
return {
|
|
404
|
+
rebound_agents: inventoryAgents.length,
|
|
405
|
+
open_invocations: openInvocations.length,
|
|
406
|
+
completed_invocations: completedInvocations,
|
|
407
|
+
rescheduled_invocations: rescheduledInvocations,
|
|
408
|
+
};
|
|
409
|
+
}
|
|
410
|
+
export async function listNodes(db, workspaceId, filters = {}) {
|
|
411
|
+
const rows = await db.select().from(nodes).where(eq(nodes.workspaceId, workspaceId));
|
|
412
|
+
return rows
|
|
413
|
+
.filter((node) => !filters.name || node.name === filters.name)
|
|
414
|
+
.filter((node) => !filters.capability || nodeHasCapability(node, filters.capability))
|
|
415
|
+
.map(publicNode);
|
|
416
|
+
}
|
|
417
|
+
export async function getPublicNode(db, workspaceId, name) {
|
|
418
|
+
const node = await getNodeByName(db, workspaceId, name);
|
|
419
|
+
return node ? publicNode(node) : null;
|
|
420
|
+
}
|
|
421
|
+
function sendControl(socket, payload) {
|
|
422
|
+
if (!socket)
|
|
423
|
+
return;
|
|
424
|
+
try {
|
|
425
|
+
socket.send(JSON.stringify(payload));
|
|
426
|
+
}
|
|
427
|
+
catch {
|
|
428
|
+
// The close handler will clean up the socket.
|
|
429
|
+
}
|
|
430
|
+
}
|
|
431
|
+
export async function handleNodeControlMessage(args) {
|
|
432
|
+
let message;
|
|
433
|
+
try {
|
|
434
|
+
message = parseFleetBrokerToRelaycastMessage(JSON.parse(args.raw));
|
|
435
|
+
}
|
|
436
|
+
catch (err) {
|
|
437
|
+
sendControl(args.socket, {
|
|
438
|
+
v: 1,
|
|
439
|
+
id: generateId(),
|
|
440
|
+
type: 'error',
|
|
441
|
+
ok: false,
|
|
442
|
+
code: 'invalid_message',
|
|
443
|
+
message: err instanceof Error ? err.message : 'Invalid node control message',
|
|
444
|
+
});
|
|
445
|
+
return;
|
|
446
|
+
}
|
|
447
|
+
try {
|
|
448
|
+
switch (message.type) {
|
|
449
|
+
case 'node.register': {
|
|
450
|
+
const registered = await registerNode(args.db, args.workspaceId, args.nodeId, message);
|
|
451
|
+
sendControl(args.socket, {
|
|
452
|
+
v: 1,
|
|
453
|
+
id: requestId(message),
|
|
454
|
+
type: 'reply',
|
|
455
|
+
ok: true,
|
|
456
|
+
data: registered,
|
|
457
|
+
});
|
|
458
|
+
// Node is now marked online: flush any queued action.invoke frames so
|
|
459
|
+
// spawns queued while it was offline can reserve capacity and dispatch.
|
|
460
|
+
await args.registry.drainNode(args.workspaceId, args.nodeId);
|
|
461
|
+
return;
|
|
462
|
+
}
|
|
463
|
+
case 'node.heartbeat':
|
|
464
|
+
await heartbeatNode(args.db, args.workspaceId, args.nodeId, message);
|
|
465
|
+
// Heartbeat refreshes online/capacity state; re-drain as a backstop in
|
|
466
|
+
// case a queued spawn could not reserve capacity at register time.
|
|
467
|
+
await args.registry.drainNode(args.workspaceId, args.nodeId);
|
|
468
|
+
return;
|
|
469
|
+
case 'node.deregister':
|
|
470
|
+
await deregisterNode(args.db, args.registry, args.workspaceId, args.nodeId);
|
|
471
|
+
return;
|
|
472
|
+
case 'agent.register': {
|
|
473
|
+
const registered = await registerAgentViaNode(args.db, args.workspaceId, args.nodeId, message);
|
|
474
|
+
// The relay broker's node_control client awaits a `reply` frame keyed by
|
|
475
|
+
// the request id (it matches `pending_agent_registrations` by `reply.id`
|
|
476
|
+
// and parses `data` as {agent_id, token, name} with deny_unknown_fields).
|
|
477
|
+
// A bare `agent.registered` frame leaves the token-authority handshake
|
|
478
|
+
// hanging until the 30s timeout, so spawn never completes. Reply in the
|
|
479
|
+
// shape the shipped broker consumes; the broker already holds the
|
|
480
|
+
// invocation_id/session_ref it sent, so only the minted identity is echoed.
|
|
481
|
+
sendControl(args.socket, {
|
|
482
|
+
v: 1,
|
|
483
|
+
id: requestId(message),
|
|
484
|
+
type: 'reply',
|
|
485
|
+
ok: true,
|
|
486
|
+
data: { agent_id: registered.agent_id, token: registered.token, name: registered.name },
|
|
487
|
+
});
|
|
488
|
+
await deliverPendingToNode(args.db, args.registry, args.workspaceId, args.nodeId);
|
|
489
|
+
return;
|
|
490
|
+
}
|
|
491
|
+
case 'agent.deregister':
|
|
492
|
+
await deregisterAgentViaNode(args.db, args.workspaceId, args.nodeId, message);
|
|
493
|
+
return;
|
|
494
|
+
case 'inventory.sync': {
|
|
495
|
+
const result = await reconcileInventory(args.db, args.registry, args.workspaceId, args.nodeId, message.agents, args.completionDeps);
|
|
496
|
+
sendControl(args.socket, {
|
|
497
|
+
v: 1,
|
|
498
|
+
id: requestId(message),
|
|
499
|
+
type: 'reply',
|
|
500
|
+
ok: true,
|
|
501
|
+
data: result,
|
|
502
|
+
});
|
|
503
|
+
await deliverPendingToNode(args.db, args.registry, args.workspaceId, args.nodeId);
|
|
504
|
+
return;
|
|
505
|
+
}
|
|
506
|
+
case 'action.result': {
|
|
507
|
+
const completed = await completeNodeInvocation(args.db, args.registry, args.workspaceId, args.nodeId, message.invocation_id, {
|
|
508
|
+
...(Object.prototype.hasOwnProperty.call(message, 'output') ? { output: message.output } : {}),
|
|
509
|
+
...(message.error ? { error: message.error } : {}),
|
|
510
|
+
});
|
|
511
|
+
if (completed && args.completionDeps) {
|
|
512
|
+
await emitInvocationCompletionEffects(args.completionDeps, args.workspaceId, completed);
|
|
513
|
+
}
|
|
514
|
+
return;
|
|
515
|
+
}
|
|
516
|
+
case 'delivery.ack':
|
|
517
|
+
await ackDeliveriesUpToSeq(args.db, args.workspaceId, args.nodeId, message.agent, message.up_to_seq);
|
|
518
|
+
return;
|
|
519
|
+
}
|
|
520
|
+
}
|
|
521
|
+
catch (err) {
|
|
522
|
+
const error = err;
|
|
523
|
+
sendControl(args.socket, {
|
|
524
|
+
v: 1,
|
|
525
|
+
id: requestId(message),
|
|
526
|
+
type: 'error',
|
|
527
|
+
ok: false,
|
|
528
|
+
code: error.code ?? 'node_control_failed',
|
|
529
|
+
message: error.message,
|
|
530
|
+
});
|
|
531
|
+
}
|
|
532
|
+
}
|
|
533
|
+
//# sourceMappingURL=node.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node.js","sourceRoot":"","sources":["../../src/engine/node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAU5D,OAAO,EAAE,kCAAkC,EAAE,MAAM,kBAAkB,CAAC;AAEtE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACtG,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAEjD,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,gCAAgC,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACjH,OAAO,EAAE,+BAA+B,EAAE,MAAM,2BAA2B,CAAC;AAE5E,OAAO,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAW3E,SAAS,cAAc,CAAC,UAA6C;IACnE,IAAI,OAAO,UAAU,KAAK,QAAQ;QAAE,OAAO,UAAU,CAAC;IACtD,OAAO,UAAU,EAAE,IAAI,IAAI,IAAI,CAAC;AAClC,CAAC;AAED,SAAS,qBAAqB,CAAC,YAA8B;IAC3D,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CACtC,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,UAAU,CACnE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,OAAwB;IACzC,OAAO,OAAO,CAAC,EAAE,IAAI,UAAU,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,UAAU,CAAC,GAAY;IAC9B,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,YAAY,EAAE,GAAG,CAAC,YAAY;QAC9B,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QACnC,IAAI;QACJ,aAAa,EAAE,IAAI,IAAI,GAAG,CAAC,YAAY;QACvC,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,aAAa,EAAE,GAAG,CAAC,YAAY;QAC/B,UAAU,EAAE,GAAG,CAAC,SAAS;QACzB,iBAAiB,EAAE,GAAG,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,IAAI;QAC7D,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,WAAW,EAAE;KACxC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,EAAM,EAAE,WAAmB,EAAE,MAAc,EAAE,YAA8B;IAChH,KAAK,MAAM,UAAU,IAAI,YAAY,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,SAAS;QACjD,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,EAAE;aACxB,MAAM,EAAE;aACR,IAAI,CAAC,OAAO,CAAC;aACb,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5E,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;gBAC9B,EAAE,EAAE,OAAO,UAAU,EAAE,EAAE;gBACzB,WAAW;gBACX,IAAI;gBACJ,WAAW,EAAE,gBAAgB,IAAI,EAAE;gBACnC,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,MAAM;gBACrB,WAAW,EAAE,EAAE;gBACf,YAAY,EAAE,EAAE;gBAChB,WAAW,EAAE,IAAI;aAClB,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,IAAI,CAAC,CAAC,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,KAAK,MAAM,CAAC,EAAE,CAAC;YACtG,MAAM,EAAE;iBACL,MAAM,CAAC,OAAO,CAAC;iBACf,GAAG,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;iBAC9C,KAAK,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,EAAM,EACN,WAAmB,EACnB,IAOC;IAED,MAAM,KAAK,GAAG,WAAW,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;IACzC,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACjE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;aACvB,MAAM,CAAC,KAAK,CAAC;aACb,GAAG,CAAC;YACH,SAAS;YACT,YAAY,EAAE,qBAAqB,CAAC,IAAI,CAAC,YAAY,IAAI,QAAQ,CAAC,YAAY,CAAC;YAC/E,SAAS,EAAE,IAAI,CAAC,UAAU,IAAI,QAAQ,CAAC,SAAS;YAChD,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI;YAChC,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO;YACzC,MAAM,EAAE,SAAS;YACjB,YAAY,EAAE,KAAK;SACpB,CAAC;aACD,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;aAChC,SAAS,EAAE,CAAC;QACf,OAAO,EAAE,GAAG,UAAU,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,MAAM,CAAC;QACN,EAAE,EAAE,IAAI,CAAC,OAAO,IAAI,QAAQ,UAAU,EAAE,EAAE;QAC1C,WAAW;QACX,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,SAAS;QACT,YAAY,EAAE,qBAAqB,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC;QAC5D,SAAS,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;QAC/B,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;QACrB,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,SAAS;QAClC,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,KAAK;QACnB,IAAI,EAAE,CAAC;QACP,YAAY,EAAE,CAAC;QACf,SAAS,EAAE,GAAG;KACf,CAAC;SACD,SAAS,EAAE,CAAC;IACf,OAAO,EAAE,GAAG,UAAU,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,EAAM,EAAE,SAAiB;IAChE,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;IACnF,OAAO,IAAI,IAAI,IAAI,CAAC;AACtB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,EAAM,EAAE,WAAmB,EAAE,IAAY;IAC3E,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/D,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,EAAE;SACpB,MAAM,EAAE;SACR,IAAI,CAAC,KAAK,CAAC;SACX,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,IAAI,IAAI,IAAI,CAAC;AACtB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,EAAM,EACN,WAAmB,EACnB,mBAA2B,EAC3B,OAAiC;IAEjC,IAAI,OAAO,CAAC,OAAO,KAAK,mBAAmB,EAAE,CAAC;QAC5C,MAAM,UAAU,CAAC,qDAAqD,EAAE,kBAAkB,EAAE,GAAG,CAAC,CAAC;IACnG,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,EAAE;SAC9B,MAAM,EAAE;SACR,IAAI,CAAC,KAAK,CAAC;SACX,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAEhF,IAAI,cAAc,IAAI,cAAc,CAAC,EAAE,KAAK,mBAAmB,EAAE,CAAC;QAChE,MAAM,UAAU,CAAC,cAAc,OAAO,CAAC,IAAI,uBAAuB,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;IACjG,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,GAAG,CAAC;QACH,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,YAAY,EAAE,OAAO,CAAC,YAAY;QAClC,SAAS,EAAE,OAAO,CAAC,UAAU;QAC7B,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,MAAM,EAAE,QAAQ;QAChB,YAAY,EAAE,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC;QAC7C,eAAe,EAAE,GAAG;KACrB,CAAC;SACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;SACjF,SAAS,EAAE,CAAC;IAEf,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,UAAU,CAAC,8CAA8C,EAAE,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1F,CAAC;IAED,MAAM,uBAAuB,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;IACjF,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,EAAM,EACN,WAAmB,EACnB,MAAc,EACd,OAAkC;IAElC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,KAAK,CAAC;SACb,GAAG,CAAC;QACH,MAAM,EAAE,QAAQ;QAChB,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,YAAY,EAAE,OAAO,CAAC,aAAa;QACnC,YAAY,EAAE,OAAO,CAAC,aAAa;QACnC,eAAe,EAAE,IAAI,IAAI,EAAE;KAC5B,CAAC;SACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;SACpE,SAAS,EAAE,CAAC;IACf,OAAO,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC9C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,EAAM,EACN,QAAgC,EAChC,WAAmB,EACnB,MAAc;IAEd,MAAM,EAAE;SACL,MAAM,CAAC,KAAK,CAAC;SACb,GAAG,CAAC;QACH,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,KAAK;QACnB,IAAI,EAAE,CAAC;QACP,YAAY,EAAE,CAAC;QACf,eAAe,EAAE,IAAI,IAAI,EAAE;KAC5B,CAAC;SACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IAExE,MAAM,EAAE;SACL,MAAM,CAAC,MAAM,CAAC;SACd,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;SAChD,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC,EACnC,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,EACnC,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,CAClC,CAAC,CAAC;IAEL,MAAM,gCAAgC,CAAC,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;AAC5E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,EAAM,EACN,QAAgC,EAChC,WAAmB,EACnB,MAAc;IAEd,MAAM,eAAe,CAAC,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,EAAM,EAAE,QAAgC;IAC9E,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC7E,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;IACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,eAAe,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,KAAK,CAAC,MAAM,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,EAAM,EAAE,WAAmB,EAAE,OAAe;IACzE,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,EAAE;SAC9B,MAAM,EAAE;SACR,IAAI,CAAC,QAAQ,CAAC;SACd,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;IACnF,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC;YACrC,SAAS,EAAE,cAAc,CAAC,EAAE;YAC5B,OAAO;YACP,IAAI,EAAE,QAAQ;SACf,CAAC,CAAC,mBAAmB,EAAE,CAAC;IAC3B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,EAAM,EACN,WAAmB,EACnB,MAAc,EACd,OAAkC;IAElC,OAAO,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;QAChC,MAAM,KAAK,GAAG,WAAW,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,aAAa,GAAG,GAAG,CAAA;mBACV,MAAM;yBACA,OAAO,CAAC,aAAa,IAAI,IAAI;yBAC7B,GAAG;MACtB,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,EAAE;aACtB,MAAM,CAAC,MAAM,CAAC;aACd,MAAM,CAAC;YACN,EAAE,EAAE,UAAU,EAAE;YAChB,WAAW;YACX,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,MAAM,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE;YAC1B,IAAI,EAAE,OAAO;YACb,SAAS;YACT,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE;YAC1G,YAAY,EAAE,UAAU;YACxB,cAAc,EAAE,MAAM;YACtB,YAAY,EAAE,MAAM;YACpB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,KAAK;YACrC,UAAU,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI;SACxC,CAAC;aACD,kBAAkB,CAAC;YAClB,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC;YACzC,GAAG,EAAE;gBACH,SAAS;gBACT,MAAM,EAAE,QAAQ;gBAChB,QAAQ,EAAE,IAAI,IAAI,EAAE;gBACpB,QAAQ,EAAE,GAAG,CAAA,uBAAuB,MAAM,CAAC,QAAQ,YAAY,aAAa,GAAG;gBAC/E,YAAY,EAAE,UAAU;gBACxB,cAAc,EAAE,MAAM;gBACtB,YAAY,EAAE,GAAG,CAAA,YAAY,MAAM,CAAC,YAAY,KAAK,MAAM,GAAG;gBAC9D,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,KAAK;gBACrC,UAAU,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI;aACxC;YACD,QAAQ,EAAE,EAAE,CACV,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAC3B,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC,CAC5E;SACF,CAAC;aACD,SAAS,EAAE,CAAC;QAEf,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,UAAU,CAAC,UAAU,OAAO,CAAC,IAAI,kCAAkC,EAAE,yBAAyB,EAAE,GAAG,CAAC,CAAC;QAC7G,CAAC;QAED,MAAM,eAAe,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;QAClD,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,KAAK;SACN,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,EAAM,EACN,WAAmB,EACnB,MAAc,EACd,OAA6C;IAE7C,MAAM,UAAU,GAAG;QACjB,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC;QACnC,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC;QACnC,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC;KAClC,CAAC;IACF,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnD,CAAC;SAAM,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,EAAE;SACvB,MAAM,CAAC,MAAM,CAAC;SACd,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;SAChD,KAAK,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;SACzB,SAAS,EAAE,CAAC;IACf,OAAO,OAAO,IAAI,IAAI,CAAC;AACzB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,EAAM,EACN,QAAgC,EAChC,WAAmB,EACnB,MAAc,EACd,eAAsC,EACtC,cAAyC;IAEzC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CACnE,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAAE,CACjD,CAAC,CAAC,CAAC;IACJ,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,8EAA8E;IAC9E,oEAAoE;IACpE,+EAA+E;IAC/E,qDAAqD;IACrD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAsC,CAAC;IACrE,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC;QACnC,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,EAAE;aACxB,MAAM,EAAE;aACR,IAAI,CAAC,MAAM,CAAC;aACZ,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,QAAQ;YAAE,SAAS;QACxB,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACxC,IAAI,QAAQ,CAAC,MAAM,KAAK,QAAQ;YAAE,SAAS;QAC3C,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,EAAE;aACzB,MAAM,EAAE;aACR,IAAI,CAAC,KAAK,CAAC;aACX,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAClC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,QAAQ,CAAC,cAAc,IAAI,EAAE,CAAC,CAC5C,CAAC,CAAC;QACL,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,QAAQ,CAAC,YAAY,KAAK,UAAU,IAAI,CAAC,QAAQ,CAAC,cAAc,IAAI,CAAC,QAAQ,CAAC,cAAc,KAAK,MAAM,IAAI,aAAa,CAAC,CAAC;QAC3I,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE;gBAC1D,YAAY,EAAE,WAAW;gBACzB,OAAO,EAAE,MAAM;gBACf,QAAQ,EAAE,QAAQ,CAAC,EAAE;gBACrB,UAAU,EAAE,QAAQ,CAAC,IAAI;gBACzB,sBAAsB,EAAE,QAAQ,CAAC,YAAY;gBAC7C,yBAAyB,EAAE,QAAQ,CAAC,cAAc;aACnD,CAAC,CAAC;YACH,MAAM,UAAU,CAAC,UAAU,IAAI,CAAC,IAAI,8CAA8C,EAAE,yBAAyB,EAAE,GAAG,CAAC,CAAC;QACtH,CAAC;IACH,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,EAAE;iBACL,MAAM,CAAC,MAAM,CAAC;iBACd,GAAG,CAAC;gBACH,MAAM,EAAE,QAAQ;gBAChB,QAAQ,EAAE,IAAI,IAAI,EAAE;gBACpB,YAAY,EAAE,UAAU;gBACxB,cAAc,EAAE,MAAM;gBACtB,YAAY,EAAE,QAAQ,CAAC,YAAY,IAAI,MAAM;gBAC7C,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI,QAAQ,CAAC,UAAU;aACpD,CAAC;iBACD,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,SAAS;QAClC,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE;YACpG,MAAM,EAAE;gBACN,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;gBACzC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI;aACtC;SACF,CAAC,CAAC;QACH,IAAI,SAAS,EAAE,CAAC;YACd,oBAAoB,EAAE,CAAC;YACvB,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,+BAA+B,CAAC,cAAc,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,EAAE;SACxB,MAAM,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;SAC5C,IAAI,CAAC,MAAM,CAAC;SACZ,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC,EACnC,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,EACnC,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,EACjC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAC5B,CAAC,CAAC;IACL,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC9F,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,EAAE;aACL,MAAM,CAAC,MAAM,CAAC;aACd,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;aAChD,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,EAAE;SAC7B,MAAM,CAAC;QACN,EAAE,EAAE,iBAAiB,CAAC,EAAE;QACxB,WAAW,EAAE,iBAAiB,CAAC,WAAW;QAC1C,UAAU,EAAE,iBAAiB,CAAC,UAAU;QACxC,QAAQ,EAAE,iBAAiB,CAAC,QAAQ;QACpC,KAAK,EAAE,iBAAiB,CAAC,KAAK;QAC9B,MAAM,EAAE,iBAAiB,CAAC,MAAM;QAChC,gBAAgB,EAAE,iBAAiB,CAAC,gBAAgB;QACpD,gBAAgB,EAAE,iBAAiB,CAAC,gBAAgB;QACpD,gBAAgB,EAAE,iBAAiB,CAAC,gBAAgB;KACrD,CAAC;SACD,IAAI,CAAC,iBAAiB,CAAC;SACvB,KAAK,CAAC,GAAG,CACR,EAAE,CAAC,iBAAiB,CAAC,WAAW,EAAE,WAAW,CAAC,EAC9C,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,MAAM,CAAC,EAC9C,OAAO,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAC7D,CAAC,CAAC;IACL,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAC/B,KAAK,MAAM,UAAU,IAAI,eAAe,EAAE,CAAC;QACzC,IAAI,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YAAE,SAAS;QACnD,IAAI,CAAC;YACH,IAAI,MAAM,wBAAwB,CAAC,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,CAAC;gBAC7D,sBAAsB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,EAAE;iBACL,MAAM,CAAC,iBAAiB,CAAC;iBACzB,GAAG,CAAC;gBACH,MAAM,EAAE,SAAS;gBACjB,gBAAgB,EAAE,IAAI;gBACtB,YAAY,EAAE,IAAI;gBAClB,YAAY,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;aAC3C,CAAC;iBACD,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,OAAO;QACL,cAAc,EAAE,eAAe,CAAC,MAAM;QACtC,gBAAgB,EAAE,eAAe,CAAC,MAAM;QACxC,qBAAqB,EAAE,oBAAoB;QAC3C,uBAAuB,EAAE,sBAAsB;KAChD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,EAAM,EACN,WAAmB,EACnB,UAAkD,EAAE;IAEpD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;IACrF,OAAO,IAAI;SACR,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC;SAC7D,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,IAAI,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;SACpF,GAAG,CAAC,UAAU,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,EAAM,EAAE,WAAmB,EAAE,IAAY;IAC3E,MAAM,IAAI,GAAG,MAAM,aAAa,CAAC,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;IACxD,OAAO,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACxC,CAAC;AAED,SAAS,WAAW,CAAC,MAAkC,EAAE,OAAgC;IACvF,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,IAAI,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IACvC,CAAC;IAAC,MAAM,CAAC;QACP,8CAA8C;IAChD,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,IAQ9C;IACC,IAAI,OAAsC,CAAC;IAC3C,IAAI,CAAC;QACH,OAAO,GAAG,kCAAkC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACrE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;YACvB,CAAC,EAAE,CAAC;YACJ,EAAE,EAAE,UAAU,EAAE;YAChB,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,KAAK;YACT,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,8BAA8B;SAC7E,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBACvF,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;oBACvB,CAAC,EAAE,CAAC;oBACJ,EAAE,EAAE,SAAS,CAAC,OAAO,CAAC;oBACtB,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,UAAU;iBACjB,CAAC,CAAC;gBACH,sEAAsE;gBACtE,wEAAwE;gBACxE,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7D,OAAO;YACT,CAAC;YACD,KAAK,gBAAgB;gBACnB,MAAM,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBACrE,uEAAuE;gBACvE,mEAAmE;gBACnE,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7D,OAAO;YACT,KAAK,iBAAiB;gBACpB,MAAM,cAAc,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC5E,OAAO;YACT,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC/F,yEAAyE;gBACzE,yEAAyE;gBACzE,0EAA0E;gBAC1E,uEAAuE;gBACvE,wEAAwE;gBACxE,kEAAkE;gBAClE,4EAA4E;gBAC5E,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;oBACvB,CAAC,EAAE,CAAC;oBACJ,EAAE,EAAE,SAAS,CAAC,OAAO,CAAC;oBACtB,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE;iBACxF,CAAC,CAAC;gBACH,MAAM,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAClF,OAAO;YACT,CAAC;YACD,KAAK,kBAAkB;gBACrB,MAAM,sBAAsB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC9E,OAAO;YACT,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;gBACpI,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;oBACvB,CAAC,EAAE,CAAC;oBACJ,EAAE,EAAE,SAAS,CAAC,OAAO,CAAC;oBACtB,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,MAAM;iBACb,CAAC,CAAC;gBACH,MAAM,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAClF,OAAO;YACT,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,EAAE;oBAC3H,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC9F,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACnD,CAAC,CAAC;gBACH,IAAI,SAAS,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBACrC,MAAM,+BAA+B,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;gBAC1F,CAAC;gBACD,OAAO;YACT,CAAC;YACD,KAAK,cAAc;gBACjB,MAAM,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBACrG,OAAO;QACX,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,GAAgC,CAAC;QAC/C,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;YACvB,CAAC,EAAE,CAAC;YACJ,EAAE,EAAE,SAAS,CAAC,OAAO,CAAC;YACtB,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,KAAK;YACT,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,qBAAqB;YACzC,OAAO,EAAE,KAAK,CAAC,OAAO;SACvB,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import type { getDb } from '../db/index.js';
|
|
2
|
+
import { nodes } from '../db/schema.js';
|
|
3
|
+
type Db = ReturnType<typeof getDb>;
|
|
4
|
+
type NodeRow = typeof nodes.$inferSelect;
|
|
5
|
+
export declare const NODE_LIVENESS_TTL_MS = 45000;
|
|
6
|
+
export declare function isNodeLive(node: Pick<NodeRow, 'status' | 'lastHeartbeatAt'>, now?: number): boolean;
|
|
7
|
+
export declare function nodeHasCapability(node: Pick<NodeRow, 'capabilities'>, capability: string): boolean;
|
|
8
|
+
export declare function nodeHasCapacity(node: Pick<NodeRow, 'maxAgents' | 'activeAgents' | 'reservedAgents'>): boolean;
|
|
9
|
+
export interface PlacementRequest {
|
|
10
|
+
actionName: string;
|
|
11
|
+
input?: Record<string, unknown>;
|
|
12
|
+
callerId?: string | null;
|
|
13
|
+
preferredNodeId?: string | null;
|
|
14
|
+
excludeNodeIds?: string[];
|
|
15
|
+
}
|
|
16
|
+
export interface PlacementResult {
|
|
17
|
+
node: NodeRow;
|
|
18
|
+
capability: string;
|
|
19
|
+
queued: boolean;
|
|
20
|
+
}
|
|
21
|
+
export declare function reserveNodeCapacity(db: Db, workspaceId: string, nodeId: string): Promise<NodeRow | null>;
|
|
22
|
+
export declare function releaseNodeCapacity(db: Db, workspaceId: string, nodeId: string, count?: number): Promise<void>;
|
|
23
|
+
export declare function claimSpawnNode(db: Db, workspaceId: string, request: PlacementRequest): Promise<PlacementResult>;
|
|
24
|
+
export declare function chooseNodeForAction(db: Db, workspaceId: string, request: PlacementRequest): Promise<PlacementResult>;
|
|
25
|
+
export {};
|
|
26
|
+
//# sourceMappingURL=placement.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"placement.d.ts","sourceRoot":"","sources":["../../src/engine/placement.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAU,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAIhD,KAAK,EAAE,GAAG,UAAU,CAAC,OAAO,KAAK,CAAC,CAAC;AACnC,KAAK,OAAO,GAAG,OAAO,KAAK,CAAC,YAAY,CAAC;AAEzC,eAAO,MAAM,oBAAoB,QAAS,CAAC;AAE3C,wBAAgB,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,GAAG,iBAAiB,CAAC,EAAE,GAAG,SAAa,GAAG,OAAO,CAMvG;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAMlG;AAED,wBAAgB,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,cAAc,GAAG,gBAAgB,CAAC,GAAG,OAAO,CAE7G;AAeD,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,OAAO,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,wBAAsB,mBAAmB,CACvC,EAAE,EAAE,EAAE,EACN,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,CAezB;AAED,wBAAsB,mBAAmB,CACvC,EAAE,EAAE,EAAE,EACN,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,SAAI,GACR,OAAO,CAAC,IAAI,CAAC,CAQf;AAED,wBAAsB,cAAc,CAClC,EAAE,EAAE,EAAE,EACN,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CAsG1B;AAED,wBAAsB,mBAAmB,CACvC,EAAE,EAAE,EAAE,EACN,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CA8F1B"}
|