feat: expose core identity transport

core/tests/identity-database.test.ts

@@ -37,7 +37,7 @@ describe("Core identity Drizzle database composition", () => {
 
   it("explicitly constructs the pool only when called, with the identity schema attached", () => {
     poolConstructor.mockImplementation(function fakePool() {
-      return { end: async () => undefined };
+      return { end: async () => undefined, on: vi.fn() };
     });
 
     const handle = createIdentityDrizzleDatabase(
@@ -52,7 +52,7 @@ describe("Core identity Drizzle database composition", () => {
 
   it("passes the connection timeout through to the pool configuration", () => {
     poolConstructor.mockImplementation(function fakePool() {
-      return { end: async () => undefined };
+      return { end: async () => undefined, on: vi.fn() };
     });
     poolConstructor.mockClear();
 
@@ -64,4 +64,21 @@ describe("Core identity Drizzle database composition", () => {
       expect.objectContaining({ connectionTimeoutMillis: 5_000 }),
     );
   });
+
+  it("registers a pool-level error listener so an idle client failure never becomes an uncaught exception", () => {
+    const on = vi.fn();
+    poolConstructor.mockImplementation(function fakePool() {
+      return { end: async () => undefined, on };
+    });
+
+    createIdentityDrizzleDatabase("postgresql://example.test/ekklesia");
+
+    expect(on).toHaveBeenCalledWith("error", expect.any(Function));
+    const handler = on.mock.calls.find((call) => call[0] === "error")?.[1] as
+      | ((error: unknown) => void)
+      | undefined;
+    expect(() =>
+      handler?.(new Error("terminating connection due to administrator command")),
+    ).not.toThrow();
+  });
 });

core/tests/identity-factory.test.ts

@@ -1,123 +0,0 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
-
-const betterAuthMock = vi.hoisted(() => vi.fn());
-const drizzleAdapterMock = vi.hoisted(() =>
-  vi.fn((db: unknown, options: unknown) => ({ __adapterOptions: options, db })),
-);
-
-vi.mock("better-auth", () => ({ betterAuth: betterAuthMock }));
-vi.mock("better-auth/adapters/drizzle", () => ({ drizzleAdapter: drizzleAdapterMock }));
-
-import type { IdentityConfig } from "../src/identity/config.js";
-import { createIdentityAuth, IdentityConstructionError } from "../src/identity/factory.js";
-import { identityAdapterSchema } from "../src/identity/schema.js";
-
-function validConfig(): IdentityConfig {
-  return Object.freeze({
-    baseUrl: "https://identity.example.test",
-    secret: "correct-horse-battery-staple-0123456789",
-    trustedOrigins: Object.freeze(["https://identity.example.test"]),
-  });
-}
-
-describe("Core identity Better Auth construction boundary", () => {
-  beforeEach(() => {
-    betterAuthMock.mockReset();
-    drizzleAdapterMock.mockClear();
-  });
-
-  it("constructs explicitly with narrow injected dependencies and returns only a success marker", () => {
-    betterAuthMock.mockReturnValue({ api: {}, handler: () => undefined });
-    const fakeDb = { __fakeDb: true };
-
-    const result = createIdentityAuth({ config: validConfig(), db: fakeDb as never });
-
-    expect(result).toEqual({ constructed: true });
-    expect(Object.keys(result)).toEqual(["constructed"]);
-    expect(Object.isFrozen(result)).toBe(true);
-  });
-
-  it("does not expose the raw database, adapter, or Better Auth instance on the returned value", () => {
-    betterAuthMock.mockReturnValue({ api: {}, handler: () => undefined });
-    const fakeDb = { __fakeDb: true };
-
-    const result = createIdentityAuth({ config: validConfig(), db: fakeDb as never });
-
-    expect(JSON.stringify(result)).not.toContain("__fakeDb");
-    expect(JSON.stringify(result)).not.toContain("handler");
-  });
-
-  it("passes the explicit ekklesia_core_* schema mapping to the Drizzle adapter", () => {
-    betterAuthMock.mockReturnValue({});
-    const fakeDb = { __fakeDb: true };
-
-    createIdentityAuth({ config: validConfig(), db: fakeDb as never });
-
-    expect(drizzleAdapterMock).toHaveBeenCalledTimes(1);
-    const [db, options] = drizzleAdapterMock.mock.calls[0] as [unknown, unknown];
-    expect(db).toBe(fakeDb);
-    expect(options).toEqual({ provider: "pg", schema: identityAdapterSchema });
-  });
-
-  it("enables no provider or plugin beyond the four core models", () => {
-    betterAuthMock.mockReturnValue({});
-
-    createIdentityAuth({ config: validConfig(), db: {} as never });
-
-    expect(betterAuthMock).toHaveBeenCalledTimes(1);
-    const [options] = betterAuthMock.mock.calls[0] as [Record<string, unknown>];
-    expect(Object.keys(options).sort()).toEqual([
-      "baseURL",
-      "database",
-      "secret",
-      "trustedOrigins",
-    ]);
-    expect(options.baseURL).toBe(validConfig().baseUrl);
-    expect(options.secret).toBe(validConfig().secret);
-    expect(options.trustedOrigins).toEqual([...validConfig().trustedOrigins]);
-  });
-
-  it("sanitizes a construction failure without leaking the underlying error, secret, or URL", () => {
-    const hostileSecret = validConfig().secret;
-    betterAuthMock.mockImplementation(() => {
-      throw new Error(
-        `Better Auth failed for secret ${hostileSecret} at https://internal.example.test`,
-      );
-    });
-
-    let captured: unknown;
-    try {
-      createIdentityAuth({ config: validConfig(), db: {} as never });
-    } catch (error) {
-      captured = error;
-    }
-
-    expect(captured).toBeInstanceOf(IdentityConstructionError);
-    expect(captured).toMatchObject({ code: "IDENTITY_CONSTRUCTION_FAILED" });
-    const publicText = `${String(captured)}\n${
-      captured instanceof Error ? (captured.stack ?? "") : ""
-    }`;
-    expect(publicText).not.toContain(hostileSecret);
-    expect(publicText).not.toContain("internal.example.test");
-  });
-
-  it("sanitizes an adapter construction failure the same way", () => {
-    drizzleAdapterMock.mockImplementation(() => {
-      throw new Error("adapter exploded with postgresql://admin:[email protected]/ekklesia");
-    });
-
-    let captured: unknown;
-    try {
-      createIdentityAuth({ config: validConfig(), db: {} as never });
-    } catch (error) {
-      captured = error;
-    }
-
-    expect(captured).toBeInstanceOf(IdentityConstructionError);
-    const publicText = `${String(captured)}\n${
-      captured instanceof Error ? (captured.stack ?? "") : ""
-    }`;
-    expect(publicText).not.toContain("db.internal");
-    expect(publicText).not.toContain("admin");
-  });
-});

core/tests/identity-runtime.test.ts

@@ -0,0 +1,288 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const betterAuthMock = vi.hoisted(() => vi.fn());
+const drizzleAdapterMock = vi.hoisted(() =>
+  vi.fn((db: unknown, options: unknown) => ({ __adapterOptions: options, db })),
+);
+
+vi.mock("better-auth", () => ({ betterAuth: betterAuthMock }));
+vi.mock("better-auth/adapters/drizzle", () => ({ drizzleAdapter: drizzleAdapterMock }));
+
+import type { IdentityConfig } from "../src/identity/config.js";
+import { createIdentityRuntime, IdentityRuntimeError } from "../src/identity/runtime.js";
+import { identityAdapterSchema } from "../src/identity/schema.js";
+
+function validConfig(): IdentityConfig {
+  return Object.freeze({
+    baseUrl: "https://identity.example.test",
+    secret: "correct-horse-battery-staple-0123456789",
+    trustedOrigins: Object.freeze(["https://identity.example.test"]),
+  });
+}
+
+function fakeDb(overrides: Partial<{ execute: ReturnType<typeof vi.fn> }> = {}) {
+  return { __fakeDb: true, execute: vi.fn().mockResolvedValue(undefined), ...overrides } as never;
+}
+
+describe("Core identity runtime construction boundary", () => {
+  beforeEach(() => {
+    betterAuthMock.mockReset();
+    drizzleAdapterMock.mockClear();
+  });
+
+  it("does not construct Better Auth or the Drizzle adapter merely because the module is imported", () => {
+    expect(betterAuthMock).not.toHaveBeenCalled();
+    expect(drizzleAdapterMock).not.toHaveBeenCalled();
+  });
+
+  it("constructs explicitly and returns only the two narrow transport capabilities", () => {
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+
+    expect(Object.keys(runtime).sort()).toEqual(["handleAuthRequest", "resolvePrincipal"]);
+    expect(typeof runtime.handleAuthRequest).toBe("function");
+    expect(typeof runtime.resolvePrincipal).toBe("function");
+    expect(Object.isFrozen(runtime)).toBe(true);
+  });
+
+  it("does not expose the raw database, adapter, or Better Auth instance on the returned value", () => {
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler: vi.fn() });
+    const db = fakeDb();
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db });
+
+    expect(JSON.stringify(runtime)).toBe("{}");
+  });
+
+  it("passes the explicit ekklesia_core_* schema mapping to the Drizzle adapter", () => {
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler: vi.fn() });
+    const db = fakeDb();
+
+    createIdentityRuntime({ config: validConfig(), db });
+
+    expect(drizzleAdapterMock).toHaveBeenCalledTimes(1);
+    const [passedDb, options] = drizzleAdapterMock.mock.calls[0] as [unknown, unknown];
+    expect(passedDb).toBe(db);
+    expect(options).toEqual({ provider: "pg", schema: identityAdapterSchema });
+  });
+
+  it("enables exactly email/password (autoSignIn disabled) and rate limiting, nothing else", () => {
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler: vi.fn() });
+
+    createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+
+    expect(betterAuthMock).toHaveBeenCalledTimes(1);
+    const [options] = betterAuthMock.mock.calls[0] as [Record<string, unknown>];
+    expect(Object.keys(options).sort()).toEqual([
+      "baseURL",
+      "database",
+      "emailAndPassword",
+      "rateLimit",
+      "secret",
+      "trustedOrigins",
+    ]);
+    expect(options.emailAndPassword).toEqual({
+      autoSignIn: false,
+      disableSignUp: false,
+      enabled: true,
+    });
+    expect(options.rateLimit).toEqual({ enabled: true, max: 100, window: 10 });
+    expect(options.baseURL).toBe(validConfig().baseUrl);
+    expect(options.secret).toBe(validConfig().secret);
+    expect(options.trustedOrigins).toEqual([...validConfig().trustedOrigins]);
+  });
+
+  it("only widens the sign-in/sign-up rate-limit window when a narrow override is explicitly injected", () => {
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler: vi.fn() });
+
+    createIdentityRuntime({
+      config: validConfig(),
+      db: fakeDb(),
+      rateLimitOverride: { max: 1000, window: 1 },
+    });
+
+    const [options] = betterAuthMock.mock.calls[0] as [Record<string, unknown>];
+    expect(options.rateLimit).toEqual({
+      customRules: {
+        "/sign-in/email": { max: 1000, window: 1 },
+        "/sign-up/email": { max: 1000, window: 1 },
+      },
+      enabled: true,
+      max: 100,
+      window: 10,
+    });
+  });
+
+  it("sanitizes a construction failure without leaking the underlying error, secret, or URL", () => {
+    const hostileSecret = validConfig().secret;
+    betterAuthMock.mockImplementation(() => {
+      throw new Error(
+        `Better Auth failed for secret ${hostileSecret} at https://internal.example.test`,
+      );
+    });
+
+    let captured: unknown;
+    try {
+      createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+    } catch (error) {
+      captured = error;
+    }
+
+    expect(captured).toBeInstanceOf(IdentityRuntimeError);
+    expect(captured).toMatchObject({ code: "IDENTITY_CONSTRUCTION_FAILED" });
+    const publicText = `${String(captured)}\n${
+      captured instanceof Error ? (captured.stack ?? "") : ""
+    }`;
+    expect(publicText).not.toContain(hostileSecret);
+    expect(publicText).not.toContain("internal.example.test");
+  });
+});
+
+describe("Core identity runtime — handleAuthRequest", () => {
+  beforeEach(() => {
+    betterAuthMock.mockReset();
+    drizzleAdapterMock.mockClear();
+  });
+
+  it("delegates directly to Better Auth's Fetch handler and passes the response through unmodified", async () => {
+    const expectedResponse = new Response("ok", { status: 200 });
+    const handler = vi.fn().mockResolvedValue(expectedResponse);
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+    const request = new Request("https://identity.example.test/api/auth/sign-in/email");
+    const response = await runtime.handleAuthRequest(request);
+
+    expect(handler).toHaveBeenCalledWith(request);
+    expect(response).toBe(expectedResponse);
+  });
+
+  it("passes through Better Auth's own error responses (e.g. 401, 403, 422) unmodified", async () => {
+    const errorResponse = new Response(JSON.stringify({ code: "INVALID_EMAIL_OR_PASSWORD" }), {
+      status: 401,
+    });
+    const handler = vi.fn().mockResolvedValue(errorResponse);
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+    const response = await runtime.handleAuthRequest(
+      new Request("https://identity.example.test/x"),
+    );
+
+    expect(response.status).toBe(401);
+  });
+
+  it("sanitizes a genuinely unexpected handler failure", async () => {
+    const handler = vi.fn().mockRejectedValue(new Error("postgresql://admin:[email protected]/x"));
+    betterAuthMock.mockReturnValue({ api: { getSession: vi.fn() }, handler });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+
+    let captured: unknown;
+    try {
+      await runtime.handleAuthRequest(new Request("https://identity.example.test/x"));
+    } catch (error) {
+      captured = error;
+    }
+
+    expect(captured).toBeInstanceOf(IdentityRuntimeError);
+    expect(captured).toMatchObject({ code: "AUTH_REQUEST_FAILED" });
+    const publicText = `${String(captured)}\n${
+      captured instanceof Error ? (captured.stack ?? "") : ""
+    }`;
+    expect(publicText).not.toContain("db.internal");
+    expect(publicText).not.toContain("admin");
+  });
+});
+
+describe("Core identity runtime — resolvePrincipal", () => {
+  beforeEach(() => {
+    betterAuthMock.mockReset();
+    drizzleAdapterMock.mockClear();
+  });
+
+  it("maps a valid session to an authenticated principal containing only the opaque user id", async () => {
+    const getSession = vi.fn().mockResolvedValue({
+      session: { id: "sess_1", token: "tok" },
+      user: { email: "[email protected]", id: "user_123", name: "Person" },
+    });
+    betterAuthMock.mockReturnValue({ api: { getSession }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+    const principal = await runtime.resolvePrincipal(
+      new Headers({ cookie: "better-auth.session_token=tok" }),
+    );
+
+    expect(principal).toEqual({ kind: "authenticated", userId: "user_123" });
+    expect(Object.keys(principal)).toEqual(["kind", "userId"]);
+    expect(Object.isFrozen(principal)).toBe(true);
+  });
+
+  it("maps a missing session with no cookie header to anonymous without probing the database", async () => {
+    const getSession = vi.fn().mockResolvedValue(null);
+    const execute = vi.fn().mockResolvedValue(undefined);
+    betterAuthMock.mockReturnValue({ api: { getSession }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb({ execute }) });
+    const principal = await runtime.resolvePrincipal(new Headers());
+
+    expect(principal).toEqual({ kind: "anonymous" });
+    expect(execute).not.toHaveBeenCalled();
+  });
+
+  it("maps an expired/revoked/malformed session (cookie present, null session) to anonymous when the database is reachable", async () => {
+    const getSession = vi.fn().mockResolvedValue(null);
+    const execute = vi.fn().mockResolvedValue(undefined);
+    betterAuthMock.mockReturnValue({ api: { getSession }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb({ execute }) });
+    const principal = await runtime.resolvePrincipal(
+      new Headers({ cookie: "better-auth.session_token=stale-or-garbage" }),
+    );
+
+    expect(principal).toEqual({ kind: "anonymous" });
+    expect(execute).toHaveBeenCalledTimes(1);
+  });
+
+  it("never downgrades an infrastructure failure to anonymous: throws when the database is unreachable", async () => {
+    const getSession = vi.fn().mockResolvedValue(null);
+    const execute = vi.fn().mockRejectedValue(new Error("connection terminated unexpectedly"));
+    betterAuthMock.mockReturnValue({ api: { getSession }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb({ execute }) });
+
+    let captured: unknown;
+    try {
+      await runtime.resolvePrincipal(new Headers({ cookie: "better-auth.session_token=x" }));
+    } catch (error) {
+      captured = error;
+    }
+
+    expect(captured).toBeInstanceOf(IdentityRuntimeError);
+    expect(captured).toMatchObject({ code: "PRINCIPAL_RESOLUTION_FAILED" });
+  });
+
+  it("treats an unexpected throw from getSession itself as a sanitized infrastructure failure", async () => {
+    const hostile = "password=hunter2 host=db.internal.example port=5432";
+    const getSession = vi.fn().mockRejectedValue(new Error(hostile));
+    betterAuthMock.mockReturnValue({ api: { getSession }, handler: vi.fn() });
+
+    const runtime = createIdentityRuntime({ config: validConfig(), db: fakeDb() });
+
+    let captured: unknown;
+    try {
+      await runtime.resolvePrincipal(new Headers({ cookie: "better-auth.session_token=x" }));
+    } catch (error) {
+      captured = error;
+    }
+
+    expect(captured).toBeInstanceOf(IdentityRuntimeError);
+    expect(captured).toMatchObject({ code: "PRINCIPAL_RESOLUTION_FAILED" });
+    const publicText = `${String(captured)}\n${
+      captured instanceof Error ? (captured.stack ?? "") : ""
+    }`;
+    expect(publicText).not.toContain("hunter2");
+    expect(publicText).not.toContain("db.internal.example");
+  });
+});