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

#include <map>
#include <string>

#include <coreinit/memory.h>
#include <SDL2/SDL_ttf.h>

#include <fa-solid-900_ttf.h>
#include "input.hpp"
#include "main.hpp"
#include "nintendo_glyphs.hpp"
#include "unlink.hpp"
#include "utils.hpp"


std::map<int, TTF_Font*> text_font_cache;
std::map<int, TTF_Font*> icon_font_cache;


void
draw_background(int r, int g, int b, int a)
{
    SDL_SetRenderDrawColor(renderer, r, g, b, a);
    SDL_RenderClear(renderer);
}


void
draw_rectangle(int x, int y, int w, int h, int r, int g, int b, int a)
{
    SDL_Rect rect = {x, y, w, h};
    SDL_SetRenderDrawColor(renderer, r, g, b, a);
    SDL_RenderFillRect(renderer, &rect);
}


TTF_Font*
get_text_font(int size)
{
    // Check if the font is already in the cache.
    if (text_font_cache.find(size) != text_font_cache.end()) {
        return text_font_cache[size];
    }

    // If not, create a new font.
    void* font_data = nullptr;
    uint32_t font_size = 0;
    OSGetSharedData(OS_SHAREDDATATYPE_FONT_STANDARD, 0, &font_data, &font_size);
    TTF_Font* font = TTF_OpenFontRW(SDL_RWFromMem((void*)font_data, font_size), 1, size);

    if (font == NULL) {
        return NULL;
    }

    // Add the new font to the cache.
    text_font_cache[size] = font;

    return font;
}


TTF_Font*
get_icon_font(int size)
{
    // Check if the font is already in the cache.
    if (icon_font_cache.find(size) != icon_font_cache.end()) {
        return icon_font_cache[size];
    }

    // If not, create a new font. fa-solid-900.ttf is a custom font file that contains icons.
    TTF_Font* font = TTF_OpenFontRW(SDL_RWFromMem((void*)fa_solid_900_ttf, fa_solid_900_ttf_size), 1, size);

    if (font == NULL) {
        return NULL;
    }

    // Add the new font to the cache.
    icon_font_cache[size] = font;
    
    return font;
}


void
draw_text(const char* text, int x, int y, int size, SDL_Color color)
{
    TTF_Font* font = get_text_font(size);
    if (font == NULL) {
        return;
    }

    // UTF8 allows us to render all the special system characters.
    SDL_Surface* surface = TTF_RenderUTF8_Blended(font, text, color);
    if (surface == NULL) {
        TTF_CloseFont(font);
        return;
    }

    SDL_Texture* texture = SDL_CreateTextureFromSurface(renderer, surface);
    if (texture == NULL) {
        SDL_FreeSurface(surface);
        TTF_CloseFont(font);
        return;
    }

    SDL_Rect rect = {x, y, surface->w, surface->h};
    SDL_RenderCopy(renderer, texture, NULL, &rect);

    SDL_DestroyTexture(texture);
    SDL_FreeSurface(surface);
}


void
draw_icon(const char* icon, int x, int y, int size, SDL_Color color)
{
    TTF_Font* font = get_icon_font(size);
    if (font == NULL) {
        return;
    }

    SDL_Surface* surface = TTF_RenderUTF8_Blended(font, icon, color);
    if (surface == NULL) {
        TTF_CloseFont(font);
        return;
    }

    SDL_Texture* texture = SDL_CreateTextureFromSurface(renderer, surface);
    if (texture == NULL) {
        SDL_FreeSurface(surface);
        TTF_CloseFont(font);
        return;
    }

    SDL_Rect rect = {x, y, surface->w, surface->h};
    SDL_RenderCopy(renderer, texture, NULL, &rect);

    SDL_DestroyTexture(texture);
    SDL_FreeSurface(surface);
}


int
get_text_size(const char* text, int size, bool get_height)
{
    TTF_Font* font = get_text_font(size);
    if (font == NULL) {
        return 0;
    }

    int width = 0;
    int height = 0;
    TTF_SizeUTF8(font, text, &width, &height); // This works with unicode characters.

    TTF_CloseFont(font);

    if (get_height)
        return height; // We don't usually need the height.
    
    return width;
}


void
close_fonts()
{
    for (auto& pair : text_font_cache) {
        TTF_CloseFont(pair.second);
    }
    for (auto& pair : icon_font_cache) {
        TTF_CloseFont(pair.second);
    }
    text_font_cache.clear();
    icon_font_cache.clear();
}


void
draw_screen_bars(bool show_confirm, bool show_controls)
{
    // These lines draw the top bar (with title, version, and author).
    draw_rectangle(0, 0, SCREEN_WIDTH, 90, 100, 0, 100, 255);
    draw_text("Wii U Account Swap", 64, 10, 50);
    draw_text(APP_VERSION, 64 + get_text_size("Wii U Account Swap", 50) + 16, 10, 50, {176, 176, 176, 255});
    draw_text("Nightkingale", SCREEN_WIDTH - 64 - get_text_size("Nightkingale", 50), 10, 50);

    // These lines draw the bottom bar (with current user and account file).
    draw_rectangle(0, 940, SCREEN_WIDTH, 140, 100, 0, 100, 255);
    draw_text("Current User: ", 64, 955, 40);
    draw_text(MII_NICKNAME.c_str(), 64 + get_text_size("Current User: ", 40), 955, 40, {176, 176, 176, 255});

    if (INKAY_EXISTS)
        // Draw the plugin checkmark icon next to the name.
        draw_icon("\uE55C", 64 + get_text_size("Current User: ", 40) + get_text_size(MII_NICKNAME.c_str(), 40) \
            + 16, 960, 40);
    else
        // Draw the plugin exit icon next to the name.
        draw_icon("\uE560", 64 + get_text_size("Current User: ", 40) + get_text_size(MII_NICKNAME.c_str(), 40) \
            + 16, 960, 40);

    draw_text(ACCOUNT_FILE.c_str(), 64, 1005, 40, {176, 176, 176, 255});

    if (show_confirm) {
        draw_rectangle(0, 940, SCREEN_WIDTH, 140, 100, 0, 100, 255);
        std::string confirm_text = std::string(NIN_GLYPH_BTN_A) + " Confirm";
        draw_text(confirm_text.c_str(), 64, 975, 50);
        std::string decline_text = std::string(NIN_GLYPH_BTN_B) + " Decline";
        draw_text(decline_text.c_str(), SCREEN_WIDTH - 64 - get_text_size(decline_text.c_str(), 50), 975, 50);
    } else if (show_controls) {
        std::string navigate_text = std::string(NIN_GLYPH_BTN_DPAD_UP_DOWN) + " Navigate";
        draw_text(navigate_text.c_str(), SCREEN_WIDTH - 64 - get_text_size(navigate_text.c_str(), 50), 975, 50);
    }
}