feat: implementation completed

src/boot.ts

@@ -0,0 +1,4 @@
+import { extendZodWithOpenApi } from '@asteasolutions/zod-to-openapi'
+import { z } from 'zod'
+
+extendZodWithOpenApi(z)

src/index.ts

@@ -0,0 +1,2 @@
+export * from './patch-nest-swagger'
+export { createZodDto, ZodValidationPipe } from '@anatine/zod-nestjs'

src/patch-nest-swagger.ts

@@ -0,0 +1,96 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/* eslint-disable @typescript-eslint/ban-types */
+
+/**
+ * This file was copied from:
+ *   https://github.com/anatine/zod-plugins/blob/main/packages/zod-nestjs/src/lib/patch-nest-swagger.ts
+ */
+import {
+  OpenApiGeneratorV3,
+  OpenAPIRegistry,
+} from '@asteasolutions/zod-to-openapi'
+import { INestApplication } from '@nestjs/common'
+import { SwaggerDocumentOptions } from '@nestjs/swagger'
+import type { SchemaObject } from 'openapi3-ts/oas31'
+
+interface Type<T = any> extends Function {
+  new (...args: any[]): T
+}
+
+interface Options {
+  sort: 'default' | 'alpha' | 'localeCompare'
+}
+
+export function patchNestjsSwagger(
+  { sort = 'default' }: Options,
+  {
+    schemaObjectFactoryModule = require('@nestjs/swagger/dist/services/schema-object-factory'),
+    swaggerScannerModule = require('@nestjs/swagger/dist/swagger-scanner'),
+  },
+) {
+  const registry = new OpenAPIRegistry()
+
+  /**
+   * Override `exploreModelSchema` from schema-object-factory
+   */
+  const { SchemaObjectFactory } = schemaObjectFactoryModule
+  const orgExploreModelSchema = SchemaObjectFactory.prototype.exploreModelSchema
+  SchemaObjectFactory.prototype.exploreModelSchema = function (
+    type: Type<unknown> | Function | any,
+    schemas: Record<string, SchemaObject>,
+    schemaRefsStack: string[] = [],
+  ) {
+    if (this.isLazyTypeFunc(type)) {
+      // eslint-disable-next-line no-param-reassign
+      type = (type as Function)()
+    }
+
+    if (!type.zodSchema) {
+      return orgExploreModelSchema.call(this, type, schemas, schemaRefsStack)
+    }
+
+    registry.register(type.name, type.zodSchema)
+
+    return type.name
+  }
+
+  /**
+   * Override `scanApplication` from swagger-scanner
+   */
+  const { SwaggerScanner } = swaggerScannerModule
+  const orgScanApplication = SwaggerScanner.prototype.scanApplication
+  SwaggerScanner.prototype.scanApplication = function (
+    app: INestApplication,
+    options: SwaggerDocumentOptions,
+  ) {
+    const openAPIObject = orgScanApplication.call(this, app, options)
+
+    const generator = new OpenApiGeneratorV3(registry.definitions)
+    const doc = generator.generateComponents()
+
+    const schemas = {
+      ...openAPIObject.components.schemas,
+      ...doc.components?.schemas,
+    }
+
+    if (sort === 'alpha') {
+      const sortedSchemas = Object.entries(schemas).sort(([aKey], [bKey]) => {
+        if (aKey < bKey) return -1
+        if (aKey > bKey) return 1
+        return 0
+      })
+
+      openAPIObject.components.schemas = Object.fromEntries(sortedSchemas)
+    } else if (sort === 'localeCompare') {
+      const sortedSchemas = Object.entries(schemas).sort(([aKey], [bKey]) =>
+        aKey.localeCompare(bKey),
+      )
+
+      openAPIObject.components.schemas = Object.fromEntries(sortedSchemas)
+    } else {
+      openAPIObject.components.schemas = schemas
+    }
+
+    return openAPIObject
+  }
+}