feature: @putout/opereate: hasParens: add

Author: coderaiser

packages/operate/README.md

@@ -276,6 +276,17 @@ Remove parens around expression depending on used `printer`:
 removeParens(path);
 ```
 
+### `hasParens(path: Path)`
+
+Check if `path` has parens around expression depending on used `printer`:
+
+- ✅ checks `node.extra.parenthesized` when `@putout/printer` used;
+- ✅ check if parent node type is `ParenthesizedExpression` or `TSParenthesizedType` when `babel` used;
+
+```js
+hasParens(path);
+```
+
 ## License
 
 MIT

packages/operate/lib/operate.js

@@ -12,7 +12,11 @@ const {rename} = require('./rename');
 const {renameProperty} = require('./rename-property');
 const {setLiteralValue} = require('./set-literal-value');
 const {getPathAfterRequires} = require('./get-path-after-requires');
-const {addParens, removeParens} = require('./parens');
+const {
+    addParens,
+    removeParens,
+    hasParens,
+} = require('./parens');
 
 const {
     getProperty,
@@ -50,6 +54,7 @@ module.exports.renameProperty = renameProperty;
 module.exports.setLiteralValue = setLiteralValue;
 
 module.exports.addParens = addParens;
+module.exports.hasParens = hasParens;
 module.exports.removeParens = removeParens;
 
 module.exports.getProperty = getProperty;

packages/operate/lib/parens.js

@@ -6,6 +6,17 @@ const {
     TSParenthesizedType,
 } = types;
 
+module.exports.hasParens = (path) => {
+    const printer = getPrinter(path);
+    
+    if (printer !== 'babel')
+        return path.node.extra?.parenthesized;
+    
+    const {type} = path.parentPath;
+    
+    return /^(TS)?Parenthesized(Expression|Type)?$/.test(type);
+};
+
 module.exports.addParens = (path) => {
     const printer = getPrinter(path);
 

packages/operate/lib/parens.spec.js

@@ -8,7 +8,11 @@ const {
     traverse,
 } = require('putout');
 
-const {addParens, removeParens} = require('./parens');
+const {
+    addParens,
+    removeParens,
+    hasParens,
+} = require('./parens');
 
 test('putout: operate: parens: removeParens: putout', (t) => {
     const source = '(b = 3)';
@@ -112,3 +116,56 @@ test('putout: operate: parens: addParens: babel: ts', (t) => {
     t.equal(result, expected);
     t.end();
 });
+
+test('putout: operate: parens: hasParens: babel: ts', (t) => {
+    let result = false;
+    const source = 'const a: (boolean) = true;';
+    const ast = parse(source, {
+        printer: 'babel',
+        isTS: true,
+    });
+    
+    traverse(ast, {
+        TSBooleanKeyword(path) {
+            result = hasParens(path);
+            path.stop();
+        },
+    });
+    
+    t.ok(result);
+    t.end();
+});
+
+test('putout: operate: parens: hasParens: babel', (t) => {
+    let result = false;
+    const source = '(b = 3)';
+    const ast = parse(source, {
+        printer: 'babel',
+    });
+    
+    traverse(ast, {
+        AssignmentExpression(path) {
+            result = hasParens(path);
+            path.stop();
+        },
+    });
+    
+    t.ok(result);
+    t.end();
+});
+
+test('putout: operate: parens: hasParens', (t) => {
+    let result = false;
+    const source = '(b = 3)';
+    const ast = parse(source);
+    
+    traverse(ast, {
+        AssignmentExpression(path) {
+            result = hasParens(path);
+            path.stop();
+        },
+    });
+    
+    t.ok(result);
+    t.end();
+});

packages/operate/test/operate.js

@@ -1092,3 +1092,18 @@ test('putout: operate: removeParens', (t) => {
     t.equal(result, expected);
     t.end();
 });
+
+test('putout: operate: hasParens', (t) => {
+    let result;
+    const source = 'b = 3';
+    const ast = parse(source);
+    
+    traverse(ast, {
+        AssignmentExpression: (path) => {
+            result = operate.hasParens(path);
+        },
+    });
+    
+    t.notOk(result);
+    t.end();
+});