#include "telnet.hpp"
#include <colors.h>
#include "../../types.hpp"
#include "../../routes.hpp"
#include <json.hpp>
#include <iostream>
#include <sstream>
#include <fstream>
#include <string>
#include <syslog.h>

#define TELNET_PORT_REQUIRED_STRING R"({"required":"[token,port]"})"

void get_telnet_port(const httplib::Request &request, httplib::Response &response) {
    std::cout << GREEN << request.path << RESET << "  " << request.method << std::endl;
    response.set_header("Access-Control-Allow-Origin", "*");
    if (request.body.empty()) {
        response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
        return;
    }

    nlohmann::json json_body = nlohmann::json::parse(request.body);
    if (json_body["token"] == nullptr) {
        response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
        return;
    }

    httplib::Client cli(AUTH_SERVICE);

    std::stringstream body;
    body <<  R"({"token":)" << json_body["token"] << R"(})";

    httplib::Result res = cli.Post("/api/access", body.str().c_str(), JSON_TYPE);
    if (res->status == httplib::OK_200) {
        nlohmann::json resjson = nlohmann::json::parse(res->body);
        if (resjson["access"] == "reject") {
            response.set_content(resjson.dump().c_str(), JSON_TYPE);
            return;
        }
        std::ifstream telnetportfile("/etc/openrouter/telnet/port");
        int telnet_port;
        telnetportfile >> telnet_port;
        telnetportfile.close();
        std::stringstream responsedata;
        responsedata << R"({"port":")" << telnet_port << R"("})";
        response.set_content(responsedata.str(), JSON_TYPE);
    }
}

void set_telnet_port(const httplib::Request &request, httplib::Response &response) {
    std::cout << GREEN << request.path << RESET << "  " << request.method << std::endl;
    response.set_header("Access-Control-Allow-Origin", "*");
    if (request.body.empty()) {
        response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
        return;
    }

    nlohmann::json json_body = nlohmann::json::parse(request.body);
    if (json_body["token"] == nullptr) {
        response.set_content(ACCESS_REQUIRED_STRING, JSON_TYPE);
        return;
    }

    if (json_body["port"] == nullptr) {
        response.set_content(TELNET_PORT_REQUIRED_STRING, JSON_TYPE);
        return;
    }

    httplib::Client cli(AUTH_SERVICE);

    std::stringstream body;
    body <<  R"({"token":)" << json_body["token"] << R"(})";

    httplib::Result res = cli.Post("/api/access", body.str().c_str(), JSON_TYPE);
    if (res->status == httplib::OK_200) {
        nlohmann::json resjson = nlohmann::json::parse(res->body);
        if (resjson["access"] == "reject") {
            response.set_content(resjson.dump().c_str(), JSON_TYPE);
            return;
        }
        std::ofstream telnetportfileout("/etc/openrouter/telnet/port");
        std::string new_telnet_port = json_body["port"];
        telnetportfileout << stoi(new_telnet_port);
        telnetportfileout.close();
        system("/etc/init.d/telnet restart");
        std::string syslogstring = "Telnet port changed to " + new_telnet_port;
        syslog(LOG_INFO, syslogstring.c_str());
        std::ifstream telnetportfile("/etc/openrouter/telnet/port");
        int telnet_port;
        telnetportfile >> telnet_port;
        telnetportfile.close();
        std::stringstream responsedata;
        responsedata << R"({"port":")" << telnet_port << R"("})";
        response.set_content(responsedata.str(), JSON_TYPE);
    }
}