@stage-labs/metro 0.1.0-beta.7 → 0.1.0-beta.71

package/runtime/node_modules/@metro-labs/telegram-bot/src/index.ts

@@ -0,0 +1,109 @@
+import { errMsg } from '@metro-labs/mcp/log';
+import { accounts, loadAccounts, tg, type Account } from './accounts.js';
+import { emit } from './wire.js';
+import {
+  emitInbound,
+  envelope,
+  reactionCountEnvelope,
+  reactionEnvelope,
+  saveMediaAndEmit,
+  type TgMsg,
+  type TgReaction,
+  type TgReactionCount,
+} from './format.js';
+import { readCalls } from '@metro-labs/mcp/trains/protocol';
+import { handleCall } from './actions.js';
+
+readCalls('telegram-bot', handleCall);
+
+interface Update {
+  update_id: number;
+  message?: TgMsg;
+  message_reaction?: TgReaction;
+  message_reaction_count?: TgReactionCount;
+}
+
+function handleUpdate(id: string, u: Update): void {
+  if (u.message && !u.message.from?.is_bot) {
+    const env = envelope(id, u.message);
+    emitInbound(emit, id, env);
+    saveMediaAndEmit(emit, id, u.message, env.id as string);
+  }
+  if (u.message_reaction) {
+    const env = reactionEnvelope(id, u.message_reaction);
+    if (env) emitInbound(emit, id, env);
+  }
+  if (u.message_reaction_count) {
+    const env = reactionCountEnvelope(id, u.message_reaction_count);
+    if (env) emitInbound(emit, id, env);
+  }
+}
+
+async function runAccount(acct: Account): Promise<void> {
+  const { id } = acct.cfg;
+  try {
+    await tg(id, 'deleteWebhook', { drop_pending_updates: false });
+  } catch (err) {
+    process.stderr.write(
+      `telegram-bot[${id}] deleteWebhook: ${errMsg(err)}\n`,
+    );
+  }
+  try {
+    const me = await tg<{ username?: string }>(id, 'getMe', {});
+    if (typeof me.username === 'string') acct.username = me.username;
+  } catch (err) {
+    process.stderr.write(`telegram-bot[${id}] getMe: ${errMsg(err)}\n`);
+  }
+
+  for (;;) {
+    try {
+      const updates = await tg<Update[]>(
+        id,
+        'getUpdates',
+        {
+          offset: acct.offset,
+          timeout: 25,
+          allowed_updates: [
+            'message',
+            'message_reaction',
+            'message_reaction_count',
+          ],
+        },
+        60_000,
+      );
+      for (const u of updates) {
+        acct.offset = u.update_id + 1;
+        handleUpdate(id, u);
+      }
+    } catch (err) {
+      process.stderr.write(
+        `telegram-bot[${id}] poll error: ${errMsg(err)}\n`,
+      );
+      await new Promise((r) => setTimeout(r, 2_000));
+    }
+  }
+}
+
+const cfgs = loadAccounts();
+for (const cfg of cfgs) {
+  accounts.set(cfg.id, {
+    cfg,
+    api: `https://api.telegram.org/bot${cfg.token}`,
+    fileApi: `https://api.telegram.org/file/bot${cfg.token}`,
+    offset: 0,
+  });
+}
+if (accounts.size === 0) {
+  process.stderr.write('telegram-bot: no accounts booted, exiting\n');
+  process.exit(2);
+}
+process.stderr.write(
+  `telegram-bot train ready (multi) — ${accounts.size} account(s): ${[...accounts.keys()].join(', ')}\n`,
+);
+
+for (const acct of accounts.values())
+  runAccount(acct).catch((err: unknown) => {
+    process.stderr.write(
+      `telegram-bot[${acct.cfg.id}] account loop failed: ${errMsg(err)}\n`,
+    );
+  });

package/runtime/node_modules/@metro-labs/telegram-bot/src/media-actions.ts

@@ -0,0 +1,166 @@
+import { tg, tgForm, targetOf } from './accounts.js';
+import { emit, mintId, respond, SELF_URI } from './wire.js';
+import { appendFile } from '@metro-labs/mcp/stations/attachments';
+
+export function emitOutbound(
+  accountId: string,
+  line: string,
+  messageId: string,
+  text: string,
+  replyTo?: string,
+): void {
+  emit({
+    kind: 'outbound',
+    id: mintId(),
+    ts: new Date().toISOString(),
+    station: 'telegram-bot',
+    line,
+    from: SELF_URI,
+    to: line,
+    message_id: messageId,
+    text,
+    reply_to: replyTo,
+    ...(replyTo ? { event: { type: 'reply', replyTo } } : {}),
+    account: accountId,
+    payload: { account: accountId },
+  });
+}
+
+export function finishSend(
+  id: string,
+  accountId: string,
+  line: string,
+  messageId: string,
+  label: string,
+  replyTo?: string,
+  extra?: Record<string, unknown>,
+): void {
+  emitOutbound(accountId, line, messageId, label, replyTo);
+  respond(id, {
+    result: { messageId, account: accountId, ...extra },
+  });
+}
+
+interface MediaArgs {
+  line: string;
+  path: string;
+  caption?: string;
+  replyTo?: string;
+  parseMode?: string;
+  account?: string;
+  name?: string;
+}
+
+export async function sendMedia(
+  method: string,
+  fieldName: string,
+  args: Record<string, unknown>,
+): Promise<{ accountId: string; message_id: number }> {
+  const {
+    line,
+    path,
+    caption,
+    replyTo,
+    parseMode,
+    account,
+    name: fileName,
+  } = args as unknown as MediaArgs;
+  const { accountId, chatId, topicId } = targetOf(line, account);
+  const form = new FormData();
+  form.append('chat_id', String(chatId));
+  if (topicId !== undefined) form.append('message_thread_id', String(topicId));
+  if (caption) form.append('caption', caption);
+  if (parseMode) form.append('parse_mode', parseMode);
+  if (replyTo)
+    form.append(
+      'reply_parameters',
+      JSON.stringify({ message_id: Number(replyTo) }),
+    );
+  const name = fileName ?? path.split('/').pop() ?? fieldName;
+  await appendFile(form, fieldName, path, name);
+  const r = await tgForm<{ message_id: number }>(accountId, method, form);
+  return { accountId, message_id: r.message_id };
+}
+
+export async function media(
+  id: string,
+  method: string,
+  field: string,
+  label: string,
+  args: Record<string, unknown>,
+): Promise<void> {
+  const { accountId, message_id } = await sendMedia(method, field, args);
+  const line = (args as { line: string }).line;
+  finishSend(
+    id,
+    accountId,
+    line,
+    String(message_id),
+    label,
+    args.replyTo as string | undefined,
+  );
+}
+
+export const MEDIA_METHOD_FIELD: Record<
+  string,
+  { method: string; field: string }
+> = {
+  image: { method: 'sendPhoto', field: 'photo' },
+  voice: { method: 'sendVoice', field: 'voice' },
+  audio: { method: 'sendAudio', field: 'audio' },
+  video: { method: 'sendVideo', field: 'video' },
+  document: { method: 'sendDocument', field: 'document' },
+};
+
+export async function sendDice(
+  id: string,
+  args: Record<string, unknown>,
+): Promise<void> {
+  const {
+    line,
+    emoji = '\U0001F3B2',
+    account,
+  } = args as { line: string; emoji?: string; account?: string };
+  const { accountId, chatId, topicId } = targetOf(line, account);
+  const body: Record<string, unknown> = { chat_id: chatId, emoji };
+  if (topicId !== undefined) body.message_thread_id = topicId;
+  const r = await tg<{ message_id: number; dice?: { value: number } }>(
+    accountId,
+    'sendDice',
+    body,
+  );
+  finishSend(
+    id,
+    accountId,
+    line,
+    String(r.message_id),
+    `[dice ${emoji} = ${r.dice?.value ?? '?'}]`,
+    undefined,
+    { value: r.dice?.value },
+  );
+}
+
+export async function sendLocation(
+  id: string,
+  args: Record<string, unknown>,
+): Promise<void> {
+  const { line, latitude, longitude, account } = args as {
+    line: string;
+    latitude: number;
+    longitude: number;
+    account?: string;
+  };
+  const { accountId, chatId } = targetOf(line, account);
+  const r = await tg<{ message_id: number }>(accountId, 'sendLocation', {
+    chat_id: chatId,
+    latitude,
+    longitude,
+  });
+  finishSend(
+    id,
+    accountId,
+    line,
+    String(r.message_id),
+    `[location: ${latitude}, ${longitude}]`,
+  );
+}

package/runtime/node_modules/@metro-labs/telegram-bot/src/members.ts

@@ -0,0 +1,47 @@
+import type { MemberList, MetroMember } from '@metro-labs/mcp/stations/types';
+
+export interface TgUser {
+  id: number;
+  username?: string;
+  first_name?: string;
+  last_name?: string;
+  is_bot?: boolean;
+}
+
+export interface TgChatMember {
+  user: TgUser;
+  status: string;
+}
+
+export function mapTgMember(m: TgChatMember): MetroMember {
+  const u = m.user;
+  const member: MetroMember = { id: String(u.id) };
+  if (u.username) member.name = u.username;
+  const display = [u.first_name, u.last_name].filter(Boolean).join(' ');
+  if (display) member.display_name = display;
+  member.is_admin = m.status === 'creator' || m.status === 'administrator';
+  member.is_bot = Boolean(u.is_bot);
+  return member;
+}
+
+const BOT_LIMIT_REASON =
+  'Telegram Bot API cannot enumerate full membership; returning administrators only.';
+
+export function adminMemberList(
+  admins: TgChatMember[],
+  total: number | null,
+): MemberList {
+  return {
+    members: admins.map(mapTgMember),
+    capability: {
+      supported: true,
+      complete: false,
+      reason: BOT_LIMIT_REASON,
+      ...(total != null ? { total } : {}),
+    },
+  };
+}
+
+export function inaccessibleMemberList(reason: string): MemberList {
+  return { members: [], capability: { supported: false, complete: false, reason } };
+}

package/runtime/node_modules/@metro-labs/telegram-bot/src/station.ts

@@ -0,0 +1,17 @@
+import type { Station, Verb } from '@metro-labs/mcp/stations/types';
+
+export const telegramBotStation: Station = {
+  name: 'telegram-bot',
+  hasAccounts: true,
+  hasTrain: true,
+  messageVerbs: new Set<Verb>([
+    'send',
+    'reply',
+    'react',
+    'unreact',
+    'edit',
+    'delete',
+  ]),
+  attachmentMode: 'canonical',
+  tools: [],
+};

package/runtime/node_modules/@metro-labs/telegram-bot/src/types.ts

@@ -0,0 +1,58 @@
+export interface TgEntity {
+  type: string;
+  offset: number;
+  length: number;
+  user?: { id: number };
+}
+
+export interface TgMsg {
+  entities?: TgEntity[];
+  caption_entities?: TgEntity[];
+  reply_to_message?: { message_id: number; from?: { id: number; username?: string; is_bot?: boolean } };
+  message_id: number;
+  date: number;
+  chat: { id: number; type: string; title?: string; first_name?: string };
+  from?: {
+    id: number;
+    username?: string;
+    first_name?: string;
+    is_bot?: boolean;
+  };
+  text?: string;
+  caption?: string;
+  message_thread_id?: number;
+  is_topic_message?: boolean;
+  photo?: { file_id: string; file_size?: number }[];
+  document?: { file_name?: string; file_id?: string; mime_type?: string };
+  voice?: { file_id?: string; duration?: number; mime_type?: string };
+  audio?: { file_id?: string; file_name?: string; mime_type?: string };
+  video?: { file_id?: string; file_name?: string; mime_type?: string };
+  animation?: { file_id?: string; file_name?: string; mime_type?: string };
+  sticker?: { file_id?: string; emoji?: string; set_name?: string };
+  location?: { latitude: number; longitude: number };
+  dice?: { emoji: string; value: number };
+}
+
+export interface TgReaction {
+  chat: { id: number; type: string };
+  message_id: number;
+  user?: {
+    id: number;
+    username?: string;
+    first_name?: string;
+    is_bot?: boolean;
+  };
+  date: number;
+  old_reaction: { type: string; emoji?: string }[];
+  new_reaction: { type: string; emoji?: string }[];
+}
+
+export interface TgReactionCount {
+  chat: { id: number; type: string };
+  message_id: number;
+  date: number;
+  reactions: {
+    type: { type: string; emoji?: string };
+    total_count: number;
+  }[];
+}

package/runtime/node_modules/@metro-labs/telegram-bot/src/verify.ts

@@ -0,0 +1,38 @@
+export class TelegramTokenError extends Error {}
+
+export interface TelegramBotIdentity {
+  botId: number;
+  username: string;
+}
+
+const API = 'https://api.telegram.org';
+const TIMEOUT_MS = 10_000;
+
+interface GetMe {
+  ok?: unknown;
+  result?: { id?: unknown; username?: unknown; is_bot?: unknown };
+}
+
+export async function verifyTelegramBotToken(
+  token: string,
+): Promise<TelegramBotIdentity> {
+  let res: Response;
+  try {
+    res = await fetch(`${API}/bot${token}/getMe`, {
+      signal: AbortSignal.timeout(TIMEOUT_MS),
+    });
+  } catch {
+    throw new TelegramTokenError(
+      'could not reach the Telegram API to check that bot token',
+    );
+  }
+  const body = (await res.json().catch(() => null)) as GetMe | null;
+  const me = body?.ok === true ? body.result : undefined;
+  if (!me) throw new TelegramTokenError('Telegram rejected that bot token');
+  if (typeof me.id !== 'number' || me.is_bot !== true)
+    throw new TelegramTokenError('that token does not belong to a Telegram bot');
+  return {
+    botId: me.id,
+    username: typeof me.username === 'string' ? me.username : '',
+  };
+}

package/runtime/node_modules/@metro-labs/telegram-bot/src/wire.ts

@@ -0,0 +1,3 @@
+export { emit, respond, mintId } from '@metro-labs/mcp/stations/station-runtime';
+
+export const SELF_URI = process.env.METRO_SELF_URI ?? '';

package/runtime/node_modules/@metro-labs/webhook/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "@metro-labs/webhook",
+  "version": "0.1.0",
+  "private": true,
+  "type": "module",
+  "exports": {
+    ".": "./src/station.ts"
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "tsc --noEmit && bun test test/"
+  },
+  "dependencies": {
+    "@metro-labs/mcp": "workspace:*"
+  },
+  "devDependencies": {
+    "@stage-labs/config": "0.1.0-beta.2",
+    "eslint": "^10.3.0",
+    "knip": "^6.15.0",
+    "madge": "^8.0.0",
+    "typescript": "^5"
+  }
+}

package/runtime/node_modules/@metro-labs/webhook/src/station.ts

@@ -0,0 +1,37 @@
+import { randomUUID } from 'node:crypto';
+import { Line } from '@metro-labs/mcp/lines';
+import { mintId, type MetroEvent } from '@metro-labs/mcp/events';
+import type { Endpoint } from '@metro-labs/mcp/endpoints';
+import type { Station, Verb } from '@metro-labs/mcp/stations/types';
+
+export const webhookStation: Station = {
+  name: 'webhook',
+  hasAccounts: true,
+  hasTrain: false,
+  messageVerbs: new Set<Verb>(),
+  attachmentMode: 'none',
+  tools: [],
+};
+
+export function webhookEntry(
+  endpoint: Endpoint,
+  headers: Record<string, string>,
+  body: unknown,
+  method: string,
+  url: string,
+): MetroEvent {
+  const line = Line.webhook(endpoint.id);
+  return {
+    id: mintId(),
+    ts: new Date().toISOString(),
+    station: 'webhook',
+    line,
+    lineName: endpoint.label,
+    from: line,
+    to: line,
+    messageId:
+      headers['x-github-delivery'] ?? headers['x-request-id'] ?? randomUUID(),
+    text: `${headers['x-github-event'] ?? headers['x-intercom-topic'] ?? 'event'} ${method} ${url}`,
+    payload: { headers, body },
+  };
+}

package/runtime/node_modules/@metro-labs/whatsapp/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "@metro-labs/whatsapp",
+  "version": "0.1.0",
+  "private": true,
+  "type": "module",
+  "exports": {
+    ".": "./src/station.ts",
+    "./train": "./src/index.ts",
+    "./login": "./src/login.ts",
+    "./tokens": "./src/token-store.ts"
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "tsc --noEmit && bun test test/"
+  },
+  "dependencies": {
+    "@metro-labs/mcp": "workspace:*",
+    "baileys": "7.0.0-rc14"
+  },
+  "devDependencies": {
+    "@stage-labs/config": "0.1.0-beta.2",
+    "eslint": "^10.3.0",
+    "knip": "^6.15.0",
+    "madge": "^8.0.0",
+    "typescript": "^5"
+  }
+}

package/runtime/node_modules/@metro-labs/whatsapp/src/accounts.ts

@@ -0,0 +1,66 @@
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import {
+  makeAccountStore,
+  resolveAccountId,
+  type Die,
+} from '@metro-labs/mcp/stations/account-store';
+import type { WhatsAppAccount } from './types.js';
+
+const ACCOUNTS_FILE =
+  process.env.WHATSAPP_ACCOUNTS_FILE ??
+  join(homedir(), '.metro', 'whatsapp-accounts.json');
+
+const isJid = (s: string): boolean => s.includes('@');
+
+function validateAccount(
+  a: WhatsAppAccount,
+  seen: Set<string>,
+  die: Die,
+): void {
+  if (!a.id) die('account missing id');
+  if (!a.phone || typeof a.phone !== 'string')
+    die(`account '${a.id}' missing phone`);
+  if (seen.has(a.id)) die(`duplicate account id '${a.id}'`);
+  seen.add(a.id);
+}
+
+export const { loadAccounts } = makeAccountStore<WhatsAppAccount>({
+  prefix: 'whatsapp',
+  file: ACCOUNTS_FILE,
+  validate(raw, die) {
+    const seen = new Set<string>();
+    for (const a of raw) validateAccount(a, seen, die);
+  },
+});
+
+export const accounts = new Map<string, WhatsAppAccount>();
+
+export function accountFor(args: { account?: string; line?: string }): string {
+  return resolveAccountId(accounts, args, (line) => targetOf(line)?.accountId);
+}
+
+export function lineOf(accountId: string, jid: string): string {
+  return `metro://whatsapp/${accountId}/${jid}`;
+}
+
+interface Target {
+  accountId: string;
+  jid: string;
+}
+
+function splitScoped(path: string[]): { accountId: string; jid?: string } {
+  const first = path[0];
+  if (path.length >= 2 && first !== undefined && !isJid(first))
+    return { accountId: first, jid: path[1] };
+  return { accountId: 'default', jid: first };
+}
+
+export function targetOf(line: string): Target | undefined {
+  const prefix = 'metro://whatsapp/';
+  if (!line.startsWith(prefix)) return undefined;
+  const path = line.slice(prefix.length).split('/').filter(Boolean);
+  const { accountId, jid } = splitScoped(path);
+  if (jid === undefined || !isJid(jid)) return undefined;
+  return { accountId, jid };
+}

package/runtime/node_modules/@metro-labs/whatsapp/src/ack.ts

@@ -0,0 +1,101 @@
+import { TrainError } from '@metro-labs/mcp/train-error';
+import type { BinaryNode, WASocket } from 'baileys';
+
+const DEFAULT_ACK_WAIT_MS = 5000;
+
+export function ackWaitMs(raw: string | undefined): number {
+  if (raw === undefined || raw.trim() === '') return DEFAULT_ACK_WAIT_MS;
+  const ms = Number(raw);
+  return Number.isFinite(ms) && ms >= 0 ? ms : DEFAULT_ACK_WAIT_MS;
+}
+
+export const ACK_WAIT_MS = ackWaitMs(process.env.METRO_WHATSAPP_ACK_WAIT_MS);
+
+const RECENT_MAX = 256;
+
+const RESTRICTION = '463';
+
+export interface SendAck {
+  messageId: string;
+  jid: string;
+  error?: string;
+}
+
+export function ackOf(node: BinaryNode): SendAck | undefined {
+  const { id, from, error } = node.attrs;
+  if (!id || !from) return undefined;
+  return { messageId: id, jid: from, ...(error ? { error } : {}) };
+}
+
+export function rejection(ack: SendAck): TrainError | undefined {
+  if (!ack.error) return undefined;
+  if (ack.error === RESTRICTION)
+    return new TrainError(
+      'whatsapp_account_restricted',
+      `WhatsApp refused the message to ${ack.jid} with ack error 463 (MessageAccountRestriction): the send carried no trusted-contact token, so WhatsApp counted it as reaching out to a stranger and this account is under a reach-out timelock. Nothing was delivered, and it must not be retried — every retry counts as another reach-out and deepens the lock.`,
+      { retryable: false },
+    );
+  return new TrainError(
+    'whatsapp_send_refused',
+    `WhatsApp refused the message to ${ack.jid} with ack error ${ack.error}, so it was not delivered`,
+    { retryable: false },
+  );
+}
+
+export interface AckWatch {
+  record(node: BinaryNode): SendAck | undefined;
+  wait(messageId: string, timeoutMs: number): Promise<SendAck | undefined>;
+}
+
+export function makeAckWatch(max: number = RECENT_MAX): AckWatch {
+  const recent = new Map<string, SendAck>();
+  const waiting = new Map<string, (ack: SendAck) => void>();
+  return {
+    record(node) {
+      const ack = ackOf(node);
+      if (!ack) return undefined;
+      const waiter = waiting.get(ack.messageId);
+      if (waiter) {
+        waiting.delete(ack.messageId);
+        waiter(ack);
+        return ack;
+      }
+      recent.delete(ack.messageId);
+      recent.set(ack.messageId, ack);
+      while (recent.size > max) {
+        const oldest = recent.keys().next().value;
+        if (oldest === undefined) break;
+        recent.delete(oldest);
+      }
+      return ack;
+    },
+    wait(messageId, timeoutMs) {
+      const seen = recent.get(messageId);
+      if (seen) {
+        recent.delete(messageId);
+        return Promise.resolve(seen);
+      }
+      return new Promise<SendAck | undefined>((resolve) => {
+        const timer = setTimeout(() => {
+          waiting.delete(messageId);
+          resolve(undefined);
+        }, timeoutMs);
+        waiting.set(messageId, (ack) => {
+          clearTimeout(timer);
+          resolve(ack);
+        });
+      });
+    },
+  };
+}
+
+export function bindAcks(
+  sock: WASocket,
+  watch: AckWatch,
+  onError: (ack: SendAck) => void,
+): void {
+  sock.ws.on('CB:ack,class:message', (node: BinaryNode) => {
+    const ack = watch.record(node);
+    if (ack?.error) onError(ack);
+  });
+}