203 lines · 8.7 KB · TypeScript
RawDownload
// ---------------------------------------------------------------------------// Cloud admin users API — the shared, provider-neutral `AdminUsersHandlers`// backed by a WorkOS-authorized platform view, mounted at `/api/admin/users*`.//// TWO credentials reach this plane, and they are the two an operator actually// has://   1. an ORG-SCOPED api key -> `PlatformAuth`. The key IS the authority: WorkOS//      validated it and reported which org owns it, and there is no member//      behind it to check membership for. This is the machine credential//      (a customer's backend calling us).//   2. an admin SESSION member -> the console. Requires a live `getUserOrgMembership`//      whose role slug is `admin` AND whose status is `active`, matching the//      strictest existing cloud guard (`auth/handlers.ts`'s org-delete check) —//      a pending admin invite is not an admin.// A plain member session, or a USER-scoped api key, is refused: both name one// acting member, and this plane deliberately serves the whole tenant.//// The executor is built by `makePlatformExecutor` — `{ tenant, subject:// undefined, platformView: true }` — so the reads are tenant-wide and read-only// by storage policy, and no `subject` row is minted for the caller.//// Cross-tenant isolation is structural, not a check in this file: the tenant is// taken from the resolved credential (the key's own org, or the session's// authorized org), never from client input, and `reach: "tenant"` still filters// every query by that tenant.// --------------------------------------------------------------------------- import { HttpRouter } from "effect/unstable/http";import { Effect, Layer } from "effect"; import {  AdminUsersProvider,  DbProvider,  HostConfig,  PluginsProvider,  listAdminUserConnections,  listAdminUsers,  listAdminUsersWithConnections,  makeAdminUsersApiLayer,  makePlatformExecutor,  platformViewOf,  requestScopedMiddleware,  type AdminUsersHeaders,} from "@executor-js/api/server";import { AdminUsersError, AdminUsersForbidden, AdminUsersUnauthorized } from "@executor-js/api";import type { Executor } from "@executor-js/sdk"; import { ApiKeyService } from "../auth/api-keys";import { UserStoreService } from "../auth/context";import { isPlatformAuth, resolveBearerAuth } from "../auth/workos-auth-provider";import { orgSelectorFromRequest, authorizeOrganizationSelector } from "../auth/organization";import { WorkOSClient } from "../auth/workos";import { DbService } from "../db/db";import { CloudExecutionSeamsLayer } from "../engine/execution-stack"; /** * Resolve the tenant this request may read, or fail with the neutral 401/403. * * Returns only the organization id: nothing downstream needs to know WHICH of * the two credentials got the caller here, and keeping the acting member out of * the return value means no admin read can accidentally become subject-scoped. */const authorizeTenant = (  request: Request,): Effect.Effect<  string,  AdminUsersUnauthorized | AdminUsersForbidden,  WorkOSClient | ApiKeyService | UserStoreService> =>  Effect.gen(function* () {    // (1) The bearer path. `resolveBearerAuth` (not `resolveApiKeyPrincipal`,    // which rejects org keys for the product plane) is what distinguishes an    // org key from a user key.    const bearer = yield* resolveBearerAuth(request).pipe(      // Every rejected-credential and infra failure collapses to one refusal:      // this plane must not report whether a key exists, belongs to another      // org, or merely lacks privilege.      Effect.catchCause(() => Effect.succeed(null)),    );    if (bearer !== null) {      if (isPlatformAuth(bearer)) return bearer.organizationId;      // A user-scoped key authenticated fine but names one member; the platform      // plane has no honest way to serve it.      return yield* new AdminUsersForbidden();    }     // (2) The session path: a live admin membership in the selected org.    const workos = yield* WorkOSClient;    const session = yield* workos      .authenticateRequest(request)      .pipe(Effect.catchCause(() => Effect.succeed(null)));    if (!session) return yield* new AdminUsersUnauthorized();     const selector = orgSelectorFromRequest(request) ?? session.organizationId;    if (!selector) return yield* new AdminUsersForbidden();    // Re-checks live membership, so the org selector header can only ever name    // an org the caller already belongs to.    const org = yield* authorizeOrganizationSelector(session.userId, selector).pipe(      Effect.catchCause(() => Effect.succeed(null)),    );    if (!org) return yield* new AdminUsersForbidden();     const membership = yield* workos      .getUserOrgMembership(org.id, session.userId)      .pipe(Effect.catchCause(() => Effect.succeed(null)));    // A pending admin invite is not an active admin — require both.    if (!membership || membership.status !== "active" || membership.role?.slug !== "admin") {      return yield* new AdminUsersForbidden();    }    return org.id;  }); /** * Authorize, then run `body` against the tenant's platform view. * * The executor is built per request and closed after, like every other cloud * execution stack: it holds the per-request postgres socket, which Cloudflare's * I/O isolation forbids sharing across requests. */const withPlatformView = <A>(  headers: AdminUsersHeaders,  body: (executor: Executor) => Effect.Effect<A, AdminUsersError>,): Effect.Effect<  A,  AdminUsersError | AdminUsersUnauthorized | AdminUsersForbidden,  WorkOSClient | ApiKeyService | UserStoreService | DbProvider | PluginsProvider | HostConfig> =>  Effect.gen(function* () {    const organizationId = yield* authorizeTenant(      new Request("https://admin.invalid", { headers }),    );    const executor = yield* makePlatformExecutor(organizationId).pipe(      Effect.mapError(() => new AdminUsersError({ message: "Failed to open the platform view" })),    );    return yield* Effect.ensuring(body(executor), executor.close().pipe(Effect.ignore));  }); /** * Cloud's `AdminUsersProvider`, built per request so the platform executor * closes over the per-request postgres socket. */export const workosAdminUsersProvider: Layer.Layer<  AdminUsersProvider,  never,  WorkOSClient | ApiKeyService | UserStoreService | DbProvider | PluginsProvider | HostConfig> = Layer.effect(AdminUsersProvider)(  Effect.gen(function* () {    const context = yield* Effect.context<      WorkOSClient | ApiKeyService | UserStoreService | DbProvider | PluginsProvider | HostConfig    >();    return AdminUsersProvider.of({      listUsers: (headers, options) =>        withPlatformView(headers, (executor) =>          platformViewOf(executor).pipe(Effect.flatMap((admin) => listAdminUsers(admin, options))),        ).pipe(Effect.provideContext(context)),      listUsersWithConnections: (headers, options) =>        withPlatformView(headers, (executor) =>          platformViewOf(executor).pipe(            Effect.flatMap((admin) => listAdminUsersWithConnections(admin, options)),          ),        ).pipe(Effect.provideContext(context)),      listUserConnections: (headers, externalId) =>        withPlatformView(headers, (executor) =>          platformViewOf(executor).pipe(            Effect.flatMap((admin) => listAdminUserConnections(admin, externalId)),          ),        ).pipe(Effect.provideContext(context)),    });  }),); // Builds the provider per request, providing it to the handlers. Long-lived// `WorkOSClient | ApiKeyService` come from the surrounding boot context; the// per-request `DbService`/`UserStoreService` (and the execution seams built// over them) are supplied by the combined `requestScopedMiddleware`.const AdminUsersProviderMiddleware = HttpRouter.middleware<{ provides: AdminUsersProvider }>()(  Effect.gen(function* () {    const longLived = yield* Effect.context<WorkOSClient | ApiKeyService>();    return (httpEffect) =>      Effect.gen(function* () {        // Built inside the request body so the execution seams close over the        // per-request postgres socket.        const provider = yield* Effect.provide(          AdminUsersProvider.asEffect(),          workosAdminUsersProvider.pipe(Layer.provide(CloudExecutionSeamsLayer)),        );        return yield* Effect.provideService(httpEffect, AdminUsersProvider, provider);      }).pipe(Effect.provideContext(longLived));  }),); /** * The cloud admin-users route layer, mounted as an app extension under the same * `/api` prefix as the rest of the cloud router. */export const makeCloudAdminUsersRoutes = (  rsLive: Layer.Layer<DbService | UserStoreService>,  options: Parameters<typeof makeAdminUsersApiLayer>[1] = {},) =>  makeAdminUsersApiLayer(    AdminUsersProviderMiddleware.combine(requestScopedMiddleware(rsLive)).layer,    options,  );