polybar/include/components/controller.hpp

#pragma once

#include <moodycamel/blockingconcurrentqueue.h>
#include <thread>

#include "common.hpp"
#include "events/signal_fwd.hpp"
#include "events/signal_receiver.hpp"
#include "events/types.hpp"
#include "utils/file.hpp"
#include "x11/types.hpp"
#include "x11/events.hpp"

POLYBAR_NS

// fwd decl {{{

enum class alignment : uint8_t;
class bar;
class command;
class config;
class connection;
class inotify_watch;
class ipc;
class logger;
class signal_emitter;

namespace modules {
  struct module_interface;
  class input_handler;
}

using module_t = unique_ptr<modules::module_interface>;
using modulemap_t = std::map<alignment, vector<module_t>>;

// }}}

using std::thread;
namespace chrono = std::chrono;
using namespace std::chrono_literals;

namespace sig_ev = signals::eventqueue;
namespace sig_ui = signals::ui;
namespace sig_ipc = signals::ipc;

class controller : public signal_receiver<SIGN_PRIORITY_CONTROLLER, sig_ev::exit_terminate, sig_ev::exit_reload,
                       sig_ev::update, sig_ev::notify_change, sig_ev::check_state, sig_ipc::action, sig_ipc::command,
                       sig_ipc::hook, sig_ui::button_press> {
 public:
  using make_type = unique_ptr<controller>;
  static make_type make(unique_ptr<ipc>&& ipc, unique_ptr<inotify_watch>&& config_watch);

  explicit controller(connection&, signal_emitter&, const logger&, const config&, unique_ptr<bar>&&, unique_ptr<ipc>&&,
      unique_ptr<inotify_watch>&&);
  ~controller();

  bool run(bool writeback = false);

  bool enqueue(event&& evt);
  bool enqueue(string&& input_data);

 protected:
  void read_events();
  void process_eventqueue();
  void process_inputdata();

  bool on(const sig_ev::notify_change& evt);
  bool on(const sig_ev::update& evt);
  bool on(const sig_ev::exit_terminate& evt);
  bool on(const sig_ev::exit_reload& evt);
  bool on(const sig_ev::check_state& evt);
  bool on(const sig_ui::button_press& evt);
  bool on(const sig_ipc::action& evt);
  bool on(const sig_ipc::command& evt);
  bool on(const sig_ipc::hook& evt);

 private:
  connection& m_connection;
  signal_emitter& m_sig;
  const logger& m_log;
  const config& m_conf;
  unique_ptr<bar> m_bar;
  unique_ptr<ipc> m_ipc;
  unique_ptr<inotify_watch> m_confwatch;
  unique_ptr<command> m_command;

  array<unique_ptr<file_descriptor>, 2> m_queuefd{};

  /**
   * @brief Controls weather the output gets printed to stdout
   */
  bool m_writeback{false};

  /**
   * @brief Internal event queue
   */
  using queue_t = moodycamel::BlockingConcurrentQueue<event>;
  queue_t m_queue;

  /**
   * @brief Loaded modules
   */
  modulemap_t m_modules;

  /**
   * @brief Module input handlers
   */
  vector<modules::input_handler*> m_inputhandlers;

  /**
   * @brief Maximum number of subsequent events to swallow
   */
  size_t m_swallow_limit{5U};

  /**
   * @brief Time to wait for subsequent events
   */
  chrono::milliseconds m_swallow_update{10ms};

  /**
   * @brief Time to throttle input events
   */
  chrono::milliseconds m_swallow_input{30ms};

  /**
   * @brief Time of last handled input event
   */
  chrono::time_point<chrono::system_clock, chrono::milliseconds> m_lastinput;

  /**
   * @brief Input data
   */
  string m_inputdata;

  /**
   * @brief Thread for the eventqueue loop
   */
  thread m_event_thread;
};

POLYBAR_NS_END