Pass OpenAPI document to Context class constructor

.changeset/pass-openapi-document-to-context.md

@@ -0,0 +1,16 @@
+---
+"counterfact": minor
+---
+
+Pass the OpenAPI document to the Context class constructor in `_.context.ts`.
+
+The `Context` constructor now receives an `openApiDocument` property alongside the existing `loadContext` and `readJson` helpers:
+
+```ts
+// routes/_.context.ts
+export class Context {
+  constructor({ openApiDocument, loadContext, readJson }) {
+    this.openApiDocument = openApiDocument;
+  }
+}
+```

src/app.ts

@@ -85,6 +85,7 @@ export async function createMswHandlers(
     compiledPathsDirectory,
     registry,
     contextRegistry,
+    openApiDocument,
   );
   await moduleLoader.load();
   const routes = registry.routes;
@@ -129,10 +130,12 @@ export async function counterfact(config: Config) {
     config.generate,
   );
 
+  const openApiDocument = await loadOpenApiDocument(config.openApiPath);
+
   const dispatcher = new Dispatcher(
     registry,
     contextRegistry,
-    await loadOpenApiDocument(config.openApiPath),
+    openApiDocument,
     config,
   );
 
@@ -146,6 +149,7 @@ export async function counterfact(config: Config) {
     compiledPathsDirectory,
     registry,
     contextRegistry,
+    openApiDocument,
   );
 
   const middleware = koaMiddleware(dispatcher, config);

src/server/module-loader.ts

@@ -9,6 +9,7 @@ import createDebug from "debug";
 import { CHOKIDAR_OPTIONS } from "./constants.js";
 import { type Context, ContextRegistry } from "./context-registry.js";
 import { determineModuleKind } from "./determine-module-kind.js";
+import type { OpenApiDocument } from "./dispatcher.js";
 import { ModuleDependencyGraph } from "./module-dependency-graph.js";
 import type { MiddlewareFunction, Module, Registry } from "./registry.js";
 import { uncachedImport } from "./uncached-import.js";
@@ -51,6 +52,8 @@ export class ModuleLoader extends EventTarget {
 
   private readonly contextRegistry: ContextRegistry;
 
+  private readonly openApiDocument: OpenApiDocument | undefined;
+
   private readonly dependencyGraph = new ModuleDependencyGraph();
 
   private readonly uncachedImport: (moduleName: string) => Promise<unknown> =
@@ -62,11 +65,13 @@ export class ModuleLoader extends EventTarget {
     basePath: string,
     registry: Registry,
     contextRegistry = new ContextRegistry(),
+    openApiDocument?: OpenApiDocument,
   ) {
     super();
     this.basePath = basePath.replaceAll("\\", "/");
     this.registry = registry;
     this.contextRegistry = contextRegistry;
+    this.openApiDocument = openApiDocument;
   }
 
   public async watch(): Promise<void> {
@@ -268,6 +273,7 @@ export class ModuleLoader extends EventTarget {
 
           new endpoint.Context({
             loadContext,
+            openApiDocument: this.openApiDocument,
             readJson,
           }),
         );

test/server/module-loader.test.ts

@@ -363,6 +363,33 @@ describe("a module loader", () => {
     });
   });
 
+  it("passes openApiDocument to the Context constructor", async () => {
+    await usingTemporaryFiles(async ($) => {
+      await $.add(
+        "_.context.js",
+        "export class Context { constructor({ openApiDocument }) { this.openApiDocument = openApiDocument; } }",
+      );
+      await $.add("package.json", '{ "type": "module" }');
+
+      const registry: Registry = new Registry();
+      const contextRegistry: ContextRegistry = new ContextRegistry();
+      const openApiDocument = { paths: { "/hello": {} } };
+
+      const loader: ModuleLoader = new ModuleLoader(
+        $.path("."),
+        registry,
+        contextRegistry,
+        openApiDocument,
+      );
+
+      await loader.load();
+
+      const rootContext = contextRegistry.find("/") as any;
+
+      expect(rootContext?.openApiDocument).toBe(openApiDocument);
+    });
+  });
+
   // can't test because I can't get Jest to refresh modules
   it.skip("updates the registry when a dependency is updated", async () => {
     await usingTemporaryFiles(async ($) => {