fix: support chinese for enum

ferdikoomen · Nov 28, 2023 · e65de8b · e65de8b
1 parent c67d332
commit e65de8b
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/src/openApi/v2/parser/getEnum.ts b/src/openApi/v2/parser/getEnum.ts
@@ -20,7 +20,7 @@ export const getEnum = (values?: (string | number)[]): Enum[] => {
                 }
                 return {
                     name: String(value)
-                        .replace(/\W+/g, '_')
+                        .replace(/[^a-zA-Z0-9_\u4e00-\u9fa5]+/g, '_')
                         .replace(/^(\d+)/g, '_$1')
                         .replace(/([a-z])([A-Z]+)/g, '$1_$2')
                         .toUpperCase(),

diff --git a/src/openApi/v3/parser/getEnum.ts b/src/openApi/v3/parser/getEnum.ts
@@ -20,7 +20,7 @@ export const getEnum = (values?: (string | number)[]): Enum[] => {
                 }
                 return {
                     name: String(value)
-                        .replace(/\W+/g, '_')
+                        .replace(/[^a-zA-Z0-9_\u4e00-\u9fa5]+/g, '_')
                         .replace(/^(\d+)/g, '_$1')
                         .replace(/([a-z])([A-Z]+)/g, '$1_$2')
                         .toUpperCase(),

diff --git a/test/spec/v2.json b/test/spec/v2.json
@@ -965,7 +965,8 @@
                 "Warning",
                 "Error",
                 "'Single Quote'",
-                "\"Double Quotes\""
+                "\"Double Quotes\"",
+                "你好"
             ]
         },
         "EnumWithNumbers": {

diff --git a/test/spec/v3.json b/test/spec/v3.json
@@ -1561,9 +1561,11 @@
                     "Warning",
                     "Error",
                     "'Single Quote'",
-                    "\"Double Quotes\""
+                    "\"Double Quotes\"",
+                    "你好"
                 ]
             },
+
             "EnumWithNumbers": {
                 "description": "This is a simple enum with numbers",
                 "enum": [