@putout/printer 1.127.0 → 1.128.1

package/ChangeLog

@@ -1,3 +1,13 @@
+2023.06.01, v1.128.1
+
+feature:
+- 9de6e21 @putout/printer: ArrowFunctionExpression: braces
+
+2023.06.01, v1.128.0
+
+feature:
+- 118671c @putout/printer: ArrowFunctionExpression: add ability to drop braces
+
 2023.06.01, v1.127.0
 
 feature:

package/README.md

@@ -82,6 +82,8 @@ write(ast, {
         space: ' ',
         comments: true,
         splitter: '\n',
+        roundBraceOpen: '(',
+        roundBraceClose: ')',
     },
     visitors: {
         AssignmentPattern(path, {write}) {

package/lib/tokenize/expressions/assignment-pattern.js

@@ -5,7 +5,9 @@ const isArg = (path) => path.parentPath.isFunction();
 module.exports.AssignmentPattern = {
     print(path, {print, maybe}) {
         maybe.print(shouldPrint(path), '__left');
-        print(' = ');
+        print.space();
+        print('=');
+        print.space();
         print('__right');
     },
 };

package/lib/tokenize/expressions/functions/params.js

@@ -1,7 +1,9 @@
 'use strict';
 
 module.exports.printParams = (path, {print}) => {
-    print('(');
+    printBraceOpen(path, {
+        print,
+    });
     
     const params = path.get('params');
     const n = params.length;
@@ -15,5 +17,34 @@ module.exports.printParams = (path, {print}) => {
         }
     }
     
-    print(')');
+    printBraceClose(path, {
+        print,
+    });
 };
+
+function printBraceOpen(path, {print}) {
+    if (isOneArgArrow(path))
+        return print.roundBraceOpen();
+    
+    return print('(');
+}
+
+function printBraceClose(path, {print}) {
+    if (isOneArgArrow(path))
+        return print.roundBraceClose();
+    
+    print(')');
+}
+
+function isOneArgArrow(path) {
+    if (path.type !== 'ArrowFunctionExpression')
+        return false;
+    
+    const {params} = path.node;
+    const [param] = params;
+    
+    if (params.length !== 1)
+        return false;
+    
+    return param.type === 'Identifier';
+}

package/lib/tokenize/tokenize.js

@@ -34,6 +34,7 @@ const {
 
 const isString = (a) => typeof a === 'string';
 const {assign} = Object;
+const callWith = (fn, a) => () => fn(a);
 
 const traversers = {
     ...expressions,
@@ -53,6 +54,8 @@ function initFormat(format) {
         space: ' ',
         comments: true,
         splitter: '\n',
+        roundBraceOpen: '(',
+        roundBraceClose: ')',
         ...format,
     };
 }
@@ -110,6 +113,16 @@ module.exports.tokenize = (ast, overrides = {}) => {
         });
     };
     
+    const roundBraceOpen = callWith(addToken, {
+        type: TYPES.ROUND_BRACE_OPEN,
+        value: format.roundBraceOpen,
+    });
+    
+    const roundBraceClose = callWith(addToken, {
+        type: TYPES.ROUND_BRACE_Close,
+        value: format.roundBraceClose,
+    });
+    
     const linebreak = () => {
         indent();
         newline();
@@ -202,6 +215,8 @@ module.exports.tokenize = (ast, overrides = {}) => {
         assign(print, write, {
             space,
             round,
+            roundBraceOpen,
+            roundBraceClose,
         });
         
         assign(printer, {

package/lib/types.js

@@ -8,4 +8,6 @@ module.exports.TYPES = {
     INDENT: 'Indent',
     DEBUG: 'Debug',
     SPACE: 'Space',
+    ROUND_BRACE_OPEN: '(',
+    ROUND_BRACE_CLOSE: ')',
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@putout/printer",
-  "version": "1.127.0",
+  "version": "1.128.1",
   "type": "commonjs",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "Simplest possible opinionated Babel AST printer for 🐊Putout",