@putout/plugin-putout 18.3.0 → 18.4.1

package/README.md

@@ -29,6 +29,7 @@ npm i @putout/plugin-putout -D
         "putout/add-args": "on",
         "putout/add-push": "on",
         "putout/add-track-file": "on",
+        "putout/add-await-to-progress": "on",
         "putout/add-index-to-import": "on",
         "putout/check-match": "on",
         "putout/check-replace-code": "on",
@@ -768,6 +769,26 @@ module.exports.traverse = ({push}) => ({
 });
 ```
 
+## add-await-to-progress
+
+Checkout in 🐊[**Putout Editor**](https://putout.cloudcmd.io/#/gist/978d70945edfa369390ea059654ff04d/89225c04039b9c0e9057aad34852c9428264f119).
+
+### ❌ Example of incorrect code
+
+```js
+test('', ({progress}) => {
+    progress();
+});
+```
+
+### ✅ Example of correct code
+
+```js
+test('', async ({progress}) => {
+    await progress();
+});
+```
+
 ## add-track-file
 
 Checkout in 🐊[**Putout Editor**](https://putout.cloudcmd.io/#/gist/faaba1ce41e6fd274bc82a8875a52bfa/b34a22fdf9080e6b2f06703760f629d83d69ff3d).

package/lib/add-args/index.js

@@ -19,4 +19,9 @@ module.exports = addArgs({
         'test.skip("__a", async (__args) => __body)',
         'test.only("__a", async (__args) => __body)',
     ]],
+    progress: ['{progress}', [
+        'test("__a", async (__args) => __body)',
+        'test.skip("__a", async (__args) => __body)',
+        'test.only("__a", async (__args) => __body)',
+    ]],
 });

package/lib/add-await-to-progress/index.js

@@ -0,0 +1,54 @@
+'use strict';
+
+const {types, operator} = require('putout');
+const {
+    Identifier,
+    ObjectPattern,
+    ObjectProperty,
+} = types;
+
+const {remove, compare} = operator;
+const isInsideTest = ({parentPath}) => compare(parentPath, 'test(__a, __b)');
+
+const checkAwait = (vars, path) => {
+    if (!path.find(isInsideTest))
+        return false;
+    
+    return !path.parentPath.isAwaitExpression();
+};
+
+module.exports.report = () => `Add 'await' to operator 'progress()'`;
+
+module.exports.match = () => ({
+    'progress(__args)': checkAwait,
+    't.progress(__args)': checkAwait,
+});
+
+module.exports.replace = () => ({
+    'test(__a, async (t, {progress}) => __body)': 'test(__a, async ({progress}) => __body)',
+    'progress(__args)': addAwait,
+    't.progress(__args)': addAwait,
+});
+
+function addAwait(vars, path) {
+    const next = path.parentPath.getNextSibling();
+    
+    if (compare(next, 't.end()'))
+        remove(next);
+    
+    const {params} = path.scope.block;
+    const [first] = params;
+    
+    if (first.name === 't') {
+        const id = Identifier('progress');
+        
+        path.scope.block.params = [
+            ObjectPattern([
+                ObjectProperty(id, id, false, true),
+            ]),
+        ];
+    }
+    
+    path.scope.block.async = true;
+    return 'await progress(__args)';
+}

package/lib/convert-progress-to-track-file/index.js

@@ -29,7 +29,11 @@ module.exports.traverse = ({push}) => ({
         if (!statement)
             return;
         
-        const forOfCount = statement.parentPath.get('body').filter(isForOfStatement).length;
+        const forOfCount = statement
+            .parentPath
+            .get('body')
+            .filter(isForOfStatement)
+            .length;
         
         if (forOfCount > 1)
             return;

package/lib/index.js

@@ -46,6 +46,7 @@ const applyRename = require('./apply-rename');
 const applyShortProcessors = require('./apply-short-processors');
 const addTrackFile = require('./add-track-file');
 const convertProgressToTrackFile = require('./convert-progress-to-track-file');
+const addAwaitToProgress = require('./add-await-to-progress');
 
 module.exports.rules = {
     'apply-processors-destructuring': applyProcessorsDestructuring,
@@ -94,4 +95,5 @@ module.exports.rules = {
     'convert-traverse-to-scan': convertTraverseToScan,
     'add-track-file': addTrackFile,
     'convert-progress-to-track-file': convertProgressToTrackFile,
+    'add-await-to-progress': addAwaitToProgress,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@putout/plugin-putout",
-  "version": "18.3.0",
+  "version": "18.4.1",
   "type": "commonjs",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "🐊Putout plugin helps with plugins development",
@@ -37,6 +37,7 @@
     "putout"
   ],
   "devDependencies": {
+    "@putout/plugin-tape": "*",
     "@putout/test": "^8.0.0",
     "c8": "^8.0.0",
     "eslint": "^8.0.1",