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

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

@@ -0,0 +1,113 @@
+import {
+  Client,
+  type ClientOptions,
+  type Conversation,
+} from '@xmtp/node-sdk';
+import { homedir } from 'node:os';
+import { existsSync } from 'node:fs';
+import { join } from 'node:path';
+import { CODECS } from './codecs.js';
+import { expandHome, signerFor, XMTP_ENV } from './identity.js';
+import {
+  makeAccountStore,
+  resolveAccountId,
+} from '@metro-labs/mcp/stations/account-store';
+import { Line } from '@metro-labs/mcp/lines';
+
+const ACCOUNTS_FILE =
+  process.env.XMTP_ACCOUNTS_FILE ??
+  join(homedir(), '.metro', 'xmtp-accounts.json');
+
+export interface AccountConfig {
+  id: string;
+  privateKey?: string;
+  owner?: string;
+  dbPath?: string;
+}
+
+export const { loadAccounts } = makeAccountStore<AccountConfig>({
+  prefix: 'xmtp',
+  file: ACCOUNTS_FILE,
+  allowlistEnv: ['XMTP_ONLY_ACCOUNTS', 'XMTP_ACCOUNTS'],
+  validate(raw, die) {
+    const seen = new Set<string>();
+    for (const a of raw) {
+      if (!a.id) die('account missing id');
+      if (!a.privateKey) die(`account '${a.id}' needs a privateKey`);
+      if (seen.has(a.id)) die(`duplicate account id '${a.id}'`);
+      seen.add(a.id);
+    }
+  },
+});
+
+function resolvePrivateKey(cfg: AccountConfig): string {
+  const privateKey = cfg.privateKey?.trim();
+  if (!privateKey) throw new Error(`account '${cfg.id}' has no privateKey`);
+  return privateKey;
+}
+
+export interface Account {
+  cfg: AccountConfig;
+  client: Client<unknown>;
+  inboxId: string;
+  address: string;
+}
+export const accounts = new Map<string, Account>();
+
+export async function bootAccount(cfg: AccountConfig): Promise<void> {
+  const { signer, address } = signerFor(resolvePrivateKey(cfg));
+  const dbPath = expandHome(
+    cfg.dbPath ?? join(homedir(), '.metro', `xmtp-${XMTP_ENV}-${cfg.id}.db3`),
+  );
+  const fresh = !existsSync(dbPath);
+  const options: ClientOptions = { env: XMTP_ENV, codecs: CODECS(), dbPath };
+  const client: Client<unknown> = await Client.create(signer, options);
+  accounts.set(cfg.id, { cfg, client, inboxId: client.inboxId, address });
+  if (fresh)
+    process.stderr.write(
+      `xmtp[${cfg.id}] registered a NEW installation for this machine — ` +
+        'an inbox allows 10, and each machine you run metro on spends one\n',
+    );
+  process.stderr.write(
+    `xmtp[${cfg.id}] ready — inbox ${client.inboxId} (${address}, owner=${cfg.owner ?? '(broadcast)'})\n`,
+  );
+}
+
+export function lineOf(accountId: string, convId: string): string {
+  return `metro://xmtp/${accountId}/${convId}`;
+}
+
+export function parseLine(
+  line: string,
+): { accountId: string; convId: string } | null {
+  const p = Line.parseXmtp(line);
+  return p ? { accountId: p.accountId, convId: p.resource } : null;
+}
+
+export function accountForCall(args: {
+  account?: string;
+  line?: string;
+}): Account {
+  const id = resolveAccountId(
+    accounts,
+    args,
+    (line) => parseLine(line)?.accountId,
+  );
+  const acct = accounts.get(id);
+  if (!acct) throw new Error(`unknown account '${id}'`);
+  return acct;
+}
+
+export async function convOf(
+  line: string,
+): Promise<{ acct: Account; conv: Conversation | undefined }> {
+  const parsed = parseLine(line);
+  if (!parsed) throw new Error(`bad xmtp line: ${line}`);
+  const acct = accounts.get(parsed.accountId);
+  if (!acct)
+    throw new Error(`unknown account '${parsed.accountId}' in line ${line}`);
+  return {
+    acct,
+    conv: await acct.client.conversations.getConversationById(parsed.convId),
+  };
+}

package/runtime/node_modules/@metro-labs/xmtp/src/actions-close.ts

@@ -0,0 +1,110 @@
+import { errMsg } from '@metro-labs/mcp/log';
+import { accountForCall, convOf, lineOf, type Account } from './accounts.js';
+import { respond } from './wire.js';
+import type { GroupLike } from './labels.js';
+
+type Args = Record<string, unknown>;
+
+async function removeSelfFromGroup(
+  id: string,
+  group: GroupLike & { removeMembers: (ids: string[]) => Promise<void> },
+  acct: Account,
+  resolvedLine: string,
+  removed: string[],
+): Promise<boolean> {
+  try {
+    await group.removeMembers([acct.inboxId]);
+    return true;
+  } catch (e) {
+    respond(id, {
+      result: {
+        line: resolvedLine,
+        id: group.id,
+        account: acct.cfg.id,
+        removed,
+        leftSelf: false,
+        selfRemovalError: `self-removal not supported: ${errMsg(e)}`,
+      },
+    });
+    return false;
+  }
+}
+
+type ClosableGroup = GroupLike & {
+  removeMembers: (ids: string[]) => Promise<void>;
+};
+
+async function resolveCloseTarget(
+  id: string,
+  args: Args,
+  acct: Account,
+): Promise<{ group: ClosableGroup; resolvedLine: string } | null> {
+  const { line, groupId } = args as { line?: string; groupId?: string };
+  const resolvedLine =
+    line ?? (groupId ? lineOf(acct.cfg.id, groupId) : undefined);
+  if (!resolvedLine) {
+    respond(id, { error: 'closeGroup requires `line` or `groupId`' });
+    return null;
+  }
+  const { conv } = await convOf(resolvedLine);
+  if (!conv) {
+    respond(id, { error: `conversation not found for ${resolvedLine}` });
+    return null;
+  }
+  const group = conv as unknown as ClosableGroup;
+  if (typeof group.removeMembers !== 'function') {
+    respond(id, {
+      error: 'closeGroup target is not a group (no removeMembers)',
+    });
+    return null;
+  }
+  await group.sync?.().catch(() => undefined);
+  return { group, resolvedLine };
+}
+
+export async function closeGroup(id: string, args: Args): Promise<void> {
+  const { removeInboxIds, removeSelf } = args as {
+    removeInboxIds?: string[];
+    removeSelf?: boolean;
+  };
+  const acct = accountForCall(args);
+  const target = await resolveCloseTarget(id, args, acct);
+  if (!target) return;
+  const { group, resolvedLine } = target;
+  const removeMembers = group.removeMembers.bind(group);
+
+  const others = (removeInboxIds ?? []).filter(
+    (iid) => typeof iid === 'string' && iid && iid !== acct.inboxId,
+  );
+  const removed: string[] = [];
+  let leftSelf = false;
+  try {
+    if (others.length) {
+      await removeMembers(others);
+      removed.push(...others);
+    }
+    if (removeSelf) {
+      leftSelf = await removeSelfFromGroup(
+        id,
+        group,
+        acct,
+        resolvedLine,
+        removed,
+      );
+      if (!leftSelf) return;
+    }
+  } catch (err) {
+    respond(id, { error: errMsg(err) });
+    return;
+  }
+
+  respond(id, {
+    result: {
+      line: resolvedLine,
+      id: group.id,
+      account: acct.cfg.id,
+      removed,
+      leftSelf,
+    },
+  });
+}

package/runtime/node_modules/@metro-labs/xmtp/src/actions-conv.ts

@@ -0,0 +1,299 @@
+import { IdentifierKind } from '@xmtp/node-sdk';
+import {
+  accounts,
+  accountForCall,
+  convOf,
+  lineOf,
+  parseLine,
+  type Account,
+} from './accounts.js';
+import { respond } from './wire.js';
+import { TrainError } from '@metro-labs/mcp/train-error';
+import { pushHandlers } from './actions-push.js';
+import { cleanLabels, labelsBlob } from './labels.js';
+import {
+  applyMemberOp,
+  buildGroupInfo,
+  buildMemberList,
+  createGroupWithMembers,
+  ethIdentifiers,
+  parseMemberArgs,
+  resolveMembers,
+  warmGroupName,
+} from './conv-helpers.js';
+import { closeGroup } from './actions-close.js';
+import {
+  groupAddMembers,
+  groupCreate,
+  groupRemoveMembers,
+} from './group.js';
+import {
+  updateChannelMeta,
+  applyChannelMeta,
+  resolveLine,
+  setGithub,
+  setPreview,
+} from './actions-meta.js';
+
+type Args = Record<string, unknown>;
+type Handler = (id: string, args: Args) => Promise<void>;
+type Conv = Awaited<ReturnType<typeof convOf>>['conv'] & object;
+
+async function newDm(id: string, args: Args): Promise<void> {
+  const { address } = args as { address: string };
+  const acct = accountForCall(args);
+  const dm = await acct.client.conversations.createDmWithIdentifier({
+    identifier: address,
+    identifierKind: IdentifierKind.Ethereum,
+  });
+  respond(id, {
+    result: {
+      line: lineOf(acct.cfg.id, dm.id),
+      id: dm.id,
+      account: acct.cfg.id,
+    },
+  });
+}
+
+async function newGroup(id: string, args: Args): Promise<void> {
+  const { addresses, name, permissions } = args as {
+    addresses: string[];
+    name?: string;
+    permissions?: 'admin-only' | 'default';
+  };
+  const acct = accountForCall(args);
+  const opts: { groupName?: string; permissions?: number } = {};
+  if (name) opts.groupName = name;
+  if (permissions === 'admin-only') opts.permissions = 1;
+  const group = await acct.client.conversations.createGroupWithIdentifiers(
+    ethIdentifiers(addresses),
+    opts,
+  );
+  warmGroupName(group.id, name);
+  respond(id, {
+    result: {
+      line: lineOf(acct.cfg.id, group.id),
+      id: group.id,
+      account: acct.cfg.id,
+    },
+  });
+}
+
+async function createRequestGroup(id: string, args: Args): Promise<void> {
+  const { name, description, labels } = args as {
+    name: string;
+    description?: string;
+    labels?: string[];
+  };
+  const acct = accountForCall(args);
+  if (!name || typeof name !== 'string')
+    throw new TrainError('INVALID_ARGS', 'createRequestGroup requires a `name`');
+  const { addrs, inboxes } = resolveMembers(args);
+  if (addrs.length === 0 && inboxes.length === 0)
+    throw new TrainError(
+      'INVALID_ARGS',
+      'createRequestGroup requires addresses[] or inboxIds[]',
+    );
+  const opts: {
+    groupName: string;
+    groupDescription?: string;
+    appData?: string;
+  } = { groupName: name };
+  if (description) opts.groupDescription = description;
+  if (Array.isArray(labels) && labels.length)
+    opts.appData = labelsBlob(undefined, labels);
+
+  const group = await createGroupWithMembers(acct, addrs, inboxes, opts);
+  warmGroupName(group.id, name);
+  respond(id, {
+    result: {
+      line: lineOf(acct.cfg.id, group.id),
+      id: group.id,
+      account: acct.cfg.id,
+      name,
+      description: description ?? '',
+      labels: cleanLabels(labels ?? []),
+    },
+  });
+}
+
+async function setLabels(id: string, args: Args): Promise<void> {
+  const { labels, setName, setDescription, setGithub } = args as {
+    line?: string;
+    groupId?: string;
+    labels: string[];
+    setName?: string;
+    setDescription?: string;
+    setGithub?: string;
+  };
+  const resolvedLine = resolveLine(args, 'setLabels');
+  if (!Array.isArray(labels))
+    throw new TrainError('INVALID_ARGS', 'setLabels requires a `labels` array');
+  const appData: Record<string, unknown> = { labels };
+  if (typeof setGithub === 'string') appData.github = setGithub;
+  const result = await applyChannelMeta(
+    { line: resolvedLine, name: setName, description: setDescription, appData },
+    'setLabels',
+  );
+  respond(id, {
+    result: {
+      line: result.line,
+      id: result.id,
+      account: result.account,
+      labels: result.labels,
+    },
+  });
+}
+
+async function query(id: string, args: Args): Promise<void> {
+  const { line, limit } = args as { line: string; limit?: number };
+  const { conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  const lim = Math.min(Math.max(1, limit ?? 20), 200);
+  await conv.sync().catch(() => undefined);
+  const all = await conv.messages();
+  const slice = all.slice(-lim);
+  const parsed = parseLine(line);
+  if (!parsed)
+    throw new TrainError('NOT_FOUND', `could not parse line ${line}`);
+  const acctId = parsed.accountId;
+  const messages = slice.map((m) => {
+    let text = '';
+    try {
+      const cc: unknown = m.content;
+      text =
+        typeof cc === 'string' ? cc : `[${m.contentType?.typeId ?? 'unknown'}]`;
+    } catch {
+      text = `[${m.contentType?.typeId ?? 'unknown'}]`;
+    }
+    return {
+      id: m.id,
+      ts: new Date(Number(m.sentAtNs / 1_000_000n)).toISOString(),
+      from: `metro://xmtp/${acctId}/user/${m.senderInboxId}`,
+      text,
+      contentType: m.contentType?.typeId ?? 'unknown',
+    };
+  });
+  respond(id, { result: { line, count: messages.length, messages } });
+}
+
+async function addMembers(id: string, args: Args): Promise<void> {
+  const { line, addrs, inboxes } = parseMemberArgs(args, 'addMembers');
+  const { acct, conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  await applyMemberOp(conv, addrs, inboxes, 'add');
+  const info = await buildGroupInfo(line, acct, conv);
+  respond(id, {
+    result: { ...info, added: { addresses: addrs, inboxIds: inboxes } },
+  });
+}
+
+async function removeMembers(id: string, args: Args): Promise<void> {
+  const { line, addrs, inboxes } = parseMemberArgs(args, 'removeMembers');
+  const { acct, conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  await applyMemberOp(conv, addrs, inboxes, 'remove');
+  const info = await buildGroupInfo(line, acct, conv);
+  respond(id, {
+    result: { ...info, removed: { addresses: addrs, inboxIds: inboxes } },
+  });
+}
+
+async function groupInfo(id: string, args: Args): Promise<void> {
+  const { line } = args as { line: string };
+  const { acct, conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  respond(id, { result: await buildGroupInfo(line, acct, conv) });
+}
+
+async function listMembers(id: string, args: Args): Promise<void> {
+  const { line } = args as { line: string };
+  const { acct, conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  respond(id, { result: await buildMemberList(acct, conv) });
+}
+
+async function summarizeConv(acct: Account, c: Conv): Promise<unknown> {
+  const recent = await c.messages({ limit: 1, direction: 1 }).catch(() => []);
+  const last = recent[0];
+  let preview = '';
+  if (last) {
+    const cc: unknown = last.content;
+    preview =
+      typeof cc === 'string'
+        ? cc.slice(0, 80)
+        : `[${last.contentType?.typeId ?? 'unknown'}]`;
+  }
+  const isDm =
+    typeof (c as unknown as { peerInboxId?: unknown }).peerInboxId ===
+    'function';
+  const gn = (c as unknown as { name?: string | (() => Promise<string>) }).name;
+  const resolvedName =
+    typeof gn === 'function' ? await gn().catch(() => '') : (gn ?? '');
+  return {
+    line: lineOf(acct.cfg.id, c.id),
+    id: c.id,
+    account: acct.cfg.id,
+    version: isDm ? 'dm' : 'group',
+    name: resolvedName ?? '',
+    lastTs: last
+      ? new Date(Number(last.sentAtNs / 1_000_000n)).toISOString()
+      : null,
+    lastPreview: preview,
+  };
+}
+
+async function listConvs(id: string, args: Args): Promise<void> {
+  const { limit, account } = args as { limit?: number; account?: string };
+  const lim = Math.min(Math.max(1, limit ?? 50), 200);
+  const targets = account
+    ? [accounts.get(account)].filter((a): a is Account => a !== undefined)
+    : [...accounts.values()];
+  const summaries: unknown[] = [];
+  for (const acct of targets) {
+    await acct.client.conversations.syncAll();
+    const all = await acct.client.conversations.list();
+    for (const c of all.slice(0, lim))
+      summaries.push(await summarizeConv(acct, c));
+  }
+  summaries.sort((a, b) =>
+    ((b as { lastTs?: string }).lastTs ?? '').localeCompare(
+      (a as { lastTs?: string }).lastTs ?? '',
+    ),
+  );
+  respond(id, {
+    result: { count: summaries.length, conversations: summaries.slice(0, lim) },
+  });
+}
+
+export const convHandlers: Record<string, Handler> = {
+  newDm,
+  newGroup,
+  createRequestGroup,
+  addMembers,
+  removeMembers,
+  setLabels,
+  setGithub,
+  setPreview,
+  updateChannelMeta,
+  closeGroup,
+  query,
+  groupInfo,
+  listMembers,
+  listConvs,
+  groupCreate: async (id, args) => {
+    respond(id, { result: await groupCreate(args) });
+  },
+  groupAddMembers: async (id, args) => {
+    respond(id, { result: await groupAddMembers(args) });
+  },
+  groupRemoveMembers: async (id, args) => {
+    respond(id, { result: await groupRemoveMembers(args) });
+  },
+  ...pushHandlers,
+};

package/runtime/node_modules/@metro-labs/xmtp/src/actions-meta.ts

@@ -0,0 +1,157 @@
+import { accountForCall, convOf, lineOf } from './accounts.js';
+import { respond } from './wire.js';
+import { warmGroupName } from './conv-helpers.js';
+import { mergeAppData, readAppData, type GroupLike } from './labels.js';
+import { TrainError } from '@metro-labs/mcp/train-error';
+
+type Args = Record<string, unknown>;
+
+export function resolveLine(args: Args, verb: string): string {
+  const line = (args as { line?: string }).line;
+  if (line) return line;
+  const groupId = (args as { groupId?: string }).groupId;
+  if (groupId) {
+    const acct = accountForCall(args);
+    return lineOf(acct.cfg.id, groupId);
+  }
+  throw new TrainError(
+    'INVALID_ARGS',
+    `${verb} requires \`line\` or \`groupId\``,
+  );
+}
+
+async function applyNameAndDescription(
+  group: GroupLike,
+  name: string | undefined,
+  description: string | undefined,
+): Promise<void> {
+  if (
+    typeof name === 'string' &&
+    name &&
+    typeof group.updateName === 'function'
+  ) {
+    await group.updateName(name);
+    warmGroupName(group.id, name);
+  }
+  if (
+    typeof description === 'string' &&
+    typeof group.updateDescription === 'function'
+  ) {
+    await group.updateDescription(description);
+  }
+}
+
+async function applyMergedAppData(
+  group: GroupLike & { updateAppData: (s: string) => Promise<void> },
+  appData: Record<string, unknown> | undefined,
+): Promise<Record<string, unknown>> {
+  if (appData && typeof appData === 'object' && !Array.isArray(appData)) {
+    const res = mergeAppData(group.appData, appData);
+    await group.updateAppData(res.blob);
+    return res.merged;
+  }
+  return readAppData(group.appData);
+}
+
+export async function applyChannelMeta(
+  args: {
+    line: string;
+    name?: string;
+    description?: string;
+    appData?: Record<string, unknown>;
+  },
+  verb: string,
+): Promise<Record<string, unknown>> {
+  const { line, name, description, appData } = args;
+  const { acct, conv } = await convOf(line);
+  if (!conv)
+    throw new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+  const group = conv as unknown as GroupLike;
+  if (typeof group.updateAppData !== 'function') {
+    throw new TrainError(
+      'INVALID_ARGS',
+      `${verb} target is not a group (no updateAppData)`,
+    );
+  }
+  await group.sync?.().catch(() => undefined);
+
+  await applyNameAndDescription(group, name, description);
+  const merged = await applyMergedAppData(
+    group as GroupLike & { updateAppData: (s: string) => Promise<void> },
+    appData,
+  );
+
+  return {
+    line,
+    id: group.id,
+    account: acct.cfg.id,
+    ...(typeof name === 'string' && name ? { name } : {}),
+    ...metaFields(merged),
+    appData: merged,
+  };
+}
+
+function metaFields(merged: Record<string, unknown>): {
+  labels: string[];
+  github: string | undefined;
+  preview: string | undefined;
+} {
+  return {
+    labels: Array.isArray(merged.labels) ? (merged.labels as string[]) : [],
+    github: typeof merged.github === 'string' ? merged.github : undefined,
+    preview: typeof merged.preview === 'string' ? merged.preview : undefined,
+  };
+}
+
+export async function updateChannelMeta(id: string, args: Args): Promise<void> {
+  const line = resolveLine(args, 'updateChannelMeta');
+  const { name, description, appData } = args as {
+    name?: string;
+    description?: string;
+    appData?: Record<string, unknown>;
+  };
+  const result = await applyChannelMeta(
+    { line, name, description, appData },
+    'updateChannelMeta',
+  );
+  respond(id, { result });
+}
+
+export async function setGithub(id: string, args: Args): Promise<void> {
+  const line = resolveLine(args, 'setGithub');
+  const { url } = args as { url: string };
+  if (typeof url !== 'string')
+    throw new Error('setGithub requires a `url` string');
+  const result = await applyChannelMeta(
+    { line, appData: { github: url } },
+    'setGithub',
+  );
+  respond(id, {
+    result: {
+      line: result.line,
+      id: result.id,
+      account: result.account,
+      github: result.github,
+    },
+  });
+}
+
+export async function setPreview(id: string, args: Args): Promise<void> {
+  const line = resolveLine(args, 'setPreview');
+  const a = args as { preview?: unknown; url?: unknown };
+  const value = typeof a.preview === 'string' ? a.preview : a.url;
+  if (typeof value !== 'string')
+    throw new Error('setPreview requires a `preview` string');
+  const result = await applyChannelMeta(
+    { line, appData: { preview: value } },
+    'setPreview',
+  );
+  respond(id, {
+    result: {
+      line: result.line,
+      id: result.id,
+      account: result.account,
+      preview: result.preview,
+    },
+  });
+}