@inneranimalmedia/agentsam-sdk 2.6.4 → 2.6.7

package/packages/identity/src/adapters/iam-compat/index.js

@@ -0,0 +1,540 @@
+import { SESSION_POLICY, shouldRenewDesktopSession } from '../../core/constants.js';
+import { NATIVE_HANDOFF_TTL_SECONDS, SESSION_TYPES } from '../../core/session-policy.js';
+import { IdentitySchemaError } from '../../contracts/identity-store.js';
+import { IDENTITY_STORE_SCHEMA_VERSION } from '../../contracts/identity-store.js';
+import { newAccountIdentityId, newAuthUserId, newSessionId, newAuthEventId, nowUnix } from '../cloudflare-d1/ids.js';
+import { DEFAULT_COMPANY_ID, DEFAULT_COMPANY_SLUG, normalizeCompanyRow } from '../../contracts/company.js';
+
+
+/**
+ * @typedef {object} D1Database
+ * @property {(sql: string) => { bind: (...args: unknown[]) => { first: () => Promise<unknown>, run: () => Promise<unknown>, all: () => Promise<{ results?: unknown[] }> } }} prepare
+ */
+
+/**
+ * IAM compatibility IdentityStore — maps contract semantics onto hosted/IAM-shaped
+ * tables (accounts, auth_users, auth_sessions, identity_oauth_states, company, …).
+ * This is NOT the portable identity_* customer schema.
+ *
+ * @param {D1Database} db
+ * @param {{ sessionTtlSeconds?: number }} [options]
+ */
+async function hashValue(value) {
+  const v = String(value || '').trim();
+  if (!v) return null;
+  const bytes = new TextEncoder().encode(v);
+  const digest = await crypto.subtle.digest('SHA-256', bytes);
+  return Array.from(new Uint8Array(digest)).map((b) => b.toString(16).padStart(2, '0')).join('');
+}
+
+export function createIamCompatIdentityAdapter(db, options = {}) {
+  if (!db?.prepare) {
+    throw new Error('iam_compat_identity_adapter_requires_db_binding');
+  }
+  const sessionTtlSeconds = options.sessionTtlSeconds ?? SESSION_POLICY.browser.ttlSeconds;
+
+  return Object.freeze({
+    schemaVersion: IDENTITY_STORE_SCHEMA_VERSION,
+    sessionTtlSeconds,
+    backend: 'iam-compat',
+
+    /**
+     * Append-only login/logout/failed-attempt audit trail (auth_event_log).
+     * Never throws — an observability write must not break a real auth flow.
+     * status remains the event outcome only ('ok' | 'failed').
+     * Richer context (session, capabilities, activity, client) goes in metadata_json.
+     * ip/userAgent are hashed (SHA-256) — raw values are never persisted.
+     */
+    async logAuthEvent({
+      userId,
+      accountId,
+      eventType,
+      status = 'ok',
+      provider,
+      session,
+      capabilities,
+      activity,
+      client,
+      metadata,
+      request,
+    }) {
+      try {
+        const ip = request?.headers?.get?.('cf-connecting-ip') || request?.headers?.get?.('x-forwarded-for') || '';
+        const ua = request?.headers?.get?.('user-agent') || '';
+        const [ipHash, uaHash] = await Promise.all([hashValue(ip), hashValue(ua)]);
+        const envelope = {
+          ...(metadata && typeof metadata === 'object' ? metadata : {}),
+          ...(accountId ? { accountId } : {}),
+          ...(session ? { session } : {}),
+          ...(capabilities ? { capabilities } : {}),
+          ...(activity ? { activity } : {}),
+          ...(client ? { client } : {}),
+        };
+        await db.prepare(
+          `INSERT INTO auth_event_log
+           (id, user_id, event_type, status, provider, metadata_json, ip_hash, user_agent_hash)
+           VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
+        ).bind(
+          newAuthEventId(),
+          userId || null,
+          eventType,
+          status,
+          provider || null,
+          JSON.stringify(envelope),
+          ipHash,
+          uaHash,
+        ).run();
+      } catch {
+        // Never let audit logging break a real login/OAuth flow.
+      }
+    },
+
+    /**
+     * Account/session activity snapshot for auth receipts (not event outcome).
+     * @param {string} userId
+     */
+    async countAuthActivity(userId) {
+      if (!userId) {
+        return { loginCount: 0, activeSessionCount: 0, lastLoginAt: null };
+      }
+      try {
+        const logins = await db.prepare(
+          `SELECT COUNT(*) AS c, MAX(created_at) AS last_at
+           FROM auth_event_log
+           WHERE user_id = ? AND event_type = 'login' AND status = 'ok'`,
+        ).bind(userId).first();
+        const sessions = await db.prepare(
+          `SELECT COUNT(*) AS c FROM auth_sessions
+           WHERE user_id = ? AND revoked_at IS NULL AND expires_at > ?`,
+        ).bind(userId, nowUnix()).first();
+        return {
+          loginCount: Number(logins?.c || 0),
+          activeSessionCount: Number(sessions?.c || 0),
+          lastLoginAt: logins?.last_at != null ? Number(logins.last_at) : null,
+        };
+      } catch {
+        return { loginCount: 0, activeSessionCount: 0, lastLoginAt: null };
+      }
+    },
+
+    async findUserByEmail(email) {
+      const row = await db.prepare(
+        `SELECT id, email, display_name, password_hash, salt, status, created_at, updated_at
+         FROM auth_users WHERE email = ? COLLATE NOCASE LIMIT 1`,
+      ).bind(String(email || '').trim().toLowerCase()).first();
+      return row || null;
+    },
+
+    async findUserById(userId) {
+      const row = await db.prepare(
+        `SELECT id, email, display_name, password_hash, salt, status, created_at, updated_at
+         FROM auth_users WHERE id = ? LIMIT 1`,
+      ).bind(userId).first();
+      return row || null;
+    },
+
+    async createUser({ email, passwordHash, salt, displayName }) {
+      const id = newAuthUserId();
+      const ts = nowUnix();
+      const normalizedEmail = String(email || '').trim().toLowerCase();
+      // accounts = SSOT; auth_users = login principal (1:1 id in portable scaffold).
+      await db.prepare(
+        `INSERT INTO accounts (id, email, display_name, status, created_at, updated_at)
+         VALUES (?, ?, ?, 'active', ?, ?)`,
+      ).bind(id, normalizedEmail, displayName || null, ts, ts).run();
+      await db.prepare(
+        `INSERT INTO auth_users (id, email, display_name, password_hash, salt, status, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, 'active', ?, ?)`,
+      ).bind(id, normalizedEmail, displayName || null, passwordHash || null, salt || null, ts, ts).run();
+      return this.findUserById(id);
+    },
+
+    async updateUserPassword(userId, passwordHash, salt) {
+      const ts = nowUnix();
+      await db.prepare(
+        `UPDATE auth_users SET password_hash = ?, salt = ?, updated_at = ? WHERE id = ?`,
+      ).bind(passwordHash, salt, ts, userId).run();
+      await db.prepare(
+        `UPDATE accounts SET updated_at = ? WHERE id = ?`,
+      ).bind(ts, userId).run().catch(() => {});
+    },
+
+    async findUserByProvider(provider, providerSubject) {
+      const row = await db.prepare(
+        `SELECT u.id, u.email, u.display_name, u.password_hash, u.salt, u.status, u.created_at, u.updated_at
+         FROM account_identities ai
+         JOIN accounts a ON a.id = ai.account_id
+         JOIN auth_users u ON u.id = a.id
+         WHERE ai.provider = ? AND ai.provider_subject = ?
+           AND COALESCE(a.status, 'active') = 'active'
+         LIMIT 1`,
+      ).bind(provider, providerSubject).first();
+      return row || null;
+    },
+
+    async upsertProviderIdentity({ accountId, provider, providerSubject, email }) {
+      // Ensure SSOT account row exists (OAuth may create identity before password user).
+      const ts = nowUnix();
+      const account = await db.prepare(
+        `SELECT id FROM accounts WHERE id = ? LIMIT 1`,
+      ).bind(accountId).first();
+      if (!account?.id) {
+        const authUser = await this.findUserById(accountId);
+        if (!authUser) throw new Error('account_ssot_missing');
+        await db.prepare(
+          `INSERT INTO accounts (id, email, display_name, status, created_at, updated_at)
+           VALUES (?, ?, ?, 'active', ?, ?)`,
+        ).bind(
+          accountId,
+          authUser.email,
+          authUser.display_name || null,
+          authUser.created_at || ts,
+          ts,
+        ).run();
+      }
+
+      const existing = await db.prepare(
+        `SELECT id FROM account_identities WHERE provider = ? AND provider_subject = ? LIMIT 1`,
+      ).bind(provider, providerSubject).first();
+      if (existing?.id) {
+        await db.prepare(
+          `UPDATE account_identities SET account_id = ?, email = ?, updated_at = ? WHERE id = ?`,
+        ).bind(accountId, email || null, ts, existing.id).run();
+        return existing.id;
+      }
+      const id = newAccountIdentityId();
+      await db.prepare(
+        `INSERT INTO account_identities (id, account_id, provider, provider_subject, email, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?)`,
+      ).bind(id, accountId, provider, providerSubject, email || null, ts, ts).run();
+      return id;
+    },
+
+    async createSession({ userId, email, provider, providerSubject, displayName, type }) {
+      const id = newSessionId();
+      const ts = nowUnix();
+      const desktop = type === SESSION_TYPES.DESKTOP;
+      const expiresAt = ts + (desktop ? SESSION_POLICY.desktop.ttlSeconds : sessionTtlSeconds);
+      // Browser inserts stay column-identical to before (auth_sessions.type defaults
+      // to 'browser'), so this is safe on databases that predate the type column.
+      const columns = ['id', 'user_id', 'email', 'provider', 'provider_subject', 'display_name', 'expires_at', 'created_at', 'last_active_at'];
+      const values = [id, userId, email || null, provider || 'email', providerSubject || null, displayName || null, expiresAt, ts, ts];
+      if (desktop) {
+        columns.push('type');
+        values.push(SESSION_TYPES.DESKTOP);
+      }
+      await db.prepare(
+        `INSERT INTO auth_sessions (${columns.join(', ')})
+         VALUES (${columns.map(() => '?').join(', ')})`,
+      ).bind(...values).run();
+      return {
+        id,
+        type: desktop ? SESSION_TYPES.DESKTOP : SESSION_TYPES.BROWSER,
+        user_id: userId,
+        email,
+        provider: provider || 'email',
+        provider_subject: providerSubject || null,
+        display_name: displayName || null,
+        expires_at: expiresAt,
+        created_at: ts,
+      };
+    },
+
+    async getSession(sessionId) {
+      const row = await db.prepare(
+        `SELECT id, user_id, email, provider, provider_subject, display_name, expires_at, revoked_at, created_at, last_active_at, type
+         FROM auth_sessions WHERE id = ? LIMIT 1`,
+      ).bind(sessionId).first();
+      if (!row) return null;
+      if (row.revoked_at) return null;
+      const now = nowUnix();
+      if (row.expires_at <= now) return null;
+      if (shouldRenewDesktopSession(row, now)) {
+        const renewedExpiresAt = now + SESSION_POLICY.desktop.ttlSeconds;
+        await db.prepare(
+          `UPDATE auth_sessions
+           SET expires_at = ?, last_active_at = ?
+           WHERE id = ? AND revoked_at IS NULL AND expires_at > ?`,
+        ).bind(renewedExpiresAt, now, sessionId, now).run();
+        row.expires_at = renewedExpiresAt;
+        row.last_active_at = now;
+      }
+      return row;
+    },
+
+    async revokeSession(sessionId, reason = 'logout') {
+      const ts = nowUnix();
+      await db.prepare(
+        `UPDATE auth_sessions SET revoked_at = ?, last_active_at = ? WHERE id = ? AND revoked_at IS NULL`,
+      ).bind(ts, ts, sessionId).run();
+      return { ok: true, reason };
+    },
+
+    async createOAuthTransaction({ state, provider, codeVerifier, returnTo, redirectTo, appId, ttlSeconds = 600, clientType, nativeChallenge, nativeRedirect }) {
+      if (!appId) {
+        throw new IdentitySchemaError(
+          'OAUTH_TRANSACTION_APP_ID_REQUIRED',
+          'OAuth transactions must include app_id',
+        );
+      }
+      const ts = nowUnix();
+      const expiresAt = ts + ttlSeconds;
+      try {
+        const columns = ['state', 'provider', 'code_verifier', 'redirect_to', 'app_id', 'expires_at', 'created_at'];
+        const values = [state, provider, codeVerifier, returnTo ?? redirectTo ?? null, appId, expiresAt, ts];
+        // Native columns (migration 0017) are written only for native transactions,
+        // so browser OAuth is unaffected until/unless 0017 is applied.
+        if (clientType === SESSION_TYPES.DESKTOP) {
+          columns.push('client_type', 'native_challenge', 'native_redirect');
+          values.push(clientType, nativeChallenge ?? null, nativeRedirect ?? null);
+        }
+        await db.prepare(
+          `INSERT INTO identity_oauth_states (${columns.join(', ')})
+           VALUES (${columns.map(() => '?').join(', ')})`,
+        ).bind(...values).run();
+      } catch (err) {
+        throw new IdentitySchemaError(
+          'IDENTITY_SCHEMA_MIGRATION_REQUIRED',
+          'identity_oauth_states.app_id required — apply migration 0015 (no silent pre-app_id fallback)',
+          { cause: String(err?.message || err) },
+        );
+      }
+    },
+
+    /** @deprecated Use createOAuthTransaction */
+    async saveOAuthState(input) {
+      return this.createOAuthTransaction(input);
+    },
+
+    async consumeOAuthTransaction(state) {
+      let row;
+      try {
+        row = await db.prepare(
+          `SELECT * FROM identity_oauth_states WHERE state = ? LIMIT 1`,
+        ).bind(state).first();
+      } catch (err) {
+        throw new IdentitySchemaError(
+          'IDENTITY_SCHEMA_MIGRATION_REQUIRED',
+          'identity_oauth_states.app_id column missing — apply migration 0015',
+          { cause: String(err?.message || err) },
+        );
+      }
+      if (!row) return null;
+      await db.prepare(`DELETE FROM identity_oauth_states WHERE state = ?`).bind(state).run();
+      if (row.expires_at <= nowUnix()) return null;
+      if (!row.app_id) {
+        throw new IdentitySchemaError(
+          'IDENTITY_SCHEMA_MIGRATION_REQUIRED',
+          'OAuth transaction missing app_id — refuse silent pre-app_id semantics',
+        );
+      }
+      return {
+        state: row.state,
+        provider: row.provider,
+        code_verifier: row.code_verifier,
+        redirect_to: row.redirect_to,
+        return_to: row.redirect_to,
+        app_id: row.app_id,
+        client_type: row.client_type ?? null,
+        native_challenge: row.native_challenge ?? null,
+        native_redirect: row.native_redirect ?? null,
+        expires_at: row.expires_at,
+        created_at: row.created_at,
+      };
+    },
+
+    /** @deprecated Use consumeOAuthTransaction */
+    async consumeOAuthState(state) {
+      return this.consumeOAuthTransaction(state);
+    },
+
+    /** Single-use desktop session pickup (migration 0017). Only the code's sha256 is stored. */
+    async createNativeHandoff({ handoffHash, sessionId, challenge, ttlSeconds = NATIVE_HANDOFF_TTL_SECONDS }) {
+      const ts = nowUnix();
+      await db.prepare(
+        `INSERT INTO identity_native_handoffs (handoff_hash, session_id, challenge, expires_at, created_at)
+         VALUES (?, ?, ?, ?, ?)`,
+      ).bind(handoffHash, sessionId, challenge, ts + ttlSeconds, ts).run();
+    },
+
+    /** Atomically claim an unexpired, unconsumed handoff; null if none. */
+    async consumeNativeHandoff(handoffHash) {
+      const ts = nowUnix();
+      const row = await db.prepare(
+        `UPDATE identity_native_handoffs SET consumed_at = ?
+         WHERE handoff_hash = ? AND consumed_at IS NULL AND expires_at > ?
+         RETURNING session_id, challenge`,
+      ).bind(ts, handoffHash, ts).first();
+      return row || null;
+    },
+
+    async getCompanyBySlug(slug = DEFAULT_COMPANY_SLUG) {
+      const row = await db.prepare(
+        `SELECT id, slug, name, legal_name, logo_url, favicon_url, primary_color, auth_bg_color,
+                support_email, website_url, tagline, meta_json, created_at, updated_at
+         FROM company WHERE slug = ? LIMIT 1`,
+      ).bind(slug).first();
+      return normalizeCompanyRow(row);
+    },
+
+    /**
+     * Resolve portal branding by request host against company.meta.hosts (or website_url host).
+     * Product-agnostic — each tenant lists their own hosts in D1.
+     * @param {string | null | undefined} hostname
+     */
+    async getCompanyByHost(hostname) {
+      const host = String(hostname || '')
+        .trim()
+        .toLowerCase()
+        .split(':')[0];
+      if (!host) return null;
+      const { results } = await db.prepare(
+        `SELECT id, slug, name, legal_name, logo_url, favicon_url, primary_color, auth_bg_color,
+                support_email, website_url, tagline, meta_json, created_at, updated_at
+         FROM company`,
+      ).all();
+      for (const row of results || []) {
+        const company = normalizeCompanyRow(row);
+        if (!company) continue;
+        const hosts = Array.isArray(company.meta?.hosts)
+          ? company.meta.hosts.map((h) => String(h || '').trim().toLowerCase()).filter(Boolean)
+          : [];
+        if (hosts.includes(host)) return company;
+        try {
+          if (company.websiteUrl) {
+            const siteHost = new URL(company.websiteUrl).hostname.toLowerCase();
+            if (siteHost && siteHost === host) return company;
+          }
+        } catch {
+          // ignore invalid website_url
+        }
+      }
+      return null;
+    },
+
+    async getDefaultCompany() {
+      const bySlug = await this.getCompanyBySlug(DEFAULT_COMPANY_SLUG);
+      if (bySlug) return bySlug;
+      const row = await db.prepare(
+        `SELECT id, slug, name, legal_name, logo_url, favicon_url, primary_color, auth_bg_color,
+                support_email, website_url, tagline, meta_json, created_at, updated_at
+         FROM company WHERE id = ? LIMIT 1`,
+      ).bind(DEFAULT_COMPANY_ID).first();
+      return normalizeCompanyRow(row);
+    },
+
+    async upsertCompany(input) {
+      const ts = nowUnix();
+      const id = input.id || DEFAULT_COMPANY_ID;
+      const slug = input.slug || DEFAULT_COMPANY_SLUG;
+      const existing = await db.prepare(`SELECT id FROM company WHERE id = ? OR slug = ? LIMIT 1`)
+        .bind(id, slug).first();
+      const metaJson = input.meta ? JSON.stringify(input.meta) : (input.metaJson ?? null);
+      if (existing?.id) {
+        await db.prepare(
+          `UPDATE company SET
+            slug = ?, name = ?, legal_name = ?, logo_url = ?, favicon_url = ?,
+            primary_color = ?, auth_bg_color = ?, support_email = ?, website_url = ?,
+            tagline = ?, meta_json = ?, updated_at = ?
+           WHERE id = ?`,
+        ).bind(
+          slug,
+          input.name,
+          input.legalName ?? null,
+          input.logoUrl ?? null,
+          input.faviconUrl ?? null,
+          input.primaryColor ?? null,
+          input.authBgColor ?? null,
+          input.supportEmail ?? null,
+          input.websiteUrl ?? null,
+          input.tagline ?? null,
+          metaJson,
+          ts,
+          existing.id,
+        ).run();
+        return this.getCompanyBySlug(slug);
+      }
+      await db.prepare(
+        `INSERT INTO company
+         (id, slug, name, legal_name, logo_url, favicon_url, primary_color, auth_bg_color,
+          support_email, website_url, tagline, meta_json, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+      ).bind(
+        id,
+        slug,
+        input.name,
+        input.legalName ?? null,
+        input.logoUrl ?? null,
+        input.faviconUrl ?? null,
+        input.primaryColor ?? null,
+        input.authBgColor ?? null,
+        input.supportEmail ?? null,
+        input.websiteUrl ?? null,
+        input.tagline ?? null,
+        metaJson,
+        ts,
+        ts,
+      ).run();
+      return this.getCompanyBySlug(slug);
+    },
+
+    /**
+     * Provider connection / config-ref. Fixture or hosted DB must provide
+     * identity_provider_connections (credential_ref only — never raw secrets).
+     */
+    async upsertProviderConnection({
+      userId, provider, credentialRef, grantedScopes = [], expiresAt = null, refreshable = false,
+    }) {
+      if (!credentialRef) {
+        throw new IdentitySchemaError(
+          'PROVIDER_CREDENTIAL_REF_REQUIRED',
+          'Provider connections store credential_ref only — never raw secrets',
+        );
+      }
+      const existing = await db.prepare(
+        `SELECT id FROM identity_provider_connections WHERE user_id = ? AND provider = ? LIMIT 1`,
+      ).bind(userId, provider).first();
+      const ts = nowUnix();
+      const scopesJson = JSON.stringify(Array.isArray(grantedScopes) ? grantedScopes : []);
+      if (existing?.id) {
+        await db.prepare(
+          `UPDATE identity_provider_connections
+           SET granted_scopes_json = ?, credential_ref = ?, expires_at = ?, refreshable = ?, updated_at = ?
+           WHERE id = ?`,
+        ).bind(scopesJson, credentialRef, expiresAt, refreshable ? 1 : 0, ts, existing.id).run();
+        return existing.id;
+      }
+      const id = `ipc_${crypto.randomUUID().replace(/-/g, '')}`;
+      await db.prepare(
+        `INSERT INTO identity_provider_connections
+         (id, user_id, provider, granted_scopes_json, credential_ref, expires_at, refreshable, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+      ).bind(id, userId, provider, scopesJson, credentialRef, expiresAt, refreshable ? 1 : 0, ts, ts).run();
+      return id;
+    },
+
+    async getProviderConnection(userId, provider) {
+      const row = await db.prepare(
+        `SELECT id, user_id, provider, granted_scopes_json, credential_ref, expires_at, refreshable, created_at, updated_at
+         FROM identity_provider_connections WHERE user_id = ? AND provider = ? LIMIT 1`,
+      ).bind(userId, provider).first();
+      if (!row) return null;
+      let grantedScopes = [];
+      try { grantedScopes = JSON.parse(row.granted_scopes_json || '[]'); } catch { grantedScopes = []; }
+      return {
+        id: row.id,
+        user_id: row.user_id,
+        provider: row.provider,
+        granted_scopes: grantedScopes,
+        credential_ref: row.credential_ref,
+        expires_at: row.expires_at,
+        refreshable: Boolean(row.refreshable),
+        created_at: row.created_at,
+        updated_at: row.updated_at,
+      };
+    },
+  });
+}
+
+/** @deprecated Prefer createIamCompatIdentityAdapter — stable alias for existing call sites. */
+export const createCloudflareD1Adapter = createIamCompatIdentityAdapter;

package/packages/identity/src/adapters/portable-d1/index.js

@@ -0,0 +1,18 @@
+/**
+ * Portable D1 IdentityStore — SAME portable identity_* schema as SQLite.
+ * Clean customer D1 (user123). Does NOT query accounts/auth_users/company/etc.
+ *
+ * Implementation shares prepare/bind CRUD with SQLite only because both targets
+ * already speak the same D1-shaped statement API — not because D1 was forced
+ * through a Node sqlite emulator.
+ */
+import { createPortableIdentityStore } from '../portable-store.js';
+
+/**
+ * @param {{ prepare: Function }} db Cloudflare D1 binding or D1-shaped fixture
+ * @param {{ sessionTtlSeconds?: number, skipSchemaCheck?: boolean }} [options]
+ */
+export function createPortableD1IdentityAdapter(db, options = {}) {
+  if (!db?.prepare) throw new Error('portable_d1_identity_adapter_requires_db');
+  return createPortableIdentityStore(db, { ...options, backend: 'portable-d1' });
+}