/*
 * Copyright 2015 - 2022 gary@drinkingtea.net
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

#undef NDEBUG

#include <iostream>
#include <map>
#include <ox/mc/mc.hpp>
#include <ox/model/model.hpp>
#include <ox/std/std.hpp>

union TestUnion {
	static constexpr auto TypeName = "TestUnion";
	static constexpr auto TypeVersion = 1;
	bool Bool;
	uint32_t Int = 5;
	char *CString;
};

struct TestStructNest {
	static constexpr auto TypeName = "TestStructNest";
	static constexpr auto TypeVersion = 1;
	bool Bool = false;
	uint32_t Int = 0;
	ox::BString<32> BString = "";
};

struct TestStruct {
	static constexpr auto TypeName = "TestStruct";
	static constexpr auto TypeVersion = 1;
	bool Bool = false;
	int32_t Int = 0;
	int32_t Int1 = 0;
	int32_t Int2 = 0;
	int32_t Int3 = 0;
	int32_t Int4 = 0;
	int32_t Int5 = 0;
	int32_t Int6 = 0;
	int32_t Int7 = 0;
	int32_t Int8 = 0;
	int unionIdx = 1;
	TestUnion Union;
	ox::String String = "";
	ox::BString<32> BString = "";
	uint32_t List[4] = {0, 0, 0, 0};
	ox::Vector<uint32_t> Vector = {1, 2, 3, 4, 5};
	ox::HashMap<ox::String, int> Map;
	TestStructNest EmptyStruct;
	TestStructNest Struct;
	constexpr ~TestStruct() noexcept {
		if (unionIdx == 2) {
			ox::safeDelete(Union.CString);
		}
	}
};

template<typename T>
constexpr ox::Error model(T *io, ox::CommonPtrWith<TestUnion> auto *obj) noexcept {
	io->template setTypeInfo<TestUnion>();
	oxReturnError(io->field("Bool", &obj->Bool));
	oxReturnError(io->field("Int", &obj->Int));
	oxReturnError(io->fieldCString("CString", &obj->CString));
	return OxError(0);
}

oxModelBegin(TestStructNest)
	oxModelField(Bool)
	oxModelField(Int)
	oxModelField(BString)
oxModelEnd()

template<typename T>
constexpr ox::Error model(T *io, ox::CommonPtrWith<TestStruct> auto *obj) noexcept {
	io->template setTypeInfo<TestStruct>();
	oxReturnError(io->field("Vector", &obj->Vector));
	oxReturnError(io->field("Bool", &obj->Bool));
	oxReturnError(io->field("Int", &obj->Int));
	oxReturnError(io->field("Int1", &obj->Int1));
	oxReturnError(io->field("Int2", &obj->Int2));
	oxReturnError(io->field("Int3", &obj->Int3));
	oxReturnError(io->field("Int4", &obj->Int4));
	oxReturnError(io->field("Int5", &obj->Int5));
	oxReturnError(io->field("Int6", &obj->Int6));
	oxReturnError(io->field("Int7", &obj->Int7));
	oxReturnError(io->field("Int8", &obj->Int8));
	oxReturnError(io->field("unionIdx", &obj->unionIdx));
	if (ox_strcmp(io->opType(), ox::OpType::Reflect) == 0) {
		oxReturnError(io->field("Union", ox::UnionView{&obj->Union, 0}));
	} else {
		oxReturnError(io->field("Union", ox::UnionView{&obj->Union, obj->unionIdx}));
	}
	oxReturnError(io->field("String", &obj->String));
	oxReturnError(io->field("BString", &obj->BString));
	oxReturnError(io->field("List", obj->List, 4));
	oxReturnError(io->field("Map", &obj->Map));
	oxReturnError(io->field("EmptyStruct", &obj->EmptyStruct));
	oxReturnError(io->field("Struct", &obj->Struct));
	return OxError(0);
}

std::map<ox::String, ox::Error(*)()> tests = {
	{
		{
			"MetalClawWriter",
			[] {
				// This test doesn't confirm much, but it does show that the writer
				// doesn't segfault
				static constexpr size_t buffLen = 1024;
				char buff[buffLen];
				TestStruct ts;
				oxReturnError(ox::writeMC(buff, buffLen, &ts));
				return OxError(0);
			}
		},

		{
			"MetalClawReader",
			[] {
				// setup for tests
				static constexpr size_t buffLen = 1024;
				char buff[buffLen];
				TestStruct testIn, testOut;
				testIn.Bool = true;
				testIn.Int = 42;
				testIn.Union.Int = 42;
				testIn.String = "Test String 0";
				testIn.BString = "Test String 1";
				testIn.List[0] = 1;
				testIn.List[1] = 2;
				testIn.List[2] = 3;
				testIn.List[3] = 4;
				testIn.Map["asdf"] = 93;
				testIn.Map["aoeu"] = 94;
				testIn.Struct.Bool = false;
				testIn.Struct.Int = 300;
				testIn.Struct.BString = "Test String 2";
				// run tests
				oxAssert(ox::writeMC(buff, buffLen, &testIn), "writeMC failed");
				oxAssert(ox::readMC(buff, buffLen, &testOut), "readMC failed");
				//std::cout << testIn.Union.Int << "|" << testOut.Union.Int << "|\n";
				oxAssert(testIn.Bool                == testOut.Bool,                "Bool value mismatch");
				oxAssert(testIn.Int                 == testOut.Int,                 "Int value mismatch");
				oxAssert(testIn.Int1                == testOut.Int1,                "Int1 value mismatch");
				oxAssert(testIn.Int2                == testOut.Int2,                "Int2 value mismatch");
				oxAssert(testIn.Int3                == testOut.Int3,                "Int3 value mismatch");
				oxAssert(testIn.Int4                == testOut.Int4,                "Int4 value mismatch");
				oxAssert(testIn.Int5                == testOut.Int5,                "Int5 value mismatch");
				oxAssert(testIn.Int6                == testOut.Int6,                "Int6 value mismatch");
				oxAssert(testIn.Int7                == testOut.Int7,                "Int7 value mismatch");
				oxAssert(testIn.Int8                == testOut.Int8,                "Int8 value mismatch");
				oxAssert(testIn.Union.Int           == testOut.Union.Int,           "Union.Int value mismatch");
				oxAssert(testIn.String              == testOut.String,              "String value mismatch");
				oxAssert(testIn.BString             == testOut.BString,             "BString value mismatch");
				oxAssert(testIn.List[0]             == testOut.List[0],             "List[0] value mismatch");
				oxAssert(testIn.List[1]             == testOut.List[1],             "List[1] value mismatch");
				oxAssert(testIn.List[2]             == testOut.List[2],             "List[2] value mismatch");
				oxAssert(testIn.List[3]             == testOut.List[3],             "List[3] value mismatch");
				oxAssert(testIn.Vector[0]           == testOut.Vector[0],           "Vector[0] value mismatch");
				oxAssert(testIn.Vector[1]           == testOut.Vector[1],           "Vector[1] value mismatch");
				oxAssert(testIn.Vector[2]           == testOut.Vector[2],           "Vector[2] value mismatch");
				oxAssert(testIn.Vector[3]           == testOut.Vector[3],           "Vector[3] value mismatch");
				oxAssert(testIn.Map["asdf"]         == testOut.Map["asdf"],         "Map[\"asdf\"] value mismatch");
				oxAssert(testIn.Map["aoeu"]         == testOut.Map["aoeu"],         "Map[\"aoeu\"] value mismatch");
				oxAssert(testIn.EmptyStruct.Bool    == testOut.EmptyStruct.Bool,    "EmptyStruct.Bool value mismatch");
				oxAssert(testIn.EmptyStruct.Int     == testOut.EmptyStruct.Int,     "EmptyStruct.Int value mismatch");
				oxAssert(testIn.EmptyStruct.BString == testOut.EmptyStruct.BString, "EmptyStruct.BString value mismatch");
				oxAssert(testIn.Struct.Int          == testOut.Struct.Int,          "Struct.Int value mismatch");
				oxAssert(testIn.Struct.BString      == testOut.Struct.BString,      "Struct.BString value mismatch");
				oxAssert(testIn.Struct.Bool         == testOut.Struct.Bool,         "Struct.Bool value mismatch");
				return OxError(0);
			}
		},

		{
			"encodeInteger",
			[] {
				using ox::MaxValue;
				using ox::mc::McInt;
				using ox::mc::encodeInteger;
				static constexpr auto check = [](McInt val, const ox::Vector<uint8_t, 9> &expected) {
					if (val.length != expected.size()) {
						std::cout << "val.length: " << val.length << ", expected: " << expected.size() << '\n';
						return OxError(1);
					}
					for (std::size_t i = 0; i < expected.size(); i++) {
						if (expected[i] != val.data[i]) {
							std::cout << "decoded: " << static_cast<uint32_t>(val.data[i]) << ", expected: " << static_cast<uint32_t>(expected[i]) << '\n';
							std::cout << "decoded: " << i << ": " << static_cast<uint32_t>(val.data[i]) << '\n';
							return OxError(1);
						}
					}
					return OxError(0);
				};
				constexpr auto check64 = [](McInt val, auto expected) {
					if (val.length != 9) {
						std::cout << "val.length: " << val.length << '\n';
						return OxError(1);
					}
					ox::LittleEndian<decltype(expected)> decoded = *reinterpret_cast<decltype(expected)*>(&val.data[1]);
					if (expected != decoded) {
						std::cout << "decoded: " << decoded << ", expected: " << expected << '\n';
						return OxError(1);
					}
					return OxError(0);
				};
				// signed positive
				oxAssert(check(encodeInteger(int64_t(1)), {0b000'0001'0}), "Encode 1 fail");
				oxAssert(check(encodeInteger(int64_t(2)), {0b000'0010'0}), "Encode 2 fail");
				oxAssert(check(encodeInteger(int64_t(3)), {0b000'0011'0}), "Encode 3 fail");
				oxAssert(check(encodeInteger(int64_t(4)), {0b000'0100'0}), "Encode 4 fail");
				oxAssert(check(encodeInteger(int64_t(64)), {0b00'0000'01, 0b1}), "Encode 64 fail");
				oxAssert(check(encodeInteger(int64_t(128)), {0b00'0000'01, 0b10}), "Encode 128 fail");
				oxAssert(check(encodeInteger(int64_t(129)), {0b00'0001'01, 0b10}), "Encode 129 fail");
				oxAssert(check(encodeInteger(int64_t(130)), {0b00'0010'01, 0b10}), "Encode 130 fail");
				oxAssert(check(encodeInteger(int64_t(131)), {0b00'0011'01, 0b10}), "Encode 131 fail");
				// signed negative
				oxAssert(check(encodeInteger(  int64_t(-1)), {0b111'1111'0}), "Encode -1 fail");
				oxAssert(check(encodeInteger(  int64_t(-2)), {0b111'1110'0}), "Encode -2 fail");
				oxAssert(check(encodeInteger(  int64_t(-3)), {0b111'1101'0}), "Encode -3 fail");
				oxAssert(check(encodeInteger(  int64_t(-4)), {0b111'1100'0}), "Encode -4 fail");
				oxAssert(check(encodeInteger( int64_t(-64)), {0b100'0000'0}), "Encode -64 fail");
				oxAssert(check(encodeInteger(int64_t(-128)), {0b00'0000'01, 0b11'1111'10}), "Encode -128 fail");
				oxAssert(check(encodeInteger(int64_t(-129)), {0b11'1111'01, 0b11'1111'01}), "Encode -129 fail");
				oxAssert(check(encodeInteger(int64_t(-130)), {0b11'1110'01, 0b11'1111'01}), "Encode -130 fail");
				oxAssert(check(encodeInteger(int64_t(-131)), {0b11'1101'01, 0b11'1111'01}), "Encode -131 fail");
				// unsigned
				oxAssert(check(encodeInteger(uint32_t(0xffffffff)), {0b11101111, 255, 255, 255, 0b00011111}), "Encode 0xffffffff fail");
				oxAssert(check(encodeInteger(uint64_t(1)), {0b0010}), "Encode 1 fail");
				oxAssert(check(encodeInteger(uint64_t(2)), {0b0100}), "Encode 2 fail");
				oxAssert(check(encodeInteger(uint64_t(3)), {0b0110}), "Encode 3 fail");
				oxAssert(check(encodeInteger(uint64_t(4)), {0b1000}), "Encode 4 fail");
				oxAssert(check(encodeInteger(uint64_t(64)), {0b1000'000'0}), "Encode 4 fail");
				oxAssert(check(encodeInteger(uint64_t(128)), {0b0001, 0b10}), "Encode 128 fail");
				oxAssert(check(encodeInteger(uint64_t(129)), {0b0101, 0b10}), "Encode 129 fail");
				oxAssert(check(encodeInteger(uint64_t(130)), {0b1001, 0b10}), "Encode 130 fail");
				oxAssert(check(encodeInteger(uint64_t(131)), {0b1101, 0b10}), "Encode 131 fail");
				// Signed check needs lambda templates to run correctly without
				// code deduplication
				oxAssert(check64(encodeInteger(MaxValue<int64_t>), MaxValue<int64_t>), "Encode MaxValue<int64_t> fail");
				oxAssert(check64(encodeInteger(MaxValue<uint64_t>), MaxValue<uint64_t>), "Encode MaxValue<uint64_t> fail");
				return OxError(0);
			}
		},

		{
			"decodeInteger",
			[] {
				using ox::MaxValue;
				using ox::mc::McInt;
				using ox::mc::encodeInteger;
				using ox::mc::decodeInteger;
				static constexpr auto check = [](auto val) {
					auto result = decodeInteger<decltype(val)>(encodeInteger(val));
					oxReturnError(result.error);
					if (result.value != val) {
						std::cout << "Bad value: " << result.value << ", expected: " << val << '\n';
						return OxError(1);
					}
					return OxError(0);
				};
				oxAssert(check(uint32_t(14)), "Decode of 14 failed.");
				oxAssert(check(int64_t(-1)), "Decode of -1 failed.");
				oxAssert(check(int64_t(-2)), "Decode of -2 failed.");
				oxAssert(check(int64_t(-127)), "Decode of -127 failed.");
				oxAssert(check(int64_t(-128)), "Decode of -128 failed.");
				oxAssert(check(int64_t(-129)), "Decode of -129 failed.");
				oxAssert(check(int64_t(-129000)), "Decode of -129000 failed.");
				oxAssert(check(int64_t(1)), "Decode of 1 failed.");
				oxAssert(check(int64_t(2)), "Decode of 2 failed.");
				oxAssert(check(int64_t(42)), "Decode of 42 failed.");
				oxAssert(check(int64_t(130)), "Decode of 130 failed.");
				oxAssert(check(int64_t(131)), "Decode of 131 failed.");
				oxAssert(check(int64_t(131000)), "Decode of 131000 failed.");
				oxAssert(check(uint64_t(1)), "Decode of 1 failed.");
				oxAssert(check(uint64_t(2)), "Decode of 2 failed.");
				oxAssert(check(uint64_t(42)), "Decode of 42 failed.");
				oxAssert(check(uint64_t(130)), "Decode of 130 failed.");
				oxAssert(check(uint64_t(131)), "Decode of 131 failed.");
				oxAssert(check(0xffffffff), "Decode of 0xffffffff failed.");
				oxAssert(check(0xffffffffffff), "Decode of 0xffffffffffff failed.");
				oxAssert(check(0xffffffffffffffff), "Decode of U64 max failed.");
				return OxError(0);
			}
		},


		{
			"MetalClawModelValue",
			[] {
				static constexpr size_t dataBuffLen = ox::units::MB;
				ox::Buffer dataBuff(dataBuffLen);
				TestStruct testIn;
				testIn.Bool = true;
				testIn.Int = 42;
				testIn.BString = "Test String 1";
				testIn.List[0] = 1;
				testIn.List[1] = 2;
				testIn.List[2] = 3;
				testIn.List[3] = 4;
				testIn.Struct.Bool = true;
				testIn.Struct.Int = 300;
				testIn.Struct.BString = "Test String 2";
				testIn.unionIdx = 1;
				testIn.Union.Int = 93;
				oxAssert(ox::writeMC(dataBuff.data(), dataBuff.size(), &testIn), "Data generation failed");
				ox::TypeStore typeStore;
				auto type = ox::buildTypeDef(&typeStore, &testIn);
				oxAssert(type.error, "Descriptor write failed");
				ox::ModelObject testOut;
				oxReturnError(testOut.setType(type.value));
				oxAssert(ox::readMC(dataBuff.data(), dataBuff.size(), &testOut), "Data read failed");
				oxAssert(testOut["Int"].get<int>() == testIn.Int, "testOut.Int failed");
				oxAssert(testOut["Bool"].get<bool>() == testIn.Bool, "testOut.Bool failed");
				oxAssert(testOut["BString"].get<ox::String>() == testIn.BString.c_str(), "testOut.String failed");
				oxAssert(testOut["String"].get<ox::String>() == testIn.String, "testOut.String failed");
				auto &testOutStruct = testOut["Struct"].get<ox::ModelObject>();
				auto &testOutUnion = testOut["Union"].get<ox::ModelUnion>();
				auto &testOutList = testOut["List"].get<ox::ModelValueVector>();
				auto testOutStructCopy = testOut["Struct"].get<ox::ModelObject>();
				auto testOutUnionCopy = testOut["Union"].get<ox::ModelUnion>();
				auto testOutListCopy = testOut["List"].get<ox::ModelValueVector>();
				oxAssert(testOutStruct.typeName() == TestStructNest::TypeName, "ModelObject TypeName failed");
				oxAssert(testOutStruct.typeVersion() == TestStructNest::TypeVersion, "ModelObject TypeVersion failed");
				oxAssert(testOutStruct["Bool"].get<bool>() == testIn.Struct.Bool, "testOut.Struct.Bool failed");
				oxAssert(testOutStruct["BString"].get<ox::String>() == testIn.Struct.BString.c_str(), "testOut.Struct.BString failed");
				oxAssert(testOut["unionIdx"].get<int>() == testIn.unionIdx, "testOut.unionIdx failed");
				oxAssert(testOutUnion.unionIdx() == testIn.unionIdx, "testOut.Union idx wrong");
				oxAssert(testOutUnion["Int"].get<uint32_t>() == testIn.Union.Int, "testOut.Union.Int failed");
				oxAssert(testOutList[0].get<uint32_t>() == testIn.List[0], "testOut.List[0] failed");
				oxAssert(testOutList[1].get<uint32_t>() == testIn.List[1], "testOut.Struct.List[1] failed");
				oxAssert(testOutStructCopy["Bool"].get<bool>() == testIn.Struct.Bool, "testOut.Struct.Bool (copy) failed");
				oxAssert(testOutStructCopy["BString"].get<ox::String>() == testIn.Struct.BString.c_str(), "testOut.Struct.BString (copy) failed");
				oxAssert(testOutListCopy[0].get<uint32_t>() == testIn.List[0], "testOut.Struct.List[0] (copy) failed");
				oxAssert(testOutListCopy[1].get<uint32_t>() == testIn.List[1], "testOut.Struct.List[1] (copy) failed");
				return OxError(0);
			}
		},

		{
			"MetalClawDef",
			[] {
				 //constexpr size_t descBuffLen = 1024;
				 //uint8_t descBuff[descBuffLen];
				 static constexpr size_t dataBuffLen = ox::units::MB;
				 char dataBuff[dataBuffLen];
				 TestStruct testIn, testOut;
				 testIn.Bool = true;
				 testIn.Int = 42;
				 testIn.BString = "Test String 1";
				 testIn.List[0] = 1;
				 testIn.List[1] = 2;
				 testIn.List[2] = 3;
				 testIn.List[3] = 4;
				 testIn.Struct.Bool = false;
				 testIn.Struct.Int = 300;
				 testIn.Struct.BString = "Test String 2";
				 oxAssert(ox::writeMC(dataBuff, dataBuffLen, &testIn), "Data generation failed");
				 ox::TypeStore typeStore;
				 auto type = ox::buildTypeDef(&typeStore, &testIn);
				 oxAssert(type.error, "Descriptor write failed");
				 oxReturnError(ox::walkModel<ox::MetalClawReader>(type.value, dataBuff, dataBuffLen,
					[](const ox::Vector<ox::FieldName>&, const ox::Vector<ox::String>&, const ox::DescriptorField &f, ox::MetalClawReader *rdr) -> ox::Error {
						//std::cout << f.fieldName.c_str() << '\n';
						auto fieldName = f.fieldName.c_str();
						switch (f.type->primitiveType) {
							case ox::PrimitiveType::UnsignedInteger:
							  std::cout << fieldName << ":\tuint" << f.type->length * 8 << "_t:\t";
							  switch (f.type->length) {
									case 1: {
										uint8_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 2: {
										uint16_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 4: {
										uint32_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 8: {
										uint64_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
								}
								std::cout << '\n';
								break;
							case ox::PrimitiveType::SignedInteger:
								std::cout << fieldName << ":\tint" << f.type->length * 8 << "_t:\t";
								switch (f.type->length) {
									case 1: {
										int8_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 2: {
										int16_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 4: {
										int32_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
									case 8: {
										int64_t i = {};
										oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
										std::cout << i;
										break;
									}
								}
								std::cout << '\n';
								break;
							case ox::PrimitiveType::Bool: {
								bool i = {};
								oxAssert(rdr->field(fieldName, &i), "Walking model failed.");
								std::cout << fieldName << ":\t" << "bool:\t\t" << (i ? "true" : "false") << '\n';
								break;
							}
							case ox::PrimitiveType::String: {
								ox::String s;
								//std::cout << rdr->stringLength() << '\n';
								oxAssert(rdr->field(fieldName, &s), "Walking model failed.");
								oxOutf("{}:\tstring:\t\t{}\n", fieldName, s);
								break;
							}
							case ox::PrimitiveType::Struct:
								break;
							case ox::PrimitiveType::Union:
								break;
						}
						return OxError(0);
					}
				));
				return OxError(0);
			}
		},
	}
};

int main(int argc, const char **args) {
	if (argc > 0) {
		auto testName = args[1];
		if (tests.find(testName) != tests.end()) {
			oxAssert(tests[testName](), "Test failed...");
		}
	}
	return 0;
}
