package net.minecraft.server.rcon;

import java.io.ByteArrayOutputStream;
import java.io.DataOutputStream;
import java.io.IOException;

public class NetworkDataOutputStream {
	private final ByteArrayOutputStream outputStream;
	private final DataOutputStream dataOutputStream;

	public NetworkDataOutputStream(int capacity) {
		this.outputStream = new ByteArrayOutputStream(capacity);
		this.dataOutputStream = new DataOutputStream(this.outputStream);
	}

	/**
	 * Writes the given byte array to the output stream
	 */
	public void writeBytes(byte[] data) throws IOException {
		this.dataOutputStream.write(data, 0, data.length);
	}

	/**
	 * Writes the given String to the output stream
	 */
	public void writeString(String data) throws IOException {
		this.dataOutputStream.writeBytes(data);
		this.dataOutputStream.write(0);
	}

	/**
	 * Writes the given int to the output stream
	 */
	public void write(int data) throws IOException {
		this.dataOutputStream.write(data);
	}

	/**
	 * Writes the given short to the output stream
	 */
	public void writeShort(short data) throws IOException {
		this.dataOutputStream.writeShort(Short.reverseBytes(data));
	}

	public void writeInt(int data) throws IOException {
		this.dataOutputStream.writeInt(Integer.reverseBytes(data));
	}

	public void writeFloat(float data) throws IOException {
		this.dataOutputStream.writeInt(Integer.reverseBytes(Float.floatToIntBits(data)));
	}

	/**
	 * Returns the contents of the output stream as a byte array
	 */
	public byte[] toByteArray() {
		return this.outputStream.toByteArray();
	}

	/**
	 * Resets the byte array output.
	 */
	public void reset() {
		this.outputStream.reset();
	}
}