#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/json_parser.hpp>
#include <boost/version.hpp>

#include <string>
#include <set>
#include <exception>
#include <iostream>
#include <sstream>

static std::string original_problem("{\n    \"pipename\": \"quantiser(decode_lut_string=<verbatim>\\u0010\\u0000 \\u0000@\\u0000\200\\u0000\\u0000\\u0001\\u0000\\u0002\\u0000\\u0004\\u0000\\b\377\377\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000\\u0000<\\/verbatim>)\",\n    \"raw\": {\n        \"type\": \"t\",\n        \"rank\": \"3\",\n        \"shape\": {\n            \"dim\": \"256\",\n            \"dim\": \"128\",\n            \"dim\": \"128\"\n        }\n    },\n    \"encoded\": {\n        \"bytes\": \"4194304\"\n    }\n}\n",3336);

static std::string reduced("{\n    \"pipename\": \"quantiser(decode_lut_string=<verbatim>\\u0000@\\u0000\200\\u0000<\\/verbatim>)\",\n    \"raw\": {\n        \"type\": \"t\",\n        \"rank\": \"3\",\n        \"shape\": {\n            \"dim\": \"256\",\n            \"dim\": \"128\",\n            \"dim\": \"128\"\n        }\n    },\n    \"encoded\": {\n        \"bytes\": \"4194304\"\n    }\n}\n",296);

static std::string problem = reduced;

namespace pt = boost::property_tree;


struct content {

    int some_data = 42;
    std::string some_string = "howdy";
    std::string some_strange_string = problem;

    void clear(){

        some_data = 0;
        some_string.clear();
        some_strange_string.clear();

    }

    bool operator==(const content& _rhs){

        bool value = true;

        value = value && this->some_data == _rhs.some_data;
        value = value && this->some_string == _rhs.some_string;
        value = value && this->some_string.size() == _rhs.some_string.size();
        value = value && this->some_strange_string == _rhs.some_strange_string;
        value = value && this->some_strange_string.size() == _rhs.some_strange_string.size();

        return value;
    }

    friend std::ostream& operator<<(std::ostream & _cout,
                                   const content& _rhs){

        _cout << "some_data = " << _rhs.some_data << "\n";
        _cout << "some_string = " << _rhs.some_string << "\n";
        _cout << "some_strange_string = " << _rhs.some_strange_string << "\n";

        return _cout;
    }
};

std::string save(const content& _input)
{
    // Create an empty property tree object.
    pt::ptree tree;

    // Put the simple values into the tree. The integer is automatically
    // converted to a string. Note that the "debug" node is automatically
    // created if it doesn't exist.
    tree.put("data", _input.some_data);
    tree.put("string", _input.some_string);
    tree.put("strange_string", _input.some_strange_string);

    // Write property tree to JSON file
    std::stringstream output;
    pt::write_json(output, tree);

    return output.str();
}


content load(const std::string &payload)
{

    pt::ptree tree;
    content value;
    value.clear();


    std::stringstream input(payload);
    pt::read_json(input, tree);

    value.some_data = tree.get<int>("data");
    value.some_string = tree.get<std::string>("string");
    value.some_strange_string = tree.get<std::string>("strange_string");

    return value;
}


int main()
{
    try
    {
        content expected;
        std::string payload = save(expected);

        content received = load(payload);

        if(expected == received)
            std::cout << "Success (Boost version: " << BOOST_VERSION << ")\n";
        else{
            std::cout << "FAILED!! (Boost version: " << BOOST_VERSION << ")\n\n"
                      << "expected:\n" << expected << "\n\n"
                      << "received:\n" << received << "\n\n";
        }

    }
    catch (std::exception &e)
    {
        std::cout << "Error: " << e.what() << "\n(Boost version: " << BOOST_VERSION << ")\n";
    }
    return 0;
}