feature: @putout/printer: TSUnionType: always add parens when inside TSArrayType (benjamn/recast#1416)

Author: coderaiser

lib/tokenize/statements/if-statement/if-statement.js

@@ -4,10 +4,12 @@ const {types} = require('@putout/babel');
 
 const {markAfter} = require('../../mark');
 const {exists, isNext} = require('../../is');
+
 const {
     isBlockStatement,
     isFunctionDeclaration,
 } = types;
+
 const isInside = ({parentPath}) => !parentPath.parentPath.isProgram();
 const isEmptyConsequent = (path) => path.get('consequent').isEmptyStatement();
 

lib/tokenize/typescript/ts-union-type/fixture/ts-union-type-parens-fix.ts

@@ -0,0 +1 @@
+type A = (number | string)[];

lib/tokenize/typescript/ts-union-type/fixture/ts-union-type-parens.ts

@@ -0,0 +1 @@
+type A = (number | string)[];

lib/tokenize/typescript/ts-union-type/ts-union-type.js

@@ -3,14 +3,19 @@
 const {maybeParens} = require('../../maybe/maybe-parens');
 const insideTypeDeclaration = ({parentPath}) => parentPath.isTSTypeAliasDeclaration();
 
-module.exports.TSUnionType = maybeParens((path, printer, {maxTypesInOneLine}) => {
-    const types = path.get('types');
-    const {length} = types;
-    
-    if (!insideTypeDeclaration(path) || length <= maxTypesInOneLine)
-        printInOneLine(types, printer);
-    else
-        printInCoupleLines(types, printer);
+module.exports.TSUnionType = maybeParens({
+    condition: (path) => {
+        return path.parentPath.isTSArrayType();
+    },
+    print: (path, printer, {maxTypesInOneLine}) => {
+        const types = path.get('types');
+        const {length} = types;
+        
+        if (!insideTypeDeclaration(path) || length <= maxTypesInOneLine)
+            printInOneLine(types, printer);
+        else
+            printInCoupleLines(types, printer);
+    },
 });
 
 function printInOneLine(types, {traverse, write}) {

lib/tokenize/typescript/ts-union-type/ts-union-type.spec.js

@@ -2,7 +2,7 @@
 
 const {extend} = require('supertape');
 
-const {parse} = require('putout');
+const {parse, transform} = require('putout');
 const {printExtension} = require('../../../../test/printer');
 const {readFixtures} = require('../../../../test/fixture');
 
@@ -43,3 +43,27 @@ test('printer: tokenizer: ts-union-type: override', (t) => {
     t.equal(result, fixture.tsUnionTypeOverrideFix);
     t.end();
 });
+
+test('printer: tokenizer: typescript: ts-union-type: parens', (t) => {
+    const source = fixture.tsUnionTypeParens;
+    const ast = parse(source, {
+        isTS: true,
+    });
+    
+    transform(ast, source, {
+        plugins: [
+            ['remove-parens', {
+                report: () => '',
+                include: () => ['TSUnionType'],
+                fix: (path) => {
+                    delete path.node.extra;
+                },
+            }],
+        ],
+    });
+    
+    const result = print(ast);
+    
+    t.equal(result, fixture.tsUnionTypeParensFix);
+    t.end();
+});