feature: @putout/plugin-exptract-keywrods-from-variables: ArrowFunctionExpression

Author: coderaiser

packages/plugin-extract-keywords-from-variables/README.md

@@ -32,6 +32,11 @@ export const isTemplateTail = (a) => a?.type === 'TemplateTail';
 
 -const a 5;
 +const a = 5;
+
+-export const packContent = (content) {
++export const packContent = (content) => {
+    console.log(a);
+}
 ```
 
 ## License

packages/plugin-extract-keywords-from-variables/lib/extract-keywords-from-variables.js

@@ -2,6 +2,8 @@
 
 const {types, operator} = require('putout');
 const {
+    isExportDeclaration,
+    ArrowFunctionExpression,
     isArrowFunctionExpression,
     isLiteral,
     IfStatement,
@@ -14,9 +16,11 @@ const {
     isExportNamedDeclaration,
     isIdentifier,
     isMemberExpression,
+    isBlockStatement,
 } = types;
 
 const {
+    removeParens,
     remove,
     replaceWith,
     isDeclarationKeyword,
@@ -31,7 +35,13 @@ const buildDeclaration = (type) => (nextPath, path) => {
     let left;
     let right;
 
-    if (isAssignmentExpression(expression)) {
+    if (isBlockStatement(nextPath)) {
+        left = path.node.id;
+        const {node: init} = removeParens(path.get('init'));
+        const params = [init];
+        
+        right = ArrowFunctionExpression(params, nextPath.node);
+    } else if (isAssignmentExpression(expression)) {
         ({
             left,
             right,
@@ -110,6 +120,19 @@ module.exports.traverse = ({push}) => ({
                     path,
                     nextPath,
                 });
+            
+            return;
+        }
+        
+        if (isExportDeclaration(path.parentPath.parentPath)) {
+            const nextPath = path.parentPath.parentPath.getNextSibling();
+            
+            if (isBlockStatement(nextPath))
+                push({
+                    name: kind,
+                    path,
+                    nextPath,
+                });
         }
     },
 });

packages/plugin-extract-keywords-from-variables/test/fixture/arrow-fix.js

@@ -1,2 +1,6 @@
 const a = (x) => x + 1;
 fn(a);
+
+export const packContent = (content) => {
+    console.log(a);
+};

packages/plugin-extract-keywords-from-variables/test/fixture/arrow.js

@@ -1,2 +1,6 @@
 const a (x) => x + 1;
 fn(a);
+
+export const packContent = (content) {
+    console.log(a);
+}