#include <forstio/test/suite.h>
#include <forstio/codec/json/json.h>

#include <iostream>

namespace {
namespace schema {
using namespace saw::schema;

using TestTuple = Tuple<
	String,
	Int32
>;

using TestArray = Array<
	String, 1
>;

using TestMultiArray = Array<
	String, 3
>;

using TestStruct = Struct<
	Member<Int32, "foo">,
	Member<String, "bar">
>;
}

SAW_TEST("UInt8 write"){
	using namespace saw;
	data<schema::UInt8, encode::Native> native_int;
	data<schema::UInt8, encode::Json> json_int;

	native_int.set(121);

	codec<schema::UInt8, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "121";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("UInt16 write"){
	using namespace saw;
	data<schema::UInt16, encode::Native> native_int;
	data<schema::UInt16, encode::Json> json_int;

	native_int.set(24413);

	codec<schema::UInt16, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "24413";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("UInt32 write"){
	using namespace saw;
	data<schema::UInt32, encode::Native> native_int;
	data<schema::UInt32, encode::Json> json_int;

	native_int.set(44123);

	codec<schema::UInt32, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "44123";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("UInt64 write"){
	using namespace saw;
	data<schema::UInt64, encode::Native> native_int;
	data<schema::UInt64, encode::Json> json_int;

	native_int.set(243345543);

	codec<schema::UInt64, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "243345543";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("Int8 write"){
	using namespace saw;
	data<schema::Int8, encode::Native> native_int;
	data<schema::Int8, encode::Json> json_int;

	native_int.set(-121);

	codec<schema::Int8, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "-121";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("Int16 write"){
	using namespace saw;
	data<schema::Int16, encode::Native> native_int;
	data<schema::Int16, encode::Json> json_int;

	native_int.set(-24413);

	codec<schema::Int16, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "-24413";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("Int32 write"){
	using namespace saw;
	data<schema::Int32, encode::Native> native_int;
	data<schema::Int32, encode::Json> json_int;

	native_int.set(44123);

	codec<schema::Int32, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "44123";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("Int64 write"){
	using namespace saw;
	data<schema::Int64, encode::Native> native_int;
	data<schema::Int64, encode::Json> json_int;

	native_int.set(243345543);

	codec<schema::Int64, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_int, json_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_v = "243345543";
	std::string enc_val = convert_to_string(json_int.get_buffer());
	SAW_EXPECT( enc_val == str_v, std::string{"Value is not being encoded correctly. Encoded: "} + enc_val );
}

SAW_TEST("String write"){
	using namespace saw;
	data<schema::String, encode::Native> nat_str;
	data<schema::String, encode::Json> json_str;

	nat_str.set("foo");

	codec<schema::String, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(nat_str, json_str);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	std::string_view str_view = "\"foo\"";
	std::string encoded_value = convert_to_string(json_str.get_buffer());

	SAW_EXPECT(encoded_value == str_view, "String not encoded correctly");
}

SAW_TEST("Tuple write"){
	using namespace saw;
	data<schema::TestTuple, encode::Native> native_tup;
	data<schema::TestTuple, encode::Json> json_tup;

	auto& nat_zero = native_tup.template get<0>();
	auto& nat_one = native_tup.template get<1>();
	
	nat_zero.set("bar");
	nat_one.set(34);
	
	codec<schema::TestTuple, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native_tup, json_tup);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	std::string_view str_v = "[\"bar\",34]";
	std::string enc_val = convert_to_string(json_tup.get_buffer());

	SAW_EXPECT(enc_val == str_v, std::string{"Tuple not encoded correctly. Encoded: "} + enc_val + std::string{" Expected: "} + std::string{str_v});
}

SAW_TEST("Array write"){
	using namespace saw;
	data<schema::TestArray, encode::Native> native{3u};
	data<schema::TestArray, encode::Json> json;

	native.at(0).set("foo");
	native.at(1).set("bar");
	native.at(2).set("baz");

	codec<schema::TestArray, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native, json);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	std::string_view str_v = "[\"foo\",\"bar\",\"baz\"]";
	std::string enc_val = convert_to_string(json.get_buffer());

	SAW_EXPECT(enc_val == str_v, std::string{"Array not encoded correctly. Encoded: "} + enc_val + std::string{" Expected: "} + std::string{str_v});
}

SAW_TEST("Three Dim Array write"){
	using namespace saw;
	data<schema::TestMultiArray, encode::Native> native{2,1,2};
	data<schema::TestMultiArray, encode::Json> json;

	native.at(0,0,0).set("multi");
	native.at(0,0,1).set("baz");
	native.at(1,0,0).set("foo");
	native.at(1,0,1).set("bar");

	codec<schema::TestMultiArray, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native, json);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	std::string_view str_v = "[[[\"multi\",\"baz\"]],[[\"foo\",\"bar\"]]]";
	std::string enc_val = convert_to_string(json.get_buffer());

	SAW_EXPECT(enc_val == str_v, std::string{"Array not encoded correctly. Encoded: "} + enc_val + std::string{" Expected: "} + std::string{str_v});
}

SAW_TEST("Struct write"){
	using namespace saw;
	data<schema::TestStruct, encode::Native> native;
	data<schema::TestStruct, encode::Json> json;

	native.get<"foo">().set(5);
	native.get<"bar">().set("baz");

	codec<schema::TestStruct, encode::Json> json_codec;

	error_or<void> eov = json_codec.encode(native, json);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	std::string_view str_v = "{\"foo\":5,\"bar\":\"baz\"}";
	std::string enc_val = convert_to_string(json.get_buffer());

	SAW_EXPECT(enc_val == str_v, std::string{"Struct not encoded correctly. Encoded: "} + enc_val + std::string{" Expected: "} + std::string{str_v});
}

SAW_TEST("Int8 read"){
	using namespace saw;
	data<schema::Int8, encode::Native> native_int;
	data<schema::Int8, encode::Json> json_int;

	codec<schema::Int8, encode::Json> json_codec;

	error_or<void> eov = json_codec.decode(json_int, native_int);
	SAW_EXPECT(eov.is_value(), "Encoding error");

	/**
	 * Currently doing this manually. Technically I should convert to std::string for tests
	 */
	typename native_data_type<schema::Int8>::type dec_val = 43;
	SAW_EXPECT( dec_val == native_int.get(), std::string{"Value is not being encoded correctly. Encoded: "} + std::to_string(static_cast<int>(native_int.get())));
}
}