clawcenter 0.1.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/LICENSE +201 -0
- package/README.md +180 -0
- package/bin/clawcenter.js +2 -0
- package/dist/center/commands/system-commands.d.ts +15 -0
- package/dist/center/commands/system-commands.d.ts.map +1 -0
- package/dist/center/commands/system-commands.js +363 -0
- package/dist/center/commands/system-commands.js.map +1 -0
- package/dist/center/dispatcher.d.ts +32 -0
- package/dist/center/dispatcher.d.ts.map +1 -0
- package/dist/center/dispatcher.js +264 -0
- package/dist/center/dispatcher.js.map +1 -0
- package/dist/center/hub.d.ts +16 -0
- package/dist/center/hub.d.ts.map +1 -0
- package/dist/center/hub.js +214 -0
- package/dist/center/hub.js.map +1 -0
- package/dist/center/router/engine.d.ts +35 -0
- package/dist/center/router/engine.d.ts.map +1 -0
- package/dist/center/router/engine.js +106 -0
- package/dist/center/router/engine.js.map +1 -0
- package/dist/center/router/hashtag.d.ts +15 -0
- package/dist/center/router/hashtag.d.ts.map +1 -0
- package/dist/center/router/hashtag.js +21 -0
- package/dist/center/router/hashtag.js.map +1 -0
- package/dist/center/wechat/api.d.ts +131 -0
- package/dist/center/wechat/api.d.ts.map +1 -0
- package/dist/center/wechat/api.js +205 -0
- package/dist/center/wechat/api.js.map +1 -0
- package/dist/center/wechat/cdn.d.ts +13 -0
- package/dist/center/wechat/cdn.d.ts.map +1 -0
- package/dist/center/wechat/cdn.js +100 -0
- package/dist/center/wechat/cdn.js.map +1 -0
- package/dist/center/wechat/connector.d.ts +51 -0
- package/dist/center/wechat/connector.d.ts.map +1 -0
- package/dist/center/wechat/connector.js +253 -0
- package/dist/center/wechat/connector.js.map +1 -0
- package/dist/center/wechat/login.d.ts +13 -0
- package/dist/center/wechat/login.d.ts.map +1 -0
- package/dist/center/wechat/login.js +81 -0
- package/dist/center/wechat/login.js.map +1 -0
- package/dist/core/agents/adapter.d.ts +43 -0
- package/dist/core/agents/adapter.d.ts.map +1 -0
- package/dist/core/agents/adapter.js +2 -0
- package/dist/core/agents/adapter.js.map +1 -0
- package/dist/core/agents/claude-code.d.ts +19 -0
- package/dist/core/agents/claude-code.d.ts.map +1 -0
- package/dist/core/agents/claude-code.js +138 -0
- package/dist/core/agents/claude-code.js.map +1 -0
- package/dist/core/agents/claude-sdk.d.ts +17 -0
- package/dist/core/agents/claude-sdk.d.ts.map +1 -0
- package/dist/core/agents/claude-sdk.js +85 -0
- package/dist/core/agents/claude-sdk.js.map +1 -0
- package/dist/core/agents/codebuddy.d.ts +19 -0
- package/dist/core/agents/codebuddy.d.ts.map +1 -0
- package/dist/core/agents/codebuddy.js +352 -0
- package/dist/core/agents/codebuddy.js.map +1 -0
- package/dist/core/agents/codex.d.ts +18 -0
- package/dist/core/agents/codex.d.ts.map +1 -0
- package/dist/core/agents/codex.js +300 -0
- package/dist/core/agents/codex.js.map +1 -0
- package/dist/core/agents/cursor-agent.d.ts +19 -0
- package/dist/core/agents/cursor-agent.d.ts.map +1 -0
- package/dist/core/agents/cursor-agent.js +253 -0
- package/dist/core/agents/cursor-agent.js.map +1 -0
- package/dist/core/agents/http-agent.d.ts +17 -0
- package/dist/core/agents/http-agent.d.ts.map +1 -0
- package/dist/core/agents/http-agent.js +91 -0
- package/dist/core/agents/http-agent.js.map +1 -0
- package/dist/core/agents/manager.d.ts +45 -0
- package/dist/core/agents/manager.d.ts.map +1 -0
- package/dist/core/agents/manager.js +245 -0
- package/dist/core/agents/manager.js.map +1 -0
- package/dist/core/agents/openclaw.d.ts +19 -0
- package/dist/core/agents/openclaw.d.ts.map +1 -0
- package/dist/core/agents/openclaw.js +173 -0
- package/dist/core/agents/openclaw.js.map +1 -0
- package/dist/core/agents/opencode.d.ts +19 -0
- package/dist/core/agents/opencode.d.ts.map +1 -0
- package/dist/core/agents/opencode.js +175 -0
- package/dist/core/agents/opencode.js.map +1 -0
- package/dist/core/agents/worker-agent.d.ts +24 -0
- package/dist/core/agents/worker-agent.d.ts.map +1 -0
- package/dist/core/agents/worker-agent.js +40 -0
- package/dist/core/agents/worker-agent.js.map +1 -0
- package/dist/core/api/routes.d.ts +6 -0
- package/dist/core/api/routes.d.ts.map +1 -0
- package/dist/core/api/routes.js +238 -0
- package/dist/core/api/routes.js.map +1 -0
- package/dist/core/db/schema.d.ts +3 -0
- package/dist/core/db/schema.d.ts.map +1 -0
- package/dist/core/db/schema.js +97 -0
- package/dist/core/db/schema.js.map +1 -0
- package/dist/core/db/store.d.ts +167 -0
- package/dist/core/db/store.d.ts.map +1 -0
- package/dist/core/db/store.js +469 -0
- package/dist/core/db/store.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +70 -0
- package/dist/index.js.map +1 -0
- package/dist/logger.d.ts +6 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +22 -0
- package/dist/logger.js.map +1 -0
- package/dist/server.d.ts +28 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +158 -0
- package/dist/server.js.map +1 -0
- package/dist/tui/App.d.ts +78 -0
- package/dist/tui/App.d.ts.map +1 -0
- package/dist/tui/App.js +106 -0
- package/dist/tui/App.js.map +1 -0
- package/dist/tui/index.d.ts +7 -0
- package/dist/tui/index.d.ts.map +1 -0
- package/dist/tui/index.js +12 -0
- package/dist/tui/index.js.map +1 -0
- package/dist/worker/client.d.ts +28 -0
- package/dist/worker/client.d.ts.map +1 -0
- package/dist/worker/client.js +142 -0
- package/dist/worker/client.js.map +1 -0
- package/package.json +74 -0
- package/web/dist/assets/index-Bvq27OV5.js +60 -0
- package/web/dist/index.html +63 -0
|
@@ -0,0 +1,264 @@
|
|
|
1
|
+
import { EventEmitter } from "node:events";
|
|
2
|
+
import { Router, PermissionError } from "./router/engine.js";
|
|
3
|
+
import { isSystemCommand, handleSystemCommand } from "./commands/system-commands.js";
|
|
4
|
+
import { WechatConnector } from "./wechat/connector.js";
|
|
5
|
+
export class Dispatcher extends EventEmitter {
|
|
6
|
+
store;
|
|
7
|
+
agentManager;
|
|
8
|
+
router;
|
|
9
|
+
connectors = new Map();
|
|
10
|
+
constructor(store, agentManager) {
|
|
11
|
+
super();
|
|
12
|
+
this.store = store;
|
|
13
|
+
this.agentManager = agentManager;
|
|
14
|
+
const prefixFormat = store.getSetting("reply_prefix_format") ?? "[{displayName}]";
|
|
15
|
+
this.router = new Router({ store, agentManager, replyPrefixFormat: prefixFormat });
|
|
16
|
+
}
|
|
17
|
+
getConnector(wechatId) {
|
|
18
|
+
return this.connectors.get(wechatId);
|
|
19
|
+
}
|
|
20
|
+
listConnectors() {
|
|
21
|
+
return Array.from(this.connectors.values());
|
|
22
|
+
}
|
|
23
|
+
async startConnector(wechatId) {
|
|
24
|
+
if (this.connectors.has(wechatId)) {
|
|
25
|
+
throw new Error(`Connector for "${wechatId}" already running`);
|
|
26
|
+
}
|
|
27
|
+
const connector = new WechatConnector(this.store, wechatId);
|
|
28
|
+
this.connectors.set(wechatId, connector);
|
|
29
|
+
connector.on("message", (msg) => {
|
|
30
|
+
this.handleMessage(connector, msg).catch((err) => {
|
|
31
|
+
console.error(`[Dispatcher] Message handling error:`, err);
|
|
32
|
+
this.emitLog({
|
|
33
|
+
type: "error",
|
|
34
|
+
wechatId: msg.wechatAccountId,
|
|
35
|
+
userId: msg.fromUserId,
|
|
36
|
+
text: `Error: ${err.message}`,
|
|
37
|
+
});
|
|
38
|
+
});
|
|
39
|
+
});
|
|
40
|
+
connector.on("error", (err) => {
|
|
41
|
+
this.emitLog({ type: "error", wechatId, text: `Connector error: ${err.message}` });
|
|
42
|
+
});
|
|
43
|
+
connector.on("session_expired", () => {
|
|
44
|
+
this.emitLog({ type: "error", wechatId, text: "Session expired, paused for 1 hour" });
|
|
45
|
+
});
|
|
46
|
+
await connector.start();
|
|
47
|
+
this.emitLog({ type: "info", wechatId, text: "Connector started" });
|
|
48
|
+
}
|
|
49
|
+
async stopConnector(wechatId) {
|
|
50
|
+
const connector = this.connectors.get(wechatId);
|
|
51
|
+
if (!connector)
|
|
52
|
+
return;
|
|
53
|
+
connector.stop();
|
|
54
|
+
this.connectors.delete(wechatId);
|
|
55
|
+
this.emitLog({ type: "info", wechatId, text: "Connector stopped" });
|
|
56
|
+
}
|
|
57
|
+
async stopAll() {
|
|
58
|
+
for (const [id, connector] of this.connectors) {
|
|
59
|
+
connector.stop();
|
|
60
|
+
this.emitLog({ type: "info", wechatId: id, text: "Connector stopped" });
|
|
61
|
+
}
|
|
62
|
+
this.connectors.clear();
|
|
63
|
+
}
|
|
64
|
+
/** Start all connected WeChat accounts */
|
|
65
|
+
async startAllConnectors() {
|
|
66
|
+
const accounts = this.store.listWechatAccounts();
|
|
67
|
+
for (const account of accounts) {
|
|
68
|
+
if (account.token) {
|
|
69
|
+
try {
|
|
70
|
+
await this.startConnector(account.id);
|
|
71
|
+
}
|
|
72
|
+
catch (err) {
|
|
73
|
+
console.error(`[Dispatcher] Failed to start connector ${account.id}:`, err.message);
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
async handleMessage(connector, msg) {
|
|
79
|
+
this.emitLog({
|
|
80
|
+
type: "inbound",
|
|
81
|
+
wechatId: msg.wechatAccountId,
|
|
82
|
+
userId: msg.fromUserId,
|
|
83
|
+
text: msg.text,
|
|
84
|
+
});
|
|
85
|
+
// Record inbound message
|
|
86
|
+
this.store.recordMessage({
|
|
87
|
+
wechat_msg_id: msg.messageId,
|
|
88
|
+
session_id: "", // Will be updated if routed
|
|
89
|
+
agent_id: "",
|
|
90
|
+
wechat_id: msg.wechatAccountId,
|
|
91
|
+
user_id: msg.fromUserId,
|
|
92
|
+
direction: "inbound",
|
|
93
|
+
content: msg.text,
|
|
94
|
+
});
|
|
95
|
+
// System commands
|
|
96
|
+
if (isSystemCommand(msg.text)) {
|
|
97
|
+
const result = await handleSystemCommand({
|
|
98
|
+
wechatAccountId: msg.wechatAccountId,
|
|
99
|
+
fromUserId: msg.fromUserId,
|
|
100
|
+
text: msg.text,
|
|
101
|
+
store: this.store,
|
|
102
|
+
agentManager: this.agentManager,
|
|
103
|
+
});
|
|
104
|
+
this.emitLog({
|
|
105
|
+
type: "system",
|
|
106
|
+
wechatId: msg.wechatAccountId,
|
|
107
|
+
userId: msg.fromUserId,
|
|
108
|
+
text: result.reply,
|
|
109
|
+
});
|
|
110
|
+
await connector.sendText(msg.fromUserId, result.reply);
|
|
111
|
+
return;
|
|
112
|
+
}
|
|
113
|
+
// Snapshot sticky agent before routing (route() updates it)
|
|
114
|
+
const previousStickyAgent = this.store.getStickyRoute(msg.wechatAccountId, msg.fromUserId);
|
|
115
|
+
// Route to agent
|
|
116
|
+
let route;
|
|
117
|
+
try {
|
|
118
|
+
route = await this.router.route(msg);
|
|
119
|
+
}
|
|
120
|
+
catch (err) {
|
|
121
|
+
if (err instanceof PermissionError) {
|
|
122
|
+
const allowedStr = err.allowedAgents.length > 0
|
|
123
|
+
? `可用: ${err.allowedAgents.map((a) => `#${a}`).join(", ")}`
|
|
124
|
+
: "暂无已配置的 Agent。";
|
|
125
|
+
const reply = `⚠️ 无权访问 #${err.agentId}。${allowedStr}`;
|
|
126
|
+
await connector.sendText(msg.fromUserId, reply);
|
|
127
|
+
return;
|
|
128
|
+
}
|
|
129
|
+
throw err;
|
|
130
|
+
}
|
|
131
|
+
if (!route) {
|
|
132
|
+
// No agent resolved and not a system command
|
|
133
|
+
const defaultAgent = this.store.getDefaultAgent(msg.wechatAccountId, msg.fromUserId);
|
|
134
|
+
if (!defaultAgent) {
|
|
135
|
+
await connector.sendText(msg.fromUserId, "⚠️ 未配置 Agent。使用 /agents 查看可用 Agent,或联系管理员配置路由。");
|
|
136
|
+
}
|
|
137
|
+
return;
|
|
138
|
+
}
|
|
139
|
+
const agentSwitchedViaReply = route.routedVia === "reply"
|
|
140
|
+
&& previousStickyAgent != null
|
|
141
|
+
&& previousStickyAgent !== route.agentId;
|
|
142
|
+
// Update inbound message record with proper session/agent info
|
|
143
|
+
if (msg.messageId) {
|
|
144
|
+
this.store.updateMessageRouting(msg.messageId, route.sessionId, route.agentId);
|
|
145
|
+
}
|
|
146
|
+
// Auto-label session from first message
|
|
147
|
+
this.store.setSessionLabelIfEmpty(route.sessionId, route.body.slice(0, 20));
|
|
148
|
+
this.emitLog({
|
|
149
|
+
type: "thinking",
|
|
150
|
+
wechatId: msg.wechatAccountId,
|
|
151
|
+
userId: msg.fromUserId,
|
|
152
|
+
agentId: route.agentId,
|
|
153
|
+
text: `[${route.displayName}] Thinking...`,
|
|
154
|
+
});
|
|
155
|
+
// Send typing indicator
|
|
156
|
+
await connector.sendTypingIndicator(msg.fromUserId, true);
|
|
157
|
+
try {
|
|
158
|
+
// Send to agent
|
|
159
|
+
const result = await this.agentManager.sendToAgent(route.agentId, {
|
|
160
|
+
sessionId: route.sessionId,
|
|
161
|
+
agentSessionId: route.agentSessionId,
|
|
162
|
+
message: route.body,
|
|
163
|
+
onStream: (chunk) => {
|
|
164
|
+
this.emitLog({
|
|
165
|
+
type: "outbound",
|
|
166
|
+
wechatId: msg.wechatAccountId,
|
|
167
|
+
userId: msg.fromUserId,
|
|
168
|
+
agentId: route.agentId,
|
|
169
|
+
text: chunk,
|
|
170
|
+
});
|
|
171
|
+
},
|
|
172
|
+
});
|
|
173
|
+
// Update session
|
|
174
|
+
if (result.agentSessionId) {
|
|
175
|
+
this.store.updateSessionAgentSession(route.sessionId, result.agentSessionId);
|
|
176
|
+
}
|
|
177
|
+
this.store.incrementSessionMessages(route.sessionId);
|
|
178
|
+
// Stop typing
|
|
179
|
+
await connector.sendTypingIndicator(msg.fromUserId, false);
|
|
180
|
+
// Format and send reply
|
|
181
|
+
const formattedReply = this.router.formatReply(route.displayName, result.text);
|
|
182
|
+
// Chunk long messages (WeChat limit ~4000 chars)
|
|
183
|
+
const chunks = chunkText(formattedReply, 4000);
|
|
184
|
+
for (const chunk of chunks) {
|
|
185
|
+
const { clientId, serverMsgId } = await connector.sendText(msg.fromUserId, chunk);
|
|
186
|
+
// Record outbound for reply tracking (store both client_id and server-assigned wechat_msg_id)
|
|
187
|
+
this.store.recordMessage({
|
|
188
|
+
wechat_msg_id: serverMsgId,
|
|
189
|
+
client_id: clientId,
|
|
190
|
+
session_id: route.sessionId,
|
|
191
|
+
agent_id: route.agentId,
|
|
192
|
+
wechat_id: msg.wechatAccountId,
|
|
193
|
+
user_id: msg.fromUserId,
|
|
194
|
+
direction: "outbound",
|
|
195
|
+
content: chunk,
|
|
196
|
+
});
|
|
197
|
+
}
|
|
198
|
+
// Handle media in response
|
|
199
|
+
if (result.mediaUrls) {
|
|
200
|
+
for (const mediaUrl of result.mediaUrls) {
|
|
201
|
+
try {
|
|
202
|
+
if (mediaUrl.startsWith("http://") || mediaUrl.startsWith("https://")) {
|
|
203
|
+
const { downloadRemoteToTemp } = await import("./wechat/cdn.js");
|
|
204
|
+
const tempPath = await downloadRemoteToTemp(mediaUrl);
|
|
205
|
+
await connector.sendMediaFile(msg.fromUserId, tempPath);
|
|
206
|
+
}
|
|
207
|
+
else {
|
|
208
|
+
await connector.sendMediaFile(msg.fromUserId, mediaUrl);
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
catch (mediaErr) {
|
|
212
|
+
console.error(`[Dispatcher] Failed to send media:`, mediaErr);
|
|
213
|
+
}
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
// Notify user when agent switched via reply-based routing
|
|
217
|
+
if (agentSwitchedViaReply) {
|
|
218
|
+
const hint = `💡 当前对话已切换至 ${route.displayName}`;
|
|
219
|
+
await connector.sendText(msg.fromUserId, hint);
|
|
220
|
+
}
|
|
221
|
+
this.emitLog({
|
|
222
|
+
type: "outbound",
|
|
223
|
+
wechatId: msg.wechatAccountId,
|
|
224
|
+
userId: msg.fromUserId,
|
|
225
|
+
agentId: route.agentId,
|
|
226
|
+
text: result.text.slice(0, 200) + (result.text.length > 200 ? "..." : ""),
|
|
227
|
+
});
|
|
228
|
+
}
|
|
229
|
+
catch (err) {
|
|
230
|
+
await connector.sendTypingIndicator(msg.fromUserId, false);
|
|
231
|
+
const errorMsg = `⚠️ ${route.displayName} 出错: ${err.message}`;
|
|
232
|
+
await connector.sendText(msg.fromUserId, errorMsg);
|
|
233
|
+
this.emitLog({
|
|
234
|
+
type: "error",
|
|
235
|
+
wechatId: msg.wechatAccountId,
|
|
236
|
+
userId: msg.fromUserId,
|
|
237
|
+
agentId: route.agentId,
|
|
238
|
+
text: err.message,
|
|
239
|
+
});
|
|
240
|
+
}
|
|
241
|
+
}
|
|
242
|
+
emitLog(entry) {
|
|
243
|
+
this.emit("log", { ...entry, time: Date.now() });
|
|
244
|
+
}
|
|
245
|
+
}
|
|
246
|
+
function chunkText(text, maxLen) {
|
|
247
|
+
if (text.length <= maxLen)
|
|
248
|
+
return [text];
|
|
249
|
+
const chunks = [];
|
|
250
|
+
let remaining = text;
|
|
251
|
+
while (remaining.length > 0) {
|
|
252
|
+
if (remaining.length <= maxLen) {
|
|
253
|
+
chunks.push(remaining);
|
|
254
|
+
break;
|
|
255
|
+
}
|
|
256
|
+
let splitAt = remaining.lastIndexOf("\n", maxLen);
|
|
257
|
+
if (splitAt < maxLen / 2)
|
|
258
|
+
splitAt = maxLen;
|
|
259
|
+
chunks.push(remaining.slice(0, splitAt));
|
|
260
|
+
remaining = remaining.slice(splitAt);
|
|
261
|
+
}
|
|
262
|
+
return chunks;
|
|
263
|
+
}
|
|
264
|
+
//# sourceMappingURL=dispatcher.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dispatcher.js","sourceRoot":"","sources":["../../src/center/dispatcher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACrF,OAAO,EAAE,eAAe,EAAuB,MAAM,uBAAuB,CAAC;AAe7E,MAAM,OAAO,UAAW,SAAQ,YAAY;IAClC,KAAK,CAAQ;IACb,YAAY,CAAe;IAC3B,MAAM,CAAS;IACf,UAAU,GAAG,IAAI,GAAG,EAA2B,CAAC;IAExD,YAAY,KAAY,EAAE,YAA0B;QAClD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QAEjC,MAAM,YAAY,GAAG,KAAK,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,iBAAiB,CAAC;QAClF,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,YAAY,CAAC,QAAgB;QAC3B,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED,cAAc;QACZ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,mBAAmB,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC5D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAEzC,SAAS,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAmB,EAAE,EAAE;YAC9C,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/C,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,GAAG,CAAC,CAAC;gBAC3D,IAAI,CAAC,OAAO,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,QAAQ,EAAE,GAAG,CAAC,eAAe;oBAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;oBACtB,IAAI,EAAE,UAAW,GAAa,CAAC,OAAO,EAAE;iBACzC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;YACnC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,oBAAoB,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;YACnC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,oCAAoC,EAAE,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,QAAgB;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,SAAS,CAAC,IAAI,EAAE,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,KAAK,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC9C,SAAS,CAAC,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,0CAA0C;IAC1C,KAAK,CAAC,kBAAkB;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;QACjD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACxC,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,OAAO,CAAC,EAAE,GAAG,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;gBACjG,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAA0B,EAAE,GAAmB;QACzE,IAAI,CAAC,OAAO,CAAC;YACX,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,GAAG,CAAC,eAAe;YAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;YACtB,IAAI,EAAE,GAAG,CAAC,IAAI;SACf,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;YACvB,aAAa,EAAE,GAAG,CAAC,SAAS;YAC5B,UAAU,EAAE,EAAE,EAAE,4BAA4B;YAC5C,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,GAAG,CAAC,eAAe;YAC9B,OAAO,EAAE,GAAG,CAAC,UAAU;YACvB,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,GAAG,CAAC,IAAI;SAClB,CAAC,CAAC;QAEH,kBAAkB;QAClB,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC;gBACvC,eAAe,EAAE,GAAG,CAAC,eAAe;gBACpC,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,YAAY,EAAE,IAAI,CAAC,YAAY;aAChC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC;gBACX,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,GAAG,CAAC,eAAe;gBAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;gBACtB,IAAI,EAAE,MAAM,CAAC,KAAK;aACnB,CAAC,CAAC;YAEH,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,4DAA4D;QAC5D,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;QAE3F,iBAAiB;QACjB,IAAI,KAAK,CAAC;QACV,IAAI,CAAC;YACH,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACvC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,eAAe,EAAE,CAAC;gBACnC,MAAM,UAAU,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC;oBAC7C,CAAC,CAAC,OAAO,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;oBAC3D,CAAC,CAAC,eAAe,CAAC;gBACpB,MAAM,KAAK,GAAG,YAAY,GAAG,CAAC,OAAO,IAAI,UAAU,EAAE,CAAC;gBACtD,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAChD,OAAO;YACT,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,6CAA6C;YAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;YACrF,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,SAAS,CAAC,QAAQ,CACtB,GAAG,CAAC,UAAU,EACd,gDAAgD,CACjD,CAAC;YACJ,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,qBAAqB,GAAG,KAAK,CAAC,SAAS,KAAK,OAAO;eACpD,mBAAmB,IAAI,IAAI;eAC3B,mBAAmB,KAAK,KAAK,CAAC,OAAO,CAAC;QAE3C,+DAA+D;QAC/D,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACjF,CAAC;QAED,wCAAwC;QACxC,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAE5E,IAAI,CAAC,OAAO,CAAC;YACX,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,GAAG,CAAC,eAAe;YAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;YACtB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,IAAI,KAAK,CAAC,WAAW,eAAe;SAC3C,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,SAAS,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QAE1D,IAAI,CAAC;YACH,gBAAgB;YAChB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE;gBAChE,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,cAAc,EAAE,KAAK,CAAC,cAAc;gBACpC,OAAO,EAAE,KAAK,CAAC,IAAI;gBACnB,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;oBAClB,IAAI,CAAC,OAAO,CAAC;wBACX,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,GAAG,CAAC,eAAe;wBAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;wBACtB,OAAO,EAAE,KAAM,CAAC,OAAO;wBACvB,IAAI,EAAE,KAAK;qBACZ,CAAC,CAAC;gBACL,CAAC;aACF,CAAC,CAAC;YAEH,iBAAiB;YACjB,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC1B,IAAI,CAAC,KAAK,CAAC,yBAAyB,CAAC,KAAK,CAAC,SAAS,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC;YAC/E,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,wBAAwB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAErD,cAAc;YACd,MAAM,SAAS,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAE3D,wBAAwB;YACxB,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAE/E,iDAAiD;YACjD,MAAM,MAAM,GAAG,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAC/C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAElF,8FAA8F;gBAC9F,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;oBACvB,aAAa,EAAE,WAAW;oBAC1B,SAAS,EAAE,QAAQ;oBACnB,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,QAAQ,EAAE,KAAK,CAAC,OAAO;oBACvB,SAAS,EAAE,GAAG,CAAC,eAAe;oBAC9B,OAAO,EAAE,GAAG,CAAC,UAAU;oBACvB,SAAS,EAAE,UAAU;oBACrB,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;YACL,CAAC;YAED,2BAA2B;YAC3B,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACxC,IAAI,CAAC;wBACH,IAAI,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;4BACtE,MAAM,EAAE,oBAAoB,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;4BACjE,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,QAAQ,CAAC,CAAC;4BACtD,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;wBAC1D,CAAC;6BAAM,CAAC;4BACN,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;wBAC1D,CAAC;oBACH,CAAC;oBAAC,OAAO,QAAQ,EAAE,CAAC;wBAClB,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,QAAQ,CAAC,CAAC;oBAChE,CAAC;gBACH,CAAC;YACH,CAAC;YAED,0DAA0D;YAC1D,IAAI,qBAAqB,EAAE,CAAC;gBAC1B,MAAM,IAAI,GAAG,eAAe,KAAK,CAAC,WAAW,EAAE,CAAC;gBAChD,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YACjD,CAAC;YAED,IAAI,CAAC,OAAO,CAAC;gBACX,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,GAAG,CAAC,eAAe;gBAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;gBACtB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1E,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,SAAS,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,WAAW,QAAS,GAAa,CAAC,OAAO,EAAE,CAAC;YACzE,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO,CAAC;gBACX,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,GAAG,CAAC,eAAe;gBAC7B,MAAM,EAAE,GAAG,CAAC,UAAU;gBACtB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,IAAI,EAAG,GAAa,CAAC,OAAO;aAC7B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,KAA6B;QAC3C,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,EAAc,CAAC,CAAC;IAC/D,CAAC;CACF;AAED,SAAS,SAAS,CAAC,IAAY,EAAE,MAAc;IAC7C,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM;QAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,SAAS,GAAG,IAAI,CAAC;IACrB,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,SAAS,CAAC,MAAM,IAAI,MAAM,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACvB,MAAM;QACR,CAAC;QACD,IAAI,OAAO,GAAG,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAClD,IAAI,OAAO,GAAG,MAAM,GAAG,CAAC;YAAE,OAAO,GAAG,MAAM,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;QACzC,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { AgentManager } from "../core/agents/manager.js";
|
|
2
|
+
import type { Store } from "../core/db/store.js";
|
|
3
|
+
import { EventEmitter } from "node:events";
|
|
4
|
+
export declare class WorkerHub extends EventEmitter {
|
|
5
|
+
private wss;
|
|
6
|
+
private workers;
|
|
7
|
+
private agentManager;
|
|
8
|
+
private store;
|
|
9
|
+
constructor(agentManager: AgentManager, store: Store);
|
|
10
|
+
start(port: number, host?: string): void;
|
|
11
|
+
stop(): void;
|
|
12
|
+
private handleConnection;
|
|
13
|
+
private handleMessage;
|
|
14
|
+
private sendTaskToWorker;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=hub.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hub.d.ts","sourceRoot":"","sources":["../../src/center/hub.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAc3C,qBAAa,SAAU,SAAQ,YAAY;IACzC,OAAO,CAAC,GAAG,CAAgC;IAC3C,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,KAAK,CAAQ;gBAET,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK;IAMpD,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,GAAE,MAAkB,GAAG,IAAI;IAenD,IAAI,IAAI,IAAI;IAUZ,OAAO,CAAC,gBAAgB;IAiCxB,OAAO,CAAC,aAAa;IA8IrB,OAAO,CAAC,gBAAgB;CAkCzB"}
|
|
@@ -0,0 +1,214 @@
|
|
|
1
|
+
import { WebSocketServer } from "ws";
|
|
2
|
+
import { EventEmitter } from "node:events";
|
|
3
|
+
export class WorkerHub extends EventEmitter {
|
|
4
|
+
wss = null;
|
|
5
|
+
workers = new Map();
|
|
6
|
+
agentManager;
|
|
7
|
+
store;
|
|
8
|
+
constructor(agentManager, store) {
|
|
9
|
+
super();
|
|
10
|
+
this.agentManager = agentManager;
|
|
11
|
+
this.store = store;
|
|
12
|
+
}
|
|
13
|
+
start(port, host = "0.0.0.0") {
|
|
14
|
+
this.wss = new WebSocketServer({ port, host });
|
|
15
|
+
console.log(`[WorkerHub] Listening on ${host}:${port}`);
|
|
16
|
+
this.wss.on("connection", (ws, req) => {
|
|
17
|
+
const address = req.socket.remoteAddress ?? "unknown";
|
|
18
|
+
console.log(`[WorkerHub] New connection from ${address}`);
|
|
19
|
+
this.handleConnection(ws, address);
|
|
20
|
+
});
|
|
21
|
+
this.wss.on("error", (err) => {
|
|
22
|
+
console.error("[WorkerHub] Server error:", err);
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
stop() {
|
|
26
|
+
for (const [nodeId, worker] of this.workers) {
|
|
27
|
+
worker.ws.close();
|
|
28
|
+
this.agentManager.disconnectWorkerAgents(nodeId);
|
|
29
|
+
this.store.updateWorkerNodeStatus(nodeId, "disconnected");
|
|
30
|
+
}
|
|
31
|
+
this.workers.clear();
|
|
32
|
+
this.wss?.close();
|
|
33
|
+
}
|
|
34
|
+
handleConnection(ws, address) {
|
|
35
|
+
let worker = null;
|
|
36
|
+
ws.on("message", (data) => {
|
|
37
|
+
try {
|
|
38
|
+
const msg = JSON.parse(data.toString());
|
|
39
|
+
this.handleMessage(ws, msg, address, worker, (w) => { worker = w; });
|
|
40
|
+
}
|
|
41
|
+
catch (err) {
|
|
42
|
+
console.error("[WorkerHub] Invalid message:", err);
|
|
43
|
+
}
|
|
44
|
+
});
|
|
45
|
+
ws.on("close", () => {
|
|
46
|
+
if (worker) {
|
|
47
|
+
console.log(`[WorkerHub] Worker ${worker.nodeId} disconnected`);
|
|
48
|
+
this.agentManager.disconnectWorkerAgents(worker.nodeId);
|
|
49
|
+
this.store.updateWorkerNodeStatus(worker.nodeId, "disconnected");
|
|
50
|
+
this.workers.delete(worker.nodeId);
|
|
51
|
+
// Reject all pending tasks
|
|
52
|
+
for (const [, task] of worker.pendingTasks) {
|
|
53
|
+
task.reject(new Error("Worker disconnected"));
|
|
54
|
+
}
|
|
55
|
+
this.emit("worker:disconnected", worker.nodeId);
|
|
56
|
+
}
|
|
57
|
+
});
|
|
58
|
+
ws.on("error", (err) => {
|
|
59
|
+
console.error(`[WorkerHub] WebSocket error:`, err);
|
|
60
|
+
});
|
|
61
|
+
}
|
|
62
|
+
handleMessage(ws, msg, address, worker, setWorker) {
|
|
63
|
+
switch (msg.type) {
|
|
64
|
+
case "register": {
|
|
65
|
+
const nodeId = msg.nodeId;
|
|
66
|
+
if (!nodeId) {
|
|
67
|
+
ws.send(JSON.stringify({ type: "error", message: "Missing nodeId" }));
|
|
68
|
+
return;
|
|
69
|
+
}
|
|
70
|
+
const existing = this.workers.get(nodeId);
|
|
71
|
+
if (existing) {
|
|
72
|
+
existing.ws.close();
|
|
73
|
+
this.agentManager.disconnectWorkerAgents(nodeId);
|
|
74
|
+
}
|
|
75
|
+
const newWorker = {
|
|
76
|
+
ws,
|
|
77
|
+
nodeId,
|
|
78
|
+
address,
|
|
79
|
+
agents: new Map(),
|
|
80
|
+
pendingTasks: new Map(),
|
|
81
|
+
};
|
|
82
|
+
this.workers.set(nodeId, newWorker);
|
|
83
|
+
setWorker(newWorker);
|
|
84
|
+
this.store.registerWorkerNode(nodeId, address);
|
|
85
|
+
ws.send(JSON.stringify({ type: "registered", nodeId }));
|
|
86
|
+
console.log(`[WorkerHub] Worker registered: ${nodeId} (${address})`);
|
|
87
|
+
this.emit("worker:connected", nodeId);
|
|
88
|
+
break;
|
|
89
|
+
}
|
|
90
|
+
case "sync": {
|
|
91
|
+
if (!worker)
|
|
92
|
+
return;
|
|
93
|
+
const agents = msg.agents;
|
|
94
|
+
if (!agents)
|
|
95
|
+
return;
|
|
96
|
+
// Clear existing worker agents
|
|
97
|
+
this.agentManager.removeWorkerAgents(worker.nodeId);
|
|
98
|
+
worker.agents.clear();
|
|
99
|
+
for (const agent of agents) {
|
|
100
|
+
worker.agents.set(agent.id, {
|
|
101
|
+
displayName: agent.displayName,
|
|
102
|
+
type: agent.type,
|
|
103
|
+
status: agent.status,
|
|
104
|
+
});
|
|
105
|
+
if (agent.status === "running") {
|
|
106
|
+
this.agentManager.registerWorkerAgent(agent.id, agent.displayName, worker.nodeId, (agentId, params) => this.sendTaskToWorker(worker, agentId, params));
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
console.log(`[WorkerHub] Synced ${agents.length} agents from ${worker.nodeId}`);
|
|
110
|
+
this.emit("worker:synced", worker.nodeId);
|
|
111
|
+
break;
|
|
112
|
+
}
|
|
113
|
+
case "agent-added":
|
|
114
|
+
case "agent-update": {
|
|
115
|
+
if (!worker)
|
|
116
|
+
return;
|
|
117
|
+
const agent = msg.agent;
|
|
118
|
+
if (!agent)
|
|
119
|
+
return;
|
|
120
|
+
worker.agents.set(agent.id, {
|
|
121
|
+
displayName: agent.displayName,
|
|
122
|
+
type: agent.type,
|
|
123
|
+
status: agent.status,
|
|
124
|
+
});
|
|
125
|
+
if (agent.status === "running") {
|
|
126
|
+
this.agentManager.registerWorkerAgent(agent.id, agent.displayName, worker.nodeId, (agentId, params) => this.sendTaskToWorker(worker, agentId, params));
|
|
127
|
+
}
|
|
128
|
+
else {
|
|
129
|
+
const adapter = this.agentManager.getAdapter(agent.id);
|
|
130
|
+
if (adapter)
|
|
131
|
+
adapter.status = agent.status;
|
|
132
|
+
}
|
|
133
|
+
break;
|
|
134
|
+
}
|
|
135
|
+
case "agent-removed": {
|
|
136
|
+
if (!worker)
|
|
137
|
+
return;
|
|
138
|
+
const agentId = msg.agentId;
|
|
139
|
+
worker.agents.delete(agentId);
|
|
140
|
+
try {
|
|
141
|
+
this.agentManager.removeAgent(agentId).catch(() => { });
|
|
142
|
+
}
|
|
143
|
+
catch { /* ignore */ }
|
|
144
|
+
break;
|
|
145
|
+
}
|
|
146
|
+
case "stream": {
|
|
147
|
+
if (!worker)
|
|
148
|
+
return;
|
|
149
|
+
const taskId = msg.taskId;
|
|
150
|
+
const chunk = msg.chunk;
|
|
151
|
+
const task = worker.pendingTasks.get(taskId);
|
|
152
|
+
task?.onStream?.(chunk);
|
|
153
|
+
break;
|
|
154
|
+
}
|
|
155
|
+
case "result": {
|
|
156
|
+
if (!worker)
|
|
157
|
+
return;
|
|
158
|
+
const taskId = msg.taskId;
|
|
159
|
+
const task = worker.pendingTasks.get(taskId);
|
|
160
|
+
if (task) {
|
|
161
|
+
worker.pendingTasks.delete(taskId);
|
|
162
|
+
task.resolve({
|
|
163
|
+
text: msg.text ?? "",
|
|
164
|
+
agentSessionId: msg.agentSessionId,
|
|
165
|
+
mediaUrls: msg.mediaUrls,
|
|
166
|
+
});
|
|
167
|
+
}
|
|
168
|
+
break;
|
|
169
|
+
}
|
|
170
|
+
case "task-error": {
|
|
171
|
+
if (!worker)
|
|
172
|
+
return;
|
|
173
|
+
const taskId = msg.taskId;
|
|
174
|
+
const task = worker.pendingTasks.get(taskId);
|
|
175
|
+
if (task) {
|
|
176
|
+
worker.pendingTasks.delete(taskId);
|
|
177
|
+
task.reject(new Error(msg.error ?? "Unknown error"));
|
|
178
|
+
}
|
|
179
|
+
break;
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
}
|
|
183
|
+
sendTaskToWorker(worker, agentId, params) {
|
|
184
|
+
return new Promise((resolve, reject) => {
|
|
185
|
+
const taskId = `task-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
186
|
+
worker.pendingTasks.set(taskId, {
|
|
187
|
+
resolve,
|
|
188
|
+
reject,
|
|
189
|
+
onStream: params.onStream,
|
|
190
|
+
});
|
|
191
|
+
const timeout = setTimeout(() => {
|
|
192
|
+
worker.pendingTasks.delete(taskId);
|
|
193
|
+
reject(new Error("Task timeout (5 minutes)"));
|
|
194
|
+
}, 5 * 60_000);
|
|
195
|
+
const origResolve = resolve;
|
|
196
|
+
const origReject = reject;
|
|
197
|
+
worker.pendingTasks.set(taskId, {
|
|
198
|
+
resolve: (result) => { clearTimeout(timeout); origResolve(result); },
|
|
199
|
+
reject: (err) => { clearTimeout(timeout); origReject(err); },
|
|
200
|
+
onStream: params.onStream,
|
|
201
|
+
});
|
|
202
|
+
worker.ws.send(JSON.stringify({
|
|
203
|
+
type: "task",
|
|
204
|
+
taskId,
|
|
205
|
+
agentId,
|
|
206
|
+
sessionId: params.sessionId,
|
|
207
|
+
agentSessionId: params.agentSessionId,
|
|
208
|
+
message: params.message,
|
|
209
|
+
mediaPath: params.mediaPath,
|
|
210
|
+
}));
|
|
211
|
+
});
|
|
212
|
+
}
|
|
213
|
+
}
|
|
214
|
+
//# sourceMappingURL=hub.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hub.js","sourceRoot":"","sources":["../../src/center/hub.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAa,MAAM,IAAI,CAAC;AAKhD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAc3C,MAAM,OAAO,SAAU,SAAQ,YAAY;IACjC,GAAG,GAA2B,IAAI,CAAC;IACnC,OAAO,GAAG,IAAI,GAAG,EAA4B,CAAC;IAC9C,YAAY,CAAe;IAC3B,KAAK,CAAQ;IAErB,YAAY,YAA0B,EAAE,KAAY;QAClD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAY,EAAE,OAAe,SAAS;QAC1C,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;QAExD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,GAAoB,EAAE,EAAE;YAChE,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,mCAAmC,OAAO,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YAC3B,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,GAAG,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,KAAK,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5C,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC;IACpB,CAAC;IAEO,gBAAgB,CAAC,EAAa,EAAE,OAAe;QACrD,IAAI,MAAM,GAA4B,IAAI,CAAC;QAE3C,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACxC,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAC;YACrD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,GAAG,CAAC,sBAAsB,MAAM,CAAC,MAAM,eAAe,CAAC,CAAC;gBAChE,IAAI,CAAC,YAAY,CAAC,sBAAsB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACxD,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;gBACjE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAEnC,2BAA2B;gBAC3B,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBAC3C,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBAChD,CAAC;gBAED,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAClD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CACnB,EAAa,EACb,GAA4B,EAC5B,OAAe,EACf,MAA+B,EAC/B,SAAwC;QAExC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAgB,CAAC;gBACpC,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;oBACtE,OAAO;gBACT,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC1C,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;oBACpB,IAAI,CAAC,YAAY,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;gBACnD,CAAC;gBAED,MAAM,SAAS,GAAqB;oBAClC,EAAE;oBACF,MAAM;oBACN,OAAO;oBACP,MAAM,EAAE,IAAI,GAAG,EAAE;oBACjB,YAAY,EAAE,IAAI,GAAG,EAAE;iBACxB,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBACpC,SAAS,CAAC,SAAS,CAAC,CAAC;gBAErB,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC/C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,kCAAkC,MAAM,KAAK,OAAO,GAAG,CAAC,CAAC;gBACrE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;gBACtC,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAkF,CAAC;gBACtG,IAAI,CAAC,MAAM;oBAAE,OAAO;gBAEpB,+BAA+B;gBAC/B,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACpD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;gBAEtB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;wBAC1B,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;oBAEH,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;wBAC/B,IAAI,CAAC,YAAY,CAAC,mBAAmB,CACnC,KAAK,CAAC,EAAE,EACR,KAAK,CAAC,WAAW,EACjB,MAAM,CAAC,MAAM,EACb,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CACrE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO,CAAC,GAAG,CAAC,sBAAsB,MAAM,CAAC,MAAM,gBAAgB,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;gBAChF,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC1C,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC;YACnB,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,KAAK,GAAG,GAAG,CAAC,KAA0E,CAAC;gBAC7F,IAAI,CAAC,KAAK;oBAAE,OAAO;gBAEnB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE;oBAC1B,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;iBACrB,CAAC,CAAC;gBAEH,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC/B,IAAI,CAAC,YAAY,CAAC,mBAAmB,CACnC,KAAK,CAAC,EAAE,EACR,KAAK,CAAC,WAAW,EACjB,MAAM,CAAC,MAAM,EACb,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CACrE,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACvD,IAAI,OAAO;wBAAE,OAAO,CAAC,MAAM,GAAG,KAAK,CAAC,MAAyC,CAAC;gBAChF,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,OAAO,GAAG,GAAG,CAAC,OAAiB,CAAC;gBACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC9B,IAAI,CAAC;oBACH,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBACzD,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxB,MAAM;YACR,CAAC;YAED,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAgB,CAAC;gBACpC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAe,CAAC;gBAClC,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC7C,IAAI,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC;gBACxB,MAAM;YACR,CAAC;YAED,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAgB,CAAC;gBACpC,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC7C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACnC,IAAI,CAAC,OAAO,CAAC;wBACX,IAAI,EAAE,GAAG,CAAC,IAAc,IAAI,EAAE;wBAC9B,cAAc,EAAE,GAAG,CAAC,cAAoC;wBACxD,SAAS,EAAE,GAAG,CAAC,SAAiC;qBACjD,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,IAAI,CAAC,MAAM;oBAAE,OAAO;gBACpB,MAAM,MAAM,GAAG,GAAG,CAAC,MAAgB,CAAC;gBACpC,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC7C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,KAAe,IAAI,eAAe,CAAC,CAAC,CAAC;gBACjE,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,MAAwB,EAAE,OAAe,EAAE,MAAkB;QACpF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,MAAM,GAAG,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAE9E,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE;gBAC9B,OAAO;gBACP,MAAM;gBACN,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACnC,MAAM,CAAC,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;YAChD,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,CAAC;YAEf,MAAM,WAAW,GAAG,OAAO,CAAC;YAC5B,MAAM,UAAU,GAAG,MAAM,CAAC;YAC1B,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE;gBAC9B,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACpE,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC5D,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC5B,IAAI,EAAE,MAAM;gBACZ,MAAM;gBACN,OAAO;gBACP,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import type { Store } from "../../core/db/store.js";
|
|
2
|
+
import type { AgentManager } from "../../core/agents/manager.js";
|
|
3
|
+
import type { InboundMessage } from "../wechat/connector.js";
|
|
4
|
+
export type RoutedVia = "hashtag" | "reply" | "sticky" | "default";
|
|
5
|
+
export interface RouteResult {
|
|
6
|
+
agentId: string;
|
|
7
|
+
displayName: string;
|
|
8
|
+
body: string;
|
|
9
|
+
sessionId: string;
|
|
10
|
+
agentSessionId?: string;
|
|
11
|
+
routedVia: RoutedVia;
|
|
12
|
+
}
|
|
13
|
+
export interface RouterDeps {
|
|
14
|
+
store: Store;
|
|
15
|
+
agentManager: AgentManager;
|
|
16
|
+
replyPrefixFormat: string;
|
|
17
|
+
}
|
|
18
|
+
export declare class Router {
|
|
19
|
+
private deps;
|
|
20
|
+
private store;
|
|
21
|
+
private agentManager;
|
|
22
|
+
constructor(deps: RouterDeps);
|
|
23
|
+
/**
|
|
24
|
+
* Route an inbound message to the appropriate agent.
|
|
25
|
+
* Returns null if the message was handled as a system command.
|
|
26
|
+
*/
|
|
27
|
+
route(msg: InboundMessage): Promise<RouteResult | null>;
|
|
28
|
+
formatReply(displayName: string, text: string): string;
|
|
29
|
+
}
|
|
30
|
+
export declare class PermissionError extends Error {
|
|
31
|
+
readonly agentId: string;
|
|
32
|
+
readonly allowedAgents: string[];
|
|
33
|
+
constructor(agentId: string, allowedAgents: string[]);
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=engine.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"engine.d.ts","sourceRoot":"","sources":["../../../src/center/router/engine.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAK7D,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC;AAEnE,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,SAAS,CAAC;CACtB;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,KAAK,CAAC;IACb,YAAY,EAAE,YAAY,CAAC;IAC3B,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAED,qBAAa,MAAM;IAIL,OAAO,CAAC,IAAI;IAHxB,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,YAAY,CAAe;gBAEf,IAAI,EAAE,UAAU;IAKpC;;;OAGG;IACG,KAAK,CAAC,GAAG,EAAE,cAAc,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IA4F7D,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;CAIvD;AAED,qBAAa,eAAgB,SAAQ,KAAK;aAEtB,OAAO,EAAE,MAAM;aACf,aAAa,EAAE,MAAM,EAAE;gBADvB,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EAAE;CAK1C"}
|