#include <string>
#include <iostream>
#include <sstream>

#include "BlockDecoder.h"
#include "ZlibDecompressor.h"

static inline uint16_t readU16(const unsigned char *data)
{
	return data[0] << 8 | data[1];
}

static inline uint16_t readBlockContent(const unsigned char *mapData,
	u8 contentWidth, unsigned int datapos)
{
	if (contentWidth == 2) {
		size_t index = datapos << 1;
		return (mapData[index] << 8) | mapData[index + 1];
	} else {
		u8 param = mapData[datapos];
		if (param <= 0x7f)
			return param;
		else
			return (param << 4) | (mapData[datapos + 0x2000] >> 4);
	}
}

BlockDecoder::BlockDecoder()
{
	reset();
}

void BlockDecoder::reset()
{
	m_blockAirId = -1;
	m_blockIgnoreId = -1;
	m_nameMap.clear();

	m_version = 0;
	m_contentWidth = 0;
	m_mapData.clear();
}

void BlockDecoder::decode(const ustring &datastr)
{
	const unsigned char *data = datastr.c_str();
	size_t length = datastr.length();
	// TODO: bounds checks

	uint8_t version = data[0];
	if (version < 22) {
		std::ostringstream oss;
		oss << "Unsupported map version " << (int)version;
		throw std::runtime_error(oss.str());
	}
	m_version = version;

	ustring datastr2;
	if (version >= 29) {
		// decompress whole block at once
		m_zstd_decompressor.setData(data, length, 1);
		datastr2 = m_zstd_decompressor.decompress();
		data = datastr2.c_str();
		length = datastr2.size();
	}

	size_t dataOffset = 0;
	if (version >= 29)
		dataOffset = 7;
	else if (version >= 27)
		dataOffset = 4;
	else
		dataOffset = 2;

	auto decode_mapping = [&] () {
		dataOffset++; // mapping version
		uint16_t numMappings = readU16(data + dataOffset);
		dataOffset += 2;
		for (int i = 0; i < numMappings; ++i) {
			uint16_t nodeId = readU16(data + dataOffset);
			dataOffset += 2;
			uint16_t nameLen = readU16(data + dataOffset);
			dataOffset += 2;
			std::string name(reinterpret_cast<const char *>(data) + dataOffset, nameLen);
			if (name == "air")
				m_blockAirId = nodeId;
			else if (name == "ignore")
				m_blockIgnoreId = nodeId;
			else
				m_nameMap[nodeId] = name;
			dataOffset += nameLen;
		}
	};

	if (version >= 29)
		decode_mapping();

	uint8_t contentWidth = data[dataOffset];
	dataOffset++;
	uint8_t paramsWidth = data[dataOffset];
	dataOffset++;
	if (contentWidth != 1 && contentWidth != 2)
		throw std::runtime_error("unsupported map version (contentWidth)");
	if (paramsWidth != 2)
		throw std::runtime_error("unsupported map version (paramsWidth)");
	m_contentWidth = contentWidth;

	if (version >= 29) {
		m_mapData.resize((contentWidth + paramsWidth) * 4096);
		m_mapData.assign(data + dataOffset, m_mapData.size());
		return; // we have read everything we need and can return early
	}

	// version < 29
	ZlibDecompressor decompressor(data, length);
	decompressor.setSeekPos(dataOffset);
	m_mapData = decompressor.decompress();
	decompressor.decompress(); // unused metadata
	dataOffset = decompressor.seekPos();

	// Skip unused node timers
	if (version == 23)
		dataOffset += 1;
	if (version == 24) {
		uint8_t ver = data[dataOffset++];
		if (ver == 1) {
			uint16_t num = readU16(data + dataOffset);
			dataOffset += 2;
			dataOffset += 10 * num;
		}
	}

	// Skip unused static objects
	dataOffset++; // Skip static object version
	int staticObjectCount = readU16(data + dataOffset);
	dataOffset += 2;
	for (int i = 0; i < staticObjectCount; ++i) {
		dataOffset += 13;
		uint16_t dataSize = readU16(data + dataOffset);
		dataOffset += dataSize + 2;
	}
	dataOffset += 4; // Skip timestamp

	// Read mapping
	decode_mapping();
}

bool BlockDecoder::isEmpty() const
{
	// only contains ignore and air nodes?
	return m_nameMap.empty();
}

const static std::string empty;

const std::string &BlockDecoder::getNode(u8 x, u8 y, u8 z) const
{
	unsigned int position = x + (y << 4) + (z << 8);
	uint16_t content = readBlockContent(m_mapData.c_str(), m_contentWidth, position);
	if (content == m_blockAirId || content == m_blockIgnoreId)
		return empty;
	NameMap::const_iterator it = m_nameMap.find(content);
	if (it == m_nameMap.end()) {
		std::cerr << "Skipping node with invalid ID." << std::endl;
		return empty;
	}
	return it->second;
}