#include <stdio.h>
#include <string.h>
#include "mainmenu.h"
#include "../utils/util.h"
#include "utils/tools.h"
#include "../utils/btn.h"
#include "emmc/emmc.h"
#include "../storage/emummc.h"
#include "utils/script.h"

#include "common/common.h"
#include "gfx/menu.h"

#include "utils/utils.h"
#include "gfx/gfxutils.h"
#include "fs/fsutils.h"
#include "fs/fsmenu.h"
#include "emmc/emmcoperations.h"

extern bool sd_mount();
extern void sd_unmount();
extern bool return_sd_mounted(int value);
extern int launch_payload(char *path);

/*
menu_item mainmenu[MAINMENU_AMOUNT] = {
    {"[SD:/] SD CARD\n", COLOR_GREEN, SD_CARD, 1},
    {"[SYSTEM:/] EMMC", COLOR_ORANGE, EMMC_SYS, 1},
    {"[USER:/] EMMC", COLOR_ORANGE, EMMC_USR, 1},
    {"[SAFE:/] EMMC", COLOR_ORANGE, EMMC_SAF, 1},
    {"\n[SYSTEM:/] EMUMMC", COLOR_BLUE, EMUMMC_SYS, 1},
    {"[USER:/] EMUMMC", COLOR_BLUE, EMUMMC_USR, 1},
    {"[SAFE:/] EMUMMC", COLOR_BLUE, EMUMMC_SAF, 1},
    {"\nMount/Unmount SD", COLOR_WHITE, MOUNT_SD, 1},
    {"Tools", COLOR_VIOLET, TOOLS, 1},
    {"SD format", COLOR_VIOLET, SD_FORMAT, 1},
    {"\nCredits", COLOR_WHITE, CREDITS, 1},
    {"Exit", COLOR_WHITE, EXIT, 1}
};

menu_item shutdownmenu[7] = {
    {"-- EXIT --\n", COLOR_ORANGE, -1, 0},
    {"Back", COLOR_WHITE, -1, 1},
    {"\nReboot to RCM", COLOR_VIOLET, REBOOT_RCM, 1},
    {"Reboot normally", COLOR_ORANGE, REBOOT_NORMAL, 1},
    {"Power off\n", COLOR_BLUE, POWER_OFF, 1},
    {"Reboot to Hekate", COLOR_GREEN, HEKATE, -1},
    {"Reboot to Atmosphere", COLOR_GREEN, AMS, -1}
};

menu_item toolsmenu[8] = {
    {"-- TOOLS --\n", COLOR_VIOLET, -1, 0},
    {"Back", COLOR_WHITE, -1, 1},
    {"\nDisplay Console Info", COLOR_GREEN, DISPLAY_INFO, 1},
    {"Display GPIO pins", COLOR_VIOLET, DISPLAY_GPIO, 1},
    {"Dump Firmware", COLOR_BLUE, DUMPFIRMWARE, 1},
    {"Dump User Saves", COLOR_YELLOW, DUMPUSERSAVE, 1},
    {"[DEBUG] Dump bis", COLOR_RED, DUMP_BOOT, 1},
    {"[DEBUG] Restore bis", COLOR_RED, RESTORE_BOOT, 1}
};

menu_item formatmenu[4] = {
    {"-- FORMAT SD --\n", COLOR_RED, -1, 0},
    {"Back\n", COLOR_WHITE, -1, 1},
    {"Format entire SD to FAT32", COLOR_RED, FORMAT_ALL_FAT32, 1},
    {"Format for EmuMMC setup (FAT32/RAW)", COLOR_RED, FORMAT_EMUMMC, 1}
};

menu_item mmcChoice[3] = {
    {"Back\n", COLOR_WHITE, -1, 1},
    {"SysMMC", COLOR_ORANGE, SYSMMC, 1},
    {"EmuMMC", COLOR_BLUE, EMUMMC, 1}
};
*/



int res = 0, meter = 0;

void MainMenu_SDCard(){
    fileexplorer("SD:/", 0);
}

void MainMenu_EMMC(){
    gfx_clearscreen();
    gfx_printf("You're about to enter EMMC\nModifying anything here\n        can result in a BRICK!\n\nPlease only continue\n    if you know what you're doing\n\nPress Vol+/- to return\n");
    if (gfx_makewaitmenu("Press Power to enter", 4)){
        connect_mmc(SYSMMC);

        if (!mount_mmc(emmc_fs_entries[res - 2], res - 1))
            fileexplorer("emmc:/", 1);
    }
}

void MainMenu_EMUMMC(){
    connect_mmc(EMUMMC);

    if (!mount_mmc(emmc_fs_entries[res - 5], res - 4))
        fileexplorer("emmc:/", 1);   
}

void MainMenu_MountSD(){
    if (return_sd_mounted(1))
        sd_unmount();
    else
        sd_mount();
}

void MainMenu_Tools(){
    //res = makemenu(toolsmenu, 8);
    res = menu_make(mainmenu_tools, 7, "-- Tools Menu --");

    switch(res){
        case TOOLS_DISPLAY_INFO:
            displayinfo();
            break;
        case TOOLS_DISPLAY_GPIO:
            displaygpio();
            break;
        case TOOLS_DUMPFIRMWARE:
            dumpfirmware(SYSMMC);
            break;
        case TOOLS_DUMPUSERSAVE:
            if ((res = utils_mmcMenu()) > 0)
                dumpusersaves(res);

            break;
        case TOOLS_DUMP_BOOT:
            dump_emmc_parts(PART_BOOT | PART_PKG2, SYSMMC);
            break;
        case TOOLS_RESTORE_BOOT:
            /*
            gfx_printf("WARNING!\nThis will mess with your switch boot files\nMake a nand backup beforehand!\n\nThis will pull from path:\nsd:/tegraexplorer/boot.bis\n\nVol +/- to cancel\n");
            if (gfx_makewaitmenu(
                "Power to confirm",
                5
            ))
            {
                if ((res = utils_mmcMenu()) > 0)
                    restore_bis_using_file("sd:/tegraexplorer/boot.bis", res);
            }
            */
            break;
    }
}

void MainMenu_SDFormat(){
    //res = makemenu(formatmenu, 4);
    res = menu_make(mainmenu_format, 3, "-- Format Menu --");

    if (res > 0){
        gfx_clearscreen();
        gfx_printf("Are you sure you want to format your sd?\nThis will delete everything on your SD card\nThis action is irreversible!\n\nPress Vol+/- to cancel\n");
        if(gfx_makewaitmenu("Press Power to continue", 10)){
            if (format(res)){
                sd_unmount();
            }
        }
    }
}

void MainMenu_Credits(){
    if (++meter >= 3)
        gfx_errDisplay("credits", 53, 0);
    gfx_message(COLOR_WHITE, CREDITS_MESSAGE);
}

void MainMenu_Exit(){
    if (return_sd_mounted(1)){
        /*
        shutdownmenu[5].property = (checkfile("/bootloader/update.bin")) ? 1 : -1;
        shutdownmenu[6].property = (checkfile("/atmosphere/reboot_payload.bin")) ? 1 : -1;
        */

        SETBIT(mainmenu_shutdown[4].property, ISHIDE, !fsutil_checkfile("/bootloader/update.bin"));
        SETBIT(mainmenu_shutdown[5].property, ISHIDE, !fsutil_checkfile("/atmosphere/reboot_payload.bin"));
    }
    else {
        for (int i = 4; i <= 5; i++)
            SETBIT(mainmenu_shutdown[i].property, ISHIDE, 1);
    }

    //res = makemenu(shutdownmenu, 7);
    res = menu_make(mainmenu_shutdown, 6, "-- Shutdown Menu --");

    switch(res){
        case SHUTDOWN_REBOOT_RCM:
            reboot_rcm();
                    
        case SHUTDOWN_REBOOT_NORMAL:
            reboot_normal();

        case SHUTDOWN_POWER_OFF:
            power_off();

        case SHUTDOWN_HEKATE:
            launch_payload("/bootloader/update.bin");
                    
        case SHUTDOWN_AMS:
            launch_payload("/atmosphere/reboot_payload.bin");
    } //todo declock bpmp
}

part_handler mainmenu_functions[] = {
    MainMenu_SDCard,
    MainMenu_EMMC,
    MainMenu_EMMC,
    MainMenu_EMMC,
    MainMenu_EMUMMC,
    MainMenu_EMUMMC,
    MainMenu_EMUMMC,
    MainMenu_MountSD,
    MainMenu_Tools,
    MainMenu_SDFormat,
    MainMenu_Credits,
    MainMenu_Exit,
};

void RunMenuOption(int option){
    if (option != 11)
        meter = 0;
    if (option > 0)
        mainmenu_functions[option - 1]();
}

/*
void fillmainmenu(){
    int i;

    for (i = 0; i < MAINMENU_AMOUNT; i++){
        switch (i + 1) {
            case 5:
            case 6:
            case 7:
                if (mainmenu[i].property == -2)
                    continue;
            case 1:
            case 10:
                if (return_sd_mounted(i + 1))
                    mainmenu[i].property = 1;
                else
                    mainmenu[i].property = -1;
                break;
            case 8:
                if (return_sd_mounted(1)){
                    //mainmenu[i].property = 2;
                    //strcpy(mainmenu[i].name, "\nUnmount SD");
                    mainmenu_main[7].name = (menu_sd_states[0]);
                }
                else {
                    //mainmenu[i].property = 1;
                    //strcpy(mainmenu[i].name, "\nMount SD");
                    mainmenu_main[7].name = (menu_sd_states[1]);
                }
                break;
        }
    }
}
*/

void te_main(){
    int setter;

    if (dump_biskeys() == -1){
        gfx_errDisplay("dump_biskey", ERR_BISKEY_DUMP_FAILED, 0);
        for (int i = 1; i <= 3; i++)
            mainmenu_main[i].property |= ISHIDE;
    }

    if (emummc_load_cfg()){
        for (int i = 4; i <= 6; i++)
            mainmenu_main[i].property |= ISHIDE;
    }

    disconnect_mmc();

    while (1){
        //fillmainmenu();

        setter = return_sd_mounted(1);

        if (emu_cfg.enabled){
            for (int i = 4; i <= 6; i++)
                SETBIT(mainmenu_main[i].property, ISHIDE, !setter);
        }
        SETBIT(mainmenu_main[0].property, ISHIDE, !setter);
        mainmenu_main[7].name = (menu_sd_states[!setter]);

        /*
        if (return_sd_mounted(1)){
            if (emu_cfg.enabled){
                for (int i = 4; i <= 6; i++)
                    SETBIT(mainmenu_main[i], ISHIDE, 0);
            }
            SETBIT(mainmenu_main[0], ISHIDE, 0);
            mainmenu_main[7].name = (menu_sd_states[0]);
        }
        else {
            if (emu_cfg.enabled){
                for (int i = 4; i <= 6; i++)
                    SETBIT(mainmenu_main[i], ISHIDE, 1);
            }
            SETBIT(mainmenu_main[0], ISHIDE, 1);
            mainmenu_main[7].name = (menu_sd_states[1]);
        }
        */

       setter = return_sd_mounted(10);
       SETBIT(mainmenu_main[9].property, ISHIDE, !setter);

        /*
        if (return_sd_mounted(10))
            SETBIT(mainmenu_main[0], ISHIDE, 0);
        else
            SETBIT(mainmenu_main[0], ISHIDE, 1);
        */


        //res = makemenu(mainmenu, MAINMENU_AMOUNT);
        res = menu_make(mainmenu_main, 12, "-- Main Menu --") + 1;
        RunMenuOption(res);
    }
}