@oyasmi/pipiclaw 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/CHANGELOG.md +31 -0
- package/README.md +247 -0
- package/dist/agent.d.ts +18 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +938 -0
- package/dist/agent.js.map +1 -0
- package/dist/commands.d.ts +9 -0
- package/dist/commands.d.ts.map +1 -0
- package/dist/commands.js +45 -0
- package/dist/commands.js.map +1 -0
- package/dist/context.d.ts +139 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +432 -0
- package/dist/context.js.map +1 -0
- package/dist/delivery.d.ts +4 -0
- package/dist/delivery.d.ts.map +1 -0
- package/dist/delivery.js +221 -0
- package/dist/delivery.js.map +1 -0
- package/dist/dingtalk.d.ts +109 -0
- package/dist/dingtalk.d.ts.map +1 -0
- package/dist/dingtalk.js +655 -0
- package/dist/dingtalk.js.map +1 -0
- package/dist/events.d.ts +51 -0
- package/dist/events.d.ts.map +1 -0
- package/dist/events.js +287 -0
- package/dist/events.js.map +1 -0
- package/dist/log.d.ts +33 -0
- package/dist/log.d.ts.map +1 -0
- package/dist/log.js +188 -0
- package/dist/log.js.map +1 -0
- package/dist/main.d.ts +3 -0
- package/dist/main.d.ts.map +1 -0
- package/dist/main.js +298 -0
- package/dist/main.js.map +1 -0
- package/dist/paths.d.ts +8 -0
- package/dist/paths.d.ts.map +1 -0
- package/dist/paths.js +10 -0
- package/dist/paths.js.map +1 -0
- package/dist/sandbox.d.ts +34 -0
- package/dist/sandbox.d.ts.map +1 -0
- package/dist/sandbox.js +180 -0
- package/dist/sandbox.js.map +1 -0
- package/dist/shell-escape.d.ts +6 -0
- package/dist/shell-escape.d.ts.map +1 -0
- package/dist/shell-escape.js +8 -0
- package/dist/shell-escape.js.map +1 -0
- package/dist/store.d.ts +41 -0
- package/dist/store.d.ts.map +1 -0
- package/dist/store.js +110 -0
- package/dist/store.js.map +1 -0
- package/dist/tools/attach.d.ts +14 -0
- package/dist/tools/attach.d.ts.map +1 -0
- package/dist/tools/attach.js +35 -0
- package/dist/tools/attach.js.map +1 -0
- package/dist/tools/bash.d.ts +10 -0
- package/dist/tools/bash.d.ts.map +1 -0
- package/dist/tools/bash.js +78 -0
- package/dist/tools/bash.js.map +1 -0
- package/dist/tools/edit.d.ts +11 -0
- package/dist/tools/edit.d.ts.map +1 -0
- package/dist/tools/edit.js +129 -0
- package/dist/tools/edit.js.map +1 -0
- package/dist/tools/index.d.ts +5 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +15 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/read.d.ts +11 -0
- package/dist/tools/read.d.ts.map +1 -0
- package/dist/tools/read.js +132 -0
- package/dist/tools/read.js.map +1 -0
- package/dist/tools/truncate.d.ts +57 -0
- package/dist/tools/truncate.d.ts.map +1 -0
- package/dist/tools/truncate.js +184 -0
- package/dist/tools/truncate.js.map +1 -0
- package/dist/tools/write.d.ts +10 -0
- package/dist/tools/write.d.ts.map +1 -0
- package/dist/tools/write.js +31 -0
- package/dist/tools/write.js.map +1 -0
- package/package.json +54 -0
package/dist/delivery.js
ADDED
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
import * as log from "./log.js";
|
|
2
|
+
const MIN_UPDATE_INTERVAL_MS = 800;
|
|
3
|
+
class ChannelDeliveryController {
|
|
4
|
+
event;
|
|
5
|
+
bot;
|
|
6
|
+
store;
|
|
7
|
+
progressText = "";
|
|
8
|
+
mode = "progress";
|
|
9
|
+
desiredRevision = 0;
|
|
10
|
+
appliedRevision = 0;
|
|
11
|
+
running = false;
|
|
12
|
+
closed = false;
|
|
13
|
+
finalResponseDelivered = false;
|
|
14
|
+
progressWindowStartedAt = 0;
|
|
15
|
+
lastDeliveredAt = 0;
|
|
16
|
+
timer = null;
|
|
17
|
+
flushWaiters = [];
|
|
18
|
+
constructor(event, bot, store) {
|
|
19
|
+
this.event = event;
|
|
20
|
+
this.bot = bot;
|
|
21
|
+
this.store = store;
|
|
22
|
+
}
|
|
23
|
+
buildContext() {
|
|
24
|
+
return {
|
|
25
|
+
message: {
|
|
26
|
+
text: this.event.text,
|
|
27
|
+
rawText: this.event.text,
|
|
28
|
+
user: this.event.user,
|
|
29
|
+
userName: this.event.userName,
|
|
30
|
+
channel: this.event.channelId,
|
|
31
|
+
ts: this.event.ts,
|
|
32
|
+
},
|
|
33
|
+
channelName: this.event.channelId,
|
|
34
|
+
respond: async (text, shouldLog = true) => this.appendProgress(text, shouldLog),
|
|
35
|
+
respondPlain: async (text, shouldLog = true) => this.sendFinal(text, shouldLog),
|
|
36
|
+
replaceMessage: async (text) => this.replaceWithFinal(text),
|
|
37
|
+
respondInThread: async (text) => {
|
|
38
|
+
log.logInfo(`[thread] ${text.substring(0, 200)}`);
|
|
39
|
+
},
|
|
40
|
+
setTyping: async (_isTyping) => { },
|
|
41
|
+
setWorking: async (_working) => { },
|
|
42
|
+
deleteMessage: async () => this.silence(),
|
|
43
|
+
flush: async () => this.flush(),
|
|
44
|
+
close: async () => this.close(),
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
async appendProgress(text, shouldLog) {
|
|
48
|
+
if (this.closed || this.finalResponseDelivered || !text.trim())
|
|
49
|
+
return;
|
|
50
|
+
this.progressText = this.progressText ? `${this.progressText}\n\n${text}` : text;
|
|
51
|
+
if (this.progressWindowStartedAt === 0) {
|
|
52
|
+
this.progressWindowStartedAt = Date.now();
|
|
53
|
+
}
|
|
54
|
+
if (shouldLog) {
|
|
55
|
+
await this.store.logBotResponse(this.event.channelId, text, Date.now().toString());
|
|
56
|
+
}
|
|
57
|
+
this.mode = "progress";
|
|
58
|
+
this.bumpRevision(false);
|
|
59
|
+
}
|
|
60
|
+
async sendFinal(text, shouldLog) {
|
|
61
|
+
if (this.closed || this.finalResponseDelivered)
|
|
62
|
+
return this.finalResponseDelivered;
|
|
63
|
+
if (shouldLog) {
|
|
64
|
+
await this.store.logBotResponse(this.event.channelId, text, Date.now().toString());
|
|
65
|
+
}
|
|
66
|
+
const delivered = await this.bot.sendPlain(this.event.channelId, text);
|
|
67
|
+
if (!delivered) {
|
|
68
|
+
return false;
|
|
69
|
+
}
|
|
70
|
+
this.finalResponseDelivered = true;
|
|
71
|
+
this.mode = "finalize-existing";
|
|
72
|
+
this.bumpRevision(true);
|
|
73
|
+
return true;
|
|
74
|
+
}
|
|
75
|
+
async replaceWithFinal(text) {
|
|
76
|
+
if (this.closed || this.finalResponseDelivered)
|
|
77
|
+
return;
|
|
78
|
+
this.progressText = text;
|
|
79
|
+
this.mode = "finalize-with-fallback";
|
|
80
|
+
this.bumpRevision(true);
|
|
81
|
+
}
|
|
82
|
+
async silence() {
|
|
83
|
+
if (this.closed)
|
|
84
|
+
return;
|
|
85
|
+
this.finalResponseDelivered = true;
|
|
86
|
+
this.mode = "silent";
|
|
87
|
+
this.bumpRevision(true);
|
|
88
|
+
}
|
|
89
|
+
bumpRevision(forceImmediate) {
|
|
90
|
+
this.desiredRevision++;
|
|
91
|
+
this.schedule(forceImmediate);
|
|
92
|
+
}
|
|
93
|
+
schedule(forceImmediate) {
|
|
94
|
+
if (this.running)
|
|
95
|
+
return;
|
|
96
|
+
if (this.timer) {
|
|
97
|
+
clearTimeout(this.timer);
|
|
98
|
+
this.timer = null;
|
|
99
|
+
}
|
|
100
|
+
const delay = forceImmediate || this.mode !== "progress"
|
|
101
|
+
? 0
|
|
102
|
+
: Math.max(0, MIN_UPDATE_INTERVAL_MS -
|
|
103
|
+
(Date.now() - (this.lastDeliveredAt > 0 ? this.lastDeliveredAt : this.progressWindowStartedAt)));
|
|
104
|
+
if (delay === 0) {
|
|
105
|
+
void this.runSyncLoop();
|
|
106
|
+
return;
|
|
107
|
+
}
|
|
108
|
+
this.timer = setTimeout(() => {
|
|
109
|
+
this.timer = null;
|
|
110
|
+
void this.runSyncLoop();
|
|
111
|
+
}, delay);
|
|
112
|
+
}
|
|
113
|
+
async runSyncLoop() {
|
|
114
|
+
if (this.running)
|
|
115
|
+
return;
|
|
116
|
+
this.running = true;
|
|
117
|
+
try {
|
|
118
|
+
while (this.appliedRevision < this.desiredRevision) {
|
|
119
|
+
const mode = this.mode;
|
|
120
|
+
const throttleBaseAt = this.lastDeliveredAt > 0 ? this.lastDeliveredAt : this.progressWindowStartedAt;
|
|
121
|
+
if (mode === "progress" && throttleBaseAt > 0) {
|
|
122
|
+
const remaining = MIN_UPDATE_INTERVAL_MS - (Date.now() - throttleBaseAt);
|
|
123
|
+
if (remaining > 0) {
|
|
124
|
+
this.timer = setTimeout(() => {
|
|
125
|
+
this.timer = null;
|
|
126
|
+
void this.runSyncLoop();
|
|
127
|
+
}, remaining);
|
|
128
|
+
return;
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
const revision = this.desiredRevision;
|
|
132
|
+
const content = this.progressText.trim();
|
|
133
|
+
let touchedRemote = false;
|
|
134
|
+
try {
|
|
135
|
+
if (mode === "progress") {
|
|
136
|
+
if (content) {
|
|
137
|
+
touchedRemote = await this.bot.streamToCard(this.event.channelId, this.progressText);
|
|
138
|
+
if (!touchedRemote) {
|
|
139
|
+
this.bot.discardCard(this.event.channelId);
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
else if (mode === "finalize-existing") {
|
|
144
|
+
if (content) {
|
|
145
|
+
touchedRemote = await this.bot.finalizeExistingCard(this.event.channelId, this.progressText);
|
|
146
|
+
if (!touchedRemote) {
|
|
147
|
+
this.bot.discardCard(this.event.channelId);
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
else {
|
|
151
|
+
this.bot.discardCard(this.event.channelId);
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
else if (mode === "finalize-with-fallback") {
|
|
155
|
+
if (content) {
|
|
156
|
+
touchedRemote = await this.bot.finalizeCard(this.event.channelId, this.progressText);
|
|
157
|
+
if (!touchedRemote) {
|
|
158
|
+
this.bot.discardCard(this.event.channelId);
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
else {
|
|
162
|
+
this.bot.discardCard(this.event.channelId);
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
else if (mode === "silent") {
|
|
166
|
+
this.bot.discardCard(this.event.channelId);
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
catch (err) {
|
|
170
|
+
log.logWarning(`[${this.event.channelId}] Delivery sync failed`, err instanceof Error ? err.message : String(err));
|
|
171
|
+
this.bot.discardCard(this.event.channelId);
|
|
172
|
+
}
|
|
173
|
+
if (touchedRemote) {
|
|
174
|
+
this.lastDeliveredAt = Date.now();
|
|
175
|
+
}
|
|
176
|
+
if (mode !== "progress" || touchedRemote) {
|
|
177
|
+
this.progressWindowStartedAt = 0;
|
|
178
|
+
}
|
|
179
|
+
this.appliedRevision = revision;
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
finally {
|
|
183
|
+
this.running = false;
|
|
184
|
+
this.resolveFlushWaiters();
|
|
185
|
+
if (this.appliedRevision < this.desiredRevision && !this.timer) {
|
|
186
|
+
this.schedule(false);
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
}
|
|
190
|
+
isSettled() {
|
|
191
|
+
return !this.running && !this.timer && this.appliedRevision >= this.desiredRevision;
|
|
192
|
+
}
|
|
193
|
+
resolveFlushWaiters() {
|
|
194
|
+
if (!this.isSettled())
|
|
195
|
+
return;
|
|
196
|
+
const waiters = this.flushWaiters;
|
|
197
|
+
this.flushWaiters = [];
|
|
198
|
+
for (const resolve of waiters) {
|
|
199
|
+
resolve();
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
async flush() {
|
|
203
|
+
if (this.isSettled())
|
|
204
|
+
return;
|
|
205
|
+
await new Promise((resolve) => {
|
|
206
|
+
this.flushWaiters.push(resolve);
|
|
207
|
+
});
|
|
208
|
+
}
|
|
209
|
+
async close() {
|
|
210
|
+
if (this.closed) {
|
|
211
|
+
await this.flush();
|
|
212
|
+
return;
|
|
213
|
+
}
|
|
214
|
+
this.closed = true;
|
|
215
|
+
await this.flush();
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
export function createDingTalkContext(event, bot, store) {
|
|
219
|
+
return new ChannelDeliveryController(event, bot, store).buildContext();
|
|
220
|
+
}
|
|
221
|
+
//# sourceMappingURL=delivery.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delivery.js","sourceRoot":"","sources":["../src/delivery.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,GAAG,MAAM,UAAU,CAAC;AAGhC,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAInC,MAAM,yBAAyB;IAcrB,KAAK;IACL,GAAG;IACH,KAAK;IAfN,YAAY,GAAG,EAAE,CAAC;IAClB,IAAI,GAAiB,UAAU,CAAC;IAChC,eAAe,GAAG,CAAC,CAAC;IACpB,eAAe,GAAG,CAAC,CAAC;IACpB,OAAO,GAAG,KAAK,CAAC;IAChB,MAAM,GAAG,KAAK,CAAC;IACf,sBAAsB,GAAG,KAAK,CAAC;IAC/B,uBAAuB,GAAG,CAAC,CAAC;IAC5B,eAAe,GAAG,CAAC,CAAC;IACpB,KAAK,GAA0B,IAAI,CAAC;IACpC,YAAY,GAAsB,EAAE,CAAC;IAE7C,YACS,KAAoB,EACpB,GAAgB,EAChB,KAAmB,EAC1B;qBAHO,KAAK;mBACL,GAAG;qBACH,KAAK;IACX,CAAC;IAEJ,YAAY,GAAoB;QAC/B,OAAO;YACN,OAAO,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACxB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;gBAC7B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS;gBAC7B,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;aACjB;YACD,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS;YACjC,OAAO,EAAE,KAAK,EAAE,IAAY,EAAE,SAAS,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,SAAS,CAAC;YACvF,YAAY,EAAE,KAAK,EAAE,IAAY,EAAE,SAAS,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC;YACvF,cAAc,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACnE,eAAe,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE,CAAC;gBACxC,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YAAA,CAClD;YACD,SAAS,EAAE,KAAK,EAAE,SAAkB,EAAE,EAAE,CAAC,EAAC,CAAC;YAC3C,UAAU,EAAE,KAAK,EAAE,QAAiB,EAAE,EAAE,CAAC,EAAC,CAAC;YAC3C,aAAa,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE;YACzC,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE;YAC/B,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE;SAC/B,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,cAAc,CAAC,IAAY,EAAE,SAAkB,EAAiB;QAC7E,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,sBAAsB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,OAAO;QAEvE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACjF,IAAI,IAAI,CAAC,uBAAuB,KAAK,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3C,CAAC;QACD,IAAI,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC;QACvB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAAA,CACzB;IAEO,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,SAAkB,EAAoB;QAC3E,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,sBAAsB;YAAE,OAAO,IAAI,CAAC,sBAAsB,CAAC;QAEnF,IAAI,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACvE,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACd,CAAC;QAED,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,mBAAmB,CAAC;QAChC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACxB,OAAO,IAAI,CAAC;IAAA,CACZ;IAEO,KAAK,CAAC,gBAAgB,CAAC,IAAY,EAAiB;QAC3D,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,sBAAsB;YAAE,OAAO;QAEvD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;QACrC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAAA,CACxB;IAEO,KAAK,CAAC,OAAO,GAAkB;QACtC,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QAExB,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC;QACrB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAAA,CACxB;IAEO,YAAY,CAAC,cAAuB,EAAQ;QACnD,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IAAA,CAC9B;IAEO,QAAQ,CAAC,cAAuB,EAAQ;QAC/C,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QAEzB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GACV,cAAc,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU;YACzC,CAAC,CAAC,CAAC;YACH,CAAC,CAAC,IAAI,CAAC,GAAG,CACR,CAAC,EACD,sBAAsB;gBACrB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAChG,CAAC;QAEL,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YACjB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;YACxB,OAAO;QACR,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;QAAA,CACxB,EAAE,KAAK,CAAC,CAAC;IAAA,CACV;IAEO,KAAK,CAAC,WAAW,GAAkB;QAC1C,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;gBACpD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvB,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC;gBACtG,IAAI,IAAI,KAAK,UAAU,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;oBAC/C,MAAM,SAAS,GAAG,sBAAsB,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,CAAC,CAAC;oBACzE,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;wBACnB,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;4BAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;4BAClB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;wBAAA,CACxB,EAAE,SAAS,CAAC,CAAC;wBACd,OAAO;oBACR,CAAC;gBACF,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC;gBACtC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;gBACzC,IAAI,aAAa,GAAG,KAAK,CAAC;gBAE1B,IAAI,CAAC;oBACJ,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;wBACzB,IAAI,OAAO,EAAE,CAAC;4BACb,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;4BACrF,IAAI,CAAC,aAAa,EAAE,CAAC;gCACpB,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BAC5C,CAAC;wBACF,CAAC;oBACF,CAAC;yBAAM,IAAI,IAAI,KAAK,mBAAmB,EAAE,CAAC;wBACzC,IAAI,OAAO,EAAE,CAAC;4BACb,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;4BAC7F,IAAI,CAAC,aAAa,EAAE,CAAC;gCACpB,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BAC5C,CAAC;wBACF,CAAC;6BAAM,CAAC;4BACP,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;wBAC5C,CAAC;oBACF,CAAC;yBAAM,IAAI,IAAI,KAAK,wBAAwB,EAAE,CAAC;wBAC9C,IAAI,OAAO,EAAE,CAAC;4BACb,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;4BACrF,IAAI,CAAC,aAAa,EAAE,CAAC;gCACpB,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BAC5C,CAAC;wBACF,CAAC;6BAAM,CAAC;4BACP,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;wBAC5C,CAAC;oBACF,CAAC;yBAAM,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;wBAC9B,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;oBAC5C,CAAC;gBACF,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACd,GAAG,CAAC,UAAU,CACb,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,wBAAwB,EAChD,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAChD,CAAC;oBACF,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC;gBAED,IAAI,aAAa,EAAE,CAAC;oBACnB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACnC,CAAC;gBACD,IAAI,IAAI,KAAK,UAAU,IAAI,aAAa,EAAE,CAAC;oBAC1C,IAAI,CAAC,uBAAuB,GAAG,CAAC,CAAC;gBAClC,CAAC;gBACD,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;YACjC,CAAC;QACF,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;YACrB,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAE3B,IAAI,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChE,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACF,CAAC;IAAA,CACD;IAEO,SAAS,GAAY;QAC5B,OAAO,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;IAAA,CACpF;IAEO,mBAAmB,GAAS;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;QAClC,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACX,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,KAAK,GAAkB;QACpC,IAAI,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAC7B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAAA,CAChC,CAAC,CAAC;IAAA,CACH;IAEO,KAAK,CAAC,KAAK,GAAkB;QACpC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,OAAO;QACR,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IAAA,CACnB;CACD;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAoB,EAAE,GAAgB,EAAE,KAAmB,EAAmB;IACnH,OAAO,IAAI,yBAAyB,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC,YAAY,EAAE,CAAC;AAAA,CACvE","sourcesContent":["import type { DingTalkBot, DingTalkContext, DingTalkEvent } from \"./dingtalk.js\";\nimport * as log from \"./log.js\";\nimport type { ChannelStore } from \"./store.js\";\n\nconst MIN_UPDATE_INTERVAL_MS = 800;\n\ntype DeliveryMode = \"progress\" | \"finalize-existing\" | \"finalize-with-fallback\" | \"silent\";\n\nclass ChannelDeliveryController {\n\tprivate progressText = \"\";\n\tprivate mode: DeliveryMode = \"progress\";\n\tprivate desiredRevision = 0;\n\tprivate appliedRevision = 0;\n\tprivate running = false;\n\tprivate closed = false;\n\tprivate finalResponseDelivered = false;\n\tprivate progressWindowStartedAt = 0;\n\tprivate lastDeliveredAt = 0;\n\tprivate timer: NodeJS.Timeout | null = null;\n\tprivate flushWaiters: Array<() => void> = [];\n\n\tconstructor(\n\t\tprivate event: DingTalkEvent,\n\t\tprivate bot: DingTalkBot,\n\t\tprivate store: ChannelStore,\n\t) {}\n\n\tbuildContext(): DingTalkContext {\n\t\treturn {\n\t\t\tmessage: {\n\t\t\t\ttext: this.event.text,\n\t\t\t\trawText: this.event.text,\n\t\t\t\tuser: this.event.user,\n\t\t\t\tuserName: this.event.userName,\n\t\t\t\tchannel: this.event.channelId,\n\t\t\t\tts: this.event.ts,\n\t\t\t},\n\t\t\tchannelName: this.event.channelId,\n\t\t\trespond: async (text: string, shouldLog = true) => this.appendProgress(text, shouldLog),\n\t\t\trespondPlain: async (text: string, shouldLog = true) => this.sendFinal(text, shouldLog),\n\t\t\treplaceMessage: async (text: string) => this.replaceWithFinal(text),\n\t\t\trespondInThread: async (text: string) => {\n\t\t\t\tlog.logInfo(`[thread] ${text.substring(0, 200)}`);\n\t\t\t},\n\t\t\tsetTyping: async (_isTyping: boolean) => {},\n\t\t\tsetWorking: async (_working: boolean) => {},\n\t\t\tdeleteMessage: async () => this.silence(),\n\t\t\tflush: async () => this.flush(),\n\t\t\tclose: async () => this.close(),\n\t\t};\n\t}\n\n\tprivate async appendProgress(text: string, shouldLog: boolean): Promise<void> {\n\t\tif (this.closed || this.finalResponseDelivered || !text.trim()) return;\n\n\t\tthis.progressText = this.progressText ? `${this.progressText}\\n\\n${text}` : text;\n\t\tif (this.progressWindowStartedAt === 0) {\n\t\t\tthis.progressWindowStartedAt = Date.now();\n\t\t}\n\t\tif (shouldLog) {\n\t\t\tawait this.store.logBotResponse(this.event.channelId, text, Date.now().toString());\n\t\t}\n\n\t\tthis.mode = \"progress\";\n\t\tthis.bumpRevision(false);\n\t}\n\n\tprivate async sendFinal(text: string, shouldLog: boolean): Promise<boolean> {\n\t\tif (this.closed || this.finalResponseDelivered) return this.finalResponseDelivered;\n\n\t\tif (shouldLog) {\n\t\t\tawait this.store.logBotResponse(this.event.channelId, text, Date.now().toString());\n\t\t}\n\n\t\tconst delivered = await this.bot.sendPlain(this.event.channelId, text);\n\t\tif (!delivered) {\n\t\t\treturn false;\n\t\t}\n\n\t\tthis.finalResponseDelivered = true;\n\t\tthis.mode = \"finalize-existing\";\n\t\tthis.bumpRevision(true);\n\t\treturn true;\n\t}\n\n\tprivate async replaceWithFinal(text: string): Promise<void> {\n\t\tif (this.closed || this.finalResponseDelivered) return;\n\n\t\tthis.progressText = text;\n\t\tthis.mode = \"finalize-with-fallback\";\n\t\tthis.bumpRevision(true);\n\t}\n\n\tprivate async silence(): Promise<void> {\n\t\tif (this.closed) return;\n\n\t\tthis.finalResponseDelivered = true;\n\t\tthis.mode = \"silent\";\n\t\tthis.bumpRevision(true);\n\t}\n\n\tprivate bumpRevision(forceImmediate: boolean): void {\n\t\tthis.desiredRevision++;\n\t\tthis.schedule(forceImmediate);\n\t}\n\n\tprivate schedule(forceImmediate: boolean): void {\n\t\tif (this.running) return;\n\n\t\tif (this.timer) {\n\t\t\tclearTimeout(this.timer);\n\t\t\tthis.timer = null;\n\t\t}\n\n\t\tconst delay =\n\t\t\tforceImmediate || this.mode !== \"progress\"\n\t\t\t\t? 0\n\t\t\t\t: Math.max(\n\t\t\t\t\t\t0,\n\t\t\t\t\t\tMIN_UPDATE_INTERVAL_MS -\n\t\t\t\t\t\t\t(Date.now() - (this.lastDeliveredAt > 0 ? this.lastDeliveredAt : this.progressWindowStartedAt)),\n\t\t\t\t\t);\n\n\t\tif (delay === 0) {\n\t\t\tvoid this.runSyncLoop();\n\t\t\treturn;\n\t\t}\n\n\t\tthis.timer = setTimeout(() => {\n\t\t\tthis.timer = null;\n\t\t\tvoid this.runSyncLoop();\n\t\t}, delay);\n\t}\n\n\tprivate async runSyncLoop(): Promise<void> {\n\t\tif (this.running) return;\n\t\tthis.running = true;\n\n\t\ttry {\n\t\t\twhile (this.appliedRevision < this.desiredRevision) {\n\t\t\t\tconst mode = this.mode;\n\t\t\t\tconst throttleBaseAt = this.lastDeliveredAt > 0 ? this.lastDeliveredAt : this.progressWindowStartedAt;\n\t\t\t\tif (mode === \"progress\" && throttleBaseAt > 0) {\n\t\t\t\t\tconst remaining = MIN_UPDATE_INTERVAL_MS - (Date.now() - throttleBaseAt);\n\t\t\t\t\tif (remaining > 0) {\n\t\t\t\t\t\tthis.timer = setTimeout(() => {\n\t\t\t\t\t\t\tthis.timer = null;\n\t\t\t\t\t\t\tvoid this.runSyncLoop();\n\t\t\t\t\t\t}, remaining);\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tconst revision = this.desiredRevision;\n\t\t\t\tconst content = this.progressText.trim();\n\t\t\t\tlet touchedRemote = false;\n\n\t\t\t\ttry {\n\t\t\t\t\tif (mode === \"progress\") {\n\t\t\t\t\t\tif (content) {\n\t\t\t\t\t\t\ttouchedRemote = await this.bot.streamToCard(this.event.channelId, this.progressText);\n\t\t\t\t\t\t\tif (!touchedRemote) {\n\t\t\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (mode === \"finalize-existing\") {\n\t\t\t\t\t\tif (content) {\n\t\t\t\t\t\t\ttouchedRemote = await this.bot.finalizeExistingCard(this.event.channelId, this.progressText);\n\t\t\t\t\t\t\tif (!touchedRemote) {\n\t\t\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (mode === \"finalize-with-fallback\") {\n\t\t\t\t\t\tif (content) {\n\t\t\t\t\t\t\ttouchedRemote = await this.bot.finalizeCard(this.event.channelId, this.progressText);\n\t\t\t\t\t\t\tif (!touchedRemote) {\n\t\t\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (mode === \"silent\") {\n\t\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t\t}\n\t\t\t\t} catch (err) {\n\t\t\t\t\tlog.logWarning(\n\t\t\t\t\t\t`[${this.event.channelId}] Delivery sync failed`,\n\t\t\t\t\t\terr instanceof Error ? err.message : String(err),\n\t\t\t\t\t);\n\t\t\t\t\tthis.bot.discardCard(this.event.channelId);\n\t\t\t\t}\n\n\t\t\t\tif (touchedRemote) {\n\t\t\t\t\tthis.lastDeliveredAt = Date.now();\n\t\t\t\t}\n\t\t\t\tif (mode !== \"progress\" || touchedRemote) {\n\t\t\t\t\tthis.progressWindowStartedAt = 0;\n\t\t\t\t}\n\t\t\t\tthis.appliedRevision = revision;\n\t\t\t}\n\t\t} finally {\n\t\t\tthis.running = false;\n\t\t\tthis.resolveFlushWaiters();\n\n\t\t\tif (this.appliedRevision < this.desiredRevision && !this.timer) {\n\t\t\t\tthis.schedule(false);\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate isSettled(): boolean {\n\t\treturn !this.running && !this.timer && this.appliedRevision >= this.desiredRevision;\n\t}\n\n\tprivate resolveFlushWaiters(): void {\n\t\tif (!this.isSettled()) return;\n\t\tconst waiters = this.flushWaiters;\n\t\tthis.flushWaiters = [];\n\t\tfor (const resolve of waiters) {\n\t\t\tresolve();\n\t\t}\n\t}\n\n\tprivate async flush(): Promise<void> {\n\t\tif (this.isSettled()) return;\n\t\tawait new Promise<void>((resolve) => {\n\t\t\tthis.flushWaiters.push(resolve);\n\t\t});\n\t}\n\n\tprivate async close(): Promise<void> {\n\t\tif (this.closed) {\n\t\t\tawait this.flush();\n\t\t\treturn;\n\t\t}\n\n\t\tthis.closed = true;\n\t\tawait this.flush();\n\t}\n}\n\nexport function createDingTalkContext(event: DingTalkEvent, bot: DingTalkBot, store: ChannelStore): DingTalkContext {\n\treturn new ChannelDeliveryController(event, bot, store).buildContext();\n}\n"]}
|
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
export interface DingTalkConfig {
|
|
2
|
+
clientId: string;
|
|
3
|
+
clientSecret: string;
|
|
4
|
+
robotCode?: string;
|
|
5
|
+
cardTemplateId?: string;
|
|
6
|
+
cardTemplateKey?: string;
|
|
7
|
+
allowFrom?: string[];
|
|
8
|
+
stateDir?: string;
|
|
9
|
+
}
|
|
10
|
+
export interface DingTalkEvent {
|
|
11
|
+
type: "dm" | "group";
|
|
12
|
+
channelId: string;
|
|
13
|
+
ts: string;
|
|
14
|
+
user: string;
|
|
15
|
+
userName: string;
|
|
16
|
+
text: string;
|
|
17
|
+
conversationId: string;
|
|
18
|
+
conversationType: string;
|
|
19
|
+
}
|
|
20
|
+
export interface DingTalkContext {
|
|
21
|
+
message: {
|
|
22
|
+
text: string;
|
|
23
|
+
rawText: string;
|
|
24
|
+
user: string;
|
|
25
|
+
userName?: string;
|
|
26
|
+
channel: string;
|
|
27
|
+
ts: string;
|
|
28
|
+
};
|
|
29
|
+
channelName?: string;
|
|
30
|
+
respond: (text: string, shouldLog?: boolean) => Promise<void>;
|
|
31
|
+
respondPlain: (text: string, shouldLog?: boolean) => Promise<boolean>;
|
|
32
|
+
replaceMessage: (text: string) => Promise<void>;
|
|
33
|
+
respondInThread: (text: string) => Promise<void>;
|
|
34
|
+
setTyping: (isTyping: boolean) => Promise<void>;
|
|
35
|
+
setWorking: (working: boolean) => Promise<void>;
|
|
36
|
+
deleteMessage: () => Promise<void>;
|
|
37
|
+
flush: () => Promise<void>;
|
|
38
|
+
close: () => Promise<void>;
|
|
39
|
+
}
|
|
40
|
+
export interface DingTalkHandler {
|
|
41
|
+
isRunning(channelId: string): boolean;
|
|
42
|
+
handleEvent(event: DingTalkEvent, bot: DingTalkBot, isEvent?: boolean): Promise<void>;
|
|
43
|
+
handleStop(channelId: string, bot: DingTalkBot): Promise<void>;
|
|
44
|
+
}
|
|
45
|
+
export declare class DingTalkBot {
|
|
46
|
+
private handler;
|
|
47
|
+
private config;
|
|
48
|
+
private accessToken;
|
|
49
|
+
private tokenExpiry;
|
|
50
|
+
private activeCards;
|
|
51
|
+
private convMeta;
|
|
52
|
+
private queues;
|
|
53
|
+
private client;
|
|
54
|
+
private lastSocketAvailableTime;
|
|
55
|
+
private activeMessageProcessing;
|
|
56
|
+
private keepAliveTimer;
|
|
57
|
+
private isReconnecting;
|
|
58
|
+
private isStopped;
|
|
59
|
+
private reconnectAttempts;
|
|
60
|
+
private processedIds;
|
|
61
|
+
private processedIdsOrder;
|
|
62
|
+
constructor(handler: DingTalkHandler, config: DingTalkConfig);
|
|
63
|
+
/**
|
|
64
|
+
* Mark an ID as processed. Returns true if this is a new ID, false if already seen.
|
|
65
|
+
* Maintains a FIFO buffer of at most 200 entries.
|
|
66
|
+
*/
|
|
67
|
+
private markProcessed;
|
|
68
|
+
start(): Promise<void>;
|
|
69
|
+
private handleRawMessage;
|
|
70
|
+
private doReconnect;
|
|
71
|
+
stop(): void;
|
|
72
|
+
/**
|
|
73
|
+
* Enqueue an event for processing.
|
|
74
|
+
* Returns true if enqueued, false if queue is full (max 5).
|
|
75
|
+
*/
|
|
76
|
+
enqueueEvent(event: DingTalkEvent): boolean;
|
|
77
|
+
/**
|
|
78
|
+
* Get or create an AI Card for a channel.
|
|
79
|
+
*/
|
|
80
|
+
ensureCard(channelId: string): Promise<void>;
|
|
81
|
+
/**
|
|
82
|
+
* Stream content to the active AI Card for a channel.
|
|
83
|
+
*/
|
|
84
|
+
streamToCard(channelId: string, content: string, finalize?: boolean): Promise<boolean>;
|
|
85
|
+
/**
|
|
86
|
+
* Finalize the active card for a channel without falling back to a plain message.
|
|
87
|
+
* Returns true if a card was finalized, false if no active card existed.
|
|
88
|
+
*/
|
|
89
|
+
finalizeExistingCard(channelId: string, content: string): Promise<boolean>;
|
|
90
|
+
/**
|
|
91
|
+
* Finalize and remove the active card for a channel.
|
|
92
|
+
*/
|
|
93
|
+
finalizeCard(channelId: string, content: string): Promise<boolean>;
|
|
94
|
+
discardCard(channelId: string): void;
|
|
95
|
+
/**
|
|
96
|
+
* Send a normal message natively mapping DM and Group to correct endpoints (fallback when no card).
|
|
97
|
+
*/
|
|
98
|
+
sendPlain(channelId: string, text: string): Promise<boolean>;
|
|
99
|
+
private createCard;
|
|
100
|
+
private streamCard;
|
|
101
|
+
private getAccessToken;
|
|
102
|
+
private extractContent;
|
|
103
|
+
private onStreamMessage;
|
|
104
|
+
private getQueue;
|
|
105
|
+
private getConversationMeta;
|
|
106
|
+
private setConversationMeta;
|
|
107
|
+
private getConversationMetaPath;
|
|
108
|
+
}
|
|
109
|
+
//# sourceMappingURL=dingtalk.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dingtalk.d.ts","sourceRoot":"","sources":["../src/dingtalk.ts"],"names":[],"mappings":"AAmBA,MAAM,WAAW,cAAc;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,aAAa;IAC7B,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC/B,OAAO,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,EAAE,EAAE,MAAM,CAAC;KACX,CAAC;IACF,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,YAAY,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IACtE,cAAc,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,SAAS,EAAE,CAAC,QAAQ,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,UAAU,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IACnC,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3B,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B;AAED,MAAM,WAAW,eAAe;IAC/B,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC;IACtC,WAAW,CAAC,KAAK,EAAE,aAAa,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtF,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC/D;AAmED,qBAAa,WAAW;IACvB,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,MAAM,CAAiB;IAG/B,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,WAAW,CAAK;IAGxB,OAAO,CAAC,WAAW,CAA6B;IAGhD,OAAO,CAAC,QAAQ,CAAuC;IAGvD,OAAO,CAAC,MAAM,CAAmC;IAGjD,OAAO,CAAC,MAAM,CAAyB;IACvC,OAAO,CAAC,uBAAuB,CAAc;IAC7C,OAAO,CAAC,uBAAuB,CAAS;IACxC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,iBAAiB,CAAK;IAG9B,OAAO,CAAC,YAAY,CAAqB;IACzC,OAAO,CAAC,iBAAiB,CAAgB;IAEzC,YAAY,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,cAAc,EAG3D;IAED;;;OAGG;IACH,OAAO,CAAC,aAAa;IAcf,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CA6B3B;IAED,OAAO,CAAC,gBAAgB;YAgCV,WAAW;IAuFzB,IAAI,SAQH;IAED;;;OAGG;IACH,YAAY,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAiB1C;IAMD;;OAEG;IACG,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAKjD;IAED;;OAEG;IACG,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAE,OAAe,GAAG,OAAO,CAAC,OAAO,CAAC,CAiBlG;IAED;;;OAGG;IACG,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAa/E;IAED;;OAEG;IACG,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAMvE;IAED,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAEnC;IAED;;OAEG;IACG,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAkDjE;YAMa,UAAU;YAoEV,UAAU;YAyDV,cAAc;IAsC5B,OAAO,CAAC,cAAc;YAoBR,eAAe;IAuE7B,OAAO,CAAC,QAAQ;IAShB,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,mBAAmB;IAiB3B,OAAO,CAAC,uBAAuB;CAI/B","sourcesContent":["/**\n * DingTalk communication layer using dingtalk-stream SDK with AI Card streaming.\n *\n * Handles:\n * - Receiving messages via DingTalk Stream Mode (DWClient)\n * - Responding via AI Card (streaming) or plain markdown (fallback)\n * - Access token management\n * - Per-channel message queuing\n */\nimport axios from \"axios\";\nimport { DWClient, type DWClientDownStream, type RobotMessage, TOPIC_ROBOT } from \"dingtalk-stream\";\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { dirname, join } from \"path\";\nimport * as log from \"./log.js\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface DingTalkConfig {\n\tclientId: string;\n\tclientSecret: string;\n\trobotCode?: string;\n\tcardTemplateId?: string;\n\tcardTemplateKey?: string;\n\tallowFrom?: string[];\n\tstateDir?: string;\n}\n\nexport interface DingTalkEvent {\n\ttype: \"dm\" | \"group\";\n\tchannelId: string; // dm_{staffId} or group_{conversationId}\n\tts: string;\n\tuser: string; // sender staff id\n\tuserName: string; // sender nickname\n\ttext: string;\n\tconversationId: string;\n\tconversationType: string; // \"1\" = DM, \"2\" = group\n}\n\nexport interface DingTalkContext {\n\tmessage: {\n\t\ttext: string;\n\t\trawText: string;\n\t\tuser: string;\n\t\tuserName?: string;\n\t\tchannel: string;\n\t\tts: string;\n\t};\n\tchannelName?: string;\n\trespond: (text: string, shouldLog?: boolean) => Promise<void>;\n\trespondPlain: (text: string, shouldLog?: boolean) => Promise<boolean>;\n\treplaceMessage: (text: string) => Promise<void>;\n\trespondInThread: (text: string) => Promise<void>;\n\tsetTyping: (isTyping: boolean) => Promise<void>;\n\tsetWorking: (working: boolean) => Promise<void>;\n\tdeleteMessage: () => Promise<void>;\n\tflush: () => Promise<void>;\n\tclose: () => Promise<void>;\n}\n\nexport interface DingTalkHandler {\n\tisRunning(channelId: string): boolean;\n\thandleEvent(event: DingTalkEvent, bot: DingTalkBot, isEvent?: boolean): Promise<void>;\n\thandleStop(channelId: string, bot: DingTalkBot): Promise<void>;\n}\n\n// ============================================================================\n// AI Card State\n// ============================================================================\n\ninterface AICard {\n\tinstanceId: string;\n\tconversationId: string;\n\taccessToken: string;\n\ttemplateKey: string;\n\tcreatedAt: number;\n\tlastUpdated: number;\n\tcontent: string;\n\tfinished: boolean;\n}\n\ninterface ConversationMeta {\n\tconversationId: string;\n\tconversationType: string;\n\tsenderId: string;\n}\n\n// ============================================================================\n// Per-channel queue for sequential processing\n// ============================================================================\n\ntype QueuedWork = () => Promise<void>;\n\nclass ChannelQueue {\n\tprivate queue: QueuedWork[] = [];\n\tprivate processing = false;\n\n\tenqueue(work: QueuedWork): void {\n\t\tthis.queue.push(work);\n\t\tthis.processNext();\n\t}\n\n\tsize(): number {\n\t\treturn this.queue.length;\n\t}\n\n\tprivate async processNext(): Promise<void> {\n\t\tif (this.processing || this.queue.length === 0) return;\n\t\tthis.processing = true;\n\t\tconst work = this.queue.shift()!;\n\t\ttry {\n\t\t\tawait work();\n\t\t} catch (err) {\n\t\t\tlog.logWarning(\"Queue error\", err instanceof Error ? err.message : String(err));\n\t\t}\n\t\tthis.processing = false;\n\t\tthis.processNext();\n\t}\n}\n\n// ============================================================================\n// Constants\n// ============================================================================\n\nconst DINGTALK_API = \"https://api.dingtalk.com\";\nconst TOKEN_REFRESH_SECS = 90 * 60; // 1.5 hours (tokens expire after 2 hours)\n\n// ============================================================================\n// DingTalkBot\n// ============================================================================\n\nexport class DingTalkBot {\n\tprivate handler: DingTalkHandler;\n\tprivate config: DingTalkConfig;\n\n\t// Access token cache\n\tprivate accessToken: string | null = null;\n\tprivate tokenExpiry = 0;\n\n\t// Active AI cards: channelId → AICard\n\tprivate activeCards = new Map<string, AICard>();\n\n\t// Conversation metadata cache: channelId → metadata\n\tprivate convMeta = new Map<string, ConversationMeta>();\n\n\t// Per-channel queues\n\tprivate queues = new Map<string, ChannelQueue>();\n\n\t// Connection stability\n\tprivate client: DWClient | null = null;\n\tprivate lastSocketAvailableTime = Date.now();\n\tprivate activeMessageProcessing = false;\n\tprivate keepAliveTimer: NodeJS.Timeout | null = null;\n\tprivate isReconnecting = false;\n\tprivate isStopped = false;\n\tprivate reconnectAttempts = 0;\n\n\t// Deduplication cache (Set for O(1) lookup, order array for FIFO eviction)\n\tprivate processedIds = new Set<string>();\n\tprivate processedIdsOrder: string[] = [];\n\n\tconstructor(handler: DingTalkHandler, config: DingTalkConfig) {\n\t\tthis.handler = handler;\n\t\tthis.config = config;\n\t}\n\n\t/**\n\t * Mark an ID as processed. Returns true if this is a new ID, false if already seen.\n\t * Maintains a FIFO buffer of at most 200 entries.\n\t */\n\tprivate markProcessed(id: string): boolean {\n\t\tif (this.processedIds.has(id)) return false;\n\t\tthis.processedIds.add(id);\n\t\tthis.processedIdsOrder.push(id);\n\t\twhile (this.processedIdsOrder.length > 200) {\n\t\t\tthis.processedIds.delete(this.processedIdsOrder.shift()!);\n\t\t}\n\t\treturn true;\n\t}\n\n\t// ==========================================================================\n\t// Public API\n\t// ==========================================================================\n\n\tasync start(): Promise<void> {\n\t\tif (!this.config.clientId || !this.config.clientSecret) {\n\t\t\tlog.logWarning(\"DingTalk: clientId / clientSecret not configured\");\n\t\t\treturn;\n\t\t}\n\n\t\tif (!this.config.cardTemplateId) {\n\t\t\tlog.logWarning(\"DingTalk: cardTemplateId not configured — AI Card streaming will not work\");\n\t\t}\n\n\t\tlog.logInfo(`DingTalk: initializing stream (clientId=${this.config.clientId.substring(0, 8)}…)`);\n\n\t\tif (process.env.DINGTALK_FORCE_PROXY !== \"true\") {\n\t\t\taxios.defaults.proxy = false;\n\t\t}\n\n\t\tthis.client = new DWClient({\n\t\t\tclientId: this.config.clientId,\n\t\t\tclientSecret: this.config.clientSecret,\n\t\t\tautoReconnect: false,\n\t\t\tkeepAlive: false,\n\t\t} as any);\n\n\t\tthis.client.registerCallbackListener(TOPIC_ROBOT, (msg: DWClientDownStream) => {\n\t\t\treturn this.handleRawMessage(msg);\n\t\t});\n\n\t\tlog.logConnected();\n\t\tawait this.doReconnect(true); // Initial connection\n\t}\n\n\tprivate handleRawMessage(msg: DWClientDownStream): { status: \"SUCCESS\"; message: string } {\n\t\t// 1. Immediate ACK\n\t\tif (msg.headers?.messageId && this.client) {\n\t\t\tthis.client.socketCallBackResponse(msg.headers.messageId, { status: \"SUCCESS\", message: \"OK\" });\n\t\t}\n\n\t\t// 2. Protocol deduplication\n\t\tconst messageId = msg.headers?.messageId;\n\t\tif (messageId && !this.markProcessed(messageId)) {\n\t\t\treturn { status: \"SUCCESS\", message: \"OK\" };\n\t\t}\n\n\t\ttry {\n\t\t\tconst data: RobotMessage = typeof msg.data === \"string\" ? JSON.parse(msg.data) : msg.data;\n\n\t\t\t// 3. Business logic deduplication\n\t\t\tconst msgId = (data as any).msgId;\n\t\t\tif (msgId && !this.markProcessed(msgId)) {\n\t\t\t\treturn { status: \"SUCCESS\", message: \"OK\" };\n\t\t\t}\n\n\t\t\t// Fire-and-forget processing\n\t\t\tthis.onStreamMessage(data).catch((err: unknown) => {\n\t\t\t\tlog.logWarning(\"DingTalk handler error\", err instanceof Error ? err.message : String(err));\n\t\t\t});\n\t\t} catch (err) {\n\t\t\tlog.logWarning(\"DingTalk: failed to parse message\", err instanceof Error ? err.message : String(err));\n\t\t}\n\n\t\treturn { status: \"SUCCESS\", message: \"OK\" };\n\t}\n\n\tprivate async doReconnect(immediate = false) {\n\t\tif (this.isReconnecting || this.isStopped || !this.client) return;\n\t\tthis.isReconnecting = true;\n\t\tlet connectionFailed = false;\n\n\t\tif (!immediate && this.reconnectAttempts > 0) {\n\t\t\tconst delay = Math.min(1000 * 2 ** this.reconnectAttempts + Math.random() * 1000, 30000);\n\t\t\tlog.logInfo(`DingTalk: waiting ${Math.round(delay / 1000)}s before reconnecting...`);\n\t\t\tawait new Promise((resolve) => setTimeout(resolve, delay));\n\t\t}\n\n\t\ttry {\n\t\t\tconst socket = (this.client as any).socket;\n\t\t\tif (socket?.readyState === 1 || socket?.readyState === 3) {\n\t\t\t\tawait (this.client as any).disconnect();\n\t\t\t}\n\n\t\t\tawait this.client.connect();\n\n\t\t\tthis.lastSocketAvailableTime = Date.now();\n\t\t\tthis.reconnectAttempts = 0; // Success, reset backoff\n\t\t\tlog.logInfo(\"DingTalk: connected to stream.\");\n\n\t\t\t// Setup keep alive\n\t\t\tif (this.keepAliveTimer) clearInterval(this.keepAliveTimer);\n\t\t\tthis.keepAliveTimer = setInterval(() => {\n\t\t\t\tif (this.isStopped) return;\n\n\t\t\t\tconst elapsed = Date.now() - this.lastSocketAvailableTime;\n\t\t\t\tif (elapsed > 90 * 1000 && !this.activeMessageProcessing) {\n\t\t\t\t\tlog.logWarning(\"DingTalk: connection timeout detected (>90s). Keeping active where possible...\");\n\t\t\t\t}\n\n\t\t\t\ttry {\n\t\t\t\t\tconst s = (this.client as any)?.socket;\n\t\t\t\t\tif (s?.readyState === 1) {\n\t\t\t\t\t\ts.ping();\n\t\t\t\t\t}\n\t\t\t\t} catch (_err) {\n\t\t\t\t\t// Ignore\n\t\t\t\t}\n\t\t\t}, 30 * 1000);\n\n\t\t\t// Setup native socket events\n\t\t\tconst s = (this.client as any).socket;\n\n\t\t\ts?.on(\"pong\", () => {\n\t\t\t\tthis.lastSocketAvailableTime = Date.now();\n\t\t\t});\n\n\t\t\ts?.on(\"close\", (code: number, reason: string) => {\n\t\t\t\tlog.logWarning(`DingTalk: WebSocket closed: code=${code}, reason=${reason}`);\n\t\t\t\tif (this.isStopped) return;\n\t\t\t\tsetTimeout(() => {\n\t\t\t\t\tthis.doReconnect(true).catch((err) => {\n\t\t\t\t\t\tlog.logWarning(\"DingTalk: reconnect failed\", err instanceof Error ? err.message : String(err));\n\t\t\t\t\t});\n\t\t\t\t}, 1000);\n\t\t\t});\n\n\t\t\ts?.on(\"message\", (raw: any) => {\n\t\t\t\ttry {\n\t\t\t\t\tconst msg = JSON.parse(raw);\n\t\t\t\t\tif (msg.type === \"SYSTEM\" && msg.headers?.topic === \"disconnect\") {\n\t\t\t\t\t\tlog.logWarning(\"DingTalk: disconnect event received from server.\");\n\t\t\t\t\t\tif (!this.isStopped) {\n\t\t\t\t\t\t\tthis.doReconnect(true).catch(() => {});\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t} catch (_e) {\n\t\t\t\t\t// skip\n\t\t\t\t}\n\t\t\t});\n\t\t} catch (err) {\n\t\t\tthis.reconnectAttempts++;\n\t\t\tconnectionFailed = true;\n\t\t\tlog.logWarning(\"DingTalk: connection failed\", err instanceof Error ? err.message : String(err));\n\t\t} finally {\n\t\t\tthis.isReconnecting = false;\n\t\t}\n\n\t\t// Auto-retry on failure with exponential backoff\n\t\tif (connectionFailed && !this.isStopped) {\n\t\t\tthis.doReconnect().catch(() => {});\n\t\t}\n\t}\n\n\tstop() {\n\t\tthis.isStopped = true;\n\t\tif (this.keepAliveTimer) clearInterval(this.keepAliveTimer);\n\t\tif (this.client) {\n\t\t\ttry {\n\t\t\t\t(this.client as any).disconnect();\n\t\t\t} catch (_e) {}\n\t\t}\n\t}\n\n\t/**\n\t * Enqueue an event for processing.\n\t * Returns true if enqueued, false if queue is full (max 5).\n\t */\n\tenqueueEvent(event: DingTalkEvent): boolean {\n\t\tconst queue = this.getQueue(event.channelId);\n\t\tif (queue.size() >= 5) {\n\t\t\tlog.logWarning(`Event queue full for ${event.channelId}, discarding: ${event.text.substring(0, 50)}`);\n\t\t\treturn false;\n\t\t}\n\t\tlog.logInfo(`Enqueueing event for ${event.channelId}: ${event.text.substring(0, 50)}`);\n\t\tqueue.enqueue(async () => {\n\t\t\tthis.activeMessageProcessing = true;\n\t\t\ttry {\n\t\t\t\tawait this.handler.handleEvent(event, this, true);\n\t\t\t} finally {\n\t\t\t\tthis.activeMessageProcessing = false;\n\t\t\t\tthis.lastSocketAvailableTime = Date.now();\n\t\t\t}\n\t\t});\n\t\treturn true;\n\t}\n\n\t// ==========================================================================\n\t// AI Card operations\n\t// ==========================================================================\n\n\t/**\n\t * Get or create an AI Card for a channel.\n\t */\n\tasync ensureCard(channelId: string): Promise<void> {\n\t\tif (!this.config.cardTemplateId) return;\n\t\tconst existing = this.activeCards.get(channelId);\n\t\tif (existing && !existing.finished) return;\n\t\tawait this.createCard(channelId);\n\t}\n\n\t/**\n\t * Stream content to the active AI Card for a channel.\n\t */\n\tasync streamToCard(channelId: string, content: string, finalize: boolean = false): Promise<boolean> {\n\t\tlet card = this.activeCards.get(channelId);\n\t\tif ((!card || card.finished) && !finalize && this.config.cardTemplateId && content.trim()) {\n\t\t\tawait this.ensureCard(channelId);\n\t\t\tcard = this.activeCards.get(channelId);\n\t\t}\n\t\tif (!card || card.finished) {\n\t\t\tif (finalize) {\n\t\t\t\treturn this.sendPlain(channelId, content);\n\t\t\t}\n\t\t\treturn false;\n\t\t}\n\t\tconst streamed = await this.streamCard(card, content, finalize);\n\t\tif (!streamed) {\n\t\t\tthis.activeCards.delete(channelId);\n\t\t}\n\t\treturn streamed;\n\t}\n\n\t/**\n\t * Finalize the active card for a channel without falling back to a plain message.\n\t * Returns true if a card was finalized, false if no active card existed.\n\t */\n\tasync finalizeExistingCard(channelId: string, content: string): Promise<boolean> {\n\t\tlet card = this.activeCards.get(channelId);\n\t\tif ((!card || card.finished) && this.config.cardTemplateId && content.trim()) {\n\t\t\tawait this.ensureCard(channelId);\n\t\t\tcard = this.activeCards.get(channelId);\n\t\t}\n\t\tif (!card || card.finished) {\n\t\t\treturn false;\n\t\t}\n\n\t\tconst finalized = await this.streamCard(card, content, true);\n\t\tthis.activeCards.delete(channelId);\n\t\treturn finalized;\n\t}\n\n\t/**\n\t * Finalize and remove the active card for a channel.\n\t */\n\tasync finalizeCard(channelId: string, content: string): Promise<boolean> {\n\t\tconst finalized = await this.finalizeExistingCard(channelId, content);\n\t\tif (!finalized) {\n\t\t\treturn this.sendPlain(channelId, content);\n\t\t}\n\t\treturn true;\n\t}\n\n\tdiscardCard(channelId: string): void {\n\t\tthis.activeCards.delete(channelId);\n\t}\n\n\t/**\n\t * Send a normal message natively mapping DM and Group to correct endpoints (fallback when no card).\n\t */\n\tasync sendPlain(channelId: string, text: string): Promise<boolean> {\n\t\tconst token = await this.getAccessToken();\n\t\tif (!token) return false;\n\n\t\tconst meta = this.getConversationMeta(channelId);\n\t\tif (!meta) {\n\t\t\tlog.logWarning(`No conversation metadata for ${channelId}, cannot send plain message`);\n\t\t\treturn false;\n\t\t}\n\n\t\tconst robotCode = this.config.robotCode || this.config.clientId;\n\t\tconst isGroup = meta.conversationType === \"2\";\n\n\t\tconst hasMarkdown = /^#{1,6}\\s|^\\s*[-*]\\s|\\*\\*.*\\*\\*|```|`[^`]+`|\\[.*?\\]\\(.*?\\)/m.test(text);\n\n\t\tconst msgKey = hasMarkdown ? \"sampleMarkdown\" : \"sampleText\";\n\t\tconst msgParam = hasMarkdown ? JSON.stringify({ text, title: \"Bot\" }) : JSON.stringify({ content: text });\n\n\t\tconst url = isGroup\n\t\t\t? `${DINGTALK_API}/v1.0/robot/groupMessages/send`\n\t\t\t: `${DINGTALK_API}/v1.0/robot/oToMessages/batchSend`;\n\n\t\tconst body: any = {\n\t\t\trobotCode,\n\t\t\tmsgKey,\n\t\t\tmsgParam,\n\t\t};\n\n\t\tif (isGroup) {\n\t\t\tbody.openConversationId = meta.conversationId;\n\t\t} else {\n\t\t\tbody.userIds = [meta.senderId];\n\t\t}\n\n\t\ttry {\n\t\t\tawait axios.post(url, body, {\n\t\t\t\theaders: {\n\t\t\t\t\t\"x-acs-dingtalk-access-token\": token,\n\t\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\t},\n\t\t\t});\n\t\t\treturn true;\n\t\t} catch (err) {\n\t\t\tif (axios.isAxiosError(err) && err.response) {\n\t\t\t\tlog.logWarning(`DingTalk plain send failed (${err.response.status})`, JSON.stringify(err.response.data));\n\t\t\t} else {\n\t\t\t\tlog.logWarning(\"DingTalk plain send error\", err instanceof Error ? err.message : String(err));\n\t\t\t}\n\t\t\treturn false;\n\t\t}\n\t}\n\n\t// ==========================================================================\n\t// Private - AI Card implementation\n\t// ==========================================================================\n\n\tprivate async createCard(channelId: string): Promise<AICard | null> {\n\t\tconst token = await this.getAccessToken();\n\t\tif (!token) return null;\n\n\t\tconst meta = this.getConversationMeta(channelId);\n\t\tif (!meta) {\n\t\t\tlog.logWarning(`No conversation metadata for ${channelId}, cannot create card`);\n\t\t\treturn null;\n\t\t}\n\n\t\tconst isGroup = meta.conversationType === \"2\";\n\t\tconst instanceId = `card_${Date.now()}_${Math.random().toString(36).substring(2, 8)}`;\n\t\tconst robotCode = this.config.robotCode || this.config.clientId;\n\n\t\t// openSpaceId format:\n\t\t// 群聊: dtv1.card//IM_GROUP.{openConversationId}\n\t\t// 单聊: dtv1.card//IM_ROBOT.{userId}\n\t\tconst openSpaceId = isGroup\n\t\t\t? `dtv1.card//IM_GROUP.${meta.conversationId}`\n\t\t\t: `dtv1.card//IM_ROBOT.${meta.senderId}`;\n\n\t\tconst body: Record<string, unknown> = {\n\t\t\tcardTemplateId: this.config.cardTemplateId,\n\t\t\toutTrackId: instanceId,\n\t\t\tcardData: { cardParamMap: {} },\n\t\t\tcallbackType: \"STREAM\",\n\t\t\timGroupOpenSpaceModel: { supportForward: true },\n\t\t\timRobotOpenSpaceModel: { supportForward: true },\n\t\t\topenSpaceId,\n\t\t\tuserIdType: 1,\n\t\t};\n\n\t\tif (isGroup) {\n\t\t\tbody.imGroupOpenDeliverModel = { robotCode };\n\t\t} else {\n\t\t\tbody.imRobotOpenDeliverModel = { spaceType: \"IM_ROBOT\" };\n\t\t}\n\n\t\ttry {\n\t\t\tawait axios.post(`${DINGTALK_API}/v1.0/card/instances/createAndDeliver`, body, {\n\t\t\t\theaders: {\n\t\t\t\t\t\"x-acs-dingtalk-access-token\": token,\n\t\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\t},\n\t\t\t});\n\t\t} catch (err) {\n\t\t\tif (axios.isAxiosError(err) && err.response) {\n\t\t\t\tlog.logWarning(`DingTalk Card: create failed (${err.response.status})`, JSON.stringify(err.response.data));\n\t\t\t} else {\n\t\t\t\tlog.logWarning(\"DingTalk Card: create failed\", err instanceof Error ? err.message : String(err));\n\t\t\t}\n\t\t\treturn null;\n\t\t}\n\n\t\tconst card: AICard = {\n\t\t\tinstanceId,\n\t\t\tconversationId: meta.conversationId,\n\t\t\taccessToken: token,\n\t\t\ttemplateKey: this.config.cardTemplateKey || \"content\",\n\t\t\tcreatedAt: Date.now() / 1000,\n\t\t\tlastUpdated: Date.now() / 1000,\n\t\t\tcontent: \"\",\n\t\t\tfinished: false,\n\t\t};\n\t\tthis.activeCards.set(channelId, card);\n\t\treturn card;\n\t}\n\n\tprivate async streamCard(card: AICard, content: string, finalize: boolean = false): Promise<boolean> {\n\t\t// Refresh token if needed\n\t\tconst ageSecs = Date.now() / 1000 - card.createdAt;\n\t\tif (ageSecs > TOKEN_REFRESH_SECS) {\n\t\t\tconst token = await this.getAccessToken();\n\t\t\tif (token) {\n\t\t\t\tcard.accessToken = token;\n\t\t\t}\n\t\t}\n\n\t\tconst body = {\n\t\t\toutTrackId: card.instanceId,\n\t\t\tguid: `${Date.now()}_${Math.random().toString(36).substring(2, 8)}`,\n\t\t\tkey: card.templateKey,\n\t\t\tcontent,\n\t\t\tisFull: true,\n\t\t\tisFinalize: finalize,\n\t\t\tisError: false,\n\t\t};\n\n\t\tconst start = Date.now();\n\t\ttry {\n\t\t\tawait axios.put(`${DINGTALK_API}/v1.0/card/streaming`, body, {\n\t\t\t\theaders: {\n\t\t\t\t\t\"x-acs-dingtalk-access-token\": card.accessToken,\n\t\t\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tconst duration = Date.now() - start;\n\t\t\tif (duration > 1000) {\n\t\t\t\tlog.logWarning(`DingTalk Card: streaming request took ${duration}ms (slow)`);\n\t\t\t}\n\n\t\t\tcard.lastUpdated = Date.now() / 1000;\n\t\t\tcard.content = content;\n\t\t\tif (finalize) {\n\t\t\t\tcard.finished = true;\n\t\t\t}\n\t\t\treturn true;\n\t\t} catch (err) {\n\t\t\tif (axios.isAxiosError(err) && err.response) {\n\t\t\t\tlog.logWarning(\n\t\t\t\t\t`DingTalk Card: streaming failed (${err.response.status})`,\n\t\t\t\t\tJSON.stringify(err.response.data),\n\t\t\t\t);\n\t\t\t} else {\n\t\t\t\tlog.logWarning(\"DingTalk Card: streaming failed\", err instanceof Error ? err.message : String(err));\n\t\t\t}\n\t\t\treturn false;\n\t\t}\n\t}\n\n\t// ==========================================================================\n\t// Private - Access Token\n\t// ==========================================================================\n\n\tprivate async getAccessToken(): Promise<string | null> {\n\t\tif (this.accessToken && Date.now() / 1000 < this.tokenExpiry) {\n\t\t\treturn this.accessToken;\n\t\t}\n\n\t\ttry {\n\t\t\tconst resp = await axios.post(\n\t\t\t\t`${DINGTALK_API}/v1.0/oauth2/accessToken`,\n\t\t\t\t{\n\t\t\t\t\tappKey: this.config.clientId,\n\t\t\t\t\tappSecret: this.config.clientSecret,\n\t\t\t\t},\n\t\t\t\t{\n\t\t\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t\t\t},\n\t\t\t);\n\n\t\t\tconst data = resp.data as { accessToken?: string; expireIn?: number };\n\t\t\tthis.accessToken = data.accessToken || null;\n\t\t\tthis.tokenExpiry = Date.now() / 1000 + (data.expireIn || 7200) - 60;\n\t\t\treturn this.accessToken;\n\t\t} catch (err) {\n\t\t\tif (axios.isAxiosError(err) && err.response) {\n\t\t\t\tlog.logWarning(\n\t\t\t\t\t`DingTalk: failed to get access token (${err.response.status})`,\n\t\t\t\t\tJSON.stringify(err.response.data),\n\t\t\t\t);\n\t\t\t} else {\n\t\t\t\tlog.logWarning(\"DingTalk: failed to get access token\", err instanceof Error ? err.message : String(err));\n\t\t\t}\n\t\t\treturn null;\n\t\t}\n\t}\n\n\t// ==========================================================================\n\t// Private - Message handling\n\t// ==========================================================================\n\n\tprivate extractContent(data: RobotMessage): string {\n\t\t// 1. text 类型消息:从 text.content 提取\n\t\tconst textContent = (data.text?.content || \"\").trim();\n\t\tif (textContent) return textContent;\n\n\t\t// 2. richText 类型消息:从 content.richText 列表提取文本片段\n\t\tconst raw = data as unknown as Record<string, unknown>;\n\t\tconst contentObj = raw.content as { richText?: Array<Record<string, string>> } | undefined;\n\t\tif (contentObj?.richText) {\n\t\t\tconst parts: string[] = [];\n\t\t\tfor (const item of contentObj.richText) {\n\t\t\t\tif (item.text) parts.push(item.text);\n\t\t\t}\n\t\t\tconst joined = parts.join(\"\").trim();\n\t\t\tif (joined) return joined;\n\t\t}\n\n\t\treturn \"\";\n\t}\n\n\tprivate async onStreamMessage(data: RobotMessage): Promise<void> {\n\t\tconst content = this.extractContent(data);\n\t\tconst senderId = data.senderStaffId || data.senderId || \"\";\n\t\tconst senderName = data.senderNick || \"Unknown\";\n\t\tconst conversationId = data.conversationId || \"\";\n\t\tconst conversationType = data.conversationType || \"1\";\n\n\t\tif (!content) {\n\t\t\tconst msgtype = (data as unknown as Record<string, unknown>).msgtype || \"unknown\";\n\t\t\tlog.logWarning(`DingTalk: empty message (type=${msgtype})`);\n\t\t\treturn;\n\t\t}\n\n\t\tif (this.config.allowFrom && this.config.allowFrom.length > 0) {\n\t\t\tif (!this.config.allowFrom.includes(senderId)) {\n\t\t\t\tlog.logWarning(`DingTalk: ignoring message from unauthorized user ${senderName} (${senderId})`);\n\t\t\t\treturn;\n\t\t\t}\n\t\t}\n\n\t\t// Determine channel ID\n\t\tconst channelId = conversationType === \"2\" ? `group_${conversationId}` : `dm_${senderId}`;\n\n\t\tlog.logInfo(`DingTalk ← ${senderName} (${senderId}) [${channelId}]: ${content.substring(0, 80)}`);\n\n\t\t// Cache conversation metadata for card creation\n\t\tthis.setConversationMeta(channelId, {\n\t\t\tconversationId,\n\t\t\tconversationType,\n\t\t\tsenderId,\n\t\t});\n\n\t\t// Build event\n\t\tconst event: DingTalkEvent = {\n\t\t\ttype: conversationType === \"2\" ? \"group\" : \"dm\",\n\t\t\tchannelId,\n\t\t\tts: Date.now().toString(),\n\t\t\tuser: senderId,\n\t\t\tuserName: senderName,\n\t\t\ttext: content,\n\t\t\tconversationId,\n\t\t\tconversationType,\n\t\t};\n\n\t\t// Check for stop command\n\t\tif (content.toLowerCase().trim() === \"stop\") {\n\t\t\tif (this.handler.isRunning(channelId)) {\n\t\t\t\tthis.handler.handleStop(channelId, this);\n\t\t\t}\n\t\t\treturn;\n\t\t}\n\n\t\t// Check if busy\n\t\tif (this.handler.isRunning(channelId)) {\n\t\t\tconst busyMsg = \"正在处理中,请稍候。发送 `stop` 可取消当前任务。\";\n\t\t\tawait this.sendPlain(channelId, busyMsg);\n\t\t\treturn;\n\t\t}\n\n\t\t// Enqueue for processing\n\t\tthis.getQueue(channelId).enqueue(async () => {\n\t\t\tthis.activeMessageProcessing = true;\n\t\t\ttry {\n\t\t\t\tawait this.handler.handleEvent(event, this);\n\t\t\t} finally {\n\t\t\t\tthis.activeMessageProcessing = false;\n\t\t\t\tthis.lastSocketAvailableTime = Date.now();\n\t\t\t}\n\t\t});\n\t}\n\n\tprivate getQueue(channelId: string): ChannelQueue {\n\t\tlet queue = this.queues.get(channelId);\n\t\tif (!queue) {\n\t\t\tqueue = new ChannelQueue();\n\t\t\tthis.queues.set(channelId, queue);\n\t\t}\n\t\treturn queue;\n\t}\n\n\tprivate getConversationMeta(channelId: string): ConversationMeta | null {\n\t\tconst cached = this.convMeta.get(channelId);\n\t\tif (cached) return cached;\n\n\t\tconst metaPath = this.getConversationMetaPath(channelId);\n\t\tif (!metaPath || !existsSync(metaPath)) {\n\t\t\treturn null;\n\t\t}\n\n\t\ttry {\n\t\t\tconst parsed = JSON.parse(readFileSync(metaPath, \"utf-8\")) as Partial<ConversationMeta>;\n\t\t\tif (!parsed.conversationId || !parsed.conversationType || !parsed.senderId) {\n\t\t\t\treturn null;\n\t\t\t}\n\n\t\t\tconst meta: ConversationMeta = {\n\t\t\t\tconversationId: parsed.conversationId,\n\t\t\t\tconversationType: parsed.conversationType,\n\t\t\t\tsenderId: parsed.senderId,\n\t\t\t};\n\t\t\tthis.convMeta.set(channelId, meta);\n\t\t\treturn meta;\n\t\t} catch (err) {\n\t\t\tlog.logWarning(\n\t\t\t\t`Failed to load conversation metadata for ${channelId}`,\n\t\t\t\terr instanceof Error ? err.message : String(err),\n\t\t\t);\n\t\t\treturn null;\n\t\t}\n\t}\n\n\tprivate setConversationMeta(channelId: string, meta: ConversationMeta): void {\n\t\tthis.convMeta.set(channelId, meta);\n\n\t\tconst metaPath = this.getConversationMetaPath(channelId);\n\t\tif (!metaPath) return;\n\n\t\ttry {\n\t\t\tmkdirSync(dirname(metaPath), { recursive: true });\n\t\t\twriteFileSync(metaPath, JSON.stringify(meta, null, 2), \"utf-8\");\n\t\t} catch (err) {\n\t\t\tlog.logWarning(\n\t\t\t\t`Failed to persist conversation metadata for ${channelId}`,\n\t\t\t\terr instanceof Error ? err.message : String(err),\n\t\t\t);\n\t\t}\n\t}\n\n\tprivate getConversationMetaPath(channelId: string): string | null {\n\t\tif (!this.config.stateDir) return null;\n\t\treturn join(this.config.stateDir, channelId, \".channel-meta.json\");\n\t}\n}\n"]}
|