feature(@putout/plugin-madrun) convert-run-to-cut-env: add ability to convert body to array

Author: coderaiser

packages/plugin-madrun/lib/convert-run-to-cut-env/fixture/no-env-fix.js

@@ -0,0 +1,4 @@
+export default {
+    'test': () => [env, 'npm test'],
+    'coverage': async () => [env, `c8 ${await cutEnv('test')}`],
+}

packages/plugin-madrun/lib/convert-run-to-cut-env/fixture/no-env.js

@@ -0,0 +1,4 @@
+export default {
+    'test': () => [env, 'npm test'],
+    'coverage': async () => `c8 ${await run('test')}`,
+}

packages/plugin-madrun/lib/convert-run-to-cut-env/index.js

@@ -1,7 +1,10 @@
 'use strict';
 
 const {types, operator} = require('putout');
-const {isArrayExpression} = types;
+const {
+    isArrayExpression,
+    ArrayExpression,
+} = types;
 const {
     getExportDefault,
     getProperty,
@@ -10,25 +13,39 @@ const {
 module.exports.report = () => `Use 'cutEnv()' instead of 'run()'`;
 
 module.exports.replace = () => ({
-    'run(__a)': 'cutEnv(__a)',
-});
-
-module.exports.match = () => ({
-    'run(__a)': ({__a}, path) => {
-        const exportDefault = getExportDefault(path);
-        
-        if (!exportDefault)
-            return false;
-        
-        const declarationPath = exportDefault.get('declaration');
-        const property = getProperty(declarationPath, __a.value);
+    'run(__a)': (vars, path) => {
+        const {body} = path.scope.block;
 
-        if (!property)
-            return false;
+        if (!isArrayExpression(body)) {
+            const tuple = getTuple(vars, path);
+            path.scope.block.body = ArrayExpression([tuple.elements[0], body]);
+        }
 
-        const {body} = property.node.value;
-        
-        return isArrayExpression(body);
+        return 'cutEnv(__a)';
     },
 });
 
+module.exports.match = () => ({
+    'run(__a)': getTuple,
+});
+
+function getTuple({__a}, path) {
+    const exportDefault = getExportDefault(path);
+    
+    if (!exportDefault)
+        return null;
+    
+    const declarationPath = exportDefault.get('declaration');
+    const property = getProperty(declarationPath, __a.value);
+    
+    if (!property)
+        return null;
+    
+    const {body} = property.node.value;
+    
+    if (!isArrayExpression(body))
+        return null;
+    
+    return body;
+}
+

packages/plugin-madrun/lib/convert-run-to-cut-env/index.spec.js

@@ -17,6 +17,11 @@ test('madrun: convert-run-to-cut-env: transform', (t) => {
     t.end();
 });
 
+test('madrun: convert-run-to-cut-env: transform: no-env', (t) => {
+    t.transform('no-env');
+    t.end();
+});
+
 test('madrun: convert-run-to-cut-env: no transform: no export default', (t) => {
     t.noTransformCode(`export const hello = 'world'; run('hello')`);
     t.end();