#include <cJSON.h>
#include <constants.h>
#include <endpoints/api_v1_schedules.h>
#include <logger.h>
#include <models/junction_tag.h>
#include <models/schedule.h>
#include <models/tag.h>

void
api_v1_schedules_POST(struct mg_connection *c, endpoint_args_t *args, struct http_message *hm)
{
    (void)args;
    cJSON *json = cJSON_ParseWithLength(hm->body.p, hm->body.len);

    if(json == NULL)
    {
        const char *error_ptr = cJSON_GetErrorPtr();
        if (error_ptr != NULL)
        {
            LOG_ERROR("error before: %s\n", error_ptr);
        }
        cJSON_Delete(json);
    }

    cJSON *json_name = cJSON_GetObjectItemCaseSensitive(json, "name");
    if(!cJSON_IsString(json_name) || (json_name->valuestring == NULL))
    {
        LOG_DEBUG("no name for schedule provided\n");
        mg_send_head(c, 400, 2, "Content-Type: application/json\r\n" STANDARD_HEADERS);
        mg_printf(c, "{}");

        cJSON_Delete(json);
        return;
    }
    cJSON *json_period;
    cJSON *json_periods = cJSON_GetObjectItemCaseSensitive(json, "periods");

    schedule_t *new_schedule = malloc(sizeof(schedule_t));

    new_schedule->id = 0;
    uuid_generate(new_schedule->uid);

    strncpy(new_schedule->name, json_name->valuestring, MAX_NAME_LENGTH);
    new_schedule->name[MAX_NAME_LENGTH] = '\0';

    int periods_count = cJSON_GetArraySize(json_periods);
    new_schedule->periods = malloc(sizeof(period_t) * periods_count);

    int periods_valid = 0;

    cJSON_ArrayForEach(json_period, json_periods)
    {
        cJSON *json_period_start = cJSON_GetObjectItemCaseSensitive(json_period, "start");
        cJSON *json_period_end = cJSON_GetObjectItemCaseSensitive(json_period, "end");

        if(!cJSON_IsString(json_period_start) || (json_period_start->valuestring == NULL))
        {
            LOG_DEBUG("period is missing start\n");
            continue;
        }
        if(!cJSON_IsString(json_period_end) || (json_period_end->valuestring == NULL))
        {
            LOG_DEBUG("period is missing end\n");
            continue;
        }

        uint16_t start;
        uint16_t end;
        if(period_helper_parse_hhmm(json_period_start->valuestring, &start))
        {
            LOG_DEBUG("couldn't parse start '%s'\n", json_period_start->valuestring);
            continue;
        }
        if(period_helper_parse_hhmm(json_period_end->valuestring, &end))
        {
            LOG_DEBUG("couldn't parse end '%s'\n", json_period_end->valuestring);
            continue;
        }

        new_schedule->periods[periods_valid].start = start;
        new_schedule->periods[periods_valid].end = end;
        ++periods_valid;
    }

    new_schedule->periods_count = periods_valid;

    schedule_save(new_schedule);

    junction_tag_remove_for_schedule(new_schedule->id);
    cJSON *json_tag;
    cJSON *json_tags = cJSON_GetObjectItemCaseSensitive(json, "tags");
    cJSON_ArrayForEach(json_tag, json_tags)
    {
            if(!cJSON_IsString(json_tag) || (json_tag->valuestring == NULL))
            {
                LOG_DEBUG("invalid tag in tags\n");
                continue;
            }
            const char *tag = json_tag->valuestring;
            int tag_id = tag_get_id(tag);
            if(tag_id == 0)
            {
                tag_save(tag_id, tag);
                tag_id = tag_get_id(tag);
            }
            junction_tag_insert(tag_id, 0, new_schedule->id);
    }

    cJSON_Delete(json);
    json = schedule_to_json(new_schedule);

    char *json_str = cJSON_Print(json);
    if (json_str == NULL)
    {
        LOG_ERROR("failed to print schedule json\n");
        mg_send_head(c, 201, 2, "Content-Type: application/json\r\n" STANDARD_HEADERS);
        mg_printf(c, "{}");
    }
    else
    {
        mg_send_head(c, 201, strlen(json_str), "Content-Type: application/json\r\n" STANDARD_HEADERS);
        mg_printf(c, "%s", json_str);
        free(json_str);
    }
    cJSON_Delete(json);
    schedule_free(new_schedule);
}

void
api_v1_schedules_GET(struct mg_connection *c, endpoint_args_t *args, struct http_message *hm)
{
    (void)args;
    (void)hm;
    schedule_t** all_schedules = schedule_get_all();

    cJSON *json = cJSON_CreateArray();

    for(int i = 0; all_schedules[i] != NULL; ++i)
    {
        cJSON *json_schedule = schedule_to_json(all_schedules[i]);

        cJSON_AddItemToArray(json, json_schedule);
    }

    char *json_str = cJSON_Print(json);
    if (json_str == NULL)
    {
        LOG_ERROR("failed to print schedules json\n");
        mg_send_head(c, 500, 2, "Content-Type: application/json\r\n" STANDARD_HEADERS);
        mg_printf(c, "[]");
    }
    else
    {
        mg_send_head(c, 200, strlen(json_str), "Content-Type: application/json\r\n" STANDARD_HEADERS);
        mg_printf(c, "%s", json_str);
        free(json_str);
    }
    cJSON_Delete(json);
    schedule_free_list(all_schedules);
}