@stage-labs/metro 0.1.0-beta.2 → 0.1.0-beta.21

package/runtime/node_modules/@metro-labs/discord/src/group.ts

@@ -0,0 +1,115 @@
+import { errMsg } from '@metro-labs/mcp/log';
+import type {
+  GroupResult,
+  MemberOutcome,
+} from '@metro-labs/mcp/stations/types';
+import { lineOf } from './accounts.js';
+
+export type RestFn = <T = unknown>(
+  accountId: string,
+  method: string,
+  path: string,
+  body?: unknown,
+) => Promise<T>;
+
+const PUBLIC_THREAD = 11;
+const PRIVATE_THREAD = 12;
+const AUTO_ARCHIVE = 1440;
+
+function membersOf(args: Record<string, unknown>): string[] {
+  const raw = args.members;
+  if (!Array.isArray(raw)) return [];
+  return raw.filter((v) => typeof v === 'string' && v.length > 0) as string[];
+}
+
+async function addThreadMembers(
+  rest: RestFn,
+  accountId: string,
+  threadId: string,
+  members: string[],
+): Promise<MemberOutcome[]> {
+  const outcomes: MemberOutcome[] = [];
+  for (const userId of members) {
+    try {
+      await rest(
+        accountId,
+        'PUT',
+        `/channels/${threadId}/thread-members/${userId}`,
+      );
+      outcomes.push({ id: userId, status: 'added' });
+    } catch (e) {
+      outcomes.push({ id: userId, status: 'failed', reason: errMsg(e) });
+    }
+  }
+  return outcomes;
+}
+
+export async function groupCreate(
+  rest: RestFn,
+  accountId: string,
+  parentChannelId: string,
+  args: Record<string, unknown>,
+): Promise<GroupResult> {
+  const name = typeof args.name === 'string' ? args.name : '';
+  if (!name)
+    return { capability: { supported: false, reason: 'create_group requires a `name`' } };
+  const type = args.private === true ? PRIVATE_THREAD : PUBLIC_THREAD;
+  const thread = await rest<{ id: string }>(
+    accountId,
+    'POST',
+    `/channels/${parentChannelId}/threads`,
+    { name, type, auto_archive_duration: AUTO_ARCHIVE },
+  );
+  const members = await addThreadMembers(
+    rest,
+    accountId,
+    thread.id,
+    membersOf(args),
+  );
+  return {
+    capability: { supported: true },
+    line: lineOf(accountId, thread.id),
+    id: thread.id,
+    name,
+    members,
+  };
+}
+
+export async function groupAddMembers(
+  rest: RestFn,
+  accountId: string,
+  threadId: string,
+  line: string,
+  args: Record<string, unknown>,
+): Promise<GroupResult> {
+  const members = await addThreadMembers(
+    rest,
+    accountId,
+    threadId,
+    membersOf(args),
+  );
+  return { capability: { supported: true }, line, id: threadId, members };
+}
+
+export async function groupRemoveMembers(
+  rest: RestFn,
+  accountId: string,
+  threadId: string,
+  line: string,
+  args: Record<string, unknown>,
+): Promise<GroupResult> {
+  const outcomes: MemberOutcome[] = [];
+  for (const userId of membersOf(args)) {
+    try {
+      await rest(
+        accountId,
+        'DELETE',
+        `/channels/${threadId}/thread-members/${userId}`,
+      );
+      outcomes.push({ id: userId, status: 'removed' });
+    } catch (e) {
+      outcomes.push({ id: userId, status: 'failed', reason: errMsg(e) });
+    }
+  }
+  return { capability: { supported: true }, line, id: threadId, members: outcomes };
+}

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

@@ -0,0 +1,127 @@
+import { errMsg } from '@metro-labs/mcp/log';
+import {
+  Client,
+  Events,
+  GatewayIntentBits,
+  Partials,
+  type Message,
+  type MessageReaction,
+  type User,
+} from 'discord.js';
+import {
+  accounts,
+  loadAccounts,
+  lineOf,
+  type AccountConfig,
+} from './accounts.js';
+import { emitInbound, messageEnvelope, reactionEnvelope } from './format.js';
+import { mintId } from './wire.js';
+import { readCalls } from '@metro-labs/mcp/trains/protocol';
+import { handleCall } from './actions.js';
+
+readCalls('discord', handleCall);
+
+function makeClient(): Client {
+  return new Client({
+    intents: [
+      GatewayIntentBits.Guilds,
+      GatewayIntentBits.GuildMessages,
+      GatewayIntentBits.MessageContent,
+      GatewayIntentBits.GuildMessageReactions,
+      GatewayIntentBits.DirectMessages,
+      GatewayIntentBits.DirectMessageReactions,
+      GatewayIntentBits.GuildVoiceStates,
+    ],
+    partials: [
+      Partials.Channel,
+      Partials.Message,
+      Partials.Reaction,
+      Partials.User,
+    ],
+  });
+}
+
+function onEdit(accountId: string, m: Message): void {
+  if (m.author.bot) return;
+  emitInbound(accountId, {
+    kind: 'edit',
+    id: mintId(),
+    ts: new Date(m.editedTimestamp ?? Date.now()).toISOString(),
+    station: 'discord',
+    line: lineOf(accountId, m.channelId),
+    from: `metro://discord/${accountId}/user/${m.author.id}`,
+    from_name: m.author.username,
+    from_display_name: m.author.globalName ?? undefined,
+    message_id: m.id,
+    text: m.content,
+    is_private: m.guildId == null,
+    event: { type: 'edit', targetId: m.id },
+    payload: m.toJSON(),
+  });
+}
+
+async function bootAccount(cfg: AccountConfig): Promise<void> {
+  const client = makeClient();
+  const accountId = cfg.id;
+
+  client.on(Events.MessageCreate, (m) => {
+    const env = messageEnvelope(accountId, m);
+    if (env) emitInbound(accountId, env);
+  });
+
+  client.on(Events.MessageReactionAdd, (r, u) => {
+    void (async () => {
+      try {
+        if (r.partial) await r.fetch();
+        if (u.partial) await u.fetch();
+      } catch {
+      }
+      const env = reactionEnvelope(accountId, r as MessageReaction, u as User);
+      if (env) emitInbound(accountId, env);
+    })().catch((err: unknown) => {
+      process.stderr.write(
+        `discord[${accountId}] reaction handler failed: ${errMsg(err)}\n`,
+      );
+    });
+  });
+
+  client.on(Events.MessageUpdate, (_old, _new) => {
+    void (async () => {
+      try {
+        onEdit(accountId, _new.partial ? await _new.fetch() : _new);
+      } catch (err) {
+        process.stderr.write(
+          `discord[${accountId}] message update fetch failed: ${errMsg(err)}\n`,
+        );
+      }
+    })().catch((err: unknown) => {
+      process.stderr.write(
+        `discord[${accountId}] message update handler failed: ${errMsg(err)}\n`,
+      );
+    });
+  });
+
+  accounts.set(accountId, { cfg, client });
+  await client.login(cfg.token);
+  process.stderr.write(
+    `discord[${accountId}] ready — ${client.user?.tag ?? '?'} (owner=${cfg.owner ?? '(broadcast)'})\n`,
+  );
+}
+
+const cfgs = loadAccounts();
+for (const cfg of cfgs) {
+  try {
+    await bootAccount(cfg);
+  } catch (err) {
+    process.stderr.write(
+      `discord[${cfg.id}] boot FAILED: ${errMsg(err)}\n`,
+    );
+  }
+}
+if (accounts.size === 0) {
+  process.stderr.write('discord: no accounts booted, exiting\n');
+  process.exit(2);
+}
+process.stderr.write(
+  `discord train ready — ${accounts.size} account(s): ${[...accounts.keys()].join(', ')}\n`,
+);

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

@@ -0,0 +1,215 @@
+import { errMsg } from '@metro-labs/mcp/log';
+import type { MemberList, MetroMember } from '@metro-labs/mcp/stations/types';
+import { rest } from './accounts.js';
+
+export interface DiscordRole {
+  id: string;
+  name: string;
+  permissions: string;
+}
+
+export interface DiscordUser {
+  id: string;
+  username?: string;
+  global_name?: string | null;
+  bot?: boolean;
+}
+
+export interface DiscordGuildMember {
+  user?: DiscordUser;
+  nick?: string | null;
+  roles?: string[];
+}
+
+const ADMINISTRATOR = 1n << 3n;
+const PAGE = 1000;
+
+export interface RoleInfo {
+  name: string;
+  admin: boolean;
+}
+
+function roleIsAdmin(permissions: string): boolean {
+  try {
+    return (BigInt(permissions) & ADMINISTRATOR) === ADMINISTRATOR;
+  } catch {
+    return false;
+  }
+}
+
+export function indexRoles(roles: DiscordRole[]): Map<string, RoleInfo> {
+  const out = new Map<string, RoleInfo>();
+  for (const r of roles)
+    out.set(r.id, { name: r.name, admin: roleIsAdmin(r.permissions) });
+  return out;
+}
+
+function isAdmin(
+  userId: string,
+  roleIds: string[],
+  roles: Map<string, RoleInfo>,
+  ownerId?: string,
+): boolean {
+  if (ownerId !== undefined && userId === ownerId) return true;
+  return roleIds.some((id) => roles.get(id)?.admin === true);
+}
+
+export function mapGuildMember(
+  m: DiscordGuildMember,
+  roles: Map<string, RoleInfo>,
+  ownerId?: string,
+): MetroMember | null {
+  const user = m.user;
+  if (!user?.id) return null;
+  const member: MetroMember = { id: user.id };
+  if (user.username) member.name = user.username;
+  const display = m.nick ?? user.global_name ?? undefined;
+  if (display) member.display_name = display;
+  const roleIds = m.roles ?? [];
+  const roleNames = roleIds
+    .map((id) => roles.get(id)?.name)
+    .filter((n): n is string => Boolean(n));
+  if (roleNames.length) member.roles = roleNames;
+  member.is_admin = isAdmin(user.id, roleIds, roles, ownerId);
+  member.is_bot = Boolean(user.bot);
+  return member;
+}
+
+export function mapGuildMembers(
+  members: DiscordGuildMember[],
+  roles: Map<string, RoleInfo>,
+  ownerId?: string,
+): MetroMember[] {
+  return members
+    .map((m) => mapGuildMember(m, roles, ownerId))
+    .filter((x): x is MetroMember => x !== null);
+}
+
+export function mapRecipients(users: DiscordUser[]): MetroMember[] {
+  return users
+    .filter((u) => Boolean(u?.id))
+    .map((u) => {
+      const m: MetroMember = { id: u.id };
+      if (u.username) m.name = u.username;
+      if (u.global_name) m.display_name = u.global_name;
+      m.is_bot = Boolean(u.bot);
+      return m;
+    });
+}
+
+async function fetchRoles(
+  accountId: string,
+  guildId: string,
+): Promise<Map<string, RoleInfo>> {
+  try {
+    const roles = await rest<DiscordRole[]>(
+      accountId,
+      'GET',
+      `/guilds/${guildId}/roles`,
+    );
+    return indexRoles(roles);
+  } catch {
+    return new Map();
+  }
+}
+
+async function fetchOwnerId(
+  accountId: string,
+  guildId: string,
+): Promise<string | undefined> {
+  try {
+    const g = await rest<{ owner_id?: string }>(
+      accountId,
+      'GET',
+      `/guilds/${guildId}`,
+    );
+    return g.owner_id;
+  } catch {
+    return undefined;
+  }
+}
+
+interface Collected {
+  raw: DiscordGuildMember[];
+  exhausted: boolean;
+}
+
+async function paginate(
+  accountId: string,
+  guildId: string,
+  cap: number,
+): Promise<Collected> {
+  const raw: DiscordGuildMember[] = [];
+  let after: string | undefined;
+  for (;;) {
+    const want = Math.min(PAGE, cap - raw.length);
+    if (want <= 0) return { raw, exhausted: false };
+    const qs = new URLSearchParams({ limit: String(want) });
+    if (after) qs.set('after', after);
+    const page = await rest<DiscordGuildMember[]>(
+      accountId,
+      'GET',
+      `/guilds/${guildId}/members?${qs}`,
+    );
+    raw.push(...page);
+    if (page.length < want) return { raw, exhausted: true };
+    after = page[page.length - 1]?.user?.id;
+    if (!after) return { raw, exhausted: true };
+  }
+}
+
+const INTENT_HINT =
+  'Discord bot lacks the privileged GUILD_MEMBERS intent (enable it in the ' +
+  'Developer Portal) or channel access; returning what is reachable.';
+
+export async function listGuildMembers(
+  accountId: string,
+  guildId: string,
+  limit?: number,
+): Promise<MemberList> {
+  const cap = Math.min(Math.max(1, limit ?? PAGE), 10_000);
+  const roles = await fetchRoles(accountId, guildId);
+  const ownerId = await fetchOwnerId(accountId, guildId);
+  try {
+    const { raw, exhausted } = await paginate(accountId, guildId, cap);
+    const members = mapGuildMembers(raw, roles, ownerId);
+    return {
+      members,
+      capability: exhausted
+        ? { supported: true, complete: true, total: members.length }
+        : {
+            supported: true,
+            complete: false,
+            reason: `truncated at limit ${cap}`,
+          },
+    };
+  } catch (e) {
+    const msg = errMsg(e);
+    if (/\b40[13]\b/.test(msg))
+      return {
+        members: [],
+        capability: { supported: false, complete: false, reason: INTENT_HINT },
+      };
+    throw e;
+  }
+}
+
+export async function discordMembers(
+  accountId: string,
+  channelId: string,
+  limit?: number,
+): Promise<MemberList> {
+  const chan = await rest<{ guild_id?: string; recipients?: DiscordUser[] }>(
+    accountId,
+    'GET',
+    `/channels/${channelId}`,
+  );
+  if (!chan.guild_id) {
+    const members = mapRecipients(chan.recipients ?? []);
+    return {
+      members,
+      capability: { supported: true, complete: true, total: members.length },
+    };
+  }
+  return listGuildMembers(accountId, chan.guild_id, limit);
+}

package/runtime/node_modules/@metro-labs/discord/src/send-files.ts

@@ -0,0 +1,43 @@
+import { appendFile } from '@metro-labs/mcp/stations/attachments';
+
+export interface OutgoingFile {
+  path: string;
+  name: string;
+  kind: string;
+}
+
+const basenameOf = (path: string, index: number): string =>
+  path.split('/').pop() ?? `file-${index}`;
+
+export function outgoingFiles(
+  paths: string[],
+  names: string[] | undefined,
+  kinds: string[] | undefined,
+): OutgoingFile[] {
+  const out: OutgoingFile[] = [];
+  for (let i = 0; i < paths.length; i++) {
+    const path = paths[i];
+    if (path === undefined || path === '') continue;
+    const given = names?.[i];
+    out.push({
+      path,
+      name: given !== undefined && given !== '' ? given : basenameOf(path, i),
+      kind: kinds?.[i] ?? 'file',
+    });
+  }
+  return out;
+}
+
+export async function appendFiles(
+  form: FormData,
+  files: OutgoingFile[],
+): Promise<string[]> {
+  const delivered: string[] = [];
+  for (let i = 0; i < files.length; i++) {
+    const file = files[i];
+    if (!file) continue;
+    await appendFile(form, `files[${i}]`, file.path, file.name);
+    delivered.push(file.kind);
+  }
+  return delivered;
+}

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

@@ -0,0 +1,19 @@
+import type { GroupOp, Station, Verb } from '@metro-labs/mcp/stations/types';
+
+export const discordStation: Station = {
+  name: 'discord',
+  hasAccounts: true,
+  hasTrain: true,
+  messageVerbs: new Set<Verb>([
+    'send',
+    'reply',
+    'react',
+    'unreact',
+    'edit',
+    'delete',
+    'read',
+  ]),
+  groupOps: new Set<GroupOp>(['create_group', 'add_members', 'remove_members']),
+  attachmentMode: 'canonical',
+  tools: [],
+};

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

@@ -0,0 +1,82 @@
+import { API } from './api-base.js';
+
+export class DiscordTokenError extends Error {}
+
+export interface DiscordBotIdentity {
+  userId: string;
+  username: string;
+  messageContent: boolean;
+}
+
+const MESSAGE_CONTENT = 1 << 18;
+const MESSAGE_CONTENT_LIMITED = 1 << 19;
+const TIMEOUT_MS = 10_000;
+
+function botFetch(token: string, path: string): Promise<Response> {
+  return fetch(`${API}${path}`, {
+    headers: {
+      authorization: `Bot ${token}`,
+      'user-agent': 'metro-daemon (https://github.com/bonustrack/metro)',
+    },
+    signal: AbortSignal.timeout(TIMEOUT_MS),
+  });
+}
+
+async function readJson(res: Response): Promise<Record<string, unknown>> {
+  try {
+    const body: unknown = await res.json();
+    return typeof body === 'object' && body !== null
+      ? (body as Record<string, unknown>)
+      : {};
+  } catch {
+    throw new DiscordTokenError('Discord returned a response Metro could not read');
+  }
+}
+
+async function selfUser(token: string): Promise<Record<string, unknown>> {
+  let res: Response;
+  try {
+    res = await botFetch(token, '/users/@me');
+  } catch {
+    throw new DiscordTokenError(
+      'could not reach the Discord API to check that bot token',
+    );
+  }
+  if (res.status === 401 || res.status === 403)
+    throw new DiscordTokenError('Discord rejected that bot token');
+  if (!res.ok)
+    throw new DiscordTokenError(
+      `Discord answered ${res.status} while checking that bot token`,
+    );
+  return readJson(res);
+}
+
+async function messageContentEnabled(token: string): Promise<boolean> {
+  try {
+    const res = await botFetch(token, '/applications/@me');
+    if (!res.ok) return true;
+    const flags = (await readJson(res)).flags;
+    if (typeof flags !== 'number') return true;
+    return (flags & (MESSAGE_CONTENT | MESSAGE_CONTENT_LIMITED)) !== 0;
+  } catch {
+    return true;
+  }
+}
+
+export async function verifyDiscordBotToken(
+  token: string,
+): Promise<DiscordBotIdentity> {
+  const me = await selfUser(token);
+  const userId = typeof me.id === 'string' ? me.id : '';
+  if (userId === '')
+    throw new DiscordTokenError('Discord returned no bot identity for that token');
+  if (me.bot !== true)
+    throw new DiscordTokenError(
+      'that token belongs to a Discord user, not a bot application',
+    );
+  return {
+    userId,
+    username: typeof me.username === 'string' ? me.username : '',
+    messageContent: await messageContentEnabled(token),
+  };
+}