#include "flip_social_messages.h"

FlipSocialModel2 *flip_social_messages_alloc()
{
    // Allocate memory for each username only if not already allocated
    FlipSocialModel2 *users = malloc(sizeof(FlipSocialModel2));
    if (users == NULL)
    {
        FURI_LOG_E(TAG, "Failed to allocate memory for message users");
        return NULL;
    }
    for (size_t i = 0; i < MAX_MESSAGE_USERS; i++)
    {
        if (users->usernames[i] == NULL)
        {
            users->usernames[i] = malloc(MAX_USER_LENGTH);
            if (users->usernames[i] == NULL)
            {
                FURI_LOG_E(TAG, "Failed to allocate memory for username %zu", i);
                return NULL; // Return false on memory allocation failure
            }
        }
    }
    return users;
}

FlipSocialMessage *flip_social_user_messages_alloc()
{
    // Allocate memory for each username only if not already allocated
    FlipSocialMessage *messages = malloc(sizeof(FlipSocialMessage));
    if (messages == NULL)
    {
        FURI_LOG_E(TAG, "Failed to allocate memory for messages");
        return NULL;
    }
    for (size_t i = 0; i < MAX_MESSAGE_USERS; i++)
    {
        if (messages->usernames[i] == NULL)
        {
            messages->usernames[i] = malloc(MAX_USER_LENGTH);
            if (messages->usernames[i] == NULL)
            {
                FURI_LOG_E(TAG, "Failed to allocate memory for username %zu", i);
                return NULL; // Return false on memory allocation failure
            }
        }
        if (messages->messages[i] == NULL)
        {
            messages->messages[i] = malloc(MAX_MESSAGE_LENGTH);
            if (messages->messages[i] == NULL)
            {
                FURI_LOG_E(TAG, "Failed to allocate memory for message %zu", i);
                return NULL; // Return false on memory allocation failure
            }
        }
    }
    return messages;
}

void flip_social_free_message_users()
{
    if (flip_social_message_users == NULL)
    {
        return;
    }
    for (int i = 0; i < flip_social_message_users->count; i++)
    {
        if (flip_social_message_users->usernames[i])
        {
            free(flip_social_message_users->usernames[i]);
        }
    }
    free(flip_social_message_users);
    flip_social_message_users = NULL;
}

void flip_social_free_messages()
{
    if (flip_social_messages == NULL)
    {
        return;
    }
    for (int i = 0; i < flip_social_messages->count; i++)
    {
        if (flip_social_messages->usernames[i])
        {
            free(flip_social_messages->usernames[i]);
        }
        if (flip_social_messages->messages[i])
        {
            free(flip_social_messages->messages[i]);
        }
    }
    free(flip_social_messages);
    flip_social_messages = NULL;
}

bool flip_social_update_messages_submenu()
{
    if (app_instance->submenu_messages == NULL)
    {
        FURI_LOG_E(TAG, "Submenu is NULL");
        return false;
    }
    if (flip_social_message_users == NULL)
    {
        FURI_LOG_E(TAG, "Message users model is NULL");
        return false;
    }
    submenu_reset(app_instance->submenu_messages);
    submenu_set_header(app_instance->submenu_messages, "Messages");
    submenu_add_item(app_instance->submenu_messages, "[New Message]", FlipSocialSubmenuLoggedInIndexMessagesNewMessage, flip_social_callback_submenu_choices, app_instance);
    for (int i = 0; i < flip_social_message_users->count; i++)
    {
        submenu_add_item(app_instance->submenu_messages, flip_social_message_users->usernames[i], FlipSocialSubmenuLoggedInIndexMessagesUsersStart + i, flip_social_callback_submenu_choices, app_instance);
    }
    return true;
}

bool flip_social_update_submenu_user_choices()
{
    if (app_instance->submenu_messages_user_choices == NULL)
    {
        FURI_LOG_E(TAG, "Submenu is NULL");
        return false;
    }
    if (flip_social_explore == NULL)
    {
        FURI_LOG_E(TAG, "Explore model is NULL");
        return false;
    }
    submenu_reset(app_instance->submenu_messages_user_choices);
    submenu_set_header(app_instance->submenu_messages_user_choices, "Users");
    for (int i = 0; i < flip_social_explore->count; i++)
    {
        submenu_add_item(app_instance->submenu_messages_user_choices, flip_social_explore->usernames[i], FlipSocialSubmenuLoggedInIndexMessagesUserChoicesIndexStart + i, flip_social_callback_submenu_choices, app_instance);
    }
    return true;
}

// Get all the users that have sent messages to the logged in user
bool flip_social_get_message_users()
{
    if (app_instance->login_username_logged_out == NULL)
    {
        FURI_LOG_E(TAG, "Username is NULL");
        return false;
    }
    if (fhttp.state == INACTIVE)
    {
        FURI_LOG_E(TAG, "HTTP state is INACTIVE");
        return false;
    }
    char command[128];
    snprintf(
        fhttp.file_path,
        sizeof(fhttp.file_path),
        STORAGE_EXT_PATH_PREFIX "/apps_data/flip_social/message_users.json");

    fhttp.save_received_data = true;
    auth_headers_alloc();
    snprintf(command, 128, "https://www.flipsocial.net/api/messages/%s/get/list/", app_instance->login_username_logged_out);
    if (!flipper_http_get_request_with_headers(command, auth_headers))
    {
        FURI_LOG_E(TAG, "Failed to send HTTP request for messages");
        fhttp.state = ISSUE;
        return false;
    }
    fhttp.state = RECEIVING;
    return true;
}

// Get all the messages between the logged in user and the selected user
bool flip_social_get_messages_with_user()
{
    if (fhttp.state == INACTIVE)
    {
        FURI_LOG_E(TAG, "HTTP state is INACTIVE");
        return false;
    }
    if (app_instance->login_username_logged_out == NULL)
    {
        FURI_LOG_E(TAG, "Username is NULL");
        return false;
    }
    if (!flip_social_message_users->usernames[flip_social_message_users->index] || strlen(flip_social_message_users->usernames[flip_social_message_users->index]) == 0)
    {
        FURI_LOG_E(TAG, "Username is NULL");
        return false;
    }
    char command[128];
    snprintf(
        fhttp.file_path,
        sizeof(fhttp.file_path),
        STORAGE_EXT_PATH_PREFIX "/apps_data/flip_social/%s_messages.json",
        flip_social_message_users->usernames[flip_social_message_users->index]);

    fhttp.save_received_data = true;
    auth_headers_alloc();
    snprintf(command, 128, "https://www.flipsocial.net/api/messages/%s/get/%s/", app_instance->login_username_logged_out, flip_social_message_users->usernames[flip_social_message_users->index]);
    if (!flipper_http_get_request_with_headers(command, auth_headers))
    {
        FURI_LOG_E(TAG, "Failed to send HTTP request for messages");
        fhttp.state = ISSUE;
        return false;
    }
    fhttp.state = RECEIVING;
    return true;
}

// Parse the users that have sent messages to the logged-in user
bool flip_social_parse_json_message_users()
{
    // load the received data from the saved file
    FuriString *message_data = flipper_http_load_from_file(fhttp.file_path);
    if (message_data == NULL)
    {
        FURI_LOG_E(TAG, "Failed to load received data from file.");
        return false;
    }
    char *data_cstr = (char *)furi_string_get_cstr(message_data);
    if (data_cstr == NULL)
    {
        FURI_LOG_E(TAG, "Failed to get C-string from FuriString.");
        furi_string_free(message_data);
        return false;
    }

    // Allocate memory for each username only if not already allocated
    flip_social_message_users = flip_social_messages_alloc();
    if (flip_social_message_users == NULL)
    {
        FURI_LOG_E(TAG, "Failed to allocate memory for message users.");
        furi_string_free(message_data);
        free(data_cstr);
        return false;
    }

    // Remove newlines
    char *pos = data_cstr;
    while ((pos = strchr(pos, '\n')) != NULL)
    {
        *pos = ' ';
    }

    // Initialize message users count
    flip_social_message_users->count = 0;

    // Extract the users array from the JSON
    char *json_users = get_json_value("users", data_cstr, MAX_TOKENS);
    if (json_users == NULL)
    {
        FURI_LOG_E(TAG, "Failed to parse users array.");
        furi_string_free(message_data);
        free(data_cstr);
        return false;
    }

    // Manual tokenization for comma-separated values
    char *start = json_users + 1; // Skip the opening bracket
    char *end;
    while ((end = strchr(start, ',')) != NULL && flip_social_message_users->count < MAX_MESSAGE_USERS)
    {
        *end = '\0'; // Null-terminate the current token

        // Remove quotes
        if (*start == '"')
            start++;
        if (*(end - 1) == '"')
            *(end - 1) = '\0';

        // Copy username to pre-allocated memory
        snprintf(flip_social_message_users->usernames[flip_social_message_users->count], MAX_USER_LENGTH, "%s", start);
        flip_social_message_users->count++;
        start = end + 1;
    }

    // Handle the last token
    if (*start != '\0' && flip_social_message_users->count < MAX_MESSAGE_USERS)
    {
        if (*start == '"')
            start++;
        if (*(start + strlen(start) - 1) == ']')
            *(start + strlen(start) - 1) = '\0';
        if (*(start + strlen(start) - 1) == '"')
            *(start + strlen(start) - 1) = '\0';

        snprintf(flip_social_message_users->usernames[flip_social_message_users->count], MAX_USER_LENGTH, "%s", start);
        flip_social_message_users->count++;
    }

    // Add submenu items for the users
    flip_social_update_messages_submenu();

    // Free the JSON data
    free(json_users);
    free(start);
    free(end);
    furi_string_free(message_data);
    free(data_cstr);
    return true;
}

// Parse the users that the logged in user can message
bool flip_social_parse_json_message_user_choices()
{
    // load the received data from the saved file
    FuriString *user_data = flipper_http_load_from_file(fhttp.file_path);
    if (user_data == NULL)
    {
        FURI_LOG_E(TAG, "Failed to load received data from file.");
        return false;
    }
    char *data_cstr = (char *)furi_string_get_cstr(user_data);
    if (data_cstr == NULL)
    {
        FURI_LOG_E(TAG, "Failed to get C-string from FuriString.");
        furi_string_free(user_data);
        return false;
    }

    // Allocate memory for each username only if not already allocated
    flip_social_explore = flip_social_explore_alloc();
    if (flip_social_explore == NULL)
    {
        FURI_LOG_E(TAG, "Failed to allocate memory for explore usernames.");
        furi_string_free(user_data);
        free(data_cstr);
        return false;
    }

    // Remove newlines
    char *pos = data_cstr;
    while ((pos = strchr(pos, '\n')) != NULL)
    {
        *pos = ' ';
    }

    // Initialize explore count
    flip_social_explore->count = 0;

    // Extract the users array from the JSON
    char *json_users = get_json_value("users", data_cstr, MAX_TOKENS);
    if (json_users == NULL)
    {
        FURI_LOG_E(TAG, "Failed to parse users array.");
        furi_string_free(user_data);
        free(data_cstr);
        return false;
    }

    // Manual tokenization for comma-separated values
    char *start = json_users + 1; // Skip the opening bracket
    char *end;
    while ((end = strchr(start, ',')) != NULL && flip_social_explore->count < MAX_EXPLORE_USERS)
    {
        *end = '\0'; // Null-terminate the current token

        // Remove quotes
        if (*start == '"')
            start++;
        if (*(end - 1) == '"')
            *(end - 1) = '\0';

        // Copy username to pre-allocated memory
        snprintf(flip_social_explore->usernames[flip_social_explore->count], MAX_USER_LENGTH, "%s", start);
        flip_social_explore->count++;
        start = end + 1;
    }

    // Handle the last token
    if (*start != '\0' && flip_social_explore->count < MAX_EXPLORE_USERS)
    {
        if (*start == '"')
            start++;
        if (*(start + strlen(start) - 1) == ']')
            *(start + strlen(start) - 1) = '\0';
        if (*(start + strlen(start) - 1) == '"')
            *(start + strlen(start) - 1) = '\0';

        snprintf(flip_social_explore->usernames[flip_social_explore->count], MAX_USER_LENGTH, "%s", start);
        flip_social_explore->count++;
    }

    // Add submenu items for the users
    flip_social_update_submenu_user_choices();

    // Free the JSON data
    free(json_users);
    free(start);
    free(end);
    furi_string_free(user_data);
    free(data_cstr);
    return true;
}

// parse messages between the logged in user and the selected user
bool flip_social_parse_json_messages()
{
    // load the received data from the saved file
    FuriString *message_data = flipper_http_load_from_file(fhttp.file_path);
    if (message_data == NULL)
    {
        FURI_LOG_E(TAG, "Failed to load received data from file.");
        return false;
    }
    char *data_cstr = (char *)furi_string_get_cstr(message_data);
    if (data_cstr == NULL)
    {
        FURI_LOG_E(TAG, "Failed to get C-string from FuriString.");
        furi_string_free(message_data);
        return false;
    }

    // Allocate memory for each message only if not already allocated
    flip_social_messages = flip_social_user_messages_alloc();
    if (!flip_social_messages)
    {
        FURI_LOG_E(TAG, "Failed to allocate memory for messages.");
        furi_string_free(message_data);
        free(data_cstr);
        return false;
    }

    // Remove newlines
    char *pos = data_cstr;
    while ((pos = strchr(pos, '\n')) != NULL)
    {
        *pos = ' ';
    }

    // Initialize messages count
    flip_social_messages->count = 0;

    // Iterate through the messages array
    for (int i = 0; i < MAX_MESSAGES; i++)
    {
        // Parse each item in the array
        char *item = get_json_array_value("conversations", i, data_cstr, MAX_TOKENS);
        if (item == NULL)
        {
            break;
        }

        // Extract individual fields from the JSON object
        char *sender = get_json_value("sender", item, 64);
        char *content = get_json_value("content", item, 64);

        if (sender == NULL || content == NULL)
        {
            FURI_LOG_E(TAG, "Failed to parse item fields.");
            free(item);
            continue;
        }

        // Store parsed values in pre-allocated memory
        snprintf(flip_social_messages->usernames[i], MAX_USER_LENGTH, "%s", sender);
        snprintf(flip_social_messages->messages[i], MAX_MESSAGE_LENGTH, "%s", content);
        flip_social_messages->count++;

        free(item);
        free(sender);
        free(content);
    }
    furi_string_free(message_data);
    free(data_cstr);
    return true;
}