#include "Firewall.h"

ESPFirewall::ESPFirewall(int port)
{
    this->amount_of_rules = 0;
    this->head = NULL;
    log_i("Starting Firewall-API on %i", port);
    this->firewall_api = new WebServer(port);
    this->setup_routing();
}

void ESPFirewall::setup_routing()
{
    this->firewall_api->on("/api/v1/firewall", HTTP_GET, std::bind(&ESPFirewall::get_firewall_rules, this));
    this->firewall_api->begin();
}

void ESPFirewall::custom_message_response(const char *message, int response_code)
{
    cJSON *json_response = cJSON_CreateObject();
    cJSON_AddBoolToObject(json_response, "ok", true);
    cJSON_AddStringToObject(json_response, "message", message);
    this->firewall_api->send(response_code, "application/json", cJSON_Print(json_response));
    cJSON_Delete(json_response);
}

void ESPFirewall::prepare_firewall_json(cJSON *jsonResponse, firewall_rule_t *link)
{
    cJSON_AddStringToObject(jsonResponse, "status", "success");
    cJSON_AddNumberToObject(jsonResponse, "rule", link->key);
    cJSON_AddStringToObject(jsonResponse, "source", link->source);
    cJSON_AddStringToObject(jsonResponse, "destination", link->destination);
    cJSON_AddStringToObject(jsonResponse, "protocol", link->protocol);
    cJSON_AddStringToObject(jsonResponse, "target", link->target);
}

firewall_rule_t *ESPFirewall::add_rule_to_firewall(char *source, char *destination, char *protocol, char *target)
{
    firewall_rule_t *temp;
    firewall_rule_t *link = (firewall_rule_t *)malloc(sizeof(firewall_rule_t));
    link->key = ++amount_of_rules;
    strcpy(link->source, source);
    strcpy(link->destination, destination);
    strcpy(link->protocol, protocol);
    strcpy(link->target, target);
    if (head == NULL)
    {
        head = link;
        link->next = NULL;
        return link;
    }
    temp = head;
    while (temp->next != NULL)
    {
        temp = temp->next;
    }
    temp->next = link;
    link->next = NULL;
    return link;
}

void ESPFirewall::get_firewall_rules()
{
    firewall_rule_t *ptr = head;
    cJSON *json_response = cJSON_CreateArray();
    while (ptr != NULL)
    {
        cJSON *json_firewall_rule = cJSON_CreateObject();
        prepare_firewall_json(json_firewall_rule, ptr);
        cJSON_AddItemToArray(json_response, json_firewall_rule);
        ptr = ptr->next;
    }
    this->firewall_api->send(200, "application/json", cJSON_Print(json_response));
}

void ESPFirewall::handle_clients()
{
    this->firewall_api->handleClient();
}