#pragma once

#include <AK/AKString.h>
#include <AK/Function.h>
#include <AK/Retainable.h>
#include <AK/Retained.h>
#include <AK/Weakable.h>
#include <AK/Badge.h>
#include <AK/HashTable.h>
#include <SharedGraphics/GraphicsBitmap.h>
#include <LibGUI/GShortcut.h>

class GButton;
class GMenuItem;

class GAction : public Retainable<GAction>, public Weakable<GAction> {
public:
    static Retained<GAction> create(const String& text, Function<void(const GAction&)> callback)
    {
        return adopt(*new GAction(text, move(callback)));
    }
    static Retained<GAction> create(const String& text, const String& custom_data, Function<void(const GAction&)> callback)
    {
        return adopt(*new GAction(text, custom_data, move(callback)));
    }
    static Retained<GAction> create(const String& text, RetainPtr<GraphicsBitmap>&& icon, Function<void(const GAction&)> callback)
    {
        return adopt(*new GAction(text, move(icon), move(callback)));
    }
    static Retained<GAction> create(const String& text, const GShortcut& shortcut, Function<void(const GAction&)> callback)
    {
        return adopt(*new GAction(text, shortcut, move(callback)));
    }
    static Retained<GAction> create(const String& text, const GShortcut& shortcut, RetainPtr<GraphicsBitmap>&& icon, Function<void(const GAction&)> callback)
    {
        return adopt(*new GAction(text, shortcut, move(icon), move(callback)));
    }
    ~GAction();

    String text() const { return m_text; }
    GShortcut shortcut() const { return m_shortcut; }
    String custom_data() const { return m_custom_data; }
    const GraphicsBitmap* icon() const { return m_icon.ptr(); }

    Function<void(GAction&)> on_activation;

    void activate();

    bool is_enabled() const { return m_enabled; }
    void set_enabled(bool);

    void register_button(Badge<GButton>, GButton&);
    void unregister_button(Badge<GButton>, GButton&);
    void register_menu_item(Badge<GMenuItem>, GMenuItem&);
    void unregister_menu_item(Badge<GMenuItem>, GMenuItem&);

private:
    GAction(const String& text, Function<void(const GAction&)> = nullptr);
    GAction(const String& text, const GShortcut&, Function<void(const GAction&)> = nullptr);
    GAction(const String& text, const GShortcut&, RetainPtr<GraphicsBitmap>&& icon, Function<void(const GAction&)> = nullptr);
    GAction(const String& text, RetainPtr<GraphicsBitmap>&& icon, Function<void(const GAction&)> = nullptr);
    GAction(const String& text, const String& custom_data = String(), Function<void(const GAction&)> = nullptr);

    template<typename Callback> void for_each_toolbar_button(Callback);
    template<typename Callback> void for_each_menu_item(Callback);

    String m_text;
    String m_custom_data;
    RetainPtr<GraphicsBitmap> m_icon;
    GShortcut m_shortcut;
    bool m_enabled { true };

    HashTable<GButton*> m_buttons;
    HashTable<GMenuItem*> m_menu_items;
};