mirror of
https://github.com/rad4day/Waybar.git
synced 2023-12-21 10:22:59 +01:00
331 lines
12 KiB
C++
331 lines
12 KiB
C++
#include "bar.hpp"
|
|
#include <spdlog/spdlog.h>
|
|
#include "client.hpp"
|
|
#include "factory.hpp"
|
|
|
|
waybar::Bar::Bar(struct waybar_output* w_output, const Json::Value& w_config)
|
|
: output(w_output),
|
|
config(w_config),
|
|
window{Gtk::WindowType::WINDOW_TOPLEVEL},
|
|
surface(nullptr),
|
|
layer_surface(nullptr),
|
|
anchor_(ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP),
|
|
left_(Gtk::ORIENTATION_HORIZONTAL, 0),
|
|
center_(Gtk::ORIENTATION_HORIZONTAL, 0),
|
|
right_(Gtk::ORIENTATION_HORIZONTAL, 0),
|
|
box_(Gtk::ORIENTATION_HORIZONTAL, 0) {
|
|
window.set_title("waybar");
|
|
window.set_name("waybar");
|
|
window.set_decorated(false);
|
|
window.get_style_context()->add_class(output->name);
|
|
window.get_style_context()->add_class(config["name"].asString());
|
|
window.get_style_context()->add_class(config["position"].asString());
|
|
|
|
if (config["position"] == "right" || config["position"] == "left") {
|
|
height_ = 0;
|
|
width_ = 1;
|
|
}
|
|
height_ = config["height"].isUInt() ? config["height"].asUInt() : height_;
|
|
width_ = config["width"].isUInt() ? config["width"].asUInt() : width_;
|
|
|
|
window.signal_realize().connect_notify(sigc::mem_fun(*this, &Bar::onRealize));
|
|
window.signal_map_event().connect_notify(sigc::mem_fun(*this, &Bar::onMap));
|
|
window.signal_configure_event().connect_notify(sigc::mem_fun(*this, &Bar::onConfigure));
|
|
window.set_size_request(width_, height_);
|
|
|
|
if (config["position"] == "bottom") {
|
|
anchor_ = ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM;
|
|
} else if (config["position"] == "left") {
|
|
anchor_ = ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT;
|
|
} else if (config["position"] == "right") {
|
|
anchor_ = ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
|
|
}
|
|
if (anchor_ == ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM ||
|
|
anchor_ == ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP) {
|
|
anchor_ |= ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT | ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT;
|
|
} else if (anchor_ == ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT ||
|
|
anchor_ == ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT) {
|
|
anchor_ |= ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP | ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM;
|
|
left_ = Gtk::Box(Gtk::ORIENTATION_VERTICAL, 0);
|
|
center_ = Gtk::Box(Gtk::ORIENTATION_VERTICAL, 0);
|
|
right_ = Gtk::Box(Gtk::ORIENTATION_VERTICAL, 0);
|
|
box_ = Gtk::Box(Gtk::ORIENTATION_VERTICAL, 0);
|
|
vertical = true;
|
|
}
|
|
|
|
setupWidgets();
|
|
|
|
if (window.get_realized()) {
|
|
onRealize();
|
|
}
|
|
window.show_all();
|
|
}
|
|
|
|
void waybar::Bar::onConfigure(GdkEventConfigure* ev) {
|
|
auto tmp_height = height_;
|
|
auto tmp_width = width_;
|
|
if (ev->height > static_cast<int>(height_)) {
|
|
// Default minimal value
|
|
if (height_ != 1) {
|
|
spdlog::warn(MIN_HEIGHT_MSG, height_, ev->height);
|
|
}
|
|
if (config["height"].isUInt()) {
|
|
spdlog::info(SIZE_DEFINED, "Height");
|
|
} else {
|
|
tmp_height = ev->height;
|
|
}
|
|
}
|
|
if (ev->width > static_cast<int>(width_)) {
|
|
// Default minimal value
|
|
if (width_ != 1) {
|
|
spdlog::warn(MIN_WIDTH_MSG, width_, ev->width);
|
|
}
|
|
if (config["width"].isUInt()) {
|
|
spdlog::info(SIZE_DEFINED, "Width");
|
|
} else {
|
|
tmp_width = ev->width;
|
|
}
|
|
}
|
|
if (tmp_width != width_ || tmp_height != height_) {
|
|
zwlr_layer_surface_v1_set_size(layer_surface, tmp_width, tmp_height);
|
|
}
|
|
}
|
|
|
|
void waybar::Bar::onRealize() {
|
|
auto gdk_window = window.get_window()->gobj();
|
|
gdk_wayland_window_set_use_custom_surface(gdk_window);
|
|
}
|
|
|
|
void waybar::Bar::onMap(GdkEventAny* ev) {
|
|
auto gdk_window = window.get_window()->gobj();
|
|
surface = gdk_wayland_window_get_wl_surface(gdk_window);
|
|
|
|
auto client = waybar::Client::inst();
|
|
auto layer =
|
|
config["layer"] == "top" ? ZWLR_LAYER_SHELL_V1_LAYER_TOP : ZWLR_LAYER_SHELL_V1_LAYER_BOTTOM;
|
|
layer_surface = zwlr_layer_shell_v1_get_layer_surface(
|
|
client->layer_shell, surface, output->output, layer, "waybar");
|
|
|
|
zwlr_layer_surface_v1_set_keyboard_interactivity(layer_surface, false);
|
|
zwlr_layer_surface_v1_set_anchor(layer_surface, anchor_);
|
|
zwlr_layer_surface_v1_set_size(layer_surface, width_, height_);
|
|
setMarginsAndZone(height_, width_);
|
|
static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
|
|
.configure = layerSurfaceHandleConfigure,
|
|
.closed = layerSurfaceHandleClosed,
|
|
};
|
|
zwlr_layer_surface_v1_add_listener(layer_surface, &layer_surface_listener, this);
|
|
|
|
wl_surface_commit(surface);
|
|
wl_display_roundtrip(client->wl_display);
|
|
}
|
|
|
|
void waybar::Bar::setMarginsAndZone(uint32_t height, uint32_t width) {
|
|
if (config["margin-top"].isInt() || config["margin-right"].isInt() ||
|
|
config["margin-bottom"].isInt() || config["margin-left"].isInt()) {
|
|
margins_ = {
|
|
config["margin-top"].isInt() ? config["margin-top"].asInt() : 0,
|
|
config["margin-right"].isInt() ? config["margin-right"].asInt() : 0,
|
|
config["margin-bottom"].isInt() ? config["margin-bottom"].asInt() : 0,
|
|
config["margin-left"].isInt() ? config["margin-left"].asInt() : 0,
|
|
};
|
|
} else if (config["margin"].isString()) {
|
|
std::istringstream iss(config["margin"].asString());
|
|
std::vector<std::string> margins{std::istream_iterator<std::string>(iss), {}};
|
|
try {
|
|
if (margins.size() == 1) {
|
|
auto gaps = std::stoi(margins[0], nullptr, 10);
|
|
margins_ = {.top = gaps, .right = gaps, .bottom = gaps, .left = gaps};
|
|
}
|
|
if (margins.size() == 2) {
|
|
auto vertical_margins = std::stoi(margins[0], nullptr, 10);
|
|
auto horizontal_margins = std::stoi(margins[1], nullptr, 10);
|
|
margins_ = {.top = vertical_margins,
|
|
.right = horizontal_margins,
|
|
.bottom = vertical_margins,
|
|
.left = horizontal_margins};
|
|
}
|
|
if (margins.size() == 3) {
|
|
auto horizontal_margins = std::stoi(margins[1], nullptr, 10);
|
|
margins_ = {.top = std::stoi(margins[0], nullptr, 10),
|
|
.right = horizontal_margins,
|
|
.bottom = std::stoi(margins[2], nullptr, 10),
|
|
.left = horizontal_margins};
|
|
}
|
|
if (margins.size() == 4) {
|
|
margins_ = {.top = std::stoi(margins[0], nullptr, 10),
|
|
.right = std::stoi(margins[1], nullptr, 10),
|
|
.bottom = std::stoi(margins[2], nullptr, 10),
|
|
.left = std::stoi(margins[3], nullptr, 10)};
|
|
}
|
|
} catch (...) {
|
|
spdlog::warn("Invalid margins: {}", config["margin"].asString());
|
|
}
|
|
} else if (config["margin"].isInt()) {
|
|
auto gaps = config["margin"].asInt();
|
|
margins_ = {.top = gaps, .right = gaps, .bottom = gaps, .left = gaps};
|
|
}
|
|
zwlr_layer_surface_v1_set_margin(
|
|
layer_surface, margins_.top, margins_.right, margins_.bottom, margins_.left);
|
|
auto zone = vertical ? width + margins_.right : height + margins_.bottom;
|
|
zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, zone);
|
|
}
|
|
|
|
// Converting string to button code rn as to avoid doing it later
|
|
void waybar::Bar::setupAltFormatKeyForModule(const std::string& module_name) {
|
|
if (config.isMember(module_name)) {
|
|
Json::Value& module = config[module_name];
|
|
if (module.isMember("format-alt")) {
|
|
if (module.isMember("format-alt-click")) {
|
|
Json::Value& click = module["format-alt-click"];
|
|
if (click.isString()) {
|
|
if (click == "click-right") {
|
|
module["format-alt-click"] = 3U;
|
|
} else if (click == "click-middle") {
|
|
module["format-alt-click"] = 2U;
|
|
} else if (click == "click-backward") {
|
|
module["format-alt-click"] = 8U;
|
|
} else if (click == "click-forward") {
|
|
module["format-alt-click"] = 9U;
|
|
} else {
|
|
module["format-alt-click"] = 1U; // default click-left
|
|
}
|
|
} else {
|
|
module["format-alt-click"] = 1U;
|
|
}
|
|
} else {
|
|
module["format-alt-click"] = 1U;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void waybar::Bar::setupAltFormatKeyForModuleList(const char* module_list_name) {
|
|
if (config.isMember(module_list_name)) {
|
|
Json::Value& modules = config[module_list_name];
|
|
for (const Json::Value& module_name : modules) {
|
|
if (module_name.isString()) {
|
|
setupAltFormatKeyForModule(module_name.asString());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void waybar::Bar::handleSignal(int signal) {
|
|
for (auto& module : modules_left_) {
|
|
auto* custom = dynamic_cast<waybar::modules::Custom*>(module.get());
|
|
if (custom != nullptr) {
|
|
custom->refresh(signal);
|
|
}
|
|
}
|
|
for (auto& module : modules_center_) {
|
|
auto* custom = dynamic_cast<waybar::modules::Custom*>(module.get());
|
|
if (custom != nullptr) {
|
|
custom->refresh(signal);
|
|
}
|
|
}
|
|
for (auto& module : modules_right_) {
|
|
auto* custom = dynamic_cast<waybar::modules::Custom*>(module.get());
|
|
if (custom != nullptr) {
|
|
custom->refresh(signal);
|
|
}
|
|
}
|
|
}
|
|
|
|
void waybar::Bar::layerSurfaceHandleConfigure(void* data, struct zwlr_layer_surface_v1* surface,
|
|
uint32_t serial, uint32_t width, uint32_t height) {
|
|
auto o = static_cast<waybar::Bar*>(data);
|
|
if (width != o->width_ || height != o->height_) {
|
|
o->width_ = width;
|
|
o->height_ = height;
|
|
o->window.set_size_request(o->width_, o->height_);
|
|
o->window.resize(o->width_, o->height_);
|
|
auto zone = o->vertical ? width + o->margins_.right : height + o->margins_.bottom;
|
|
zwlr_layer_surface_v1_set_exclusive_zone(o->layer_surface, zone);
|
|
spdlog::info(BAR_SIZE_MSG,
|
|
o->width_ == 1 ? "auto" : std::to_string(o->width_),
|
|
o->height_ == 1 ? "auto" : std::to_string(o->height_),
|
|
o->output->name);
|
|
wl_surface_commit(o->surface);
|
|
}
|
|
zwlr_layer_surface_v1_ack_configure(surface, serial);
|
|
}
|
|
|
|
void waybar::Bar::layerSurfaceHandleClosed(void* data, struct zwlr_layer_surface_v1* /*surface*/) {
|
|
auto o = static_cast<waybar::Bar*>(data);
|
|
if (o->layer_surface) {
|
|
zwlr_layer_surface_v1_destroy(o->layer_surface);
|
|
o->layer_surface = nullptr;
|
|
}
|
|
o->modules_left_.clear();
|
|
o->modules_center_.clear();
|
|
o->modules_right_.clear();
|
|
}
|
|
|
|
auto waybar::Bar::toggle() -> void {
|
|
visible = !visible;
|
|
auto zone = visible ? height_ : 0;
|
|
if (!visible) {
|
|
window.get_style_context()->add_class("hidden");
|
|
} else {
|
|
window.get_style_context()->remove_class("hidden");
|
|
}
|
|
zwlr_layer_surface_v1_set_exclusive_zone(layer_surface, zone);
|
|
wl_surface_commit(surface);
|
|
}
|
|
|
|
void waybar::Bar::getModules(const Factory& factory, const std::string& pos) {
|
|
if (config[pos].isArray()) {
|
|
for (const auto& name : config[pos]) {
|
|
try {
|
|
auto module = factory.makeModule(name.asString());
|
|
if (pos == "modules-left") {
|
|
modules_left_.emplace_back(module);
|
|
}
|
|
if (pos == "modules-center") {
|
|
modules_center_.emplace_back(module);
|
|
}
|
|
if (pos == "modules-right") {
|
|
modules_right_.emplace_back(module);
|
|
}
|
|
module->dp.connect([module, &name] {
|
|
try {
|
|
module->update();
|
|
} catch (const std::exception& e) {
|
|
spdlog::error("{}: {}", name.asString(), e.what());
|
|
}
|
|
});
|
|
} catch (const std::exception& e) {
|
|
spdlog::warn("module {}: {}", name.asString(), e.what());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
auto waybar::Bar::setupWidgets() -> void {
|
|
window.add(box_);
|
|
box_.pack_start(left_, false, false);
|
|
box_.set_center_widget(center_);
|
|
box_.pack_end(right_, false, false);
|
|
|
|
// Convert to button code for every module that is used.
|
|
setupAltFormatKeyForModuleList("modules-left");
|
|
setupAltFormatKeyForModuleList("modules-right");
|
|
setupAltFormatKeyForModuleList("modules-center");
|
|
|
|
Factory factory(*this, config);
|
|
getModules(factory, "modules-left");
|
|
getModules(factory, "modules-center");
|
|
getModules(factory, "modules-right");
|
|
for (auto const& module : modules_left_) {
|
|
left_.pack_start(*module, false, false);
|
|
}
|
|
for (auto const& module : modules_center_) {
|
|
center_.pack_start(*module, false, false);
|
|
}
|
|
std::reverse(modules_right_.begin(), modules_right_.end());
|
|
for (auto const& module : modules_right_) {
|
|
right_.pack_end(*module, false, false);
|
|
}
|
|
}
|