refactor(estree-to-babel) set-literal

coderaiser · Jan 28, 2021 · 88338b4 · 88338b4
1 parent 70a74f3
commit 88338b4
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/lib/set-literal.js b/lib/set-literal.js
@@ -13,17 +13,6 @@ module.exports = (node) => {
         return;
     }
 
-    if (node.regex) {
-        node.type = 'RegExpLiteral';
-
-        for (const key of Object.keys(node.regex)) {
-            node[key] = node.regex[key];
-        }
-
-        delete node.regex;
-        return;
-    }
-
     if (isString(value)) {
         node.type = 'StringLiteral';
         return;
@@ -38,4 +27,19 @@ module.exports = (node) => {
         node.type = 'BooleanLiteral';
         return;
     }
+
+    if (node.regex) {
+        transformRegExp(node);
+        return;
+    }
 };
+
+function transformRegExp(node) {
+    node.type = 'RegExpLiteral';
+
+    for (const key of Object.keys(node.regex)) {
+        node[key] = node.regex[key];
+    }
+
+    delete node.regex;
+}