mirror of
https://github.com/hansjone/dsh-im-ops.git
synced 2026-10-14 14:20:45 +08:00
Internalize all IM channel implementations
This commit is contained in:
xmanrui
parent
8996476693
commit
bd469f58f8
95 changed files with 25012 additions and 100 deletions
271
test/channels/dingtalk/dingtalk-bridge.test.mjs
Normal file
271
test/channels/dingtalk/dingtalk-bridge.test.mjs
Normal file
|
|
@ -0,0 +1,271 @@
|
|||
import assert from 'node:assert/strict';
|
||||
import test from 'node:test';
|
||||
|
||||
import {
|
||||
createDingtalkBridgeStatus,
|
||||
DingtalkHarnessBridge,
|
||||
} from '../../../src/channels/dingtalk/dingtalk-bridge.mjs';
|
||||
|
||||
function message(id, text, overrides = {}) {
|
||||
return {
|
||||
msgId: id,
|
||||
msgtype: 'text',
|
||||
text: { content: text },
|
||||
conversationType: '1',
|
||||
conversationId: `conversation-${id}`,
|
||||
senderStaffId: 'staff-approved',
|
||||
senderNick: '钉钉用户',
|
||||
sessionWebhook: `https://oapi.dingtalk.com/robot/reply?ticket=${id}`,
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
function stateFixture() {
|
||||
const sessions = new Map();
|
||||
const seen = new Set();
|
||||
const pending = new Map();
|
||||
return {
|
||||
sessions,
|
||||
seen,
|
||||
pending,
|
||||
state: {
|
||||
hasSeen: (id) => seen.has(id),
|
||||
markSeen: async (id) => seen.add(id),
|
||||
sessionFor: (key) => sessions.get(key) ?? null,
|
||||
setSession: async (key, sessionId) => sessions.set(key, sessionId),
|
||||
clearSession: async (key) => sessions.delete(key),
|
||||
pendingSenders: () => [...pending.values()].map((entry) => structuredClone(entry)),
|
||||
recordPendingSender: async ({ staffId, displayName, lastSeenAt }) => {
|
||||
const existing = [...pending.values()].find((entry) => entry.staffId === staffId);
|
||||
const entry = {
|
||||
requestId: existing?.requestId ?? `request-${staffId}`,
|
||||
staffId,
|
||||
displayName,
|
||||
requestedAt: existing?.requestedAt ?? lastSeenAt,
|
||||
lastSeenAt,
|
||||
};
|
||||
pending.set(entry.requestId, entry);
|
||||
return structuredClone(entry);
|
||||
},
|
||||
removePendingSenderByStaffId: async (staffId) => {
|
||||
const entry = [...pending.values()].find((value) => value.staffId === staffId);
|
||||
if (!entry) return false;
|
||||
pending.delete(entry.requestId);
|
||||
return true;
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
test('bridge maps a DingTalk direct conversation to one persistent Harness session', async () => {
|
||||
const fixture = stateFixture();
|
||||
const sent = [];
|
||||
const asked = [];
|
||||
const status = createDingtalkBridgeStatus();
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: { sendText: async (request) => sent.push(request) },
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
sessionExists: async (sessionId) => sessionId === 'session-one',
|
||||
createSession: async () => 'session-one',
|
||||
ask: async (sessionId, text) => {
|
||||
asked.push({ sessionId, text });
|
||||
return 'Harness 回答';
|
||||
},
|
||||
},
|
||||
state: fixture.state,
|
||||
status,
|
||||
});
|
||||
|
||||
await Promise.all([
|
||||
bridge.accept(message('one', '你好')),
|
||||
bridge.accept(message('one', '重复消息')),
|
||||
]);
|
||||
await bridge.accept(message('two', '继续'));
|
||||
|
||||
assert.equal(fixture.sessions.get('p2p:staff-approved'), 'session-one');
|
||||
assert.deepEqual(asked, [
|
||||
{ sessionId: 'session-one', text: '你好' },
|
||||
{ sessionId: 'session-one', text: '继续' },
|
||||
]);
|
||||
assert.deepEqual(sent.map(({ text, sessionWebhook }) => ({ text, sessionWebhook })), [
|
||||
{ text: 'Harness 回答', sessionWebhook: 'https://oapi.dingtalk.com/robot/reply?ticket=one' },
|
||||
{ text: 'Harness 回答', sessionWebhook: 'https://oapi.dingtalk.com/robot/reply?ticket=two' },
|
||||
]);
|
||||
assert.equal(status.messagesReceived, 2);
|
||||
assert.equal(status.messagesReplied, 2);
|
||||
assert.equal(status.stats.messagesReplied, 2);
|
||||
});
|
||||
|
||||
test('senders in the bot visibility scope enter Harness without local approval', async () => {
|
||||
const fixture = stateFixture();
|
||||
const sent = [];
|
||||
const asked = [];
|
||||
const status = createDingtalkBridgeStatus();
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: { sendText: async (request) => sent.push(request) },
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
sessionExists: async () => false,
|
||||
createSession: async () => 'session-visible-sender',
|
||||
ask: async (sessionId, text) => {
|
||||
asked.push({ sessionId, text });
|
||||
return '直接回答';
|
||||
},
|
||||
},
|
||||
state: fixture.state,
|
||||
status,
|
||||
});
|
||||
|
||||
await bridge.accept(message('visible', '可见范围内的问题', {
|
||||
senderStaffId: 'raw-staff-id',
|
||||
senderNick: '可见范围用户',
|
||||
}));
|
||||
|
||||
assert.deepEqual(asked, [{
|
||||
sessionId: 'session-visible-sender',
|
||||
text: '可见范围内的问题',
|
||||
}]);
|
||||
assert.equal(sent.length, 1);
|
||||
assert.equal(sent[0].text, '直接回答');
|
||||
assert.equal(fixture.pending.size, 0);
|
||||
assert.deepEqual(status.pendingSenders, []);
|
||||
assert.equal(status.messagesRejected, 0);
|
||||
assert.equal(status.messagesReplied, 1);
|
||||
});
|
||||
|
||||
test('group messages require an explicit bot mention before Harness work', async () => {
|
||||
const fixture = stateFixture();
|
||||
const sent = [];
|
||||
const asked = [];
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: { sendText: async (request) => sent.push(request) },
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
sessionExists: async () => true,
|
||||
createSession: async () => 'session-group',
|
||||
ask: async (_sessionId, text) => {
|
||||
asked.push(text);
|
||||
return '群聊回答';
|
||||
},
|
||||
},
|
||||
state: fixture.state,
|
||||
});
|
||||
const group = {
|
||||
conversationType: '2',
|
||||
conversationId: 'group-one',
|
||||
};
|
||||
|
||||
await bridge.accept(message('not-mentioned', '群聊噪音', { ...group, isInAtList: false }));
|
||||
await bridge.accept(message('mentioned', '明确问题', { ...group, isInAtList: true }));
|
||||
|
||||
assert.deepEqual(asked, ['明确问题']);
|
||||
assert.deepEqual(sent.map(({ text }) => text), ['群聊回答']);
|
||||
assert.equal(bridge.status.messagesIgnored, 1);
|
||||
assert.equal(fixture.sessions.get('group:group-one'), 'session-group');
|
||||
});
|
||||
|
||||
test('bridge streams Harness snapshots into one DingTalk AI Card and finalizes it', async () => {
|
||||
const fixture = stateFixture();
|
||||
const calls = { create: [], update: [], finish: [], text: [] };
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: {
|
||||
sendText: async (request) => calls.text.push(request),
|
||||
createAiCard: async (request) => {
|
||||
calls.create.push(request);
|
||||
return { cardInstanceId: 'card-one' };
|
||||
},
|
||||
updateAiCard: async (request) => calls.update.push(request),
|
||||
finishAiCard: async (request) => {
|
||||
calls.finish.push(request);
|
||||
return { delivered: true, completed: false };
|
||||
},
|
||||
},
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
sessionExists: async () => false,
|
||||
createSession: async () => 'session-stream',
|
||||
ask: async (_sessionId, _text, options) => {
|
||||
options.onUpdate({ type: 'text', text: '生成中的完整快照' });
|
||||
await new Promise((resolve) => setTimeout(resolve, 510));
|
||||
return '最终完整回答';
|
||||
},
|
||||
},
|
||||
state: fixture.state,
|
||||
});
|
||||
|
||||
await bridge.accept(message('stream', '请流式回答'));
|
||||
|
||||
assert.equal(calls.create.length, 1);
|
||||
assert.deepEqual(calls.create[0].target, { type: 'user', userId: 'staff-approved' });
|
||||
assert.equal(calls.update.at(-1).text, '生成中的完整快照');
|
||||
assert.equal(calls.finish.length, 1);
|
||||
assert.equal(calls.finish[0].text, '最终完整回答');
|
||||
assert.equal(calls.text.length, 0);
|
||||
assert.equal(bridge.status.messagesReplied, 1);
|
||||
});
|
||||
|
||||
test('bridge asks Harness once and falls back to final text when AI Card creation fails', async () => {
|
||||
const fixture = stateFixture();
|
||||
const sent = [];
|
||||
let asks = 0;
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: {
|
||||
sendText: async (request) => sent.push(request.text),
|
||||
createAiCard: async () => { throw new Error('card unavailable'); },
|
||||
updateAiCard: async () => undefined,
|
||||
finishAiCard: async () => undefined,
|
||||
},
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
sessionExists: async () => false,
|
||||
createSession: async () => 'session-fallback',
|
||||
ask: async () => {
|
||||
asks += 1;
|
||||
return '文本降级回答';
|
||||
},
|
||||
},
|
||||
state: fixture.state,
|
||||
logger: { error() {} },
|
||||
});
|
||||
|
||||
await bridge.accept(message('fallback', '卡片失败也要回答'));
|
||||
|
||||
assert.equal(asks, 1);
|
||||
assert.deepEqual(sent, ['文本降级回答']);
|
||||
assert.equal(bridge.status.messagesReplied, 1);
|
||||
});
|
||||
|
||||
test('commands stay local and unsafe session webhooks are rejected before Harness', async () => {
|
||||
const fixture = stateFixture();
|
||||
fixture.sessions.set('p2p:staff-approved', 'old-session');
|
||||
const sent = [];
|
||||
let asked = 0;
|
||||
const bridge = new DingtalkHarnessBridge({
|
||||
api: { sendText: async (request) => sent.push(request.text) },
|
||||
clientId: 'ding-client',
|
||||
clientSecret: 'host-secret',
|
||||
harness: {
|
||||
ensureRunning: async () => true,
|
||||
sessionExists: async () => true,
|
||||
ask: async () => { asked += 1; },
|
||||
},
|
||||
state: fixture.state,
|
||||
logger: { warn() {}, error() {} },
|
||||
});
|
||||
|
||||
await bridge.accept(message('new', '/new'));
|
||||
assert.equal(fixture.sessions.has('p2p:staff-approved'), false);
|
||||
await bridge.accept(message('unsafe', '不应执行', {
|
||||
sessionWebhook: 'https://oapi.dingtalk.com.attacker.example/reply?private=one',
|
||||
}));
|
||||
assert.equal(asked, 0);
|
||||
assert.equal(sent[0], '已开启新会话。请发送你的问题。');
|
||||
assert.equal(sent.length, 1);
|
||||
assert.equal(bridge.status.lastError, '钉钉消息没有安全的回复地址。');
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue