@acrool/rtk-query-codegen-openapi 1.2.0-alpha.0 → 1.2.0-alpha.2

package/lib/index.mjs

@@ -905,7 +905,7 @@ function getTypeFromParameter(param, schemaTypeMap = {}) {
 init_esm_shims();
 function generateRtkQueryFile(endpointInfos, options) {
   const { groupKey } = options;
-  const httpClientTypeName = options.httpClient?.importReturnTypeName || options.httpClient?.importName || "IRestFulEndpointsQueryReturn";
+  const httpClientTypeName = options.httpClient?.importReturnTypeName || "IRestFulEndpointsQueryReturn";
   const endpoints = endpointInfos.map((info) => {
     const methodType = info.isQuery ? "query" : "mutation";
     const argType = info.isVoidArg ? "void" : `${httpClientTypeName}<${info.argTypeName}>`;
@@ -930,7 +930,7 @@ ${paramsLines}
     }
     let tagsSection = "";
     if (info.tags && info.tags.length > 0) {
-      const tagsArray = info.tags.map((tag) => `"${tag}"`).join(", ");
+      const tagsArray = info.tags.map((tag) => `ECacheTagTypes.${tag}`).join(", ");
       if (info.isQuery) {
         tagsSection = `
             providesTags: [${tagsArray}],`;
@@ -966,10 +966,13 @@ ${paramsLines}
   const httpClientImport = options.httpClient ? `import {${options.httpClient.importReturnTypeName}} from "${options.httpClient.file}";
 ` : `import {IRestFulEndpointsQueryReturn} from "@acrool/react-fetcher";
 `;
+  const hasTags = endpointInfos.some((info) => info.tags && info.tags.length > 0);
+  const tagTypesImport = hasTags ? `import {ECacheTagTypes} from "../tagTypes";
+` : "";
   return `/* eslint-disable */
 // [Warning] Generated automatically - do not edit manually
 
-${apiImport}${httpClientImport}
+${apiImport}${httpClientImport}${tagTypesImport}
 ${typeImportStatement}
 
 
@@ -1005,7 +1008,6 @@ export default injectedRtkApi;
 // src/generators/rtk-enhance-endpoints-generator.ts
 init_esm_shims();
 function generateRtkEnhanceEndpointsFile(endpointInfos, options) {
-  const { groupKey, cacheTagTypes } = options;
   const endpointConfigs = endpointInfos.map((info) => {
     if (info.isQuery) {
       return `    ${info.operationName}: {
@@ -1017,15 +1019,10 @@ function generateRtkEnhanceEndpointsFile(endpointInfos, options) {
     },`;
     }
   }).join("\n");
-  let importStatements = `/* eslint-disable */
+  const importStatements = `/* eslint-disable */
 // [Warning] Generated automatically - do not edit manually
 
-`;
-  if (cacheTagTypes) {
-    importStatements += `import { ${cacheTagTypes.importReturnTypeName} } from "${cacheTagTypes.file}";
-`;
-  }
-  importStatements += `import api from "./query.generated";
+import api from "./query.generated";
 `;
   return `${importStatements}
 const enhancedApi = api.enhanceEndpoints({
@@ -1063,8 +1060,15 @@ var ApiCodeGenerator = class {
     const enhanceEndpointsContent = generateRtkEnhanceEndpointsFile(endpointInfos, this.options);
     const indexContent = this.generateIndex();
     const operationNames = endpointInfos.map((info) => info.operationName);
+    const allTags = /* @__PURE__ */ new Set();
+    endpointInfos.forEach((info) => {
+      if (info.tags && Array.isArray(info.tags)) {
+        info.tags.forEach((tag) => allTags.add(tag));
+      }
+    });
     return {
       operationNames,
+      tags: Array.from(allTags),
       files: {
         types: typesContent,
         queryService: rtkQueryContent,
@@ -1148,6 +1152,27 @@ export function withoutUndefined(obj?: Record<string, any>) {
 `;
 }
 
+// src/generators/tag-types-generator.ts
+init_esm_shims();
+function generateTagTypesFile(tags) {
+  if (tags.length === 0) {
+    return `/* eslint-disable */
+// [Warning] Generated automatically - do not edit manually
+
+export enum ECacheTagTypes {
+}
+`;
+  }
+  const enumEntries = tags.sort().map((tag) => `    ${tag} = '${tag}',`).join("\n");
+  return `/* eslint-disable */
+// [Warning] Generated automatically - do not edit manually
+
+export enum ECacheTagTypes {
+${enumEntries}
+}
+`;
+}
+
 // src/services/unified-code-generator.ts
 var UnifiedCodeGenerator = class {
   _options;
@@ -1165,8 +1190,11 @@ var UnifiedCodeGenerator = class {
     commonTypes: "",
     componentSchema: "",
     doNotModify: "",
-    utils: ""
+    utils: "",
+    tagTypes: ""
   };
+  // 收集所有 tags
+  allTags = /* @__PURE__ */ new Set();
   constructor(options) {
     this._options = options;
   }
@@ -1180,6 +1208,7 @@ var UnifiedCodeGenerator = class {
     this.generateSchemaContent();
     this.generateUtilsContent();
     this.generateDoNotModifyContent();
+    this.generateTagTypesContent();
     return await this.release();
   }
   /**
@@ -1232,6 +1261,9 @@ var UnifiedCodeGenerator = class {
             outputPath: groupInfo.outputPath,
             content: groupContent
           });
+          if (groupContent.tags && Array.isArray(groupContent.tags)) {
+            groupContent.tags.forEach((tag) => this.allTags.add(tag));
+          }
         }
       } catch (error) {
         throw new Error(`\u7FA4\u7D44 ${groupInfo.groupKey} \u751F\u6210\u5931\u6557: ${error}`);
@@ -1262,6 +1294,13 @@ var UnifiedCodeGenerator = class {
   async generateUtilsContent() {
     this.generatedContent.utils = generateUtilsFile();
   }
+  /**
+   * 生成 Tag Types
+   */
+  async generateTagTypesContent() {
+    const tagsArray = Array.from(this.allTags);
+    this.generatedContent.tagTypes = generateTagTypesFile(tagsArray);
+  }
   /**
    * 發佈階段：統一寫入所有檔案
    */
@@ -1302,6 +1341,13 @@ var UnifiedCodeGenerator = class {
         );
         results.push(...sharedResults);
       }
+      if (this.generatedContent.tagTypes) {
+        const tagTypesResult = await this.fileWriterService.writeFile(
+          path4.join(outputDir, "tagTypes.ts"),
+          this.generatedContent.tagTypes
+        );
+        results.push(tagTypesResult);
+      }
       if (this.generatedContent.componentSchema) {
         const schemaResults = await this.fileWriterService.writeSchemaFile(
           outputDir,