Waybar/src/modules/sni/host.cpp

144 lines
5.0 KiB
C++
Raw Normal View History

2018-11-22 14:47:23 +00:00
#include "modules/sni/host.hpp"
2019-05-18 23:44:45 +00:00
#include <spdlog/spdlog.h>
#include <fmt/ostream.h>
2019-04-24 10:37:24 +00:00
namespace waybar::modules::SNI {
2018-09-04 21:50:08 +00:00
Host::Host(const std::size_t id, const Json::Value& config,
const std::function<void(std::unique_ptr<Item>&)>& on_add,
const std::function<void(std::unique_ptr<Item>&)>& on_remove)
: bus_name_("org.kde.StatusNotifierHost-" + std::to_string(getpid()) + "-" +
std::to_string(id)),
object_path_("/StatusNotifierHost/" + std::to_string(id)),
bus_name_id_(Gio::DBus::own_name(Gio::DBus::BusType::BUS_TYPE_SESSION, bus_name_,
sigc::mem_fun(*this, &Host::busAcquired))),
config_(config),
on_add_(on_add),
on_remove_(on_remove) {}
Host::~Host() {
if (bus_name_id_ > 0) {
Gio::DBus::unwatch_name(bus_name_id_);
bus_name_id_ = 0;
}
if (watcher_id_ > 0) {
Gio::DBus::unwatch_name(watcher_id_);
watcher_id_ = 0;
}
g_cancellable_cancel(cancellable_);
g_clear_object(&cancellable_);
g_clear_object(&watcher_);
}
void Host::busAcquired(const Glib::RefPtr<Gio::DBus::Connection>& conn, Glib::ustring name) {
2019-04-18 15:52:00 +00:00
watcher_id_ = Gio::DBus::watch_name(conn,
"org.kde.StatusNotifierWatcher",
sigc::mem_fun(*this, &Host::nameAppeared),
sigc::mem_fun(*this, &Host::nameVanished));
2018-11-22 14:47:23 +00:00
}
void Host::nameAppeared(const Glib::RefPtr<Gio::DBus::Connection>& conn, const Glib::ustring name,
const Glib::ustring& name_owner) {
2018-11-22 14:47:23 +00:00
if (cancellable_ != nullptr) {
2018-10-25 11:49:30 +00:00
// TODO
2018-10-29 20:52:53 +00:00
return;
}
2018-11-22 14:47:23 +00:00
cancellable_ = g_cancellable_new();
2019-04-18 15:52:00 +00:00
sn_watcher_proxy_new(conn->gobj(),
G_DBUS_PROXY_FLAGS_NONE,
"org.kde.StatusNotifierWatcher",
"/StatusNotifierWatcher",
cancellable_,
&Host::proxyReady,
this);
}
void Host::nameVanished(const Glib::RefPtr<Gio::DBus::Connection>& conn, const Glib::ustring name) {
2018-11-22 14:47:23 +00:00
g_cancellable_cancel(cancellable_);
g_clear_object(&cancellable_);
g_clear_object(&watcher_);
items_.clear();
}
void Host::proxyReady(GObject* src, GAsyncResult* res, gpointer data) {
2019-04-18 15:52:00 +00:00
GError* error = nullptr;
2018-10-26 08:05:54 +00:00
SnWatcher* watcher = sn_watcher_proxy_new_finish(res, &error);
if (g_error_matches(error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
2019-05-18 23:44:45 +00:00
spdlog::error("Host: {}", error->message);
g_error_free(error);
return;
}
auto host = static_cast<SNI::Host*>(data);
host->watcher_ = watcher;
if (error != nullptr) {
2019-05-18 23:44:45 +00:00
spdlog::error("Host: {}", error->message);
g_error_free(error);
return;
}
2019-04-18 15:52:00 +00:00
sn_watcher_call_register_host(
host->watcher_, host->object_path_.c_str(), host->cancellable_, &Host::registerHost, data);
}
void Host::registerHost(GObject* src, GAsyncResult* res, gpointer data) {
GError* error = nullptr;
2018-10-26 08:05:54 +00:00
sn_watcher_call_register_host_finish(SN_WATCHER(src), res, &error);
if (g_error_matches(error, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
2019-05-18 23:44:45 +00:00
spdlog::error("Host: {}", error->message);
g_error_free(error);
return;
}
auto host = static_cast<SNI::Host*>(data);
if (error != nullptr) {
2019-05-18 23:44:45 +00:00
spdlog::error("Host: {}", error->message);
g_error_free(error);
return;
}
g_signal_connect(host->watcher_, "item-registered", G_CALLBACK(&Host::itemRegistered), data);
g_signal_connect(host->watcher_, "item-unregistered", G_CALLBACK(&Host::itemUnregistered), data);
2018-10-26 08:05:54 +00:00
auto items = sn_watcher_dup_registered_items(host->watcher_);
2019-04-24 10:37:24 +00:00
if (items != nullptr) {
for (uint32_t i = 0; items[i] != nullptr; i += 1) {
host->addRegisteredItem(items[i]);
}
}
g_strfreev(items);
}
void Host::itemRegistered(SnWatcher* watcher, const gchar* service, gpointer data) {
auto host = static_cast<SNI::Host*>(data);
host->addRegisteredItem(service);
}
void Host::itemUnregistered(SnWatcher* watcher, const gchar* service, gpointer data) {
auto host = static_cast<SNI::Host*>(data);
2018-09-04 21:50:08 +00:00
auto [bus_name, object_path] = host->getBusNameAndObjectPath(service);
2018-11-22 14:47:23 +00:00
for (auto it = host->items_.begin(); it != host->items_.end(); ++it) {
if ((*it)->bus_name == bus_name && (*it)->object_path == object_path) {
host->on_remove_(*it);
host->items_.erase(it);
2018-09-04 21:50:08 +00:00
break;
}
}
}
std::tuple<std::string, std::string> Host::getBusNameAndObjectPath(const std::string service) {
2019-04-24 10:37:24 +00:00
auto it = service.find('/');
2018-11-22 14:47:23 +00:00
if (it != std::string::npos) {
return {service.substr(0, it), service.substr(it)};
}
2018-11-22 14:47:23 +00:00
return {service, "/StatusNotifierItem"};
}
void Host::addRegisteredItem(std::string service) {
2018-09-04 21:50:08 +00:00
auto [bus_name, object_path] = getBusNameAndObjectPath(service);
2019-05-13 12:27:01 +00:00
auto it = std::find_if(items_.begin(), items_.end(), [&bus_name, &object_path](const auto& item) {
return bus_name == item->bus_name && object_path == item->object_path;
});
if (it == items_.end()) {
items_.emplace_back(new Item(bus_name, object_path, config_));
on_add_(items_.back());
}
2018-10-04 16:03:01 +00:00
}
2019-04-24 10:37:24 +00:00
2019-05-18 23:44:45 +00:00
} // namespace waybar::modules::SNI