Reload all devices after wake from suspend

This commit is contained in:
Erik Reider 2022-03-15 22:49:40 +01:00
parent 5396ffb71f
commit d47073ef13
3 changed files with 46 additions and 4 deletions

View File

@ -25,6 +25,10 @@ class UPower : public AModule {
static void deviceAdded_cb(UpClient *client, UpDevice *device, gpointer data); static void deviceAdded_cb(UpClient *client, UpDevice *device, gpointer data);
static void deviceRemoved_cb(UpClient *client, const gchar *object_path, gpointer data); static void deviceRemoved_cb(UpClient *client, const gchar *object_path, gpointer data);
static void deviceNotify_cb(gpointer data); static void deviceNotify_cb(gpointer data);
static void prepareForSleep_cb(GDBusConnection *system_bus, const gchar *sender_name,
const gchar *object_path, const gchar *interface_name,
const gchar *signal_name, GVariant *parameters,
gpointer user_data);
void removeDevice(const std::string devicePath); void removeDevice(const std::string devicePath);
void addDevice(UpDevice *device); void addDevice(UpDevice *device);
void setDisplayDevice(); void setDisplayDevice();
@ -41,6 +45,8 @@ class UPower : public AModule {
UpClient *client = NULL; UpClient *client = NULL;
UpDevice *displayDevice = NULL; UpDevice *displayDevice = NULL;
std::map<std::string, UpDevice *> devices; std::map<std::string, UpDevice *> devices;
guint login1_id;
GDBusConnection *login1_connection;
}; };
} // namespace waybar::modules } // namespace waybar::modules

View File

@ -86,7 +86,7 @@ wayland_cursor = dependency('wayland-cursor')
wayland_protos = dependency('wayland-protocols') wayland_protos = dependency('wayland-protocols')
gtkmm = dependency('gtkmm-3.0', version : ['>=3.22.0']) gtkmm = dependency('gtkmm-3.0', version : ['>=3.22.0'])
dbusmenu_gtk = dependency('dbusmenu-gtk3-0.4', required: get_option('dbusmenu-gtk')) dbusmenu_gtk = dependency('dbusmenu-gtk3-0.4', required: get_option('dbusmenu-gtk'))
giounix = dependency('gio-unix-2.0', required: (get_option('dbusmenu-gtk').enabled() or get_option('logind').enabled())) giounix = dependency('gio-unix-2.0', required: (get_option('dbusmenu-gtk').enabled() or get_option('logind').enabled() or get_option('upower_glib').enabled()))
jsoncpp = dependency('jsoncpp') jsoncpp = dependency('jsoncpp')
sigcpp = dependency('sigc++-2.0') sigcpp = dependency('sigc++-2.0')
libepoll = dependency('epoll-shim', required: false) libepoll = dependency('epoll-shim', required: false)
@ -204,7 +204,7 @@ if libnl.found() and libnlgen.found()
src_files += 'src/modules/network.cpp' src_files += 'src/modules/network.cpp'
endif endif
if upower_glib.found() if (upower_glib.found() and giounix.found() and not get_option('logind').disabled())
add_project_arguments('-DHAVE_UPOWER', language: 'cpp') add_project_arguments('-DHAVE_UPOWER', language: 'cpp')
src_files += 'src/modules/upower/upower.cpp' src_files += 'src/modules/upower/upower.cpp'
endif endif

View File

@ -1,5 +1,7 @@
#include "modules/upower/upower.hpp" #include "modules/upower/upower.hpp"
#include <gio/gio.h>
#include <iostream> #include <iostream>
#include <map> #include <map>
#include <string> #include <string>
@ -35,7 +37,22 @@ UPower::UPower(const std::string& id, const Json::Value& config)
throw std::runtime_error("Unable to create UPower client!"); throw std::runtime_error("Unable to create UPower client!");
} }
// TODO: Connect to login1 prepare_for_sleep signal // Connect to Login1 PrepareForSleep signal
login1_connection = g_bus_get_sync(G_BUS_TYPE_SYSTEM, NULL, &error);
if (!login1_connection) {
throw std::runtime_error("Unable to connect to the SYSTEM Bus!...");
} else {
login1_id = g_dbus_connection_signal_subscribe(login1_connection,
"org.freedesktop.login1",
"org.freedesktop.login1.Manager",
"PrepareForSleep",
"/org/freedesktop/login1",
NULL,
G_DBUS_SIGNAL_FLAGS_NONE,
prepareForSleep_cb,
this,
NULL);
}
g_signal_connect(client, "device-added", G_CALLBACK(deviceAdded_cb), this); g_signal_connect(client, "device-added", G_CALLBACK(deviceAdded_cb), this);
g_signal_connect(client, "device-removed", G_CALLBACK(deviceRemoved_cb), this); g_signal_connect(client, "device-removed", G_CALLBACK(deviceRemoved_cb), this);
@ -46,7 +63,13 @@ UPower::UPower(const std::string& id, const Json::Value& config)
dp.emit(); dp.emit();
} }
UPower::~UPower() {} UPower::~UPower() {
if (client != NULL) g_object_unref(client);
if (login1_id > 0) {
g_dbus_connection_signal_unsubscribe(login1_connection, login1_id);
login1_id = 0;
}
}
void UPower::deviceAdded_cb(UpClient* client, UpDevice* device, gpointer data) { void UPower::deviceAdded_cb(UpClient* client, UpDevice* device, gpointer data) {
UPower* up = static_cast<UPower*>(data); UPower* up = static_cast<UPower*>(data);
@ -67,6 +90,19 @@ void UPower::deviceNotify_cb(gpointer data) {
// Update the widget // Update the widget
up->dp.emit(); up->dp.emit();
} }
void UPower::prepareForSleep_cb(GDBusConnection* system_bus, const gchar* sender_name,
const gchar* object_path, const gchar* interface_name,
const gchar* signal_name, GVariant* parameters, gpointer data) {
if (g_variant_is_of_type(parameters, G_VARIANT_TYPE("(b)"))) {
gboolean sleeping;
g_variant_get(parameters, "(b)", &sleeping);
if (!sleeping) {
UPower* up = static_cast<UPower*>(data);
up->resetDevices();
}
}
}
void UPower::removeDevice(const std::string devicePath) { devices.erase(devicePath); } void UPower::removeDevice(const std::string devicePath) { devices.erase(devicePath); }