#include <map>
#include <string>

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

#include <fa-solid-900_ttf.h>

#include "utils.hpp"

#include "main.hpp"
#include "nintendo_glyphs.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) {
        return;
    }

    SDL_Texture* texture = SDL_CreateTextureFromSurface(renderer, surface);
    if (texture == NULL) {
        SDL_FreeSurface(surface);
        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) {
        return;
    }

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

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

    SDL_DestroyTexture(texture);
    SDL_FreeSurface(surface);
}


void
draw_button(const std::string& glyph, const std::string& text, int x, int y, int size, bool align_right = false)
{
    std::string full_text = glyph + " " + text;
    int text_width = get_text_size(full_text.c_str(), size);
    if (align_right) {
        x = x - text_width;
    }
    draw_text(full_text.c_str(), x, y, size);
}


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.


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


void
clean_font_caches()
{
    for (auto& pair : text_font_cache) {
        TTF_CloseFont(pair.second);
    }
    text_font_cache.clear();
    
    for (auto& pair : icon_font_cache) {
        TTF_CloseFont(pair.second);
    }
    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});

    // Calculate the common x-coordinate offset for the text and icon.
    int offset = 64 + get_text_size("Current User: ", 40) + get_text_size(MII_NICKNAME.c_str(), 40) + 32;

    // If the account ID is not empty, display "Network ID: (ID)" next to the name.
    std::string account_display = ACCOUNT_ID.empty() ? "N/A" : ACCOUNT_ID;
    draw_text("Network ID: ", offset, 955, 40);
    offset += get_text_size("Network ID: ", 40);
    draw_text(account_display.c_str(), offset, 955, 40, {176, 176, 176, 255});

    // Draw the plugin icon next to the name.
    std::string icon = INKAY_EXISTS ? "\uE55C" : "\uE560";
    offset += get_text_size(account_display.c_str(), 40) + 32; // Only add the size of the account ID.
    draw_icon(icon.c_str(), offset, 960, 40);

    // Draw the account file name.
    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);
        draw_button(NIN_GLYPH_BTN_A, "Confirm", 64, 975, 50);
        draw_button(NIN_GLYPH_BTN_B, "Decline", SCREEN_WIDTH - 64, 975, 50, true);
    } else if (show_controls) {
        draw_button(NIN_GLYPH_BTN_DPAD_UP_DOWN, "Navigate", SCREEN_WIDTH - 64, 975, 50, true);
    }
}