feature: @putout/plugin-parens: add-missing-for-assign: add

Author: coderaiser

docs/syntax-errors.md

@@ -264,3 +264,12 @@ const a = {
 ```
 
 </details>
+
+<details><summary>wrong brace</summary>
+
+```diff
+a && b = a;
+a && (b = a);
+```
+
+</details>

packages/plugin-parens/README.md

@@ -25,6 +25,25 @@ npm i @putout/plugin-parens
 }
 ```
 
+## assign
+
+To disable use:
+
+```json
+{
+    "rules": {
+        "parens/add-missing-for-assign": "off"
+    }
+}
+```
+
+## ❌ Example of incorrect code
+
+```diff
+-a && b = a;
++a && (b = a);
+```
+
 ## await
 
 To disable use:

packages/plugin-parens/lib/add-missing-for-assign/fixture/add-missing-for-assign-fix.js

@@ -0,0 +1 @@
+a && (b = a);

packages/plugin-parens/lib/add-missing-for-assign/fixture/add-missing-for-assign.js

@@ -0,0 +1 @@
+a && b = a;

packages/plugin-parens/lib/add-missing-for-assign/fixture/binary-fix.js

@@ -0,0 +1 @@
+a === (b = c);

packages/plugin-parens/lib/add-missing-for-assign/fixture/binary.js

@@ -0,0 +1 @@
+a === b = c

packages/plugin-parens/lib/add-missing-for-assign/fixture/unary-fix.js

@@ -0,0 +1 @@
+!(b = c);

packages/plugin-parens/lib/add-missing-for-assign/fixture/unary.js

@@ -0,0 +1 @@
+!b = c

packages/plugin-parens/lib/add-missing-for-assign/index.js

@@ -0,0 +1,69 @@
+'use strict';
+
+const {types, operator} = require('putout');
+const {
+    BinaryExpression,
+    LogicalExpression,
+    UnaryExpression,
+    AssignmentExpression,
+    isBinaryExpression,
+    isUnaryExpression,
+    isLogicalExpression,
+} = types;
+
+const {replaceWith} = operator;
+
+module.exports.report = () => `SyntaxError: Invalid left-hand side in assignment expression`;
+
+module.exports.fix = (path) => {
+    const {
+        left,
+        right,
+        operator,
+    } = path.node;
+    
+    const logicalOperator = left.operator;
+    const logicalLeft = left.left;
+    
+    if (isLogicalExpression(left)) {
+        const logicalRight = AssignmentExpression(operator, left.right, right);
+        
+        replaceWith(path, LogicalExpression(
+            logicalOperator,
+            logicalLeft,
+            logicalRight,
+        ));
+    } else if (isBinaryExpression(left)) {
+        const logicalRight = AssignmentExpression(operator, left.right, right);
+        
+        replaceWith(path, BinaryExpression(
+            logicalOperator,
+            logicalLeft,
+            logicalRight,
+        ));
+    } else if (isUnaryExpression(left)) {
+        const logicalRight = AssignmentExpression(operator, left.argument, right);
+        
+        replaceWith(path, UnaryExpression(
+            logicalOperator,
+            logicalRight,
+        ));
+    }
+    
+    path.node.extra = {
+        parenthesized: true,
+    };
+};
+
+module.exports.traverse = ({push}) => ({
+    AssignmentExpression(path) {
+        if (isLogicalExpression(path.node.left))
+            push(path);
+        
+        if (isBinaryExpression(path.node.left))
+            push(path);
+        
+        if (isUnaryExpression(path.node.left))
+            push(path);
+    },
+});

packages/plugin-parens/lib/add-missing-for-assign/index.spec.js

@@ -0,0 +1,30 @@
+'use strict';
+
+const {createTest} = require('@putout/test');
+const plugin = require('.');
+
+const test = createTest(__dirname, {
+    plugins: [
+        ['add-missing-for-assign', plugin],
+    ],
+});
+
+test('putout: plugin-parens: add-missing-for-assign: report', (t) => {
+    t.report('add-missing-for-assign', `SyntaxError: Invalid left-hand side in assignment expression`);
+    t.end();
+});
+
+test('putout: plugin-parens: add-missing-for-assign: transform', (t) => {
+    t.transform('add-missing-for-assign');
+    t.end();
+});
+
+test('putout: plugin-parens: add-missing-for-assign: transform: binary', (t) => {
+    t.transform('binary');
+    t.end();
+});
+
+test('putout: plugin-parens: add-missing-for-assign: transform: unary', (t) => {
+    t.transform('unary');
+    t.end();
+});