mirror of
				https://github.com/rad4day/Waybar.git
				synced 2025-11-04 09:42:42 +01:00 
			
		
		
		
	Finish base implementation
This commit is contained in:
		@@ -21,29 +21,31 @@ class WorkspaceGroup;
 | 
			
		||||
class Workspace {
 | 
			
		||||
 public:
 | 
			
		||||
  Workspace(const waybar::Bar &bar, const Json::Value &config, WorkspaceGroup &workspace_group,
 | 
			
		||||
            zwlr_workspace_handle_v1 *workspace);
 | 
			
		||||
            zwlr_workspace_handle_v1 *workspace, uint32_t id);
 | 
			
		||||
  ~Workspace();
 | 
			
		||||
  auto update() -> void;
 | 
			
		||||
 | 
			
		||||
  auto id() const -> uint32_t { return id_; }
 | 
			
		||||
  auto is_active() const -> bool { return state_ & static_cast<uint32_t>(State::ACTIVE); }
 | 
			
		||||
  // wlr stuff
 | 
			
		||||
  auto handle_name(const std::string &name) -> void { name_ = name; }
 | 
			
		||||
  auto handle_coordinates(const std::vector<uint32_t> &coordinates) -> void {
 | 
			
		||||
    coordinates_ = coordinates;
 | 
			
		||||
  }
 | 
			
		||||
  auto handle_name(const std::string &name) -> void;
 | 
			
		||||
  auto handle_coordinates(const std::vector<uint32_t> &coordinates) -> void;
 | 
			
		||||
  auto handle_state(const std::vector<uint32_t> &state) -> void;
 | 
			
		||||
  auto handle_remove() -> void;
 | 
			
		||||
 | 
			
		||||
  auto handle_done() -> void;
 | 
			
		||||
  auto handle_clicked() -> void;
 | 
			
		||||
  auto show() -> void { button_.show(); }
 | 
			
		||||
  auto hide() -> void { button_.hide(); }
 | 
			
		||||
  auto get_button_ref() -> Gtk::Button & { return button_; }
 | 
			
		||||
  auto get_name() -> std::string & { return name_; }
 | 
			
		||||
  auto get_coords() -> std::vector<uint32_t> & { return coordinates_; }
 | 
			
		||||
 | 
			
		||||
  enum class State { ACTIVE = 1 << 0 };
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  auto get_icon() -> std::string;
 | 
			
		||||
 | 
			
		||||
  static uint32_t    workspace_global_id;
 | 
			
		||||
  const Bar &        bar_;
 | 
			
		||||
  const Json::Value &config_;
 | 
			
		||||
  WorkspaceGroup &   workspace_group_;
 | 
			
		||||
@@ -66,8 +68,9 @@ class Workspace {
 | 
			
		||||
 | 
			
		||||
class WorkspaceGroup {
 | 
			
		||||
 public:
 | 
			
		||||
  WorkspaceGroup(const waybar::Bar &bar, const Json::Value &config, WorkspaceManager &manager,
 | 
			
		||||
                 zwlr_workspace_group_handle_v1 *workspace_group_handle);
 | 
			
		||||
  WorkspaceGroup(const waybar::Bar &bar, Gtk::Box &box, const Json::Value &config,
 | 
			
		||||
                 WorkspaceManager &manager, zwlr_workspace_group_handle_v1 *workspace_group_handle,
 | 
			
		||||
                 uint32_t id);
 | 
			
		||||
  ~WorkspaceGroup();
 | 
			
		||||
  auto update() -> void;
 | 
			
		||||
 | 
			
		||||
@@ -84,10 +87,12 @@ class WorkspaceGroup {
 | 
			
		||||
  auto add_button(Gtk::Button &button) -> void;
 | 
			
		||||
  auto handle_done() -> void;
 | 
			
		||||
  auto commit() -> void;
 | 
			
		||||
  auto sort_workspaces() -> void;
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
  static uint32_t    group_global_id;
 | 
			
		||||
  static uint32_t    workspace_global_id;
 | 
			
		||||
  const waybar::Bar &bar_;
 | 
			
		||||
  Gtk::Box &         box_;
 | 
			
		||||
  const Json::Value &config_;
 | 
			
		||||
  WorkspaceManager & workspace_manager_;
 | 
			
		||||
 | 
			
		||||
@@ -97,6 +102,8 @@ class WorkspaceGroup {
 | 
			
		||||
 | 
			
		||||
  uint32_t                                id_;
 | 
			
		||||
  std::vector<std::unique_ptr<Workspace>> workspaces_;
 | 
			
		||||
  bool                                    sort_by_name = true;
 | 
			
		||||
  bool                                    sort_by_coordinates = true;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
class WorkspaceManager : public AModule {
 | 
			
		||||
@@ -114,7 +121,6 @@ class WorkspaceManager : public AModule {
 | 
			
		||||
  auto handle_done() -> void;
 | 
			
		||||
  auto handle_finished() -> void;
 | 
			
		||||
 | 
			
		||||
  auto add_button(Gtk::Button &button) -> void { box_.pack_start(button, false, false); }
 | 
			
		||||
  auto commit() -> void;
 | 
			
		||||
 | 
			
		||||
 private:
 | 
			
		||||
@@ -124,6 +130,8 @@ class WorkspaceManager : public AModule {
 | 
			
		||||
 | 
			
		||||
  // wlr stuff
 | 
			
		||||
  zwlr_workspace_manager_v1 *workspace_manager_ = nullptr;
 | 
			
		||||
 | 
			
		||||
  static uint32_t group_global_id;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}  // namespace waybar::modules::wlr
 | 
			
		||||
 
 | 
			
		||||
@@ -9,8 +9,8 @@
 | 
			
		||||
 | 
			
		||||
namespace waybar::modules::wlr {
 | 
			
		||||
 | 
			
		||||
uint32_t                           Workspace::workspace_global_id = 0;
 | 
			
		||||
uint32_t                           WorkspaceGroup::group_global_id = 0;
 | 
			
		||||
uint32_t                           WorkspaceGroup::workspace_global_id = 0;
 | 
			
		||||
uint32_t                           WorkspaceManager::group_global_id = 0;
 | 
			
		||||
std::map<std::string, std::string> Workspace::icons_map_;
 | 
			
		||||
 | 
			
		||||
WorkspaceManager::WorkspaceManager(const std::string &id, const waybar::Bar &bar,
 | 
			
		||||
@@ -44,8 +44,10 @@ auto WorkspaceManager::register_manager(wl_registry *registry, uint32_t name, ui
 | 
			
		||||
 | 
			
		||||
auto WorkspaceManager::handle_workspace_group_create(
 | 
			
		||||
    zwlr_workspace_group_handle_v1 *workspace_group_handle) -> void {
 | 
			
		||||
  groups_.push_back(std::make_unique<WorkspaceGroup>(bar_, config_, *this, workspace_group_handle));
 | 
			
		||||
  spdlog::debug("Workspace group {} created", groups_.back()->id());
 | 
			
		||||
  auto new_id = ++group_global_id;
 | 
			
		||||
  groups_.push_back(
 | 
			
		||||
      std::make_unique<WorkspaceGroup>(bar_, box_, config_, *this, workspace_group_handle, new_id));
 | 
			
		||||
  spdlog::debug("Workspace group {} created", new_id);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceManager::handle_finished() -> void {
 | 
			
		||||
@@ -89,17 +91,28 @@ auto WorkspaceManager::remove_workspace_group(uint32_t id) -> void {
 | 
			
		||||
}
 | 
			
		||||
auto WorkspaceManager::commit() -> void { zwlr_workspace_manager_v1_commit(workspace_manager_); }
 | 
			
		||||
 | 
			
		||||
WorkspaceGroup::WorkspaceGroup(const Bar &bar, const Json::Value &config, WorkspaceManager &manager,
 | 
			
		||||
                               zwlr_workspace_group_handle_v1 *workspace_group_handle)
 | 
			
		||||
WorkspaceGroup::WorkspaceGroup(const Bar &bar, Gtk::Box &box, const Json::Value &config,
 | 
			
		||||
                               WorkspaceManager &              manager,
 | 
			
		||||
                               zwlr_workspace_group_handle_v1 *workspace_group_handle, uint32_t id)
 | 
			
		||||
    : bar_(bar),
 | 
			
		||||
      box_(box),
 | 
			
		||||
      config_(config),
 | 
			
		||||
      workspace_manager_(manager),
 | 
			
		||||
      workspace_group_handle_(workspace_group_handle),
 | 
			
		||||
      id_(++group_global_id) {
 | 
			
		||||
      id_(id) {
 | 
			
		||||
  add_workspace_group_listener(workspace_group_handle, this);
 | 
			
		||||
  auto config_sort_by_name = config_["sort_by_name"];
 | 
			
		||||
  if (config_sort_by_name.isBool()) {
 | 
			
		||||
    sort_by_name = config_sort_by_name.asBool();
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  auto config_sort_by_coordinates = config_["sort_by_coordinates"];
 | 
			
		||||
  if (config_sort_by_coordinates.isBool()) {
 | 
			
		||||
    sort_by_coordinates = config_sort_by_coordinates.asBool();
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
auto WorkspaceGroup::add_button(Gtk::Button &button) -> void {
 | 
			
		||||
  workspace_manager_.add_button(button);
 | 
			
		||||
  box_.pack_start(button, false, false);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
WorkspaceGroup::~WorkspaceGroup() {
 | 
			
		||||
@@ -112,8 +125,9 @@ WorkspaceGroup::~WorkspaceGroup() {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::handle_workspace_create(zwlr_workspace_handle_v1 *workspace) -> void {
 | 
			
		||||
  workspaces_.push_back(std::make_unique<Workspace>(bar_, config_, *this, workspace));
 | 
			
		||||
  spdlog::debug("Workspace {} created", workspaces_.back()->id());
 | 
			
		||||
  auto new_id = ++workspace_global_id;
 | 
			
		||||
  workspaces_.push_back(std::make_unique<Workspace>(bar_, config_, *this, workspace, new_id));
 | 
			
		||||
  spdlog::debug("Workspace {} created", new_id);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::handle_remove() -> void {
 | 
			
		||||
@@ -124,12 +138,18 @@ auto WorkspaceGroup::handle_remove() -> void {
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::handle_output_enter(wl_output *output) -> void {
 | 
			
		||||
  spdlog::debug("Output {} assigned to {} group", (void *)output, id_);
 | 
			
		||||
  for (auto &workspace : workspaces_) {
 | 
			
		||||
    workspace->show();
 | 
			
		||||
  }
 | 
			
		||||
  output_ = output;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::handle_output_leave() -> void {
 | 
			
		||||
  spdlog::debug("Output {} remove from {} group", (void *)output_, id_);
 | 
			
		||||
  output_ = nullptr;
 | 
			
		||||
  for (auto &workspace : workspaces_) {
 | 
			
		||||
    workspace->hide();
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::update() -> void {
 | 
			
		||||
@@ -150,22 +170,37 @@ auto WorkspaceGroup::remove_workspace(uint32_t id) -> void {
 | 
			
		||||
 | 
			
		||||
  workspaces_.erase(it);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::handle_done() -> void {
 | 
			
		||||
  if (is_visible()) {
 | 
			
		||||
    for (auto &workspace : workspaces_) {
 | 
			
		||||
      workspace->handle_done();
 | 
			
		||||
    }
 | 
			
		||||
  for (auto &workspace : workspaces_) {
 | 
			
		||||
    workspace->handle_done();
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
auto WorkspaceGroup::commit() -> void { workspace_manager_.commit(); }
 | 
			
		||||
 | 
			
		||||
auto WorkspaceGroup::sort_workspaces() -> void {
 | 
			
		||||
  auto cmp = [=](std::unique_ptr<Workspace> &lhs, std::unique_ptr<Workspace> &rhs) {
 | 
			
		||||
    if (sort_by_name && lhs->get_name() != rhs->get_name()) {
 | 
			
		||||
      return lhs->get_name() < rhs->get_name();
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return lhs->get_coords() < rhs->get_coords();
 | 
			
		||||
  };
 | 
			
		||||
  std::sort(workspaces_.begin(), workspaces_.end(), cmp);
 | 
			
		||||
  for (size_t i = 0; i < workspaces_.size(); ++i) {
 | 
			
		||||
    for (auto &workspace : workspaces_) {
 | 
			
		||||
      box_.reorder_child(workspace->get_button_ref(), i);
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
Workspace::Workspace(const Bar &bar, const Json::Value &config, WorkspaceGroup &workspace_group,
 | 
			
		||||
                     zwlr_workspace_handle_v1 *workspace)
 | 
			
		||||
                     zwlr_workspace_handle_v1 *workspace, uint32_t id)
 | 
			
		||||
    : bar_(bar),
 | 
			
		||||
      config_(config),
 | 
			
		||||
      workspace_group_(workspace_group),
 | 
			
		||||
      workspace_handle_(workspace),
 | 
			
		||||
      id_(++workspace_global_id) {
 | 
			
		||||
      id_(id) {
 | 
			
		||||
  add_workspace_listener(workspace, this);
 | 
			
		||||
 | 
			
		||||
  auto config_format = config["format"];
 | 
			
		||||
@@ -186,6 +221,10 @@ Workspace::Workspace(const Bar &bar, const Json::Value &config, WorkspaceGroup &
 | 
			
		||||
  button_.set_relief(Gtk::RELIEF_NONE);
 | 
			
		||||
  content_.set_center_widget(label_);
 | 
			
		||||
  button_.add(content_);
 | 
			
		||||
  if (!workspace_group.is_visible()) {
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  button_.show();
 | 
			
		||||
  label_.show();
 | 
			
		||||
  content_.show();
 | 
			
		||||
@@ -221,6 +260,7 @@ auto Workspace::handle_remove() -> void {
 | 
			
		||||
  workspace_handle_ = nullptr;
 | 
			
		||||
  workspace_group_.remove_workspace(id_);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto Workspace::handle_done() -> void {
 | 
			
		||||
  spdlog::debug("Workspace {} changed to state {}", id_, state_);
 | 
			
		||||
  auto style_context = button_.get_style_context();
 | 
			
		||||
@@ -230,6 +270,7 @@ auto Workspace::handle_done() -> void {
 | 
			
		||||
    style_context->remove_class("focused");
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto Workspace::get_icon() -> std::string {
 | 
			
		||||
  if (is_active()) {
 | 
			
		||||
    auto focused_icon_it = icons_map_.find("focused");
 | 
			
		||||
@@ -250,9 +291,20 @@ auto Workspace::get_icon() -> std::string {
 | 
			
		||||
 | 
			
		||||
  return name_;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto Workspace::handle_clicked() -> void {
 | 
			
		||||
  spdlog::debug("Workspace {} clicked", (void*)workspace_handle_);
 | 
			
		||||
  spdlog::debug("Workspace {} clicked", (void *)workspace_handle_);
 | 
			
		||||
  zwlr_workspace_handle_v1_activate(workspace_handle_);
 | 
			
		||||
  workspace_group_.commit();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto Workspace::handle_name(const std::string &name) -> void {
 | 
			
		||||
  name_ = name;
 | 
			
		||||
  workspace_group_.sort_workspaces();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
auto Workspace::handle_coordinates(const std::vector<uint32_t> &coordinates) -> void {
 | 
			
		||||
  coordinates_ = coordinates;
 | 
			
		||||
  workspace_group_.sort_workspaces();
 | 
			
		||||
}
 | 
			
		||||
}  // namespace waybar::modules::wlr
 | 
			
		||||
@@ -107,12 +107,18 @@ void workspace_handle_coordinates(void *data, struct zwlr_workspace_handle_v1 *_
 | 
			
		||||
  static_cast<Workspace *>(data)->handle_coordinates(coords_vec);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void workspace_handle_state(void *data, struct zwlr_workspace_handle_v1 *_,
 | 
			
		||||
void workspace_handle_state(void *data, struct zwlr_workspace_handle_v1 *workspace_handle,
 | 
			
		||||
                            struct wl_array *state) {
 | 
			
		||||
  std::vector<uint32_t> state_vec;
 | 
			
		||||
  auto                  coords = static_cast<uint32_t *>(state->data);
 | 
			
		||||
  auto                  states = static_cast<uint32_t *>(state->data);
 | 
			
		||||
  for (size_t i = 0; i < state->size; ++i) {
 | 
			
		||||
    state_vec.push_back(coords[i]);
 | 
			
		||||
    // To sync server and pending states
 | 
			
		||||
    if (states[i] == ZWLR_WORKSPACE_HANDLE_V1_STATE_ACTIVE) {
 | 
			
		||||
      zwlr_workspace_handle_v1_activate(workspace_handle);
 | 
			
		||||
    } else {
 | 
			
		||||
      zwlr_workspace_handle_v1_deactivate(workspace_handle);
 | 
			
		||||
    }
 | 
			
		||||
    state_vec.push_back(states[i]);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  static_cast<Workspace *>(data)->handle_state(state_vec);
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user