#include "nbt.h"

static bool nbt_print_header(const tag_t *tag, const stream_t *stream, bool named) {
	if (stream_write_i8(stream, tag->type) == false)
		return false;
	if (!named)
		return true;
	if (stream_write_u16(stream, tag->name_len) == false)
		return false;
	if (tag->name_len > 0)
		if (stream_write(stream, tag->name, tag->name_len) == false)
			return false;
	return true;
}

static bool nbt_print_byte_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->b_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->b_arr.size; i++)
		if (stream_write_i8(stream, data->b_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_int_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->i_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->i_arr.size; i++)
		if (stream_write_i32(stream, data->i_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_long_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->l_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->l_arr.size; i++)
		if (stream_write_i64(stream, data->l_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_string(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_u16(stream, data->string.size) == false)
		return false;
	if (data->string.size < 1)
		return true;
	if (stream_write(stream, data->string.data, data->string.size) == false)
		return false;
	return true;
}

static bool nbt_print_data(const tag_t *tag, const stream_t *stream);

static bool nbt_print_list(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i8(stream, data->list.type) == false)
		return false;
	if (stream_write_i32(stream, data->list.size) == false)
		return false;
	for (int32_t i = 0; i < data->list.size; i++)
		if (nbt_print_data(&data->list.tags[i], stream) == false)
			return false;
	return true;
}

static bool nbt_print_compound(const tagdata_t *data, const stream_t *stream) {
	for (int32_t i = 0; i < data->compound.size; i++)
		if (nbt_print(&data->compound.tags[i], stream) == false)
			return false;
	if (stream_write_i8(stream, TAG_END) == false)
		return false;
	return true;
}

static bool nbt_print_data(const tag_t *tag, const stream_t *stream) {
	bool ok = true;

	switch (tag->type) {
        case TAG_END:
			// tag end has no data
			break;
        case TAG_BYTE:
			ok = stream_write_i8(stream, tag->data.b);
			break;
        case TAG_SHORT:
			ok = stream_write_i16(stream, tag->data.s);
			break;
        case TAG_INT:
        case TAG_FLOAT:
			ok = stream_write_i32(stream, tag->data.i);
			break;
        case TAG_LONG:
        case TAG_DOUBLE:
			ok = stream_write_i64(stream, tag->data.l);
			break;
        case TAG_BYTE_ARRAY:
			ok = nbt_print_byte_array(&tag->data, stream);
			break;
        case TAG_STRING:
			ok = nbt_print_string(&tag->data, stream);
			break;
        case TAG_LIST:
			ok = nbt_print_list(&tag->data, stream);
			break;
        case TAG_COMPOUND:
			ok = nbt_print_compound(&tag->data, stream);
			break;
        case TAG_INT_ARRAY:
			ok = nbt_print_int_array(&tag->data, stream);
			break;
        case TAG_LONG_ARRAY:
			ok = nbt_print_long_array(&tag->data, stream);
			break;
        }

	return ok;
}

bool nbt_print(const tag_t *tag, const stream_t *stream) {
	if (nbt_print_header(tag, stream, true) == false)
		return false;
	if (nbt_print_data(tag, stream) == false)
		return false;
	return true;
}