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

package/src/generators/react-hooks.ts

@@ -20,7 +20,7 @@ type CreateBindingParams = {
 };
 
 const createBinding = ({
-  operationDefinition: { verb, path, operation },
+  operationDefinition: { verb, path },
   overrides,
   isLazy = false,
 }: CreateBindingParams) =>
@@ -28,7 +28,7 @@ const createBinding = ({
     undefined,
     undefined,
     factory.createIdentifier(
-      `use${isLazy ? 'Lazy' : ''}${capitalize(getOperationName(verb, path, operation.operationId))}${
+      `use${isLazy ? 'Lazy' : ''}${capitalize(getOperationName(verb, path, undefined))}${
         isQuery(verb, overrides) ? 'Query' : 'Mutation'
       }`
     ),

package/src/index.ts

@@ -4,10 +4,84 @@ import path from 'node:path';
 import { generateApi } from './generate';
 import type { CommonOptions, ConfigFile, GenerationOptions, OutputFileOptions } from './types';
 import { isValidUrl, prettify } from './utils';
-export type { ConfigFile } from './types';
+import camelCase from 'lodash.camelcase';
+export type { OutputFilesConfig, ConfigFile } from './types';
 
 const require = createRequire(__filename);
 
+
+
+// 確保目錄存在的函數
+async function ensureDirectoryExists(filePath: string) {
+  const dirname = path.dirname(filePath);
+  if (!fs.existsSync(dirname)) {
+    await fs.promises.mkdir(dirname, { recursive: true });
+  }
+}
+
+
+// 檢查檔案是否存在的函數
+function fileExists(filePath: string): boolean {
+  try {
+    return fs.statSync(filePath).isFile();
+  } catch {
+    return false;
+  }
+}
+
+// 獲取資料夾名稱並轉換為 API 名稱
+function getApiNameFromDir(dirPath: string): string {
+  const dirName = path.basename(dirPath);
+  return `${dirName}Api`;
+}
+
+// 確保基礎文件存在的函數
+async function ensureBaseFilesExist(outputDir: string) {
+  const enhanceEndpointsPath = path.join(outputDir, 'enhanceEndpoints.ts');
+  const indexPath = path.join(outputDir, 'index.ts');
+  const apiName = getApiNameFromDir(outputDir);
+
+  // 如果 enhanceEndpoints.ts 不存在，創建它
+  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');
+  }
+
+  // 如果 index.ts 不存在，創建它
+  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(path: string, pattern: RegExp): string {
+  // console.log('pattern', pattern);
+
+  const match = path.match(pattern);
+  // console.log('match', path, match);
+
+  if (match && match[1]) {
+    return camelCase(match[1]);
+  }
+  return 'common';
+}
+
+
+
+
+
 export async function generateEndpoints(options: GenerationOptions): Promise<string | void> {
   const schemaLocation = options.schemaFile;
 
@@ -20,8 +94,15 @@ export async function generateEndpoints(options: GenerationOptions): Promise<str
   });
   const { outputFile, prettierConfigFile } = options;
   if (outputFile) {
+    const outputPath = path.resolve(process.cwd(), outputFile);
+    await ensureDirectoryExists(outputPath);
+
+    // 確保基礎文件存在
+    const outputDir = path.dirname(outputPath);
+    await ensureBaseFilesExist(outputDir);
+
     fs.writeFileSync(
-      path.resolve(process.cwd(), outputFile),
+      outputPath,
       await prettify(outputFile, sourceCode, prettierConfigFile)
     );
   } else {
@@ -29,18 +110,49 @@ export async function generateEndpoints(options: GenerationOptions): Promise<str
   }
 }
 
+
 export function parseConfig(fullConfig: ConfigFile) {
   const outFiles: (CommonOptions & OutputFileOptions)[] = [];
 
   if ('outputFiles' in fullConfig) {
     const { outputFiles, ...commonConfig } = fullConfig;
-    for (const [outputFile, specificConfig] of Object.entries(outputFiles)) {
-      outFiles.push({
-        ...commonConfig,
-        ...specificConfig,
-        outputFile,
+
+    // 讀取 OpenAPI 文檔
+    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, path) => {
+
+        const groupName = getGroupNameFromPath(path, pattern);
+        if (!acc[groupName]) {
+          acc[groupName] = [];
+        }
+        acc[groupName].push(path);
+        return acc;
+      }, {} as Record<string, string[]>);
+
+      // 為每個分類生成配置
+      Object.entries(groupedPaths).forEach(([groupName, paths]) => {
+        const finalOutputPath = outputPath.replace('$1', groupName);
+
+        const filterEndpoints = filterEndpoint(groupName);
+        outFiles.push({
+          ...commonConfig,
+          outputFile: finalOutputPath,
+          filterEndpoints: [filterEndpoints],
+        });
       });
-    }
+
   } else {
     outFiles.push(fullConfig);
   }

package/src/types.ts

@@ -132,6 +132,7 @@ export interface OutputFileOptions extends Partial<CommonOptions> {
    * If passed as true it will generate TS enums instead of union of strings
    */
   useEnumType?: boolean;
+  sharedTypesFile?: string;
 }
 
 export type EndpointOverrides = {
@@ -141,10 +142,18 @@ export type EndpointOverrides = {
   parameterFilter: ParameterMatcher;
 }>;
 
+export type OutputFilesConfig = {
+  [outputFile: string]: {
+      groupMatch: RegExp,
+      filterEndpoint: (groupName: string) => RegExp
+    }
+};
+
 export type ConfigFile =
   | Id<Require<CommonOptions & OutputFileOptions, 'outputFile'>>
   | Id<
       Omit<CommonOptions, 'outputFile'> & {
-        outputFiles: { [outputFile: string]: Omit<OutputFileOptions, 'outputFile'> };
+        // outputFiles: { [outputFile: string]: Omit<OutputFileOptions, 'outputFile'> };
+        outputFiles: OutputFilesConfig
       }
     >;