SplitStream.java

package jaci.openrio.toast.lib.log;

import jaci.openrio.toast.lib.util.Pretty;

import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;

/**
 * Acts as a splitter for {@link java.io.OutputStream} objects, splitting the output
 * between all the given instances in the constructor
 *
 * @author Jaci
 */
public class SplitStream extends OutputStream {

    ArrayList<OutputStream> outs;

    public SplitStream(OutputStream... streams) {
        this.outs = new ArrayList<>(Arrays.asList(streams));
    }

    /**
     * Add an output stream to the splitstream. Use this to delegate logger and
     * system.out output to your own handler or file.
     */
    public void add(OutputStream stream) {
        outs.add(stream);
    }

    /**
     * Write a single byte to all the output streams created for this SplitStream
     */
    @Override
    public void write(int b) throws IOException {
        for (OutputStream stream : outs)
            stream.write(b);
    }

    /**
     * Write a byte array to all the output streams created for this SplitStream
     */
    @Override
    public void write(byte b[]) throws IOException {
        for (OutputStream stream : outs)
            stream.write(b);
    }

    /**
     * Write a byte array with the offset and length given in this method to all
     * output streams created for this SplitStream
     */
    @Override
    public void write(byte b[], int off, int len) throws IOException {
        for (OutputStream stream : outs) {
            stream.write(b, off, len);
        }
    }

    /**
     * Flush the output streams registered for this SplitStream
     */
    @Override
    public void flush() throws IOException {
        for (OutputStream stream : outs)
            stream.flush();
    }

    /**
     * Close this stream and all it's children output streams.
     */
    @Override
    public void close() throws IOException {
        for (OutputStream stream : outs)
            stream.close();
    }

    /**
     * Print a color stream. This outputs color to the first output, but an escaped version to everything else
     * so your files don't get weird escape chars in them.
     */
    public void color(String s) throws IOException {
        outs.get(0).write(s.getBytes());
        String noc = Pretty.strip(s);
        for (int i = 1; i < outs.size(); i++) {
            OutputStream stream = outs.get(i);
            if (stream instanceof ColorPrint.ColorStream)
                stream.write(s.getBytes());
            else
                stream.write(noc.getBytes());
        }
    }
}