feat: classNameMode 支持 回调函数的形式

Author: WangMingHua111

README.md

@@ -41,7 +41,7 @@ export interface ISwaggerOptions {
   serviceNameSuffix?: string
   enumNamePrefix?: string
   methodNameMode?: 'operationId' | 'path' | 'shortOperationId' | ((reqProps: IRequestMethod) => string)
-  classNameMode?: 'parentPath' | 'normal'
+  classNameMode?: 'parentPath' | 'normal' | ((path: string, tag: string[]) => string)
   /** only effect classNameMode='parentPath' */
   pathClassNameDefaultName?: string
   outputDir?: string

src/baseInterfaces.ts

@@ -4,7 +4,7 @@ export interface ISwaggerOptions {
   serviceNameSuffix?: string
   enumNamePrefix?: string
   methodNameMode?: 'operationId' | 'path' | 'shortOperationId' | ((reqProps: IRequestMethod) => string)
-  classNameMode?: 'parentPath' | 'normal'
+  classNameMode?: 'parentPath' | 'normal' | ((path: string, tag: string[]) => string)
   /** only effect classNameMode='parentPath' */
   pathClassNameDefaultName?: string
   outputDir?: string

src/requestCodegen/index.ts

@@ -1,11 +1,11 @@
-import { getClassNameByPath, getMethodNameByPath, RemoveSpecialCharacters } from '../utils'
-import { IPaths } from '../swaggerInterfaces'
-import { getRequestParameters } from './getRequestParameters'
-import { getResponseType } from './getResponseType'
 import camelcase from 'camelcase'
-import { getRequestBody } from './getRequestBody'
 import { ISwaggerOptions } from '../baseInterfaces'
+import { IPaths } from '../swaggerInterfaces'
+import { getClassNameByPath, getMethodNameByPath, RemoveSpecialCharacters } from '../utils'
 import { getContentType } from './getContentType'
+import { getRequestBody } from './getRequestBody'
+import { getRequestParameters } from './getRequestParameters'
+import { getResponseType } from './getResponseType'
 import { mapFormDataToV2 } from './mapFormDataToV2'
 
 export interface IRequestClass {
@@ -29,10 +29,10 @@ export function requestCodegen(paths: IPaths, isV3: boolean, options: ISwaggerOp
           options.methodNameMode === 'operationId'
             ? reqProps.operationId
             : options.methodNameMode === 'shortOperationId'
-              ? trimSuffix(reqProps.operationId, reqProps.tags?.[0])
-              : typeof options.methodNameMode === 'function'
-                ? options.methodNameMode(reqProps)
-                : methodName
+            ? trimSuffix(reqProps.operationId, reqProps.tags?.[0])
+            : typeof options.methodNameMode === 'function'
+            ? options.methodNameMode(reqProps)
+            : methodName
         if (!methodName) {
           // console.warn('method Name is null：', path);
           continue
@@ -50,6 +50,8 @@ export function requestCodegen(paths: IPaths, isV3: boolean, options: ISwaggerOp
           } else {
             className = camelcase(className, { pascalCase: true })
           }
+        } else if (typeof options.classNameMode === 'function') {
+          className = options.classNameMode(path, reqProps.tags || [])
         } else {
           if (!reqProps.tags) continue
           className = camelcase(RemoveSpecialCharacters(reqProps.tags[0]), { pascalCase: true })
@@ -59,8 +61,6 @@ export function requestCodegen(paths: IPaths, isV3: boolean, options: ISwaggerOp
         // 如果是数字开头，则加上下划线
         if (!Number.isNaN(Number(className[0]))) className = '_' + className
 
-
-
         // 是否存在
         if (!requestClasses[className]) {
           requestClasses[className] = []
@@ -138,7 +138,7 @@ export function requestCodegen(paths: IPaths, isV3: boolean, options: ISwaggerOp
           }
         } catch (error) {
           // 获取方法名字失败，跳过该请求
-          console.log('error request, url: ', path);
+          console.log('error request, url: ', path)
           continue
         }