#include <libreborn/libreborn.h>
#include <symbols/minecraft.h>

#include <mods/init/init.h>
#include <mods/feature/feature.h>
#include <mods/misc/misc.h>
#include <mods/creative/creative.h>

#ifndef MCPI_SERVER_MODE
// Add Item To Inventory
static void inventory_add_item(FillingContainer *inventory, Item *item) {
    ItemInstance *item_instance = new ItemInstance;
    ALLOC_CHECK(item_instance);
    item_instance = item_instance->constructor_item(item);
    inventory->addItem(item_instance);
}
static void inventory_add_item(FillingContainer *inventory, Tile *item) {
    ItemInstance *item_instance = new ItemInstance;
    ALLOC_CHECK(item_instance);
    item_instance = item_instance->constructor_tile(item);
    inventory->addItem(item_instance);
}

// Expand Creative Inventory
static void Inventory_setupDefault_FillingContainer_addItem_call_injection(FillingContainer *filling_container) {
    // Add Items
    inventory_add_item(filling_container, Item_flintAndSteel);
    inventory_add_item(filling_container, Item_snowball);
    inventory_add_item(filling_container, Item_egg);
    inventory_add_item(filling_container, Item_shears);
    // Dyes
    for (int i = 0; i < 16; i++) {
        if (i == 15) {
            // Bonemeal Is Already In The Creative Inventory
            continue;
        }
        ItemInstance *new_item_instance = new ItemInstance;
        ALLOC_CHECK(new_item_instance);
        new_item_instance = new_item_instance->constructor_item_extra(Item_dye_powder, 1, i);
        filling_container->addItem(new_item_instance);
    }
    inventory_add_item(filling_container, Item_camera);
    // Add Tiles
    inventory_add_item(filling_container, Tile_water);
    inventory_add_item(filling_container, Tile_lava);
    inventory_add_item(filling_container, Tile_calmWater);
    inventory_add_item(filling_container, Tile_calmLava);
    inventory_add_item(filling_container, Tile_glowingObsidian);
    inventory_add_item(filling_container, Tile_web);
    inventory_add_item(filling_container, Tile_topSnow);
    inventory_add_item(filling_container, Tile_ice);
    inventory_add_item(filling_container, Tile_invisible_bedrock);
    inventory_add_item(filling_container, Tile_bedrock);
    inventory_add_item(filling_container, Tile_info_updateGame1);
    inventory_add_item(filling_container, Tile_info_updateGame2);
    // Nether Reactor
    for (int i = 0; i < 3; i++) {
        if (i == 0) {
            // Default Is Already In The Creative Inventory
            continue;
        }
        ItemInstance *new_item_instance = new ItemInstance;
        ALLOC_CHECK(new_item_instance);
        new_item_instance = new_item_instance->constructor_tile_extra(Tile_netherReactor, 1, i);
        filling_container->addItem(new_item_instance);
    }
    // Tall Grass
    for (int i = 0; i < 4; i++) {
        if (i == 2) {
            // Identical To Previous Auxiliary Value
            continue;
        }
        ItemInstance *new_item_instance = new ItemInstance;
        ALLOC_CHECK(new_item_instance);
        new_item_instance = new_item_instance->constructor_tile_extra(Tile_tallgrass, 1, i);
        filling_container->addItem(new_item_instance);
    }
    // Smooth Stone Slab
    {
        ItemInstance *new_item_instance = new ItemInstance;
        ALLOC_CHECK(new_item_instance);
        new_item_instance = new_item_instance->constructor_tile_extra(Tile_stoneSlab, 1, 6);
        filling_container->addItem(new_item_instance);
    }
}
#endif

// Hook Specific TileItem Constructor
static TileItem *Tile_initTiles_TileItem_injection(TileItem *tile_item, int32_t id) {
    // Call Original Method
    tile_item->constructor(id);

    // Switch VTable
    tile_item->vtable = (TileItem_vtable *) AuxDataTileItem_vtable_base;
    // Configure Item
    tile_item->is_stacked_by_data = true;
    tile_item->max_damage = 0;
    ((AuxDataTileItem *) tile_item)->icon_tile = Tile_tiles[id + 0x100];

    // Return
    return tile_item;
}

// Check Restriction Status
static int is_restricted = 1;
int creative_is_restricted() {
    return is_restricted;
}

// Init
void init_creative() {
    // Add Extra Items To Creative Inventory (Only Replace Specific Function Call)
    if (feature_has("Expand Creative Mode Inventory", server_enabled)) {
#ifndef MCPI_SERVER_MODE
        misc_run_on_creative_inventory_setup(Inventory_setupDefault_FillingContainer_addItem_call_injection);
#endif

        // Use AuxDataTileItem by default instead of TileItem, so tiles in the Creative
        // Inventory can have arbitrary auxiliary values.
        {
            // Fix Size
            unsigned char size_patch[4] = {AUX_DATA_TILE_ITEM_SIZE, 0x00, 0xa0, 0xe3}; // "mov r0, #AUX_DATA_TILE_ITEM_SIZE"
            patch((void *) 0xc6f64, size_patch);
            // Hook Constructor
            overwrite_call((void *) 0xc6f74, (void *) Tile_initTiles_TileItem_injection);
        }
    }

    // Remove Creative Mode Restrictions (Opening Chests, Crafting, Etc)
    unsigned char nop_patch[4] = {0x00, 0xf0, 0x20, 0xe3}; // "nop"
    if (feature_has("Remove Creative Mode Restrictions", server_enabled)) {
        // Remove Restrictions
        patch((void *) 0x43ee8, nop_patch);
        patch((void *) 0x43f3c, nop_patch);
        patch((void *) 0x43f8c, nop_patch);
        patch((void *) 0x43fd8, nop_patch);
        patch((void *) 0x99010, nop_patch);
        // Allow Nether Reactor
        patch((void *) 0xc0290, nop_patch);
        // Disable Other Restrictions
        is_restricted = 0;
    }

    // Inventory Behavior
    if (feature_has("Force Survival Mode Inventory Behavior", server_enabled)) {
        patch((void *) 0x8d080, nop_patch); // Inventory::add
        patch((void *) 0x92828, nop_patch); // FillingContainer::add
        patch((void *) 0x91d48, nop_patch); // FillingContainer::hasResource
        patch((void *) 0x92098, nop_patch); // FillingContainer::removeResource(int)
        unsigned char inv_creative_check_r3_patch[4] = {0x03, 0x00, 0x53, 0xe1}; // "cmp r3, r3"
        patch((void *) 0x923c0, inv_creative_check_r3_patch); // FillingContainer::removeResource(ItemInstance const&, bool)
    }

    // "Craft" And "Armor" Buttons
    if (feature_has("Force Survival Mode Inventory UI", server_enabled)) {
        patch((void *) 0x341c0, nop_patch); // Add "Armor" Button To Classic Inventory
        unsigned char inv_creative_check_r5_patch[4] = {0x05, 0x00, 0x55, 0xe1}; // "cmp r5, r5"
        patch((void *) 0x3adb0, inv_creative_check_r5_patch); // Reposition "Select blocks" In Touch Inventory
        patch((void *) 0x3b374, nop_patch); // Add "Armor" And "Craft" Buttons To Touch Inventory
    }

    // Display Slot Count
    if (feature_has("Display Slot Count In Creative Mode", server_enabled)) {
        patch((void *) 0x1e3f4, nop_patch);
        unsigned char slot_count_patch[4] = {0x18, 0x00, 0x00, 0xea}; // "b 0x27110"
        patch((void *) 0x270a8, slot_count_patch);
        patch((void *) 0x33954, nop_patch);
    }

    // Maximize Creative Inventory Stack Size
    if (feature_has("Maximize Creative Mode Inventory Stack Size", server_enabled)) {
        unsigned char maximize_stack_patch[4] = {0xff, 0xc0, 0xa0, 0xe3}; // "mov r12, 0xff"
        patch((void *) 0x8e104, maximize_stack_patch);
    }
}