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

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

@@ -206,6 +206,7 @@ export type ConfigFile =
 
 export type GenerateApiResult = {
   operationNames: string[];
+  tags: string[]; // 收集到的所有 tags
   files: {
     types: string;
     queryService: string; // RTK Query generated file