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

package/src/services/api-code-generator.ts

@@ -60,8 +60,17 @@ export class ApiCodeGenerator {
     // 步驟 6: 收集操作名稱
     const operationNames = endpointInfos.map(info => info.operationName);
 
+    // 步驟 7: 收集所有 tags
+    const allTags = new Set<string>();
+    endpointInfos.forEach(info => {
+      if (info.tags && Array.isArray(info.tags)) {
+        info.tags.forEach((tag: string) => allTags.add(tag));
+      }
+    });
+
     return {
       operationNames,
+      tags: Array.from(allTags),
       files: {
         types: typesContent,
         queryService: rtkQueryContent, // RTK Query 檔案

package/src/services/unified-code-generator.ts

@@ -11,6 +11,7 @@ import { generateDoNotModifyFile } from '../generators/do-not-modify-generator';
 import type { GenerationOptions, CommonOptions } from '../types';
 import { ApiCodeGenerator } from './api-code-generator';
 import { generateUtilsFile } from '../generators/utils-generator';
+import { generateTagTypesFile } from '../generators/tag-types-generator';
 
 /**
  * 統一代碼生成器選項
@@ -68,14 +69,19 @@ export class UnifiedCodeGenerator {
     componentSchema: string;
     doNotModify: string;
     utils: string;
+    tagTypes: string;
   } = {
     groups: [],
     commonTypes: '',
     componentSchema: '',
     doNotModify: '',
-    utils: ''
+    utils: '',
+    tagTypes: ''
   };
 
+  // 收集所有 tags
+  private allTags: Set<string> = new Set();
+
 
   constructor(options: UnifiedGenerationOptions) {
     this._options = options;
@@ -98,6 +104,7 @@ export class UnifiedCodeGenerator {
     this.generateSchemaContent()
     this.generateUtilsContent()
     this.generateDoNotModifyContent()
+    this.generateTagTypesContent()
 
     return await this.release();
   }
@@ -176,6 +183,11 @@ export class UnifiedCodeGenerator {
             outputPath: groupInfo.outputPath,
             content: groupContent
           });
+
+          // 收集此群組的所有 tags
+          if (groupContent.tags && Array.isArray(groupContent.tags)) {
+            groupContent.tags.forEach((tag: string) => this.allTags.add(tag));
+          }
         }
         // 如果沒有任何 endpoint，則跳過此群組，不創建資料夾
       } catch (error) {
@@ -217,6 +229,14 @@ export class UnifiedCodeGenerator {
     this.generatedContent.utils = generateUtilsFile();
   }
 
+  /**
+   * 生成 Tag Types
+   */
+  private async generateTagTypesContent(): Promise<void> {
+    const tagsArray = Array.from(this.allTags);
+    this.generatedContent.tagTypes = generateTagTypesFile(tagsArray);
+  }
+
 
 
 
@@ -273,6 +293,15 @@ export class UnifiedCodeGenerator {
         results.push(...sharedResults);
       }
 
+      // 寫入 tagTypes.ts
+      if (this.generatedContent.tagTypes) {
+        const tagTypesResult = await this.fileWriterService.writeFile(
+          path.join(outputDir, 'tagTypes.ts'),
+          this.generatedContent.tagTypes
+        );
+        results.push(tagTypesResult);
+      }
+
       // 寫入 component schema
       if (this.generatedContent.componentSchema) {
         const schemaResults = await this.fileWriterService.writeSchemaFile(

package/src/types.ts

@@ -73,15 +73,6 @@ export interface CommonOptions {
     file: string;
     importReturnTypeName: string; // 用於指定別名導入，例如 IRestFulEndpointsQueryReturn
   };
-  /**
-   * Cache tag types configuration for RTK Query cache invalidation
-   * If provided, will import the specified type from the given file in enhanceEndpoints.ts
-   * Example: { file: "@/store/tagTypes", importName: "ECacheTagTypes" }
-   */
-  cacheTagTypes?: {
-    file: string;
-    importReturnTypeName: string;
-  };
   /**
    * defaults to "enhancedApi"
    */
@@ -206,6 +197,7 @@ export type ConfigFile =
 
 export type GenerateApiResult = {
   operationNames: string[];
+  tags: string[]; // 收集到的所有 tags
   files: {
     types: string;
     queryService: string; // RTK Query generated file