Wii-U-Account-Swap/source/main.cpp

#include <filesystem>

#include <coreinit/screen.h>
#include <coreinit/thread.h>
#include <mocha/mocha.h>
#include <nn/act.h>
#include <SDL2/SDL.h>
#include <SDL2/SDL_ttf.h>
#include <sndcore2/core.h>
#include <sysapp/launch.h>
#include <whb/log.h>
#include <whb/log_console.h>
#include <whb/proc.h>

#include "main.hpp"

#include "input.hpp"
#include "start_screen.hpp"
#include "sub_screens.hpp"
#include "utils.hpp"


bool INKAY_EXISTS; // Whether the Inkay plugin config exists.
std::string NNID_BACKUP; // The backup path to the Nintendo Network ID account.dat.
std::string PNID_BACKUP; // The backup path to the Pretendo Network ID account.dat.
std::string MII_NICKNAME; // The current user's Mii nickname.
std::string PERSISTENT_ID; // The current user persistent ID.
std::string ACCOUNT_ID; // The current user's account name.
std::string ACCOUNT_FILE; // The path to the current account.dat.
std::string INKAY_CONFIG; // The path to the Inkay configuration file.

SDL_Window* window = nullptr; // Global window variable.
SDL_Renderer* renderer = nullptr; // Global renderer variable.

int client_handle = 0; // The client handle for flushing the filesystem.


void
get_user_information()
{
    // Grab the user's Mii name.
    nn::act::Initialize();
    int16_t mii_name[256];
    nn::act::GetMiiName(mii_name);
    MII_NICKNAME = std::string(mii_name, mii_name + sizeof(mii_name) / sizeof(mii_name[0]));
    
    // Grab the user's account ID.
    char account_id[256];
    nn::act::GetAccountId(account_id);
    ACCOUNT_ID = std::string(account_id);

    // Get the user's persistent ID.
    unsigned int persistent_id_result = nn::act::GetPersistentId();
    char persistent_id_hex[9];
    snprintf(persistent_id_hex, sizeof persistent_id_hex, "%08x", persistent_id_result);
    PERSISTENT_ID = std::string(persistent_id_hex);

    // Set the account file path.
    ACCOUNT_FILE = "storage_mlc:/usr/save/system/act/" + PERSISTENT_ID + "/account.dat";

    // Set the backup file paths.
    NNID_BACKUP = "fs:/vol/external01/wiiu/accounts/" + PERSISTENT_ID + "/nnid_account.dat";
    PNID_BACKUP = "fs:/vol/external01/wiiu/accounts/" + PERSISTENT_ID + "/pnid_account.dat";

    // Get the environment path and set the Inkay configuration file path.
    char environment_path_buffer[0x100];
    Mocha_GetEnvironmentPath(environment_path_buffer, sizeof(environment_path_buffer));
    INKAY_CONFIG = std::string(environment_path_buffer) + std::string("/plugins/config/inkay.json");

    // Check if the Inkay configuration file exists.
    INKAY_EXISTS = std::filesystem::exists(INKAY_CONFIG);
}


void
deinitialize()
{
    FSAFlushVolume(client_handle, "/vol/external01"); // Probably not necessary, but doesn't hurt.
    FSAFlushVolume(client_handle, "/vol/storage_mlc01"); // MLC actually does need to be flushed.
    FSADelClient(client_handle);

    nn::act::Finalize();
    Mocha_UnmountFS("storage_mlc");
    Mocha_DeInitLibrary();
    WHBLogConsoleFree();
    VPADShutdown();
    KPADShutdown();
    AXQuit();

    if (renderer != nullptr)
        SDL_DestroyRenderer(renderer);
    if (window != nullptr)
        SDL_DestroyWindow(window);
    
    clean_font_caches();

    TTF_Quit();
    SDL_Quit();
}


bool
initialize_program()
{
    WHBProcInit();
    VPADInit(); // Wii U GamePad support.
    KPADInit(); // External controller support.
    WPADEnableURCC(1); // Allow Pro Controller input.
    WHBLogConsoleInit(); // Initialize the log console.
    FSAInit(); // Allows for flushing the filesystem.
    AXInit(); // Disable playing audio.

    // Initialize the Mocha library.
    if (Mocha_InitLibrary() != MOCHA_RESULT_SUCCESS) {
        return false;
    }

    // Initialize the FSA handler to flush the filesystem.
    client_handle = FSAAddClient(NULL);
    if (client_handle == 0) {
        return false;
    }

    Mocha_MountFS("storage_mlc", NULL, "/vol/storage_mlc01");
    get_user_information();

    return true;
}


bool
initialize_graphics()
{
    SDL_Init(SDL_INIT_VIDEO);
    TTF_Init();

    window = SDL_CreateWindow("Wii U Account Swap", SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED, \
        SCREEN_WIDTH, SCREEN_HEIGHT, 0);
    if (!window)
        return false;

    renderer = SDL_CreateRenderer(window, -1, SDL_RENDERER_ACCELERATED);
    if (!renderer)
        return false;

    return true;
}


bool
check_initialization(bool condition, const char* error_message)
{
    if (!condition) {
        WHBLogConsoleSetColor(0x99000000);
        WHBLogPrint(error_message);
        WHBLogConsoleDraw();
        OSSleepTicks(OSMillisecondsToTicks(5000));
        SYSLaunchMenu();
        return true;
    }
    return false;
}


int
main()
{
    bool error_occurred = false;

    error_occurred = check_initialization(initialize_program(), "Error initializing program!") ||
        check_initialization(initialize_graphics(), "Error initializing graphics!");

    while (WHBProcIsRunning()) {
        if (!error_occurred)
            process_screens();
    }

    deinitialize();
    WHBProcShutdown();
    return 0;
}