mirror of
https://github.com/hansjone/dsh-im-ops.git
synced 2026-10-10 23:10:46 +08:00
762 lines
26 KiB
JavaScript
762 lines
26 KiB
JavaScript
import {
|
|
conversationKey,
|
|
extractText,
|
|
isAllowedSender,
|
|
isBotSender,
|
|
splitText,
|
|
} from './message-utils.mjs';
|
|
import {
|
|
harnessAnswerForQuestion,
|
|
harnessQuestionText,
|
|
validHarnessQuestion,
|
|
} from '../shared/harness-question.mjs';
|
|
import { HarnessApprovalQueue } from '../shared/harness-approval.mjs';
|
|
import { runWorkspaceCommand } from '../shared/workspace-command.mjs';
|
|
import { askInWorkspaceSession } from '../shared/workspace-session.mjs';
|
|
|
|
const INTERACTION_RESOLVED_TEXT = '这个问题已在其他客户端处理，无需再次回答。';
|
|
const RESOLVED_REPLY_TTL_MS = 30 * 60_000;
|
|
|
|
const HELP_TEXT = [
|
|
'北汇星河 AIOS 已连接 DeepSeek Harness。',
|
|
'',
|
|
'直接发送问题即可继续当前会话。',
|
|
'/new 开启一个全新会话',
|
|
'/workspace 工作区绝对路径 切换工作区',
|
|
'/workspacelist 列出工作区绝对路径',
|
|
'/sessionlist [工作区序号或绝对路径] 列出会话 ID 和标题',
|
|
'/session Session ID 将当前聊天绑定到指定会话',
|
|
'/status 检查连接状态',
|
|
'/help 显示本帮助',
|
|
].join('\n');
|
|
|
|
function nonEmptyString(value) {
|
|
return typeof value === 'string' && value.trim() ? value.trim() : null;
|
|
}
|
|
|
|
function senderOpenId(event) {
|
|
return nonEmptyString(event?.sender?.sender_id?.open_id)
|
|
?? nonEmptyString(event?.sender?.sender_id?.user_id);
|
|
}
|
|
|
|
function canClaimInteractionReply(event, pending) {
|
|
return pending.needsPresentation !== true
|
|
&& pending.questions[pending.index]
|
|
&& senderOpenId(event) === pending.actor
|
|
&& event?.message?.message_type === 'text'
|
|
&& nonEmptyString(extractText(event));
|
|
}
|
|
|
|
function ensureStatus(status) {
|
|
for (const key of ['messagesReceived', 'messagesReplied', 'messagesRejected']) {
|
|
status[key] ??= 0;
|
|
}
|
|
status.lastMessageAt ??= null;
|
|
status.lastReplyAt ??= null;
|
|
status.lastRejectedAt ??= null;
|
|
status.lastError ??= null;
|
|
}
|
|
|
|
export class FeishuHarnessBridge {
|
|
#client;
|
|
#channel;
|
|
#harness;
|
|
#state;
|
|
#queues = new Map();
|
|
#pendingInteractions = new Map();
|
|
#interactionKeys = new Map();
|
|
#resolvedQuestionReplies = new Map();
|
|
#acceptedMessageIds = new Set();
|
|
#interactionTasks = new Set();
|
|
#approvals;
|
|
#status;
|
|
#allowedSenderOpenIds;
|
|
#replyTimeoutMs;
|
|
#logger;
|
|
#signal;
|
|
|
|
constructor({
|
|
client,
|
|
channel,
|
|
harness,
|
|
state,
|
|
status,
|
|
allowedSenderOpenIds = new Set(),
|
|
replyTimeoutMs = 600_000,
|
|
logger = console,
|
|
signal,
|
|
}) {
|
|
if (!client || !harness || !state || !status) {
|
|
throw new TypeError('Feishu bridge dependencies are required');
|
|
}
|
|
this.#client = client;
|
|
this.#channel = channel;
|
|
this.#harness = harness;
|
|
this.#state = state;
|
|
this.#status = status;
|
|
this.#allowedSenderOpenIds = allowedSenderOpenIds;
|
|
this.#replyTimeoutMs = replyTimeoutMs;
|
|
this.#logger = logger;
|
|
this.#approvals = new HarnessApprovalQueue({ label: 'Feishu', logger });
|
|
this.#signal = signal;
|
|
ensureStatus(this.#status);
|
|
}
|
|
|
|
accept(event) {
|
|
if (this.#signal?.aborted) return Promise.resolve();
|
|
const messageId = nonEmptyString(event?.message?.message_id);
|
|
if (!messageId || isBotSender(event)) return Promise.resolve();
|
|
if (!isAllowedSender(event, this.#allowedSenderOpenIds)) {
|
|
this.#status.messagesRejected += 1;
|
|
this.#status.lastRejectedAt = new Date().toISOString();
|
|
this.#logger.warn?.('[dsh-feishu] ignored a message from a sender outside the allowlist');
|
|
return Promise.resolve();
|
|
}
|
|
if (this.#state.hasSeen(messageId) || this.#acceptedMessageIds.has(messageId)) {
|
|
return Promise.resolve();
|
|
}
|
|
|
|
let key;
|
|
try {
|
|
key = conversationKey(event);
|
|
} catch {
|
|
this.#status.messagesRejected += 1;
|
|
this.#status.lastRejectedAt = new Date().toISOString();
|
|
return Promise.resolve();
|
|
}
|
|
|
|
this.#acceptedMessageIds.add(messageId);
|
|
const processingReaction = this.#addReaction(messageId, 'OnIt');
|
|
if (this.#isResolvedQuestionReply(event, key)) {
|
|
const current = Promise.resolve()
|
|
.then(() => this.#discardResolvedInteractionReply(event, messageId))
|
|
.then(() => this.#finishReaction(messageId, processingReaction, 'DONE'))
|
|
.catch((error) => this.#handleMessageFailure(
|
|
event,
|
|
messageId,
|
|
processingReaction,
|
|
error,
|
|
))
|
|
.finally(() => this.#acceptedMessageIds.delete(messageId));
|
|
return current;
|
|
}
|
|
const pending = this.#pendingInteractions.get(key);
|
|
const approvalReply = this.#approvals.claimReply({
|
|
key,
|
|
actor: senderOpenId(event),
|
|
messageId,
|
|
text: extractText(event) ?? '',
|
|
addressed: event?.message?.chat_type === 'p2p'
|
|
|| (Array.isArray(event?.message?.mentions) && event.message.mentions.length > 0),
|
|
hasPendingQuestion: Boolean(pending),
|
|
questionCompletion: pending?.submitting || pending?.claimedReplyMessageId
|
|
? pending.queue
|
|
: null,
|
|
isQuestionPending: () => this.#pendingInteractions.has(key),
|
|
send: (text) => this.#send(event.message.chat_id, text),
|
|
});
|
|
if (approvalReply) {
|
|
const processing = approvalReply.process(async () => {
|
|
if (this.#state.hasSeen(messageId)) return false;
|
|
await this.#state.markSeen(messageId);
|
|
this.#status.lastMessageAt = new Date().toISOString();
|
|
this.#status.messagesReceived += 1;
|
|
return true;
|
|
});
|
|
let current;
|
|
current = processing
|
|
.then(() => this.#finishReaction(messageId, processingReaction, 'DONE'))
|
|
.catch((error) => this.#handleMessageFailure(
|
|
event,
|
|
messageId,
|
|
processingReaction,
|
|
error,
|
|
))
|
|
.finally(() => {
|
|
this.#acceptedMessageIds.delete(messageId);
|
|
this.#interactionTasks.delete(current);
|
|
});
|
|
this.#interactionTasks.add(current);
|
|
return current;
|
|
}
|
|
if (pending && senderOpenId(event) !== pending.actor) {
|
|
return this.#enqueueMessage(event, messageId, key, processingReaction);
|
|
}
|
|
if (pending?.submitting || pending?.claimedReplyMessageId) {
|
|
return this.#enqueueMessage(event, messageId, key, processingReaction);
|
|
}
|
|
if (pending) {
|
|
if (canClaimInteractionReply(event, pending)) pending.claimedReplyMessageId = messageId;
|
|
const previous = pending.queue ?? Promise.resolve();
|
|
const processing = previous
|
|
.catch(() => undefined)
|
|
.then(() => this.#processInteractionReply(
|
|
event,
|
|
messageId,
|
|
key,
|
|
pending,
|
|
processingReaction,
|
|
));
|
|
pending.queue = processing;
|
|
|
|
const releaseInteraction = () => {
|
|
if (pending.claimedReplyMessageId === messageId) {
|
|
pending.claimedReplyMessageId = null;
|
|
}
|
|
if (pending.queue === processing) pending.queue = null;
|
|
};
|
|
let current;
|
|
current = processing
|
|
.then(
|
|
() => {
|
|
releaseInteraction();
|
|
return this.#finishReaction(messageId, processingReaction, 'DONE');
|
|
},
|
|
(error) => {
|
|
releaseInteraction();
|
|
return this.#handleMessageFailure(
|
|
event,
|
|
messageId,
|
|
processingReaction,
|
|
error,
|
|
);
|
|
},
|
|
)
|
|
.finally(() => {
|
|
releaseInteraction();
|
|
this.#acceptedMessageIds.delete(messageId);
|
|
this.#interactionTasks.delete(current);
|
|
});
|
|
this.#interactionTasks.add(current);
|
|
return current;
|
|
}
|
|
return this.#enqueueMessage(event, messageId, key, processingReaction);
|
|
}
|
|
|
|
#enqueueMessage(event, messageId, key, processingReaction, {
|
|
releaseMessageId = true,
|
|
alreadyRecorded = false,
|
|
finalize = true,
|
|
} = {}) {
|
|
const previous = this.#queues.get(key) ?? Promise.resolve();
|
|
const work = previous
|
|
.catch(() => undefined)
|
|
.then(() => this.#handle(event, key, { alreadyRecorded }));
|
|
const settled = finalize
|
|
? work
|
|
.then(() => this.#finishReaction(messageId, processingReaction, 'DONE'))
|
|
.catch((error) => this.#handleMessageFailure(
|
|
event,
|
|
messageId,
|
|
processingReaction,
|
|
error,
|
|
))
|
|
: work;
|
|
let current;
|
|
current = settled.finally(() => {
|
|
if (releaseMessageId) this.#acceptedMessageIds.delete(messageId);
|
|
if (this.#queues.get(key) === current) this.#queues.delete(key);
|
|
});
|
|
this.#queues.set(key, current);
|
|
return current;
|
|
}
|
|
|
|
async #handleMessageFailure(event, messageId, processingReaction, error) {
|
|
if (this.#signal?.aborted) {
|
|
await this.#removeProcessingReaction(messageId, processingReaction);
|
|
return;
|
|
}
|
|
this.#logger.error?.('[dsh-feishu] message handling failed:', error?.message ?? String(error));
|
|
this.#status.lastError = error?.message ?? String(error);
|
|
await this.#finishReaction(messageId, processingReaction, 'ERROR');
|
|
await this.#send(
|
|
event.message.chat_id,
|
|
'处理失败，请稍后重试。如果问题持续，请在 DeepSeek Harness 的飞书插件页面检查连接状态。',
|
|
).catch(() => undefined);
|
|
}
|
|
|
|
async waitForIdle() {
|
|
await Promise.allSettled([
|
|
...this.#queues.values(),
|
|
...[...this.#pendingInteractions.values()].flatMap((pending) => (
|
|
pending.queue ? [pending.queue] : []
|
|
)),
|
|
...this.#interactionTasks,
|
|
]);
|
|
}
|
|
|
|
async #handle(event, key, { alreadyRecorded = false } = {}) {
|
|
this.#signal?.throwIfAborted();
|
|
const messageId = event.message.message_id;
|
|
if (!alreadyRecorded) {
|
|
if (this.#state.hasSeen(messageId)) return;
|
|
await this.#state.markSeen(messageId);
|
|
this.#status.lastMessageAt = new Date().toISOString();
|
|
this.#status.messagesReceived += 1;
|
|
}
|
|
|
|
const text = extractText(event);
|
|
if (!text) {
|
|
await this.#send(event.message.chat_id, '目前仅支持文字消息。');
|
|
return;
|
|
}
|
|
|
|
if (text === '/help') {
|
|
await this.#send(event.message.chat_id, HELP_TEXT);
|
|
return;
|
|
}
|
|
if (text === '/new') {
|
|
await this.#state.clearSession(key);
|
|
await this.#send(event.message.chat_id, '已开启全新 Harness 会话。');
|
|
return;
|
|
}
|
|
if (text === '/status') {
|
|
await this.#harness.ensureRunning({ signal: this.#signal });
|
|
await this.#send(event.message.chat_id, '飞书机器人与 DeepSeek Harness 连接正常。');
|
|
return;
|
|
}
|
|
const workspaceCommand = await runWorkspaceCommand(text, this.#harness, key);
|
|
if (workspaceCommand) {
|
|
for (const reply of workspaceCommand.messages ?? [workspaceCommand.message]) {
|
|
await this.#send(event.message.chat_id, reply);
|
|
}
|
|
return;
|
|
}
|
|
|
|
this.#logger.info?.(`[dsh-feishu] processing ${event.message.chat_type} message ${messageId}`);
|
|
try {
|
|
await this.#answerWithStream(event, key, text);
|
|
this.#status.messagesReplied += 1;
|
|
this.#status.lastReplyAt = new Date().toISOString();
|
|
this.#status.lastError = null;
|
|
} finally {
|
|
await this.#cancelPendingInteraction(key);
|
|
await this.#approvals.closeRoute(key);
|
|
}
|
|
}
|
|
|
|
#interactionAskOptions(event, key) {
|
|
return {
|
|
timeoutMs: this.#replyTimeoutMs,
|
|
signal: this.#signal,
|
|
onInteraction: (interaction) => this.#handleInteraction(interaction, {
|
|
key,
|
|
actor: senderOpenId(event),
|
|
chatId: event.message.chat_id,
|
|
requiresMention: event.message.chat_type !== 'p2p',
|
|
}),
|
|
onInteractionResolved: (resolution) => this.#handleInteractionResolved(resolution),
|
|
};
|
|
}
|
|
|
|
async #answerWithStream(event, key, text) {
|
|
const chatId = event.message.chat_id;
|
|
const messageId = event.message.message_id;
|
|
if (!this.#channel?.stream) {
|
|
const { answer } = await askInWorkspaceSession({
|
|
harness: this.#harness,
|
|
state: this.#state,
|
|
key,
|
|
text,
|
|
createOptions: { signal: this.#signal },
|
|
existsOptions: { signal: this.#signal },
|
|
askOptions: this.#interactionAskOptions(event, key),
|
|
});
|
|
for (const chunk of splitText(answer)) await this.#send(chatId, chunk);
|
|
this.#status.streamFallbacks = (this.#status.streamFallbacks ?? 0) + 1;
|
|
return;
|
|
}
|
|
|
|
let promptStarted = false;
|
|
let completedAnswer = '';
|
|
try {
|
|
await this.#channel.stream(chatId, {
|
|
markdown: async (controller) => {
|
|
promptStarted = true;
|
|
const askOptions = {
|
|
...this.#interactionAskOptions(event, key),
|
|
onUpdate: async (update) => {
|
|
await controller.setContent(this.#progressText(update));
|
|
this.#status.streamUpdates = (this.#status.streamUpdates ?? 0) + 1;
|
|
},
|
|
};
|
|
({ answer: completedAnswer } = await askInWorkspaceSession({
|
|
harness: this.#harness,
|
|
state: this.#state,
|
|
key,
|
|
text,
|
|
createOptions: { signal: this.#signal },
|
|
existsOptions: { signal: this.#signal },
|
|
askOptions,
|
|
}));
|
|
await controller.setContent(completedAnswer);
|
|
},
|
|
}, { replyTo: messageId });
|
|
this.#status.streamResponses = (this.#status.streamResponses ?? 0) + 1;
|
|
} catch (error) {
|
|
this.#status.streamErrors = (this.#status.streamErrors ?? 0) + 1;
|
|
if (completedAnswer) {
|
|
this.#logger.warn?.(
|
|
'[dsh-feishu] native stream failed after generation; sending final text:',
|
|
error.message,
|
|
);
|
|
for (const chunk of splitText(completedAnswer)) await this.#send(chatId, chunk);
|
|
this.#status.streamFallbacks = (this.#status.streamFallbacks ?? 0) + 1;
|
|
return;
|
|
}
|
|
if (promptStarted) throw error;
|
|
|
|
this.#logger.warn?.('[dsh-feishu] native stream unavailable; using text fallback:', error.message);
|
|
const { answer } = await askInWorkspaceSession({
|
|
harness: this.#harness,
|
|
state: this.#state,
|
|
key,
|
|
text,
|
|
createOptions: { signal: this.#signal },
|
|
existsOptions: { signal: this.#signal },
|
|
askOptions: this.#interactionAskOptions(event, key),
|
|
});
|
|
for (const chunk of splitText(answer)) await this.#send(chatId, chunk);
|
|
this.#status.streamFallbacks = (this.#status.streamFallbacks ?? 0) + 1;
|
|
}
|
|
}
|
|
|
|
async #processInteractionReply(event, messageId, key, expected, processingReaction) {
|
|
this.#signal?.throwIfAborted();
|
|
const current = this.#pendingInteractions.get(key);
|
|
const claimed = expected.claimedReplyMessageId === messageId;
|
|
if (!current || current !== expected || current.submitting) {
|
|
if (this.#isResolvedQuestionReply(event, key)) {
|
|
return this.#discardResolvedInteractionReply(event, messageId);
|
|
}
|
|
if (claimed && (!current || current !== expected)) {
|
|
return this.#discardResolvedInteractionReply(event, messageId);
|
|
}
|
|
return this.#enqueueMessage(event, messageId, key, processingReaction, {
|
|
releaseMessageId: false,
|
|
finalize: false,
|
|
});
|
|
}
|
|
if (this.#state.hasSeen(messageId)) return;
|
|
await this.#state.markSeen(messageId);
|
|
this.#status.lastMessageAt = new Date().toISOString();
|
|
this.#status.messagesReceived += 1;
|
|
|
|
const text = extractText(event);
|
|
if (!text) {
|
|
await this.#send(event.message.chat_id, '请用文字回答当前问题。');
|
|
return;
|
|
}
|
|
|
|
const pending = this.#pendingInteractions.get(key);
|
|
if (!pending || pending !== expected || pending.submitting) {
|
|
if (this.#isResolvedQuestionReply(event, key)) {
|
|
await this.#send(event.message.chat_id, INTERACTION_RESOLVED_TEXT).catch(() => undefined);
|
|
return;
|
|
}
|
|
if (claimed && (!pending || pending !== expected)) {
|
|
await this.#send(event.message.chat_id, INTERACTION_RESOLVED_TEXT);
|
|
return;
|
|
}
|
|
return this.#enqueueMessage(event, messageId, key, processingReaction, {
|
|
releaseMessageId: false,
|
|
alreadyRecorded: true,
|
|
finalize: false,
|
|
});
|
|
}
|
|
pending.chatId = event.message.chat_id;
|
|
if (pending.needsPresentation) {
|
|
try {
|
|
await this.#presentInteraction(pending);
|
|
} catch {
|
|
this.#status.lastError = '飞书交互问题发送失败。';
|
|
this.#logger.error?.('[dsh-feishu] failed to retry an interaction question');
|
|
pending.interaction.reconnect?.();
|
|
}
|
|
return;
|
|
}
|
|
const question = pending.questions[pending.index];
|
|
if (!question) return;
|
|
|
|
pending.answers.push(harnessAnswerForQuestion(question, text));
|
|
pending.index += 1;
|
|
if (pending.index < pending.questions.length) {
|
|
if (pending.claimedReplyMessageId === messageId) {
|
|
pending.claimedReplyMessageId = null;
|
|
}
|
|
pending.needsPresentation = true;
|
|
try {
|
|
await this.#presentInteraction(pending);
|
|
} catch {
|
|
this.#status.lastError = '飞书交互问题发送失败。';
|
|
this.#logger.error?.('[dsh-feishu] failed to send the next interaction question');
|
|
pending.interaction.reconnect?.();
|
|
}
|
|
return;
|
|
}
|
|
|
|
pending.submitting = true;
|
|
try {
|
|
await pending.interaction.respond({
|
|
ok: true,
|
|
value: {
|
|
sessionId: pending.sessionId,
|
|
answer: { answers: pending.answers },
|
|
},
|
|
});
|
|
this.#rememberResolvedInteraction(key, pending);
|
|
this.#clearPendingInteraction(key, pending.interactionId);
|
|
this.#status.lastError = null;
|
|
} catch (error) {
|
|
if (this.#signal?.aborted) return;
|
|
if (this.#pendingInteractions.get(key) !== pending) return;
|
|
if (error?.code === 'interaction-not-pending') {
|
|
this.#rememberResolvedInteraction(key, pending);
|
|
this.#clearPendingInteraction(key, pending.interactionId);
|
|
await this.#send(event.message.chat_id, INTERACTION_RESOLVED_TEXT).catch(() => undefined);
|
|
return;
|
|
}
|
|
pending.submitting = false;
|
|
pending.answers.pop();
|
|
pending.index -= 1;
|
|
this.#status.lastError = '回答提交失败。';
|
|
this.#logger.error?.('[dsh-feishu] failed to answer a Harness interaction');
|
|
await this.#send(event.message.chat_id, '回答提交失败，请重新发送当前问题的答案。')
|
|
.catch(() => undefined);
|
|
}
|
|
}
|
|
|
|
async #handleInteraction(interaction, {
|
|
key,
|
|
actor,
|
|
chatId,
|
|
requiresMention,
|
|
}) {
|
|
if (await this.#approvals.handleRequested(interaction, {
|
|
key,
|
|
actor,
|
|
requiresMention,
|
|
send: (text) => this.#send(chatId, text),
|
|
})) return;
|
|
|
|
// Approval requests return above; the existing question state machine stays unchanged.
|
|
if (interaction?.kind !== 'question') return;
|
|
const questions = interaction?.payload?.questions;
|
|
const interactionId = typeof interaction?.interactionId === 'string'
|
|
? interaction.interactionId
|
|
: interaction?.rpcId;
|
|
if (typeof interaction.rpcId !== 'string'
|
|
|| typeof interactionId !== 'string'
|
|
|| typeof interaction.sessionId !== 'string'
|
|
|| !Array.isArray(questions)
|
|
|| questions.length === 0
|
|
|| questions.some((question) => !validHarnessQuestion(question))) {
|
|
this.#logger.warn?.('[dsh-feishu] ignored an invalid Harness question interaction');
|
|
return;
|
|
}
|
|
|
|
if (interaction.recovered === true) {
|
|
await interaction.respond({
|
|
ok: false,
|
|
error: {
|
|
code: 'cancelled',
|
|
message: 'Feishu safely cancelled an interaction left by an earlier client.',
|
|
details: {},
|
|
},
|
|
});
|
|
await this.#send(
|
|
chatId,
|
|
'检测到这个 Session 中遗留的待回答问题，已安全取消并继续处理你刚才的消息。',
|
|
).catch(() => undefined);
|
|
return;
|
|
}
|
|
|
|
const existing = this.#pendingInteractions.get(key);
|
|
if (existing?.interactionId === interactionId) {
|
|
existing.interaction = interaction;
|
|
if (existing.needsPresentation) await this.#presentInteraction(existing);
|
|
return;
|
|
}
|
|
if (this.#interactionKeys.has(interactionId)) return;
|
|
if (existing) {
|
|
await interaction.respond({
|
|
ok: false,
|
|
error: {
|
|
code: 'cancelled',
|
|
message: 'Feishu is already handling another user interaction.',
|
|
details: {},
|
|
},
|
|
});
|
|
return;
|
|
}
|
|
|
|
const pending = {
|
|
kind: 'question',
|
|
interactionId,
|
|
sessionId: interaction.sessionId,
|
|
interaction,
|
|
key,
|
|
actor,
|
|
requiresMention,
|
|
questions,
|
|
answers: [],
|
|
index: 0,
|
|
chatId,
|
|
queue: null,
|
|
claimedReplyMessageId: null,
|
|
submitting: false,
|
|
needsPresentation: true,
|
|
questionMessageIds: new Set(),
|
|
inactive: false,
|
|
};
|
|
this.#pendingInteractions.set(key, pending);
|
|
this.#interactionKeys.set(pending.interactionId, key);
|
|
await this.#presentInteraction(pending);
|
|
}
|
|
|
|
async #handleInteractionResolved(resolution) {
|
|
if (await this.#approvals.handleResolved(resolution)) return;
|
|
const interactionId = resolution?.interactionId;
|
|
if (resolution?.kind !== 'question' || typeof interactionId !== 'string') return;
|
|
const key = this.#interactionKeys.get(interactionId);
|
|
if (!key) return;
|
|
const pending = this.#pendingInteractions.get(key);
|
|
if (pending) this.#rememberResolvedInteraction(key, pending);
|
|
this.#clearPendingInteraction(key, interactionId);
|
|
}
|
|
|
|
async #presentInteraction(pending) {
|
|
const question = pending.questions[pending.index];
|
|
if (!question) return;
|
|
const messageId = await this.#send(
|
|
pending.chatId,
|
|
harnessQuestionText(
|
|
question,
|
|
pending.index,
|
|
pending.questions.length,
|
|
{ requiresMention: pending.requiresMention },
|
|
),
|
|
);
|
|
if (messageId) {
|
|
pending.questionMessageIds.add(messageId);
|
|
if (pending.inactive) this.#rememberResolvedInteraction(pending.key, pending);
|
|
}
|
|
pending.needsPresentation = false;
|
|
}
|
|
|
|
#rememberResolvedInteraction(key, pending) {
|
|
const expiresAt = Date.now() + RESOLVED_REPLY_TTL_MS;
|
|
for (const messageId of pending.questionMessageIds ?? []) {
|
|
this.#resolvedQuestionReplies.set(messageId, { key, expiresAt });
|
|
}
|
|
}
|
|
|
|
#isResolvedQuestionReply(event, key) {
|
|
const now = Date.now();
|
|
for (const [messageId, resolution] of this.#resolvedQuestionReplies) {
|
|
if (resolution.expiresAt <= now) this.#resolvedQuestionReplies.delete(messageId);
|
|
}
|
|
for (const reference of [event?.message?.parent_id, event?.message?.root_id]) {
|
|
const resolution = this.#resolvedQuestionReplies.get(reference);
|
|
if (resolution?.key === key && resolution.expiresAt > now) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
async #discardResolvedInteractionReply(event, messageId) {
|
|
if (this.#state.hasSeen(messageId)) return;
|
|
await this.#state.markSeen(messageId);
|
|
this.#status.lastMessageAt = new Date().toISOString();
|
|
this.#status.messagesReceived += 1;
|
|
await this.#send(event.message.chat_id, INTERACTION_RESOLVED_TEXT).catch(() => undefined);
|
|
}
|
|
|
|
#takePendingInteraction(key, interactionId) {
|
|
const pending = this.#pendingInteractions.get(key);
|
|
if (!pending
|
|
|| (interactionId !== undefined && pending.interactionId !== interactionId)) return null;
|
|
this.#pendingInteractions.delete(key);
|
|
this.#interactionKeys.delete(pending.interactionId);
|
|
pending.inactive = true;
|
|
return pending;
|
|
}
|
|
|
|
#clearPendingInteraction(key, interactionId) {
|
|
return this.#takePendingInteraction(key, interactionId) !== null;
|
|
}
|
|
|
|
async #cancelPendingInteraction(key) {
|
|
const pending = this.#takePendingInteraction(key);
|
|
if (!pending || pending.kind !== 'question') return;
|
|
this.#rememberResolvedInteraction(key, pending);
|
|
try {
|
|
await pending.interaction.respond({
|
|
ok: false,
|
|
error: {
|
|
code: 'cancelled',
|
|
message: 'The Feishu interaction ended before the user answered.',
|
|
details: {},
|
|
},
|
|
}, { signal: AbortSignal.timeout(5_000) });
|
|
} catch (error) {
|
|
if (error?.code !== 'interaction-not-pending') {
|
|
this.#logger.warn?.('[dsh-feishu] failed to cancel a pending Harness interaction');
|
|
}
|
|
}
|
|
}
|
|
|
|
#progressText(update) {
|
|
if (update.type === 'text' && update.text) return update.text;
|
|
if (update.type === 'tool') {
|
|
if (update.name === 'web_search') return '_正在搜索网络并整理信息…_';
|
|
return `_正在使用 ${update.name || '工具'}…_`;
|
|
}
|
|
return `_${update.text || '正在处理…'}_`;
|
|
}
|
|
|
|
async #addReaction(messageId, emojiType) {
|
|
if (!this.#channel?.addReaction) return null;
|
|
try {
|
|
const reactionId = await this.#channel.addReaction(messageId, emojiType);
|
|
this.#status.reactionsAdded = (this.#status.reactionsAdded ?? 0) + 1;
|
|
return reactionId;
|
|
} catch (error) {
|
|
this.#status.reactionErrors = (this.#status.reactionErrors ?? 0) + 1;
|
|
this.#logger.warn?.(`[dsh-feishu] unable to add ${emojiType} reaction:`, error.message);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
async #removeProcessingReaction(messageId, processingReaction) {
|
|
const reactionId = await processingReaction;
|
|
if (reactionId && this.#channel?.removeReaction) {
|
|
try {
|
|
await this.#channel.removeReaction(messageId, reactionId);
|
|
this.#status.reactionsRemoved = (this.#status.reactionsRemoved ?? 0) + 1;
|
|
} catch (error) {
|
|
this.#status.reactionErrors = (this.#status.reactionErrors ?? 0) + 1;
|
|
this.#logger.warn?.('[dsh-feishu] unable to remove processing reaction:', error.message);
|
|
}
|
|
}
|
|
}
|
|
|
|
async #finishReaction(messageId, processingReaction, finalEmojiType) {
|
|
await this.#removeProcessingReaction(messageId, processingReaction);
|
|
await this.#addReaction(messageId, finalEmojiType);
|
|
}
|
|
|
|
async #send(chatId, text) {
|
|
const response = await this.#client.im.v1.message.create({
|
|
params: { receive_id_type: 'chat_id' },
|
|
data: {
|
|
receive_id: chatId,
|
|
msg_type: 'text',
|
|
content: JSON.stringify({ text }),
|
|
},
|
|
});
|
|
if (response?.code && response.code !== 0) {
|
|
throw new Error(`Feishu send failed: ${response.msg || response.code}`);
|
|
}
|
|
return nonEmptyString(response?.data?.message_id);
|
|
}
|
|
}
|