#ifndef FRONTEND_INVENTORY_VIEW_H_
#define FRONTEND_INVENTORY_VIEW_H_

#include <vector>
#include <functional>
#include <glm/glm.hpp>

#include "../frontend/gui/UINode.h"
#include "../frontend/gui/containers.h"
#include "../frontend/gui/controls.h"
#include "../items/ItemStack.h"
#include "../typedefs.h"

class Assets;
class GfxContext;
class Content;
class ContentIndices;
class LevelFrontend;
class Inventory;

namespace gui {
    class UiXmlReader;
}

namespace scripting {
    class Environment;
}

using slotcallback = std::function<void(uint, ItemStack&)>;

class InventoryInteraction {
    ItemStack grabbedItem;
public:
    InventoryInteraction() = default;

    ItemStack& getGrabbedItem() {
        return grabbedItem;
    }
};

struct SlotLayout {
    int index;
    glm::vec2 position;
    bool background;
    bool itemSource;
    slotcallback shareFunc;
    slotcallback rightClick;
    int padding = 0;

    SlotLayout(int index,
               glm::vec2 position, 
               bool background,
               bool itemSource,
               slotcallback shareFunc,
               slotcallback rightClick);
};

class SlotView : public gui::UINode {
    LevelFrontend* frontend = nullptr;
    InventoryInteraction* interaction = nullptr;
    const Content* content;
    SlotLayout layout;
    bool highlighted = false;

    int64_t inventoryid = 0;
    ItemStack* bound = nullptr;
public:
    SlotView(SlotLayout layout);

    virtual void draw(const GfxContext* pctx, Assets* assets) override;

    void setHighlighted(bool flag);
    bool isHighlighted() const;

    virtual void clicked(gui::GUI*, int) override;
    virtual void focus(gui::GUI*) override;

    void bind(
        int64_t inventoryid,
        ItemStack& stack,
        LevelFrontend* frontend, 
        InventoryInteraction* interaction
    );

    const SlotLayout& getLayout() const;
};

class InventoryView : public gui::Container {
    const Content* content;
    const ContentIndices* indices;
    
    std::shared_ptr<Inventory> inventory;
    LevelFrontend* frontend = nullptr;
    InventoryInteraction* interaction = nullptr;

    std::vector<SlotView*> slots;
    glm::vec2 origin {};
public:
    InventoryView();
    virtual ~InventoryView();

    void setInventory(std::shared_ptr<Inventory> inventory);

    virtual void setCoord(glm::vec2 coord) override;

    void setOrigin(glm::vec2 origin);
    glm::vec2 getOrigin() const;

    void setSelected(int index);

    void bind(
        std::shared_ptr<Inventory> inventory,
        LevelFrontend* frontend, 
        InventoryInteraction* interaction
    );

    std::shared_ptr<SlotView> addSlot(SlotLayout layout);

    std::shared_ptr<Inventory> getInventory() const;

    size_t getSlotsCount() const;

    static void createReaders(gui::UiXmlReader& reader);

    static const int SLOT_INTERVAL = 4;
    static const int SLOT_SIZE = ITEM_ICON_SIZE;
};

class InventoryBuilder {
    std::shared_ptr<InventoryView> view;
public:
    InventoryBuilder();

    void addGrid(
        int cols, int count, 
        glm::vec2 coord, 
        int padding,
        bool addpanel,
        SlotLayout slotLayout
    );
    
    void add(SlotLayout slotLayout);
    std::shared_ptr<InventoryView> build();
};

#endif // FRONTEND_INVENTORY_VIEW_H_