264 lines
8.7 KiB
C++
264 lines
8.7 KiB
C++
#include "modules/sway/workspaces.hpp"
|
|
|
|
waybar::modules::sway::Workspaces::Workspaces(const std::string &id, const Bar &bar,
|
|
const Json::Value &config)
|
|
: bar_(bar),
|
|
config_(config),
|
|
box_(bar.vertical ? Gtk::ORIENTATION_VERTICAL : Gtk::ORIENTATION_HORIZONTAL, 0),
|
|
scrolling_(false) {
|
|
box_.set_name("workspaces");
|
|
if (!id.empty()) {
|
|
box_.get_style_context()->add_class(id);
|
|
}
|
|
ipc_.subscribe("[ \"workspace\" ]");
|
|
// Launch worker
|
|
worker();
|
|
}
|
|
|
|
void waybar::modules::sway::Workspaces::worker() {
|
|
thread_ = [this] {
|
|
try {
|
|
if (!workspaces_.empty()) {
|
|
ipc_.handleEvent();
|
|
}
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
auto res = ipc_.sendCmd(IPC_GET_WORKSPACES);
|
|
if (thread_.isRunning()) {
|
|
workspaces_ = parser_.parse(res.payload);
|
|
}
|
|
}
|
|
dp.emit();
|
|
} catch (const std::exception &e) {
|
|
std::cerr << "Workspaces: " << e.what() << std::endl;
|
|
}
|
|
};
|
|
}
|
|
|
|
auto waybar::modules::sway::Workspaces::update() -> void {
|
|
bool needReorder = false;
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
for (auto it = buttons_.begin(); it != buttons_.end();) {
|
|
auto ws = std::find_if(workspaces_.begin(), workspaces_.end(), [it](auto node) -> bool {
|
|
return node["name"].asString() == it->first;
|
|
});
|
|
if (ws == workspaces_.end() ||
|
|
(!config_["all-outputs"].asBool() && (*ws)["output"].asString() != bar_.output->name)) {
|
|
it = buttons_.erase(it);
|
|
needReorder = true;
|
|
} else {
|
|
++it;
|
|
}
|
|
}
|
|
for (auto const &node : workspaces_) {
|
|
if (!config_["all-outputs"].asBool() && bar_.output->name != node["output"].asString()) {
|
|
continue;
|
|
}
|
|
auto it = buttons_.find(node["name"].asString());
|
|
if (it == buttons_.end()) {
|
|
addWorkspace(node);
|
|
needReorder = true;
|
|
} else {
|
|
auto &button = it->second;
|
|
if (node["focused"].asBool()) {
|
|
button.get_style_context()->add_class("focused");
|
|
} else {
|
|
button.get_style_context()->remove_class("focused");
|
|
}
|
|
if (node["visible"].asBool()) {
|
|
button.get_style_context()->add_class("visible");
|
|
} else {
|
|
button.get_style_context()->remove_class("visible");
|
|
}
|
|
if (node["urgent"].asBool()) {
|
|
button.get_style_context()->add_class("urgent");
|
|
} else {
|
|
button.get_style_context()->remove_class("urgent");
|
|
}
|
|
if (needReorder) {
|
|
box_.reorder_child(button, getWorkspaceIndex(node["name"].asString()));
|
|
}
|
|
auto icon = getIcon(node["name"].asString(), node);
|
|
std::string output = icon;
|
|
if (config_["format"].isString()) {
|
|
auto format = config_["format"].asString();
|
|
output = fmt::format(format,
|
|
fmt::arg("icon", icon),
|
|
fmt::arg("name", trimWorkspaceName(node["name"].asString())),
|
|
fmt::arg("index", node["num"].asString()));
|
|
}
|
|
if (!config_["disable-markup"].asBool()) {
|
|
static_cast<Gtk::Label *>(button.get_children()[0])->set_markup(output);
|
|
} else {
|
|
button.set_label(output);
|
|
}
|
|
onButtonReady(node, button);
|
|
}
|
|
}
|
|
if (scrolling_) {
|
|
scrolling_ = false;
|
|
}
|
|
}
|
|
|
|
void waybar::modules::sway::Workspaces::addWorkspace(const Json::Value &node) {
|
|
auto icon = getIcon(node["name"].asString(), node);
|
|
auto format = config_["format"].isString()
|
|
? fmt::format(config_["format"].asString(),
|
|
fmt::arg("icon", icon),
|
|
fmt::arg("name", trimWorkspaceName(node["name"].asString())),
|
|
fmt::arg("index", node["num"].asString()))
|
|
: icon;
|
|
auto pair = buttons_.emplace(node["name"].asString(), format);
|
|
auto &button = pair.first->second;
|
|
if (!config_["disable-markup"].asBool()) {
|
|
static_cast<Gtk::Label *>(button.get_children()[0])->set_markup(format);
|
|
}
|
|
box_.pack_start(button, false, false, 0);
|
|
button.set_relief(Gtk::RELIEF_NONE);
|
|
button.signal_clicked().connect([this, pair] {
|
|
try {
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
auto cmd = fmt::format("workspace \"{}\"", pair.first->first);
|
|
ipc_.sendCmd(IPC_COMMAND, cmd);
|
|
} catch (const std::exception &e) {
|
|
std::cerr << e.what() << std::endl;
|
|
}
|
|
});
|
|
if (!config_["disable-scroll"].asBool()) {
|
|
button.add_events(Gdk::SCROLL_MASK | Gdk::SMOOTH_SCROLL_MASK);
|
|
button.signal_scroll_event().connect(sigc::mem_fun(*this, &Workspaces::handleScroll));
|
|
}
|
|
box_.reorder_child(button, getWorkspaceIndex(node["name"].asString()));
|
|
if (node["focused"].asBool()) {
|
|
button.get_style_context()->add_class("focused");
|
|
}
|
|
if (node["visible"].asBool()) {
|
|
button.get_style_context()->add_class("visible");
|
|
}
|
|
if (node["urgent"].asBool()) {
|
|
button.get_style_context()->add_class("urgent");
|
|
}
|
|
|
|
onButtonReady(node, button);
|
|
}
|
|
|
|
std::string waybar::modules::sway::Workspaces::getIcon(const std::string &name,
|
|
const Json::Value &node) {
|
|
std::vector<std::string> keys = {name, "urgent", "focused", "visible", "default"};
|
|
for (auto const &key : keys) {
|
|
if (key == "focused" || key == "visible" || key == "urgent") {
|
|
if (config_["format-icons"][key].isString() && node[key].asBool()) {
|
|
return config_["format-icons"][key].asString();
|
|
}
|
|
} else if (config_["format-icons"][key].isString()) {
|
|
return config_["format-icons"][key].asString();
|
|
}
|
|
}
|
|
return name;
|
|
}
|
|
|
|
bool waybar::modules::sway::Workspaces::handleScroll(GdkEventScroll *e) {
|
|
// Avoid concurrent scroll event
|
|
if (scrolling_) {
|
|
return false;
|
|
}
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
uint8_t idx;
|
|
scrolling_ = true;
|
|
for (idx = 0; idx < workspaces_.size(); idx += 1) {
|
|
if (workspaces_[idx]["focused"].asBool()) {
|
|
break;
|
|
}
|
|
}
|
|
if (idx == workspaces_.size()) {
|
|
scrolling_ = false;
|
|
return false;
|
|
}
|
|
std::string name;
|
|
if (e->direction == GDK_SCROLL_UP) {
|
|
name = getCycleWorkspace(idx, true);
|
|
}
|
|
if (e->direction == GDK_SCROLL_DOWN) {
|
|
name = getCycleWorkspace(idx, false);
|
|
}
|
|
if (e->direction == GDK_SCROLL_SMOOTH) {
|
|
gdouble delta_x, delta_y;
|
|
gdk_event_get_scroll_deltas(reinterpret_cast<const GdkEvent *>(e), &delta_x, &delta_y);
|
|
if (delta_y < 0) {
|
|
name = getCycleWorkspace(idx, true);
|
|
} else if (delta_y > 0) {
|
|
name = getCycleWorkspace(idx, false);
|
|
}
|
|
}
|
|
if (name.empty() || name == workspaces_[idx]["name"].asString()) {
|
|
scrolling_ = false;
|
|
return false;
|
|
}
|
|
ipc_.sendCmd(IPC_COMMAND, fmt::format("workspace \"{}\"", name));
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(150));
|
|
return true;
|
|
}
|
|
|
|
const std::string waybar::modules::sway::Workspaces::getCycleWorkspace(uint8_t focused_workspace,
|
|
bool prev) const {
|
|
auto inc = prev ? -1 : 1;
|
|
int size = workspaces_.size();
|
|
uint8_t idx = 0;
|
|
for (int i = focused_workspace; i < size && i >= 0; i += inc) {
|
|
bool same_output = (workspaces_[i]["output"].asString() == bar_.output->name &&
|
|
!config_["all-outputs"].asBool()) ||
|
|
config_["all-outputs"].asBool();
|
|
bool same_name =
|
|
workspaces_[i]["name"].asString() == workspaces_[focused_workspace]["name"].asString();
|
|
if (same_output && !same_name) {
|
|
return workspaces_[i]["name"].asString();
|
|
}
|
|
if (prev && i - 1 < 0) {
|
|
i = size;
|
|
} else if (!prev && i + 1 >= size) {
|
|
i = -1;
|
|
} else if (idx >= workspaces_.size()) {
|
|
return "";
|
|
}
|
|
idx += 1;
|
|
}
|
|
return "";
|
|
}
|
|
|
|
uint16_t waybar::modules::sway::Workspaces::getWorkspaceIndex(const std::string &name) const {
|
|
uint16_t idx = 0;
|
|
for (const auto &workspace : workspaces_) {
|
|
if (workspace["name"].asString() == name) {
|
|
return idx;
|
|
}
|
|
if (!(!config_["all-outputs"].asBool() &&
|
|
workspace["output"].asString() != bar_.output->name)) {
|
|
idx += 1;
|
|
}
|
|
}
|
|
return workspaces_.size();
|
|
}
|
|
|
|
std::string waybar::modules::sway::Workspaces::trimWorkspaceName(std::string name) {
|
|
std::size_t found = name.find(":");
|
|
if (found != std::string::npos) {
|
|
return name.substr(found + 1);
|
|
}
|
|
return name;
|
|
}
|
|
|
|
void waybar::modules::sway::Workspaces::onButtonReady(const Json::Value &node,
|
|
Gtk::Button & button) {
|
|
if (config_["current-only"].asBool()) {
|
|
if (node["focused"].asBool()) {
|
|
button.show();
|
|
} else {
|
|
button.hide();
|
|
}
|
|
} else {
|
|
button.show();
|
|
}
|
|
}
|
|
|
|
waybar::modules::sway::Workspaces::operator Gtk::Widget &() { return box_; }
|