From 6bf52b74c06f08f81da4739c64e6eeb366c42c80 Mon Sep 17 00:00:00 2001
From: Logan Smyth <loganfsmyth@gmail.com>
Date: Sat, 16 Jul 2016 13:41:54 -0700
Subject: [PATCH] Include newline insertion in the call sites of rightBrace.

---
 packages/babel-generator/src/generators/base.js    | 3 +++
 packages/babel-generator/src/generators/classes.js | 2 ++
 packages/babel-generator/src/printer.js            | 2 --
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/babel-generator/src/generators/base.js b/packages/babel-generator/src/generators/base.js
index d291683b6b..a7d1bd35b2 100644
--- a/packages/babel-generator/src/generators/base.js
+++ b/packages/babel-generator/src/generators/base.js
@@ -24,6 +24,9 @@ export function BlockStatement(node: Object) {
     this.removeTrailingNewline();
 
     this.source("end", node.loc);
+
+    if (!this.endsWith("\n")) this.newline();
+
     this.rightBrace();
   } else {
     this.source("end", node.loc);
diff --git a/packages/babel-generator/src/generators/classes.js b/packages/babel-generator/src/generators/classes.js
index f7b75a321e..841473324a 100644
--- a/packages/babel-generator/src/generators/classes.js
+++ b/packages/babel-generator/src/generators/classes.js
@@ -42,6 +42,8 @@ export function ClassBody(node: Object) {
     this.printSequence(node.body, node);
     this.dedent();
 
+    if (!this.endsWith("\n")) this.newline();
+
     this.rightBrace();
   }
 }
diff --git a/packages/babel-generator/src/printer.js b/packages/babel-generator/src/printer.js
index 14139c2d16..facf0fb4c0 100644
--- a/packages/babel-generator/src/printer.js
+++ b/packages/babel-generator/src/printer.js
@@ -76,8 +76,6 @@ export default class Printer {
    */
 
   rightBrace(): void {
-    if (!this.endsWith("\n")) this.newline();
-
     if (this.format.minified) {
       this._buf.removeLastSemicolon();
     }