feature: goldstein: printer: add ability to print goldstein: try

packages/keyword-try/index.js

@@ -43,10 +43,20 @@ export default function keywordTry(Parser) {
                         expression.callee,
                         ...expression.arguments,
                     ],
+                    goldstein: createGoldsteinNode({
+                        await: false,
+                        callee: expression.callee,
+                        arguments: expression.arguments,
+                    }),
                 };
             else if (isAwaitExpression(expression))
                 node.expression = {
                     type: 'AwaitExpression',
+                    goldstein: createGoldsteinNode({
+                        await: true,
+                        callee: expression.argument.callee,
+                        arguments: expression.argument.arguments,
+                    }),
                     argument: {
                         type: 'CallExpression',
                         callee: {
@@ -100,3 +110,15 @@ export default function keywordTry(Parser) {
     };
 }
 
+function createGoldsteinNode(args) {
+    return {
+        type: 'TryStatement',
+        expression: true,
+        await: args.await,
+        argument: {
+            type: 'CallExpression',
+            callee: args.callee,
+            arguments: args.arguments,
+        },
+    };
+}

packages/printer/index.js

@@ -0,0 +1,18 @@
+import {print as printJS} from '@putout/printer';
+
+import {AwaitExpression} from './visitors/await-expression.js';
+import {CallExpression} from './visitors/call-expression.js';
+import {TryStatement} from './visitors/try-statement.js';
+import {fixEmpty} from '../goldstein/index.js';
+
+export const print = (ast) => {
+    const code = printJS(ast, {
+        visitors: {
+            CallExpression,
+            TryStatement,
+            AwaitExpression,
+        },
+    });
+
+    return fixEmpty(code);
+};

packages/printer/index.spec.js

@@ -0,0 +1,31 @@
+import {test} from 'supertape';
+import montag from 'montag';
+import estreeToBabel from 'estree-to-babel';
+
+import {print} from './index.js';
+import {parse} from '../goldstein/index.js';
+
+test('goldstein: printer: try: await', (t) => {
+    const source = `const a = try await f('hello')`;
+    const ast = estreeToBabel(parse(source));
+    const result = print(ast);
+    const expected = montag`
+        const a = try await f('hello');\n
+    `;
+
+    t.equal(result, expected);
+    t.end();
+});
+
+test('goldstein: printer: try', (t) => {
+    const source = `const a = try f('hello')`;
+    const ast = estreeToBabel(parse(source));
+    const result = print(ast);
+    const expected = montag`
+        const a = try f('hello');\n
+    `;
+
+    t.equal(result, expected);
+    t.end();
+});
+

packages/printer/visitors/await-expression.js

@@ -0,0 +1,10 @@
+import {visitors as v} from '@putout/printer';
+
+export const AwaitExpression = (path, printer, semantics) => {
+    const {print} = printer;
+
+    if (!path.node.goldstein)
+        return v.AwaitExpression(path, printer, semantics);
+
+    print('__goldstein');
+};

packages/printer/visitors/call-expression.js

@@ -0,0 +1,10 @@
+import {visitors as v} from '@putout/printer';
+
+export const CallExpression = (path, printer, semantics) => {
+    const {print} = printer;
+
+    if (!path.node.goldstein)
+        return v.CallExpression(path, printer, semantics);
+
+    print('__goldstein');
+};

packages/printer/visitors/try-statement.js

@@ -0,0 +1,12 @@
+import {visitors as v} from '@putout/printer';
+export const TryStatement = (path, printer, semantics) => {
+    const {maybe, print} = printer;
+    const {node} = path;
+
+    if (!node.expression)
+        return v.TryStatement(path, printer, semantics);
+
+    print('try ');
+    maybe.print(node.await, 'await ');
+    print('__argument');
+};