RRRosmontis
20898d5d21
A throw from the converter used to escape the try/catch and drop the message; degrade to plain text for that chunk instead.
1326 lines
56 KiB
JavaScript
1326 lines
56 KiB
JavaScript
import { mkdirSync, readFileSync, writeFileSync } from 'node:fs';
|
||
import { homedir } from 'node:os';
|
||
import { dirname, join } from 'node:path';
|
||
import { createUserMessage } from '@deepseek-ai/dsh-llm/message';
|
||
import { SessionId } from '@deepseek-ai/dsh-session/types';
|
||
import { isAuthorized } from './auth.js';
|
||
import { resolveApiProxy } from './apiproxy.js';
|
||
import { catalogFromLiveAgents, loadCatalog, truncateButton, visibleSessionsForWorkspace, workspacesWithVisibleSessions, } from './catalog.js';
|
||
import { TelegramClient, } from './client.js';
|
||
import { MSG, LAST_CB, parseCommand } from './commands.js';
|
||
import { markdownToHtml, splitMessage } from './format.js';
|
||
import { splitRichMarkdown } from './rich-format.js';
|
||
import { formatLastTurn, loadLastTurn } from './history.js';
|
||
import { describeAgent, displayLabel } from './label.js';
|
||
import { formatModel, loadSessionModels, selectSessionModel, } from './models.js';
|
||
const WS_CB = 'ws:';
|
||
const SID_CB = 'sid:';
|
||
const BACK_WS_CB = 'wb';
|
||
const MODEL_CB = 'mdl:';
|
||
/** Use eff: (not me:) — short prefix, no collision with other callbacks. */
|
||
const EFFORT_CB = 'eff:';
|
||
const BACK_MODEL_CB = 'mb';
|
||
const MAX_BUTTONS = 40;
|
||
function lastContextKeyboard() {
|
||
return {
|
||
inline_keyboard: [[{ text: '查看上次对话', callback_data: LAST_CB }]],
|
||
};
|
||
}
|
||
const defaultSleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
|
||
function contentToText(content) {
|
||
return content
|
||
.filter((block) => block.type === 'text')
|
||
.map((block) => block.text)
|
||
.join('');
|
||
}
|
||
function isRichUnsupportedError(err) {
|
||
const message = err instanceof Error ? err.message : String(err);
|
||
return /method not found|unknown method|not found|404|bad request|can't parse|rich message/i.test(message);
|
||
}
|
||
export class TelegramBridge {
|
||
ctx;
|
||
token;
|
||
allowedUserIds;
|
||
allowAllUsers;
|
||
client;
|
||
sleep;
|
||
maxMessageLength;
|
||
renderingMode;
|
||
bindings = new Map();
|
||
pickers = new Map();
|
||
/** chatId → model awaiting reasoning-effort pick (kept outside picker so list refreshes won't drop it). */
|
||
pendingModels = new Map();
|
||
polling = false;
|
||
offset;
|
||
pollPromise;
|
||
pollAbort;
|
||
disposeSessionListener;
|
||
// ── Stability & interactivity additions ──
|
||
/** sessionIds mid-turn (busy feedback, /stop state) */
|
||
busySessions = new Set();
|
||
/** chatId → typing heartbeat interval handle */
|
||
heartbeats = new Map();
|
||
/** chatId → serialized notice chain (prevents 429 storms) */
|
||
noticeQueue = new Map();
|
||
pendingAsks = new Map();
|
||
hookTimer;
|
||
pendingApprovalsTG = new Map();
|
||
disposeApprovalHook;
|
||
/** sessionId → thinking indicator state (one notice per reasoning phase) */
|
||
thinkingSessions = new Map();
|
||
/** callId → tool name (tool/result failure notices) */
|
||
callNames = new Map();
|
||
/** sessionId → latest todo snapshot (/mission) */
|
||
lastTodos = new Map();
|
||
constructor(ctx, options) {
|
||
this.ctx = ctx;
|
||
this.token = options.token;
|
||
this.allowedUserIds = options.allowedUserIds;
|
||
this.allowAllUsers = options.allowAllUsers;
|
||
this.client = options.client ?? new TelegramClient(options.token, {
|
||
pollingTimeoutSec: options.pollingTimeoutSec ?? 30,
|
||
});
|
||
this.sleep = options.sleep ?? defaultSleep;
|
||
this.maxMessageLength = options.maxMessageLength ?? 4096;
|
||
this.renderingMode = options.rendering === 'html' ? 'html' : 'rich';
|
||
}
|
||
start() {
|
||
this.disposeSessionListener?.();
|
||
this.disposeSessionListener = this.ctx.on('session/event', (session, event) => {
|
||
void this.onSessionEvent(session, event).catch((err) => {
|
||
this.ctx.logger.error(this.redact(err));
|
||
});
|
||
});
|
||
// Restore persisted bindings (survive hot reload / restart) and install
|
||
// the TG answering hooks (ask_user dual-path + approval dual-path).
|
||
this.loadBindings();
|
||
this.hookUserQuestions();
|
||
this.disposeApprovalHook?.();
|
||
this.disposeApprovalHook = this.ctx.on('approval/request', (req, next) => this.onApprovalRequest(req, next));
|
||
void this.client.setMyCommands([
|
||
{ command: 'start', description: '欢迎与用法' },
|
||
{ command: 'sessions', description: '按工作区列出并附着会话' },
|
||
{ command: 'last', description: '查看上次问答（续接上下文）' },
|
||
{ command: 'model', description: '切换当前绑定会话的模型' },
|
||
{ command: 'status', description: '查看当前绑定' },
|
||
{ command: 'unbind', description: '断开手机绑定（不关闭本机会话）' },
|
||
{ command: 'stop', description: '中止当前正在运行的任务' },
|
||
{ command: 'mission', description: '查看任务清单与完成情况' },
|
||
{ command: 'new', description: '在当前工作区新开对话并附着' },
|
||
{ command: 'cancel', description: '关闭 TG 端作答/审批（Web 端仍可答）' },
|
||
{ command: 'help', description: '显示帮助' },
|
||
]).then(() => {
|
||
this.ctx.logger.info('dsh-telegram-channel: bot commands registered');
|
||
}).catch((err) => {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: setMyCommands failed: ${this.redact(err)}`);
|
||
});
|
||
if (!this.polling) {
|
||
this.polling = true;
|
||
this.pollAbort = new AbortController();
|
||
this.ctx.logger.info('dsh-telegram-channel: long-polling started');
|
||
this.pollPromise = this.pollLoop();
|
||
}
|
||
}
|
||
async stop() {
|
||
this.polling = false;
|
||
this.pollAbort?.abort();
|
||
this.pollAbort = undefined;
|
||
this.disposeSessionListener?.();
|
||
this.disposeSessionListener = undefined;
|
||
// Never dispose host agents — only clear remote bindings.
|
||
this.bindings.clear();
|
||
this.pickers.clear();
|
||
this.pendingModels.clear();
|
||
this.stopAllHeartbeats();
|
||
this.busySessions.clear();
|
||
this.noticeQueue.clear();
|
||
this.pendingAsks.clear();
|
||
this.pendingApprovalsTG.clear();
|
||
this.thinkingSessions.clear();
|
||
this.callNames.clear();
|
||
this.lastTodos.clear();
|
||
if (this.hookTimer) {
|
||
clearTimeout(this.hookTimer);
|
||
this.hookTimer = undefined;
|
||
}
|
||
this.disposeApprovalHook?.();
|
||
this.disposeApprovalHook = undefined;
|
||
if (this.pollPromise) {
|
||
await this.pollPromise.catch(() => { });
|
||
this.pollPromise = undefined;
|
||
}
|
||
}
|
||
async processUpdate(update) {
|
||
if (update.callback_query) {
|
||
await this.handleCallback(update);
|
||
return;
|
||
}
|
||
const message = update.message;
|
||
if (!message)
|
||
return;
|
||
const chatId = message.chat.id;
|
||
const userId = message.from?.id;
|
||
if (!isAuthorized({ allowAllUsers: this.allowAllUsers, allowedUserIds: this.allowedUserIds, userId })) {
|
||
await this.client.sendMessage(chatId, MSG.DENIED);
|
||
return;
|
||
}
|
||
// Non-text messages: tell the (authorized) user instead of dropping silently.
|
||
if (!message.text) {
|
||
await this.enqueueNotice(chatId, '暂只支持文本消息（图片/语音/文件等媒体暂不处理）');
|
||
return;
|
||
}
|
||
const parsed = parseCommand(message.text);
|
||
switch (parsed.type) {
|
||
case 'start':
|
||
await this.client.sendMessage(chatId, MSG.WELCOME);
|
||
return;
|
||
case 'help':
|
||
await this.client.sendMessage(chatId, MSG.HELP);
|
||
return;
|
||
case 'sessions':
|
||
await this.sendWorkspacePicker(chatId);
|
||
return;
|
||
case 'last':
|
||
await this.sendLastTurn(chatId);
|
||
return;
|
||
case 'model':
|
||
await this.sendModelPicker(chatId);
|
||
return;
|
||
case 'status':
|
||
await this.sendStatus(chatId);
|
||
return;
|
||
case 'unbind':
|
||
this.bindings.delete(String(chatId));
|
||
this.pickers.delete(String(chatId));
|
||
this.pendingAsks.delete(String(chatId));
|
||
this.saveBindings();
|
||
await this.client.sendMessage(chatId, MSG.UNBOUND);
|
||
return;
|
||
case 'stop':
|
||
await this.stopBound(chatId);
|
||
return;
|
||
case 'mission':
|
||
await this.sendMission(chatId);
|
||
return;
|
||
case 'new':
|
||
await this.newSessionHere(chatId);
|
||
return;
|
||
case 'cancel': {
|
||
// /cancel closes the TG-side answer/approval channel; the UI side stays open.
|
||
let cancelled = false;
|
||
if (this.pendingApprovalsTG.delete(String(chatId)))
|
||
cancelled = true;
|
||
if (this.pendingAsks.delete(String(chatId)))
|
||
cancelled = true;
|
||
if (cancelled)
|
||
await this.enqueueNotice(chatId, '已关闭 TG 端作答/审批（该问题仍可在 Web 端回答）');
|
||
else
|
||
await this.client.sendMessage(chatId, '当前没有待作答的提问或审批');
|
||
return;
|
||
}
|
||
case 'unknown':
|
||
await this.client.sendMessage(chatId, MSG.unknown(parsed.command));
|
||
return;
|
||
case 'plain': {
|
||
// A pending approval/ask on TG consumes the message as an answer.
|
||
const approval = this.pendingApprovalsTG.get(String(chatId));
|
||
if (approval) {
|
||
await this.handleTgApproval(chatId, approval, parsed.text);
|
||
return;
|
||
}
|
||
const pending = this.pendingAsks.get(String(chatId));
|
||
if (pending) {
|
||
await this.handleTgAnswer(chatId, pending, parsed.text);
|
||
return;
|
||
}
|
||
await this.followupBound(chatId, parsed.text);
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
async handleCallback(update) {
|
||
const cq = update.callback_query;
|
||
const userId = cq.from.id;
|
||
const chatId = cq.message?.chat.id;
|
||
if (chatId === undefined) {
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
return;
|
||
}
|
||
if (!isAuthorized({ allowAllUsers: this.allowAllUsers, allowedUserIds: this.allowedUserIds, userId })) {
|
||
await this.client.answerCallbackQuery(cq.id, MSG.DENIED);
|
||
await this.client.sendMessage(chatId, MSG.DENIED);
|
||
return;
|
||
}
|
||
const data = cq.data ?? '';
|
||
const picker = this.pickers.get(String(chatId));
|
||
if (data === LAST_CB) {
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
await this.sendLastTurn(chatId);
|
||
return;
|
||
}
|
||
if (data === BACK_WS_CB) {
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
await this.sendWorkspacePicker(chatId, picker?.catalog);
|
||
return;
|
||
}
|
||
if (data === BACK_MODEL_CB) {
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
await this.sendModelPicker(chatId);
|
||
return;
|
||
}
|
||
if (data.startsWith(WS_CB)) {
|
||
const index = Number(data.slice(WS_CB.length));
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
await this.sendSessionPicker(chatId, index);
|
||
return;
|
||
}
|
||
if (data.startsWith(SID_CB)) {
|
||
const index = Number(data.slice(SID_CB.length));
|
||
const row = picker?.sessions[index];
|
||
if (!row) {
|
||
await this.client.answerCallbackQuery(cq.id, '会话已过期');
|
||
await this.client.sendMessage(chatId, MSG.PICKER_STALE);
|
||
return;
|
||
}
|
||
await this.bindSession(chatId, cq.id, row);
|
||
return;
|
||
}
|
||
if (data.startsWith(MODEL_CB)) {
|
||
const index = Number(data.slice(MODEL_CB.length));
|
||
const option = picker?.models?.[index];
|
||
if (!option) {
|
||
await this.client.answerCallbackQuery(cq.id, '列表已过期');
|
||
await this.client.sendMessage(chatId, MSG.PICKER_STALE);
|
||
return;
|
||
}
|
||
const efforts = (option.efforts ?? []).filter((e) => e.id);
|
||
if (efforts.length === 1) {
|
||
// Single effort (often only "off") — apply immediately, no second tap.
|
||
await this.applyModel(chatId, cq.id, { ...option, efforts }, efforts[0].id);
|
||
return;
|
||
}
|
||
if (efforts.length > 1) {
|
||
const pending = { ...option, efforts };
|
||
this.pendingModels.set(String(chatId), pending);
|
||
if (picker)
|
||
picker.pendingModel = pending;
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
await this.sendEffortPicker(chatId, pending);
|
||
return;
|
||
}
|
||
await this.applyModel(chatId, cq.id, option);
|
||
return;
|
||
}
|
||
if (data.startsWith(EFFORT_CB) || data.startsWith('me:')) {
|
||
// Accept legacy "me:" callbacks from older bot messages.
|
||
const rawIndex = data.startsWith(EFFORT_CB)
|
||
? data.slice(EFFORT_CB.length)
|
||
: data.slice('me:'.length);
|
||
const index = Number(rawIndex);
|
||
const pending = this.pendingModels.get(String(chatId)) ?? picker?.pendingModel;
|
||
const effort = pending?.efforts?.[index];
|
||
if (!pending || !effort?.id) {
|
||
await this.client.answerCallbackQuery(cq.id, '列表已过期');
|
||
await this.client.sendMessage(chatId, MSG.PICKER_STALE);
|
||
return;
|
||
}
|
||
await this.applyModel(chatId, cq.id, pending, effort.id);
|
||
return;
|
||
}
|
||
// Legacy bind:<sessionId> callbacks (older messages / tests)
|
||
if (data.startsWith('bind:')) {
|
||
const sessionId = data.slice('bind:'.length);
|
||
await this.bindSession(chatId, cq.id, {
|
||
sessionId,
|
||
title: sessionId,
|
||
blank: false,
|
||
running: true,
|
||
updatedAt: 0,
|
||
});
|
||
return;
|
||
}
|
||
await this.client.answerCallbackQuery(cq.id);
|
||
}
|
||
async resolveCatalog() {
|
||
try {
|
||
const fromApi = await loadCatalog(this.ctx);
|
||
if (fromApi)
|
||
return fromApi;
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: catalog via apiProxy failed: ${this.redact(err)}`);
|
||
}
|
||
return catalogFromLiveAgents(this.liveAgents(), this.ctx);
|
||
}
|
||
async sendWorkspacePicker(chatId, existing) {
|
||
const catalog = existing ?? await this.resolveCatalog();
|
||
const workspaces = workspacesWithVisibleSessions(catalog);
|
||
if (workspaces.length === 0) {
|
||
this.pickers.delete(String(chatId));
|
||
await this.client.sendMessage(chatId, MSG.NO_SESSIONS);
|
||
return;
|
||
}
|
||
const shown = workspaces.slice(0, MAX_BUTTONS);
|
||
this.pickers.set(String(chatId), { workspaces: shown, sessions: [], catalog });
|
||
const keyboard = {
|
||
inline_keyboard: shown.map((ws, i) => ([{
|
||
text: truncateButton(`${i + 1}. ${ws.title}`),
|
||
callback_data: `${WS_CB}${i}`,
|
||
}])),
|
||
};
|
||
const body = [
|
||
catalog.complete
|
||
? `选择工作区（共 ${workspaces.length} 个，与 Web 对齐，已排除归档）：`
|
||
: `选择工作区（共 ${workspaces.length} 个）⚠️ 仅运行中的会话（apiProxy 未就绪，完整列表需插件 ≥0.3.2 并重启 dsh web）：`,
|
||
'',
|
||
...shown.map((ws, i) => {
|
||
const n = visibleSessionsForWorkspace(catalog, ws).length;
|
||
return `${i + 1}. ${ws.title}\n ${ws.path}\n 会话：${n}`;
|
||
}),
|
||
workspaces.length > MAX_BUTTONS ? `\n仅显示前 ${MAX_BUTTONS} 个工作区。` : '',
|
||
'',
|
||
'点下方按钮进入该工作区的会话列表。',
|
||
].filter(Boolean).join('\n');
|
||
await this.client.sendMessage(chatId, body, undefined, keyboard);
|
||
}
|
||
async sendSessionPicker(chatId, workspaceIndex) {
|
||
const picker = this.pickers.get(String(chatId));
|
||
const catalog = picker?.catalog ?? await this.resolveCatalog();
|
||
const workspaces = picker?.workspaces?.length
|
||
? picker.workspaces
|
||
: workspacesWithVisibleSessions(catalog);
|
||
const workspace = workspaces[workspaceIndex];
|
||
if (!workspace) {
|
||
await this.client.sendMessage(chatId, MSG.PICKER_STALE);
|
||
return;
|
||
}
|
||
const sessions = visibleSessionsForWorkspace(catalog, workspace)
|
||
.slice()
|
||
.sort((a, b) => b.updatedAt - a.updatedAt);
|
||
if (sessions.length === 0) {
|
||
await this.client.sendMessage(chatId, MSG.NO_SESSIONS_IN_WS(workspace.title));
|
||
return;
|
||
}
|
||
const shown = sessions.slice(0, MAX_BUTTONS);
|
||
this.pickers.set(String(chatId), { workspaces, sessions: shown, catalog });
|
||
const keyboard = {
|
||
inline_keyboard: [
|
||
...shown.map((row, i) => ([{
|
||
text: truncateButton(`${i + 1}. ${row.title}${row.running ? '' : ' · 冷'}`),
|
||
callback_data: `${SID_CB}${i}`,
|
||
}])),
|
||
[{ text: '← 返回工作区', callback_data: BACK_WS_CB }],
|
||
],
|
||
};
|
||
const body = [
|
||
`工作区：${workspace.title}`,
|
||
workspace.path,
|
||
'',
|
||
`选择会话（共 ${sessions.length} 个）：`,
|
||
'',
|
||
...shown.map((row, i) => {
|
||
const mark = row.running ? '运行中' : '未附着';
|
||
return `${i + 1}. ${row.title}\n ${mark} · …${row.sessionId.slice(-12)}`;
|
||
}),
|
||
sessions.length > MAX_BUTTONS ? `\n仅显示前 ${MAX_BUTTONS} 个会话。` : '',
|
||
'',
|
||
'点下方按钮附着；冷会话会自动 resume（不关闭 Web）。',
|
||
].filter(Boolean).join('\n');
|
||
await this.client.sendMessage(chatId, body, undefined, keyboard);
|
||
}
|
||
async bindSession(chatId, callbackId, row) {
|
||
const agent = await this.ensureLiveAgent(row.sessionId);
|
||
if (!agent) {
|
||
await this.client.answerCallbackQuery(callbackId, '无法附着');
|
||
await this.client.sendMessage(chatId, MSG.RESUME_FAILED);
|
||
return;
|
||
}
|
||
const parts = describeAgent(agent, 0, this.ctx);
|
||
const label = row.title && row.title !== row.sessionId
|
||
? displayLabel({ ...parts, title: row.title })
|
||
: displayLabel(parts);
|
||
this.bindings.set(String(chatId), { chatId, sessionId: String(agent.id), label });
|
||
// Drop stale pending asks when rebinding to another session; persist bindings.
|
||
const stale = this.pendingAsks.get(String(chatId));
|
||
if (stale && stale.sessionId !== String(agent.id))
|
||
this.pendingAsks.delete(String(chatId));
|
||
this.saveBindings();
|
||
await this.client.answerCallbackQuery(callbackId, '已附着');
|
||
await this.client.sendMessage(chatId, MSG.BOUND(label), undefined, lastContextKeyboard());
|
||
}
|
||
async sendLastTurn(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
const agent = await this.ensureLiveAgent(binding.sessionId);
|
||
try {
|
||
const turn = await loadLastTurn(this.ctx, binding.sessionId, agent);
|
||
const text = formatLastTurn(turn);
|
||
await this.deliver(chatId, text);
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: /last failed: ${this.redact(err)}`);
|
||
await this.client.sendMessage(chatId, MSG.LAST_FAILED);
|
||
}
|
||
}
|
||
async sendModelPicker(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
const agent = await this.ensureLiveAgent(binding.sessionId);
|
||
if (!agent) {
|
||
this.bindings.delete(String(chatId));
|
||
await this.client.sendMessage(chatId, MSG.GONE);
|
||
return;
|
||
}
|
||
try {
|
||
const snap = await loadSessionModels(this.ctx, binding.sessionId);
|
||
if (!snap.routable) {
|
||
await this.client.sendMessage(chatId, MSG.MODEL_UNROUTABLE(formatModel(snap.current)));
|
||
return;
|
||
}
|
||
if (snap.options.length === 0) {
|
||
await this.client.sendMessage(chatId, MSG.MODEL_EMPTY(formatModel(snap.current)));
|
||
return;
|
||
}
|
||
const shown = snap.options.slice(0, MAX_BUTTONS);
|
||
const prev = this.pickers.get(String(chatId));
|
||
this.pickers.set(String(chatId), {
|
||
workspaces: prev?.workspaces ?? [],
|
||
sessions: prev?.sessions ?? [],
|
||
catalog: prev?.catalog,
|
||
models: shown,
|
||
});
|
||
const keyboard = {
|
||
inline_keyboard: shown.map((opt, i) => ([{
|
||
text: truncateButton(`${opt.label}${opt.provider === snap.current.provider && opt.model === snap.current.model ? ' ✓' : ''}`),
|
||
callback_data: `${MODEL_CB}${i}`,
|
||
}])),
|
||
};
|
||
const body = [
|
||
`当前模型：${formatModel(snap.current)}`,
|
||
`会话：${binding.label}`,
|
||
'',
|
||
'选择新模型（下一回合生效）：',
|
||
].join('\n');
|
||
await this.client.sendMessage(chatId, body, undefined, keyboard);
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: /model failed: ${this.redact(err)}`);
|
||
const detail = err instanceof Error ? err.message : String(err);
|
||
await this.client.sendMessage(chatId, MSG.MODEL_UNAVAILABLE(detail));
|
||
}
|
||
}
|
||
async sendEffortPicker(chatId, option) {
|
||
const efforts = option.efforts ?? [];
|
||
const keyboard = {
|
||
inline_keyboard: [
|
||
...efforts.map((e, i) => ([{
|
||
text: truncateButton(e.name || e.id),
|
||
callback_data: `${EFFORT_CB}${i}`,
|
||
}])),
|
||
[{ text: '← 返回模型列表', callback_data: BACK_MODEL_CB }],
|
||
],
|
||
};
|
||
await this.client.sendMessage(chatId, `已选 ${option.label}\n请选择 reasoning effort：`, undefined, keyboard);
|
||
}
|
||
async applyModel(chatId, callbackId, option, reasoningEffort) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.answerCallbackQuery(callbackId, '未绑定');
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
// Ensure session is live before selectModel (same as followup path).
|
||
await this.ensureLiveAgent(binding.sessionId);
|
||
try {
|
||
const selected = await selectSessionModel(this.ctx, binding.sessionId, {
|
||
provider: option.provider,
|
||
model: option.model,
|
||
reasoningEffort,
|
||
});
|
||
this.pendingModels.delete(String(chatId));
|
||
const picker = this.pickers.get(String(chatId));
|
||
if (picker)
|
||
picker.pendingModel = undefined;
|
||
await this.client.answerCallbackQuery(callbackId, '已切换');
|
||
await this.client.sendMessage(chatId, MSG.MODEL_SET(formatModel(selected)));
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: selectModel failed: ${this.redact(err)}`);
|
||
const detail = err instanceof Error ? err.message : String(err);
|
||
await this.client.answerCallbackQuery(callbackId, '切换失败');
|
||
await this.client.sendMessage(chatId, MSG.MODEL_FAILED(detail));
|
||
}
|
||
}
|
||
liveAgents() {
|
||
const agents = this.ctx.agents;
|
||
if (typeof agents.roots === 'function') {
|
||
const roots = agents.roots();
|
||
if (roots.length > 0)
|
||
return roots;
|
||
}
|
||
if (typeof agents.list === 'function')
|
||
return agents.list();
|
||
return [];
|
||
}
|
||
findLiveAgent(sessionId) {
|
||
const agents = this.ctx.agents;
|
||
if (typeof agents.get === 'function') {
|
||
try {
|
||
const found = agents.get(SessionId(sessionId));
|
||
if (found)
|
||
return found;
|
||
}
|
||
catch {
|
||
// fall through to list scan
|
||
}
|
||
}
|
||
return this.liveAgents().find((a) => String(a.id) === sessionId);
|
||
}
|
||
/** Resume cold sessions when needed; never dispose the returned handle. */
|
||
async ensureLiveAgent(sessionId) {
|
||
const live = this.findLiveAgent(sessionId);
|
||
if (live)
|
||
return live;
|
||
const agents = this.ctx.agents;
|
||
if (typeof agents.resume !== 'function')
|
||
return undefined;
|
||
try {
|
||
const handle = await agents.resume({ resumeSessionId: SessionId(sessionId) });
|
||
return handle.agent;
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: resume failed for ${sessionId}: ${this.redact(err)}`);
|
||
return undefined;
|
||
}
|
||
}
|
||
async sendStatus(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.STATUS_NONE);
|
||
return;
|
||
}
|
||
const stillLive = this.findLiveAgent(binding.sessionId);
|
||
if (!stillLive) {
|
||
await this.client.sendMessage(chatId, MSG.STATUS_BOUND_COLD(binding.label));
|
||
return;
|
||
}
|
||
await this.client.sendMessage(chatId, MSG.STATUS_BOUND(binding.label));
|
||
}
|
||
async followupBound(chatId, text) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
const agent = await this.ensureLiveAgent(binding.sessionId);
|
||
if (!agent) {
|
||
this.bindings.delete(String(chatId));
|
||
await this.client.sendMessage(chatId, MSG.GONE);
|
||
return;
|
||
}
|
||
const message = createUserMessage({
|
||
content: [{ type: 'text', text }],
|
||
source: { kind: 'user' },
|
||
});
|
||
// Busy feedback: mid-turn followups queue silently by design — tell the user.
|
||
const busy = this.busySessions.has(binding.sessionId)
|
||
|| Boolean(agent.phase?.kind
|
||
&& agent.phase?.kind !== 'idle');
|
||
agent.followup(message);
|
||
if (busy) {
|
||
await this.enqueueNotice(chatId, '已加入队列：当前任务完成后将处理这条消息');
|
||
}
|
||
}
|
||
async pollLoop() {
|
||
const signal = this.pollAbort?.signal;
|
||
let errorCount = 0;
|
||
while (this.polling) {
|
||
try {
|
||
const updates = await this.client.getUpdates(this.offset);
|
||
if (!this.polling)
|
||
break;
|
||
errorCount = 0;
|
||
if (updates.length === 0) {
|
||
await this.interruptibleDelay(50, signal);
|
||
continue;
|
||
}
|
||
for (const update of updates) {
|
||
if (!this.polling)
|
||
break;
|
||
await this.processUpdate(update);
|
||
this.offset = update.update_id + 1;
|
||
}
|
||
}
|
||
catch (err) {
|
||
if (!this.polling)
|
||
break;
|
||
errorCount += 1;
|
||
this.ctx.logger.error(this.redact(err));
|
||
await this.interruptibleSleep(Math.min(1000 * errorCount, 10_000), signal);
|
||
}
|
||
}
|
||
}
|
||
interruptibleDelay(ms, signal) {
|
||
if (signal?.aborted || !this.polling)
|
||
return Promise.resolve();
|
||
return new Promise((resolve) => {
|
||
const timer = setTimeout(resolve, ms);
|
||
signal?.addEventListener('abort', () => {
|
||
clearTimeout(timer);
|
||
resolve();
|
||
}, { once: true });
|
||
});
|
||
}
|
||
async interruptibleSleep(ms, signal) {
|
||
if (signal?.aborted || !this.polling)
|
||
return;
|
||
await Promise.race([
|
||
this.sleep(ms),
|
||
new Promise((resolve) => {
|
||
if (signal?.aborted) {
|
||
resolve();
|
||
return;
|
||
}
|
||
signal?.addEventListener('abort', () => resolve(), { once: true });
|
||
}),
|
||
]);
|
||
}
|
||
async onSessionEvent(session, event) {
|
||
const id = String(session.id);
|
||
const targets = [...this.bindings.values()].filter((b) => b.sessionId === id);
|
||
if (targets.length === 0)
|
||
return;
|
||
if (event.type === 'turn/start') {
|
||
// Typing heartbeat — one sendChatAction expires in ~5s, so refresh it
|
||
// for the whole turn instead of only at start.
|
||
this.busySessions.add(id);
|
||
for (const b of targets)
|
||
this.startTypingHeartbeat(b.chatId);
|
||
return;
|
||
}
|
||
if (event.type === 'turn/end') {
|
||
this.busySessions.delete(id);
|
||
this.thinkingSessions.delete(id);
|
||
for (const b of targets)
|
||
this.stopTypingHeartbeat(b.chatId);
|
||
// Abnormal endings (API failure / abort / crash-recovery) must be
|
||
// visible on the phone — previously they ended in silence.
|
||
const reason = event.data?.reason;
|
||
const kind = reason?.kind;
|
||
if (kind === 'error') {
|
||
const detail = String(reason?.error?.message ?? reason?.error?.code ?? '').replace(/\s+/g, ' ').slice(0, 300);
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, `本轮异常结束（API/内部错误）${detail ? `：${detail}` : ''}`);
|
||
}
|
||
else if (kind === 'aborted') {
|
||
const manual = reason?.reason?.kind === 'user';
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, manual ? '本轮已被手动中止' : '本轮已被中断');
|
||
}
|
||
else if (kind === 'interrupted') {
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, '会话曾被异常退出（崩溃恢复），本轮被标记中断');
|
||
}
|
||
return;
|
||
}
|
||
if (event.type === 'tool/call') {
|
||
// Progress summary; ask_user is owned by the provider hook (avoids duplicates).
|
||
const name = event.data?.name ?? 'tool';
|
||
if (name === 'ask_user_question')
|
||
return;
|
||
this.thinkingSessions.set(id, false);
|
||
if (event.data?.callId !== undefined) {
|
||
if (this.callNames.size > 400) {
|
||
const first = this.callNames.keys().next().value;
|
||
if (first !== undefined)
|
||
this.callNames.delete(first);
|
||
}
|
||
this.callNames.set(String(event.data.callId), String(name));
|
||
}
|
||
const args = typeof event.data?.arguments === 'string' ? event.data.arguments : '';
|
||
const brief = args.replace(/\s+/g, ' ').trim();
|
||
const shown = brief.length > 60 ? `${brief.slice(0, 60)}…` : brief;
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, `> ${name}${shown ? ` ${shown}` : ''}`);
|
||
return;
|
||
}
|
||
if (event.type === 'assistant/message') {
|
||
const text = contentToText(event.data.message.content);
|
||
this.thinkingSessions.set(id, false);
|
||
if (!text)
|
||
return;
|
||
await Promise.all(targets.map((b) => this.deliver(b.chatId, text)));
|
||
return;
|
||
}
|
||
if (event.type === 'assistant/chunk') {
|
||
// Thinking indicator: once per reasoning phase, status only (no content).
|
||
const chunk = event.data?.chunk;
|
||
const ctype = String(chunk?.type ?? '');
|
||
const btype = String(chunk?.block?.type ?? '');
|
||
const isReasoning = ctype.includes('reasoning') || btype.includes('reasoning') || btype.includes('think');
|
||
if (isReasoning && this.thinkingSessions.get(id) !== true) {
|
||
this.thinkingSessions.set(id, true);
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, '> Thinking…');
|
||
}
|
||
return;
|
||
}
|
||
if (event.type === 'tool/result') {
|
||
// Only failures — successful results are too noisy for the phone.
|
||
const data = event.data;
|
||
if (data?.error) {
|
||
const callKey = String(data.message?.tool_use_id ?? data.message?.callId ?? '');
|
||
const name = this.callNames.get(callKey) ?? 'tool';
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, `> 工具 ${name} 失败：${String(data.error.name ?? 'Error')}${data.error.code ? `/${String(data.error.code)}` : ''}`);
|
||
}
|
||
return;
|
||
}
|
||
if (event.type === 'todo/write') {
|
||
const todos = Array.isArray(event.data?.todos) ? event.data.todos : [];
|
||
this.lastTodos.set(id, todos);
|
||
if (todos.length === 0)
|
||
return;
|
||
const done = todos.filter((t) => t?.status === 'completed').length;
|
||
const current = todos.find((t) => t?.status === 'in_progress') ?? todos.find((t) => t?.status === 'pending');
|
||
const cur = current ? String(current.content ?? '').slice(0, 50) : '';
|
||
for (const b of targets)
|
||
this.enqueueNotice(b.chatId, `> 进度 ${done}/${todos.length}${cur ? `：${cur}` : ''}`);
|
||
return;
|
||
}
|
||
}
|
||
async deliver(chatId, markdown) {
|
||
if (this.renderingMode !== 'html') {
|
||
let sentAny = false;
|
||
try {
|
||
const chunks = splitRichMarkdown(markdown);
|
||
for (const chunk of chunks) {
|
||
try {
|
||
// Per-chunk retry: transient network blips must not drop a reply.
|
||
await this.withRetry(() => this.client.sendRichMessage(chatId, chunk));
|
||
sentAny = true;
|
||
}
|
||
catch (err) {
|
||
if (!sentAny && isRichUnsupportedError(err)) {
|
||
this.ctx.logger.warn('dsh-telegram-channel: Rich Message API unavailable, falling back to HTML rendering');
|
||
this.renderingMode = 'html';
|
||
await this.deliverHtml(chatId, markdown);
|
||
return;
|
||
}
|
||
throw err;
|
||
}
|
||
}
|
||
return;
|
||
}
|
||
catch (err) {
|
||
// Persistent rich failure → HTML fallback instead of dropping the message.
|
||
this.ctx.logger.warn(`dsh-telegram-channel: rich deliver failed (${this.redact(err)}), falling back to HTML`);
|
||
}
|
||
}
|
||
await this.deliverHtml(chatId, markdown);
|
||
}
|
||
async deliverHtml(chatId, markdown) {
|
||
const chunks = splitMessage(markdown, this.maxMessageLength);
|
||
for (const chunk of chunks) {
|
||
let html;
|
||
try {
|
||
html = markdownToHtml(chunk);
|
||
}
|
||
catch {
|
||
html = chunk; // Pathological markdown input: degrade to plain text instead of dropping.
|
||
}
|
||
try {
|
||
await this.withRetry(() => this.client.sendMessage(chatId, html, 'HTML'));
|
||
}
|
||
catch {
|
||
try {
|
||
await this.withRetry(() => this.client.sendMessage(chatId, chunk));
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.error(this.redact(err));
|
||
}
|
||
}
|
||
}
|
||
}
|
||
// ── Stability & interactivity helpers ──
|
||
/** Retry an outbound call with capped linear backoff (500ms, 1s, 2s… max 4s). */
|
||
async withRetry(fn, tries = 3) {
|
||
let lastErr;
|
||
for (let attempt = 1; attempt <= tries; attempt++) {
|
||
try {
|
||
return await fn();
|
||
}
|
||
catch (err) {
|
||
lastErr = err;
|
||
if (attempt < tries)
|
||
await this.sleep(Math.min(500 * 2 ** (attempt - 1), 4000));
|
||
}
|
||
}
|
||
throw lastErr;
|
||
}
|
||
escHtml(s) {
|
||
return String(s)
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>');
|
||
}
|
||
/**
|
||
* Send a notice; a leading `> ` renders as a Telegram blockquote (HTML).
|
||
* Falls back to plain text when the HTML send fails — a notice is never lost.
|
||
*/
|
||
async deliverNotice(chatId, text) {
|
||
const quoted = text.startsWith('> ');
|
||
const body = quoted ? text.slice(2) : text;
|
||
try {
|
||
await this.withRetry(() => {
|
||
if (!quoted)
|
||
return this.client.sendMessage(chatId, body);
|
||
return this.client.sendMessage(chatId, `<blockquote>${this.escHtml(body)}</blockquote>`, 'HTML');
|
||
});
|
||
return;
|
||
}
|
||
catch {
|
||
try {
|
||
await this.client.sendMessage(chatId, quoted ? `> ${body}` : body);
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.error(this.redact(err));
|
||
}
|
||
}
|
||
}
|
||
/** Serialized per-chat notice chain — bursts can't race into Telegram 429s. */
|
||
enqueueNotice(chatId, text) {
|
||
const key = String(chatId);
|
||
const prev = this.noticeQueue.get(key) ?? Promise.resolve();
|
||
const next = prev.then(() => this.deliverNotice(chatId, text)).catch(() => { });
|
||
this.noticeQueue.set(key, next);
|
||
return next;
|
||
}
|
||
startTypingHeartbeat(chatId) {
|
||
this.stopTypingHeartbeat(chatId);
|
||
const send = () => {
|
||
void this.client.sendChatAction(chatId, 'typing').catch(() => { });
|
||
};
|
||
send();
|
||
this.heartbeats.set(String(chatId), setInterval(send, 4000));
|
||
}
|
||
stopTypingHeartbeat(chatId) {
|
||
const h = this.heartbeats.get(String(chatId));
|
||
if (h) {
|
||
clearInterval(h);
|
||
this.heartbeats.delete(String(chatId));
|
||
}
|
||
}
|
||
stopAllHeartbeats() {
|
||
for (const h of this.heartbeats.values())
|
||
clearInterval(h);
|
||
this.heartbeats.clear();
|
||
}
|
||
// ── ask_user_question: TG answering via dual-path race with the UI ──
|
||
userQuestions() {
|
||
return this.ctx.userQuestions;
|
||
}
|
||
/**
|
||
* Wrap the UI provider's ask() so Telegram gets a parallel answer path.
|
||
* `Promise.race` decides; the UI path is untouched. The TG promise NEVER
|
||
* settles when there is no bound chat — race would kill the UI's window
|
||
* with that early rejection.
|
||
*/
|
||
hookUserQuestions(attempt = 0) {
|
||
try {
|
||
const provider = this.userQuestions()?.provider;
|
||
if (provider) {
|
||
if (provider.__tgHooked && provider.__tgRealAsk) {
|
||
// Hot reload: re-point the wrapper at THIS bridge instance.
|
||
const realAsk = provider.__tgRealAsk;
|
||
const self = this;
|
||
provider.ask = function (request) {
|
||
const tg = self.registerTgAsk(request);
|
||
let gui;
|
||
try {
|
||
gui = realAsk(request);
|
||
}
|
||
catch (err) {
|
||
tg.reject(err);
|
||
throw err;
|
||
}
|
||
void gui.then(() => self.settleGuiSide(request), () => self.settleGuiSide(request));
|
||
return Promise.race([tg.promise, gui]);
|
||
};
|
||
return;
|
||
}
|
||
const realAsk = provider.ask.bind(provider);
|
||
provider.__tgRealAsk = realAsk;
|
||
const self = this;
|
||
provider.ask = function (request) {
|
||
const tg = self.registerTgAsk(request);
|
||
let gui;
|
||
try {
|
||
gui = realAsk(request);
|
||
}
|
||
catch (err) {
|
||
tg.reject(err);
|
||
throw err;
|
||
}
|
||
void gui.then(() => self.settleGuiSide(request), () => self.settleGuiSide(request));
|
||
return Promise.race([tg.promise, gui]);
|
||
};
|
||
provider.__tgHooked = true;
|
||
this.ctx.logger.info('dsh-telegram-channel: ask_user TG answering hook installed');
|
||
return;
|
||
}
|
||
}
|
||
catch (err) {
|
||
if (attempt === 0)
|
||
this.ctx.logger.warn(`dsh-telegram-channel: userQuestions hook deferred: ${this.redact(err)}`);
|
||
}
|
||
if (attempt >= 30) {
|
||
this.ctx.logger.warn('dsh-telegram-channel: user-questions provider never appeared; TG answering disabled');
|
||
return;
|
||
}
|
||
this.hookTimer = setTimeout(() => this.hookUserQuestions(attempt + 1), 2000);
|
||
}
|
||
registerTgAsk(request) {
|
||
let resolveFn = () => { };
|
||
let rejectFn = () => { };
|
||
const promise = new Promise((resolve, reject) => {
|
||
resolveFn = resolve;
|
||
rejectFn = reject;
|
||
});
|
||
const ask = { resolve: resolveFn, reject: rejectFn, answered: false };
|
||
const req = request;
|
||
const sessionId = req.agent?.id !== undefined ? String(req.agent.id) : undefined;
|
||
const questions = Array.isArray(req.questions) ? req.questions : [];
|
||
const onAbort = () => {
|
||
for (const [chatId, p] of this.pendingAsks) {
|
||
if (p.ask === ask)
|
||
this.pendingAsks.delete(chatId);
|
||
}
|
||
rejectFn(new Error('ask aborted'));
|
||
};
|
||
if (req.signal?.aborted) {
|
||
rejectFn(new Error('ask aborted'));
|
||
return { promise, reject: rejectFn };
|
||
}
|
||
req.signal?.addEventListener('abort', onAbort, { once: true });
|
||
if (sessionId !== undefined && questions.length > 0) {
|
||
let asked = false;
|
||
for (const [chatId, binding] of this.bindings) {
|
||
if (binding.sessionId !== sessionId)
|
||
continue;
|
||
this.pendingAsks.set(String(chatId), { req: request, sessionId, questions, ask, at: Date.now() });
|
||
this.enqueueNotice(Number(chatId), this.formatAskPending(questions));
|
||
asked = true;
|
||
}
|
||
if (asked)
|
||
return { promise, reject: rejectFn };
|
||
}
|
||
// No bound TG chat: never settle — the UI remains the only answer path.
|
||
return { promise, reject: rejectFn };
|
||
}
|
||
settleGuiSide(request) {
|
||
for (const [chatId, p] of [...this.pendingAsks]) {
|
||
if (p.req !== request)
|
||
continue;
|
||
this.pendingAsks.delete(chatId);
|
||
if (!p.ask.answered) {
|
||
this.enqueueNotice(Number(chatId), '该问题已在 Web 端作答，TG 作答通道关闭');
|
||
}
|
||
}
|
||
}
|
||
formatAskPending(questions) {
|
||
const lines = ['待回答（回复字母选项，或直接输入自定义答案；/cancel 关闭 TG 作答）'];
|
||
questions.forEach((q, i) => {
|
||
const prefix = questions.length > 1 ? `Q${i + 1}. ` : '';
|
||
lines.push(`${prefix}${q.question ?? ''}`);
|
||
const opts = Array.isArray(q.options) ? q.options : [];
|
||
if (opts.length === 0) {
|
||
lines.push(' （开放问题，直接输入答案）');
|
||
}
|
||
else {
|
||
opts.forEach((o, j) => lines.push(` [${String.fromCharCode(65 + j)}] ${o.label}${o.description ? ` — ${o.description}` : ''}`));
|
||
if (q.multiSelect)
|
||
lines.push(' （可多选，如：A,B）');
|
||
}
|
||
});
|
||
return lines.join('\n');
|
||
}
|
||
async handleTgAnswer(chatId, pending, text) {
|
||
const { questions, ask } = pending;
|
||
let parts;
|
||
if (questions.length === 1) {
|
||
parts = [[0, text.trim()]];
|
||
}
|
||
else {
|
||
parts = [];
|
||
const lines = text.split(/\n|；|;/).map((s) => s.trim()).filter(Boolean);
|
||
for (const line of lines) {
|
||
const m = line.match(/^([0-9]+)\s*[:：.、)]\s*(.+)$/);
|
||
if (m)
|
||
parts.push([Number(m[1]) - 1, m[2].trim()]);
|
||
else
|
||
parts.push([null, line]);
|
||
}
|
||
if (parts.some(([idx]) => idx === null)) {
|
||
await this.enqueueNotice(chatId, `共 ${questions.length} 个问题，请每行用 1:/2: 前缀分别作答`);
|
||
return;
|
||
}
|
||
}
|
||
const answers = [];
|
||
for (const [idx, raw] of parts) {
|
||
if (idx === null || idx < 0 || idx >= questions.length) {
|
||
await this.enqueueNotice(chatId, `题号 ${(idx ?? 0) + 1} 不存在（共 ${questions.length} 题）`);
|
||
return;
|
||
}
|
||
const q = questions[idx];
|
||
const cleaned = raw.replace(/^\[+/, '').replace(/\]+$/, '').trim();
|
||
const letters = cleaned.split(/[,，、\s]+/).map((s) => s.trim()).filter(Boolean);
|
||
const opts = Array.isArray(q.options) ? q.options : [];
|
||
const isLetterChoice = letters.length > 0
|
||
&& letters.every((s) => /^[a-zA-Z]$/.test(s))
|
||
&& opts.length > 0;
|
||
if (isLetterChoice) {
|
||
const selected = [];
|
||
for (const s of letters) {
|
||
const i = s.toUpperCase().charCodeAt(0) - 65;
|
||
if (i < 0 || i >= opts.length) {
|
||
await this.enqueueNotice(chatId, `选项 ${s.toUpperCase()} 不存在（范围 A-${String.fromCharCode(64 + opts.length)}），请重答或 /cancel`);
|
||
return;
|
||
}
|
||
selected.push(opts[i].label);
|
||
}
|
||
answers.push({ id: q.id, selected });
|
||
}
|
||
else {
|
||
answers.push({ id: q.id, selected: [], custom: cleaned });
|
||
}
|
||
}
|
||
ask.answered = true;
|
||
for (const [cid, p] of this.pendingAsks) {
|
||
if (p.ask === ask)
|
||
this.pendingAsks.delete(cid);
|
||
}
|
||
ask.resolve({ answers });
|
||
this.enqueueNotice(chatId, '已提交你的回答');
|
||
}
|
||
// ── approval: TG answering via dual-path race on the request waterfall ──
|
||
async onApprovalRequest(req, next) {
|
||
const sessionId = req?.agent?.id !== undefined ? String(req.agent.id) : undefined;
|
||
const targets = sessionId !== undefined
|
||
? [...this.bindings.values()].filter((b) => b.sessionId === sessionId).map((b) => String(b.chatId))
|
||
: [];
|
||
let resolveFn = () => { };
|
||
let rejectFn = () => { };
|
||
const promise = new Promise((resolve, reject) => {
|
||
resolveFn = resolve;
|
||
rejectFn = reject;
|
||
});
|
||
const ask = { resolve: resolveFn, reject: rejectFn, answered: false };
|
||
const onAbort = () => {
|
||
for (const [chatId, p] of this.pendingApprovalsTG) {
|
||
if (p.ask === ask)
|
||
this.pendingApprovalsTG.delete(chatId);
|
||
}
|
||
rejectFn(new Error('approval withdrawn'));
|
||
};
|
||
if (req?.signal?.aborted) {
|
||
rejectFn(new Error('approval withdrawn'));
|
||
return next();
|
||
}
|
||
req?.signal?.addEventListener('abort', onAbort, { once: true });
|
||
if (targets.length > 0) {
|
||
const toolName = String(req?.toolName ?? 'tool');
|
||
const reason = req?.reason ? String(req.reason).slice(0, 200) : '';
|
||
for (const chatId of targets) {
|
||
this.pendingApprovalsTG.set(chatId, { toolName, ask });
|
||
this.enqueueNotice(Number(chatId), `权限审批：${toolName}${reason ? `\n(${reason})` : ''}\n回复 [A] 允许一次 / [B] 拒绝（/cancel 关闭 TG 审批）`);
|
||
}
|
||
}
|
||
// No bound TG chat: promise never settles — the UI remains the only answer path.
|
||
const gui = next();
|
||
void gui.then(() => {
|
||
for (const [chatId, p] of [...this.pendingApprovalsTG]) {
|
||
if (p.ask !== ask)
|
||
continue;
|
||
this.pendingApprovalsTG.delete(chatId);
|
||
if (!p.ask.answered) {
|
||
p.ask.reject(new Error('superseded-by-gui'));
|
||
this.enqueueNotice(Number(chatId), '该审批已在 Web 端处理');
|
||
}
|
||
}
|
||
}, () => {
|
||
for (const [chatId, p] of [...this.pendingApprovalsTG]) {
|
||
if (p.ask === ask)
|
||
this.pendingApprovalsTG.delete(chatId);
|
||
}
|
||
});
|
||
return Promise.race([promise, gui]);
|
||
}
|
||
async handleTgApproval(chatId, approval, text) {
|
||
const cleaned = text.trim().replace(/^\[+/, '').replace(/\]+$/, '').toUpperCase();
|
||
if (cleaned === 'A' || cleaned === 'B') {
|
||
approval.ask.answered = true;
|
||
for (const [cid, p] of this.pendingApprovalsTG) {
|
||
if (p.ask === approval.ask)
|
||
this.pendingApprovalsTG.delete(cid);
|
||
}
|
||
if (cleaned === 'A') {
|
||
approval.ask.resolve('allowed-once');
|
||
await this.enqueueNotice(chatId, `已允许 ${approval.toolName} 执行一次`);
|
||
}
|
||
else {
|
||
approval.ask.resolve('rejected');
|
||
await this.enqueueNotice(chatId, '已拒绝该工具执行');
|
||
}
|
||
return;
|
||
}
|
||
await this.enqueueNotice(chatId, '请回复 [A] 允许一次 或 [B] 拒绝（/cancel 关闭）');
|
||
}
|
||
// ── /stop /mission /new commands ──
|
||
async stopBound(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
const agent = this.findLiveAgent(binding.sessionId);
|
||
if (!agent) {
|
||
await this.client.sendMessage(chatId, '当前会话没有正在运行的任务');
|
||
return;
|
||
}
|
||
try {
|
||
agent.cancel({ kind: 'user' });
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: stop failed: ${this.redact(err)}`);
|
||
}
|
||
await this.enqueueNotice(chatId, '已请求中止当前轮…');
|
||
}
|
||
async sendMission(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
const todos = this.lastTodos.get(binding.sessionId) ?? [];
|
||
const lines = ['任务清单'];
|
||
if (todos.length === 0) {
|
||
lines.push('（暂无任务记录，agent 使用 todo 工具后这里会显示）');
|
||
}
|
||
else {
|
||
const mark = { completed: '✓', in_progress: '⟳', pending: '·' };
|
||
todos.forEach((t, i) => lines.push(`${mark[t?.status ?? ''] ?? '·'} ${i + 1}. ${String(t?.content ?? '')}`));
|
||
const done = todos.filter((t) => t?.status === 'completed').length;
|
||
lines.push(`—— ${done}/${todos.length} 已完成`);
|
||
}
|
||
lines.push(this.busySessions.has(binding.sessionId) ? '（当前任务：运行中，/stop 可中止）' : '（当前空闲）');
|
||
await this.client.sendMessage(chatId, lines.join('\n'));
|
||
}
|
||
/** /new: create a session in the bound session's workspace and attach to it. */
|
||
async newSessionHere(chatId) {
|
||
const binding = this.bindings.get(String(chatId));
|
||
if (!binding) {
|
||
await this.client.sendMessage(chatId, MSG.NEED_BIND);
|
||
return;
|
||
}
|
||
let workspaceId;
|
||
let fallbackCwd;
|
||
try {
|
||
const catalog = await this.resolveCatalog();
|
||
for (const ws of catalog?.workspaces ?? []) {
|
||
if ((ws.sessionIds ?? []).includes(binding.sessionId)) {
|
||
workspaceId = ws.id;
|
||
break;
|
||
}
|
||
}
|
||
if (workspaceId === undefined)
|
||
fallbackCwd = catalog?.sessionsById?.get(binding.sessionId)?.cwd;
|
||
}
|
||
catch {
|
||
// Catalog unavailable → fall back to the host default workspace.
|
||
}
|
||
const proxy = resolveApiProxy(this.ctx);
|
||
if (!proxy?.sessions?.create) {
|
||
await this.client.sendMessage(chatId, '当前宿主不支持会话创建接口');
|
||
return;
|
||
}
|
||
let res;
|
||
try {
|
||
const payload = workspaceId !== undefined ? { workspaceId } : (fallbackCwd ? { cwd: fallbackCwd } : {});
|
||
res = await proxy.sessions.create({
|
||
rpcId: `tg-${Date.now()}-${Math.random().toString(36).slice(2)}`,
|
||
payload,
|
||
});
|
||
}
|
||
catch (err) {
|
||
await this.client.sendMessage(chatId, `创建失败：${this.redact(err)}`);
|
||
return;
|
||
}
|
||
const newId = res?.result?.ok ? res.result.value?.sessionId : undefined;
|
||
if (!newId) {
|
||
const detail = res?.result?.error?.message ?? JSON.stringify(res?.result ?? res);
|
||
await this.client.sendMessage(chatId, `创建失败：${String(detail).slice(0, 200)}`);
|
||
return;
|
||
}
|
||
const stale = this.pendingAsks.get(String(chatId));
|
||
if (stale && stale.sessionId !== String(newId))
|
||
this.pendingAsks.delete(String(chatId));
|
||
this.bindings.set(String(chatId), { chatId, sessionId: String(newId), label: binding.label });
|
||
this.saveBindings();
|
||
const where = workspaceId !== undefined ? '当前工作区' : (fallbackCwd ? `目录 ${fallbackCwd}` : '默认工作区');
|
||
await this.client.sendMessage(chatId, `已在${where}新开对话并附着：\n${String(newId)}\n直接发消息即可开始`);
|
||
}
|
||
// ── Binding persistence (survives hot reload / restart) ──
|
||
bindingsPath() {
|
||
// DSH_TELEGRAM_BINDINGS_FILE overrides the location (tests must isolate
|
||
// from the real profile data — parallel test files share this file).
|
||
const override = process.env.DSH_TELEGRAM_BINDINGS_FILE;
|
||
if (override)
|
||
return override;
|
||
const home = process.env.DSH_HOME ?? join(homedir(), '.dsh');
|
||
return join(home, 'telegram-channel-bindings.json');
|
||
}
|
||
saveBindings() {
|
||
try {
|
||
const obj = Object.fromEntries(this.bindings);
|
||
mkdirSync(dirname(this.bindingsPath()), { recursive: true });
|
||
writeFileSync(this.bindingsPath(), JSON.stringify(obj, null, 2));
|
||
}
|
||
catch (err) {
|
||
this.ctx.logger.warn(`dsh-telegram-channel: saveBindings failed: ${this.redact(err)}`);
|
||
}
|
||
}
|
||
loadBindings() {
|
||
try {
|
||
const raw = readFileSync(this.bindingsPath(), 'utf8');
|
||
const obj = JSON.parse(raw);
|
||
let n = 0;
|
||
for (const [chatId, b] of Object.entries(obj ?? {})) {
|
||
if (b && b.sessionId) {
|
||
this.bindings.set(String(chatId), {
|
||
chatId: b.chatId ?? Number(chatId),
|
||
sessionId: String(b.sessionId),
|
||
label: b.label ?? String(b.sessionId),
|
||
});
|
||
n += 1;
|
||
}
|
||
}
|
||
if (n > 0)
|
||
this.ctx.logger.info(`dsh-telegram-channel: restored ${n} binding(s) from disk`);
|
||
}
|
||
catch {
|
||
// First run: no file yet.
|
||
}
|
||
}
|
||
redact(value) {
|
||
const message = value instanceof Error ? value.message : String(value);
|
||
return message.split(this.token).join('***');
|
||
}
|
||
}
|