feat: internal ability to detect which type is extracted

Author: Сергей Волков

src/configuration.ts

@@ -5,6 +5,8 @@ import * as typescript from "typescript";
 import type {
   ExtractingOptions,
   GenerateApiConfiguration,
+  Hooks,
+  SchemaComponent,
 } from "../types/index.js";
 import { ComponentTypeNameResolver } from "./component-type-name-resolver.js";
 import * as CONSTANTS from "./constants.js";
@@ -86,11 +88,11 @@ export class CodeGenConfig {
   };
   routeNameDuplicatesMap = new Map();
   prettierOptions = { ...CONSTANTS.PRETTIER_OPTIONS };
-  hooks = {
+  hooks: Hooks = {
     onPreBuildRoutePath: (_routePath: unknown) => void 0,
     onBuildRoutePath: (_routeData: unknown) => void 0,
     onInsertPathParam: (_pathParam: unknown) => void 0,
-    onCreateComponent: (schema: unknown) => schema,
+    onCreateComponent: (schema: SchemaComponent) => schema,
     onPreParseSchema: (
       _originalSchema: unknown,
       _typeName: unknown,

src/schema-components-map.ts

@@ -21,11 +21,16 @@ export class SchemaComponentsMap {
     return ref.split("/");
   };
 
-  createComponent($ref: string, rawTypeData: string) {
+  createComponent(
+    $ref: string,
+    rawTypeData: SchemaComponent["rawTypeData"],
+  ): SchemaComponent {
     const parsed = this.parseRef($ref);
-    const typeName = parsed[parsed.length - 1];
-    const componentName = parsed[parsed.length - 2];
-    const componentSchema = {
+    const typeName = parsed[parsed.length - 1]!;
+    const componentName = parsed[
+      parsed.length - 2
+    ] as SchemaComponent["componentName"];
+    const componentSchema: SchemaComponent = {
       $ref,
       typeName,
       rawTypeData,
@@ -52,7 +57,7 @@ export class SchemaComponentsMap {
     return this._data;
   }
 
-  filter(...componentNames: string[]) {
+  filter(...componentNames: (string[] | string)[]) {
     return this._data.filter((it) =>
       componentNames.some((componentName) =>
         it.$ref.startsWith(`#/components/${componentName}`),

src/schema-parser/schema-parser-fabric.ts

@@ -1,5 +1,6 @@
 import type {
   ParsedSchema,
+  SchemaComponent,
   SchemaTypeEnumContent,
   SchemaTypeObjectContent,
   SchemaTypePrimitiveContent,
@@ -63,7 +64,11 @@ export class SchemaParserFabric {
     return parser.schema;
   };
 
-  createParsedComponent = ({ typeName, schema, schemaPath }) => {
+  createParsedComponent = ({
+    typeName,
+    schema,
+    schemaPath,
+  }): SchemaComponent => {
     const schemaCopy = structuredClone(schema);
     const customComponent = this.schemaComponentsMap.createComponent(
       this.schemaComponentsMap.createRef(["components", "schemas", typeName]),

src/schema-routes/schema-routes.ts

@@ -597,6 +597,10 @@ export class SchemaRoutes {
         typeName,
         schemaPath: [operationId],
       });
+
+      if (schema?.typeData) {
+        schema.typeData.isExtractedRequestBody = true;
+      }
       content = this.schemaParserFabric.getInlineParseContent({
         $ref: schema.$ref,
       });
@@ -670,10 +674,16 @@ export class SchemaRoutes {
         },
       );
 
-      return this.schemaParserFabric.createParsedComponent({
+      const component = this.schemaParserFabric.createParsedComponent({
         typeName: generatedTypeName,
         schema: schema,
       });
+
+      if (component.typeData) {
+        component.typeData.isExtractedRequestParams = true;
+      }
+
+      return component;
     }
 
     return schema;
@@ -705,6 +715,9 @@ export class SchemaRoutes {
           schemaPath: [routeInfo.operationId],
         });
         successResponse.schema.contentKind = contentKind;
+        if (successResponse.schema.typeData) {
+          successResponse.schema.typeData.isExtractedResponseBody = true;
+        }
         successResponse.type = this.schemaParserFabric.getInlineParseContent({
           $ref: successResponse.schema.$ref,
         });
@@ -756,6 +769,9 @@ export class SchemaRoutes {
         { ...schema },
       );
       responseBodyInfo.error.schemas = [component];
+      if (component.typeData) {
+        component.typeData.isExtractedResponseError = true;
+      }
       responseBodyInfo.error.type = this.typeNameFormatter.format(
         component.typeName,
       );

types/index.ts

@@ -442,6 +442,10 @@ export interface ParsedSchema<C> {
   description?: string;
   allFieldsAreOptional?: boolean;
   content: C;
+  isExtractedRequestParams?: boolean;
+  isExtractedRequestBody?: boolean;
+  isExtractedResponseBody?: boolean;
+  isExtractedResponseError?: boolean;
 }
 
 export interface PathArgInfo {