feature: @putout/engine-parser: mark used printer in ast

Author: coderaiser

packages/engine-parser/lib/parse.js

@@ -2,6 +2,7 @@
 
 const toBabel = require('estree-to-babel');
 const customParser = require('./custom-parser');
+const {assign} = Object;
 
 module.exports = (source, options) => {
     const {
@@ -23,11 +24,18 @@ module.exports = (source, options) => {
 
 const getParser = ({parser = 'babel', isTS, isJSX, printer}) => ({
     parse(source) {
+        const options = {};
+        
+        if (printer === 'babel')
+            assign(options, {
+                convertParens: false,
+            });
+        
         const ast = toBabel(customParser(source, parser, {
             isTS,
             isJSX,
             printer,
-        }));
+        }), options);
 
         return ast;
     },

packages/engine-parser/lib/parse.spec.js

@@ -102,3 +102,24 @@ test('putout: engine-parser: parse + generate = sourcemap', (t) => {
     t.ok(map, 'returns map');
     t.end();
 });
+
+test('putout: engine-parser: parse: set extra.__putout_printer', (t) => {
+    const source = `const hello = 'world';`;
+    const ast = parse(source);
+    const {__putout_printer} = ast.program.extra;
+    
+    t.equal(__putout_printer, 'putout');
+    t.end();
+});
+
+test('putout: engine-parser: parse: set extra.__putout_printer: babel', (t) => {
+    const source = `const hello = 'world';`;
+    const ast = parse(source, {
+        printer: 'babel',
+    });
+    
+    const {__putout_printer} = ast.program.extra;
+    
+    t.equal(__putout_printer, 'babel');
+    t.end();
+});

packages/engine-parser/lib/parsers/babel/index.js

@@ -41,7 +41,10 @@ module.exports.parse = function babelParse(source, overrides) {
             createParenthesizedExpressions: true,
         });
 
-    return parse(source, parserOptions);
+    const ast = parse(source, parserOptions);
+    
+    ast.program.extra.__putout_printer = printer;
+    return ast;
 };
 
 function getBabelLangExts({isTS, isFlow, isJSX}) {

packages/engine-parser/lib/printers/babel.spec.js

@@ -134,6 +134,7 @@ test('putout: parser: print: printer: babel: preserve format: parens: parser: no
 
 test('putout: parser: print: printer: babel: preserve format: parens', (t) => {
     const source = montag`
+        a && (a = b);
         const a: (boolean | number)[] = [false, 1];
     `;
 
@@ -148,6 +149,7 @@ test('putout: parser: print: printer: babel: preserve format: parens', (t) => {
     });
 
     const expected = montag`
+        a && (a = b);
         const a: (boolean | number)[] = [false, 1];\n
     `;