openclaw-rocketchat 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 +21 -0
- package/README.en.md +694 -0
- package/README.md +730 -0
- package/dist/cli/add-bot.d.ts +2 -0
- package/dist/cli/add-bot.d.ts.map +1 -0
- package/dist/cli/add-bot.js +138 -0
- package/dist/cli/add-bot.js.map +1 -0
- package/dist/cli/add-group.d.ts +2 -0
- package/dist/cli/add-group.d.ts.map +1 -0
- package/dist/cli/add-group.js +145 -0
- package/dist/cli/add-group.js.map +1 -0
- package/dist/cli/add-user.d.ts +2 -0
- package/dist/cli/add-user.d.ts.map +1 -0
- package/dist/cli/add-user.js +186 -0
- package/dist/cli/add-user.js.map +1 -0
- package/dist/cli/invite.d.ts +2 -0
- package/dist/cli/invite.d.ts.map +1 -0
- package/dist/cli/invite.js +194 -0
- package/dist/cli/invite.js.map +1 -0
- package/dist/cli/prompts.d.ts +24 -0
- package/dist/cli/prompts.d.ts.map +1 -0
- package/dist/cli/prompts.js +134 -0
- package/dist/cli/prompts.js.map +1 -0
- package/dist/cli/remove-user.d.ts +2 -0
- package/dist/cli/remove-user.d.ts.map +1 -0
- package/dist/cli/remove-user.js +108 -0
- package/dist/cli/remove-user.js.map +1 -0
- package/dist/cli/setup.d.ts +2 -0
- package/dist/cli/setup.d.ts.map +1 -0
- package/dist/cli/setup.js +446 -0
- package/dist/cli/setup.js.map +1 -0
- package/dist/cli/status.d.ts +2 -0
- package/dist/cli/status.d.ts.map +1 -0
- package/dist/cli/status.js +161 -0
- package/dist/cli/status.js.map +1 -0
- package/dist/cli/uninstall.d.ts +2 -0
- package/dist/cli/uninstall.d.ts.map +1 -0
- package/dist/cli/uninstall.js +81 -0
- package/dist/cli/uninstall.js.map +1 -0
- package/dist/config/credentials.d.ts +26 -0
- package/dist/config/credentials.d.ts.map +1 -0
- package/dist/config/credentials.js +169 -0
- package/dist/config/credentials.js.map +1 -0
- package/dist/config/writer.d.ts +52 -0
- package/dist/config/writer.d.ts.map +1 -0
- package/dist/config/writer.js +170 -0
- package/dist/config/writer.js.map +1 -0
- package/dist/docker/compose.d.ts +10 -0
- package/dist/docker/compose.d.ts.map +1 -0
- package/dist/docker/compose.js +56 -0
- package/dist/docker/compose.js.map +1 -0
- package/dist/docker/manager.d.ts +38 -0
- package/dist/docker/manager.d.ts.map +1 -0
- package/dist/docker/manager.js +219 -0
- package/dist/docker/manager.js.map +1 -0
- package/dist/index.d.ts +5 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +203 -0
- package/dist/index.js.map +1 -0
- package/dist/rc-api/rest-client.d.ts +64 -0
- package/dist/rc-api/rest-client.d.ts.map +1 -0
- package/dist/rc-api/rest-client.js +268 -0
- package/dist/rc-api/rest-client.js.map +1 -0
- package/dist/rc-api/types.d.ts +158 -0
- package/dist/rc-api/types.d.ts.map +1 -0
- package/dist/rc-api/types.js +5 -0
- package/dist/rc-api/types.js.map +1 -0
- package/dist/rc-api/ws-client.d.ts +47 -0
- package/dist/rc-api/ws-client.d.ts.map +1 -0
- package/dist/rc-api/ws-client.js +308 -0
- package/dist/rc-api/ws-client.js.map +1 -0
- package/dist/service/bot-manager.d.ts +34 -0
- package/dist/service/bot-manager.d.ts.map +1 -0
- package/dist/service/bot-manager.js +156 -0
- package/dist/service/bot-manager.js.map +1 -0
- package/dist/service/channel.d.ts +35 -0
- package/dist/service/channel.d.ts.map +1 -0
- package/dist/service/channel.js +157 -0
- package/dist/service/channel.js.map +1 -0
- package/dist/service/message-handler.d.ts +45 -0
- package/dist/service/message-handler.d.ts.map +1 -0
- package/dist/service/message-handler.js +207 -0
- package/dist/service/message-handler.js.map +1 -0
- package/openclaw.plugin.json +45 -0
- package/package.json +61 -0
|
@@ -0,0 +1,308 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// Rocket.Chat WebSocket (DDP) 客户端
|
|
3
|
+
// 实时消息订阅,自动重连
|
|
4
|
+
// ============================================================
|
|
5
|
+
import WebSocket from "ws";
|
|
6
|
+
/** 重连配置 */
|
|
7
|
+
const RECONNECT_MIN_MS = 1000;
|
|
8
|
+
const RECONNECT_MAX_MS = 30000;
|
|
9
|
+
const PING_TIMEOUT_MS = 10000;
|
|
10
|
+
export class RocketChatWsClient {
|
|
11
|
+
serverUrl;
|
|
12
|
+
ws = null;
|
|
13
|
+
messageCounter = 0;
|
|
14
|
+
pendingCalls = new Map();
|
|
15
|
+
subscriptions = new Map(); // subId -> roomId
|
|
16
|
+
messageHandlers = [];
|
|
17
|
+
connectionHandlers = [];
|
|
18
|
+
reconnecting = false;
|
|
19
|
+
reconnectAttempt = 0;
|
|
20
|
+
reconnectTimer = null;
|
|
21
|
+
pingTimer = null;
|
|
22
|
+
disposed = false;
|
|
23
|
+
loginUser = "";
|
|
24
|
+
loginPassword = "";
|
|
25
|
+
autoReconnect = true;
|
|
26
|
+
constructor(serverUrl) {
|
|
27
|
+
// 转换 http -> ws
|
|
28
|
+
this.serverUrl = serverUrl
|
|
29
|
+
.replace(/^https:/, "wss:")
|
|
30
|
+
.replace(/^http:/, "ws:")
|
|
31
|
+
.replace(/\/+$/, "");
|
|
32
|
+
}
|
|
33
|
+
// ----------------------------------------------------------
|
|
34
|
+
// 连接管理
|
|
35
|
+
// ----------------------------------------------------------
|
|
36
|
+
/** 建立 WebSocket 连接 */
|
|
37
|
+
async connect() {
|
|
38
|
+
return new Promise((resolve, reject) => {
|
|
39
|
+
const wsUrl = `${this.serverUrl}/websocket`;
|
|
40
|
+
this.ws = new WebSocket(wsUrl);
|
|
41
|
+
this.ws.on("open", () => {
|
|
42
|
+
// DDP 握手
|
|
43
|
+
this.sendRaw({
|
|
44
|
+
msg: "connect",
|
|
45
|
+
version: "1",
|
|
46
|
+
support: ["1"],
|
|
47
|
+
});
|
|
48
|
+
});
|
|
49
|
+
this.ws.on("message", (data) => {
|
|
50
|
+
const raw = data.toString();
|
|
51
|
+
if (!raw)
|
|
52
|
+
return;
|
|
53
|
+
let msg;
|
|
54
|
+
try {
|
|
55
|
+
msg = JSON.parse(raw);
|
|
56
|
+
}
|
|
57
|
+
catch {
|
|
58
|
+
return;
|
|
59
|
+
}
|
|
60
|
+
this.handleDdpMessage(msg, resolve, reject);
|
|
61
|
+
});
|
|
62
|
+
this.ws.on("close", () => {
|
|
63
|
+
this.clearPingTimer();
|
|
64
|
+
// 清理所有等待中的调用,防止永久挂起
|
|
65
|
+
for (const [id, pending] of this.pendingCalls) {
|
|
66
|
+
pending.reject(new Error("WebSocket connection closed"));
|
|
67
|
+
}
|
|
68
|
+
this.pendingCalls.clear();
|
|
69
|
+
this.notifyConnection(false);
|
|
70
|
+
if (this.autoReconnect && !this.disposed) {
|
|
71
|
+
this.scheduleReconnect();
|
|
72
|
+
}
|
|
73
|
+
});
|
|
74
|
+
this.ws.on("error", (err) => {
|
|
75
|
+
if (!this.reconnecting) {
|
|
76
|
+
reject(err);
|
|
77
|
+
}
|
|
78
|
+
});
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
/** DDP 登录 */
|
|
82
|
+
async login(user, password) {
|
|
83
|
+
this.loginUser = user;
|
|
84
|
+
this.loginPassword = password;
|
|
85
|
+
const result = await this.callMethod("login", [
|
|
86
|
+
{
|
|
87
|
+
user: { username: user },
|
|
88
|
+
password: {
|
|
89
|
+
digest: await this.sha256(password),
|
|
90
|
+
algorithm: "sha-256",
|
|
91
|
+
},
|
|
92
|
+
},
|
|
93
|
+
]);
|
|
94
|
+
if (!result) {
|
|
95
|
+
throw new Error("Login failed: no result");
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
/** 断开连接 */
|
|
99
|
+
disconnect() {
|
|
100
|
+
this.disposed = true;
|
|
101
|
+
this.autoReconnect = false;
|
|
102
|
+
this.clearPingTimer();
|
|
103
|
+
if (this.reconnectTimer) {
|
|
104
|
+
clearTimeout(this.reconnectTimer);
|
|
105
|
+
this.reconnectTimer = null;
|
|
106
|
+
}
|
|
107
|
+
if (this.ws) {
|
|
108
|
+
this.ws.close();
|
|
109
|
+
this.ws = null;
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
// ----------------------------------------------------------
|
|
113
|
+
// 消息订阅
|
|
114
|
+
// ----------------------------------------------------------
|
|
115
|
+
/** 订阅房间消息 */
|
|
116
|
+
async subscribeRoomMessages(roomId) {
|
|
117
|
+
const subId = this.nextId();
|
|
118
|
+
return new Promise((resolve, reject) => {
|
|
119
|
+
this.pendingCalls.set(subId, {
|
|
120
|
+
resolve: () => {
|
|
121
|
+
this.subscriptions.set(subId, roomId);
|
|
122
|
+
resolve(subId);
|
|
123
|
+
},
|
|
124
|
+
reject,
|
|
125
|
+
});
|
|
126
|
+
this.sendRaw({
|
|
127
|
+
msg: "sub",
|
|
128
|
+
id: subId,
|
|
129
|
+
name: "stream-room-messages",
|
|
130
|
+
params: [roomId, false],
|
|
131
|
+
});
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
/** 取消订阅 */
|
|
135
|
+
async unsubscribe(subId) {
|
|
136
|
+
this.subscriptions.delete(subId);
|
|
137
|
+
this.sendRaw({ msg: "unsub", id: subId });
|
|
138
|
+
}
|
|
139
|
+
// ----------------------------------------------------------
|
|
140
|
+
// 事件监听
|
|
141
|
+
// ----------------------------------------------------------
|
|
142
|
+
/** 监听消息事件 */
|
|
143
|
+
onMessage(handler) {
|
|
144
|
+
this.messageHandlers.push(handler);
|
|
145
|
+
}
|
|
146
|
+
/** 监听连接状态变化 */
|
|
147
|
+
onConnectionChange(handler) {
|
|
148
|
+
this.connectionHandlers.push(handler);
|
|
149
|
+
}
|
|
150
|
+
// ----------------------------------------------------------
|
|
151
|
+
// 内部:DDP 协议处理
|
|
152
|
+
// ----------------------------------------------------------
|
|
153
|
+
handleDdpMessage(msg, connectResolve, connectReject) {
|
|
154
|
+
switch (msg.msg) {
|
|
155
|
+
case "connected":
|
|
156
|
+
this.reconnectAttempt = 0;
|
|
157
|
+
this.reconnecting = false;
|
|
158
|
+
this.startPingTimer();
|
|
159
|
+
this.notifyConnection(true);
|
|
160
|
+
connectResolve?.();
|
|
161
|
+
break;
|
|
162
|
+
case "ping":
|
|
163
|
+
this.sendRaw({ msg: "pong" });
|
|
164
|
+
break;
|
|
165
|
+
case "pong":
|
|
166
|
+
// 心跳响应,无需处理
|
|
167
|
+
break;
|
|
168
|
+
case "result": {
|
|
169
|
+
const pending = this.pendingCalls.get(msg.id || "");
|
|
170
|
+
if (pending) {
|
|
171
|
+
this.pendingCalls.delete(msg.id || "");
|
|
172
|
+
if (msg.error) {
|
|
173
|
+
pending.reject(new Error(typeof msg.error === "string"
|
|
174
|
+
? msg.error
|
|
175
|
+
: JSON.stringify(msg.error)));
|
|
176
|
+
}
|
|
177
|
+
else {
|
|
178
|
+
pending.resolve(msg.result);
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
break;
|
|
182
|
+
}
|
|
183
|
+
case "ready": {
|
|
184
|
+
// 订阅就绪
|
|
185
|
+
const subIds = msg.subs;
|
|
186
|
+
if (subIds) {
|
|
187
|
+
for (const subId of subIds) {
|
|
188
|
+
const pending = this.pendingCalls.get(subId);
|
|
189
|
+
if (pending) {
|
|
190
|
+
this.pendingCalls.delete(subId);
|
|
191
|
+
pending.resolve(undefined);
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
break;
|
|
196
|
+
}
|
|
197
|
+
case "changed": {
|
|
198
|
+
// 实时消息推送
|
|
199
|
+
if (msg.collection === "stream-room-messages" && msg.fields?.args) {
|
|
200
|
+
for (const arg of msg.fields.args) {
|
|
201
|
+
const rcMsg = arg;
|
|
202
|
+
if (rcMsg && rcMsg._id && rcMsg.rid) {
|
|
203
|
+
this.notifyMessage(rcMsg, rcMsg.rid);
|
|
204
|
+
}
|
|
205
|
+
}
|
|
206
|
+
}
|
|
207
|
+
break;
|
|
208
|
+
}
|
|
209
|
+
case "failed":
|
|
210
|
+
connectReject?.(new Error("DDP connection failed"));
|
|
211
|
+
break;
|
|
212
|
+
}
|
|
213
|
+
}
|
|
214
|
+
async callMethod(method, params) {
|
|
215
|
+
const id = this.nextId();
|
|
216
|
+
return new Promise((resolve, reject) => {
|
|
217
|
+
this.pendingCalls.set(id, { resolve, reject });
|
|
218
|
+
this.sendRaw({ msg: "method", method, id, params });
|
|
219
|
+
});
|
|
220
|
+
}
|
|
221
|
+
sendRaw(data) {
|
|
222
|
+
if (this.ws?.readyState === WebSocket.OPEN) {
|
|
223
|
+
this.ws.send(JSON.stringify(data));
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
nextId() {
|
|
227
|
+
return String(++this.messageCounter);
|
|
228
|
+
}
|
|
229
|
+
// ----------------------------------------------------------
|
|
230
|
+
// 内部:心跳
|
|
231
|
+
// ----------------------------------------------------------
|
|
232
|
+
startPingTimer() {
|
|
233
|
+
this.clearPingTimer();
|
|
234
|
+
this.pingTimer = setInterval(() => {
|
|
235
|
+
this.sendRaw({ msg: "ping" });
|
|
236
|
+
}, PING_TIMEOUT_MS);
|
|
237
|
+
}
|
|
238
|
+
clearPingTimer() {
|
|
239
|
+
if (this.pingTimer) {
|
|
240
|
+
clearInterval(this.pingTimer);
|
|
241
|
+
this.pingTimer = null;
|
|
242
|
+
}
|
|
243
|
+
}
|
|
244
|
+
// ----------------------------------------------------------
|
|
245
|
+
// 内部:自动重连
|
|
246
|
+
// ----------------------------------------------------------
|
|
247
|
+
scheduleReconnect() {
|
|
248
|
+
if (this.disposed || this.reconnecting)
|
|
249
|
+
return;
|
|
250
|
+
this.reconnecting = true;
|
|
251
|
+
const delay = Math.min(RECONNECT_MIN_MS * Math.pow(2, this.reconnectAttempt), RECONNECT_MAX_MS);
|
|
252
|
+
this.reconnectAttempt++;
|
|
253
|
+
this.reconnectTimer = setTimeout(async () => {
|
|
254
|
+
try {
|
|
255
|
+
await this.connect();
|
|
256
|
+
// 重新登录
|
|
257
|
+
if (this.loginUser && this.loginPassword) {
|
|
258
|
+
await this.login(this.loginUser, this.loginPassword);
|
|
259
|
+
}
|
|
260
|
+
// 重新订阅所有房间
|
|
261
|
+
const rooms = [...this.subscriptions.values()];
|
|
262
|
+
this.subscriptions.clear();
|
|
263
|
+
for (const roomId of rooms) {
|
|
264
|
+
await this.subscribeRoomMessages(roomId);
|
|
265
|
+
}
|
|
266
|
+
// 重连成功,重置状态
|
|
267
|
+
this.reconnecting = false;
|
|
268
|
+
this.reconnectAttempt = 0;
|
|
269
|
+
}
|
|
270
|
+
catch {
|
|
271
|
+
// 继续重试
|
|
272
|
+
this.reconnecting = false;
|
|
273
|
+
this.scheduleReconnect();
|
|
274
|
+
}
|
|
275
|
+
}, delay);
|
|
276
|
+
}
|
|
277
|
+
// ----------------------------------------------------------
|
|
278
|
+
// 内部:事件通知
|
|
279
|
+
// ----------------------------------------------------------
|
|
280
|
+
notifyMessage(msg, roomId) {
|
|
281
|
+
for (const handler of this.messageHandlers) {
|
|
282
|
+
try {
|
|
283
|
+
handler(msg, roomId);
|
|
284
|
+
}
|
|
285
|
+
catch {
|
|
286
|
+
// 忽略 handler 错误
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
}
|
|
290
|
+
notifyConnection(connected) {
|
|
291
|
+
for (const handler of this.connectionHandlers) {
|
|
292
|
+
try {
|
|
293
|
+
handler(connected);
|
|
294
|
+
}
|
|
295
|
+
catch {
|
|
296
|
+
// 忽略 handler 错误
|
|
297
|
+
}
|
|
298
|
+
}
|
|
299
|
+
}
|
|
300
|
+
// ----------------------------------------------------------
|
|
301
|
+
// 工具
|
|
302
|
+
// ----------------------------------------------------------
|
|
303
|
+
async sha256(text) {
|
|
304
|
+
const { createHash } = await import("node:crypto");
|
|
305
|
+
return createHash("sha256").update(text).digest("hex");
|
|
306
|
+
}
|
|
307
|
+
}
|
|
308
|
+
//# sourceMappingURL=ws-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ws-client.js","sourceRoot":"","sources":["../../src/rc-api/ws-client.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,kCAAkC;AAClC,cAAc;AACd,+DAA+D;AAE/D,OAAO,SAAS,MAAM,IAAI,CAAC;AAsB3B,WAAW;AACX,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAC9B,MAAM,gBAAgB,GAAG,KAAK,CAAC;AAC/B,MAAM,eAAe,GAAG,KAAK,CAAC;AAE9B,MAAM,OAAO,kBAAkB;IACrB,SAAS,CAAS;IAClB,EAAE,GAAqB,IAAI,CAAC;IAC5B,cAAc,GAAG,CAAC,CAAC;IACnB,YAAY,GAAG,IAAI,GAAG,EAG3B,CAAC;IACI,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,kBAAkB;IAC7D,eAAe,GAAqB,EAAE,CAAC;IACvC,kBAAkB,GAAwB,EAAE,CAAC;IAC7C,YAAY,GAAG,KAAK,CAAC;IACrB,gBAAgB,GAAG,CAAC,CAAC;IACrB,cAAc,GAAyC,IAAI,CAAC;IAC5D,SAAS,GAA0C,IAAI,CAAC;IACxD,QAAQ,GAAG,KAAK,CAAC;IAEjB,SAAS,GAAG,EAAE,CAAC;IACf,aAAa,GAAG,EAAE,CAAC;IACnB,aAAa,GAAG,IAAI,CAAC;IAE7B,YAAY,SAAiB;QAC3B,gBAAgB;QAChB,IAAI,CAAC,SAAS,GAAG,SAAS;aACvB,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC;aAC1B,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC;aACxB,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACzB,CAAC;IAED,6DAA6D;IAC7D,OAAO;IACP,6DAA6D;IAE7D,sBAAsB;IACtB,KAAK,CAAC,OAAO;QACX,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,KAAK,GAAG,GAAG,IAAI,CAAC,SAAS,YAAY,CAAC;YAE5C,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC;YAE/B,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACtB,SAAS;gBACT,IAAI,CAAC,OAAO,CAAC;oBACX,GAAG,EAAE,SAAS;oBACd,OAAO,EAAE,GAAG;oBACZ,OAAO,EAAE,CAAC,GAAG,CAAC;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC5B,IAAI,CAAC,GAAG;oBAAE,OAAO;gBAEjB,IAAI,GAAe,CAAC;gBACpB,IAAI,CAAC;oBACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACxB,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,oBAAoB;gBACpB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBAC9C,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC,CAAC;gBAC3D,CAAC;gBACD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;gBAC1B,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;gBAC7B,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACzC,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;oBACvB,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,aAAa;IACb,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,QAAgB;QACxC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE;YAC5C;gBACE,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;gBACxB,QAAQ,EAAE;oBACR,MAAM,EAAE,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;oBACnC,SAAS,EAAE,SAAS;iBACrB;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,WAAW;IACX,UAAU;QACR,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,OAAO;IACP,6DAA6D;IAE7D,aAAa;IACb,KAAK,CAAC,qBAAqB,CAAC,MAAc;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAE5B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE;gBAC3B,OAAO,EAAE,GAAG,EAAE;oBACZ,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;oBACtC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjB,CAAC;gBACD,MAAM;aACP,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC;gBACX,GAAG,EAAE,KAAK;gBACV,EAAE,EAAE,KAAK;gBACT,IAAI,EAAE,sBAAsB;gBAC5B,MAAM,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;aACxB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,WAAW;IACX,KAAK,CAAC,WAAW,CAAC,KAAa;QAC7B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,6DAA6D;IAC7D,OAAO;IACP,6DAA6D;IAE7D,aAAa;IACb,SAAS,CAAC,OAAuB;QAC/B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAED,eAAe;IACf,kBAAkB,CAAC,OAA0B;QAC3C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,6DAA6D;IAC7D,cAAc;IACd,6DAA6D;IAErD,gBAAgB,CACtB,GAAe,EACf,cAA0D,EAC1D,aAAyC;QAEzC,QAAQ,GAAG,CAAC,GAAG,EAAE,CAAC;YAChB,KAAK,WAAW;gBACd,IAAI,CAAC,gBAAgB,GAAG,CAAC,CAAC;gBAC1B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;gBAC5B,cAAc,EAAE,EAAE,CAAC;gBACnB,MAAM;YAER,KAAK,MAAM;gBACT,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC9B,MAAM;YAER,KAAK,MAAM;gBACT,YAAY;gBACZ,MAAM;YAER,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;gBACpD,IAAI,OAAO,EAAE,CAAC;oBACZ,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;oBACvC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;wBACd,OAAO,CAAC,MAAM,CACZ,IAAI,KAAK,CACP,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ;4BAC3B,CAAC,CAAC,GAAG,CAAC,KAAK;4BACX,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAC9B,CACF,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,OAAO;gBACP,MAAM,MAAM,GAAI,GAAqC,CAAC,IAAI,CAAC;gBAC3D,IAAI,MAAM,EAAE,CAAC;oBACX,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;wBAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;wBAC7C,IAAI,OAAO,EAAE,CAAC;4BACZ,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;4BAChC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;wBAC7B,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,SAAS;gBACT,IAAI,GAAG,CAAC,UAAU,KAAK,sBAAsB,IAAI,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;oBAClE,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;wBAClC,MAAM,KAAK,GAAG,GAAgB,CAAC;wBAC/B,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;4BACpC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;wBACvC,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;YAED,KAAK,QAAQ;gBACX,aAAa,EAAE,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBACpD,MAAM;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,MAAiB;QACxD,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC/C,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,OAAO,CAAC,IAAa;QAC3B,IAAI,IAAI,CAAC,EAAE,EAAE,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,MAAM;QACZ,OAAO,MAAM,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IACvC,CAAC;IAED,6DAA6D;IAC7D,QAAQ;IACR,6DAA6D;IAErD,cAAc;QACpB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;YAChC,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;QAChC,CAAC,EAAE,eAAe,CAAC,CAAC;IACtB,CAAC;IAEO,cAAc;QACpB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,UAAU;IACV,6DAA6D;IAErD,iBAAiB;QACvB,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC/C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,gBAAgB,CAAC,EACrD,gBAAgB,CACjB,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAExB,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YAC1C,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;gBACrB,OAAO;gBACP,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;gBACvD,CAAC;gBACD,WAAW;gBACX,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC/C,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;gBAC3B,KAAK,MAAM,MAAM,IAAI,KAAK,EAAE,CAAC;oBAC3B,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBAC3C,CAAC;gBACD,YAAY;gBACZ,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,IAAI,CAAC,gBAAgB,GAAG,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO;gBACP,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAED,6DAA6D;IAC7D,UAAU;IACV,6DAA6D;IAErD,aAAa,CAAC,GAAc,EAAE,MAAc;QAClD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,gBAAgB;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,SAAkB;QACzC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC9C,IAAI,CAAC;gBACH,OAAO,CAAC,SAAS,CAAC,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,gBAAgB;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,KAAK;IACL,6DAA6D;IAErD,KAAK,CAAC,MAAM,CAAC,IAAY;QAC/B,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACnD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACzD,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import { RocketChatRestClient } from "../rc-api/rest-client.js";
|
|
2
|
+
import type { RcMessage } from "../rc-api/types.js";
|
|
3
|
+
type IncomingMessageHandler = (message: RcMessage, roomId: string, botUsername: string, agentId: string) => void;
|
|
4
|
+
export declare class BotManager {
|
|
5
|
+
private serverUrl;
|
|
6
|
+
private bots;
|
|
7
|
+
private messageHandler;
|
|
8
|
+
private logger;
|
|
9
|
+
constructor(serverUrl: string, logger?: {
|
|
10
|
+
info: (msg: string) => void;
|
|
11
|
+
error: (msg: string) => void;
|
|
12
|
+
});
|
|
13
|
+
/** 设置消息回调 */
|
|
14
|
+
onMessage(handler: IncomingMessageHandler): void;
|
|
15
|
+
/** 添加并连接一个机器人 */
|
|
16
|
+
addBot(username: string, password: string, userId: string, agentId: string, displayName?: string): Promise<void>;
|
|
17
|
+
/** 订阅房间消息 */
|
|
18
|
+
subscribeRoom(botUsername: string, roomId: string): Promise<void>;
|
|
19
|
+
/** 通过机器人发送消息 */
|
|
20
|
+
sendMessage(botUsername: string, roomId: string, text: string): Promise<void>;
|
|
21
|
+
/** 通过机器人上传文件 */
|
|
22
|
+
uploadFile(botUsername: string, roomId: string, fileBuffer: Buffer, filename: string): Promise<void>;
|
|
23
|
+
/** 获取机器人的 REST 客户端 */
|
|
24
|
+
getRestClient(botUsername: string): RocketChatRestClient | undefined;
|
|
25
|
+
/** 获取所有已连接的机器人名 */
|
|
26
|
+
getConnectedBots(): string[];
|
|
27
|
+
/** 获取机器人绑定的 Agent ID */
|
|
28
|
+
getAgentId(botUsername: string): string | undefined;
|
|
29
|
+
/** 断开所有机器人 */
|
|
30
|
+
disconnectAll(): void;
|
|
31
|
+
private handleIncomingMessage;
|
|
32
|
+
}
|
|
33
|
+
export {};
|
|
34
|
+
//# sourceMappingURL=bot-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bot-manager.d.ts","sourceRoot":"","sources":["../../src/service/bot-manager.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAEhE,OAAO,KAAK,EAAE,SAAS,EAAkB,MAAM,oBAAoB,CAAC;AAapE,KAAK,sBAAsB,GAAG,CAC5B,OAAO,EAAE,SAAS,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,KACZ,IAAI,CAAC;AAEV,qBAAa,UAAU;IACrB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,IAAI,CAAoC;IAChD,OAAO,CAAC,cAAc,CAAuC;IAC7D,OAAO,CAAC,MAAM,CAAgE;gBAG5E,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE;QAAE,IAAI,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC;QAAC,KAAK,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAA;KAAE;IASxE,aAAa;IACb,SAAS,CAAC,OAAO,EAAE,sBAAsB,GAAG,IAAI;IAQhD,iBAAiB;IACX,MAAM,CACV,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAuDhB,aAAa;IACP,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBvE,gBAAgB;IACV,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,IAAI,CAAC;IAShB,gBAAgB;IACV,UAAU,CACd,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAShB,sBAAsB;IACtB,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,oBAAoB,GAAG,SAAS;IAIpE,mBAAmB;IACnB,gBAAgB,IAAI,MAAM,EAAE;IAM5B,wBAAwB;IACxB,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAInD,cAAc;IACd,aAAa,IAAI,IAAI;IAgBrB,OAAO,CAAC,qBAAqB;CA0B9B"}
|
|
@@ -0,0 +1,156 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// 多机器人连接管理器
|
|
3
|
+
// 每个机器人一个 WebSocket 连接,管理所有 DM 和群组的消息订阅
|
|
4
|
+
// ============================================================
|
|
5
|
+
import { RocketChatRestClient } from "../rc-api/rest-client.js";
|
|
6
|
+
import { RocketChatWsClient } from "../rc-api/ws-client.js";
|
|
7
|
+
export class BotManager {
|
|
8
|
+
serverUrl;
|
|
9
|
+
bots = new Map();
|
|
10
|
+
messageHandler = null;
|
|
11
|
+
logger;
|
|
12
|
+
constructor(serverUrl, logger) {
|
|
13
|
+
this.serverUrl = serverUrl;
|
|
14
|
+
this.logger = logger || {
|
|
15
|
+
info: (msg) => console.log(`[RC BotManager] ${msg}`),
|
|
16
|
+
error: (msg) => console.error(`[RC BotManager] ${msg}`),
|
|
17
|
+
};
|
|
18
|
+
}
|
|
19
|
+
/** 设置消息回调 */
|
|
20
|
+
onMessage(handler) {
|
|
21
|
+
this.messageHandler = handler;
|
|
22
|
+
}
|
|
23
|
+
// ----------------------------------------------------------
|
|
24
|
+
// 机器人生命周期
|
|
25
|
+
// ----------------------------------------------------------
|
|
26
|
+
/** 添加并连接一个机器人 */
|
|
27
|
+
async addBot(username, password, userId, agentId, displayName) {
|
|
28
|
+
if (this.bots.has(username)) {
|
|
29
|
+
this.logger.info(`机器人 ${username} 已连接,跳过`);
|
|
30
|
+
return;
|
|
31
|
+
}
|
|
32
|
+
const restClient = new RocketChatRestClient(this.serverUrl);
|
|
33
|
+
const wsClient = new RocketChatWsClient(this.serverUrl);
|
|
34
|
+
const conn = {
|
|
35
|
+
username,
|
|
36
|
+
displayName,
|
|
37
|
+
agentId,
|
|
38
|
+
restClient,
|
|
39
|
+
wsClient,
|
|
40
|
+
subscribedRooms: new Set(),
|
|
41
|
+
connected: false,
|
|
42
|
+
};
|
|
43
|
+
this.bots.set(username, conn);
|
|
44
|
+
// 连接 WebSocket
|
|
45
|
+
try {
|
|
46
|
+
await wsClient.connect();
|
|
47
|
+
await wsClient.login(username, password);
|
|
48
|
+
conn.connected = true;
|
|
49
|
+
// 同时设置 REST 客户端的认证
|
|
50
|
+
const authResult = await restClient.login(username, password);
|
|
51
|
+
restClient.setAuth(authResult.userId, authResult.authToken);
|
|
52
|
+
// 监听消息
|
|
53
|
+
wsClient.onMessage((msg, roomId) => {
|
|
54
|
+
this.handleIncomingMessage(msg, roomId, username, agentId);
|
|
55
|
+
});
|
|
56
|
+
// 监听连接状态
|
|
57
|
+
wsClient.onConnectionChange((connected) => {
|
|
58
|
+
conn.connected = connected;
|
|
59
|
+
if (connected) {
|
|
60
|
+
this.logger.info(`机器人 ${username} 已重新连接`);
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
this.logger.info(`机器人 ${username} 连接断开,等待重连...`);
|
|
64
|
+
}
|
|
65
|
+
});
|
|
66
|
+
this.logger.info(`机器人 ${username} 已连接`);
|
|
67
|
+
}
|
|
68
|
+
catch (err) {
|
|
69
|
+
this.logger.error(`机器人 ${username} 连接失败: ${err.message}`);
|
|
70
|
+
throw err;
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
/** 订阅房间消息 */
|
|
74
|
+
async subscribeRoom(botUsername, roomId) {
|
|
75
|
+
const conn = this.bots.get(botUsername);
|
|
76
|
+
if (!conn) {
|
|
77
|
+
throw new Error(`机器人 ${botUsername} 未找到`);
|
|
78
|
+
}
|
|
79
|
+
if (conn.subscribedRooms.has(roomId)) {
|
|
80
|
+
return; // 已订阅
|
|
81
|
+
}
|
|
82
|
+
try {
|
|
83
|
+
await conn.wsClient.subscribeRoomMessages(roomId);
|
|
84
|
+
conn.subscribedRooms.add(roomId);
|
|
85
|
+
this.logger.info(`${botUsername}: 已订阅房间 ${roomId}`);
|
|
86
|
+
}
|
|
87
|
+
catch (err) {
|
|
88
|
+
this.logger.error(`${botUsername}: 订阅房间 ${roomId} 失败: ${err.message}`);
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
/** 通过机器人发送消息 */
|
|
92
|
+
async sendMessage(botUsername, roomId, text) {
|
|
93
|
+
const conn = this.bots.get(botUsername);
|
|
94
|
+
if (!conn) {
|
|
95
|
+
throw new Error(`机器人 ${botUsername} 未找到`);
|
|
96
|
+
}
|
|
97
|
+
await conn.restClient.sendMessage(roomId, text);
|
|
98
|
+
}
|
|
99
|
+
/** 通过机器人上传文件 */
|
|
100
|
+
async uploadFile(botUsername, roomId, fileBuffer, filename) {
|
|
101
|
+
const conn = this.bots.get(botUsername);
|
|
102
|
+
if (!conn) {
|
|
103
|
+
throw new Error(`机器人 ${botUsername} 未找到`);
|
|
104
|
+
}
|
|
105
|
+
await conn.restClient.uploadFile(roomId, fileBuffer, filename);
|
|
106
|
+
}
|
|
107
|
+
/** 获取机器人的 REST 客户端 */
|
|
108
|
+
getRestClient(botUsername) {
|
|
109
|
+
return this.bots.get(botUsername)?.restClient;
|
|
110
|
+
}
|
|
111
|
+
/** 获取所有已连接的机器人名 */
|
|
112
|
+
getConnectedBots() {
|
|
113
|
+
return [...this.bots.entries()]
|
|
114
|
+
.filter(([, conn]) => conn.connected)
|
|
115
|
+
.map(([username]) => username);
|
|
116
|
+
}
|
|
117
|
+
/** 获取机器人绑定的 Agent ID */
|
|
118
|
+
getAgentId(botUsername) {
|
|
119
|
+
return this.bots.get(botUsername)?.agentId;
|
|
120
|
+
}
|
|
121
|
+
/** 断开所有机器人 */
|
|
122
|
+
disconnectAll() {
|
|
123
|
+
for (const [username, conn] of this.bots) {
|
|
124
|
+
try {
|
|
125
|
+
conn.wsClient.disconnect();
|
|
126
|
+
this.logger.info(`机器人 ${username} 已断开`);
|
|
127
|
+
}
|
|
128
|
+
catch {
|
|
129
|
+
// 忽略断开错误
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
this.bots.clear();
|
|
133
|
+
}
|
|
134
|
+
// ----------------------------------------------------------
|
|
135
|
+
// 内部:消息处理
|
|
136
|
+
// ----------------------------------------------------------
|
|
137
|
+
handleIncomingMessage(msg, roomId, botUsername, agentId) {
|
|
138
|
+
// 忽略机器人自己发的消息
|
|
139
|
+
if (msg.u?.username === botUsername) {
|
|
140
|
+
return;
|
|
141
|
+
}
|
|
142
|
+
// 忽略系统消息
|
|
143
|
+
if (msg.t) {
|
|
144
|
+
return;
|
|
145
|
+
}
|
|
146
|
+
if (this.messageHandler) {
|
|
147
|
+
try {
|
|
148
|
+
this.messageHandler(msg, roomId, botUsername, agentId);
|
|
149
|
+
}
|
|
150
|
+
catch (err) {
|
|
151
|
+
this.logger.error(`消息处理错误: ${err.message}`);
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
//# sourceMappingURL=bot-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bot-manager.js","sourceRoot":"","sources":["../../src/service/bot-manager.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,YAAY;AACZ,wCAAwC;AACxC,+DAA+D;AAE/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAqB5D,MAAM,OAAO,UAAU;IACb,SAAS,CAAS;IAClB,IAAI,GAAG,IAAI,GAAG,EAAyB,CAAC;IACxC,cAAc,GAAkC,IAAI,CAAC;IACrD,MAAM,CAAgE;IAE9E,YACE,SAAiB,EACjB,MAAsE;QAEtE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI;YACtB,IAAI,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,EAAE,CAAC;YAC5D,KAAK,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC;SAChE,CAAC;IACJ,CAAC;IAED,aAAa;IACb,SAAS,CAAC,OAA+B;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAED,6DAA6D;IAC7D,UAAU;IACV,6DAA6D;IAE7D,iBAAiB;IACjB,KAAK,CAAC,MAAM,CACV,QAAgB,EAChB,QAAgB,EAChB,MAAc,EACd,OAAe,EACf,WAAoB;QAEpB,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,SAAS,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5D,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAExD,MAAM,IAAI,GAAkB;YAC1B,QAAQ;YACR,WAAW;YACX,OAAO;YACP,UAAU;YACV,QAAQ;YACR,eAAe,EAAE,IAAI,GAAG,EAAE;YAC1B,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAE9B,eAAe;QACf,IAAI,CAAC;YACH,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzB,MAAM,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACzC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YAEtB,mBAAmB;YACnB,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC9D,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC;YAE5D,OAAO;YACP,QAAQ,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE;gBACjC,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC7D,CAAC,CAAC,CAAC;YAEH,SAAS;YACT,QAAQ,CAAC,kBAAkB,CAAC,CAAC,SAAS,EAAE,EAAE;gBACxC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;gBAC3B,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,QAAQ,CAAC,CAAC;gBAC5C,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,eAAe,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,MAAM,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,OAAO,QAAQ,UAAW,GAAa,CAAC,OAAO,EAAE,CAClD,CAAC;YACF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,aAAa;IACb,KAAK,CAAC,aAAa,CAAC,WAAmB,EAAE,MAAc;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,OAAO,WAAW,MAAM,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrC,OAAO,CAAC,MAAM;QAChB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YAClD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,WAAW,MAAM,EAAE,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,GAAG,WAAW,UAAU,MAAM,QAAS,GAAa,CAAC,OAAO,EAAE,CAC/D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,gBAAgB;IAChB,KAAK,CAAC,WAAW,CACf,WAAmB,EACnB,MAAc,EACd,IAAY;QAEZ,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,OAAO,WAAW,MAAM,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IAED,gBAAgB;IAChB,KAAK,CAAC,UAAU,CACd,WAAmB,EACnB,MAAc,EACd,UAAkB,EAClB,QAAgB;QAEhB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,OAAO,WAAW,MAAM,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED,sBAAsB;IACtB,aAAa,CAAC,WAAmB;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,UAAU,CAAC;IAChD,CAAC;IAED,mBAAmB;IACnB,gBAAgB;QACd,OAAO,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;aAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC;aACpC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,wBAAwB;IACxB,UAAU,CAAC,WAAmB;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,OAAO,CAAC;IAC7C,CAAC;IAED,cAAc;IACd,aAAa;QACX,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACzC,IAAI,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;gBAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,QAAQ,MAAM,CAAC,CAAC;YAC1C,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;IAED,6DAA6D;IAC7D,UAAU;IACV,6DAA6D;IAErD,qBAAqB,CAC3B,GAAc,EACd,MAAc,EACd,WAAmB,EACnB,OAAe;QAEf,cAAc;QACd,IAAI,GAAG,CAAC,CAAC,EAAE,QAAQ,KAAK,WAAW,EAAE,CAAC;YACpC,OAAO;QACT,CAAC;QAED,SAAS;QACT,IAAI,GAAG,CAAC,CAAC,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;YACzD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,WAAY,GAAa,CAAC,OAAO,EAAE,CACpC,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import { BotManager } from "./bot-manager.js";
|
|
2
|
+
import { MessageHandler } from "./message-handler.js";
|
|
3
|
+
import type { RocketchatChannelConfig, OpenClawBinding } from "../rc-api/types.js";
|
|
4
|
+
interface ChannelServiceOptions {
|
|
5
|
+
config: Partial<RocketchatChannelConfig>;
|
|
6
|
+
bindings?: OpenClawBinding[];
|
|
7
|
+
logger?: {
|
|
8
|
+
info: (msg: string) => void;
|
|
9
|
+
error: (msg: string) => void;
|
|
10
|
+
};
|
|
11
|
+
}
|
|
12
|
+
export declare class ChannelService {
|
|
13
|
+
private config;
|
|
14
|
+
private bindings;
|
|
15
|
+
private botManager;
|
|
16
|
+
private messageHandler;
|
|
17
|
+
private adminClient;
|
|
18
|
+
private logger;
|
|
19
|
+
private running;
|
|
20
|
+
constructor(options: ChannelServiceOptions);
|
|
21
|
+
/** 启动服务 */
|
|
22
|
+
start(): Promise<void>;
|
|
23
|
+
/** 停止服务 */
|
|
24
|
+
stop(): void;
|
|
25
|
+
/** 获取 MessageHandler(用于出站消息) */
|
|
26
|
+
getMessageHandler(): MessageHandler | null;
|
|
27
|
+
/** 获取 BotManager */
|
|
28
|
+
getBotManager(): BotManager | null;
|
|
29
|
+
private connectBots;
|
|
30
|
+
/** 订阅机器人的所有 DM */
|
|
31
|
+
private subscribeBotDMs;
|
|
32
|
+
private ensureGroups;
|
|
33
|
+
}
|
|
34
|
+
export {};
|
|
35
|
+
//# sourceMappingURL=channel.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"channel.d.ts","sourceRoot":"","sources":["../../src/service/channel.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAKtD,OAAO,KAAK,EAAE,uBAAuB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAEnF,UAAU,qBAAqB;IAC7B,MAAM,EAAE,OAAO,CAAC,uBAAuB,CAAC,CAAC;IACzC,QAAQ,CAAC,EAAE,eAAe,EAAE,CAAC;IAC7B,MAAM,CAAC,EAAE;QAAE,IAAI,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC;QAAC,KAAK,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAA;KAAE,CAAC;CACxE;AAED,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAmC;IACjD,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,UAAU,CAA2B;IAC7C,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,WAAW,CAAqC;IACxD,OAAO,CAAC,MAAM,CAAgE;IAC9E,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,EAAE,qBAAqB;IAa1C,WAAW;IACL,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA4D5B,WAAW;IACX,IAAI,IAAI,IAAI;IAQZ,gCAAgC;IAChC,iBAAiB,IAAI,cAAc,GAAG,IAAI;IAI1C,oBAAoB;IACpB,aAAa,IAAI,UAAU,GAAG,IAAI;YAQpB,WAAW;IAsCzB,kBAAkB;YACJ,eAAe;YAoBf,YAAY;CA4B3B"}
|