Waybar/src/client.cpp

146 lines
4.7 KiB
C++
Raw Normal View History

2018-08-08 21:54:58 +00:00
#include "client.hpp"
2018-12-18 16:30:54 +00:00
#include "util/clara.hpp"
2018-11-15 13:44:43 +00:00
#include <iostream>
2018-08-08 21:54:58 +00:00
waybar::Client::Client(int argc, char* argv[])
2018-12-18 16:30:54 +00:00
: gtk_main(argc, argv),
2018-08-29 19:07:58 +00:00
gdk_display(Gdk::Display::get_default())
{
2018-08-29 19:07:58 +00:00
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());
}
2018-08-08 21:54:58 +00:00
2018-11-24 10:04:56 +00:00
const std::string waybar::Client::getValidPath(std::vector<std::string> paths)
{
wordexp_t p;
for (const std::string &path: paths) {
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();
}
2018-08-16 12:29:41 +00:00
void waybar::Client::handleGlobal(void *data, struct wl_registry *registry,
2018-08-09 18:22:01 +00:00
uint32_t name, const char *interface, uint32_t version)
{
2018-08-16 12:29:41 +00:00
auto o = static_cast<waybar::Client *>(data);
if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
o->layer_shell = static_cast<struct zwlr_layer_shell_v1 *>(
wl_registry_bind(registry, name, &zwlr_layer_shell_v1_interface, version));
} else if (strcmp(interface, wl_output_interface.name) == 0) {
2018-08-14 23:53:43 +00:00
auto output = std::make_unique<struct wl_output *>();
2018-08-16 12:29:41 +00:00
*output = static_cast<struct wl_output *>(wl_registry_bind(registry, name,
&wl_output_interface, version));
if (o->xdg_output_manager != nullptr) {
2018-08-19 11:39:57 +00:00
o->bars.emplace_back(std::make_unique<Bar>(*o, std::move(output), name));
2018-08-16 12:29:41 +00:00
}
} else if (strcmp(interface, wl_seat_interface.name) == 0) {
o->seat = static_cast<struct wl_seat *>(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<struct zxdg_output_manager_v1 *>(
wl_registry_bind(registry, name,
&zxdg_output_manager_v1_interface, ZXDG_OUTPUT_V1_NAME_SINCE_VERSION));
}
2018-08-09 18:22:01 +00:00
}
2018-08-19 11:39:57 +00:00
void waybar::Client::handleGlobalRemove(void* data,
struct wl_registry* /*registry*/, uint32_t name)
2018-08-09 18:22:01 +00:00
{
2018-08-19 11:39:57 +00:00
auto o = static_cast<waybar::Client *>(data);
for (auto it = o->bars.begin(); it != o->bars.end(); ++it) {
if ((**it).wl_name == name) {
o->bars.erase(it);
break;
}
}
2018-08-09 18:22:01 +00:00
}
2018-12-18 16:30:54 +00:00
void waybar::Client::setupConfigs(const std::string& config, const std::string& style)
{
config_file = config.empty() ? getValidPath({
"$XDG_CONFIG_HOME/waybar/config",
"$HOME/.config/waybar/config",
"$HOME/waybar/config",
"/etc/xdg/waybar/config",
"./resources/config",
}) : config;
css_file = style.empty() ? getValidPath({
"$XDG_CONFIG_HOME/waybar/style.css",
"$HOME/.config/waybar/style.css",
"$HOME/waybar/style.css",
"/etc/xdg/waybar/style.css",
"./resources/style.css",
}) : style;
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;
}
2018-08-16 12:29:41 +00:00
void waybar::Client::bindInterfaces()
2018-08-08 21:54:58 +00:00
{
2018-08-16 12:29:41 +00:00
registry = wl_display_get_registry(wl_display);
2018-08-09 18:22:01 +00:00
static const struct wl_registry_listener registry_listener = {
2018-08-16 12:29:41 +00:00
.global = handleGlobal,
.global_remove = handleGlobalRemove,
2018-08-09 18:22:01 +00:00
};
2018-08-08 21:54:58 +00:00
wl_registry_add_listener(registry, &registry_listener, this);
2018-08-16 12:29:41 +00:00
wl_display_roundtrip(wl_display);
2018-11-16 09:02:12 +00:00
if (!layer_shell || !seat || !xdg_output_manager) {
throw std::runtime_error("Failed to acquire required resources.");
}
wl_display_roundtrip(wl_display);
2018-08-08 21:54:58 +00:00
}
2018-12-18 16:30:54 +00:00
int waybar::Client::main(int argc, char* argv[])
2018-08-08 21:54:58 +00:00
{
2018-12-18 16:30:54 +00:00
bool show_help = false;
bool show_version = false;
std::string config;
std::string style;
2019-01-28 18:26:16 +00:00
std::string bar_id;
2018-12-18 16:30:54 +00:00
auto cli = clara::detail::Help(show_help)
| clara::detail::Opt(show_version)["-v"]["--version"]("Show version")
| clara::detail::Opt(config, "config")["-c"]["--config"]("Config path")
2019-01-28 18:26:16 +00:00
| clara::detail::Opt(style, "style")["-s"]["--style"]("Style path")
| clara::detail::Opt(bar_id, "id")["-b"]["--bar"]("Bar id");
2018-12-18 16:30:54 +00:00
auto res = cli.parse(clara::detail::Args(argc, argv));
if (!res) {
std::cerr << "Error in command line: " << res.errorMessage() << std::endl;
return 1;
}
if (show_help) {
std::cout << cli << std::endl;
return 0;
}
if (show_version) {
std::cout << "Waybar v" << VERSION << std::endl;
return 0;
}
setupConfigs(config, style);
2018-08-16 12:29:41 +00:00
bindInterfaces();
2018-12-18 16:30:54 +00:00
gtk_main.run();
2018-08-17 18:28:26 +00:00
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);
2018-08-08 21:54:58 +00:00
return 0;
}