#include <update/update.h>
#include <flip_storage/flip_wifi_storage.h>
#include <storage/storage.h>

static bool update_is_str(const char *src, const char *dst) { return strcmp(src, dst) == 0; }
static bool update_json_to_datetime(DateTime *rtc_time, FuriString *str)
{
    if (!rtc_time || !str)
    {
        FURI_LOG_E(TAG, "rtc_time or str is NULL");
        return false;
    }
    FuriString *hour = get_json_value_furi("hour", str);
    if (hour)
    {
        rtc_time->hour = atoi(furi_string_get_cstr(hour));
        furi_string_free(hour);
    }
    FuriString *minute = get_json_value_furi("minute", str);
    if (minute)
    {
        rtc_time->minute = atoi(furi_string_get_cstr(minute));
        furi_string_free(minute);
    }
    FuriString *second = get_json_value_furi("second", str);
    if (second)
    {
        rtc_time->second = atoi(furi_string_get_cstr(second));
        furi_string_free(second);
    }
    FuriString *day = get_json_value_furi("day", str);
    if (day)
    {
        rtc_time->day = atoi(furi_string_get_cstr(day));
        furi_string_free(day);
    }
    FuriString *month = get_json_value_furi("month", str);
    if (month)
    {
        rtc_time->month = atoi(furi_string_get_cstr(month));
        furi_string_free(month);
    }
    FuriString *year = get_json_value_furi("year", str);
    if (year)
    {
        rtc_time->year = atoi(furi_string_get_cstr(year));
        furi_string_free(year);
    }
    FuriString *weekday = get_json_value_furi("weekday", str);
    if (weekday)
    {
        rtc_time->weekday = atoi(furi_string_get_cstr(weekday));
        furi_string_free(weekday);
    }
    return datetime_validate_datetime(rtc_time);
}

static FuriString *update_datetime_to_json(DateTime *rtc_time)
{
    if (!rtc_time)
    {
        FURI_LOG_E(TAG, "rtc_time is NULL");
        return NULL;
    }
    char json[256];
    snprintf(
        json,
        sizeof(json),
        "{\"hour\":%d,\"minute\":%d,\"second\":%d,\"day\":%d,\"month\":%d,\"year\":%d,\"weekday\":%d}",
        rtc_time->hour,
        rtc_time->minute,
        rtc_time->second,
        rtc_time->day,
        rtc_time->month,
        rtc_time->year,
        rtc_time->weekday);
    return furi_string_alloc_set_str(json);
}

static bool update_save_rtc_time(DateTime *rtc_time)
{
    if (!rtc_time)
    {
        FURI_LOG_E(TAG, "rtc_time is NULL");
        return false;
    }
    FuriString *json = update_datetime_to_json(rtc_time);
    if (!json)
    {
        FURI_LOG_E(TAG, "Failed to convert DateTime to JSON");
        return false;
    }
    save_char("last_checked", furi_string_get_cstr(json));
    furi_string_free(json);
    return true;
}

//
// Returns true if time_current is one hour (or more) later than the stored last_updated time
//
static bool update_is_update_time(DateTime *time_current)
{
    if (!time_current)
    {
        FURI_LOG_E(TAG, "time_current is NULL");
        return false;
    }
    char last_updated_old[128];
    if (!load_char("last_updated", last_updated_old, sizeof(last_updated_old)))
    {
        FURI_LOG_E(TAG, "Failed to load last_updated");
        FuriString *json = update_datetime_to_json(time_current);
        if (json)
        {
            save_char("last_updated", furi_string_get_cstr(json));
            furi_string_free(json);
        }
        return false;
    }

    DateTime last_updated_time;

    FuriString *last_updated_furi = char_to_furi_string(last_updated_old);
    if (!last_updated_furi)
    {
        FURI_LOG_E(TAG, "Failed to convert char to FuriString");
        return false;
    }
    if (!update_json_to_datetime(&last_updated_time, last_updated_furi))
    {
        FURI_LOG_E(TAG, "Failed to convert JSON to DateTime");
        furi_string_free(last_updated_furi);
        return false;
    }
    furi_string_free(last_updated_furi); // Free after usage.

    bool time_diff = false;
    // If the date is different assume more than one hour has passed.
    if (time_current->year != last_updated_time.year ||
        time_current->month != last_updated_time.month ||
        time_current->day != last_updated_time.day)
    {
        time_diff = true;
    }
    else
    {
        // For the same day, compute seconds from midnight.
        int seconds_current = time_current->hour * 3600 + time_current->minute * 60 + time_current->second;
        int seconds_last = last_updated_time.hour * 3600 + last_updated_time.minute * 60 + last_updated_time.second;
        if ((seconds_current - seconds_last) >= 3600)
        {
            time_diff = true;
        }
    }

    return time_diff;
}

// Sends a request to fetch the last updated date of the app.
static bool update_last_app_update(FlipperHTTP *fhttp, bool flipper_server)
{
    if (!fhttp)
    {
        FURI_LOG_E(TAG, "fhttp is NULL");
        return false;
    }
    char url[256];
    if (flipper_server)
    {
        // make sure folder is created
        char directory_path[256];
        snprintf(directory_path, sizeof(directory_path), STORAGE_EXT_PATH_PREFIX "/apps_data/flip_store");

        // Create the directory
        Storage *storage = furi_record_open(RECORD_STORAGE);
        storage_common_mkdir(storage, directory_path);
        snprintf(directory_path, sizeof(directory_path), STORAGE_EXT_PATH_PREFIX "/apps_data/flip_store/data");
        storage_common_mkdir(storage, directory_path);
        snprintf(directory_path, sizeof(directory_path), STORAGE_EXT_PATH_PREFIX "/apps_data/flip_store/data/last_update_request.txt");
        storage_simply_remove_recursive(storage, directory_path); // ensure the file is empty
        furi_record_close(RECORD_STORAGE);

        fhttp->save_received_data = true;
        fhttp->is_bytes_request = false;

        snprintf(fhttp->file_path, sizeof(fhttp->file_path), STORAGE_EXT_PATH_PREFIX "/apps_data/flip_store/data/last_update_request.txt");
        snprintf(url, sizeof(url), "https://catalog.flipperzero.one/api/v0/0/application/%s?is_latest_release_version=true", BUILD_ID);
        return flipper_http_request(fhttp, GET, url, "{\"Content-Type\":\"application/json\"}", NULL);
    }
    else
    {
        snprintf(url, sizeof(url), "https://www.jblanked.com/flipper/api/app/last-updated/flip_downloader/");
        return flipper_http_request(fhttp, GET, url, "{\"Content-Type\":\"application/json\"}", NULL);
    }
}

// Parses the server response and returns true if an update is available.
static bool update_parse_last_app_update(FlipperHTTP *fhttp, DateTime *time_current, bool flipper_server)
{
    if (!fhttp)
    {
        FURI_LOG_E(TAG, "fhttp is NULL");
        return false;
    }
    if (fhttp->state == ISSUE)
    {
        FURI_LOG_E(TAG, "Failed to fetch last app update");
        return false;
    }
    char version_str[32];
    if (!flipper_server)
    {
        if (fhttp->last_response == NULL || strlen(fhttp->last_response) == 0)
        {
            FURI_LOG_E(TAG, "fhttp->last_response is NULL or empty");
            return false;
        }

        char *app_version = get_json_value("version", fhttp->last_response);
        if (app_version)
        {
            // Save the server app version: it should save something like: 0.8
            save_char("server_app_version", app_version);
            snprintf(version_str, sizeof(version_str), "%s", app_version);
            free(app_version);
        }
        else
        {
            FURI_LOG_E(TAG, "Failed to get app version");
            return false;
        }
    }
    else
    {
        FuriString *app_data = flipper_http_load_from_file_with_limit(fhttp->file_path, memmgr_heap_get_max_free_block());
        if (!app_data)
        {
            FURI_LOG_E(TAG, "Failed to load app data");
            return false;
        }
        FuriString *current_version = get_json_value_furi("current_version", app_data);
        if (!current_version)
        {
            FURI_LOG_E(TAG, "Failed to get current version");
            furi_string_free(app_data);
            return false;
        }
        furi_string_free(app_data);
        FuriString *version = get_json_value_furi("version", current_version);
        if (!version)
        {
            FURI_LOG_E(TAG, "Failed to get version");
            furi_string_free(current_version);
            furi_string_free(app_data);
            return false;
        }
        // Save the server app version: it should save something like: 0.8
        save_char("server_app_version", furi_string_get_cstr(version));
        snprintf(version_str, sizeof(version_str), "%s", furi_string_get_cstr(version));
        furi_string_free(current_version);
        furi_string_free(version);
        // furi_string_free(app_data);
    }
    // Only check for an update if an hour or more has passed.
    if (update_is_update_time(time_current))
    {
        char app_version[32];
        if (!load_char("app_version", app_version, sizeof(app_version)))
        {
            FURI_LOG_E(TAG, "Failed to load app version");
            return false;
        }
        FURI_LOG_I(TAG, "App version: %s", app_version);
        FURI_LOG_I(TAG, "Server version: %s", version_str);
        // Check if the app version is different from the server version.
        if (!update_is_str(app_version, version_str))
        {
            easy_flipper_dialog("Update available", "New update available!\nPress BACK to download.");
            return true; // Update available.
        }
        FURI_LOG_I(TAG, "No update available");
        return false; // No update available.
    }
    FURI_LOG_I(TAG, "Not enough time has passed since the last update check");
    return false; // Not yet time to update.
}

static bool update_get_fap_file(FlipperHTTP *fhttp, bool flipper_server)
{
    if (!fhttp)
    {
        FURI_LOG_E(TAG, "FlipperHTTP is NULL.");
        return false;
    }
    char url[256];
    fhttp->save_received_data = false;
    fhttp->is_bytes_request = true;
#ifndef FW_ORIGIN_Momentum
    snprintf(
        fhttp->file_path,
        sizeof(fhttp->file_path),
        STORAGE_EXT_PATH_PREFIX "/apps/GPIO/flip_downloader.fap");
#else
    snprintf(
        fhttp->file_path,
        sizeof(fhttp->file_path),
        STORAGE_EXT_PATH_PREFIX "/apps/GPIO/FlipperHTTP/flip_downloader.fap");
#endif
    if (flipper_server)
    {
        char build_id[32];
        snprintf(build_id, sizeof(build_id), "%s", BUILD_ID);
        uint8_t target;
        target = furi_hal_version_get_hw_target();
        uint16_t api_major, api_minor;
        furi_hal_info_get_api_version(&api_major, &api_minor);
        snprintf(
            url,
            sizeof(url),
            "https://catalog.flipperzero.one/api/v0/application/version/%s/build/compatible?target=f%d&api=%d.%d",
            build_id,
            target,
            api_major,
            api_minor);
    }
    else
    {
        snprintf(url, sizeof(url), "https://www.jblanked.com/flipper/api/app/download/flip_downloader/");
    }
    return flipper_http_request(fhttp, BYTES, url, "{\"Content-Type\": \"application/octet-stream\"}", NULL);
}

// Updates the app. Uses the supplied current time for validating if update check should proceed.
static bool update_update_app(FlipperHTTP *fhttp, DateTime *time_current, bool use_flipper_api)
{
    if (!fhttp)
    {
        FURI_LOG_E(TAG, "fhttp is NULL");
        return false;
    }
    if (!update_last_app_update(fhttp, use_flipper_api))
    {
        FURI_LOG_E(TAG, "Failed to fetch last app update");
        return false;
    }
    fhttp->state = RECEIVING;
    furi_timer_start(fhttp->get_timeout_timer, TIMEOUT_DURATION_TICKS);
    while (fhttp->state == RECEIVING && furi_timer_is_running(fhttp->get_timeout_timer) > 0)
    {
        furi_delay_ms(100);
    }
    furi_timer_stop(fhttp->get_timeout_timer);
    if (update_parse_last_app_update(fhttp, time_current, use_flipper_api))
    {
        if (!update_get_fap_file(fhttp, false))
        {
            FURI_LOG_E(TAG, "Failed to fetch fap file 1");
            return false;
        }
        fhttp->state = RECEIVING;

        while (fhttp->state == RECEIVING)
        {
            furi_delay_ms(100);
        }

        if (fhttp->state == ISSUE)
        {
            FURI_LOG_E(TAG, "Failed to fetch fap file 2");
            easy_flipper_dialog("Update Error", "Failed to download the\nupdate file.\nPlease try again.");
            return false;
        }
        return true;
    }

    FURI_LOG_I(TAG, "No update available");
    return false; // No update available.
}

// Handles the app update routine. This function obtains the current RTC time,
// checks the "last_checked" value, and if it is more than one hour old, calls for an update.
bool update_is_ready(FlipperHTTP *fhttp, bool use_flipper_api)
{
    if (!fhttp)
    {
        FURI_LOG_E(TAG, "fhttp is NULL");
        return false;
    }
    DateTime rtc_time;
    furi_hal_rtc_get_datetime(&rtc_time);
    char last_checked[32];
    if (!load_char("last_checked", last_checked, sizeof(last_checked)))
    {
        // First time – save the current time and check for an update.
        if (!update_save_rtc_time(&rtc_time))
        {
            FURI_LOG_E(TAG, "Failed to save RTC time");
            return false;
        }
        return update_update_app(fhttp, &rtc_time, use_flipper_api);
    }
    else
    {
        // Check if the current RTC time is at least one hour past the stored time.
        if (update_is_update_time(&rtc_time))
        {
            if (!update_update_app(fhttp, &rtc_time, use_flipper_api))
            {
                FURI_LOG_E(TAG, "Failed to update app");
                // save the current time for the next check.
                if (!update_save_rtc_time(&rtc_time))
                {
                    FURI_LOG_E(TAG, "Failed to save RTC time");
                    return false;
                }
                return false;
            }
            // Save the current time for the next check.
            if (!update_save_rtc_time(&rtc_time))
            {
                FURI_LOG_E(TAG, "Failed to save RTC time");
                return false;
            }
            return true;
        }
        return false; // No update necessary.
    }
}