#pragma once

#include "modules/meta/event_module.hpp"
#include "utils/bspwm.hpp"

POLYBAR_NS

namespace modules {
  class bspwm_module : public event_module<bspwm_module> {
   public:
    enum class state {
      NONE = 0U,
      EMPTY,
      OCCUPIED,
      FOCUSED,
      URGENT,
      DIMMED,  // used when the monitor is out of focus
    };

    enum class mode {
      NONE = 0U,
      LAYOUT_MONOCLE,
      LAYOUT_TILED,
      STATE_FULLSCREEN,
      STATE_FLOATING,
      NODE_LOCKED,
      NODE_STICKY,
      NODE_PRIVATE
    };

    struct bspwm_monitor {
      vector<pair<uint32_t, label_t>> workspaces;
      vector<label_t> modes;
      label_t label;
      string name;
      bool focused{false};
    };

    using event_module::event_module;

    void setup();
    void stop();
    bool has_event();
    bool update();
    string get_output();
    bool build(builder* builder, const string& tag) const;
    bool handle_event(string cmd);
    bool receive_events() const {
      return true;
    }

   private:
    static constexpr auto DEFAULT_ICON = "ws-icon-default";
    static constexpr auto DEFAULT_LABEL = "%icon% %name%";
    static constexpr auto DEFAULT_MONITOR_LABEL = "%name%";

    static constexpr auto TAG_LABEL_MONITOR = "<label-monitor>";
    static constexpr auto TAG_LABEL_STATE = "<label-state>";
    static constexpr auto TAG_LABEL_MODE = "<label-mode>";

    static constexpr const char* EVENT_PREFIX{"bspwm-desk"};
    static constexpr const char* EVENT_CLICK{"bspwm-deskfocus"};
    static constexpr const char* EVENT_SCROLL_UP{"bspwm-desknext"};
    static constexpr const char* EVENT_SCROLL_DOWN{"bspwm-deskprev"};

    bspwm_util::connection_t m_subscriber;

    vector<unique_ptr<bspwm_monitor>> m_monitors;

    map<mode, label_t> m_modelabels;
    map<uint32_t, label_t> m_statelabels;
    label_t m_monitorlabel;
    iconset_t m_icons;

    bool m_click{true};
    bool m_scroll{true};
    bool m_pinworkspaces{true};
    string_util::hash_type m_hash{0U};

    // used while formatting output
    size_t m_index{0U};
  };
}

POLYBAR_NS_END