#include "client.hpp" #include waybar::Client::Client(int argc, char* argv[]) : gtk_app(Gtk::Application::create(argc, argv, "fr.arouillard.waybar")), gdk_display(Gdk::Display::get_default()) { if (!gdk_display) { throw std::runtime_error("Can't find display"); } if (!GDK_IS_WAYLAND_DISPLAY(gdk_display->gobj())) { throw std::runtime_error("Bar need to run under Wayland"); } wl_display = gdk_wayland_display_get_wl_display(gdk_display->gobj()); auto getFirstValidPath = [] (std::vector possiblePaths) { wordexp_t p; for (const std::string &path: possiblePaths) { if (wordexp(path.c_str(), &p, 0) == 0) { if (access(*p.we_wordv, F_OK) == 0) { std::string result = *p.we_wordv; wordfree(&p); return result; } wordfree(&p); } } return std::string(); }; config_file = getFirstValidPath({ "$XDG_CONFIG_HOME/waybar/config", "$HOME/waybar/config", "/etc/xdg/waybar/config", "./resources/config", }); css_file = getFirstValidPath({ "$XDG_CONFIG_HOME/waybar/style.css", "$HOME/waybar/style.css", "/etc/xdg/waybar/style.css", "./resources/style.css", }); if (css_file.empty() || config_file.empty()) { throw std::runtime_error("Missing required resources files"); } std::cout << "Resources files: " + config_file + ", " + css_file << std::endl; } void waybar::Client::handleGlobal(void *data, struct wl_registry *registry, uint32_t name, const char *interface, uint32_t version) { auto o = static_cast(data); if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) { o->layer_shell = static_cast( wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, version)); } else if (strcmp(interface, wl_output_interface.name) == 0) { auto output = std::make_unique(); *output = static_cast(wl_registry_bind(registry, name, &wl_output_interface, version)); if (o->xdg_output_manager != nullptr) { o->bars.emplace_back(std::make_unique(*o, std::move(output), name)); } } else if (strcmp(interface, wl_seat_interface.name) == 0) { o->seat = static_cast(wl_registry_bind(registry, name, &wl_seat_interface, version)); } else if (strcmp(interface, zxdg_output_manager_v1_interface.name) == 0 && version >= ZXDG_OUTPUT_V1_NAME_SINCE_VERSION) { o->xdg_output_manager = static_cast( wl_registry_bind(registry, name, &zxdg_output_manager_v1_interface, ZXDG_OUTPUT_V1_NAME_SINCE_VERSION)); } } void waybar::Client::handleGlobalRemove(void* data, struct wl_registry* /*registry*/, uint32_t name) { auto o = static_cast(data); for (auto it = o->bars.begin(); it != o->bars.end(); ++it) { if ((**it).wl_name == name) { o->bars.erase(it); break; } } } void waybar::Client::bindInterfaces() { registry = wl_display_get_registry(wl_display); static const struct wl_registry_listener registry_listener = { .global = handleGlobal, .global_remove = handleGlobalRemove, }; wl_registry_add_listener(registry, ®istry_listener, this); wl_display_roundtrip(wl_display); if (!layer_shell || !seat || !xdg_output_manager) { throw std::runtime_error("Failed to acquire required resources."); } wl_display_roundtrip(wl_display); } int waybar::Client::main(int /*argc*/, char* /*argv*/[]) { bindInterfaces(); gtk_app->hold(); gtk_app->run(); bars.clear(); zxdg_output_manager_v1_destroy(xdg_output_manager); zwlr_layer_shell_v1_destroy(layer_shell); wl_registry_destroy(registry); wl_seat_destroy(seat); wl_display_disconnect(wl_display); return 0; }