feat: add supportParseEnumDescByReg func

Author: rookie-luochao

src/bin/openapi.ts

@@ -97,6 +97,10 @@ const params = program
     'parse enum description to generate enum label',
     false
   )
+  .option(
+    '--supportParseEnumDescByReg <string>',
+    'custom regex for parsing enum description, e.g., "(\\S+)\\s*=\\s*(\\S+)"'
+  )
   .parse(process.argv)
   .opts();
 
@@ -144,6 +148,9 @@ const baseGenerate = (_params_: OptionValues): GenerateServiceProps => {
     isCamelCase: JSON.parse(_params_.isCamelCase as string) === true,
     isSupportParseEnumDesc:
       JSON.parse(_params_.isSupportParseEnumDesc as string) === true,
+    supportParseEnumDescByReg: _params_.supportParseEnumDescByReg as
+      | string
+      | RegExp,
   };
 
   return options;

src/generator/serviceGenarator.ts

@@ -100,6 +100,7 @@ import {
   isSchemaObject,
   markAllowedSchema,
   parseDescriptionEnum,
+  parseDescriptionEnumByReg,
   replaceDot,
   resolveRefs,
   resolveTypeName,
@@ -1551,8 +1552,17 @@ export default class ServiceGenerator {
       }).join(',')}}`;
     } else {
       if (numberEnum.includes(schemaObject.type) || isAllNumber(enumArray)) {
-        if (this.config.isSupportParseEnumDesc && schemaObject.description) {
-          const enumMap = parseDescriptionEnum(schemaObject.description);
+        if (
+          (this.config.isSupportParseEnumDesc ||
+            this.config.supportParseEnumDescByReg) &&
+          schemaObject.description
+        ) {
+          const enumMap = this.config.isSupportParseEnumDesc
+            ? parseDescriptionEnum(schemaObject.description)
+            : parseDescriptionEnumByReg(
+                schemaObject.description,
+                this.config.supportParseEnumDescByReg
+              );
           enumLabelTypeStr = `{${map(enumArray, (value) => {
             const enumLabel = enumMap.get(Number(value));
             return `${Number(value)}:"${enumLabel}"`;

src/generator/util.ts

@@ -573,6 +573,57 @@ export const parseDescriptionEnum = (
   return enumMap;
 };
 
+/**
+ * 通过自定义正则表达式解析 description 中的枚举翻译
+ * @param description 描述文本
+ * @param regex 自定义正则表达式，用于匹配枚举项，例如：(\S+)\s*=\s*(\S+)
+ * @returns Map<number, string> 枚举值到标签的映射
+ */
+export const parseDescriptionEnumByReg = (
+  description: string,
+  regex: string | RegExp
+): Map<number, string> => {
+  const enumMap = new Map<number, string>();
+  if (!description) return enumMap;
+
+  // 将字符串正则转换为 RegExp 对象，确保有全局标志
+  let regExp: RegExp;
+  if (typeof regex === 'string') {
+    const flagsMatch = regex.match(/\/([gimsuy]*)$/);
+    if (flagsMatch) {
+      const parts = regex.split('/');
+      const pattern = parts.slice(1, -1).join('/');
+      const flags = flagsMatch[1] || '';
+      regExp = new RegExp(pattern, flags.includes('g') ? flags : flags + 'g');
+    } else {
+      regExp = new RegExp(regex, 'g');
+    }
+  } else {
+    const flags = regex.flags;
+    regExp = flags.includes('g')
+      ? regex
+      : new RegExp(regex.source, flags + 'g');
+  }
+
+  // 匹配所有结果，然后对每个匹配结果按 = 拆分
+  let match: RegExpExecArray | null;
+  while ((match = regExp.exec(description)) !== null) {
+    const fullMatch = match[0].trim();
+    // 按 = 拆分，左边是标签，右边是值
+    const parts = fullMatch.split(/\s*=\s*/);
+    if (parts.length >= 2) {
+      const label = parts[0].trim();
+      const valueStr = parts[1].trim();
+      const numValue = Number(valueStr);
+      if (label && valueStr && !isNaN(numValue)) {
+        enumMap.set(numValue, label);
+      }
+    }
+  }
+
+  return enumMap;
+};
+
 /**
  * 获取默认的二进制媒体类型列表
  */

src/index.ts

@@ -159,6 +159,12 @@ export type GenerateServiceProps = {
    * 是否使用 description 中的枚举定义
    */
   isSupportParseEnumDesc?: boolean;
+  /**
+   * 自定义正则表达式，用于解析 description 中的枚举定义
+   * 如果设置了此参数，将使用此正则表达式替代默认的 parseDescriptionEnum 方法
+   * 例如：/([^\s=<>/&;]+(?:\s+[^\s=<>/&;]+)*)\s*=\s*(\d+)/g 可以匹配 "普通 = 0" 或 "SampleMaker = 1" 这样的格式
+   */
+  supportParseEnumDescByReg?: string | RegExp;
   /**
    * 命名空间名称，默认为API，不需要关注
    */