com.googlecode.lanterna.terminal.win32.WindowsConsoleOutputStream Maven / Gradle / Ivy

Go to download
package com.googlecode.lanterna.terminal.win32;

import java.io.ByteArrayOutputStream;
import java.io.EOFException;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;

import com.sun.jna.platform.win32.WinNT.HANDLE;
import com.sun.jna.ptr.IntByReference;

public class WindowsConsoleOutputStream extends OutputStream {

	private final HANDLE hConsoleOutput;
	private final Charset decoderCharset;
	private final ByteArrayOutputStream buffer = new ByteArrayOutputStream();

	public WindowsConsoleOutputStream(Charset decoder) {
		this(Wincon.INSTANCE.GetStdHandle(Wincon.STD_OUTPUT_HANDLE), decoder);
	}

	public WindowsConsoleOutputStream(HANDLE hConsoleOutput, Charset decoderCharset) {
		this.hConsoleOutput = hConsoleOutput;
		this.decoderCharset = decoderCharset;
	}

	public HANDLE getHandle() {
		return hConsoleOutput;
	}

	public Charset getDecoderCharset() {
		return decoderCharset;
	}


	@Override
	public synchronized void write(int b) {
		buffer.write(b);
	}

	@Override
	public synchronized void write(byte[] b, int off, int len) {
		buffer.write(b, off, len);
	}

	@Override
	public synchronized void flush() throws IOException {
		String characters = buffer.toString(decoderCharset.name());
		buffer.reset();

		IntByReference lpNumberOfCharsWritten = new IntByReference();
		while (!characters.isEmpty()) {
			if (!Wincon.INSTANCE.WriteConsole(hConsoleOutput, characters, characters.length(), lpNumberOfCharsWritten, null)) {
				throw new EOFException();
			}
			characters = characters.substring(lpNumberOfCharsWritten.getValue());
		}
	}

}