polybar/include/registry.hpp

#ifndef _REGISTRY_HPP_
#define _REGISTRY_HPP_

#include <condition_variable>

#include "exception.hpp"
#include "modules/base.hpp"

DefineBaseException(RegistryError);
DefineChildException(ModuleNotFound, RegistryError);

struct ModuleEntry
{
  concurrency::Atomic<bool> warmedup;
  std::unique_ptr<modules::ModuleInterface> module;

  ModuleEntry(std::unique_ptr<modules::ModuleInterface> &&module)
  {
    this->warmedup = false;
    this->module.swap(module);
  }
};

class Registry
{
  // Stopped and no loaded modules
  const int STAGE_1 = 1;
  // Modules loaded but waiting for initial broadcast
  const int STAGE_2 = 2;
  // Running
  const int STAGE_3 = 3;
  // Unloading modules
  const int STAGE_4 = 4;

  concurrency::Atomic<int> stage;

  std::vector<std::unique_ptr<ModuleEntry>> modules;

  std::mutex wait_mtx;
  std::condition_variable wait_cv;

  public:
    Registry();

    bool ready();
    void insert(std::unique_ptr<modules::ModuleInterface> &&module);
    void load();
    void unload();
    bool wait();
    void notify(const std::string& module_name);
    std::string get(const std::string& module_name);
    std::unique_ptr<ModuleEntry>& find(const std::string& module_name) throw(ModuleNotFound);
};

std::shared_ptr<Registry> &get_registry();

#endif