feature: @putout/operator-parens: addParens/removeParens: return path

Author: coderaiser

packages/operator-parens/README.md

@@ -13,7 +13,7 @@ npm i putout @putout/operator-parens
 
 ## API
 
-### `addParens(path: Path)`
+### `addParens(path: Path): Path`
 
 Add parens around expression depending on used `printer`:
 
@@ -28,7 +28,7 @@ const {addParens} = operator;
 addParens(path);
 ```
 
-### `removeParens(path: Path)`
+### `removeParens(path: Path): Path`
 
 Remove parens around expression depending on used `printer`:
 
@@ -43,7 +43,7 @@ const {removeParens} = operator;
 removeParens(path);
 ```
 
-### `hasParens(path: Path)`
+### `hasParens(path: Path): Boolean`
 
 Check if `path` has parens around expression depending on used `printer`:
 

packages/operator-parens/lib/parens.js

@@ -26,7 +26,7 @@ module.exports.addParens = (path) => {
         extra.parenthesized = true;
         path.node.extra = extra;
 
-        return;
+        return path;
     }
 
     const {node} = path;
@@ -35,16 +35,22 @@ module.exports.addParens = (path) => {
         return path.replaceWith(TSParenthesizedType(node));
 
     path.replaceWith(ParenthesizedExpression(node));
+    
+    return path;
 };
 
 module.exports.removeParens = (path) => {
     const printer = getPrinter(path);
 
-    if (printer !== 'babel')
-        return path.node.extra.parenthesized = false;
+    if (printer !== 'babel') {
+        path.node.extra.parenthesized = false;
+        return path;
+    }
 
     const {node} = path;
     path.parentPath.replaceWith(node);
+    
+    return path;
 };
 
 function getPrinter(path) {

packages/operator-parens/lib/parens.spec.js

@@ -19,7 +19,9 @@ test('putout: operator: parens: removeParens: putout', (t) => {
     const ast = parse(source);
 
     traverse(ast, {
-        AssignmentExpression: removeParens,
+        AssignmentExpression: (path) => {
+            removeParens(path);
+        },
     });
 
     const result = print(ast);
@@ -29,6 +31,21 @@ test('putout: operator: parens: removeParens: putout', (t) => {
     t.end();
 });
 
+test('putout: operator: parens: removeParens: putout: return', (t) => {
+    const source = '(b = 3)';
+    const ast = parse(source);
+    let result;
+    
+    traverse(ast, {
+        AssignmentExpression: (path) => {
+            result = removeParens(path);
+        },
+    });
+    
+    t.equal(result.type, 'AssignmentExpression');
+    t.end();
+});
+
 test('putout: operator: parens: removeParens: babel', (t) => {
     const source = '(b = 3)';
     const ast = parse(source, {
@@ -52,6 +69,25 @@ test('putout: operator: parens: removeParens: babel', (t) => {
     t.end();
 });
 
+test('putout: operator: parens: removeParens: babel: return', (t) => {
+    const source = '(b = 3)';
+    const ast = parse(source, {
+        printer: 'babel',
+    });
+    
+    let result;
+    
+    traverse(ast, {
+        AssignmentExpression(path) {
+            result = removeParens(path);
+            path.stop();
+        },
+    });
+    
+    t.equal(result.type, 'AssignmentExpression');
+    t.end();
+});
+
 test('putout: operator: parens: addParens', (t) => {
     const source = 'const b = a';
     const ast = parse(source);
@@ -70,6 +106,22 @@ test('putout: operator: parens: addParens', (t) => {
     t.end();
 });
 
+test('putout: operator: parens: addParens: return', (t) => {
+    const source = 'const b = a';
+    const ast = parse(source);
+    let result;
+    
+    traverse(ast, {
+        VariableDeclarator(path) {
+            result = addParens(path.get('init'));
+            path.stop();
+        },
+    });
+    
+    t.equal(result.type, 'Identifier');
+    t.end();
+});
+
 test('putout: operator: parens: addParens: babel', (t) => {
     const source = 'b = 3';
     const ast = parse(source, {
@@ -93,6 +145,25 @@ test('putout: operator: parens: addParens: babel', (t) => {
     t.end();
 });
 
+test('putout: operator: parens: addParens: babel: return', (t) => {
+    const source = 'b = 3';
+    const ast = parse(source, {
+        printer: 'babel',
+    });
+    
+    let result;
+    
+    traverse(ast, {
+        AssignmentExpression(path) {
+            result = addParens(path);
+            path.stop();
+        },
+    });
+    
+    t.equal(result.type, 'ParenthesizedExpression');
+    t.end();
+});
+
 test('putout: operator: parens: addParens: babel: ts', (t) => {
     const source = 'const a: boolean = true;';
     const ast = parse(source, {