feature: @putout/printer: ForOfStatement: await

putoutjs · Jul 7, 2023 · 0c2a787 · 0c2a787
1 parent 177567d
commit 0c2a787
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 6 deletions.
diff --git a/lib/tokenize/statements/for-of-statement/fixture/for-of-await.js b/lib/tokenize/statements/for-of-statement/fixture/for-of-await.js
@@ -0,0 +1 @@
+for await (x of y) {}
diff --git a/...enize/statements/fixture/for-of-inside.js → ...for-of-statement/fixture/for-of-inside.js b/...enize/statements/fixture/for-of-inside.js → ...for-of-statement/fixture/for-of-inside.js
diff --git a/...nize/statements/fixture/for-of-newline.js → ...or-of-statement/fixture/for-of-newline.js b/...nize/statements/fixture/for-of-newline.js → ...or-of-statement/fixture/for-of-newline.js
diff --git a/...ze/statements/fixture/for-of-space-fix.js → ...-of-statement/fixture/for-of-space-fix.js b/...ze/statements/fixture/for-of-space-fix.js → ...-of-statement/fixture/for-of-space-fix.js
diff --git a/...kenize/statements/fixture/for-of-space.js → .../for-of-statement/fixture/for-of-space.js b/...kenize/statements/fixture/for-of-space.js → .../for-of-statement/fixture/for-of-space.js
diff --git a/lib/tokenize/statements/for-of-statement.js → ...ents/for-of-statement/for-of-statement.js b/lib/tokenize/statements/for-of-statement.js → ...ents/for-of-statement/for-of-statement.js
@@ -5,13 +5,13 @@ const {
     markAfter,
     markBefore,
     isMarkedAfter,
-} = require('../mark');
+} = require('../../mark');
 
 const {
     isFirst,
     isNext,
     isLast,
-} = require('../is');
+} = require('../../is');
 
 module.exports.ForOfStatement = {
     beforeIf(path) {
@@ -27,8 +27,11 @@ module.exports.ForOfStatement = {
         markBefore(path);
     },
     print(path, {indent, print, maybe, traverse}) {
+        const {node} = path;
+
         indent();
         print('for');
+        maybe.print(node.await, ' await');
         print.space();
         print('(');
         print('__left');

diff --git a/...enize/statements/for-of-statement.spec.js → ...for-of-statement/for-of-statement.spec.js b/...enize/statements/for-of-statement.spec.js → ...for-of-statement/for-of-statement.spec.js
@@ -1,10 +1,10 @@
 'use strict';
 
 const {extend} = require('supertape');
-const {printExtension} = require('../../../test/printer');
-const {readFixtures} = require('../../../test/fixture');
+const {printExtension} = require('../../../../test/printer');
+const {readFixtures} = require('../../../../test/fixture');
 const {parse} = require('putout');
-const {print} = require('../../printer');
+const {print} = require('../../../printer');
 const fixture = readFixtures(__dirname);
 
 const test = extend({
@@ -21,6 +21,11 @@ test('printer: tokenizer: statement: for-of: inside', (t) => {
     t.end();
 });
 
+test('printer: tokenizer: statement: for-of: await', (t) => {
+    t.print(fixture.forOfAwait);
+    t.end();
+});
+
 test('printer: tokenizer: statement: for-of: space', (t) => {
     const ast = parse(fixture.forOfSpace);
 

diff --git a/lib/tokenize/statements/index.js b/lib/tokenize/statements/index.js
@@ -3,7 +3,7 @@
 const {ExpressionStatement} = require('./expression-statement');
 const {VariableDeclaration} = require('./variable-declaration/variable-declaration');
 const {IfStatement} = require('./if-statement/if-statement');
-const {ForOfStatement} = require('./for-of-statement');
+const {ForOfStatement} = require('./for-of-statement/for-of-statement');
 const {BlockStatement} = require('./block-statement/block-statement');
 const {ReturnStatement} = require('./return-statement/return-statement');
 const TryStatements = require('./try-statements');