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

package/lib/index.mjs

@@ -305,14 +305,14 @@ function removeUndefined(t) {
 
 // src/generators/react-hooks.ts
 var createBinding = ({
-  operationDefinition: { verb, path: path4, operation },
+  operationDefinition: { verb, path: path4 },
   overrides,
   isLazy = false
 }) => factory.createBindingElement(
   void 0,
   void 0,
   factory.createIdentifier(
-    `use${isLazy ? "Lazy" : ""}${capitalize(getOperationName(verb, path4, operation.operationId))}${isQuery(verb, overrides) ? "Query" : "Mutation"}`
+    `use${isLazy ? "Lazy" : ""}${capitalize(getOperationName(verb, path4, void 0))}${isQuery(verb, overrides) ? "Query" : "Mutation"}`
   ),
   void 0
 );
@@ -366,8 +366,8 @@ function defaultIsDataResponse(code, includeDefault) {
   const parsedCode = Number(code);
   return !Number.isNaN(parsedCode) && parsedCode >= 200 && parsedCode < 300;
 }
-function getOperationName2({ verb, path: path4, operation }) {
-  return _getOperationName(verb, path4, operation.operationId);
+function getOperationName2({ verb, path: path4 }) {
+  return _getOperationName(verb, path4, void 0);
 }
 function getTags({ verb, pathItem }) {
   return verb ? pathItem[verb]?.tags || [] : [];
@@ -432,7 +432,8 @@ async function generateApi(spec, {
   includeDefault = false,
   useEnumType = false,
   mergeReadWriteOnly = false,
-  httpResolverOptions
+  httpResolverOptions,
+  sharedTypesFile
 }) {
   const v3Doc = v3DocCache[spec] ??= await getV3Doc(spec, httpResolverOptions);
   const apiGen = new ApiGenerator(v3Doc, {
@@ -440,6 +441,62 @@ async function generateApi(spec, {
     useEnumType,
     mergeReadWriteOnly
   });
+  const schemeTypeNames = /* @__PURE__ */ new Set();
+  function addSchemeTypeName(name) {
+    schemeTypeNames.add(name);
+    schemeTypeNames.add(camelCase(name));
+    schemeTypeNames.add(capitalize(camelCase(name)));
+  }
+  if (sharedTypesFile) {
+    const resultFile2 = ts4.createSourceFile(
+      "sharedTypes.ts",
+      "",
+      ts4.ScriptTarget.Latest,
+      /*setParentNodes*/
+      false,
+      ts4.ScriptKind.TS
+    );
+    const printer2 = ts4.createPrinter({ newLine: ts4.NewLineKind.LineFeed });
+    const allTypeDefinitions = [];
+    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));
+          const typeNode = wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(def));
+          return factory.createTypeAliasDeclaration(
+            [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+            factory.createIdentifier(typeName),
+            void 0,
+            typeNode
+          );
+        });
+        return factory.createModuleDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          factory.createIdentifier("Scheme"),
+          factory.createModuleBlock(typeEntries),
+          ts4.NodeFlags.Namespace
+        );
+      });
+      allTypeDefinitions.push(...componentDefinitions);
+    }
+    if (useEnumType) {
+      allTypeDefinitions.push(...apiGen.enumAliases);
+    }
+    allTypeDefinitions.push(...apiGen.aliases);
+    const output = printer2.printNode(
+      ts4.EmitHint.Unspecified,
+      factory.createSourceFile(
+        allTypeDefinitions,
+        factory.createToken(ts4.SyntaxKind.EndOfFileToken),
+        ts4.NodeFlags.None
+      ),
+      resultFile2
+    );
+    const fs2 = await import("node:fs/promises");
+    await fs2.writeFile(sharedTypesFile, output, "utf-8");
+  }
   if (apiGen.spec.components?.schemas) {
     apiGen.preprocessComponents(apiGen.spec.components.schemas);
   }
@@ -471,12 +528,18 @@ async function generateApi(spec, {
     }
   }
   apiFile = apiFile.replace(/\.[jt]sx?$/, "");
+  const sharedTypesImportPath = sharedTypesFile && outputFile ? (() => {
+    let rel = path2.relative(path2.dirname(outputFile), sharedTypesFile).replace(/\\/g, "/").replace(/\.[jt]sx?$/, "");
+    if (!rel.startsWith(".")) rel = "./" + rel;
+    return rel;
+  })() : "./shared-types";
   return printer.printNode(
     ts4.EmitHint.Unspecified,
     factory.createSourceFile(
       [
         generateImportNode(apiFile, { [apiImport]: "api" }),
         generateImportNode("@acrool/react-fetcher", { IRestFulEndpointsQueryReturn: "IRestFulEndpointsQueryReturn" }),
+        ...sharedTypesFile ? [generateImportNode(sharedTypesImportPath, { Scheme: "Scheme" })] : [],
         ...tag ? [generateTagTypes({ addTagTypes: extractAllTagTypes({ operationDefinitions }) })] : [],
         generateCreateApiCall({
           tag,
@@ -484,7 +547,8 @@ async function generateApi(spec, {
             operationDefinitions.map(
               (operationDefinition) => generateEndpoint({
                 operationDefinition,
-                overrides: getOverrides(operationDefinition, endpointOverrides)
+                overrides: getOverrides(operationDefinition, endpointOverrides),
+                sharedTypesFile: !!sharedTypesFile
               })
             ),
             true
@@ -496,8 +560,7 @@ async function generateApi(spec, {
           factory.createIdentifier(generatedApiName)
         ),
         ...Object.values(interfaces),
-        ...apiGen.aliases,
-        ...apiGen.enumAliases,
+        ...sharedTypesFile ? [] : [...apiGen.aliases, ...apiGen.enumAliases],
         ...hooks ? [
           generateReactHooks({
             exportName: generatedApiName,
@@ -524,7 +587,8 @@ async function generateApi(spec, {
   }
   function generateEndpoint({
     operationDefinition,
-    overrides
+    overrides,
+    sharedTypesFile: sharedTypesFile2
   }) {
     const {
       verb,
@@ -533,7 +597,7 @@ async function generateApi(spec, {
       operation,
       operation: { responses, requestBody }
     } = operationDefinition;
-    const operationName = getOperationName2({ verb, path: path4, operation });
+    const operationName = getOperationName2({ verb, path: path4 });
     const tags = tag ? getTags({ verb, pathItem }) : [];
     const isQuery2 = isQuery(verb, overrides);
     const returnsJson = apiGen.getResponseType(responses) === "json";
@@ -543,18 +607,39 @@ async function generateApi(spec, {
         ([code, response]) => [
           code,
           apiGen.resolve(response),
-          apiGen.getTypeFromResponse(response, "readOnly") || factory.createKeywordTypeNode(ts4.SyntaxKind.UndefinedKeyword)
+          wrapWithSchemeIfComponent(
+            apiGen.getTypeFromResponse(response, "readOnly") || factory.createKeywordTypeNode(ts4.SyntaxKind.UndefinedKeyword)
+          )
         ]
       ).filter(
         ([status, response]) => isDataResponse(status, includeDefault, apiGen.resolve(response), responses || {})
-      ).filter(([_1, _2, type]) => type !== keywordType.void).map(
-        ([code, response, type]) => ts4.addSyntheticLeadingComment(
-          { ...type },
+      ).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,
           `* status ${code} ${response.description} `,
           false
-        )
-      );
+        );
+      });
       if (returnTypes.length > 0) {
         ResponseType = factory.createUnionTypeNode(returnTypes);
       }
@@ -595,7 +680,7 @@ async function generateApi(spec, {
         origin: "param",
         name,
         originalName: param.name,
-        type: apiGen.getTypeFromSchema(isReference(param) ? param : param.schema, void 0, "writeOnly"),
+        type: wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(isReference(param) ? param : param.schema, void 0, "writeOnly")),
         required: param.required,
         param
       };
@@ -603,7 +688,7 @@ async function generateApi(spec, {
     if (requestBody) {
       const body = apiGen.resolve(requestBody);
       const schema = apiGen.getSchemaFromContent(body.content);
-      const type = apiGen.getTypeFromSchema(schema);
+      const type = wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(schema));
       const schemaName = camelCase(
         type.name || getReferenceName(schema) || typeof schema === "object" && "title" in schema && schema.title || "body"
       );
@@ -612,7 +697,7 @@ async function generateApi(spec, {
         origin: "body",
         name,
         originalName: schemaName,
-        type: apiGen.getTypeFromSchema(schema, void 0, "writeOnly"),
+        type: wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(schema, void 0, "writeOnly")),
         required: true,
         body
       };
@@ -753,6 +838,49 @@ async function generateApi(spec, {
   function generateMutationEndpointProps({}) {
     return {};
   }
+  function wrapWithSchemeIfComponent(typeNode) {
+    if (ts4.isTypeReferenceNode(typeNode) && ts4.isIdentifier(typeNode.typeName)) {
+      const typeName = typeNode.typeName.text;
+      if (schemeTypeNames.has(typeName)) {
+        return factory.createTypeReferenceNode(
+          factory.createQualifiedName(
+            factory.createIdentifier("Scheme"),
+            typeNode.typeName
+          ),
+          typeNode.typeArguments?.map(wrapWithSchemeIfComponent)
+        );
+      }
+      if (typeNode.typeArguments) {
+        return factory.createTypeReferenceNode(
+          typeNode.typeName,
+          typeNode.typeArguments.map(wrapWithSchemeIfComponent)
+        );
+      }
+    }
+    if (ts4.isArrayTypeNode(typeNode)) {
+      return factory.createArrayTypeNode(wrapWithSchemeIfComponent(typeNode.elementType));
+    }
+    if (ts4.isUnionTypeNode(typeNode)) {
+      return factory.createUnionTypeNode(typeNode.types.map(wrapWithSchemeIfComponent));
+    }
+    if (ts4.isTypeLiteralNode(typeNode)) {
+      return factory.createTypeLiteralNode(
+        typeNode.members.map((member) => {
+          if (ts4.isPropertySignature(member) && member.type) {
+            return factory.updatePropertySignature(
+              member,
+              member.modifiers,
+              member.name,
+              member.questionToken,
+              wrapWithSchemeIfComponent(member.type)
+            );
+          }
+          return member;
+        })
+      );
+    }
+    return typeNode;
+  }
 }
 function generatePathExpression(path4, pathParameters, rootObject, isFlatArg, encodePathParams) {
   const expressions = [];
@@ -780,7 +908,55 @@ function generatePathExpression(path4, pathParameters, rootObject, isFlatArg, en
 }
 
 // src/index.ts
+import camelCase2 from "lodash.camelcase";
 var require2 = createRequire(__filename);
+async function ensureDirectoryExists(filePath) {
+  const dirname = path3.dirname(filePath);
+  if (!fs.existsSync(dirname)) {
+    await fs.promises.mkdir(dirname, { recursive: true });
+  }
+}
+function fileExists(filePath) {
+  try {
+    return fs.statSync(filePath).isFile();
+  } catch {
+    return false;
+  }
+}
+function getApiNameFromDir(dirPath) {
+  const dirName = path3.basename(dirPath);
+  return `${dirName}Api`;
+}
+async function ensureBaseFilesExist(outputDir) {
+  const enhanceEndpointsPath = path3.join(outputDir, "enhanceEndpoints.ts");
+  const indexPath = path3.join(outputDir, "index.ts");
+  const apiName = getApiNameFromDir(outputDir);
+  if (!fileExists(enhanceEndpointsPath)) {
+    const enhanceEndpointsContent = `import api from './query.generated';
+
+const enhancedApi = api.enhanceEndpoints({
+    endpoints: {
+    },
+});
+
+export default enhancedApi;
+`;
+    await fs.promises.writeFile(enhanceEndpointsPath, enhanceEndpointsContent, "utf-8");
+  }
+  if (!fileExists(indexPath)) {
+    const indexContent = `export * from './query.generated';
+export {default as ${apiName}} from './enhanceEndpoints';
+`;
+    await fs.promises.writeFile(indexPath, indexContent, "utf-8");
+  }
+}
+function getGroupNameFromPath(path4, pattern) {
+  const match = path4.match(pattern);
+  if (match && match[1]) {
+    return camelCase2(match[1]);
+  }
+  return "common";
+}
 async function generateEndpoints(options) {
   const schemaLocation = options.schemaFile;
   const schemaAbsPath = isValidUrl(options.schemaFile) ? options.schemaFile : path3.resolve(process.cwd(), schemaLocation);
@@ -789,8 +965,12 @@ async function generateEndpoints(options) {
   });
   const { outputFile, prettierConfigFile } = options;
   if (outputFile) {
+    const outputPath = path3.resolve(process.cwd(), outputFile);
+    await ensureDirectoryExists(outputPath);
+    const outputDir = path3.dirname(outputPath);
+    await ensureBaseFilesExist(outputDir);
     fs.writeFileSync(
-      path3.resolve(process.cwd(), outputFile),
+      outputPath,
       await prettify(outputFile, sourceCode, prettierConfigFile)
     );
   } else {
@@ -801,13 +981,29 @@ function parseConfig(fullConfig) {
   const outFiles = [];
   if ("outputFiles" in fullConfig) {
     const { outputFiles, ...commonConfig } = fullConfig;
-    for (const [outputFile, specificConfig] of Object.entries(outputFiles)) {
+    const openApiDoc = JSON.parse(fs.readFileSync(fullConfig.schemaFile, "utf-8"));
+    const paths = Object.keys(openApiDoc.paths);
+    const [outputPath, config] = Object.entries(outputFiles)[0];
+    const patterns = config.groupMatch;
+    const filterEndpoint = config.filterEndpoint;
+    const pattern = patterns;
+    const groupedPaths = paths.reduce((acc, path4) => {
+      const groupName = getGroupNameFromPath(path4, pattern);
+      if (!acc[groupName]) {
+        acc[groupName] = [];
+      }
+      acc[groupName].push(path4);
+      return acc;
+    }, {});
+    Object.entries(groupedPaths).forEach(([groupName, paths2]) => {
+      const finalOutputPath = outputPath.replace("$1", groupName);
+      const filterEndpoints = filterEndpoint(groupName);
       outFiles.push({
         ...commonConfig,
-        ...specificConfig,
-        outputFile
+        outputFile: finalOutputPath,
+        filterEndpoints: [filterEndpoints]
       });
-    }
+    });
   } else {
     outFiles.push(fullConfig);
   }