package example5_3javaiodecorator;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

/**
 * @author hornick
 * This output stream decorator adds line numbers to a wrapped output stream
 */

public class LineNumberDecorator extends FilterOutputStream {
	private int lineNumber;

	/**
	 * constructs the output stream decorator
	 * @param wrappedStream
	 */
	public LineNumberDecorator(OutputStream wrappedStream) {
		super(wrappedStream);
		lineNumber = 0; // initialize line number
	}

	/**
	 * accept an array of bytes and forward them to the wrapped output stream
	 * Update Jay Urbain 1/14/2014
	 */
	@Override
	public void write( byte[] b ) throws IOException {

		// First, do the line number decorating
		String ln = "" + lineNumber++ + ":";
		for(int j=0; j<ln.length(); super.write(ln.charAt(j++)) );
		boolean wasCR=false;
		for( int i=0; i<b.length; i++) {
			if( wasCR ) {
				wasCR = false;
				ln = "" + lineNumber++ + ":";
				for(int j=0; j<ln.length(); super.write(ln.charAt(j++)) );
			}
			super.write(b[i]);
			if( b[i] == '\n') {
				wasCR = true;
			}
		}			
	}


	/**
	 * accept an array of bytes and forward them to the wrapped output stream
	 * Update Jay Urbain 1/14/2014
	 */
	@Override
	
	public void write( byte[] b, int offset, int len ) throws IOException {

		// First, do the line number decorating
		String ln = "" + lineNumber++ + ":";
		for(int j=0; j<ln.length(); super.write(ln.charAt(j++)) );
		boolean wasCR=false;
		for( int i=0; i<len; i++) {
			if( wasCR ) {
				wasCR = false;
				ln = "" + lineNumber++ + ":";
				for(int j=0; j<ln.length(); super.write(ln.charAt(j++)) );
			}
			super.write(b[offset+i]);
			if( b[i] == '\n') {
				wasCR = true;
			}
		}		
	}

/*
	public void write( byte[] b, int offset, int len ) throws IOException {

		for( int i=0; i<len; i++) {
			if( lineNumber==0 ) {
				super.write(lineNumber++);
				super.write(':');
				super.write(' ');
				super.write(b[i]);
			}
			else if( b[i] == '\n' ) {
				super.write(b[i]);
				super.write(lineNumber++);
				super.write(':');
				super.write(' ');
			}
		}
	}
*/
}