@stage-labs/metro 0.1.0-beta.20 → 0.1.0-beta.201

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

@@ -1,37 +0,0 @@
-import { drizzle, type PostgresJsDatabase } from 'drizzle-orm/postgres-js';
-import postgres from 'postgres';
-import { log } from '../daemon/log.js';
-import * as schema from './schema.js';
-
-export type Db = PostgresJsDatabase<typeof schema>;
-
-let sql: ReturnType<typeof postgres> | null = null;
-let db: Db | null = null;
-
-export function databaseUrl(): string | undefined {
-  const url = process.env.DATABASE_URL?.trim();
-  if (!url) return undefined;
-  return url;
-}
-
-export function getDb(): Db {
-  const url = databaseUrl();
-  if (!url) throw new Error('DATABASE_URL is not set');
-  if (!db) {
-    sql = postgres(url, {
-      max: 4,
-      prepare: false,
-      onnotice: (notice) => {
-        log.debug({ notice: notice.message }, 'postgres notice');
-      },
-    });
-    db = drizzle(sql, { schema });
-  }
-  return db;
-}
-
-export async function closeDb(): Promise<void> {
-  if (sql) await sql.end({ timeout: 5 });
-  sql = null;
-  db = null;
-}

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

@@ -1,218 +0,0 @@
-import { isUniqueViolation } from './users.js';
-import { projectIdOrThrow } from './projects.js';
-import { and, asc, eq, inArray, ne } from 'drizzle-orm';
-import { getDb } from './client.js';
-import { newId } from './ids.js';
-import { ConnectorError, connectorName } from './connector-config.js';
-import { collectionItems, collections, connectors } from './schema.js';
-
-export interface ConnectorCollectionRow {
-  id: string;
-  name: string;
-  connectorIds: string[];
-}
-
-interface CollectionRow {
-  id: string;
-  projectId: string;
-  name: string;
-}
-
-const missing = (): ConnectorError =>
-  new ConnectorError('no such collection', 404);
-
-const duplicate = (name: string): ConnectorError =>
-  new ConnectorError(`you already have a collection named '${name}'`, 409);
-
-const nameClash = (name: string, collection: string): ConnectorError =>
-  new ConnectorError(
-    `the collection '${collection}' already has a connector named '${name}'`,
-    409,
-  );
-
-async function collidingCollection(
-  collectionIds: string[],
-  name: string,
-  exceptConnectorId: string,
-): Promise<string | null> {
-  if (collectionIds.length === 0) return null;
-  const rows = await getDb()
-    .select({ collection: collections.name })
-    .from(collectionItems)
-    .innerJoin(connectors, eq(connectors.id, collectionItems.connectorId))
-    .innerJoin(collections, eq(collections.id, collectionItems.collectionId))
-    .where(
-      and(
-        inArray(collectionItems.collectionId, collectionIds),
-        eq(connectors.name, name),
-        ne(connectors.id, exceptConnectorId),
-      ),
-    )
-    .limit(1);
-  return rows[0]?.collection ?? null;
-}
-
-export async function assertRenameFreeOfClash(
-  connectorId: string,
-  name: string,
-): Promise<void> {
-  const rows = await getDb()
-    .select({ collectionId: collectionItems.collectionId })
-    .from(collectionItems)
-    .where(eq(collectionItems.connectorId, connectorId));
-  const clash = await collidingCollection(
-    rows.map((r) => r.collectionId),
-    name,
-    connectorId,
-  );
-  if (clash !== null) throw nameClash(name, clash);
-}
-
-async function ownedCollectionOrThrow(
-  email: string,
-  id: string,
-): Promise<CollectionRow> {
-  const rows = await getDb()
-    .select()
-    .from(collections)
-    .where(eq(collections.id, id));
-  const row = rows[0];
-  if (row === undefined) throw missing();
-  try {
-    await projectIdOrThrow(email, row.projectId);
-  } catch {
-    throw missing();
-  }
-  return row;
-}
-
-async function itemsOf(collectionId: string): Promise<string[]> {
-  const rows = await getDb()
-    .select({ connectorId: collectionItems.connectorId })
-    .from(collectionItems)
-    .where(eq(collectionItems.collectionId, collectionId))
-    .orderBy(asc(collectionItems.id));
-  return rows.map((r) => r.connectorId);
-}
-
-async function withItems(row: CollectionRow): Promise<ConnectorCollectionRow> {
-  return { id: row.id, name: row.name, connectorIds: await itemsOf(row.id) };
-}
-
-export async function listCollectionsForEmail(
-  email: string,
-  project: string,
-): Promise<ConnectorCollectionRow[]> {
-  const projectId = await projectIdOrThrow(email, project);
-  const rows = await getDb()
-    .select()
-    .from(collections)
-    .where(eq(collections.projectId, projectId))
-    .orderBy(asc(collections.id));
-  return Promise.all(rows.map(withItems));
-}
-
-export async function getCollectionForEmail(
-  email: string,
-  id: string,
-): Promise<ConnectorCollectionRow> {
-  return withItems(await ownedCollectionOrThrow(email, id));
-}
-
-export async function createCollectionForEmail(
-  email: string,
-  project: string,
-  raw: string,
-): Promise<ConnectorCollectionRow> {
-  const name = connectorName(raw);
-  const projectId = await projectIdOrThrow(email, project);
-  try {
-    const rows = await getDb()
-      .insert(collections)
-      .values({ id: newId(), projectId, name })
-      .returning();
-    const row = rows[0];
-    if (row === undefined)
-      throw new ConnectorError('collection insert returned no row', 500);
-    return { id: row.id, name: row.name, connectorIds: [] };
-  } catch (err) {
-    if (!isUniqueViolation(err)) throw err;
-    throw duplicate(name);
-  }
-}
-
-export async function renameCollectionForEmail(
-  email: string,
-  id: string,
-  raw: string,
-): Promise<ConnectorCollectionRow> {
-  const name = connectorName(raw);
-  const row = await ownedCollectionOrThrow(email, id);
-  try {
-    await getDb()
-      .update(collections)
-      .set({ name })
-      .where(eq(collections.id, row.id));
-  } catch (err) {
-    if (!isUniqueViolation(err)) throw err;
-    throw duplicate(name);
-  }
-  return withItems({ ...row, name });
-}
-
-export async function deleteCollectionForEmail(
-  email: string,
-  id: string,
-): Promise<{ id: string; name: string }> {
-  const row = await ownedCollectionOrThrow(email, id);
-  await getDb().delete(collections).where(eq(collections.id, row.id));
-  return { id: row.id, name: row.name };
-}
-
-async function connectorNameIn(
-  projectId: string,
-  connectorId: string,
-): Promise<string | null> {
-  const rows = await getDb()
-    .select({ name: connectors.name })
-    .from(connectors)
-    .where(and(eq(connectors.id, connectorId), eq(connectors.projectId, projectId)));
-  return rows[0]?.name ?? null;
-}
-
-export async function addToCollectionForEmail(
-  email: string,
-  id: string,
-  connectorId: string,
-): Promise<ConnectorCollectionRow> {
-  const row = await ownedCollectionOrThrow(email, id);
-  const name = await connectorNameIn(row.projectId, connectorId);
-  if (name === null) throw new ConnectorError('no such connector', 404);
-  const clash = await collidingCollection([row.id], name, connectorId);
-  if (clash !== null) throw nameClash(name, clash);
-  try {
-    await getDb()
-      .insert(collectionItems)
-      .values({ id: newId(), collectionId: row.id, connectorId });
-  } catch (err) {
-    if (!isUniqueViolation(err)) throw err;
-  }
-  return withItems(row);
-}
-
-export async function removeFromCollectionForEmail(
-  email: string,
-  id: string,
-  connectorId: string,
-): Promise<ConnectorCollectionRow> {
-  const row = await ownedCollectionOrThrow(email, id);
-  await getDb()
-    .delete(collectionItems)
-    .where(
-      and(
-        eq(collectionItems.collectionId, row.id),
-        eq(collectionItems.connectorId, connectorId),
-      ),
-    );
-  return withItems(row);
-}

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

@@ -1,329 +0,0 @@
-import { assertRenameFreeOfClash } from './connector-collections.js';
-import { projectIdOrThrow } from './projects.js';
-import { and, asc, eq, inArray } from 'drizzle-orm';
-import {
-  connectorUrlText,
-  ConnectorVerifyError,
-  parseConnectorUrl,
-  verifyRemoteMcp,
-  type ConnectorAuth,
-  type OAuthAuth,
-  type VerifiedRecord,
-} from '../daemon/connector-verify.js';
-import { oauthExpired, refreshOAuth } from '../daemon/connector-oauth.js';
-import { advertisesOAuth } from '../daemon/oauth-discovery.js';
-import {
-  ConnectorError,
-  connectorAuth,
-  connectorName,
-  readConfig,
-  signInState,
-  stamp,
-  type ConnectorConfig,
-  type ConnectorSignIn,
-} from './connector-config.js';
-import { errMsg, log } from '../daemon/log.js';
-import { newId } from './ids.js';
-import { getDb } from './client.js';
-import { connectors, type ConnectorTransport } from './schema.js';
-
-export interface Connector {
-  id: string;
-  name: string;
-  url: string;
-  transport: ConnectorTransport;
-  auth: ConnectorAuth['kind'];
-  header: string | null;
-  secret: string | null;
-  bearer: string | null;
-  expiresAt: number | null;
-  signIn: ConnectorSignIn;
-  verified: VerifiedRecord;
-}
-
-export interface ConnectorInput {
-  name: unknown;
-  url: unknown;
-  header: unknown;
-  value: unknown;
-}
-
-export type ConnectorCheck =
-  | { id: string; name: string; ok: true; verified: VerifiedRecord }
-  | { id: string; name: string; ok: false; reason: string };
-
-export interface DeletedConnector {
-  id: string;
-  name: string;
-}
-
-type ConnectorRow = typeof connectors.$inferSelect;
-
-function toConnector(row: ConnectorRow): Connector {
-  const config = readConfig(row.config);
-  const auth = config.auth;
-  return {
-    id: row.id,
-    name: row.name,
-    url: row.url,
-    transport: row.transport,
-    auth: auth.kind,
-    header: auth.kind === 'header' ? auth.name : null,
-    secret: auth.kind === 'header' ? auth.value : null,
-    bearer: auth.kind === 'oauth' ? auth.accessToken : null,
-    expiresAt: auth.kind === 'oauth' ? (auth.expiresAt ?? null) : null,
-    signIn: signInState(config),
-    verified: config.verified,
-  };
-}
-
-const missing = (): ConnectorError =>
-  new ConnectorError('no such connector', 404);
-
-async function ownedConnectorOrThrow(
-  email: string,
-  id: string,
-): Promise<ConnectorRow> {
-  const rows = await getDb()
-    .select()
-    .from(connectors)
-    .where(eq(connectors.id, id));
-  const row = rows[0];
-  if (row === undefined) throw missing();
-  try {
-    await projectIdOrThrow(email, row.projectId);
-  } catch {
-    throw missing();
-  }
-  return row;
-}
-
-async function saveConfig(
-  row: ConnectorRow,
-  config: ConnectorConfig,
-): Promise<Connector> {
-  const rows = await getDb()
-    .update(connectors)
-    .set({ config })
-    .where(and(eq(connectors.id, row.id), eq(connectors.projectId, row.projectId)))
-    .returning();
-  const saved = rows[0];
-  if (saved === undefined) throw missing();
-  return toConnector(saved);
-}
-
-async function connectorRowsFor(
-  email: string,
-  project: string,
-): Promise<ConnectorRow[]> {
-  const projectId = await projectIdOrThrow(email, project);
-  return getDb()
-    .select()
-    .from(connectors)
-    .where(eq(connectors.projectId, projectId))
-    .orderBy(asc(connectors.id));
-}
-
-export async function listConnectorsForEmail(
-  email: string,
-  project: string,
-): Promise<Connector[]> {
-  return (await connectorRowsFor(email, project)).map(toConnector);
-}
-
-async function withFreshToken(row: ConnectorRow): Promise<Connector> {
-  const config = readConfig(row.config);
-  if (config.auth.kind !== 'oauth' || !oauthExpired(config.auth))
-    return toConnector(row);
-  const url = parseConnectorUrl(row.url);
-  const auth = await refreshOAuth(config.auth, url.toString());
-  return saveConfig(row, { ...config, auth });
-}
-
-async function rowsByIds(ids: string[]): Promise<ConnectorRow[]> {
-  if (ids.length === 0) return [];
-  return getDb()
-    .select()
-    .from(connectors)
-    .where(inArray(connectors.id, ids))
-    .orderBy(asc(connectors.id));
-}
-
-export async function listFreshConnectorsByIds(
-  ids: string[],
-): Promise<Connector[]> {
-  const rows = await rowsByIds(ids);
-  return Promise.all(
-    rows.map(async (row) =>
-      withFreshToken(row).catch((err: unknown) => {
-        log.warn(
-          { id: row.id, err: errMsg(err) },
-          'connectors: could not refresh the access token',
-        );
-        return toConnector(row);
-      }),
-    ),
-  );
-}
-
-export async function getConnectorForEmail(
-  email: string,
-  id: string,
-): Promise<Connector> {
-  return toConnector(await ownedConnectorOrThrow(email, id));
-}
-
-async function insertConnector(
-  projectId: string,
-  name: string,
-  url: URL,
-  config: ConnectorConfig,
-): Promise<Connector> {
-  const rows = await getDb()
-    .insert(connectors)
-    .values({
-      id: newId(),
-      projectId,
-      name,
-      url: connectorUrlText(url),
-      transport: 'http',
-      config,
-    })
-    .returning();
-  const row = rows[0];
-  if (row === undefined)
-    throw new ConnectorError('connector insert returned no row', 500);
-  return toConnector(row);
-}
-
-const UNVERIFIED = {
-  at: '',
-  server: '',
-  version: '',
-  protocol: '',
-  icon: '',
-  tools: 0,
-  catalog: [],
-};
-
-export async function createPendingConnectorForEmail(
-  email: string,
-  project: string,
-  input: { name: unknown; url: unknown },
-): Promise<Connector> {
-  const name = connectorName(input.name);
-  const url = parseConnectorUrl(input.url);
-  const projectId = await projectIdOrThrow(email, project);
-  return insertConnector(projectId, name, url, {
-    auth: { kind: 'none' },
-    createdAt: new Date().toISOString(),
-    verified: UNVERIFIED,
-    oauth: true,
-  });
-}
-
-export async function createConnectorForEmail(
-  email: string,
-  project: string,
-  input: ConnectorInput,
-): Promise<Connector> {
-  const name = connectorName(input.name);
-  const url = parseConnectorUrl(input.url);
-  const auth = connectorAuth(input.header, input.value);
-  const projectId = await projectIdOrThrow(email, project);
-  const verified = stamp(await verifyRemoteMcp(url, auth));
-  return insertConnector(projectId, name, url, {
-    auth,
-    createdAt: new Date().toISOString(),
-    verified,
-    oauth: await oauthCapable(url, auth),
-  });
-}
-
-async function oauthCapable(url: URL, auth: ConnectorAuth): Promise<boolean> {
-  if (auth.kind === 'oauth') return true;
-  if (auth.kind === 'header') return false;
-  return advertisesOAuth(url);
-}
-
-async function freshAuth(
-  auth: ConnectorAuth,
-  resource: string,
-): Promise<ConnectorAuth> {
-  if (auth.kind !== 'oauth' || !oauthExpired(auth)) return auth;
-  return refreshOAuth(auth, resource);
-}
-
-export async function reconnectConnectorForEmail(
-  email: string,
-  id: string,
-  auth: OAuthAuth,
-): Promise<Connector> {
-  const row = await ownedConnectorOrThrow(email, id);
-  const config = readConfig(row.config);
-  const url = parseConnectorUrl(row.url);
-  const verified = stamp(await verifyRemoteMcp(url, auth));
-  return saveConfig(row, { ...config, auth, verified, oauth: true });
-}
-
-export async function disconnectConnectorForEmail(
-  email: string,
-  id: string,
-): Promise<Connector> {
-  const row = await ownedConnectorOrThrow(email, id);
-  const config = readConfig(row.config);
-  if (config.auth.kind !== 'oauth')
-    throw new ConnectorError('that connector is not signed in', 400);
-  return saveConfig(row, { ...config, auth: { kind: 'none' }, oauth: true });
-}
-
-export async function verifyConnectorForEmail(
-  email: string,
-  id: string,
-): Promise<ConnectorCheck> {
-  const row = await ownedConnectorOrThrow(email, id);
-  const config = readConfig(row.config);
-  try {
-    const url = parseConnectorUrl(row.url);
-    const auth = await freshAuth(config.auth, url.toString());
-    const verified = stamp(await verifyRemoteMcp(url, auth));
-    const oauth = config.oauth || (await oauthCapable(url, auth));
-    await saveConfig(row, { ...config, auth, verified, oauth });
-    return { id: row.id, name: row.name, ok: true, verified };
-  } catch (err) {
-    if (!(err instanceof ConnectorVerifyError)) throw err;
-    return { id: row.id, name: row.name, ok: false, reason: err.message };
-  }
-}
-
-export async function renameConnectorForEmail(
-  email: string,
-  id: string,
-  raw: string,
-): Promise<Connector> {
-  const name = connectorName(raw);
-  const row = await ownedConnectorOrThrow(email, id);
-  await assertRenameFreeOfClash(row.id, name);
-  const rows = await getDb()
-    .update(connectors)
-    .set({ name })
-    .where(and(eq(connectors.id, row.id), eq(connectors.projectId, row.projectId)))
-    .returning();
-  const saved = rows[0];
-  if (saved === undefined) throw missing();
-  return toConnector(saved);
-}
-
-export async function deleteConnectorForEmail(
-  email: string,
-  id: string,
-): Promise<DeletedConnector> {
-  const row = await ownedConnectorOrThrow(email, id);
-  const gone = await getDb()
-    .delete(connectors)
-    .where(eq(connectors.id, row.id))
-    .returning({ id: connectors.id, name: connectors.name });
-  const deleted = gone[0];
-  if (deleted === undefined) throw missing();
-  return deleted;
-}