#include <flip_wifi.h>
#include <alloc/flip_wifi_alloc.h>
#include <update/update.h>
#include <flip_storage/flip_wifi_storage.h>

// Entry point for the FlipWiFi application
int32_t flip_wifi_main(void *p)
{
    UNUSED(p);

    // Initialize the FlipWiFi application
    FlipWiFiApp *app = flip_wifi_app_alloc();
    if (!app)
    {
        FURI_LOG_E(TAG, "Failed to allocate FlipWiFiApp");
        return -1;
    }

    save_char("app_version", VERSION);

    // check if board is connected (Derek Jamison)
    FlipperHTTP *fhttp = flipper_http_alloc();
    if (!fhttp)
    {
        easy_flipper_dialog("FlipperHTTP Error", "The UART is likely busy.\nEnsure you have the correct\nflash for your board then\nrestart your Flipper Zero.");
        return -1;
    }

    if (!flipper_http_send_command(fhttp, HTTP_CMD_PING))
    {
        FURI_LOG_E(TAG, "Failed to ping the device");
        flipper_http_free(fhttp);
        return -1;
    }

    furi_delay_ms(100);

    // Try to wait for pong response.
    uint32_t counter = 10;
    while (fhttp->state == INACTIVE && --counter > 0)
    {
        FURI_LOG_D(TAG, "Waiting for PONG");
        furi_delay_ms(100);
    }

    // last response should be PONG
    if (!fhttp->last_response || strcmp(fhttp->last_response, "[PONG]") != 0)
    {
        easy_flipper_dialog("FlipperHTTP Error", "Ensure your WiFi Developer\nBoard or Pico W is connected\nand the latest FlipperHTTP\nfirmware is installed.");
        FURI_LOG_E(TAG, "Failed to receive PONG");
    }
    else
    {
        // for now use the catalog API until I implement caching on the server
        if (update_is_ready(fhttp, true))
        {
            easy_flipper_dialog("Update Status", "Complete.\nRestart your Flipper Zero.");
        }
    }

    flipper_http_free(fhttp);

    // Run the view dispatcher
    view_dispatcher_run(app->view_dispatcher);

    // Free the resources used by the FlipWiFi application
    flip_wifi_app_free(app);

    // Return 0 to indicate success
    return 0;
}