#include <string>
#include <fstream>
#include <streambuf>
#include <vector>

#include <unistd.h>

#include <libcore/libcore.h>

#include "extra.h"
#include "cxx11_util.h"

#include <cstdio>

extern "C" {
    // Read Asset File
    static cxx11_string AppPlatform_readAssetFile(__attribute__((unused)) unsigned char *app_platform, std::string const& path) {
        std::string full_path("./data/");
        full_path.append(path);
        std::ifstream stream(full_path);
        std::string str((std::istreambuf_iterator<char>(stream)), std::istreambuf_iterator<char>());
        return create_cxx11_string(str.c_str());
    }

    typedef unsigned char *(*TextEditScreen_t)(unsigned char *, unsigned char *);
    static TextEditScreen_t TextEditScreen = (TextEditScreen_t) 0x3a840;

    typedef void (*Minecraft_setScreen_t)(unsigned char *, unsigned char *);
    static Minecraft_setScreen_t Minecraft_setScreen = (Minecraft_setScreen_t) 0x15d6c;

    // Open Sign Screen
    static void LocalPlayer_openTextEdit(unsigned char *local_player, unsigned char *sign) {
        if (*(int *)(sign + 0x18) == 4) {
            unsigned char *minecraft = *(unsigned char **) (local_player + 0xc90);
            unsigned char *screen = (unsigned char *) ::operator new(0xd0);
            screen = (*TextEditScreen)(screen, sign);
            (*Minecraft_setScreen)(minecraft, screen);
        }
    }

    #define BACKSPACE_KEY 8

    static int is_valid_key(char key) {
        return (key >= 32 && key <= 126) || key == BACKSPACE_KEY;
    }

    // Store Text Input
    std::vector<char> input;
    void extra_key_press(char key) {
        if (is_valid_key(key)) {
            input.push_back(key);
        }
    }
    void extra_clear_input() {
        input.clear();
    }

    typedef void (*Screen_updateEvents_t)(unsigned char *screen);
    static Screen_updateEvents_t Screen_updateEvents = (Screen_updateEvents_t) 0x28eb8;

    typedef void (*Screen_keyboardNewChar_t)(unsigned char *screen, char key);
    typedef void (*Screen_keyPressed_t)(unsigned char *screen, int32_t key);

    // Handle Text Input
    static void Screen_updateEvents_injection(unsigned char *screen) {
        // Call Original Method
        (*Screen_updateEvents)(screen);

        if (*(char *)(screen + 4) == '\0') {
            uint32_t vtable = *((uint32_t *) screen);
            for (char key : input) {
                if (key == BACKSPACE_KEY) {
                    // Handle Backspace
                    (*(Screen_keyPressed_t *) (vtable + 0x6c))(screen, BACKSPACE_KEY);
                } else {
                    // Handle Nrmal Key
                    (*(Screen_keyboardNewChar_t *) (vtable + 0x70))(screen, key);
                }
            }
        }
        extra_clear_input();
    }

    typedef unsigned char *(*ItemInstance_t)(unsigned char *item_instance, unsigned char *item);
    static ItemInstance_t ItemInstance_item = (ItemInstance_t) 0x9992c;
    static ItemInstance_t ItemInstance_tile = (ItemInstance_t) 0x998e4;
    typedef unsigned char *(*ItemInstance_damage_t)(unsigned char *item_instance, unsigned char *item, int32_t count, int32_t damage);
    static ItemInstance_damage_t ItemInstance_damage = (ItemInstance_damage_t) 0x99960;

    typedef int32_t (*FillingContainer_addItem_t)(unsigned char *filling_container, unsigned char *item_instance);
    static FillingContainer_addItem_t FillingContainer_addItem = (FillingContainer_addItem_t) 0x92aa0;

    static void inventory_add_item(unsigned char *inventory, unsigned char *item, bool is_tile) {
        unsigned char *item_instance = (unsigned char *) ::operator new(0xc);
        item_instance = (*(is_tile ? ItemInstance_tile : ItemInstance_item))(item_instance, item);
        (*FillingContainer_addItem)(inventory, item_instance);
    }

    // Items
    static unsigned char **item_flintAndSteel = (unsigned char **) 0x17ba70;
    static unsigned char **item_snowball = (unsigned char **) 0x17bbb0;
    static unsigned char **item_shears = (unsigned char **) 0x17bbf0;
    static unsigned char **item_egg = (unsigned char **) 0x17bbd0;
    static unsigned char **item_dye_powder = (unsigned char **) 0x17bbe0;
    // Tiles
    static unsigned char **tile_water = (unsigned char **) 0x181b3c;
    static unsigned char **tile_lava = (unsigned char **) 0x181cc8;
    static unsigned char **tile_calmWater = (unsigned char **) 0x181b40;
    static unsigned char **tile_calmLava = (unsigned char **) 0x181ccc;
    static unsigned char **tile_glowingObsidian = (unsigned char **) 0x181dcc;
    static unsigned char **tile_topSnow = (unsigned char **) 0x181b30;
    static unsigned char **tile_ice = (unsigned char **) 0x181d80;
    static unsigned char **tile_invisible_bedrock = (unsigned char **) 0x181d94;

    static int32_t FillingContainer_addItem_injection(unsigned char *filling_container, unsigned char *item_instance) {
        // Call Original
        int32_t ret = (*FillingContainer_addItem)(filling_container, item_instance);

        // Add Items
        inventory_add_item(filling_container, *item_flintAndSteel, false);
        inventory_add_item(filling_container, *item_snowball, false);
        inventory_add_item(filling_container, *item_egg, false);
        inventory_add_item(filling_container, *item_shears, false);
        for (int i = 0; i < 15; i++) {
            unsigned char *item_instance = (unsigned char *) ::operator new(0xc);
            item_instance = (*ItemInstance_damage)(item_instance, *item_dye_powder, 1, i);
            (*FillingContainer_addItem)(filling_container, item_instance);
        }
        // Add Tiles
        inventory_add_item(filling_container, *tile_water, true);
        inventory_add_item(filling_container, *tile_lava, true);
        inventory_add_item(filling_container, *tile_calmWater, true);
        inventory_add_item(filling_container, *tile_calmLava, true);
        inventory_add_item(filling_container, *tile_glowingObsidian, true);
        inventory_add_item(filling_container, *tile_topSnow, true);
        inventory_add_item(filling_container, *tile_ice, true);
        inventory_add_item(filling_container, *tile_invisible_bedrock, true);

        return ret;
    }

    typedef void (*Minecraft_tick_t)(unsigned char *minecraft, int32_t param_1, int32_t param_2);
    static Minecraft_tick_t Minecraft_tick = (Minecraft_tick_t) 0x16934;

    typedef void (*Textures_tick_t)(unsigned char *textures, bool param_1);
    static Textures_tick_t Textures_tick = (Textures_tick_t) 0x531c4;

    static void Minecraft_tick_injection(unsigned char *minecraft, int32_t param_1, int32_t param_2) {
        // Call Original Method
        (*Minecraft_tick)(minecraft, param_1, param_2);

        // Tick Dynamic Textures
        unsigned char *textures = *(unsigned char **) (minecraft + 0x164);
        if (textures != NULL) {
            (*Textures_tick)(textures, true);
        }
    }

    __attribute((constructor)) static void init() {
        // Implement AppPlatform::AppPlatform_AppPlatform_readAssetFile So Translations Work
        overwrite((void *) 0x12b10, (void *) AppPlatform_readAssetFile);

        if (extra_has_feature("Fix Sign Placement")) {
            // Fix Signs
            patch_address((void *) 0x106460, (void *) LocalPlayer_openTextEdit);
            patch_address((void *) 0x10531c, (void *) Screen_updateEvents_injection);
        }

        if (extra_has_feature("Expand Creative Inventory")) {
            // Add Extra Items To Creative Inventory (Only Replace Specific Function Call)
            overwrite_call((void *) 0x8e0fc, (void *) FillingContainer_addItem_injection);
        }

        if (extra_has_feature("Animated Water")) {
            // Tick Dynamic Textures (Animated Water)
            overwrite_calls((void *) Minecraft_tick, (void *) Minecraft_tick_injection);
        }
    }
}