// bson-validate.cpp

#include <iostream>
#include <nlohmann/json-schema.hpp>
#include <nlohmann/json.hpp>

static int error_count = 0;

#define EXPECT_EQ(a, b)                                              \
	do {                                                             \
		if (a != b) {                                                \
			std::cerr << "Failed: '" << a << "' != '" << b << "'\n"; \
			error_count++;                                           \
		}                                                            \
	} while (0)

using json = nlohmann::json;
using validator = nlohmann::json_schema::json_validator;

// check binary data validation
const json bson_schema = json::parse(R"(
{
  "type": "object",
  "properties": {
    "standard_string": {
      "type": "string"
    },
    "binary_data": {
      "type": "string",
      "contentEncoding": "binary"
    }
  },
  "additionalProperties": false
}
)");

const json array_of_types = json::parse(R"(
{
  "type": "object",
  "properties": {
    "something": {
      "type": ["string", "number", "boolean"],
      "contentEncoding": "binary"
    }
  }
}
)");

const json array_of_types_without_binary = json::parse(R"(
{
  "type": "object",
  "properties": {
    "something": {
      "type": ["string", "number", "boolean"]
    }
  }
}
)");

class store_ptr_err_handler : public nlohmann::json_schema::basic_error_handler
{
	void error(const nlohmann::json::json_pointer &ptr, const json &, const std::string &message) override
	{
		nlohmann::json_schema::basic_error_handler::error(ptr, "", message);
		std::cerr << "ERROR: '" << ptr << "' - '"
		          << ""
		          << "': " << message << "\n";
		failed_pointers.push_back(ptr);
	}

public:
	std::vector<nlohmann::json::json_pointer> failed_pointers;

	void reset() override
	{
		nlohmann::json_schema::basic_error_handler::reset();
		failed_pointers.clear();
	}
};

static void content(const std::string &contentEncoding, const std::string &contentMediaType, const json &instance)
{
	if (instance.type() != json::value_t::binary)
		throw std::invalid_argument("invalid instance type for binary content checker");

	std::cerr << "mediaType: '" << contentMediaType << "', encoding: '" << contentEncoding << "'\n";

	if (contentMediaType != "")
		throw std::invalid_argument("unable to check for contentMediaType " + contentMediaType);

	if (contentEncoding == "binary") {

	} else if (contentEncoding != "")
		throw std::invalid_argument("unable to check for contentEncoding " + contentEncoding);
}

int main()
{
	validator val(nullptr, nullptr, content);

	// create some bson doc
	json::binary_t arr;
	std::string as_binary = "hello world";
	std::copy(as_binary.begin(), as_binary.end(), std::back_inserter(arr));

	json binary = json::binary(arr);

	store_ptr_err_handler err;

	/////////////////////////////////////
	val.set_root_schema(bson_schema);

	// all right
	val.validate({{"standard_string", "some string"}, {"binary_data", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 0);
	err.reset();

	// invalid binary data
	val.validate({{"binary_data", "string, but expect binary data"}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 2);
	EXPECT_EQ(err.failed_pointers[0].to_string(), "/binary_data");
	EXPECT_EQ(err.failed_pointers[1].to_string(), "/binary_data"); // second error comes from content-checker
	err.reset();

	// also check that simple string not accept binary data
	val.validate({{"standard_string", binary}, {"binary_data", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 1);
	EXPECT_EQ(err.failed_pointers[0].to_string(), "/standard_string");
	err.reset();

	/////////////////////////////////////
	// check with array of types

	// check simple types
	val.set_root_schema(array_of_types);
	val.validate({{"something", "string"}}, err);
	val.validate({{"something", 1}}, err);
	val.validate({{"something", false}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 4); // binary encoding invalidated all other types
	err.reset();

	// check binary data
	val.validate({{"something", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 0);
	err.reset();

	/////////////////////////////////////
	// and check that you can't set binary data if contentEncoding don't set
	val.set_root_schema(array_of_types_without_binary);
	val.validate({{"something", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 1);
	EXPECT_EQ(err.failed_pointers[0], "/something");
	err.reset();

	// check without content-callback everything fails
	validator val_no_content;

	/////////////////////////////////////
	val_no_content.set_root_schema(bson_schema);

	// all right
	val_no_content.validate({{"standard_string", "some string"}, {"binary_data", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 1);
	err.reset();

	// invalid binary data
	val_no_content.validate({{"binary_data", "string, but expect binary data"}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 2);
	EXPECT_EQ(err.failed_pointers[0].to_string(), "/binary_data");
	EXPECT_EQ(err.failed_pointers[1].to_string(), "/binary_data"); // second error comes from content-checker
	err.reset();

	// also check that simple string not accept binary data
	val_no_content.validate({{"standard_string", binary}, {"binary_data", binary}}, err);
	EXPECT_EQ(err.failed_pointers.size(), 2);
	EXPECT_EQ(err.failed_pointers[0].to_string(), "/binary_data");
	EXPECT_EQ(err.failed_pointers[1].to_string(), "/standard_string");
	err.reset();


	return error_count;
}