#pragma once

#include <dwmipcpp/connection.hpp>

#include "modules/meta/event_module.hpp"
#include "modules/meta/input_handler.hpp"

POLYBAR_NS

namespace modules {
  class dwm_module : public event_module<dwm_module>, public input_handler {
   public:
    explicit dwm_module(const bar_settings&, string);

    using tag_mask_t = unsigned int;
    using window_t = unsigned int;

    enum class state_t : uint8_t {
      FOCUSED,    ///< Monitor is selected and tag is selected, overrides all below
      URGENT,     ///< Tag is urgent, overrides all below
      UNFOCUSED,  ///< Monitor is not selected, but tag is selected
      VISIBLE,    ///< Tag is not selected, but occupied
      EMPTY       ///< Tag is unoccupied and unselected
    };

    struct tag_t {
      tag_t(string& name, tag_mask_t bit_mask, state_t state, label_t&& label)
          : name(name), bit_mask(bit_mask), state(state), label(forward<label_t>(label)) {}

      string name;
      tag_mask_t bit_mask;
      state_t state;
      label_t label;
    };

    auto stop() -> void override;
    auto has_event() -> bool;
    auto update() -> bool;
    auto build(builder* builder, const string& tag) const -> bool;

   protected:
    auto input(string&& cmd) -> bool override;

   private:
    static constexpr const char* DEFAULT_FORMAT_TAGS{"<label-state> <label-layout> <label-title>"};
    static constexpr const char* DEFAULT_STATE_LABEL{"%name%"};

    static constexpr const char* TAG_LABEL_STATE{"<label-state>"};
    static constexpr const char* TAG_LABEL_LAYOUT{"<label-layout>"};
    static constexpr const char* TAG_LABEL_TITLE{"<label-title>"};

    static constexpr const char* EVENT_PREFIX{"dwm-"};
    static constexpr const char* EVENT_LCLICK{"view"};
    static constexpr const char* EVENT_RCLICK{"toggleview"};

    void on_layout_change(const dwmipc::LayoutChangeEvent& ev);
    void on_monitor_focus_change(const dwmipc::MonitorFocusChangeEvent& ev);
    void on_tag_change(const dwmipc::TagChangeEvent& ev);
    void on_client_focus_change(const dwmipc::ClientFocusChangeEvent& ev);
    void on_focused_title_change(const dwmipc::FocusedTitleChangeEvent& ev);

    void update_monitor_ref();
    void update_tag_labels();
    void update_title_label(window_t client_id);

    auto get_state(tag_mask_t bit_mask) const -> state_t;
    auto check_send_cmd(string cmd, const string& ev_name) -> bool;
    auto reconnect_dwm() -> bool;

    bool m_click{true};

    const dwmipc::Monitor* m_active_mon = nullptr;
    const dwmipc::Monitor* m_bar_mon = nullptr;
    window_t m_focused_client_id = 0;

    label_t m_layout_label;
    label_t m_seperator_label;
    label_t m_title_label;

    unique_ptr<dwmipc::Connection> m_ipc;
    shared_ptr<std::vector<dwmipc::Monitor>> m_monitors;
    std::unordered_map<state_t, label_t> m_state_labels;
    vector<tag_t> m_tags;
  };
}  // namespace modules

POLYBAR_NS_END