// Include necessary header files
#include <expresso/core/server.h>
#include <expresso/middleware/cacher.h>
#include <expresso/middleware/cookie_parser.h>
#include <expresso/middleware/cors.h>
#include <expresso/middleware/static_serve.h>

#include <brewtils/env.h>

// Personally, I don't encourange using namespaces, but, I left it here just so
// that the code could be more readable ¯\_(ツ)_/¯
using namespace expresso::core;
using namespace expresso::enums;
using namespace expresso::messages;
using namespace expresso::middleware;

// Global variable, just for fun :)
int port;

// About handler
void about(Request &req, Response &res) {
  json::object response;
  response["framework"] = "Expresso";
  response["repository"] = "https://github.com/coding-cpp/expresso";
  response["language"] = "C++";
  response["creator"]["name"] = "Adit Jain";
  response["creator"]["github"] = "https://github.com/jadit19";
  response["submodules"].resize(5);

  response["submodules"][0]["repository"] =
      "https://github.com/coding-cpp/logger";
  response["submodules"][0]["work"] = "Logging library for C++";
  response["submodules"][0]["location"] = "lib/logger";
  response["submodules"][0]["name"] = "Logger";

  response["submodules"][1]["repository"] =
      "https://github.com/coding-cpp/nexus";
  response["submodules"][1]["work"] = "Thread pool library for C++";
  response["submodules"][1]["location"] = "lib/nexus";
  response["submodules"][1]["name"] = "Nexus";

  response["submodules"][2]["repository"] =
      "https://github.com/coding-cpp/json";
  response["submodules"][2]["work"] = "JSON library for C++";
  response["submodules"][2]["location"] = "lib/json";
  response["submodules"][2]["name"] = "JSON";

  response["submodules"][3]["repository"] =
      "https://github.com/coding-cpp/brewtils";
  response["submodules"][3]["work"] = "Utilities library for C++";
  response["submodules"][3]["location"] = "lib/brewtils";
  response["submodules"][3]["name"] = "Brewtils";

  response["submodules"][4]["repository"] =
      "https://github.com/coding-cpp/zippuccino";
  response["submodules"][4]["work"] = "Zipping library for C++";
  response["submodules"][4]["location"] = "lib/zippuccino";
  response["submodules"][4]["name"] = "Zippuccino";

  json::object mochios;
  mochios["repository"] = "https://github.com/coding-cpp/mochios";
  mochios["work"] = "HTTP request library for C++";
  mochios["location"] = "lib/mochios";
  mochios["name"] = "Mochios";
  response["submodules"].push_back(mochios);

  res.status(STATUS_CODE::OK).json(response).end();
  return;
}

int main(int argc, char **argv) {
  brewtils::env::init("../.env");
  port = std::stoi(brewtils::env::get("PORT", "8000"));

  Server app = Server();

  // CORS middleware, applied across all routes
  std::unique_ptr<expresso::middleware::Cors> cors = std::make_unique<Cors>();
  cors->allowOrigin("*");
  cors->allowCredentials(true);
  app.use(std::move(cors));

  // Cookie Parser, applied across all routes
  std::unique_ptr<expresso::middleware::CookieParser> cookieParser =
      std::make_unique<CookieParser>();
  app.use(std::move(cookieParser));

  // Cache middleware, applied across all routes
  std::unique_ptr<expresso::middleware::Cacher> cacher =
      std::make_unique<Cacher>(3600, false);
  app.use(std::move(cacher));

  // Static serve middleware
  std::unique_ptr<expresso::middleware::StaticServe> staticServe =
      std::make_unique<StaticServe>("../assets");
  app.use(std::move(staticServe));

  // Route handling like normal
  app.get("/health", [](Request &req, Response &res) {
    res.status(STATUS_CODE::OK).send("Running healthy");
  });

  // Creating and using a router
  Router router;
  router.get("/", about);
  app.use("/about", &router);

  // Listing directories
  std::unique_ptr<expresso::middleware::StaticServe> pictureServe =
      std::make_unique<StaticServe>("../assets/github", true);
  Router pictureRouter;
  pictureRouter.use(std::move(pictureServe));
  app.use("/pictures", &pictureRouter);

  // Sending multiple files as single zip
  app.get("/download", [](Request &req, Response &res) {
    // You can put folders too, it will zip all the files in the folder
    std::set<std::string> files = {"../assets"};
    res.sendFiles(files, "assets.zip");
  });

  app.post("/print", [](Request &req, Response &res) {
    logger::info(req.body.dumps(2));
    json::object responseJson;
    responseJson["data"] = req.body;
    responseJson["timestamp"] = brewtils::date::getCurrentUTC();
    responseJson["printed"] = true;
    res.status(STATUS_CODE::OK).json(responseJson);
  });

  // Starting the server
  app.listen(port, []() {
    logger::success("Server is running on port " + std::to_string(port));
  });

  return 0;
}