refactor: introduce ensureNoLineTerminator

babel · Sep 29, 2022 · 9dfae22 · 9dfae22
1 parent 09b3980
commit 9dfae22
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/packages/babel-generator/src/printer.ts b/packages/babel-generator/src/printer.ts
@@ -465,6 +465,16 @@ class Printer {
     return this._indentRepeat * this._indent;
   }
 
+  ensureNoLineTerminator(fn: () => void) {
+    const { _noLineTerminator } = this;
+    this._noLineTerminator = true;
+    try {
+      fn();
+    } finally {
+      this._noLineTerminator = _noLineTerminator;
+    }
+  }
+
   printTerminatorless(node: t.Node, parent: t.Node, isLabel: boolean) {
     /**
      * Set some state that will be modified if a newline has been inserted before any
@@ -482,9 +492,9 @@ class Printer {
      *  `undefined` will be returned and not `foo` due to the terminator.
      */
     if (isLabel) {
-      this._noLineTerminator = true;
-      this.print(node, parent);
-      this._noLineTerminator = false;
+      this.ensureNoLineTerminator(() => {
+        this.print(node, parent);
+      });
     } else {
       const terminatorState = {
         printed: false,