@acrool/rtk-query-codegen-openapi 0.0.2-test.7 → 0.0.2-test.9

package/lib/index.mjs

@@ -458,12 +458,14 @@ async function generateApi(spec, {
     );
     const printer2 = ts4.createPrinter({ newLine: ts4.NewLineKind.LineFeed });
     const allTypeDefinitions = [];
+    const definedTypeNames = /* @__PURE__ */ new Set();
     const components = v3Doc.components;
     if (components) {
       const componentDefinitions = Object.entries(components).map(([componentType, componentDefs]) => {
         const typeEntries = Object.entries(componentDefs).map(([name, def]) => {
           addSchemeTypeName(name);
           const typeName = capitalize(camelCase(name));
+          definedTypeNames.add(typeName);
           const typeNode = wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(def));
           return factory.createTypeAliasDeclaration(
             [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
@@ -481,10 +483,101 @@ async function generateApi(spec, {
       });
       allTypeDefinitions.push(...componentDefinitions);
     }
-    if (useEnumType) {
-      allTypeDefinitions.push(...apiGen.enumAliases);
+    const enumEntries = [
+      ...apiGen.enumAliases.filter((e) => ts4.isEnumDeclaration(e)),
+      ...apiGen.enumAliases.filter((e) => ts4.isTypeAliasDeclaration(e))
+    ].map((enumDecl) => {
+      if (ts4.isEnumDeclaration(enumDecl)) {
+        return factory.createEnumDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          enumDecl.name,
+          enumDecl.members
+        );
+      } else if (ts4.isTypeAliasDeclaration(enumDecl)) {
+        return factory.createTypeAliasDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          enumDecl.name,
+          enumDecl.typeParameters,
+          enumDecl.type
+        );
+      }
+      return enumDecl;
+    });
+    const unionTypeEnums = apiGen.aliases.filter((alias) => {
+      if (ts4.isTypeAliasDeclaration(alias) && alias.type) {
+        return ts4.isUnionTypeNode(alias.type);
+      }
+      return false;
+    }).map((alias) => {
+      if (ts4.isTypeAliasDeclaration(alias)) {
+        return factory.createTypeAliasDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          alias.name,
+          alias.typeParameters,
+          alias.type
+        );
+      }
+      return alias;
+    });
+    const allEnumEntries = [...enumEntries, ...unionTypeEnums];
+    if (allEnumEntries.length > 0) {
+      allTypeDefinitions.push(
+        factory.createModuleDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          factory.createIdentifier("Enum"),
+          factory.createModuleBlock(allEnumEntries),
+          ts4.NodeFlags.Namespace
+        )
+      );
+    }
+    if (apiGen.aliases.length > 0) {
+      const aliasEntries = apiGen.aliases.filter((alias) => {
+        if (ts4.isTypeAliasDeclaration(alias)) {
+          const isDefinedInComponents = definedTypeNames.has(alias.name.text);
+          const isUnionTypeEnum = ts4.isUnionTypeNode(alias.type);
+          return !isDefinedInComponents && !isUnionTypeEnum;
+        }
+        return false;
+      }).map((alias) => {
+        if (ts4.isTypeAliasDeclaration(alias)) {
+          return factory.createTypeAliasDeclaration(
+            [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+            alias.name,
+            alias.typeParameters,
+            alias.type
+          );
+        }
+        return alias;
+      });
+      if (aliasEntries.length > 0) {
+        const existingSchemeIndex = allTypeDefinitions.findIndex(
+          (def) => ts4.isModuleDeclaration(def) && ts4.isIdentifier(def.name) && def.name.text === "Scheme"
+        );
+        if (existingSchemeIndex >= 0) {
+          const existingScheme = allTypeDefinitions[existingSchemeIndex];
+          const mergedMembers = [...existingScheme.body.statements, ...aliasEntries];
+          allTypeDefinitions[existingSchemeIndex] = factory.createModuleDeclaration(
+            [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+            factory.createIdentifier("Scheme"),
+            factory.createModuleBlock(mergedMembers),
+            ts4.NodeFlags.Namespace
+          );
+        } else {
+          allTypeDefinitions.push(
+            factory.createModuleDeclaration(
+              [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+              factory.createIdentifier("Scheme"),
+              factory.createModuleBlock(aliasEntries),
+              ts4.NodeFlags.Namespace
+            )
+          );
+        }
+      }
     }
-    allTypeDefinitions.push(...apiGen.aliases);
+    const fs2 = await import("node:fs/promises");
+    const path4 = await import("node:path");
+    const sharedTypesDir = path4.dirname(sharedTypesFile);
+    await fs2.mkdir(sharedTypesDir, { recursive: true });
     const output = printer2.printNode(
       ts4.EmitHint.Unspecified,
       factory.createSourceFile(
@@ -494,7 +587,6 @@ async function generateApi(spec, {
       ),
       resultFile2
     );
-    const fs2 = await import("node:fs/promises");
     await fs2.writeFile(sharedTypesFile, output, "utf-8");
   }
   if (apiGen.spec.components?.schemas) {
@@ -539,7 +631,12 @@ async function generateApi(spec, {
       [
         generateImportNode(apiFile, { [apiImport]: "api" }),
         generateImportNode("@acrool/react-fetcher", { IRestFulEndpointsQueryReturn: "IRestFulEndpointsQueryReturn" }),
-        ...sharedTypesFile ? [generateImportNode(sharedTypesImportPath, { Scheme: "Scheme" })] : [],
+        ...sharedTypesFile ? [
+          generateImportNode(sharedTypesImportPath, {
+            Scheme: "Scheme",
+            ...useEnumType ? { Enum: "Enum" } : {}
+          })
+        ] : [],
         ...tag ? [generateTagTypes({ addTagTypes: extractAllTagTypes({ operationDefinitions }) })] : [],
         generateCreateApiCall({
           tag,
@@ -614,25 +711,6 @@ async function generateApi(spec, {
       ).filter(
         ([status, response]) => isDataResponse(status, includeDefault, apiGen.resolve(response), responses || {})
       ).filter(([_1, _2, type]) => type !== keywordType.void).map(([code, response, type]) => {
-        if (sharedTypesFile2 && ts4.isTypeReferenceNode(type) && type.typeName) {
-          if (ts4.isIdentifier(type.typeName)) {
-            const typeName = type.typeName.text;
-            if (typeName in apiGen.aliases || typeName in apiGen.enumAliases) {
-              return ts4.addSyntheticLeadingComment(
-                factory.createTypeReferenceNode(
-                  factory.createQualifiedName(
-                    factory.createIdentifier("sharedTypes"),
-                    factory.createIdentifier(camelCase(typeName))
-                  ),
-                  type.typeArguments
-                ),
-                ts4.SyntaxKind.MultiLineCommentTrivia,
-                `* status ${code} ${response.description} `,
-                false
-              );
-            }
-          }
-        }
         return ts4.addSyntheticLeadingComment(
           type,
           ts4.SyntaxKind.MultiLineCommentTrivia,
@@ -841,6 +919,28 @@ async function generateApi(spec, {
   function wrapWithSchemeIfComponent(typeNode) {
     if (ts4.isTypeReferenceNode(typeNode) && ts4.isIdentifier(typeNode.typeName)) {
       const typeName = typeNode.typeName.text;
+      const isEnumType = useEnumType && (apiGen.enumAliases.some((enumDecl) => {
+        if (ts4.isEnumDeclaration(enumDecl) || ts4.isTypeAliasDeclaration(enumDecl)) {
+          return enumDecl.name.text === typeName;
+        }
+        return false;
+      }) || apiGen.aliases.some((alias) => {
+        if (ts4.isTypeAliasDeclaration(alias) && alias.type) {
+          if (ts4.isUnionTypeNode(alias.type)) {
+            return alias.name.text === typeName;
+          }
+        }
+        return false;
+      }));
+      if (isEnumType) {
+        return factory.createTypeReferenceNode(
+          factory.createQualifiedName(
+            factory.createIdentifier("Enum"),
+            typeNode.typeName
+          ),
+          typeNode.typeArguments?.map(wrapWithSchemeIfComponent)
+        );
+      }
       if (schemeTypeNames.has(typeName)) {
         return factory.createTypeReferenceNode(
           factory.createQualifiedName(
@@ -861,6 +961,40 @@ async function generateApi(spec, {
       return factory.createArrayTypeNode(wrapWithSchemeIfComponent(typeNode.elementType));
     }
     if (ts4.isUnionTypeNode(typeNode)) {
+      const unionTypes = typeNode.types;
+      if (unionTypes.length > 0 && unionTypes.every(
+        (type) => ts4.isLiteralTypeNode(type) && (ts4.isStringLiteral(type.literal) || ts4.isNumericLiteral(type.literal))
+      )) {
+        const enumValues = unionTypes.map((type) => {
+          if (ts4.isLiteralTypeNode(type)) {
+            if (ts4.isStringLiteral(type.literal)) {
+              return type.literal.text;
+            } else if (ts4.isNumericLiteral(type.literal)) {
+              return type.literal.text;
+            }
+          }
+          return null;
+        }).filter(Boolean);
+        const matchingEnum = apiGen.aliases.find((alias) => {
+          if (ts4.isTypeAliasDeclaration(alias) && ts4.isUnionTypeNode(alias.type)) {
+            const aliasValues = alias.type.types.map((type) => {
+              if (ts4.isLiteralTypeNode(type)) {
+                if (ts4.isStringLiteral(type.literal)) {
+                  return type.literal.text;
+                } else if (ts4.isNumericLiteral(type.literal)) {
+                  return type.literal.text;
+                }
+              }
+              return null;
+            }).filter(Boolean);
+            return aliasValues.length === enumValues.length && aliasValues.every((val) => enumValues.includes(val));
+          }
+          return false;
+        });
+        if (matchingEnum && ts4.isTypeAliasDeclaration(matchingEnum)) {
+          return typeNode;
+        }
+      }
       return factory.createUnionTypeNode(typeNode.types.map(wrapWithSchemeIfComponent));
     }
     if (ts4.isTypeLiteralNode(typeNode)) {
@@ -997,12 +1131,36 @@ function parseConfig(fullConfig) {
     }, {});
     Object.entries(groupedPaths).forEach(([groupName, paths2]) => {
       const finalOutputPath = outputPath.replace("$1", groupName);
-      const filterEndpoints = filterEndpoint(groupName);
-      outFiles.push({
-        ...commonConfig,
-        outputFile: finalOutputPath,
-        filterEndpoints: [filterEndpoints]
-      });
+      if (filterEndpoint) {
+        const pathBasedFilter = (operationName, operationDefinition) => {
+          const path4 = operationDefinition.path;
+          const pathGroupName = getGroupNameFromPath(path4, pattern);
+          if (pathGroupName !== groupName) {
+            return false;
+          }
+          const endpointFilter = filterEndpoint(groupName);
+          if (endpointFilter instanceof RegExp) {
+            return endpointFilter.test(operationName);
+          }
+          return true;
+        };
+        outFiles.push({
+          ...commonConfig,
+          outputFile: finalOutputPath,
+          filterEndpoints: pathBasedFilter
+        });
+      } else {
+        const pathBasedFilter = (operationName, operationDefinition) => {
+          const path4 = operationDefinition.path;
+          const pathGroupName = getGroupNameFromPath(path4, pattern);
+          return pathGroupName === groupName;
+        };
+        outFiles.push({
+          ...commonConfig,
+          outputFile: finalOutputPath,
+          filterEndpoints: pathBasedFilter
+        });
+      }
     });
   } else {
     outFiles.push(fullConfig);