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

void
api_v1_schedules_STR_GET(struct mg_connection *nc, struct http_message *hm, endpoint_args_t *args, endpoint_response_t *response)
{
    (void)hm;
    (void)nc;

    uuid_t target_uid;
    if(schedule_uid_parse(args[0].value.v_str, target_uid))
    {
        M_RESPONSE_400_NO_VALID_ID(response);
        return;
    }

    schedule_t* schedule = schedule_get_by_uid(target_uid);

    if(!schedule)
    {
        M_RESPONSE_404_NO_SCHEDULE_FOUND_FOR_ID(response);
        return;
    }

    cJSON *json = schedule_to_json(schedule);

    endpoint_response_json(response, 200, json);
    cJSON_Delete(json);
    schedule_free(schedule);
}

void
api_v1_schedules_STR_PUT(struct mg_connection *nc, struct http_message *hm, endpoint_args_t *args, endpoint_response_t *response)
{
    (void)hm;
    (void)nc;

    uuid_t target_uid;
    if(schedule_uid_parse(args[0].value.v_str, target_uid))
    {
        M_RESPONSE_400_NO_VALID_ID(response);
        return;
    }

    schedule_t* schedule = schedule_get_by_uid(target_uid);

    if(!schedule)
    {
        M_RESPONSE_404_NO_SCHEDULE_FOUND_FOR_ID(response);
        return;
    }

    cJSON *json = cJSON_ParseWithLength(hm->body.p, hm->body.len);

    if(json == NULL)
    {
        M_RESPONSE_400_NO_VALID_JSON(response);
        return;
    }

    cJSON *json_name = cJSON_GetObjectItemCaseSensitive(json, "name");
    if(cJSON_IsString(json_name) && json_name->valuestring)
    {
        strncpy(schedule->name, json_name->valuestring, MAX_NAME_LENGTH);
        schedule->name[MAX_NAME_LENGTH] = '\0';
    }

    if(!schedule_is_protected(schedule))
    {
        cJSON *json_period;
        cJSON *json_periods = cJSON_GetObjectItemCaseSensitive(json, "periods");

        int periods_count = cJSON_GetArraySize(json_periods);
        free(schedule->periods);
        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))
            {
                LOGGER_DEBUG("period is missing start\n");
                continue;
            }
            if(!cJSON_IsString(json_period_end) || (json_period_end->valuestring == NULL))
            {
                LOGGER_DEBUG("period is missing end\n");
                continue;
            }

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

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

        schedule->periods_count = periods_valid;
    }

    if(schedule_save(schedule))
    {
        free(schedule);
        cJSON_Delete(json);

        M_RESPONSE_500_FAILED_TO_SAVE_TO_DATABASE(response);
        return;
    }

    command_schedule_update(schedule);

    cJSON *json_tag;
    cJSON *json_tags = cJSON_GetObjectItemCaseSensitive(json, "tags");
    if(cJSON_IsArray(json_tags))
    {
        junction_tag_remove_for_schedule(schedule->id);
    }
    cJSON_ArrayForEach(json_tag, json_tags)
    {
        if(!cJSON_IsString(json_tag) || (json_tag->valuestring == NULL))
        {
            LOGGER_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, schedule->id);
    }

    cJSON_Delete(json);
    json = schedule_to_json(schedule);

    endpoint_response_json(response, 200, json);
    cJSON_Delete(json);
    schedule_free(schedule);
}

void
api_v1_schedules_STR_DELETE(struct mg_connection *nc, struct http_message *hm, endpoint_args_t *args, endpoint_response_t *response)
{
    (void)hm;
    (void)nc;

    const char *target_uid_str = args[0].value.v_str;

    uuid_t target_uid;
    if(schedule_uid_parse(target_uid_str, target_uid))
    {
        M_RESPONSE_400_NO_VALID_ID(response);
        return;
    }

    schedule_t* schedule = schedule_get_by_uid(target_uid);

    if(!schedule)
    {
        M_RESPONSE_404_NO_SCHEDULE_FOUND_FOR_ID(response);
        return;
    }

    if(schedule_is_protected(schedule))
    {
        schedule_free(schedule);

        M_RESPONSE_403_PROTECTED_SCHEDULE(response);
        return;
    }

    if(schedule_remove(schedule))
    {
        M_RESPONSE_500_FAILED_TO_DELETE_FROM_DATABASE(response);
    }
    else
    {
        M_RESPONSE_MSG(LOGGER_DEBUG, response, 200, "the target schedule got deleted");
    }
    schedule_free(schedule);
}