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

package/runtime/node_modules/@metro-labs/mcp/src/daemon/local-mode.ts

@@ -0,0 +1,158 @@
+import { hostname } from 'node:os';
+import { ApiError } from './api-error.js';
+import { errMsg, log } from './log.js';
+import { AttachSessions } from './attach-session.js';
+import type { AgentApiDeps } from './agent-api.js';
+import { ATTACHABLE, type AccountApiDeps } from './account-api.js';
+import type { AgentConnectorApiDeps } from './agent-connector-api.js';
+import type { ProjectApiDeps } from './project-api.js';
+import type { SessionApis } from './session-apis.js';
+import type { ModeInfo } from './mode-api.js';
+import {
+  claimLocalOwner,
+  LOCAL_PROJECT_ID,
+  localAttachAccount,
+  localCreateAgent,
+  localDeleteAgent,
+  localDetachAccount,
+  localListAgents,
+  localOwnedAgentOrThrow,
+  localOwner,
+  localResetAgentKey,
+} from '../db/file-admin.js';
+import { normalizeAddress } from '../db/users.js';
+import type { Project } from '../db/projects.js';
+import type { StationName } from '../db/schema.js';
+
+export interface LocalModeDeps {
+  syncStations: (station: StationName) => Promise<void>;
+  closeAgentSession: (id: string) => Promise<boolean>;
+  gatherAccounts: AgentApiDeps['gatherAccounts'];
+  capabilities: AgentApiDeps['capabilities'];
+  liveness: AgentApiDeps['liveness'];
+  prepareAccount: AccountApiDeps['prepareAccount'];
+}
+
+const notHere = (what: string): Promise<never> =>
+  Promise.reject(
+    new ApiError(`${what} are managed on metro.box, not on a local daemon`, 400),
+  );
+
+function localProject(): Project {
+  return {
+    id: LOCAL_PROJECT_ID,
+    name: hostname(),
+    isDefault: true,
+    owner: true,
+    role: 'admin',
+  };
+}
+
+function attachSessions(deps: LocalModeDeps): AttachSessions {
+  return new AttachSessions({
+    authorize: async (owner) => {
+      await localOwnedAgentOrThrow(owner.subject, owner.agentId);
+    },
+    complete: async (owner, station, config) => {
+      const ref = await localAttachAccount(owner.subject, owner.agentId, station, config);
+      const activated = await deps.syncStations(station).then(
+        () => true,
+        (err: unknown) => {
+          log.warn(
+            { station, err: errMsg(err) },
+            'attach-session: station reload failed, the change lands at the next boot',
+          );
+          return false;
+        },
+      );
+      return { accountId: ref.accountId, activated };
+    },
+  });
+}
+
+function agentApi(deps: LocalModeDeps): AgentApiDeps {
+  return {
+    attachSessions: attachSessions(deps),
+    listAgents: localListAgents,
+    createAgent: localCreateAgent,
+    deleteAgent: localDeleteAgent,
+    resetKey: async (subject, id) => {
+      const reset = await localResetAgentKey(subject, id);
+      const closed = await deps.closeAgentSession(id);
+      log.info({ agent: reset.name, id, sessionClosed: closed }, 'local: key rotated');
+      return reset;
+    },
+    gatherAccounts: deps.gatherAccounts,
+    capabilities: deps.capabilities,
+    attachable: ATTACHABLE.filter((s) => s !== 'webhook'),
+    liveness: deps.liveness,
+    releaseRuntime: () => notHere('runtime leases'),
+    runtimes: () => Promise.resolve(new Map()),
+    connectorIds: () => Promise.resolve(new Map()),
+    prepareAccount: deps.prepareAccount,
+    attachAccount: localAttachAccount,
+    detachAccount: localDetachAccount,
+    syncStations: deps.syncStations,
+  };
+}
+
+const projectApi: ProjectApiDeps = {
+  listProjects: (subject) => {
+    const owner = localOwner();
+    const mine = owner !== null && owner === normalizeAddress(subject);
+    return Promise.resolve(mine ? [localProject()] : []);
+  },
+  createProject: () => notHere('projects'),
+  renameProject: () => notHere('projects'),
+  deleteProject: () => notHere('projects'),
+  listMembers: (subject, id) => {
+    const owner = localOwner();
+    if (id !== LOCAL_PROJECT_ID || owner === null || owner !== normalizeAddress(subject))
+      return Promise.reject(new ApiError('no such project', 404));
+    return Promise.resolve([
+      { id: 'owner', address: owner, email: null, role: 'admin', owner: true },
+    ]);
+  },
+  addMember: () => notHere('members'),
+  setMemberRole: () => notHere('members'),
+  removeMember: () => notHere('members'),
+};
+
+const agentConnectorApi: AgentConnectorApiDeps = {
+  agentConnectors: async (subject, id) => {
+    const { agent } = await localOwnedAgentOrThrow(subject, id);
+    return { ...agent, connectorIds: [] };
+  },
+  addConnector: () => notHere('connectors'),
+  removeConnector: () => notHere('connectors'),
+  mintCode: () =>
+    Promise.reject(new ApiError('a local daemon has no pairing codes', 400)),
+};
+
+export function localConnectHint(port: number): string {
+  const ui = process.env.METRO_UI_URL ?? 'https://metro.box';
+  const here = `http://127.0.0.1:${String(port)}`;
+  const forward = `ssh -L ${String(port)}:127.0.0.1:${String(port)} ${hostname()}`;
+  return [
+    'Manage this machine from the web UI:',
+    '',
+    `  ${ui}/#/connect/${encodeURIComponent(here)}`,
+    '',
+    `From another computer, forward the port first:  ${forward}`,
+    '',
+  ].join('\n');
+}
+
+export function localModeInfo(): ModeInfo {
+  return { mode: 'local', owner: localOwner(), project: LOCAL_PROJECT_ID };
+}
+
+export function localSessionApis(deps: LocalModeDeps): SessionApis {
+  return {
+    agentApi: agentApi(deps),
+    agentConnectorApi,
+    projectApi,
+    siwe: { ensureUser: claimLocalOwner },
+    mode: localModeInfo,
+  };
+}

package/runtime/node_modules/@metro-labs/mcp/src/daemon/local-secret.ts

@@ -0,0 +1,21 @@
+import { randomBytes } from 'node:crypto';
+import { existsSync, readFileSync } from 'node:fs';
+import { join } from 'node:path';
+import { agentsDir } from '../db/file-source.js';
+import { ensureSecureDir, writeSecure } from './secure-fs.js';
+
+const SECRET_FILE = '.session-secret';
+
+export function ensureLocalSessionSecret(dir = agentsDir()): string {
+  const configured = process.env.METRO_SESSION_SECRET?.trim() ?? '';
+  if (configured !== '') return configured;
+  const path = join(dir, SECRET_FILE);
+  let secret = existsSync(path) ? readFileSync(path, 'utf8').trim() : '';
+  if (secret === '') {
+    secret = randomBytes(32).toString('base64url');
+    ensureSecureDir(dir);
+    writeSecure(path, `${secret}\n`);
+  }
+  process.env.METRO_SESSION_SECRET = secret;
+  return secret;
+}

package/runtime/node_modules/@metro-labs/mcp/src/daemon/mode-api.ts

@@ -0,0 +1,31 @@
+import type { IncomingMessage, ServerResponse } from 'node:http';
+import { cors, sendJson } from './api-http.js';
+
+const PATH = '/api/mode';
+
+export type DaemonMode = 'hosted' | 'linked' | 'local';
+
+export interface ModeInfo {
+  mode: DaemonMode;
+  owner: string | null;
+  project: string | null;
+}
+
+export function handleModeRequest(
+  req: IncomingMessage,
+  res: ServerResponse,
+  info: () => ModeInfo,
+): boolean {
+  const path = (req.url ?? '').split('?')[0] ?? '';
+  if (path !== PATH) return false;
+  if (req.method === 'OPTIONS') {
+    res.writeHead(204, cors(req)).end();
+    return true;
+  }
+  if (req.method !== 'GET') {
+    sendJson(req, res, 405, { error: 'method not allowed' });
+    return true;
+  }
+  sendJson(req, res, 200, info());
+  return true;
+}

package/runtime/node_modules/@metro-labs/mcp/src/daemon/oauth-pending.ts

@@ -7,7 +7,7 @@ const SWEEP_MS = 60_000;
 const MAX_PENDING = 100;
 
 export interface PendingAuth {
-  email: string;
+  subject: string;
   name: string;
   url: string;
   resource: string;

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

@@ -10,6 +10,7 @@ import {
 } from 'node:fs';
 import { homedir } from 'node:os';
 import { join } from 'node:path';
+import { fileURLToPath } from 'node:url';
 import { log } from './log.js';
 
 export const STATE_DIR =
@@ -23,6 +24,15 @@ export const CONFIG_ENV_FILE = join(CONFIG_DIR, '.env');
 
 export const TRAINS_ENV_FILE = join(homedir(), '.metro', '.env');
 
+export function trainsDir(): string {
+  const env = process.env.METRO_TRAINS_DIR?.trim();
+  if (env !== undefined && env !== '') return env;
+  return fileURLToPath(new URL('../../trains', import.meta.url));
+}
+
+export const isLocalMode = (): boolean =>
+  process.env.METRO_MODE?.trim() === 'local';
+
 export const envSources = (): readonly string[] => [
   join(process.cwd(), '.env'),
   TRAINS_ENV_FILE,

package/runtime/node_modules/@metro-labs/mcp/src/daemon/project-api.ts

@@ -15,32 +15,32 @@ import type { Member, Project } from '../db/projects.js';
 const PREFIX = '/api/projects';
 
 export interface ProjectApiDeps {
-  listProjects: (email: string) => Promise<Project[]>;
-  createProject: (email: string, name: unknown) => Promise<Project>;
+  listProjects: (subject: string) => Promise<Project[]>;
+  createProject: (subject: string, name: unknown) => Promise<Project>;
   renameProject: (
-    email: string,
+    subject: string,
     id: string,
     name: unknown,
   ) => Promise<Project>;
   deleteProject: (
-    email: string,
+    subject: string,
     id: string,
   ) => Promise<{ id: string; name: string }>;
-  listMembers: (email: string, id: string) => Promise<Member[]>;
+  listMembers: (subject: string, id: string) => Promise<Member[]>;
   addMember: (
-    email: string,
+    subject: string,
     id: string,
     invited: unknown,
     role: unknown,
   ) => Promise<Member[]>;
   setMemberRole: (
-    email: string,
+    subject: string,
     id: string,
     memberId: string,
     role: unknown,
   ) => Promise<Member[]>;
   removeMember: (
-    email: string,
+    subject: string,
     id: string,
     memberId: string,
   ) => Promise<Member[]>;
@@ -98,11 +98,11 @@ async function handleIndex(
   session: ApiSession,
 ): Promise<void> {
   if (req.method === 'GET') {
-    sendJson(req, res, 200, { projects: await deps.listProjects(session.email) });
+    sendJson(req, res, 200, { projects: await deps.listProjects(session.subject) });
     return;
   }
   const made = await deps.createProject(
-    session.email,
+    session.subject,
     bodyField(await readJsonBody(req), 'name'),
   );
   log.info({ id: made.id, name: made.name }, 'project-api: created project');
@@ -117,14 +117,14 @@ async function handleMembers(
   id: string,
 ): Promise<void> {
   if (req.method === 'GET') {
-    sendJson(req, res, 200, { members: await deps.listMembers(session.email, id) });
+    sendJson(req, res, 200, { members: await deps.listMembers(session.subject, id) });
     return;
   }
   const body = await readJsonBody(req);
   const members = await deps.addMember(
-    session.email,
+    session.subject,
     id,
-    bodyField(body, 'email'),
+    bodyField(body, 'address'),
     bodyField(body, 'role'),
   );
   sendJson(req, res, 200, { members });
@@ -138,12 +138,12 @@ async function handleMember(
   tgt: { id: string; memberId: string },
 ): Promise<void> {
   if (req.method === 'DELETE') {
-    const members = await deps.removeMember(session.email, tgt.id, tgt.memberId);
+    const members = await deps.removeMember(session.subject, tgt.id, tgt.memberId);
     sendJson(req, res, 200, { members });
     return;
   }
   const members = await deps.setMemberRole(
-    session.email,
+    session.subject,
     tgt.id,
     tgt.memberId,
     bodyField(await readJsonBody(req), 'role'),
@@ -170,12 +170,12 @@ async function route(
         res,
         200,
         await deps.renameProject(
-          session.email,
+          session.subject,
           tgt.id,
           bodyField(await readJsonBody(req), 'name'),
         ),
       );
-    else sendJson(req, res, 200, await deps.deleteProject(session.email, tgt.id));
+    else sendJson(req, res, 200, await deps.deleteProject(session.subject, tgt.id));
   } catch (err) {
     apiFailure(req, res, err, 'project-api');
   }

package/runtime/node_modules/@metro-labs/mcp/src/daemon/run-api.ts

@@ -9,7 +9,7 @@ import {
   sendJson,
 } from './api-http.js';
 import { AGENT_CODE_RE, takeAgentCode } from './agent-pair.js';
-import { sessionTtlFromEnv } from './google-oauth.js';
+import { sessionTtlFromEnv } from './session-config.js';
 import { signRunToken } from './session.js';
 import type { LoadedAgent } from '../db/materialize.js';
 import type { RuntimeLease } from '../db/runtimes.js';
@@ -78,7 +78,7 @@ async function handleClaim(
   }
   const lease = await deps.claimRuntime(taken.agentId, labelOf(body));
   const token = signRunToken(
-    { email: taken.email, agentId: lease.agentId, runtimeId: lease.runtimeId },
+    { subject: taken.subject, agentId: lease.agentId, runtimeId: lease.runtimeId },
     secret,
     { ttlSec: sessionTtlFromEnv() },
   );

package/runtime/node_modules/@metro-labs/mcp/src/daemon/session-api.ts

@@ -22,6 +22,6 @@ export function handleSessionApiRequest(
     sendJson(req, res, 401, { error: 'unauthorized' });
     return true;
   }
-  sendJson(req, res, 200, { email: session.email });
+  sendJson(req, res, 200, { subject: session.subject });
   return true;
 }

package/runtime/node_modules/@metro-labs/mcp/src/daemon/session-apis.ts

@@ -16,6 +16,8 @@ import {
   handleAgentConnectorRequest,
   type AgentConnectorApiDeps,
 } from './agent-connector-api.js';
+import type { SiweRouteDeps } from './siwe-routes.js';
+import type { ModeInfo } from './mode-api.js';
 
 export interface SessionApis {
   agentApi?: AgentApiDeps;
@@ -24,6 +26,8 @@ export interface SessionApis {
   projectApi?: ProjectApiDeps;
   runApi?: RunApiDeps;
   relayApi?: RelayApiDeps;
+  siwe?: SiweRouteDeps;
+  mode?: () => ModeInfo;
 }
 
 export function handleSessionApis(

package/runtime/node_modules/@metro-labs/mcp/src/daemon/session-config.ts

@@ -0,0 +1,27 @@
+const DEFAULT_SESSION_TTL_SEC = 30 * 24 * 3600;
+const PREVIEW = /^([a-z0-9-]+--)?metro-ui\.netlify\.app$/;
+
+export function sessionTtlFromEnv(): number {
+  const ttl = Number(process.env.METRO_SESSION_TTL_SEC);
+  return Number.isFinite(ttl) && ttl > 0 ? ttl : DEFAULT_SESSION_TTL_SEC;
+}
+
+export function allowedWebHost(host: string): boolean {
+  if (host === 'metro.box' || host === 'localhost' || host === '127.0.0.1')
+    return true;
+  return PREVIEW.test(host);
+}
+
+export function validateReturnTo(returnTo: string): boolean {
+  let url: URL;
+  try {
+    url = new URL(returnTo);
+  } catch {
+    return false;
+  }
+  const host = url.hostname;
+  if (url.protocol === 'http:' && (host === 'localhost' || host === '127.0.0.1'))
+    return true;
+  if (url.protocol !== 'https:') return false;
+  return allowedWebHost(host);
+}

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

@@ -95,17 +95,17 @@ export function verifyState(
 }
 
 interface SessionClaims {
-  email: string;
+  subject: string;
   agentIds: string[];
 }
 
 export interface AgentClaims {
-  email: string;
+  subject: string;
   agentId: string;
 }
 
 export interface RunClaims {
-  email: string;
+  subject: string;
   agentId: string;
   runtimeId: string;
 }
@@ -119,7 +119,7 @@ export function signSession(
   return sign(
     {
       typ: 'session',
-      sub: claims.email,
+      sub: claims.subject,
       agent_ids: claims.agentIds,
       iat,
       exp: iat + (opts.ttlSec ?? 30 * 24 * 3600),
@@ -144,7 +144,7 @@ export function verifySession(
     ids.some((a) => typeof a !== 'string' || !ID_RE.test(a))
   )
     throw new SessionError('malformed session');
-  return { email: p.sub, agentIds: ids as string[] };
+  return { subject: p.sub, agentIds: ids as string[] };
 }
 
 export function signAgentToken(
@@ -156,7 +156,7 @@ export function signAgentToken(
   return sign(
     {
       typ: 'agent',
-      sub: claims.email,
+      sub: claims.subject,
       agent: claims.agentId,
       iat,
       exp: iat + (opts.ttlSec ?? 30 * 24 * 3600),
@@ -176,7 +176,7 @@ export function verifyAgentToken(
     throw new SessionError('agent token expired');
   if (typeof p.sub !== 'string' || typeof p.agent !== 'string')
     throw new SessionError('malformed agent token');
-  return { email: p.sub, agentId: p.agent };
+  return { subject: p.sub, agentId: p.agent };
 }
 
 export function signRunToken(
@@ -188,7 +188,7 @@ export function signRunToken(
   return sign(
     {
       typ: 'run',
-      sub: claims.email,
+      sub: claims.subject,
       agent: claims.agentId,
       rt: claims.runtimeId,
       iat,
@@ -214,5 +214,5 @@ export function verifyRunToken(
     !ID_RE.test(p.agent)
   )
     throw new SessionError('malformed run token');
-  return { email: p.sub, agentId: p.agent, runtimeId: p.rt };
+  return { subject: p.sub, agentId: p.agent, runtimeId: p.rt };
 }

package/runtime/node_modules/@metro-labs/mcp/src/daemon/siwe-auth.ts

@@ -0,0 +1,82 @@
+import { verifyMessage, type Hex } from 'viem';
+import { parseSiweMessage, validateSiweMessage } from 'viem/siwe';
+import { ApiError } from './api-error.js';
+import { allowedWebHost } from './session-config.js';
+
+export class SiweError extends ApiError {}
+
+export interface SiweLogin {
+  message: string;
+  signature: string;
+}
+
+export interface SiweVerifyArgs {
+  address: `0x${string}`;
+  message: string;
+  signature: Hex;
+}
+
+export interface SiweDeps {
+  takeNonce: (nonce: string) => boolean;
+  now?: number;
+  verify?: (args: SiweVerifyArgs) => Promise<boolean>;
+}
+
+const HEX_SIGNATURE = /^0x[0-9a-fA-F]{130}$/;
+
+export function allowedSiweDomain(domain: string): boolean {
+  return allowedWebHost(domain.replace(/:\d+$/, ''));
+}
+
+function hostOf(uri: string): string | null {
+  try {
+    return new URL(uri).host;
+  } catch {
+    return null;
+  }
+}
+
+function fieldsOrThrow(message: string): {
+  address: `0x${string}`;
+  domain: string;
+  nonce: string;
+  parsed: ReturnType<typeof parseSiweMessage>;
+} {
+  const parsed = parseSiweMessage(message);
+  const { address, domain, nonce, uri, version } = parsed;
+  if (
+    address === undefined ||
+    domain === undefined ||
+    nonce === undefined ||
+    uri === undefined ||
+    version !== '1'
+  )
+    throw new SiweError('that is not a sign-in message', 400);
+  if (!allowedSiweDomain(domain))
+    throw new SiweError(`sign-in messages for ${domain} are not accepted here`, 400);
+  if (hostOf(uri) !== domain)
+    throw new SiweError('the sign-in message names a different site', 400);
+  return { address, domain, nonce, parsed };
+}
+
+export async function verifySiweLogin(
+  input: SiweLogin,
+  deps: SiweDeps,
+): Promise<string> {
+  if (!HEX_SIGNATURE.test(input.signature))
+    throw new SiweError('that is not an Ethereum signature', 400);
+  const { address, domain, nonce, parsed } = fieldsOrThrow(input.message);
+  const time = new Date(deps.now ?? Date.now());
+  if (!validateSiweMessage({ message: parsed, domain, nonce, time }))
+    throw new SiweError('this sign-in message is expired or does not match', 400);
+  if (!deps.takeNonce(nonce))
+    throw new SiweError('that sign-in has expired or was already used', 400);
+  const verify = deps.verify ?? verifyMessage;
+  const signature = input.signature as Hex;
+  if (!(await verify({ address, message: input.message, signature })))
+    throw new SiweError(
+      'the signature does not match the address; smart-contract wallets are not supported yet',
+      401,
+    );
+  return address.toLowerCase();
+}

package/runtime/node_modules/@metro-labs/mcp/src/daemon/siwe-nonces.ts

@@ -0,0 +1,43 @@
+import { generateSiweNonce } from 'viem/siwe';
+
+const TTL_MS = 10 * 60_000;
+const SWEEP_MS = 60_000;
+const MAX_LIVE = 500;
+
+const live = new Map<string, number>();
+let sweeper: ReturnType<typeof setInterval> | null = null;
+
+function sweep(now = Date.now()): void {
+  for (const [nonce, expiresAt] of live) if (expiresAt <= now) live.delete(nonce);
+}
+
+function ensureSweeper(): void {
+  if (sweeper !== null) return;
+  sweeper = setInterval(sweep, SWEEP_MS);
+  sweeper.unref?.();
+}
+
+export function mintNonce(now = Date.now()): string {
+  sweep(now);
+  while (live.size >= MAX_LIVE) {
+    const oldest = live.keys().next();
+    if (oldest.done) break;
+    live.delete(oldest.value);
+  }
+  const nonce = generateSiweNonce();
+  live.set(nonce, now + TTL_MS);
+  ensureSweeper();
+  return nonce;
+}
+
+export function takeNonce(nonce: string, now = Date.now()): boolean {
+  const expiresAt = live.get(nonce);
+  if (expiresAt === undefined) return false;
+  live.delete(nonce);
+  return expiresAt > now;
+}
+
+export function nonceCount(): number {
+  sweep();
+  return live.size;
+}