#include "parse.hpp"

namespace binparse {

UnexpectedEOS::UnexpectedEOS()
: std::runtime_error("Unexpected end of stream.")
{}

UnexpectedEOS::UnexpectedEOS(std::__cxx11::string location)
: std::runtime_error("Unexpected end of stream after " + location)
{}

template<>
uint8_t parse<uint8_t>(std::istream& is, std::string name) {
	if(!is) {
		throw UnexpectedEOS();
	}
	char data = 0;
	is.read(&data, 1);
	if(!is) {
		throw UnexpectedEOS(name);
	}
	
	return (unsigned char) data;
}

template<>
uint16_t parse<uint16_t>(std::istream& is, std::string name) {
	if(!is) {
		throw UnexpectedEOS();
	}
	char data[2] = {0, 0};
	is.read(data, 2);
	if(!is) {
		throw UnexpectedEOS(name);
	}
	
	return (
		  ((unsigned char) data[1]) << 8)
		| (unsigned char) data[0];
}

template<>
uint32_t parse<uint32_t>(std::istream& is, std::string name) {
	if(!is) {
		throw UnexpectedEOS();
	}
	char data[4] = {0, 0};
	is.read(data, sizeof(data));
	if(!is) {
		throw UnexpectedEOS(name);
	}
	
	return (
		  ((unsigned char) data[3]) << 24
		| ((unsigned char) data[2]) << 16
		| ((unsigned char) data[1]) << 8
		| ((unsigned char) data[0])
	);
}

template<>
Value8 parse<Value8>(std::istream& is, std::string name) {
	return Value8(parse<uint8_t>(is, name));
}

template<>
Value16 parse<Value16>(std::istream& is, std::string name) {
	return Value16(parse<uint16_t>(is, name));
}

template<>
Magic16 parse<Magic16>(std::istream& is, std::string name) {
	return Magic16(parse<uint16_t>(is, name));
}

template<>
Offset16 parse<Offset16>(std::istream& is, std::string name) {
	return Offset16(parse<uint16_t>(is, name));
}

template<>
Value24 parse<Value24>(std::istream& is, std::__cxx11::string name)
{
	return Value24(parse<std::array<uint8_t, 3>>(is, name));	
}

template<>
Value32 parse<Value32>(std::istream& is, std::string name) {
	return Value32(parse<uint32_t>(is, name));
}

template<>
Offset32 parse<Offset32>(std::istream& is, std::string name) {
	return Offset32(parse<uint32_t>(is, name));
}

template<>
PString8 parse<PString8>(std::istream& is, std::string name) {
	uint8_t len = parse<uint8_t>(is, name + "_len");
	std::vector<char> data(len, 0);
	is.read(data.data(), len);
	return PString8(std::string(data.begin(), data.end()));
}

template<typename T, unsigned int N>
std::array<T, N> parse(std::istream& is, std::string name) {
	if(!is) {
		throw UnexpectedEOS();
	}
	
	std::array<T, N> buffer;
	is.read(reinterpret_cast<char*>(buffer.data()), buffer.size());
	
	if(!is) {
		throw UnexpectedEOS(name);
	}
	
	return buffer;
}

template<>
std::array<uint8_t, 3> parse<std::array<uint8_t, 3>>(std::istream& is, std::string name) {
	return parse<uint8_t, 3>(is, name);
}

template<>
std::array<uint8_t, 8> parse<std::array<uint8_t, 8>>(std::istream& is, std::string name) {
	return parse<uint8_t, 8>(is, name);
}

template<>
std::array<uint8_t, 22> parse<std::array<uint8_t, 22>>(std::istream& is, std::string name) {
	return parse<uint8_t, 22>(is, name);
}

void dump_bytes(std::istream& is, std::vector<uint8_t>& buffer, std::string name)
{
	if(!is) {
		throw UnexpectedEOS();
	}
	
	is.read(reinterpret_cast<char*>(buffer.data()), buffer.size());
	
	if(!is) {
		throw UnexpectedEOS(name);
	}
}

}