Waybar/include/modules/wlr/workspaces.hpp

83 lines
2.3 KiB
C++

#pragma once
#include <fmt/format.h>
#include <gtkmm/image.h>
#include <memory>
#include <vector>
#include "AModule.hpp"
#include "bar.hpp"
#include "wlr-workspace-unstable-v1-client-protocol.h"
namespace waybar::modules::wlr {
class WorkspaceManager;
class WorkspaceGroup;
class Workspace {
public:
Workspace(const waybar::Bar &bar, const Json::Value &config, WorkspaceGroup &workspace_group,
zwlr_workspace_handle_v1 &workspace);
~Workspace() = default;
auto update() -> void;
enum class State { ACTIVE = 1 << 1 };
private:
const Bar & bar_;
const Gtk::Box box_;
const Gtk::Image icon_;
const Json::Value & config_;
const WorkspaceGroup &workspace_group_;
// wlr stuff
zwlr_workspace_handle_v1 &workspace_handle_;
uint32_t state_;
};
class WorkspaceGroup {
public:
WorkspaceGroup(const waybar::Bar &bar, const Json::Value &config, WorkspaceManager &manager,
zwlr_workspace_group_handle_v1 &workspace_group_handle);
auto update() -> void;
// wlr stuff
auto handle_workspace_create(zwlr_workspace_handle_v1 &workspace_handle) -> void;
auto handle_remove() -> void;
auto handle_output_enter(wl_output &output) -> void;
auto handle_output_leave(wl_output &output) -> void;
private:
const waybar::Bar & bar_;
const Json::Value & config_;
const WorkspaceManager & workspace_manager_;
std::vector<std::unique_ptr<Workspace>> workspaces_;
// wlr stuff
zwlr_workspace_group_handle_v1 &workspace_group_handle_;
};
class WorkspaceManager : public AModule {
public:
WorkspaceManager(const std::string &id, const waybar::Bar &bar, const Json::Value &config);
auto update() -> void override;
// wlr stuff
auto register_manager(wl_registry *registry, uint32_t name, uint32_t version) -> void;
auto handle_workspace_group_create(zwlr_workspace_group_handle_v1 *workspace_group_handle)
-> void;
auto handle_done() -> void;
auto handle_finished() -> void;
private:
const waybar::Bar & bar_;
Gtk::Box box_;
std::vector<std::unique_ptr<WorkspaceGroup>> groups_;
// wlr stuff
zwlr_workspace_manager_v1 *workspace_manager_ = nullptr;
};
} // namespace waybar::modules::wlr