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

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

@@ -0,0 +1,244 @@
+import { ReactionAction, ReactionSchema } from '@xmtp/node-sdk';
+import type { Reply } from '@xmtp/node-bindings';
+import {
+  AttachmentCodec,
+  type Attachment,
+} from '@xmtp/content-type-remote-attachment';
+import { convOf } from './accounts.js';
+import { resolveMsgId, respond } from './wire.js';
+import { emitOutbound } from './emit.js';
+import { PollCodec, buildPollContent } from './codecs.js';
+import { convHandlers } from './actions-conv.js';
+import { normalizeXmtp } from '@metro-labs/core/stations/messaging-normalize';
+import { TrainError } from '@metro-labs/core/train-error';
+import { claimNameAction, nameAction, setProfile } from './profile.js';
+import { profileAction } from './sender.js';
+import { makeStation, type CallMsg } from '@metro-labs/core/stations/station-runtime';
+
+type Args = Record<string, unknown>;
+
+const noConv = (line: string): TrainError =>
+  new TrainError('NOT_FOUND', `conversation not found for ${line}`);
+const badArgs = (message: string): TrainError =>
+  new TrainError('INVALID_ARGS', message);
+
+async function send(id: string, args: Args): Promise<void> {
+  const { line, text } = args as { line: string; text: string };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const messageId = await conv.sendText(text);
+  emitOutbound(acct.cfg.id, line, messageId, text);
+  respond(id, { result: { messageId } });
+}
+
+async function ask(id: string, args: Args): Promise<void> {
+  const { line, pollId } = args as { line: string; pollId?: string };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const fallbackId = `poll_${Date.now()}_${Math.random().toString(36).slice(2, 10)}`;
+  const mintedId =
+    pollId ??
+    (typeof crypto?.randomUUID === 'function'
+      ? crypto.randomUUID()
+      : fallbackId);
+  const { poll, title } = buildPollContent(args, mintedId);
+  const sentId = await conv.send(new PollCodec().encode(poll));
+  emitOutbound(acct.cfg.id, line, sentId, `📊 Poll: ${title}`);
+  respond(id, { result: { messageId: sentId, pollId: mintedId } });
+}
+
+async function react(id: string, args: Args): Promise<void> {
+  const {
+    line,
+    messageId,
+    emoji,
+    action: reactAction,
+    schema: reactSchema,
+  } = args as {
+    line: string;
+    messageId: string;
+    emoji: string;
+    action?: 'added' | 'removed';
+    schema?: string;
+    referenceInboxId?: string;
+  };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const xmtpMsgId = resolveMsgId(messageId);
+  let refInbox = (args as { referenceInboxId?: string }).referenceInboxId;
+  if (!refInbox) {
+    const recent = await conv.messages({ limit: 200, direction: 1 });
+    refInbox = recent.find((m) => m.id === xmtpMsgId)?.senderInboxId;
+    if (!refInbox)
+      throw new TrainError(
+        'NOT_FOUND',
+        `could not resolve referenceInboxId for ${xmtpMsgId}`,
+      );
+  }
+  const schemaEnum =
+    reactSchema === 'custom'
+      ? ReactionSchema.Custom
+      : reactSchema === 'shortcode'
+        ? ReactionSchema.Shortcode
+        : ReactionSchema.Unicode;
+  const sentId = await conv.sendReaction({
+    reference: xmtpMsgId,
+    referenceInboxId: refInbox,
+    action:
+      reactAction === 'removed' ? ReactionAction.Removed : ReactionAction.Added,
+    content: emoji,
+    schema: schemaEnum,
+  });
+  emitOutbound(
+    acct.cfg.id,
+    line,
+    sentId,
+    `[react ${emoji}${reactAction === 'removed' ? ' (removed)' : ''}]`,
+    { type: 'react', emoji, targetId: xmtpMsgId },
+  );
+  respond(id, { result: { messageId: sentId } });
+}
+
+async function reply(id: string, args: Args): Promise<void> {
+  const { line, replyTo, text } = args as {
+    line: string;
+    replyTo: string;
+    text: string;
+  };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const { encodeText } = await import('@xmtp/node-bindings');
+  const xmtpReplyTo = resolveMsgId(replyTo);
+  const sentId = await conv.sendReply({
+    reference: xmtpReplyTo,
+    content: encodeText(text),
+    contentType: {
+      authorityId: 'xmtp.org',
+      typeId: 'text',
+      versionMajor: 1,
+      versionMinor: 0,
+    },
+  } as unknown as Reply);
+  emitOutbound(acct.cfg.id, line, sentId, text, {
+    type: 'reply',
+    replyTo: xmtpReplyTo,
+  });
+  respond(id, { result: { messageId: sentId } });
+}
+
+async function sendAttachment(id: string, args: Args): Promise<void> {
+  const { line, name, mime, dataB64 } = args as {
+    line: string;
+    name: string;
+    mime: string;
+    dataB64: string;
+  };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const sentId = await conv.sendAttachment({
+    filename: name,
+    mimeType: mime,
+    content: new Uint8Array(Buffer.from(dataB64, 'base64')),
+  });
+  emitOutbound(acct.cfg.id, line, sentId, `[${mime.split('/')[0]}: ${name}]`);
+  respond(id, { result: { messageId: sentId } });
+}
+
+const IMG_MIME_BY_EXT: Record<string, string> = {
+  jpg: 'image/jpeg',
+  jpeg: 'image/jpeg',
+  gif: 'image/gif',
+  webp: 'image/webp',
+};
+
+async function loadImageBytes(
+  path: string | undefined,
+  dataB64: string | undefined,
+): Promise<Uint8Array> {
+  if (path) {
+    const { readFileSync } = await import('node:fs');
+    return new Uint8Array(readFileSync(path));
+  }
+  if (dataB64) return new Uint8Array(Buffer.from(dataB64, 'base64'));
+  throw badArgs('sendImage requires path or dataB64');
+}
+
+function imageMime(
+  mimeType: string | undefined,
+  filename: string | undefined,
+  path: string | undefined,
+): string {
+  if (mimeType) return mimeType;
+  const ext = (filename ?? path ?? '').toLowerCase().split('.').pop() ?? '';
+  return IMG_MIME_BY_EXT[ext] ?? 'image/png';
+}
+
+function imageFilename(
+  filename: string | undefined,
+  path: string | undefined,
+): string {
+  if (filename) return filename;
+  const baseName = path ? path.split('/').pop() : undefined;
+  return baseName != null && baseName !== '' ? baseName : 'image.png';
+}
+
+async function sendImage(id: string, args: Args): Promise<void> {
+  const { line, path, dataB64, filename, mimeType } = args as {
+    line: string;
+    path?: string;
+    dataB64?: string;
+    filename?: string;
+    mimeType?: string;
+  };
+  const { acct, conv } = await convOf(line);
+  if (!conv) throw noConv(line);
+  const bytes = await loadImageBytes(path, dataB64);
+  const mime = imageMime(mimeType, filename, path);
+  const fname = imageFilename(filename, path);
+  const attachment: Attachment = {
+    filename: fname,
+    mimeType: mime,
+    data: bytes,
+  };
+  const sentId = await conv.send(new AttachmentCodec().encode(attachment));
+  emitOutbound(acct.cfg.id, line, sentId, `[${mime.split('/')[0]}: ${fname}]`);
+  respond(id, { result: { messageId: sentId } });
+}
+
+async function accountsAction(id: string): Promise<void> {
+  const { accounts } = await import('./accounts.js');
+  respond(id, {
+    result: {
+      accounts: [...accounts.values()].map((a) => ({
+        id: a.cfg.id,
+        handle: a.address,
+        url: `https://stage.box/#/${a.address}`,
+        address: a.address,
+        inboxId: a.inboxId,
+        smart: a.smart !== null,
+      })),
+    },
+  });
+}
+
+const handlers: Record<string, (id: string, args: Args) => Promise<void>> = {
+  accounts: (id) => accountsAction(id),
+  set_profile: setProfile,
+  claim_name: claimNameAction,
+  name: nameAction,
+  profile: profileAction,
+  send,
+  ask,
+  react,
+  reply,
+  sendAttachment,
+  sendImage,
+  ...convHandlers,
+};
+
+export type { CallMsg };
+
+export const handleCall = makeStation({
+  handlers,
+  normalize: normalizeXmtp,
+});

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

@@ -0,0 +1,107 @@
+import {
+  AttachmentCodec,
+  RemoteAttachmentCodec,
+  ContentTypeAttachment,
+} from '@xmtp/content-type-remote-attachment';
+import {
+  saveBufferToCache,
+  assertAttachmentSize,
+} from '@metro-labs/core/stations/attachments';
+import type { SavedAttachment } from '@metro-labs/core/stations/attachments';
+import { errMsg } from '@metro-labs/core/log';
+
+export type { SavedAttachment };
+
+export const REMOTE_FETCH_ATTEMPTS = 3;
+const REMOTE_FETCH_BACKOFF_MS = 400;
+
+const sleep = (ms: number): Promise<void> =>
+  new Promise((resolve) => setTimeout(resolve, ms));
+
+const attachmentCodec = new AttachmentCodec();
+const loadRegistry = {
+  codecFor: (ct: { typeId?: string }) =>
+    ct.typeId === ContentTypeAttachment.typeId ? attachmentCodec : undefined,
+};
+
+export interface RemoteEntry {
+  url: string;
+  filename?: string;
+  contentDigest?: string;
+  nonce?: Uint8Array;
+  salt?: Uint8Array;
+  secret?: Uint8Array;
+  scheme?: string;
+  contentLength?: number;
+}
+
+export async function saveInlineAttachment(
+  a: { filename?: string; mimeType?: string; content: Uint8Array },
+  messageId: string,
+  index = 0,
+): Promise<SavedAttachment> {
+  return saveBufferToCache(a.content, messageId, index, {
+    mime: a.mimeType,
+    name: a.filename,
+  });
+}
+
+function toRemoteDescriptor(r: RemoteEntry): {
+  url: string;
+  contentDigest: string;
+  salt: Uint8Array;
+  nonce: Uint8Array;
+  secret: Uint8Array;
+  scheme: string;
+  contentLength: number;
+  filename: string;
+} {
+  return {
+    url: r.url,
+    contentDigest: r.contentDigest ?? '',
+    salt: r.salt ?? new Uint8Array(),
+    nonce: r.nonce ?? new Uint8Array(),
+    secret: r.secret ?? new Uint8Array(),
+    scheme: r.scheme ?? 'https://',
+    contentLength: r.contentLength ?? 0,
+    filename: r.filename ?? '',
+  };
+}
+
+interface DecodedAttachment {
+  filename?: string;
+  mimeType?: string;
+  data: Uint8Array;
+}
+
+async function loadRemote(r: RemoteEntry): Promise<DecodedAttachment> {
+  let last = '';
+  for (let attempt = 1; attempt <= REMOTE_FETCH_ATTEMPTS; attempt += 1) {
+    try {
+      return await RemoteAttachmentCodec.load<DecodedAttachment>(
+        toRemoteDescriptor(r),
+        loadRegistry,
+      );
+    } catch (err) {
+      last = errMsg(err);
+      if (attempt < REMOTE_FETCH_ATTEMPTS)
+        await sleep(REMOTE_FETCH_BACKOFF_MS * attempt);
+    }
+  }
+  throw new Error(
+    `xmtp remote attachment fetch failed after ${REMOTE_FETCH_ATTEMPTS} attempts: ${last}`,
+  );
+}
+
+export async function saveRemoteAttachment(
+  r: RemoteEntry,
+  messageId: string,
+  index = 0,
+): Promise<SavedAttachment> {
+  if (r.contentLength) assertAttachmentSize(r.contentLength);
+  const decoded = await loadRemote(r);
+  return saveBufferToCache(decoded.data, messageId, index, {
+    mime: decoded.mimeType,
+    name: decoded.filename ?? r.filename,
+  });
+}

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

@@ -0,0 +1,171 @@
+import type {
+  ContentTypeId,
+  ContentCodec,
+  EncodedContent,
+} from '@xmtp/content-type-primitives';
+
+const enc = (o: unknown): Uint8Array =>
+  new TextEncoder().encode(JSON.stringify(o));
+const dec = (e: EncodedContent): unknown =>
+  JSON.parse(new TextDecoder().decode(e.content));
+
+const makeJsonCodec = <T>(
+  contentType: ContentTypeId,
+  fallbackFn: (c: T) => string,
+) =>
+  class {
+    get contentType() {
+      return contentType;
+    }
+    encode(c: T): EncodedContent {
+      return {
+        type: contentType,
+        parameters: {},
+        fallback: fallbackFn(c),
+        content: enc(c),
+      };
+    }
+    decode(e: EncodedContent): T {
+      return dec(e) as T;
+    }
+    fallback(c: T) {
+      return fallbackFn(c);
+    }
+    shouldPush() {
+      return true;
+    }
+  };
+
+export const ContentTypePoll: ContentTypeId = {
+  authorityId: 'metro.box',
+  typeId: 'poll',
+  versionMajor: 1,
+  versionMinor: 0,
+};
+export interface PollOption {
+  label: string;
+  description?: string;
+}
+export interface PollQuestion {
+  question: string;
+  header?: string;
+  multiSelect?: boolean;
+  open?: boolean;
+  options?: (PollOption | string)[];
+}
+export interface PollContent {
+  questions?: PollQuestion[];
+  question?: string;
+  header?: string;
+  multiSelect?: boolean;
+  pollId?: string;
+  options?: (PollOption | string)[];
+  [k: string]: unknown;
+}
+const pollTitle = (c: PollContent): string =>
+  c.questions?.[0]?.question ?? c.question ?? 'Poll';
+export const PollCodec = makeJsonCodec<PollContent>(
+  ContentTypePoll,
+  (c) => `📊 Poll: ${pollTitle(c)}`,
+);
+
+const normOpts = (o: (string | PollOption)[]): PollOption[] =>
+  o.map((x) =>
+    typeof x === 'string'
+      ? { label: x }
+      : { label: x.label, description: x.description },
+  );
+
+export function buildPollContent(
+  args: Record<string, unknown>,
+  pollId: string,
+): { poll: PollContent; title: string } {
+  const { question, options, header, multiSelect, questions } = args as {
+    question?: string;
+    options?: (string | PollOption)[];
+    header?: string;
+    multiSelect?: boolean;
+    questions?: PollQuestion[];
+  };
+  if (Array.isArray(questions) && questions.length > 0) {
+    const norm: PollQuestion[] = questions.map((q, i) => {
+      if (!q || typeof q.question !== 'string' || !q.question)
+        throw new Error(`ask questions[${i}] requires a question`);
+      const open = q.open === true;
+      const opts = Array.isArray(q.options) ? q.options : [];
+      if (!open && opts.length === 0)
+        throw new Error(
+          `ask questions[${i}] requires a non-empty options array (or open:true for free-text)`,
+        );
+      return {
+        question: q.question,
+        options: normOpts(opts),
+        multiSelect: !!q.multiSelect,
+        ...(open ? { open: true } : {}),
+        ...(q.header ? { header: q.header } : {}),
+      };
+    });
+    const first = norm[0];
+    if (!first) throw new Error('ask requires at least one question');
+    return { poll: { questions: norm, pollId }, title: first.question };
+  }
+  if (!question || typeof question !== 'string')
+    throw new Error('ask requires a question (or a questions[] array)');
+  if (!Array.isArray(options) || options.length === 0)
+    throw new Error('ask requires a non-empty options array');
+  return {
+    poll: {
+      question,
+      options: normOpts(options),
+      multiSelect: !!multiSelect,
+      pollId,
+      ...(header ? { header } : {}),
+    },
+    title: question,
+  };
+}
+
+export const ContentTypeSignatureRequest: ContentTypeId = {
+  authorityId: 'metro.box',
+  typeId: 'signatureRequest',
+  versionMajor: 1,
+  versionMinor: 0,
+};
+export interface SignatureRequestContent {
+  id?: string;
+  kind?: 'eip712' | 'personal';
+  eip712?: unknown;
+  message?: string;
+  description?: string;
+  [k: string]: unknown;
+}
+export const SignatureRequestCodec = makeJsonCodec<SignatureRequestContent>(
+  ContentTypeSignatureRequest,
+  (c) =>
+    c.description
+      ? `[Signature request] ${c.description}`
+      : '[Signature request]',
+);
+
+export const ContentTypeSignatureReference: ContentTypeId = {
+  authorityId: 'metro.box',
+  typeId: 'signatureReference',
+  versionMajor: 1,
+  versionMinor: 0,
+};
+export interface SignatureReferenceContent {
+  requestId?: string;
+  signature: string;
+  signer?: string;
+  [k: string]: unknown;
+}
+export const SignatureReferenceCodec =
+  makeJsonCodec<SignatureReferenceContent>(ContentTypeSignatureReference, (c) =>
+    c.signature ? `[Signature] ${c.signature}` : '[Signature]',
+  );
+
+export const CODECS = (): ContentCodec[] => [
+  new PollCodec(),
+  new SignatureRequestCodec(),
+  new SignatureReferenceCodec(),
+];