package org.xmlhammer.gui.util;

 import java.awt.Color;

 import java.awt.Font;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.io.PrintStream;

 import java.util.HashMap;

 import javax.swing.JTextPane;

 import javax.swing.text.BadLocationException;

 import javax.swing.text.Document;

 import javax.swing.text.SimpleAttributeSet;

 import javax.swing.text.StyleConstants;

 import javax.swing.text.StyledDocument;

 public class ConsolePane extends JTextPane {

     private static final long serialVersionUID = 1513033943583921042L;

     private static final PrintStream ORIGINAL_OUT = System.out;

     private static final PrintStream ORIGINAL_ERR = System.err;

     private static OutputStreamProxy errOutputStreamProxy = null;

     private static OutputStreamProxy outOutputStreamProxy = null;

     public ConsolePane() {

         super();

         init();

     }

     public ConsolePane(StyledDocument arg0) {

         super(arg0);

         init();

     }

     private void init() {

         Font f = getFont();

         setFont(new Font("Monospaced", f.getStyle(), f.getSize()));

     }

     public void start(Thread thread) {

         getErrStreamProxy().put(thread, createErrStream());

         getOutStreamProxy().put(thread, createOutStream());

     }

     public void stop(Thread thread) {

         getErrStreamProxy().remove(thread);

         getOutStreamProxy().remove(thread);

     }

     private static OutputStreamProxy getErrStreamProxy() {

         if (errOutputStreamProxy == null) {

             errOutputStreamProxy = new OutputStreamProxy(ORIGINAL_ERR);

             System.setErr(new PrintStream(errOutputStreamProxy));

         }

         return errOutputStreamProxy;

     }

     private static OutputStreamProxy getOutStreamProxy() {

         if (outOutputStreamProxy == null) {

             outOutputStreamProxy = new OutputStreamProxy(ORIGINAL_OUT);

             System.setOut(new PrintStream(outOutputStreamProxy));

         }

         return outOutputStreamProxy;

     }

     private PrintStream createErrStream() {

         SimpleAttributeSet errAttributes = new SimpleAttributeSet();

         StyleConstants.setForeground(errAttributes, Color.red);

         return new PrintStream(new OutputStreamAdapter(this.getDocument(), errAttributes));

     }

     private PrintStream createOutStream() {

         SimpleAttributeSet attributes = new SimpleAttributeSet();

         StyleConstants.setForeground(attributes, Color.blue);

         return new PrintStream(new OutputStreamAdapter(this.getDocument(), attributes));

     }

     public class OutputStreamAdapter extends OutputStream {

         private Document document = null;

         private SimpleAttributeSet attributes = null;

         public OutputStreamAdapter(Document document, SimpleAttributeSet attributes) {

             this.document = document;

             this.attributes = attributes;

         }

         public void write(byte bytes[], int off, int len) {

             ORIGINAL_OUT.println("ID = "+Thread.currentThread().getId());

             try {

                 document.insertString(document.getLength(), new String(bytes, off, len), attributes);

             } catch (BadLocationException e) {

                 // Should never happen

             }

         }

         public void write(int c) {

             ORIGINAL_OUT.println("ID = "+Thread.currentThread().getId());

             try {

                 document.insertString(document.getLength(), new String(new byte[] {(byte)c}), attributes);

             } catch (BadLocationException e) {

                 // Should never happen

             }

         }

     }

     public static class OutputStreamProxy extends OutputStream {

         private OutputStream original = null;

         private HashMap<Long, OutputStream> map = null; 

         public OutputStreamProxy(OutputStream original) {

             map = new HashMap<Long, OutputStream>();

             this.original = original;

         }

         public void write(byte bytes[], int off, int len) throws IOException {

             OutputStream stream = map.get(Thread.currentThread().getId());

             if (stream == null) {

                 original.write(bytes, off, len);

             } else {

                 stream.write(bytes, off, len);

             }

         }

         public void write(int c) throws IOException {

             OutputStream stream = map.get(Thread.currentThread().getId());

             if (stream == null) {

                 original.write(c);

             } else {

                 stream.write(c);

             }

         }

         public void put(Thread thread, OutputStream stream) {

             map.put(thread.getId(), stream);

         }

         public void remove(Thread thread) {

             map.remove(thread.getId());

         }

     }

 }