[Truffle] Newlines on global assign.

jruby · Nov 23, 2015 · 9f7fefc · 9f7fefc
1 parent 080d30a
commit 9f7fefc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/truffle/src/main/java/org/jruby/truffle/translator/BodyTranslator.java b/truffle/src/main/java/org/jruby/truffle/translator/BodyTranslator.java
@@ -1430,12 +1430,12 @@ public RubyNode visitGlobalAsgnNode(org.jruby.ast.GlobalAsgnNode node) {
 
         final boolean inCore = rhs.getSourceSection().getSource().getPath().startsWith(context.getCoreLibrary().getCoreLoadPath() + "/core/");
         if (!inCore && READ_ONLY_GLOBAL_VARIABLES.contains(name)) {
-            return new WriteReadOnlyGlobalNode(context, sourceSection, name, rhs);
+            return addNewlineIfNeeded(node, new WriteReadOnlyGlobalNode(context, sourceSection, name, rhs));
         }
 
         if (THREAD_LOCAL_GLOBAL_VARIABLES.contains(name)) {
             final ThreadLocalObjectNode threadLocalVariablesObjectNode = new ThreadLocalObjectNode(context, sourceSection);
-            return new WriteInstanceVariableNode(context, sourceSection, name, threadLocalVariablesObjectNode, rhs, true);
+            return addNewlineIfNeeded(node, new WriteInstanceVariableNode(context, sourceSection, name, threadLocalVariablesObjectNode, rhs, true));
         } else if (FRAME_LOCAL_GLOBAL_VARIABLES.contains(name)) {
             if (environment.getNeverAssignInParentScope()) {
                 environment.declareVar(name);
@@ -1468,10 +1468,10 @@ public RubyNode visitGlobalAsgnNode(org.jruby.ast.GlobalAsgnNode node) {
                 assignment = GetFromThreadLocalNodeGen.create(context, sourceSection, assignment);
             }
 
-            return assignment;
+            return addNewlineIfNeeded(node, assignment);
         } else {
             final LiteralNode globalVariablesObjectNode = new LiteralNode(context, sourceSection, context.getCoreLibrary().getGlobalVariablesObject());
-            return new WriteInstanceVariableNode(context, sourceSection, name, globalVariablesObjectNode, rhs, true);
+            return addNewlineIfNeeded(node, new WriteInstanceVariableNode(context, sourceSection, name, globalVariablesObjectNode, rhs, true));
 
         }
     }