#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "custom_menu.h"
#include "custom_menu_system.h"
#include "pc/network/network.h"

#include "pc/configfile.h"
#include "pc/controller/controller_keyboard.h"
#include "game/object_list_processor.h"
#include "game/object_helpers.h"
#include "game/ingame_menu.h"
#include "game/game_init.h"
#include "game/segment2.h"
#include "object_fields.h"
#include "model_ids.h"
#include "behavior_data.h"
#include "audio_defines.h"
#include "audio/external.h"
#include "config.h"
#include "pc/network/version.h"
#include "pc/network/discord/discord.h"

#define MAIN_MENU_HEADER_TEXT "SM64 COOP"

char sConnectionJoinError[128] = { 0 };
char gConnectionText[128] = { 0 };
struct CustomMenu* sConnectMenu = NULL;

u8 gOpenConnectMenu = FALSE;
s8 sGotoGame = 0;

static void menu_main_draw_strings(void) {
    char* subtitle = "Still in development.";
    s16 subtitleX = (SCREEN_WIDTH - get_generic_ascii_string_width(subtitle)) / 2;
    print_generic_ascii_string(subtitleX, 150, subtitle);

    char* versionString = get_version();
    gDPSetEnvColor(gDisplayListHead++, 255, 255, 255, 120);
    print_generic_ascii_string(25, 25, versionString);

}

static void host_menu_draw_strings(void) {
    #ifdef DISCORD_SDK
        #define HOST_MENU_MAX_ITEMS 6
    #else
        #define HOST_MENU_MAX_ITEMS 5
    #endif

    // set up server setting strings
    char* buttonText[HOST_MENU_MAX_ITEMS];

    switch (configPlayerInteraction) {
        case 0:  buttonText[0] = "Non-solid players."; break;
        case 1:  buttonText[0] = "Solid players.";     break;
        case 2:  buttonText[0] = "Friendly fire.";     break;
        default: buttonText[0] = "UNKNOWN";            break;
    }
    if (configPlayerKnockbackStrength <= 20) {
        buttonText[1] = "Weak knockback.";
    } else if (configPlayerKnockbackStrength <= 40) {
        buttonText[1] = "Normal knockback.";
    } else {
        buttonText[1] = "Too much knockback.";
    }

    buttonText[2] = configStayInLevelAfterStar ? "Stay in level after star." : "Leave level after star.";

    buttonText[3] = configSkipIntro ? "Skip intro cutscene." : "Play intro cutscene.";

    buttonText[4] = configShareLives ? "Share lives." : "Lives are not shared.";

    #ifdef DISCORD_SDK
        buttonText[5] = (configNetworkSystem == 0) ? "Host through Discord." : "Host direct connection.";
    #endif

    // display server setting strings
    for (int i = 0; i < HOST_MENU_MAX_ITEMS; i++) {
        print_generic_ascii_string(95, 173 + -29 * i, buttonText[i]);
    }

    // display direct connection warning
    if (configNetworkSystem != 0) {
        f32 red = (f32)fabs(sin(gGlobalTimer / 20.0f));
        gDPSetEnvColor(gDisplayListHead++, 222, 222 * red, 222 * red, gMenuStringAlpha);
        char warning[128];
        snprintf(warning, 127, "Port forward '%d' in network router settings or use Hamachi.", configHostPort);
        print_generic_ascii_string(0, 5, warning);
    } else if ((configNetworkSystem == 0) && gDiscordFailed) {
        f32 red = (f32)fabs(sin(gGlobalTimer / 20.0f));
        gDPSetEnvColor(gDisplayListHead++, 222, 222 * red, 222 * red, gMenuStringAlpha);
        char warning[128];
        snprintf(warning, 127, "Discord failed to initialize.");
        print_generic_ascii_string(0, 15, warning);
    }
}

static void host_menu_do_host(void) {
#ifndef DISCORD_SDK
    configNetworkSystem = 1;
#endif
    if (configNetworkSystem == 0) {
        network_set_system(NS_DISCORD);
    } else {
        network_set_system(NS_SOCKET);
    }
    custom_menu_close_system();
}

static void host_menu_setting_network_system(void) {
#ifdef DISCORD_SDK
    configNetworkSystem = (configNetworkSystem == 0) ? 1 : 0;
#else
    configNetworkSystem = 1;
#endif
}

static void host_menu_setting_interaction(void) {
    switch (configPlayerInteraction) {
        case 0: configPlayerInteraction = 1; break;
        case 1: configPlayerInteraction = 2; break;
        default: configPlayerInteraction = 0; break;
    }
}

static void host_menu_setting_knockback(void) {
    if (configPlayerKnockbackStrength <= 20) {
        configPlayerKnockbackStrength = 25;
    } else if (configPlayerKnockbackStrength <= 40) {
        configPlayerKnockbackStrength = 75;
    } else {
        configPlayerKnockbackStrength = 10;
    }
}

static void host_menu_setting_stay_in_level(void) {
    configStayInLevelAfterStar = (configStayInLevelAfterStar == 0) ? 1 : 0;
}

static void host_menu_setting_skip_intro(void) {
    configSkipIntro = (configSkipIntro == 1) ? 0 : 1;
}

static void host_menu_setting_share_lives(void) {
    configShareLives = (configShareLives == 0) ? 1 : 0;
}

#ifdef DISCORD_SDK
static void join_menu_draw_strings(void) {
    print_generic_ascii_string(30, 155, "Accept a Discord game invite in order to join.");
    print_generic_ascii_string(30, 130, "For direct connections, click connect to type in an IP.");
}
#endif

static void connect_menu_draw_strings(void) {
    if (gNetworkType == NT_CLIENT) {
        f32 alpha = (f32)fabs(sin(gGlobalTimer / 20.0f));
        gDPSetEnvColor(gDisplayListHead++, 222, 222, 222, gMenuStringAlpha * alpha);
        print_generic_ascii_string(130, 130, "Connecting...");
        return;
    }

    if (*sConnectionJoinError) {
        f32 red = (f32)fabs(sin(gGlobalTimer / 20.0f));
        gDPSetEnvColor(gDisplayListHead++, 222, 222 * red, 222 * red, gMenuStringAlpha);
        f32 messageX = (SCREEN_WIDTH - get_generic_ascii_string_width(sConnectionJoinError)) / 2.0;
        f32 messageY = (SCREEN_HEIGHT + get_generic_ascii_string_height(sConnectionJoinError)) / 2.0;
        print_generic_ascii_string(messageX, messageY, sConnectionJoinError);
        return;
    }

    print_generic_ascii_string(30, 175, "Type in or paste the host's IP.");
    print_generic_ascii_string(30, 160, "Note - the host must forward a port on their router.");

    if (keyboard_in_text_input()) {
        if (strlen(gTextInput) >= 7) {
            print_generic_ascii_string(30, 100, "Press (ENTER) to connect.");
        } else {
            print_generic_ascii_string(30, 100, "Press (ESC) to cancel.");
        }
    }

    gDPSetEnvColor(gDisplayListHead++, 130, 222, 140, gMenuStringAlpha);
    print_generic_ascii_string(30, 130, gTextInput);
}

static void connect_menu_on_connection_attempt(void) {
    play_sound(SOUND_GENERAL_COIN, gDefaultSoundArgs);

    keyboard_stop_text_input();
    if (gNetworkType != NT_NONE) { return; }

    char delims[2] = " ";

    // copy input
    char buffer[MAX_TEXT_INPUT] = { 0 };
    strncpy(buffer, gTextInput, MAX_TEXT_INPUT);
    char* text = buffer;

    // trim whitespace
    while (*text == ' ') { text++; }

    // grab IP
    char* ip = strtok(text, delims);
    if (ip == NULL) { custom_menu_close(); return; }
    strncpy(configJoinIp, ip, MAX_CONFIG_STRING);

    // grab port
    char* port = strtok(NULL, delims);
    if (port != NULL) {
        unsigned int intPort = atoi(port);
        if (intPort == 0 || intPort > 65535) { configJoinPort = DEFAULT_PORT; custom_menu_close(); return; }
        configJoinPort = intPort;
    }
    else {
        configJoinPort = DEFAULT_PORT;
    }

    network_set_system(NS_SOCKET);
    network_init(NT_CLIENT);

}

static void connect_menu_on_click(void) {
    sConnectionJoinError[0] = '\0';

    keyboard_start_text_input(TIM_IP, MAX_TEXT_INPUT, custom_menu_close, connect_menu_on_connection_attempt);

    // fill in our last attempt
    if (configJoinPort == 0 || configJoinPort > 65535) { configJoinPort = DEFAULT_PORT; }

    // only print custom port
    if (configJoinPort == DEFAULT_PORT) {
        sprintf(gTextInput, "%s", configJoinIp);
    }
    else {
        sprintf(gTextInput, "%s %d", configJoinIp, configJoinPort);
    }
}

static void connect_menu_on_close(void) {
    keyboard_stop_text_input();
    network_shutdown();
}

void custom_menu_init(struct CustomMenu* head) {

    // set the header text
    head->me->label = calloc(strlen(MAIN_MENU_HEADER_TEXT) + 1, sizeof(char));
    strcpy(head->me->label, MAIN_MENU_HEADER_TEXT);

    // set main menu settings
    head->headerY = 55;
    head->draw_strings = menu_main_draw_strings;

    // create sub menus and buttons
    struct CustomMenu* hostMenu = custom_menu_create(head, "HOST", -266, 0, gButtonScale.large);
    hostMenu->headerY = 30;
    hostMenu->draw_strings = host_menu_draw_strings;
    custom_menu_create_button(hostMenu, "CANCEL", 700, -196 + (210 * 3), gButtonScale.large, SOUND_MENU_CAMERA_ZOOM_OUT, custom_menu_close);
    custom_menu_create_button(hostMenu, "HOST", 700, -220, gButtonScale.large, SOUND_MENU_CAMERA_ZOOM_IN, host_menu_do_host);
    custom_menu_create_button(hostMenu, "", -700, -180 + (210 * 3), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_interaction);
    custom_menu_create_button(hostMenu, "", -700, -180 + (210 * 2), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_knockback);
    custom_menu_create_button(hostMenu, "", -700, -180 + (210 * 1), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_stay_in_level);
    custom_menu_create_button(hostMenu, "", -700, -180 + (210 * 0), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_skip_intro);
    custom_menu_create_button(hostMenu, "", -700, -180 + (210 * -1), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_share_lives);
    #ifdef DISCORD_SDK
        custom_menu_create_button(hostMenu, "", -700, -180 + (210 * -2), gButtonScale.medium, SOUND_ACTION_BONK, host_menu_setting_network_system);
    #endif

    #ifdef DISCORD_SDK
        struct CustomMenu* joinMenu = custom_menu_create(head, "JOIN", 266, 0, gButtonScale.large);
        custom_menu_create_button(joinMenu, "CANCEL", -266, -320, gButtonScale.large, SOUND_MENU_CAMERA_ZOOM_OUT, custom_menu_close);
        joinMenu->draw_strings = join_menu_draw_strings;
        struct CustomMenu* connectMenu = custom_menu_create(joinMenu, "CONNECT", 266, -320, gButtonScale.large);
    #else
        struct CustomMenu* connectMenu = custom_menu_create(head, "CONNECT", 266, 0, gButtonScale.large);
    #endif
    connectMenu->me->on_click = connect_menu_on_click;
    connectMenu->on_close = connect_menu_on_close;
    connectMenu->draw_strings = connect_menu_draw_strings;
    custom_menu_create_button(connectMenu, "CANCEL", 0, -400, gButtonScale.large, SOUND_MENU_CAMERA_ZOOM_OUT, custom_menu_close);
    sConnectMenu = connectMenu;
}

void custom_menu_loop(void) {
    // we've received an event that makes us exit the menus
    if (sGotoGame) { sSelectedFileNum = sGotoGame; }

    // force-start the load when command-line server hosting
    if (gNetworkType == NT_SERVER && sSelectedFileNum == 0) {
        sSelectedFileNum = 1;
    }

    if (gOpenConnectMenu && sConnectMenu != NULL) {
        gOpenConnectMenu = FALSE;
        custom_menu_open(sConnectMenu);
    }
}

void custom_menu_on_load_save_file(s8 saveFileNum) {
    if (gNetworkType != NT_CLIENT && saveFileNum != 0) {
        configHostSaveSlot = saveFileNum;
        network_init(NT_SERVER);
    }
}

void custom_menu_goto_game(s16 saveFileNum) {
    sGotoGame = saveFileNum;
}

void custom_menu_connection_error(char* message) {
    play_sound(SOUND_MARIO_MAMA_MIA, gDefaultSoundArgs);
    strcpy(sConnectionJoinError, message);
    network_shutdown();
}