waybar/src/modules/sni/snh.cpp

151 lines
4.2 KiB
C++
Raw Normal View History

#include "modules/sni/snh.hpp"
#include <iostream>
2018-09-04 23:50:08 +02:00
using namespace waybar::modules::SNI;
Host::Host(Glib::Dispatcher* dp)
: dp_(dp)
{
GBusNameOwnerFlags flags = static_cast<GBusNameOwnerFlags>(
G_BUS_NAME_OWNER_FLAGS_NONE);
bus_name_ = "org.kde.StatusNotifierHost-" + std::to_string(getpid());
object_path_ = "/StatusNotifierHost";
bus_name_id_ = g_bus_own_name(G_BUS_TYPE_SESSION,
bus_name_.c_str(), flags,
&Host::busAcquired, nullptr, nullptr, this, nullptr);
}
2018-09-04 23:50:08 +02:00
void Host::busAcquired(GDBusConnection* connection,
const gchar* name, gpointer data)
{
auto host = static_cast<SNI::Host *>(data);
host->watcher_id_ = g_bus_watch_name(
G_BUS_TYPE_SESSION,
"org.kde.StatusNotifierWatcher",
G_BUS_NAME_WATCHER_FLAGS_NONE,
&Host::nameAppeared, &Host::nameVanished, data, nullptr);
}
2018-09-04 23:50:08 +02:00
void Host::nameAppeared(GDBusConnection* connection,
const gchar* name, const gchar* name_owner, gpointer data)
{
auto host = static_cast<SNI::Host *>(data);
if (host->cancellable_ != nullptr) {
2018-10-25 13:49:30 +02:00
// TODO
}
host->cancellable_ = g_cancellable_new();
2018-10-26 10:05:54 +02:00
sn_watcher_proxy_new(
connection,
G_DBUS_PROXY_FLAGS_NONE,
"org.kde.StatusNotifierWatcher",
"/StatusNotifierWatcher",
host->cancellable_, &Host::proxyReady, data);
}
2018-09-04 23:50:08 +02:00
void Host::nameVanished(GDBusConnection* connection,
const gchar* name, gpointer data)
{
auto host = static_cast<SNI::Host *>(data);
g_cancellable_cancel(host->cancellable_);
g_clear_object(&host->cancellable_);
g_clear_object(&host->watcher_);
host->items.clear();
}
2018-09-04 23:50:08 +02:00
void Host::proxyReady(GObject* src, GAsyncResult* res,
gpointer data)
{
GError* error = nullptr;
2018-10-26 10:05:54 +02:00
SnWatcher* watcher = sn_watcher_proxy_new_finish(res, &error);
if (g_error_matches(error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
std::cerr << error->message << std::endl;
g_error_free(error);
return;
}
auto host = static_cast<SNI::Host *>(data);
host->watcher_ = watcher;
if (error != nullptr) {
std::cerr << error->message << std::endl;
g_error_free(error);
return;
}
2018-10-26 10:05:54 +02:00
sn_watcher_call_register_host(
host->watcher_, host->object_path_.c_str(), host->cancellable_,
&Host::registerHost, data);
}
2018-09-04 23:50:08 +02:00
void Host::registerHost(GObject* src, GAsyncResult* res,
gpointer data)
{
GError* error = nullptr;
2018-10-26 10:05:54 +02:00
sn_watcher_call_register_host_finish(SN_WATCHER(src), res, &error);
if (g_error_matches(error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
std::cerr << error->message << std::endl;
g_error_free(error);
return;
}
auto host = static_cast<SNI::Host *>(data);
if (error != nullptr) {
std::cerr << error->message << std::endl;
g_error_free(error);
return;
}
2018-10-26 10:05:54 +02:00
g_signal_connect(host->watcher_, "item-registered",
G_CALLBACK(&Host::itemRegistered), data);
2018-10-26 10:05:54 +02:00
g_signal_connect(host->watcher_, "item-unregistered",
G_CALLBACK(&Host::itemUnregistered), data);
2018-10-26 10:05:54 +02:00
auto items = sn_watcher_dup_registered_items(host->watcher_);
if (items) {
for (uint32_t i = 0; items[i] != nullptr; i += 1) {
host->addRegisteredItem(items[i]);
}
}
g_strfreev(items);
}
2018-09-04 23:50:08 +02:00
void Host::itemRegistered(
2018-10-26 10:05:54 +02:00
SnWatcher* watcher, const gchar* service, gpointer data)
{
auto host = static_cast<SNI::Host *>(data);
host->addRegisteredItem(service);
}
2018-09-04 23:50:08 +02:00
void Host::itemUnregistered(
2018-10-26 10:05:54 +02:00
SnWatcher* watcher, const gchar* service, gpointer data)
{
2018-09-04 23:50:08 +02:00
auto host = static_cast<SNI::Host *>(data);
auto [bus_name, object_path] = host->getBusNameAndObjectPath(service);
for (auto it = host->items.begin(); it != host->items.end(); ++it) {
if (it->bus_name == bus_name && it->object_path == object_path) {
host->items.erase(it);
break;
}
}
host->dp_->emit();
}
2018-09-04 23:50:08 +02:00
std::tuple<std::string, std::string> Host::getBusNameAndObjectPath(
const gchar* service)
{
2018-09-04 23:50:08 +02:00
std::string bus_name;
std::string object_path;
gchar* tmp = g_strstr_len(service, -1, "/");
if (tmp != nullptr) {
gchar** str = g_strsplit(service, "/", 2);
2018-09-04 23:50:08 +02:00
bus_name = str[0];
object_path = tmp;
g_strfreev(str);
} else {
2018-09-04 23:50:08 +02:00
bus_name = service;
object_path = "/StatusNotifierItem";
}
2018-09-04 23:50:08 +02:00
return { bus_name, object_path };
}
2018-09-04 23:50:08 +02:00
void Host::addRegisteredItem(const gchar* service)
{
2018-09-04 23:50:08 +02:00
auto [bus_name, object_path] = getBusNameAndObjectPath(service);
items.emplace_back(bus_name, object_path, dp_);
2018-10-04 18:03:01 +02:00
}