fix: ship prebuilt lib and drop prepare for one-click git install
Avoids pnpm allowBuilds block on dsh plugin add; README install steps expanded for first-time users. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
owo
+305
@@ -0,0 +1,305 @@
|
||||
import { createUserMessage } from '@deepseek-ai/dsh-llm/message';
|
||||
import { SessionId } from '@deepseek-ai/dsh-session/types';
|
||||
import { isAuthorized } from './auth.js';
|
||||
import { TelegramClient, } from './client.js';
|
||||
import { BIND_CB_PREFIX, MSG, parseCommand } from './commands.js';
|
||||
import { markdownToHtml, splitMessage } from './format.js';
|
||||
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 shortLabel(agent, index) {
|
||||
const id = String(agent.id);
|
||||
const tail = id.length > 12 ? id.slice(-12) : id;
|
||||
const cwd = agent.session?.meta?.cwd;
|
||||
const cwdBit = cwd ? cwd.split(/[/\\]/).filter(Boolean).slice(-1)[0] : '';
|
||||
const base = cwdBit ? `${cwdBit} · ${tail}` : `session ${index + 1} · ${tail}`;
|
||||
return base.length > 60 ? `${base.slice(0, 57)}...` : base;
|
||||
}
|
||||
export class TelegramBridge {
|
||||
ctx;
|
||||
token;
|
||||
allowedUserIds;
|
||||
allowAllUsers;
|
||||
client;
|
||||
sleep;
|
||||
maxMessageLength;
|
||||
bindings = new Map();
|
||||
polling = false;
|
||||
offset;
|
||||
pollPromise;
|
||||
pollAbort;
|
||||
disposeSessionListener;
|
||||
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;
|
||||
}
|
||||
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));
|
||||
});
|
||||
});
|
||||
if (!this.polling) {
|
||||
this.polling = true;
|
||||
this.pollAbort = new AbortController();
|
||||
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();
|
||||
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?.text)
|
||||
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;
|
||||
}
|
||||
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.sendSessionPicker(chatId);
|
||||
return;
|
||||
case 'status':
|
||||
await this.sendStatus(chatId);
|
||||
return;
|
||||
case 'unbind':
|
||||
this.bindings.delete(String(chatId));
|
||||
await this.client.sendMessage(chatId, MSG.UNBOUND);
|
||||
return;
|
||||
case 'unknown':
|
||||
await this.client.sendMessage(chatId, MSG.unknown(parsed.command));
|
||||
return;
|
||||
case 'plain':
|
||||
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 ?? '';
|
||||
if (!data.startsWith(BIND_CB_PREFIX)) {
|
||||
await this.client.answerCallbackQuery(cq.id);
|
||||
return;
|
||||
}
|
||||
const sessionId = data.slice(BIND_CB_PREFIX.length);
|
||||
const agent = this.findLiveAgent(sessionId);
|
||||
if (!agent) {
|
||||
await this.client.answerCallbackQuery(cq.id, '会话不存在');
|
||||
await this.client.sendMessage(chatId, MSG.GONE);
|
||||
return;
|
||||
}
|
||||
const label = shortLabel(agent, 0);
|
||||
this.bindings.set(String(chatId), { chatId, sessionId: String(agent.id), label });
|
||||
await this.client.answerCallbackQuery(cq.id, '已附着');
|
||||
await this.client.sendMessage(chatId, MSG.BOUND(label));
|
||||
}
|
||||
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);
|
||||
}
|
||||
async sendSessionPicker(chatId) {
|
||||
const agents = this.liveAgents();
|
||||
if (agents.length === 0) {
|
||||
await this.client.sendMessage(chatId, MSG.NO_LIVE);
|
||||
return;
|
||||
}
|
||||
const keyboard = {
|
||||
inline_keyboard: agents.map((agent, i) => {
|
||||
const label = shortLabel(agent, i);
|
||||
return [{ text: label, callback_data: `${BIND_CB_PREFIX}${String(agent.id)}` }];
|
||||
}),
|
||||
};
|
||||
await this.client.sendMessage(chatId, `选择要遥控的本机会话（共 ${agents.length} 个）：`, undefined, keyboard);
|
||||
}
|
||||
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) {
|
||||
this.bindings.delete(String(chatId));
|
||||
await this.client.sendMessage(chatId, MSG.GONE);
|
||||
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 = this.findLiveAgent(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' },
|
||||
});
|
||||
agent.followup(message);
|
||||
}
|
||||
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') {
|
||||
await Promise.all(targets.map((b) => this.client.sendChatAction(b.chatId, 'typing')));
|
||||
return;
|
||||
}
|
||||
if (event.type === 'assistant/message') {
|
||||
const text = contentToText(event.data.message.content);
|
||||
if (!text)
|
||||
return;
|
||||
await Promise.all(targets.map((b) => this.deliver(b.chatId, text)));
|
||||
}
|
||||
}
|
||||
async deliver(chatId, markdown) {
|
||||
const chunks = splitMessage(markdown, this.maxMessageLength);
|
||||
for (const chunk of chunks) {
|
||||
const html = markdownToHtml(chunk);
|
||||
try {
|
||||
await this.client.sendMessage(chatId, html, 'HTML');
|
||||
}
|
||||
catch {
|
||||
try {
|
||||
await this.client.sendMessage(chatId, chunk);
|
||||
}
|
||||
catch (err) {
|
||||
this.ctx.logger.error(this.redact(err));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
redact(value) {
|
||||
const message = value instanceof Error ? value.message : String(value);
|
||||
return message.split(this.token).join('***');
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user