#include "client_https.hpp"
#include "server_https.hpp"

//Added for the json-example
#define BOOST_SPIRIT_THREADSAFE
#include <boost/property_tree/json_parser.hpp>
#include <boost/property_tree/ptree.hpp>

//Added for the default_resource example
#include "crypto.hpp"
#include <algorithm>
#include <boost/filesystem.hpp>
#include <fstream>
#include <vector>

using namespace std;
//Added for the json-example:
using namespace boost::property_tree;

typedef SimpleWeb::Server<SimpleWeb::HTTPS> HttpsServer;
typedef SimpleWeb::Client<SimpleWeb::HTTPS> HttpsClient;

//Added for the default_resource example
void default_resource_send(const HttpsServer &server, const shared_ptr<HttpsServer::Response> &response, const shared_ptr<ifstream> &ifs);

int main() {
  //HTTPS-server at port 8080 using 1 thread
  //Unless you do more heavy non-threaded processing in the resources,
  //1 thread is usually faster than several threads
  HttpsServer server("server.crt", "server.key");
  server.config.port = 8080;

  //Add resources using path-regex and method-string, and an anonymous function
  //POST-example for the path /string, responds the posted string
  server.resource["^/string$"]["POST"] = [](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> request) {
    //Retrieve string:
    auto content = request->content.string();
    //request->content.string() is a convenience function for:
    //stringstream ss;
    //ss << request->content.rdbuf();
    //auto content=ss.str();

    *response << "HTTP/1.1 200 OK\r\nContent-Length: " << content.length() << "\r\n\r\n"
              << content;


    // Alternatively, use one of the convenience functions, for instance:
    // response->write(content);
  };

  //POST-example for the path /json, responds firstName+" "+lastName from the posted json
  //Responds with an appropriate error message if the posted json is not valid, or if firstName or lastName is missing
  //Example posted json:
  //{
  //  "firstName": "John",
  //  "lastName": "Smith",
  //  "age": 25
  //}
  server.resource["^/json$"]["POST"] = [](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> request) {
    try {
      ptree pt;
      read_json(request->content, pt);

      auto name = pt.get<string>("firstName") + " " + pt.get<string>("lastName");

      *response << "HTTP/1.1 200 OK\r\n"
                << "Content-Type: application/json\r\n"
                << "Content-Length: " << name.length() << "\r\n\r\n"
                << name;
    }
    catch(const exception &e) {
      *response << "HTTP/1.1 400 Bad Request\r\nContent-Length: " << strlen(e.what()) << "\r\n\r\n"
                << e.what();
    }


    // Alternatively, using a convenience function:
    // try {
    //     ptree pt;
    //     read_json(request->content, pt);

    //     auto name=pt.get<string>("firstName")+" "+pt.get<string>("lastName");
    //     response->write(name, {{"Content-Type", "application/json"}});
    // }
    // catch(const exception &e) {
    //     response->write(SimpleWeb::StatusCode::client_error_bad_request, e.what());
    // }
  };

  //GET-example for the path /info
  //Responds with request-information
  server.resource["^/info$"]["GET"] = [](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> request) {
    stringstream stream;
    stream << "<h1>Request from " << request->remote_endpoint_address << " (" << request->remote_endpoint_port << ")</h1>";
    stream << request->method << " " << request->path << " HTTP/" << request->http_version << "<br>";
    for(auto &header : request->header)
      stream << header.first << ": " << header.second << "<br>";

    //find length of content_stream (length received using content_stream.tellp())
    stream.seekp(0, ios::end);

    *response << "HTTP/1.1 200 OK\r\nContent-Length: " << stream.tellp() << "\r\n\r\n"
              << stream.rdbuf();


    // Alternatively, using a convenience function:
    // stringstream stream;
    // stream << "<h1>Request from " << request->remote_endpoint_address << " (" << request->remote_endpoint_port << ")</h1>";
    // stream << request->method << " " << request->path << " HTTP/" << request->http_version << "<br>";
    // for(auto &header: request->header)
    //     stream << header.first << ": " << header.second << "<br>";
    // response->write(stream);
  };

  //GET-example for the path /match/[number], responds with the matched string in path (number)
  //For instance a request GET /match/123 will receive: 123
  server.resource["^/match/([0-9]+)$"]["GET"] = [](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> request) {
    string number = request->path_match[1];
    *response << "HTTP/1.1 200 OK\r\nContent-Length: " << number.length() << "\r\n\r\n"
              << number;


    // Alternatively, using a convenience function:
    // response->write(request->path_match[1]);
  };

  //Get example simulating heavy work in a separate thread
  server.resource["^/work$"]["GET"] = [](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> /*request*/) {
    thread work_thread([response] {
      this_thread::sleep_for(chrono::seconds(5));
      response->write("Work done");
    });
    work_thread.detach();
  };

  //Default GET-example. If no other matches, this anonymous function will be called.
  //Will respond with content in the web/-directory, and its subdirectories.
  //Default file: index.html
  //Can for instance be used to retrieve an HTML 5 client that uses REST-resources on this server
  server.default_resource["GET"] = [&server](shared_ptr<HttpsServer::Response> response, shared_ptr<HttpsServer::Request> request) {
    try {
      auto web_root_path = boost::filesystem::canonical("web");
      auto path = boost::filesystem::canonical(web_root_path / request->path);
      //Check if path is within web_root_path
      if(distance(web_root_path.begin(), web_root_path.end()) > distance(path.begin(), path.end()) ||
         !equal(web_root_path.begin(), web_root_path.end(), path.begin()))
        throw invalid_argument("path must be within root path");
      if(boost::filesystem::is_directory(path))
        path /= "index.html";

      SimpleWeb::CaseInsensitiveMultimap header;

//    Uncomment the following line to enable Cache-Control
//    header.emplace("Cache-Control", "max-age=86400");

#ifdef HAVE_OPENSSL
//    Uncomment the following lines to enable ETag
//    {
//      ifstream ifs(path.string(), ifstream::in | ios::binary);
//      if(ifs) {
//        auto hash = SimpleWeb::Crypto::to_hex_string(SimpleWeb::Crypto::md5(ifs));
//        header.emplace("ETag", "\"" + hash + "\"");
//        auto it = request->header.find("If-None-Match");
//        if(it != request->header.end()) {
//          if(!it->second.empty() && it->second.compare(1, hash.size(), hash) == 0) {
//            response->write(SimpleWeb::StatusCode::redirection_not_modified, header);
//            return;
//          }
//        }
//      }
//      else
//        throw invalid_argument("could not read file");
//    }
#endif

      auto ifs = make_shared<ifstream>();
      ifs->open(path.string(), ifstream::in | ios::binary | ios::ate);

      if(*ifs) {
        auto length = ifs->tellg();
        ifs->seekg(0, ios::beg);

        header.emplace("Content-Length", to_string(length));
        response->write(header);
        default_resource_send(server, response, ifs);
      }
      else
        throw invalid_argument("could not read file");
    }
    catch(const exception &e) {
      response->write(SimpleWeb::StatusCode::client_error_bad_request, "Could not open path " + request->path + ": " + e.what());
    }
  };

  server.on_error = [](std::shared_ptr<HttpsServer::Request> /*request*/, const SimpleWeb::error_code & /*ec*/) {
    // handle errors here
  };

  thread server_thread([&server]() {
    //Start server
    server.start();
  });

  //Wait for server to start so that the client can connect
  this_thread::sleep_for(chrono::seconds(1));

  //Client examples
  //Second Client() parameter set to false: no certificate verification
  HttpsClient client("localhost:8080", false);

  // synchronous request examples
  auto r1 = client.request("GET", "/match/123");
  cout << r1->content.rdbuf() << endl; // Alternatively, use the convenience function r1->content.string()

  string json_string = "{\"firstName\": \"John\",\"lastName\": \"Smith\",\"age\": 25}";
  auto r2 = client.request("POST", "/string", json_string);
  cout << r2->content.rdbuf() << endl;

  // asynchronous request example
  client.request("POST", "/json", json_string, [](std::shared_ptr<HttpsClient::Response> response, const SimpleWeb::error_code &ec) {
    if(!ec)
      cout << response->content.rdbuf() << endl;
  });
  client.io_service->reset(); // needed because the io_service has been run already in the synchronous examples
  client.io_service->run();

  server_thread.join();
}

void default_resource_send(const HttpsServer &server, const shared_ptr<HttpsServer::Response> &response, const shared_ptr<ifstream> &ifs) {
  //read and send 128 KB at a time
  static vector<char> buffer(131072); // Safe when server is running on one thread
  streamsize read_length;
  if((read_length = ifs->read(&buffer[0], buffer.size()).gcount()) > 0) {
    response->write(&buffer[0], read_length);
    if(read_length == static_cast<streamsize>(buffer.size())) {
      server.send(response, [&server, response, ifs](const SimpleWeb::error_code &ec) {
        if(!ec)
          default_resource_send(server, response, ifs);
        else
          cerr << "Connection interrupted" << endl;
      });
    }
  }
}