From cf94008b6fded5f58cea764d48beb0dcbbd4bb97 Mon Sep 17 00:00:00 2001
From: Christopher Vogt <oss.nsp@cvogt.org>
Date: Thu, 13 Oct 2016 01:29:48 +0000
Subject: swap out System.out and System.err in a way that affects JDK and
 Scala Before it only affected jdk, because scala.Console captures our and err
 before the swap.

This is needed when running main classes like Scaladoc or the compiler
and wanting to redirect output to standard error
---
 nailgun_launcher/NailgunLauncher.java         | 19 ++++++++++++++++-
 nailgun_launcher/ThreadLocalOutputStream.java | 30 +++++++++++++++++++++++++++
 2 files changed, 48 insertions(+), 1 deletion(-)
 create mode 100644 nailgun_launcher/ThreadLocalOutputStream.java

(limited to 'nailgun_launcher')

diff --git a/nailgun_launcher/NailgunLauncher.java b/nailgun_launcher/NailgunLauncher.java
index 37bfa06..b1daf2a 100644
--- a/nailgun_launcher/NailgunLauncher.java
+++ b/nailgun_launcher/NailgunLauncher.java
@@ -59,7 +59,24 @@ public class NailgunLauncher{
     installProxySettings();
     String[] diff = args[0].split("\\.");
     long start = _start - (Long.parseLong(diff[0]) * 1000L) - Long.parseLong(diff[1]);
-    
+
+    // if nailgun didn't install it's threadLocal stdout/err replacements, install CBT's.
+    // this hack allows to later swap out System.out/err while still affecting things like
+    // scala.Console, which captured them at startup
+    try{
+      System.out.getClass().getDeclaredField("streams"); // nailgun ThreadLocalPrintStream
+      assert(System.out.getClass().getName() == "com.martiansoftware.nailgun.ThreadLocalPrintStream");
+    } catch( NoSuchFieldException e ){
+      System.setOut( new PrintStream(new ThreadLocalOutputStream(System.out)) );
+    }
+    try{
+      System.err.getClass().getDeclaredField("streams"); // nailgun ThreadLocalPrintStream
+      assert(System.out.getClass().getName() == "com.martiansoftware.nailgun.ThreadLocalPrintStream");
+    } catch( NoSuchFieldException e ){
+      System.setErr( new PrintStream(new ThreadLocalOutputStream(System.err)) );
+    }
+    // ---------------------
+
     _assert(System.getenv("CBT_HOME") != null, "environment variable CBT_HOME not defined");
     String CBT_HOME = System.getenv("CBT_HOME");
     String cache = CBT_HOME + "/cache/";
diff --git a/nailgun_launcher/ThreadLocalOutputStream.java b/nailgun_launcher/ThreadLocalOutputStream.java
new file mode 100644
index 0000000..c12b775
--- /dev/null
+++ b/nailgun_launcher/ThreadLocalOutputStream.java
@@ -0,0 +1,30 @@
+package cbt;
+import java.io.*;
+
+public class ThreadLocalOutputStream extends OutputStream{
+  final public ThreadLocal<OutputStream> threadLocal; 
+  final private OutputStream initialValue;
+
+  public ThreadLocalOutputStream( OutputStream initialValue ){
+    this.initialValue = initialValue;
+    threadLocal = new ThreadLocal<OutputStream>() {
+      @Override protected OutputStream initialValue() {
+        return ThreadLocalOutputStream.this.initialValue;
+      }
+    };
+  }
+  
+  public OutputStream get(){
+    return threadLocal.get();
+  }
+  
+  public void set( OutputStream outputStream ){
+    threadLocal.set( outputStream );
+  }
+  
+  public void write( int b ) throws IOException{
+    // after implementing this I realized NailgunLauncher uses the same hack,
+    // so probably this is not a problem performance
+    get().write(b);
+  }
+}
-- 
cgit v1.2.3