Waybar/src/modules/workspaces.cpp

94 lines
2.8 KiB
C++
Raw Normal View History

2018-08-08 21:54:58 +00:00
#include "modules/workspaces.hpp"
#include "ipc/client.hpp"
2018-08-09 10:05:48 +00:00
waybar::modules::Workspaces::Workspaces(Bar &bar)
2018-08-10 21:21:21 +00:00
: _bar(bar)
2018-08-08 21:54:58 +00:00
{
2018-08-10 21:21:21 +00:00
_box.get_style_context()->add_class("workspaces");
2018-08-08 21:54:58 +00:00
std::string socketPath = get_socketpath();
_ipcSocketfd = ipc_open_socket(socketPath);
_ipcEventSocketfd = ipc_open_socket(socketPath);
const char *subscribe = "[ \"workspace\", \"mode\" ]";
uint32_t len = strlen(subscribe);
ipc_single_command(_ipcEventSocketfd, IPC_SUBSCRIBE, subscribe, &len);
2018-08-10 21:21:21 +00:00
_thread = [this] {
Glib::signal_idle().connect_once([this] {
update();
});
_thread.sleep_for(chrono::milliseconds(250));
2018-08-09 18:22:01 +00:00
};
2018-08-08 21:54:58 +00:00
}
2018-08-10 21:21:21 +00:00
2018-08-09 10:05:48 +00:00
auto waybar::modules::Workspaces::update() -> void
2018-08-08 21:54:58 +00:00
{
Json::Value workspaces = _getWorkspaces();
2018-08-10 16:02:12 +00:00
bool needReorder = false;
2018-08-08 21:54:58 +00:00
for (auto it = _buttons.begin(); it != _buttons.end(); ++it) {
auto ws = std::find_if(workspaces.begin(), workspaces.end(),
[it](auto node) -> bool { return node["num"].asInt() == it->first; });
if (ws == workspaces.end()) {
2018-08-10 21:21:21 +00:00
it = _buttons.erase(it);
2018-08-10 16:02:12 +00:00
needReorder = true;
2018-08-08 21:54:58 +00:00
}
}
for (auto node : workspaces) {
auto it = _buttons.find(node["num"].asInt());
if (it == _buttons.end()) {
_addWorkspace(node);
2018-08-10 16:02:12 +00:00
needReorder = true;
2018-08-08 21:54:58 +00:00
} else {
auto styleContext = it->second.get_style_context();
bool isCurrent = node["focused"].asBool();
if (!isCurrent) {
2018-08-08 21:54:58 +00:00
styleContext->remove_class("current");
} else if (isCurrent) {
2018-08-08 21:54:58 +00:00
styleContext->add_class("current");
}
2018-08-10 16:02:12 +00:00
if (needReorder)
2018-08-10 21:21:21 +00:00
_box.reorder_child(it->second, node["num"].asInt() - 1);
2018-08-08 21:54:58 +00:00
it->second.show();
}
}
}
2018-08-09 10:05:48 +00:00
void waybar::modules::Workspaces::_addWorkspace(Json::Value node)
2018-08-08 21:54:58 +00:00
{
auto pair = _buttons.emplace(node["num"].asInt(), node["name"].asString());
auto &button = pair.first->second;
2018-08-10 21:21:21 +00:00
_box.pack_start(button, false, false, 0);
2018-08-08 21:54:58 +00:00
button.set_relief(Gtk::RELIEF_NONE);
button.signal_clicked().connect([this, pair] {
auto value = fmt::format("workspace \"{}\"", pair.first->first);
uint32_t size = value.size();
ipc_single_command(_ipcSocketfd, IPC_COMMAND, value.c_str(), &size);
});
2018-08-10 21:21:21 +00:00
_box.reorder_child(button, node["num"].asInt() - 1);
2018-08-08 21:54:58 +00:00
if (node["focused"].asBool()) {
button.get_style_context()->add_class("current");
}
button.show();
}
2018-08-09 10:05:48 +00:00
Json::Value waybar::modules::Workspaces::_getWorkspaces()
2018-08-08 21:54:58 +00:00
{
uint32_t len = 0;
Json::Value root;
Json::CharReaderBuilder builder;
Json::CharReader* reader = builder.newCharReader();
std::string err;
std::string str = ipc_single_command(_ipcSocketfd, IPC_GET_WORKSPACES,
nullptr, &len);
bool res = reader->parse(str.c_str(), str.c_str() + str.size(), &root, &err);
delete reader;
if (!res) {
std::cerr << err << std::endl;
return nullptr;
}
return root;
}
2018-08-09 10:05:48 +00:00
waybar::modules::Workspaces::operator Gtk::Widget &() {
2018-08-10 21:21:21 +00:00
return _box;
2018-08-08 21:54:58 +00:00
}