130 lines · 6.1 KB · TypeScript
RawDownload
// ---------------------------------------------------------------------------// Cloud execution-stack seams.//// The shared `makeExecutionStack` (@executor-js/api/server) owns the body://   makeScopedExecutor -> createExecutionEngine -> EngineDecorator.decorate.// Used by the protected HTTP API (per-request) and the MCP session DO// (per-session) so changes to the stack flow to both. Cloud supplies the five// seam Layers it reads from; the only cloud-specific differences are the// Cloudflare dynamic-worker code substrate and the usage-metering decorator.////   - DbProvider          -> cloudDbProviderLayer: rebuilds the postgres-js fuma//                            client per request off the request-scoped//                            `DbService.db` (Hyperdrive forbids sharing an I/O//                            handle across requests). The shared factory reads//                            `db` without caching, preserving per-request rebuild.//   - PluginsProvider      -> fresh per-request plugins with the Worker env's//                            WorkOS credentials.//   - HostConfig           -> `allowLocalNetwork` is config-driven (the//                            `ALLOW_LOCAL_NETWORK` var; production leaves it unset//                            -> `false`, the test workers set it `"true"`). It is//                            an SSRF/private-network guard, so it MUST NOT key off//                            a test flag. `webBaseUrl` is `VITE_PUBLIC_SITE_URL ??//                            executor.sh`.//   - CodeExecutorProvider -> `makeDynamicWorkerExecutor({ loader: env.LOADER })`.//   - EngineDecorator      -> the billing decorator that meters each execution//                            to Autumn. BOTH cloud execution planes (the HTTP//                            `/api/*` executor plane AND the MCP session DO) use//                            the metered stack (`CloudMeteredExecutionStackLayer`,//                            ../engine/execution-stack-metered.ts), since the MCP//                            server is the primary execution surface. Billing//                            still lives in the cloud app, not this neutral//                            seams module; the decorator is composed on top.// --------------------------------------------------------------------------- import { env } from "cloudflare:workers";import { Layer } from "effect"; import {  CodeExecutorProvider,  DbProvider,  HostConfig,  PluginsProvider,  collectTables,} from "@executor-js/api/server";import { makeDynamicWorkerExecutor } from "@executor-js/runtime-dynamic-worker";import type { AnyPlugin } from "@executor-js/sdk"; import executorConfig from "../../executor.config";import { DbService } from "../db/db";import { cloudDbProviderLayer } from "../db/fuma"; export { makeExecutionStack } from "@executor-js/api/server"; // The executor table set is fixed (plugin-independent), so the per-request// DbProvider rebuilds the fuma client over the same schema.export const CloudDbProvider = cloudDbProviderLayer(collectTables()); const cloudPluginFactory = executorConfig.plugins as (deps: {  readonly workosCredentials: {    readonly apiKey: string;    readonly clientId: string;    readonly apiUrl?: string;  };  readonly activeToolkitSlug?: string;  readonly allowLocalNetwork?: boolean;  readonly workerLoader?: WorkerLoader;}) => readonly AnyPlugin[]; // Fresh plugin instances per request, carrying the Worker env's WorkOS Vault// credentials. Matches the old `createScopedExecutor`'s `orgPlugins()`.export const CloudPluginsProvider: Layer.Layer<PluginsProvider> = Layer.succeed(PluginsProvider)({  plugins: (context) =>    cloudPluginFactory({      workosCredentials: {        apiKey: env.WORKOS_API_KEY,        clientId: env.WORKOS_CLIENT_ID,        apiUrl: env.WORKOS_API_URL,      },      activeToolkitSlug:        context?.mcpResource?.kind === "toolkit" ? context.mcpResource.slug : undefined,      allowLocalNetwork: env.ALLOW_LOCAL_NETWORK === "true",      workerLoader: env.LOADER,    }),}); /** * The path prefix the cloud mounts its typed API under. SINGLE SOURCE OF TRUTH: * `app.ts` passes this as `ExecutorApp.make({ config: { mountPrefix } })`, and * `make` derives the OAuth callback (`${webBaseUrl}${CLOUD_MOUNT_PREFIX}/oauth/callback`) * from that same `mountPrefix`, so the redirect URI the host sends to providers * always matches the route that actually serves the callback — no second knob. */export const CLOUD_MOUNT_PREFIX = "/api" as const; export const CloudHostConfig: Layer.Layer<HostConfig> = Layer.sync(HostConfig, () => ({  // SSRF / private-network egress guard. Config-driven, NOT a test flag:  // production leaves `ALLOW_LOCAL_NETWORK` unset so the guard stays ON (`false`);  // the e2e dev-server env opts in with `"true"` so in-scenario fixture  // servers on localhost are reachable. See `hosted-http-client.ts`.  allowLocalNetwork: env.ALLOW_LOCAL_NETWORK === "true",  webBaseUrl: env.VITE_PUBLIC_SITE_URL ?? "https://executor.sh",  oauthCallbackPath: `${CLOUD_MOUNT_PREFIX}/oauth/callback`,  // WorkOS Vault is cloud's credential storage implementation detail, not a  // user-selectable provider surface.  exposeCredentialProviders: false,})); export const CloudCodeExecutorProvider: Layer.Layer<CodeExecutorProvider> = Layer.sync(  CodeExecutorProvider,  () => makeDynamicWorkerExecutor({ loader: env.LOADER }),); /** * The four billing-free execution-stack seams (db / plugins / host-config / * code-executor): everything `makeExecutionStack` reads EXCEPT the * `EngineDecorator`. Both cloud planes compose this with the billing decorator * via `CloudMeteredExecutionStackLayer` (../engine/execution-stack-metered.ts); * exported so that overlay builds over the SAME four seams. There is no neutral * no-op-decorator variant anymore: every cloud execution meters. */export const CloudExecutionSeamsLayer: Layer.Layer<  DbProvider | PluginsProvider | HostConfig | CodeExecutorProvider,  never,  DbService> = Layer.mergeAll(  CloudDbProvider,  CloudPluginsProvider,  CloudHostConfig,  CloudCodeExecutorProvider,);