234 lines · 10.2 KB · TypeScript
RawDownload
// ---------------------------------------------------------------------------// Shared executor-API ExecutionStackMiddleware.//// Cloud and self-host had a structurally identical `HttpRouter` middleware that,// per request://   1. reads the inbound `HttpServerRequest`, converts it to a web `Request`,//   2. resolves identity (api-key/session for cloud, cookie/bearer/x-api-key for//      self-host) into a neutral `Principal`,//   3. builds the per-(user, org) executor + engine via `makeExecutionStack`,//   4. provides `AuthContext` + the execution-stack services + every plugin//      extension Service to the wrapped handler.//// This factory owns that common body. The differences are injected://   - `authenticate`     — the provider's resolve fn. BOTH apps yield the neutral//                          `Principal` and fail the SHARED `Unauthorized |//                          NoOrganization | Unavailable` (cloud: WorkOS api-key///                          sealed-session; self-host: Better Auth cookie/bearer///                          x-api-key). The credential precedence stays INSIDE each//                          impl.//   - `renderFailure`    — the failure-rendering strategy. Cloud renders the//                          shared errors as its exact `{ error, code }` JSON at//                          401/403/503; self-host catches them into 401/403/503//                          text. The seam (request -> Principal | shared error) is//                          identical; only the rendering differs.//   - `plugins`          — the host's plugin tuple (typed extension Services).//   - `stackLayer`       — the host's `makeExecutionStack` seam Layer (cloud://                          `CloudExecutionStackLayer`; self-host://                          `SelfHostExecutionStackLayer`).//// `LongLived` is the boot-scoped context captured at layer-build time (the// provider tag + the stack's long-lived deps) so the per-request function only// depends on `HttpRouter`-provided context. The returned value is the// `HttpRouter.middleware` (NOT `.layer`) so a host can still `.combine(...)` a// request-scoped middleware into it (cloud folds its per-request DB layer).// --------------------------------------------------------------------------- import { HttpRouter, HttpServerRequest, HttpServerResponse } from "effect/unstable/http";import { Context, Effect, Layer } from "effect"; import type { AnyPlugin } from "@executor-js/sdk"; import type { DbProvider } from "./executor-fuma-db";import { RequestWebOrigin, type HostConfig, type PluginsProvider } from "./scoped-executor";import { ExecutionEngineService, ExecutorService } from "../services";import { providePluginExtensions, type PluginExtensionServices } from "../plugin-routes";import {  authContextFromPrincipal,  AuthContext,  type IdentityFailure,  type Principal,} from "./identity";import {  makeExecutionStack,  type CodeExecutorProvider,  type EngineDecorator,} from "./execution-stack"; /** * A failure-rendering strategy. `renderFailure` runs on the result of * `authenticate`: it MUST either re-raise the failure (so a `Respondable` typed * error reaches the framework's response pipeline — cloud) or recover it into a * concrete `HttpServerResponse` (self-host's explicit 401/403 text). `RR` is the * residual requirement the strategy adds (always `never` in practice). */export interface FailureRenderingStrategy<E, RR = never> {  readonly renderFailure: <R>(    effect: Effect.Effect<Principal, E, R>,  ) => Effect.Effect<Principal | HttpServerResponse.HttpServerResponse, E, R | RR>;} /** * Self-host's strategy: this is an `HttpRouter` middleware (not an `HttpApi` * endpoint), so a failed typed error would surface as a 500 — recover * `Unauthorized` -> 401 text and `NoOrganization` -> 403 text instead. Self-host * never produces `Unavailable`, but the shared channel now includes it, so it is * recovered to a 503 text for total coverage. */export const textFailureStrategy: FailureRenderingStrategy<IdentityFailure> = {  renderFailure: (effect) =>    effect.pipe(      Effect.catchTags({        Unauthorized: () =>          Effect.succeed(HttpServerResponse.text("Unauthorized", { status: 401 })),        NoOrganization: () =>          Effect.succeed(            HttpServerResponse.text("No organization for this account", {              status: 403,            }),          ),        Unavailable: () =>          Effect.succeed(            HttpServerResponse.text("Authentication temporarily unavailable", {              status: 503,            }),          ),      }),    ),}; export interface MakeExecutionStackMiddlewareOptions<  TPlugins extends readonly AnyPlugin[],  E,  RLong,  RStack,  RStrategy,> {  /** The host's plugin tuple — drives the typed extension Services and binding. */  readonly plugins: TPlugins;  /**   * Resolve the inbound web `Request` to a neutral `Principal`. Adapter-specific   * credential precedence stays inside this function.   */  readonly authenticate: (request: Request) => Effect.Effect<Principal, E, RLong>;  /** Render `authenticate` failures (passthrough for cloud, text for self-host). */  readonly strategy: FailureRenderingStrategy<E, RStrategy>;  /** The host's `makeExecutionStack` seam Layer. */  readonly stackLayer: Layer.Layer<    DbProvider | PluginsProvider | HostConfig | CodeExecutorProvider | EngineDecorator,    never,    RStack  >;} /** * Build the shared `ExecutionStackMiddleware`. `RCapture` is the boot-scoped * context captured ONCE at layer-build time; anything the per-request body still * needs (`RLong | RStack | RStrategy` minus `RCapture`) stays a residual * requirement of the returned middleware, satisfied per request by the host. * *   - self-host captures everything (`AuthProvider | SelfHostDb`): no residual, *     so `.layer` is a complete Layer. *   - cloud captures only the boot-scoped services (its identity provider + the *     app-only billing service its metered stack reads) and leaves `DbService` *     residual, satisfied per request by `.combine(requestScopedMiddleware(rsLive))` *     (so the postgres.js socket lives in the request fiber's scope). * * The returned value is the `HttpRouter.middleware` (NOT `.layer`) so cloud can * still `.combine(...)`. */export const makeExecutionStackMiddleware = <  const TPlugins extends readonly AnyPlugin[],  E,  RLong = never,  RStack = never,  RStrategy = never,  RCapture = RLong | RStack | RStrategy,>(  options: MakeExecutionStackMiddlewareOptions<TPlugins, E, RLong, RStack, RStrategy>,) => {  const provideExecutorExtensions = providePluginExtensions(options.plugins);  return HttpRouter.middleware<{    provides:      | AuthContext      | ExecutorService      | ExecutionEngineService      | PluginExtensionServices<TPlugins>;  }>()(    Effect.gen(function* () {      const captured = yield* Effect.context<RCapture>();      return (httpEffect) =>        Effect.gen(function* () {          const request = yield* HttpServerRequest.HttpServerRequest;          const webRequest = yield* HttpServerRequest.toWeb(request);          const resolved = yield* options.strategy.renderFailure(options.authenticate(webRequest));          // The strategy recovered the failure into a Response — return it.          if (!isPrincipal(resolved)) return resolved;          const auth = AuthContext.of(authContextFromPrincipal(resolved));          // The public origin the caller actually hit, so a host with no static          // web base URL (a Worker) derives one zero-config. An explicit          // `HostConfig.webBaseUrl` still wins; we deliberately read `request.url`          // (not a spoofable `X-Forwarded-Host`).          const { executor, engine } = yield* makeExecutionStack<TPlugins>(            resolved.accountId,            resolved.organizationId,            resolved.organizationName,          ).pipe(            Effect.provide(options.stackLayer),            Effect.provideService(RequestWebOrigin, {              origin: requestWebOriginFromRequest(webRequest),            }),          );          return yield* httpEffect.pipe(            Effect.provideService(AuthContext, auth),            Effect.provideService(ExecutorService, executor),            Effect.provideService(ExecutionEngineService, engine),            provideExecutorExtensions(executor),          );          // Provide the boot-captured context; uncaptured deps (cloud's          // request-scoped `DbService`) remain residual and flow through here.        }).pipe(Effect.provideContext(captured as Context.Context<RCapture>));    }),  );}; // `renderFailure` yields either the resolved `Principal` (proceed) or an// already-built `HttpServerResponse` (the strategy recovered the failure). A// `Principal` is a plain object with `accountId`; a response is tagged. Discern// by the marker the response framework brands its values with.const isPrincipal = (  value: Principal | HttpServerResponse.HttpServerResponse,): value is Principal => !HttpServerResponse.isHttpServerResponse(value); const LOOPBACK_HOSTNAMES = new Set(["localhost", "127.0.0.1", "::1", "[::1]"]); const parseOrigin = (value: string): URL | null => {  // oxlint-disable-next-line executor/no-try-catch-or-throw -- boundary: new URL() throws on malformed origin; no Effect equivalent for this sync parse  try {    const parsed = new URL(value);    parsed.pathname = "/";    parsed.search = "";    parsed.hash = "";    return parsed;  } catch {    return null;  }}; const isLoopbackOrigin = (value: URL): boolean => LOOPBACK_HOSTNAMES.has(value.hostname); const originString = (value: URL): string => value.origin; export const requestWebOriginFromRequest = (request: Request): string => {  const requestUrl = new URL(request.url);  const requestOrigin = requestUrl.origin;  const browserOriginHeader = request.headers.get("origin");  if (!browserOriginHeader) return requestOrigin;   const browserOrigin = parseOrigin(browserOriginHeader);  if (!browserOrigin) return requestOrigin;  if (!isLoopbackOrigin(requestUrl) || !isLoopbackOrigin(browserOrigin)) return requestOrigin;  if (requestUrl.protocol !== browserOrigin.protocol) return requestOrigin;  if (requestUrl.port !== browserOrigin.port) return requestOrigin;  return originString(browserOrigin);};