#pragma once

#include <giomm/menu.h>
#include <giomm/menuitem.h>
#include <giomm/menumodel.h>
#include <giomm/simpleaction.h>
#include <giomm/simpleactiongroup.h>
#include <gtkmm/box.h>
#include <gtkmm/button.h>
#include <gtkmm/gestureclick.h>
#include <gtkmm/image.h>
#include <gtkmm/picture.h>
#include <gtkmm/popovermenu.h>

#include <map>
#include <memory>
#include <string>

#include "services/tray.hpp"

class TrayIconWidget : public Gtk::Button {
  public:
    TrayIconWidget(TrayService &service, std::string id);

    void update(const TrayService::Item &item);

  private:
    TrayService &service;
    std::string id;
    Gtk::Box container;
    Gtk::Picture picture;
    Gtk::Image image;
    Glib::RefPtr<Gtk::GestureClick> primaryGesture;
    Glib::RefPtr<Gtk::GestureClick> secondaryGesture;
    Glib::RefPtr<Gtk::PopoverMenu> menuPopover;
    Glib::RefPtr<Gio::SimpleActionGroup> menuActions;
    Glib::RefPtr<Gio::MenuModel> menuModel;
    sigc::connection menuChangedConnection;
    bool menuPopupPending = false;
    double pendingX = 0.0;
    double pendingY = 0.0;

    void on_primary_released(int n_press, double x, double y);
    void on_secondary_released(int n_press, double x, double y);
    bool ensure_menu();
    void on_menu_items_changed(guint position, guint removed, guint added);
    void try_popup();
    void
    populate_menu_items(const std::vector<TrayService::MenuNode> &nodes,
                        const Glib::RefPtr<Gio::Menu> &menu,
                        const Glib::RefPtr<Gio::SimpleActionGroup> &actions);
    void on_menu_action(const Glib::VariantBase &parameter, int itemId);
};

class TrayWidget : public Gtk::Box {
  public:
    explicit TrayWidget(TrayService &service);
    ~TrayWidget() override;

  private:
    TrayService &service;
    std::map<std::string, std::unique_ptr<TrayIconWidget>> icons;

    sigc::connection addConnection;
    sigc::connection removeConnection;
    sigc::connection updateConnection;

    void on_item_added(const TrayService::Item &item);
    void on_item_removed(const std::string &id);
    void on_item_updated(const TrayService::Item &item);

    void rebuild_existing();
};