#include <stdarg.h>
#include <stdio.h>

#include "nbt.h"

static char buf[1024];

__attribute__((format(printf, 3, 4)))
static bool printi(const stream_t *stream, int depth, const char *format, ...) {
	for (int i = 0; i < depth; i++)
		if (stream_write(stream, "\t", 1) == false)
			return false;
	va_list list;
	va_start(list, format);
	int len;
	if ((len = vsnprintf(buf, 1024, format, list)) < 0)
		return false;
	if (stream_write(stream, buf, len) == false)
		return false;
	return true;
}

static bool snbt_print_impl(const tag_t *tag, const stream_t *stream, int depth);

static bool snbt_print_byte_array(const tagdata_t *data, const stream_t *stream) {
	if (printi(stream, 0, "[B;") == false)
		return false;
	for (int32_t i = 0; i < data->b_arr.size; i++) {
		if (i != 0) 
			if (printi(stream, 0, ",") == false)
				return false;
		if (printi(stream, 0, "%hhdb", data->b_arr.data[i]) == false)
			return false;
	}
	if (printi(stream, 0, "]") == false)
		return false;
	return true;
}

static bool snbt_print_int_array(const tagdata_t *data, const stream_t *stream) {
	if (printi(stream, 0, "[I;") == false)
		return false;
	for (int32_t i = 0; i < data->i_arr.size; i++) {
		if (i != 0) 
			if (printi(stream, 0, ",") == false)
				return false;
		if (printi(stream, 0, "%d", data->i_arr.data[i]) == false)
			return false;
	}
	if (printi(stream, 0, "]") == false)
		return false;
	return true;
}

static bool snbt_print_long_array(const tagdata_t *data, const stream_t *stream) {
	if (printi(stream, 0, "[L;") == false)
		return false;
	for (int32_t i = 0; i < data->l_arr.size; i++) {
		if (i != 0) 
			if (printi(stream, 0, ",") == false)
				return false;
		if (printi(stream, 0, "%ldL", data->l_arr.data[i]) == false)
			return false;
	}
	if (printi(stream, 0, "]") == false)
		return false;
	return true;
}

static bool snbt_string_unquoted(const char *text, uint16_t len) {
	for (uint16_t i = 0; i < len; i++) {
		char c = text[i];
		if (snbt_allowed_ident(c))
			continue;
		return false;
	}
	return true;
}

static bool snbt_print_string_impl(const stream_t *stream, int depth, char *text, uint16_t len) {
	if (len > 0 && snbt_string_unquoted(text, len)) {
		if (printi(stream, depth, "%.*s", len, text) == false)
			return false;
	} else if (len > 0) {
		if (printi(stream, depth, "\"") == false)
			return false;
		for (uint16_t i = 0; i < len; i++) {
			char c = text[i];
			if (c == '\\' || c == '"') {
				if (printi(stream, 0, "\\%c", c) == false)
					return false;
			} else {
				if (printi(stream, 0, "%c", c) == false)
					return false;
			}
		}
		if (printi(stream, 0, "\"") == false)
			return false;
	} else {
		if (printi(stream, depth, "''") == false)
			return false;
	}
	return true;
}

static bool snbt_print_string(const tagdata_t *data, const stream_t *stream) {
	char *text = data->string.data;
	uint16_t len = data->string.size;
	return snbt_print_string_impl(stream, 0, text, len);
}

static bool snbt_print_compound(const tagdata_t *data, const stream_t *stream, int depth) {
	if (printi(stream, 0, "{\n") == false)
		return false;
	bool first = true;
	for (uint32_t i = 0; i < data->compound.capacity; i++) {
		if (data->compound.entries[i].name == NULL)
			continue;
		if (!first && printi(stream, 0, ",\n") == false)
			return false;
		first = false;
		if (snbt_print_impl(&data->compound.entries[i], stream, depth + 1) == false)
			return false;
	}
	if (printi(stream, 0, "\n") == false || printi(stream, depth, "}") == false)
		return false;
	return true;
}

static bool snbt_print_list(const tagdata_t *data, const stream_t *stream, int depth) {
	if (printi(stream, 0, "[\n") == false)
		return false;
	for (int32_t i = 0; i < data->list.size; i++) {
		if (i != 0 && printi(stream, 0, ",\n") == false)
			return false;
		if (snbt_print_impl(&data->list.tags[i], stream, depth + 1) == false)
			return false;
	}
	if (printi(stream, 0, "\n") == false || printi(stream, depth, "]") == false)
		return false;
	return true;
}

static bool snbt_print_data(const tag_t *tag, const stream_t *stream, int depth) {

	bool ok = true;

	switch (tag->type) {
        case TAG_BYTE:
			ok = printi(stream, 0, "%hhdb", tag->data.b);
			break;
        case TAG_SHORT:
			ok = printi(stream, 0, "%hds", tag->data.s);
			break;
        case TAG_INT:
			ok = printi(stream, 0, "%d", tag->data.i);
			break;
        case TAG_LONG:
			ok = printi(stream, 0, "%ldL", tag->data.l);
			break;
        case TAG_FLOAT:
			ok = printi(stream, 0, "%.9gf", tag->data.f);
			break;
        case TAG_DOUBLE:
			ok = printi(stream, 0, "%.17g", tag->data.d);
			break;
        case TAG_BYTE_ARRAY:
			ok = snbt_print_byte_array(&tag->data, stream);
			break;
        case TAG_STRING:
			ok = snbt_print_string(&tag->data, stream);
			break;
        case TAG_LIST:
			ok = snbt_print_list(&tag->data, stream, depth);
			break;
        case TAG_COMPOUND:
			ok = snbt_print_compound(&tag->data, stream, depth);
			break;
        case TAG_INT_ARRAY:
			ok = snbt_print_int_array(&tag->data, stream);
			break;
        case TAG_LONG_ARRAY:
			ok = snbt_print_long_array(&tag->data, stream);
			break;
        case TAG_END:
            break;
        }

	return ok;
}

static bool snbt_print_impl(const tag_t *tag, const stream_t *stream, int depth) {
	if (tag->name_len > 0) {
		if (snbt_print_string_impl(stream, depth, tag->name, tag->name_len) == false)
			return false;
		if (printi(stream, 0, ": ") == false)
			return false;
	} else {
		for (int i = 0; i < depth; i++) 
			printi(stream, 0, "\t");
	}
	return snbt_print_data(tag, stream, depth);
}

bool snbt_print(const tag_t *tag, const stream_t *stream) {
	if (snbt_print_impl(tag, stream, 0) == false)
		return false;
	if (stream_write(stream, "\n", 1) == false)
		return false;
	return true;
}