@acrool/rtk-query-codegen-openapi 0.0.2-test.6 → 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 || [] : [];
@@ -441,41 +441,61 @@ 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 resultFile2 = ts4.createSourceFile(
-        "sharedTypes.ts",
-        "",
-        ts4.ScriptTarget.Latest,
-        /*setParentNodes*/
-        false,
-        ts4.ScriptKind.TS
-      );
-      const printer2 = ts4.createPrinter({ newLine: ts4.NewLineKind.LineFeed });
-      const typeDefinitions = Object.entries(components).flatMap(([_, componentDefs]) => {
-        return Object.entries(componentDefs).map(([name, def]) => {
-          const typeNode = apiGen.getTypeFromSchema(def);
+      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(name),
+            factory.createIdentifier(typeName),
             void 0,
             typeNode
           );
         });
+        return factory.createModuleDeclaration(
+          [factory.createModifier(ts4.SyntaxKind.ExportKeyword)],
+          factory.createIdentifier("Scheme"),
+          factory.createModuleBlock(typeEntries),
+          ts4.NodeFlags.Namespace
+        );
       });
-      const output = printer2.printNode(
-        ts4.EmitHint.Unspecified,
-        factory.createSourceFile(
-          typeDefinitions,
-          factory.createToken(ts4.SyntaxKind.EndOfFileToken),
-          ts4.NodeFlags.None
-        ),
-        resultFile2
-      );
-      const fs2 = await import("node:fs/promises");
-      await fs2.writeFile(sharedTypesFile, output, "utf-8");
+      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);
@@ -506,34 +526,20 @@ async function generateApi(spec, {
       apiFile = apiFile.replace(/\\/g, "/");
       if (!apiFile.startsWith(".")) apiFile = `./${apiFile}`;
     }
-    if (sharedTypesFile && sharedTypesFile.startsWith(".")) {
-      sharedTypesFile = path2.relative(path2.dirname(outputFile), sharedTypesFile);
-      sharedTypesFile = sharedTypesFile.replace(/\\/g, "/");
-      if (!sharedTypesFile.startsWith(".")) sharedTypesFile = `./${sharedTypesFile}`;
-    }
   }
   apiFile = apiFile.replace(/\.[jt]sx?$/, "");
-  if (sharedTypesFile) {
-    sharedTypesFile = sharedTypesFile.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 ? [
-          factory.createImportDeclaration(
-            void 0,
-            factory.createImportClause(
-              false,
-              void 0,
-              factory.createNamespaceImport(factory.createIdentifier("SharedTypes"))
-            ),
-            factory.createStringLiteral(sharedTypesFile),
-            void 0
-          )
-        ] : [],
+        ...sharedTypesFile ? [generateImportNode(sharedTypesImportPath, { Scheme: "Scheme" })] : [],
         ...tag ? [generateTagTypes({ addTagTypes: extractAllTagTypes({ operationDefinitions }) })] : [],
         generateCreateApiCall({
           tag,
@@ -541,7 +547,8 @@ async function generateApi(spec, {
             operationDefinitions.map(
               (operationDefinition) => generateEndpoint({
                 operationDefinition,
-                overrides: getOverrides(operationDefinition, endpointOverrides)
+                overrides: getOverrides(operationDefinition, endpointOverrides),
+                sharedTypesFile: !!sharedTypesFile
               })
             ),
             true
@@ -580,7 +587,8 @@ async function generateApi(spec, {
   }
   function generateEndpoint({
     operationDefinition,
-    overrides
+    overrides,
+    sharedTypesFile: sharedTypesFile2
   }) {
     const {
       verb,
@@ -589,60 +597,49 @@ 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";
     let ResponseType = factory.createKeywordTypeNode(ts4.SyntaxKind.UnknownKeyword);
-    function replaceReferences(schema) {
-      if (!schema) return factory.createKeywordTypeNode(ts4.SyntaxKind.UnknownKeyword);
-      const refName = getReferenceName(schema);
-      if (refName && sharedTypesFile) {
-        return factory.createTypeReferenceNode(
-          factory.createQualifiedName(
-            factory.createIdentifier("SharedTypes"),
-            factory.createIdentifier(refName)
-          ),
-          void 0
-        );
-      }
-      if (schema.type === "object" && schema.properties) {
-        const members = Object.entries(schema.properties).map(([key, value]) => {
-          return factory.createPropertySignature(
-            void 0,
-            factory.createIdentifier(key),
-            schema.required?.includes(key) ? void 0 : factory.createToken(ts4.SyntaxKind.QuestionToken),
-            replaceReferences(value)
-          );
-        });
-        return factory.createTypeLiteralNode(members);
-      }
-      if (schema.type === "array" && schema.items) {
-        return factory.createArrayTypeNode(replaceReferences(schema.items));
-      }
-      return apiGen.getTypeFromSchema(schema);
-    }
     if (returnsJson) {
       const returnTypes = Object.entries(responses || {}).map(
-        ([code, response]) => {
-          const resolvedResponse = apiGen.resolve(response);
-          if (!resolvedResponse.content?.["application/json"]?.schema) {
-            return [code, resolvedResponse, factory.createKeywordTypeNode(ts4.SyntaxKind.UndefinedKeyword)];
-          }
-          const schema = resolvedResponse.content["application/json"].schema;
-          const type = replaceReferences(schema);
-          return [code, resolvedResponse, type];
-        }
+        ([code, response]) => [
+          code,
+          apiGen.resolve(response),
+          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);
       }
@@ -677,10 +674,21 @@ async function generateApi(spec, {
       }
       return name;
     }
+    for (const param of parameters) {
+      const name = generateName(param.name, param.in);
+      queryArg[name] = {
+        origin: "param",
+        name,
+        originalName: param.name,
+        type: wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(isReference(param) ? param : param.schema, void 0, "writeOnly")),
+        required: param.required,
+        param
+      };
+    }
     if (requestBody) {
       const body = apiGen.resolve(requestBody);
       const schema = apiGen.getSchemaFromContent(body.content);
-      const type = replaceReferences(schema);
+      const type = wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(schema));
       const schemaName = camelCase(
         type.name || getReferenceName(schema) || typeof schema === "object" && "title" in schema && schema.title || "body"
       );
@@ -689,24 +697,11 @@ async function generateApi(spec, {
         origin: "body",
         name,
         originalName: schemaName,
-        type,
+        type: wrapWithSchemeIfComponent(apiGen.getTypeFromSchema(schema, void 0, "writeOnly")),
         required: true,
         body
       };
     }
-    for (const param of parameters) {
-      const name = generateName(param.name, param.in);
-      const paramSchema = isReference(param) ? param : param.schema;
-      const type = replaceReferences(paramSchema);
-      queryArg[name] = {
-        origin: "param",
-        name,
-        originalName: param.name,
-        type,
-        required: param.required,
-        param
-      };
-    }
     const propertyName = (name) => {
       if (typeof name === "string") {
         return isValidIdentifier(name) ? factory.createIdentifier(name) : factory.createStringLiteral(name);
@@ -843,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 = [];
@@ -870,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);
@@ -879,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 {
@@ -891,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);
   }