@stage-labs/metro 0.1.0-beta.32 → 0.1.0-beta.321

package/runtime/node_modules/@metro-labs/mcp/src/db/schema.ts

@@ -1,118 +0,0 @@
-import {
-  boolean,
-  jsonb,
-  pgTable,
-  text,
-  unique,
-} from 'drizzle-orm/pg-core';
-
-export const STATIONS = [
-  'xmtp',
-  'telegram-bot',
-  'telegram',
-  'discord-bot',
-  'whatsapp',
-  'webhook',
-] as const;
-
-export type StationName = (typeof STATIONS)[number];
-
-export const users = pgTable('users', {
-  id: text('id').primaryKey(),
-  email: text('email').unique(),
-  address: text('address').unique(),
-});
-
-export const PROJECT_ROLES = ['admin', 'member'] as const;
-
-export type ProjectRole = (typeof PROJECT_ROLES)[number];
-
-export const projects = pgTable('projects', {
-  id: text('id').primaryKey(),
-  name: text('name').notNull(),
-  ownerId: text('owner_id')
-    .notNull()
-    .references(() => users.id, { onDelete: 'restrict' }),
-  isDefault: boolean('is_default').notNull().default(false),
-});
-
-export const projectMembers = pgTable(
-  'project_members',
-  {
-    id: text('id').primaryKey(),
-    projectId: text('project_id')
-      .notNull()
-      .references(() => projects.id, { onDelete: 'cascade' }),
-    userId: text('user_id')
-      .notNull()
-      .references(() => users.id, { onDelete: 'restrict' }),
-    role: text('role').$type<ProjectRole>().notNull(),
-  },
-  (t) => [unique('project_members_project_id_user_id_unique').on(t.projectId, t.userId)],
-);
-
-export const agents = pgTable('agents', {
-  id: text('id').primaryKey(),
-  name: text('name').notNull(),
-  projectId: text('project_id')
-    .notNull()
-    .references(() => projects.id, { onDelete: 'restrict' }),
-  key: text('key').unique(),
-  runtimeId: text('runtime_id'),
-});
-
-export const runtimes = pgTable('runtimes', {
-  id: text('id').primaryKey(),
-  agentId: text('agent_id')
-    .notNull()
-    .references(() => agents.id, { onDelete: 'cascade' }),
-  label: text('label').notNull(),
-  createdAt: text('created_at').notNull(),
-  lastSeenAt: text('last_seen_at'),
-  revokedAt: text('revoked_at'),
-});
-
-export const stations = pgTable('stations', {
-  id: text('id').primaryKey(),
-  agentId: text('agent_id').notNull(),
-  station: text('station').$type<StationName>().notNull(),
-  allowlist: text('allowlist').array().default(['*']),
-  config: jsonb('config').notNull(),
-});
-
-export const CONNECTOR_TRANSPORTS = ['http', 'sse'] as const;
-
-export type ConnectorTransport = (typeof CONNECTOR_TRANSPORTS)[number];
-
-export const connectors = pgTable(
-  'connectors',
-  {
-    id: text('id').primaryKey(),
-    projectId: text('project_id')
-      .notNull()
-      .references(() => projects.id, { onDelete: 'restrict' }),
-    name: text('name').notNull(),
-    url: text('url').notNull(),
-    transport: text('transport').$type<ConnectorTransport>().notNull(),
-    config: jsonb('config').notNull(),
-  },
-);
-
-export const agentConnectors = pgTable(
-  'agent_connectors',
-  {
-    id: text('id').primaryKey(),
-    agentId: text('agent_id')
-      .notNull()
-      .references(() => agents.id, { onDelete: 'cascade' }),
-    connectorId: text('connector_id')
-      .notNull()
-      .references(() => connectors.id, { onDelete: 'cascade' }),
-  },
-  (t) => [
-    unique('agent_connectors_agent_id_connector_id_unique').on(
-      t.agentId,
-      t.connectorId,
-    ),
-  ],
-);

package/runtime/node_modules/@metro-labs/mcp/src/db/token-carry.ts

@@ -1,45 +0,0 @@
-import { existsSync, readFileSync } from 'node:fs';
-import { tokenStorePath } from '@metro-labs/whatsapp/tokens';
-import { errMsg, log } from '../daemon/log.js';
-import { writeSecure } from '../daemon/secure-fs.js';
-
-export const PREVIOUS_ACCOUNT_ID = 'previousAccountId';
-
-export interface CarriedStation {
-  station: string;
-  id: string;
-  config: Record<string, unknown>;
-}
-
-export function previousAccountId(
-  config: Record<string, unknown>,
-): string | undefined {
-  const raw = config[PREVIOUS_ACCOUNT_ID];
-  return typeof raw === 'string' && raw !== '' ? raw : undefined;
-}
-
-export function carryTokenStores(list: CarriedStation[]): number {
-  let carried = 0;
-  for (const s of list) {
-    if (s.station !== 'whatsapp') continue;
-    const previous = previousAccountId(s.config);
-    if (previous === undefined || previous === s.id) continue;
-    const from = tokenStorePath(previous);
-    const to = tokenStorePath(s.id);
-    if (!existsSync(from) || existsSync(to)) continue;
-    try {
-      writeSecure(to, readFileSync(from, 'utf8'));
-      carried += 1;
-      log.info(
-        { from: previous, to: s.id },
-        'db: carried the whatsapp token store to the station id',
-      );
-    } catch (err) {
-      log.error(
-        { from: previous, to: s.id, err: errMsg(err) },
-        'db: could not carry the whatsapp token store',
-      );
-    }
-  }
-  return carried;
-}

package/runtime/node_modules/@metro-labs/mcp/src/db/users.ts

@@ -1,72 +0,0 @@
-import { eq, type SQL } from 'drizzle-orm';
-import { ApiError } from '../daemon/api-error.js';
-import { getDb } from './client.js';
-import { newId } from './ids.js';
-import { users } from './schema.js';
-
-const UNIQUE_VIOLATION = '23505';
-const ADDRESS_RE = /^0x[0-9a-f]{40}$/;
-
-export class UserError extends ApiError {}
-
-const CAUSE_DEPTH = 5;
-
-export function isUniqueViolation(err: unknown): boolean {
-  let current: unknown = err;
-  for (let depth = 0; depth < CAUSE_DEPTH; depth += 1) {
-    if (typeof current !== 'object' || current === null) return false;
-    const { code, cause } = current as { code?: unknown; cause?: unknown };
-    if (code === UNIQUE_VIOLATION) return true;
-    current = cause;
-  }
-  return false;
-}
-
-export function normalizeEmail(raw: string): string {
-  return raw.trim().toLowerCase();
-}
-
-export function normalizeAddress(raw: string): string | null {
-  const address = raw.trim().toLowerCase();
-  return ADDRESS_RE.test(address) ? address : null;
-}
-
-async function userIdWhere(where: SQL): Promise<string | null> {
-  const rows = await getDb().select({ id: users.id }).from(users).where(where);
-  return rows[0]?.id ?? null;
-}
-
-export async function userIdForSubject(subject: string): Promise<string | null> {
-  const address = normalizeAddress(subject);
-  if (address !== null) return userIdWhere(eq(users.address, address));
-  return userIdWhere(eq(users.email, normalizeEmail(subject)));
-}
-
-export async function resolveUserId(
-  insert: () => Promise<string | undefined>,
-  lookup: () => Promise<string | null>,
-): Promise<string> {
-  const inserted = await insert();
-  if (inserted !== undefined) return inserted;
-  const existing = await lookup();
-  if (existing === null)
-    throw new UserError('user lookup returned no id', 500);
-  return existing;
-}
-
-export async function ensureUserByAddress(raw: string): Promise<string> {
-  const address = normalizeAddress(raw);
-  if (address === null)
-    throw new UserError('an Ethereum address is required', 400);
-  return resolveUserId(
-    async () => {
-      const rows = await getDb()
-        .insert(users)
-        .values({ id: newId(), address })
-        .onConflictDoNothing({ target: users.address })
-        .returning({ id: users.id });
-      return rows[0]?.id;
-    },
-    () => userIdWhere(eq(users.address, address)),
-  );
-}

package/runtime/node_modules/@metro-labs/mcp/src/db/whatsapp-login.ts

@@ -1,18 +0,0 @@
-import { and, eq, sql } from 'drizzle-orm';
-import { getDb } from './client.js';
-import { stations } from './schema.js';
-
-export async function writeWhatsappCredentials(
-  accountId: string,
-  credentials: unknown,
-): Promise<void> {
-  const db = getDb();
-  await db
-    .update(stations)
-    .set({
-      config: sql`${stations.config} || ${JSON.stringify({ credentials })}::jsonb`,
-    })
-    .where(
-      and(eq(stations.station, 'whatsapp'), eq(stations.id, accountId)),
-    );
-}

package/runtime/node_modules/@metro-labs/mcp/src/mcp/channel-owner.ts

@@ -1,29 +0,0 @@
-import { eventInScope } from '../db/agent-scope.js';
-import { allowedAgents, type RequestIdentity } from './request-identity.js';
-import { sessionScopeKey } from './session-route.js';
-
-export class ChannelOwner {
-  private stream: RequestIdentity | undefined;
-
-  bindStream(identity: RequestIdentity): void {
-    this.stream = identity;
-  }
-
-  releaseStream(): void {
-    this.stream = undefined;
-  }
-
-  scope(): Set<string> {
-    return allowedAgents(this.stream);
-  }
-
-  streamBelongsTo(identity: RequestIdentity): boolean {
-    if (this.stream === undefined) return false;
-    return sessionScopeKey(this.stream) === sessionScopeKey(identity);
-  }
-
-  inScope(line: string): boolean {
-    if (this.stream === undefined) return false;
-    return eventInScope(this.scope(), line);
-  }
-}

package/runtime/node_modules/@metro-labs/mcp/src/mcp/frame-scope.ts

@@ -1,21 +0,0 @@
-import type { JSONRPCMessage } from '@modelcontextprotocol/sdk/types.js';
-import { eventInScope } from '../db/agent-scope.js';
-import { str } from './str.js';
-
-export interface FrameOrigin {
-  line: string | undefined;
-  owner: Set<string>;
-}
-
-export function frameLine(message: JSONRPCMessage): string | undefined {
-  const params: unknown = (message as { params?: unknown }).params;
-  if (typeof params !== 'object' || params === null) return undefined;
-  const meta: unknown = (params as { meta?: unknown }).meta;
-  if (typeof meta !== 'object' || meta === null) return undefined;
-  return str((meta as { line?: unknown }).line) || undefined;
-}
-
-export function frameInScope(scope: Set<string>, origin: FrameOrigin): boolean {
-  if (origin.line !== undefined) return eventInScope(scope, origin.line);
-  return origin.owner.size > 0 && [...origin.owner].every((id) => scope.has(id));
-}

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

@@ -1,192 +0,0 @@
-import type { IncomingMessage, ServerResponse } from 'node:http';
-import {
-  BodyTooLargeError,
-  readBody as readBodyBuffer,
-} from '../daemon/http.js';
-import {
-  allowedAgents,
-  authConfigFromEnv,
-  authenticate,
-  runWithIdentity,
-  type RequestIdentity,
-} from './request-identity.js';
-import {
-  headerValue,
-  isStandaloneGet,
-  serveChannelGet,
-} from './raw-get-stream.js';
-import { channelLog, type McpSession } from './session.js';
-import {
-  SessionCapacityError,
-  SessionRegistry,
-  type AgentLiveness,
-} from './session-registry.js';
-import { routeSession, sessionScopeKey } from './session-route.js';
-
-const isInitialize = (b: unknown): boolean =>
-  !!b &&
-  typeof b === 'object' &&
-  (b as { method?: string }).method === 'initialize';
-
-const MCP_BODY_MAX = 32 * 1024 * 1024;
-
-async function readBody(req: IncomingMessage): Promise<unknown> {
-  const raw = (await readBodyBuffer(req, MCP_BODY_MAX)).toString('utf8');
-  try {
-    return raw ? JSON.parse(raw) : undefined;
-  } catch {
-    return undefined;
-  }
-}
-
-type ParsedBody = { ok: true; body: unknown } | { ok: false };
-
-async function readOrReject(
-  req: IncomingMessage,
-  res: ServerResponse,
-): Promise<ParsedBody> {
-  try {
-    const body = req.method === 'POST' ? await readBody(req) : undefined;
-    return { ok: true, body };
-  } catch (err) {
-    if (!(err instanceof BodyTooLargeError)) throw err;
-    res.writeHead(413).end('payload too large');
-    return { ok: false };
-  }
-}
-
-async function resolveSession(
-  registry: SessionRegistry,
-  req: IncomingMessage,
-  body: unknown,
-  identity: RequestIdentity,
-  res: ServerResponse,
-): Promise<McpSession | undefined> {
-  const presented = headerValue(req, 'mcp-session-id');
-  const scopeKey = sessionScopeKey(identity);
-  const route = routeSession({
-    isInitialize: isInitialize(body),
-    presented,
-    ownership: registry.ownership(presented, scopeKey),
-    hasOwnSession: registry.forScope(scopeKey) !== undefined,
-  });
-  if (route.kind === 'reject') {
-    channelLog(
-      'session: refused',
-      'status',
-      route.status,
-      'presented',
-      presented ?? '(none)',
-      'scope',
-      scopeKey,
-    );
-    res.writeHead(route.status).end(route.message);
-    return undefined;
-  }
-  if (route.kind === 'use') {
-    const found =
-      presented === undefined
-        ? registry.forScope(scopeKey)
-        : registry.get(presented);
-    if (found) return found;
-    res.writeHead(404).end('Session not found');
-    return undefined;
-  }
-  try {
-    return await registry.create(identity, route.adoptId);
-  } catch (err) {
-    if (!(err instanceof SessionCapacityError)) throw err;
-    res.writeHead(503).end(err.message);
-    return undefined;
-  }
-}
-
-async function serveGet(
-  session: McpSession,
-  req: IncomingMessage,
-  res: ServerResponse,
-  identity: RequestIdentity,
-): Promise<void> {
-  if (session.currentSink && !session.owner.streamBelongsTo(identity)) {
-    channelLog(
-      'session: refused stream takeover',
-      'session',
-      session.id,
-      'scope',
-      session.scopeKey,
-    );
-    res.writeHead(409).end('Conflict: stream held by another identity');
-    return;
-  }
-  const served = await serveChannelGet({
-    transport: session.transport,
-    eventStore: session.eventStore,
-    scope: allowedAgents(identity),
-    req,
-    res,
-    previous: session.currentSink,
-    log: channelLog,
-    registerSink: (sink) => {
-      session.bindSink(sink, identity);
-    },
-  });
-  if (served) session.channel.replayMissed();
-}
-
-let activeRegistry: SessionRegistry | undefined;
-
-export function agentLiveness(): Map<string, AgentLiveness> {
-  return activeRegistry?.liveness() ?? new Map<string, AgentLiveness>();
-}
-
-export async function closeAgentSession(agentId: string): Promise<boolean> {
-  const scopeKey = sessionScopeKey({ kind: 'agent', agentId });
-  return (await activeRegistry?.closeScope(scopeKey)) ?? false;
-}
-
-export async function createMetroMcp(): Promise<{
-  httpHandler: (req: IncomingMessage, res: ServerResponse) => Promise<void>;
-  startInbound: () => void;
-}> {
-  await activeRegistry?.closeAll();
-  const registry = new SessionRegistry(channelLog);
-  activeRegistry = registry;
-
-  const httpHandler = async (
-    req: IncomingMessage,
-    res: ServerResponse,
-  ): Promise<void> => {
-    const identity = authenticate(req, authConfigFromEnv());
-    if (!identity) {
-      res.writeHead(401).end('unauthorized');
-      return;
-    }
-    const parsed = await readOrReject(req, res);
-    if (!parsed.ok) return;
-    await runWithIdentity(identity, async () => {
-      const session = await resolveSession(
-        registry,
-        req,
-        parsed.body,
-        identity,
-        res,
-      );
-      if (!session) return;
-      session.touch();
-      if (isStandaloneGet(req)) {
-        await serveGet(session, req, res, identity);
-        return;
-      }
-      await session.transport.handleRequest(req, res, parsed.body);
-    });
-  };
-
-  const startInbound = (): void => {
-    registry.startInbound();
-    channelLog(
-      'inbound: per-session bus subscriptions (bounded replay on reconnect)',
-    );
-  };
-
-  return { httpHandler, startInbound };
-}

package/runtime/node_modules/@metro-labs/mcp/src/mcp/permission-relay.ts

@@ -1,66 +0,0 @@
-import type { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { z } from 'zod';
-import type { InboundRelay } from '../channels/inbound.js';
-import type { ChannelOwner } from './channel-owner.js';
-import { metroCall } from './ctx.js';
-
-const PermissionRequestSchema = z.object({
-  method: z.literal('notifications/claude/channel/permission_request'),
-  params: z.object({
-    request_id: z.string(),
-    tool_name: z.string(),
-    description: z.string(),
-    input_preview: z.string(),
-  }),
-});
-
-type PermissionRequest = z.infer<typeof PermissionRequestSchema>;
-
-const trainOf = (line: string): string => line.split('/')[2] ?? '';
-
-async function metroSend(line: string, text: string): Promise<void> {
-  await metroCall(trainOf(line), 'send', { line, text });
-}
-
-function promptBody(params: PermissionRequest['params']): string {
-  return (
-    `Claude wants to run ${params.tool_name}: ${params.description}\n` +
-    (params.input_preview ? `\n${params.input_preview}\n` : '') +
-    `\nReply "yes ${params.request_id}" or "no ${params.request_id}"`
-  );
-}
-
-export interface PermissionRelayDeps {
-  mcp: Server;
-  relay: InboundRelay;
-  owner: ChannelOwner;
-  log: (...a: unknown[]) => void;
-}
-
-export function registerPermissionRelay(deps: PermissionRelayDeps): void {
-  const { mcp, relay, owner, log } = deps;
-  mcp.setNotificationHandler(
-    PermissionRequestSchema as never,
-    async (n: PermissionRequest) => {
-      const { params } = n;
-      const line = relay.knownLine;
-      if (!line) {
-        log('permission_request but no known line to relay to', params.request_id);
-        return;
-      }
-      if (!owner.inScope(line)) {
-        log(
-          'permission_request: known line is outside the channel session scope',
-          params.request_id,
-        );
-        return;
-      }
-      relay.registerPermission(params.request_id);
-      try {
-        await metroSend(line, promptBody(params));
-      } catch (e) {
-        log('relay send failed', e);
-      }
-    },
-  );
-}

package/runtime/node_modules/@metro-labs/mcp/src/mcp/request-identity.ts

@@ -1,73 +0,0 @@
-import { AsyncLocalStorage } from 'node:async_hooks';
-import type { IncomingMessage } from 'node:http';
-import { verifySession } from '../daemon/session.js';
-import { agentIdForKey } from '../db/key-map.js';
-
-export type RequestIdentity =
-  | { kind: 'agent'; agentId: string }
-  | { kind: 'session'; subject: string; agentIds: string[] };
-
-const storage = new AsyncLocalStorage<RequestIdentity>();
-
-export function runWithIdentity<T>(
-  identity: RequestIdentity,
-  fn: () => Promise<T>,
-): Promise<T> {
-  return storage.run(identity, fn);
-}
-
-export function currentIdentity(): RequestIdentity | undefined {
-  return storage.getStore();
-}
-
-export interface AuthConfig {
-  sessionSecret: string;
-}
-
-export function authConfigFromEnv(): AuthConfig {
-  return { sessionSecret: process.env.METRO_SESSION_SECRET?.trim() ?? '' };
-}
-
-export function extractToken(req: IncomingMessage): string | undefined {
-  const header = req.headers.authorization;
-  if (header?.startsWith('Bearer ')) {
-    const t = header.slice(7).trim();
-    if (t) return t;
-  }
-  const qt = new URL(req.url ?? '/', 'http://localhost').searchParams.get(
-    'token',
-  );
-  return qt ?? undefined;
-}
-
-const looksLikeJwt = (token: string): boolean =>
-  /^[\w-]+\.[\w-]+\.[\w-]+$/.test(token);
-
-export function authenticate(
-  req: IncomingMessage,
-  cfg: AuthConfig,
-): RequestIdentity | null {
-  const token = extractToken(req);
-  if (!token) return null;
-
-  const agentId = agentIdForKey(token);
-  if (agentId !== undefined) return { kind: 'agent', agentId };
-
-  if (cfg.sessionSecret !== '' && looksLikeJwt(token)) {
-    try {
-      const { subject, agentIds } = verifySession(token, cfg.sessionSecret);
-      return { kind: 'session', subject, agentIds };
-    } catch {
-      return null;
-    }
-  }
-  return null;
-}
-
-export function allowedAgents(
-  identity: RequestIdentity | undefined,
-): Set<string> {
-  if (identity?.kind === 'session') return new Set(identity.agentIds);
-  if (identity?.kind === 'agent') return new Set([identity.agentId]);
-  return new Set();
-}