feat: register all components/schemas from openapi spec

cloudventure-io · Nov 24, 2023 · 3ed3296 · 3ed3296
1 parent 8290a9d
commit 3ed3296
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/http-api/HttpApi.ts b/src/http-api/HttpApi.ts
@@ -281,7 +281,7 @@ export class HttpApi<OperationType extends object = object> extends Construct {
     }
   }
 
-  private registerSchema(operation: ParsedOperation<OperationType>): string {
+  private registerOperationSchema(operation: ParsedOperation<OperationType>): string {
     const {
       operationId,
       request: {
@@ -361,7 +361,7 @@ export class HttpApi<OperationType extends object = object> extends Construct {
   >): Promise<BundlerTypeScriptHandler> {
     // get dereferenced version of the operation
     const operation = await this.operationParser.parseOperation(pathPattern, method)
-    const operationTitle = this.registerSchema(operation)
+    const operationTitle = this.registerOperationSchema(operation)
 
     // render lambda function entry point, handler and validator
     const entryPointPath = await this.renderLambdaFiles(operation, operationTitle)
@@ -471,6 +471,12 @@ export class HttpApi<OperationType extends object = object> extends Construct {
   public async _preSynth() {
     // write the OpenAPI document
     await writeFile(this.documentOutputPath, JSON.stringify(this.config.document, null, 2))
+
+    const doc = await this.operationParser.document
+
+    if (doc.components?.schemas) {
+      Object.values(doc.components.schemas).forEach(schema => this.bundler.registerSchema(schema))
+    }
   }
 
   public async _synth() {

diff --git a/src/http-api/openapi/OperationParser.ts b/src/http-api/openapi/OperationParser.ts
@@ -127,7 +127,7 @@ export class OperationParser<OperationType extends object = object> {
     }
   }
 
-  private get document(): PromiseLike<DereferencedDocument<OperationType>> {
+  public get document(): PromiseLike<DereferencedDocument<OperationType>> {
     if (!this.dereferencedDocument) {
       this.dereferencedDocument = new Promise<DereferencedDocument<OperationType>>(resolve => {
         SwaggerParser.dereference(JSON.parse(JSON.stringify(this.rawDocument))).then(doc =>