#include <map>
#include <string>
#include <switch.h>
#include "filesystem/fileParser.hpp"
#include "ui/strings.hpp"
#include "log.hpp"
#include "stringUtil.hpp"

namespace
{
    // This map holds the strings.
    std::map<std::pair<std::string, int>, std::string> s_UIStringMap;
}

static void addUIString(const std::string &name, int index, const std::string &newString)
{
    s_UIStringMap[std::make_pair(name, index)] = newString;
}

static void replaceButtonsInString(std::string &str)
{
    stringUtil::replaceInString(str, "[A]", "\ue0e0");
    stringUtil::replaceInString(str, "[B]", "\ue0e1");
    stringUtil::replaceInString(str, "[X]", "\ue0e2");
    stringUtil::replaceInString(str, "[Y]", "\ue0e3");
    stringUtil::replaceInString(str, "[L]", "\ue0e4");
    stringUtil::replaceInString(str, "[R]", "\ue0e5");
    stringUtil::replaceInString(str, "[ZL]", "\ue0e6");
    stringUtil::replaceInString(str, "[ZR]", "\ue0e7");
    stringUtil::replaceInString(str, "[SL]", "\ue0e8");
    stringUtil::replaceInString(str, "[SR]", "\ue0e9");
    stringUtil::replaceInString(str, "[DPAD]", "\ue0ea");
    stringUtil::replaceInString(str, "[DUP]", "\ue0eb");
    stringUtil::replaceInString(str, "[DDOWN]", "\ue0ec");
    stringUtil::replaceInString(str, "[DLEFT]", "\ue0ed");
    stringUtil::replaceInString(str, "[DRIGHT]", "\ue0ee");
    stringUtil::replaceInString(str, "[+]", "\ue0ef");
    stringUtil::replaceInString(str, "[-]", "\ue0f0");
}

static void loadTranslationFile(const std::string &filePath)
{
    fs::fileParser translationFile(filePath);

    while(translationFile.readLine())
    {
        // This is the "name" of the string
        std::string stringName = translationFile.getCurrentItemName();
        // This is the index of the string to use. This is mostly for menus.
        int stringIndex = translationFile.getNextValueAsInt();
        // This is the actual string JKSV will use.
        std::string uiString = translationFile.getNextValueAsString();
        // Add it to map.
        addUIString(stringName, stringIndex, uiString);
    }
}

// Initializes default US English strings.
static void initStringsToENUS(void)
{
    // Init strings
    addUIString("author", 0, "NULL");
    addUIString("helpUser", 0, "[A] Select   [Y] Dump All Saves   [X] User Options");
    addUIString("helpTitle", 0, "[A] Select   [L][R] Jump   [Y] Favorite   [X] Title Options  [B] Back");
    addUIString("helpFolder", 0, "[A] Select  [Y] Restore  [X] Delete   [ZR] Upload  [B] Close");
    addUIString("helpSettings", 0, "[A] Toggle   [X] Defaults   [B] Back");

    // Y/N On/Off
    addUIString("dialogYes", 0, "Yes [A]");
    addUIString("dialogNo", 0, "No [B]");
    addUIString("dialogOK", 0, "OK [A]");
    addUIString("settingsOn", 0, ">On>");
    addUIString("settingsOff", 0, "Off");
    addUIString("holdingText", 0, "(Hold) ");
    addUIString("holdingText", 1, "(Keep Holding) ");
    addUIString("holdingText", 2, "(Almost There!) ");

    // Confirmation Strings
    addUIString("confirmBlacklist", 0, "Are you sure you want to add #%s# to your blacklist?");
    addUIString("confirmOverwrite", 0, "Are you sure you want to overwrite #%s#?");
    addUIString("confirmRestore", 0, "Are you sure you want to restore #%s#?");
    addUIString("confirmDelete", 0, "Are you sure you want to delete #%s#? *This is permanent*!");
    addUIString("confirmCopy", 0, "Are you sure you want to copy #%s# to #%s#?");
    addUIString("confirmDeleteSaveData", 0, "*WARNING*: This *will* erase the save data for #%s# *from your system*. Are you sure you want to do this?");
    addUIString("confirmResetSaveData", 0, "*WARNING*: This *will* reset the save data for this game as if it was never ran before. Are you sure you want to do this?");
    addUIString("confirmCreateAllSaveData", 0, "Are you sure you would like to create all save data on this system for #%s#? This can take a while depending on how many titles are found.");
    addUIString("confirmDeleteBackupsTitle", 0, "Are you sure you would like to delete all save backups for #%s#?");
    addUIString("confirmDeleteBackupsAll", 0, "Are you sure you would like to delete *all* of your save backups for all of your games?");
    addUIString("confirmDriveOverwrite", 0, "Downloading this backup from drive will overwrite the one on your SD card. Continue?");

    // Save Data related strings
    addUIString("saveDataNoneFound", 0, "No saves found for #%s#!");
    addUIString("saveDataCreatedForUser", 0, "Save data created for %s!");
    addUIString("saveDataCreationFailed", 0, "Save data creation failed!");
    addUIString("saveDataResetSuccess", 0, "Save for #%s# reset!");
    addUIString("saveDataDeleteSuccess", 0, "Save data for #%s# deleted!");
    addUIString("saveDataExtendSuccess", 0, "Save data for #%s# extended!");
    addUIString("saveDataExtendFailed", 0, "Failed to extend save data.");
    addUIString("saveDataDeleteAllUser", 0, "*ARE YOU SURE YOU WANT TO DELETE ALL SAVE DATA FOR %s?*");
    addUIString("saveDataBackupDeleted", 0, "#%s# has been deleted.");
    addUIString("saveDataBackupMovedToTrash", 0, "#%s# has been moved to trash.");
    addUIString("saveDataErrorMounting", 0, "Error mounting save data!");

    // This is redundant. Need to merge and use one or the other...
    addUIString("saveDataTypeText", 0, "System");
    addUIString("saveDataTypeText", 1, "Account");
    addUIString("saveDataTypeText", 2, "BCAT");
    addUIString("saveDataTypeText", 3, "Device");
    addUIString("saveDataTypeText", 4, "Temporary");
    addUIString("saveDataTypeText", 5, "Cache");
    addUIString("saveDataTypeText", 6, "System BCAT");

    // Internet Related
    addUIString("onlineErrorConnecting", 0, "Error Connecting!");
    addUIString("onlineNoUpdates", 0, "No Updates Available.");

    // File mode menu strings
    addUIString("fileModeMenu", 0, "Copy To ");
    addUIString("fileModeMenu", 1, "Delete");
    addUIString("fileModeMenu", 2, "Rename");
    addUIString("fileModeMenu", 3, "Make Dir");
    addUIString("fileModeMenu", 4, "Properties");
    addUIString("fileModeMenu", 5, "Close");
    addUIString("fileModeMenu", 6, "Add to Path Filters");
    addUIString("fileModeMenuMkDir", 0, "New");

    // New folder pop menu strings
    addUIString("folderMenuNew", 0, "New Backup");

    // File mode properties string
    addUIString("fileModeFileProperties", 0, "Path: %s\nSize: %s");
    addUIString("fileModeFolderProperties", 0, "Path: %s\nSub Folders: %u\nFile Count: %u\nTotal Size: %s");

    // Settings menu
    addUIString("settingsMenu", 0, "Empty Trash Bin");
    addUIString("settingsMenu", 1, "Check for Updates");
    addUIString("settingsMenu", 2, "Set JKSV Save Output Folder");
    addUIString("settingsMenu", 3, "Edit Blacklisted Titles");
    addUIString("settingsMenu", 4, "Delete All Save Backups");
    addUIString("settingsMenu", 5, "Include Device Saves With Users: ");
    addUIString("settingsMenu", 6, "Auto Backup On Restore: ");
    addUIString("settingsMenu", 7, "Auto-Name Backups: ");
    addUIString("settingsMenu", 8, "Overclock/CPU Boost: ");
    addUIString("settingsMenu", 9, "Hold To Delete: ");
    addUIString("settingsMenu", 10, "Hold To Restore: ");
    addUIString("settingsMenu", 11, "Hold To Overwrite: ");
    addUIString("settingsMenu", 12, "Force Mount: ");
    addUIString("settingsMenu", 13, "Account System Saves: ");
    addUIString("settingsMenu", 14, "Enable Writing to System Saves: ");
    addUIString("settingsMenu", 15, "Use FS Commands Directly: ");
    addUIString("settingsMenu", 16, "Export Saves to ZIP: ");
    addUIString("settingsMenu", 17, "Force English To Be Used: ");
    addUIString("settingsMenu", 18, "Enable Trash Bin: ");
    addUIString("settingsMenu", 19, "Title Sorting Type: ");
    addUIString("settingsMenu", 20, "Animation Scale: ");
    addUIString("settingsMenu", 21, "Auto-upload to Drive: ");

    // Main menu
    addUIString("mainMenuSettings", 0, "Settings");
    addUIString("mainMenuExtras", 0, "Extras");

    // Translator in main page
    addUIString("translationMainPage", 0, "Translation: ");

    // Loading page
    addUIString("loadingStartPage", 0, "Loading...");

    // Sort Strings for ^
    addUIString("sortType", 0, "Alphabetical");
    addUIString("sortType", 1, "Time Played");
    addUIString("sortType", 2, "Last Played");

    // Extras
    addUIString("extrasMenu", 0, "SD to SD Browser");
    addUIString("extrasMenu", 1, "BIS: ProdInfoF");
    addUIString("extrasMenu", 2, "BIS: Safe");
    addUIString("extrasMenu", 3, "BIS: System");
    addUIString("extrasMenu", 4, "BIS: User");
    addUIString("extrasMenu", 5, "Remove Pending Update");
    addUIString("extrasMenu", 6, "Terminate Process");
    addUIString("extrasMenu", 7, "Mount System Save");
    addUIString("extrasMenu", 8, "Rescan Titles");
    addUIString("extrasMenu", 9, "Mount Process RomFS");
    addUIString("extrasMenu", 10, "Backup JKSV Folder");
    addUIString("extrasMenu", 11, "*[DEV]* Output en-US");

    // User Options
    addUIString("userOptions", 0, "Dump All For %s");
    addUIString("userOptions", 1, "Create Save Data");
    addUIString("userOptions", 2, "Create All Save Data");
    addUIString("userOptions", 3, "Delete All User Saves");

    // Title Options
    addUIString("titleOptions", 0, "Information");
    addUIString("titleOptions", 1, "Blacklist");
    addUIString("titleOptions", 2, "Change Output Folder");
    addUIString("titleOptions", 3, "Open in File Mode");
    addUIString("titleOptions", 4, "Delete All Save Backups");
    addUIString("titleOptions", 5, "Reset Save Data");
    addUIString("titleOptions", 6, "Delete Save Data");
    addUIString("titleOptions", 7, "Extend Save Data");
    addUIString("titleOptions", 8, "Export SVI");

    // Thread Status Strings
    addUIString("threadStatusCreatingSaveData", 0, "Creating save data for #%s#...");
    addUIString("threadStatusCopyingFile", 0, "Copying '#%s#'...");
    addUIString("threadStatusDeletingFile", 0, "Deleting...");
    addUIString("threadStatusOpeningFolder", 0, "Opening '#%s#'...");
    addUIString("threadStatusAddingFileToZip", 0, "Adding '#%s#' to ZIP...");
    addUIString("threadStatusDecompressingFile", 0, "Decompressing '#%s#'...");
    addUIString("threadStatusDeletingSaveData", 0, "Deleting Save Data for #%s#...");
    addUIString("threadStatusExtendingSaveData", 0, "Extending Save Data for #%s#...");
    addUIString("threadStatusCreatingSaveData", 0, "Creating Save Data for #%s#...");
    addUIString("threadStatusResettingSaveData", 0, "Resetting save data...");
    addUIString("threadStatusDeletingUpdate", 0, "Deleting pending update...");
    addUIString("threadStatusCheckingForUpdate", 0, "Checking for updates...");
    addUIString("threadStatusDownloadingUpdate", 0, "Downloading update...");
    addUIString("threadStatusGetDirProps", 0, "Getting Folder Properties...");
    addUIString("threadStatusPackingJKSV", 0, "Writing JKSV folder contents to ZIP...");
    addUIString("threadStatusSavingTranslations", 0, "Saving the file master...");
    addUIString("threadStatusCalculatingSaveSize", 0, "Calculating save data size...");
    addUIString("threadStatusUploadingFile", 0, "Uploading #%s#...");
    addUIString("threadStatusDownloadingFile", 0, "Downloading #%s#...");
    addUIString("threadStatusCompressingSaveForUpload", 0, "Compressing #%s# for upload...");

    // Random leftover pop-ups
    addUIString("popCPUBoostEnabled", 0, "CPU Boost Enabled for ZIP.");
    addUIString("popErrorCommittingFile", 0, "Error committing file to save!");
    addUIString("popZipIsEmpty", 0, "ZIP file is empty!");
    addUIString("popFolderIsEmpty", 0, "Folder is empty!");
    addUIString("popSaveIsEmpty", 0, "Save data is empty!");
    addUIString("popProcessShutdown", 0, "#%s# successfully shutdown.");
    addUIString("popAddedToPathFilter", 0, "'#%s#' added to path filters.");
    addUIString("popChangeOutputFolder", 0, "#%s# changed to #%s#");
    addUIString("popChangeOutputError", 0, "#%s# contains illegal or non-ASCII characters.");
    addUIString("popTrashEmptied", 0, "Trash emptied");
    addUIString("popSVIExported", 0, "SVI Exported.");
    addUIString("popDriveStarted", 0, "Google Drive started successfully.");
    addUIString("popDriveFailed", 0, "Failed to start Google Drive.");
    addUIString("popDriveNotActive", 0, "Google Drive is not available");
    addUIString("popErrorMountingSave", 0, "Error mounting save data!");

    // Keyboard hints
    addUIString("swkbdEnterName", 0, "Enter a new name");
    addUIString("swkbdSaveIndex", 0, "Enter Cache Index");
    addUIString("swkbdSetWorkDir", 0, "Enter a new Output Path");
    addUIString("swkbdProcessID", 0, "Enter Process ID");
    addUIString("swkbdSysSavID", 0, "Enter System Save ID");
    addUIString("swkbdRename", 0, "Enter a new name for item");
    addUIString("swkbdMkDir", 0, "Enter a folder name");
    addUIString("swkbdNewSafeTitle", 0, "Input New Output Folder");
    addUIString("swkbdExpandSize", 0, "Enter New Size in MB");

    // Status informations
    addUIString("infoStatus", 0, "TID: %016lX");
    addUIString("infoStatus", 1, "SID: %016lX");
    addUIString("infoStatus", 2, "Play Time: %02d:%02d");
    addUIString("infoStatus", 3, "Total Launches: %u");
    addUIString("infoStatus", 4, "Publisher: %s");
    addUIString("infoStatus", 5, "Save Type: %s");
    addUIString("infoStatus", 6, "Cache Index: %u");
    addUIString("infoStatus", 7, "User: %s");

    addUIString("debugStatus", 0, "User Count: ");
    addUIString("debugStatus", 1, "Current User: ");
    addUIString("debugStatus", 2, "Current Title: ");
    addUIString("debugStatus", 3, "Safe Title: ");
    addUIString("debugStatus", 4, "Sort Type: ");

    addUIString("appletModeWarning", 0, "*WARNING*: You are running JKSV in applet mode. Certain functions may not work.");
}

void ui::strings::init(void)
{
    // Get system language.
    SetLanguage systemLanguage = static_cast<SetLanguage>(0);
    uint64_t languageCode = 0;
    Result getLanguageCode = setGetLanguageCode(&languageCode);
    Result makeLanguageCode = setMakeLanguage(languageCode, &systemLanguage);
    // If either failed, or language is US...
    if(R_FAILED(getLanguageCode) || R_FAILED(makeLanguageCode) || systemLanguage == SetLanguage_ENUS || systemLanguage == SetLanguage_ENGB)
    {
        // Init default
        initStringsToENUS();
    }
    else
    {
        // Everything went as it should, determine path.
        std::string translationFilePath = "romfs:/Translations/";
        switch(systemLanguage)
        {
            case SetLanguage_JA:
            {
                translationFilePath += "ja.txt";
            }
            break;
            // Skip US/UK
            case SetLanguage_FR:
            {
                translationFilePath += "fr.txt";
            }
            break;

            case SetLanguage_DE:
            {
                translationFilePath += "de.txt";
            }
            break;

            case SetLanguage_IT:
            {
                translationFilePath += "it.txt";
            }
            break;

            case SetLanguage_ES:
            {
                translationFilePath += "es.txt";
            }
            break;

            case SetLanguage_ZHCN:
            case SetLanguage_ZHHANS:
            {
                translationFilePath += "zh-CN.txt";
            }
            break;

            case SetLanguage_KO:
            {
                translationFilePath += "ko.txt";
            }
            break;

            case SetLanguage_NL:
            {
                translationFilePath += "nl.txt";
            }
            break;

            case SetLanguage_PT:
            {
                translationFilePath += "pt.txt";
            }
            break;

            case SetLanguage_RU:
            {
                translationFilePath += "ru.txt";
            }
            break;

            case SetLanguage_ZHTW:
            case SetLanguage_ZHHANT:
            {
                translationFilePath += "zh-TW.txt";
            }
            break;

            case SetLanguage_FRCA:
            {
                translationFilePath += "fr-CA.txt";
            }
            break;

            case SetLanguage_ES419:
            {
                translationFilePath += "es-419.txt";
            }
            break;

            // I have trouble testing/seeing this one?
            case SetLanguage_PTBR:
            {
                translationFilePath += "pt-BR.txt";
            }
            break;

            // Just pray and use this.
            default:
            {
                translationFilePath += "en-US.txt";
            }
            break;
        }
        // That was long. Load it now.
        loadTranslationFile(translationFilePath);
    }

    // Should be init'd/loaded now...
    replaceButtonsInString(s_UIStringMap[std::make_pair("helpUser", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("helpTitle", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("helpFolder", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("helpSettings", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("dialogYes", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("dialogNo", 0)]);
    replaceButtonsInString(s_UIStringMap[std::make_pair("dialogOK", 0)]);

    logger::log("ui::strings::init(): Succeeded.");
}

std::string ui::strings::getString(const std::string &name, int index)
{
    return s_UIStringMap[std::make_pair(name, index)];
}

const char *ui::strings::getCString(const std::string &name, int index)
{
    return s_UIStringMap[std::make_pair(name, index)].c_str();
}