#include <libreborn/libreborn.h>

#include <mods/feature/feature.h>
#include <mods/init/init.h>
#include <mods/touch/touch.h>

#include <symbols/minecraft.h>

// Enable Touch GUI
static int32_t Minecraft_isTouchscreen_injection(__attribute__((unused)) Minecraft *minecraft) {
    return 1;
}

// IngameBlockSelectionScreen Memory Allocation Override
static unsigned char *operator_new_IngameBlockSelectionScreen_injection(__attribute__((unused)) uint32_t size) {
    return (unsigned char *) ::operator new(TOUCH_INGAME_BLOCK_SELECTION_SCREEN_SIZE);
}

// Improved Button Hover Behavior
static int32_t Button_hovered_injection(__attribute__((unused)) Button *button, __attribute__((unused)) Minecraft *minecraft, __attribute__((unused)) int32_t click_x, __attribute__((unused)) int32_t click_y) {
    // Get Mouse Position
    int32_t x = Mouse_getX() * Gui_InvGuiScale;
    int32_t y = Mouse_getY() * Gui_InvGuiScale;

    // Get Button Position
    int32_t button_x1 = button->x;
    int32_t button_y1 = button->y;
    int32_t button_x2 = button_x1 + button->width;
    int32_t button_y2 = button_y1 + button->height;

    // Check
    return x >= button_x1 && x < button_x2 && y >= button_y1 && y < button_y2;
}
static void LargeImageButton_render_GuiComponent_drawCenteredString_injection(GuiComponent *component, Font *font, std::string *text, int32_t x, int32_t y, int32_t color) {
    // Change Color On Hover
    if (color == 0xe0e0e0 && Button_hovered_injection((Button *) component, NULL, 0, 0)) {
        color = 0xffffa0;
    }

    // Call Original Method
    GuiComponent_drawCenteredString(component, font, text, x, y, color);
}

// Create Button
int touch_gui = 0;
Button *touch_create_button(int id, std::string text) {
    Button *button = nullptr;
    if (touch_gui) {
        button = (Button *) new Touch_TButton;
    } else {
        button = new Button;
    }
    ALLOC_CHECK(button);
    if (touch_gui) {
        Touch_TButton_constructor((Touch_TButton *) button, id, &text);
    } else {
        Button_constructor(button, id, &text);
    }
    return button;
}

// Init
void init_touch() {
    touch_gui = feature_has("Full Touch GUI", server_disabled);
    int touch_buttons = touch_gui;
    if (touch_gui) {
        // Main UI
        overwrite((void *) Minecraft_isTouchscreen, (void *) Minecraft_isTouchscreen_injection);

        // Force Correct Toolbar Size
        unsigned char toolbar_patch[4] = {0x01, 0x00, 0x50, 0xe3}; // "cmp r0, #0x1"
        patch((void *) 0x257b0, toolbar_patch);
    } else {
        // Force Touch Inventory
        if (feature_has("Force Touch GUI Inventory", server_disabled)) {
            overwrite_call((void *) 0x2943c, (void *) operator_new_IngameBlockSelectionScreen_injection);
            overwrite_call((void *) 0x29444, (void *) Touch_IngameBlockSelectionScreen_constructor);
            // Make "Craft" And "Armor" Buttons Use Classic GUI Style (Button And TButton Have The Same Size)
            overwrite_call((void *) 0x3b060, (void *) Button_constructor);
            overwrite_call((void *) 0x3b08c, (void *) Button_constructor);
        }

        // Force Touch Button Behavior
        if (feature_has("Force Touch GUI Button Behavior", server_disabled)) {
            touch_buttons = 1;
            overwrite_call((void *) 0x1baf4, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x1be40, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x1c470, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x1e868, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x290b8, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x29168, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x3e314, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x2cbc0, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x2ea7c, (void *) Minecraft_isTouchscreen_injection);
            overwrite_call((void *) 0x4a438, (void *) Minecraft_isTouchscreen_injection);
        }
    }

    // Improved Button Hover Behavior
    if (touch_buttons && feature_has("Improved Button Hover Behavior", server_disabled)) {
        overwrite((void *) Button_hovered, (void *) Button_hovered_injection);
        overwrite_call((void *) 0x1ebd4, (void *) LargeImageButton_render_GuiComponent_drawCenteredString_injection);
    }

    // Show Block Outlines
    int block_outlines = feature_has("Show Block Outlines", server_disabled);
    unsigned char outline_patch[4] = {(unsigned char) (block_outlines ? !touch_gui : touch_gui), 0x00, 0x50, 0xe3}; // "cmp r0, #0x1" or "cmp r0, #0x0"
    patch((void *) 0x4a210, outline_patch);
}