1 files changed, 125 insertions, 0 deletions
diff --git a/src/nbt/print.c b/src/nbt/print.c
new file mode 100644
index 0000000..4eac31f
--- /dev/null
+++ b/src/nbt/print.c
@@ -0,0 +1,125 @@
+#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;
+}