127 lines
3.2 KiB
C++
127 lines
3.2 KiB
C++
#include "modules/upower/upower.hpp"
|
|
|
|
#include <iostream>
|
|
#include <map>
|
|
#include <string>
|
|
|
|
namespace waybar::modules {
|
|
UPower::UPower(const std::string& id, const Json::Value& config)
|
|
: AModule(config, "tray", id),
|
|
box_(Gtk::ORIENTATION_HORIZONTAL, 0),
|
|
icon_(),
|
|
label_(),
|
|
devices() {
|
|
box_.pack_start(icon_);
|
|
box_.pack_start(label_);
|
|
event_box_.add(box_);
|
|
|
|
GError* error = NULL;
|
|
client = up_client_new_full(NULL, &error);
|
|
if (client == NULL) {
|
|
throw std::runtime_error("Unable to create UPower client!");
|
|
}
|
|
|
|
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, "notify", G_CALLBACK(deviceNotify_cb), this);
|
|
|
|
resetDevices();
|
|
|
|
dp.emit();
|
|
}
|
|
|
|
UPower::~UPower() {}
|
|
|
|
void UPower::clicked() { printf("CLICK\n"); }
|
|
|
|
void UPower::deviceAdded_cb(UpClient* client, UpDevice* device, gpointer data) {
|
|
UPower* up = static_cast<UPower*>(data);
|
|
up->addDevice(device);
|
|
up->setDisplayDevice();
|
|
// Update the widget
|
|
up->dp.emit();
|
|
}
|
|
void UPower::deviceRemoved_cb(UpClient* client, const gchar* object_path, gpointer data) {
|
|
UPower* up = static_cast<UPower*>(data);
|
|
up->removeDevice(object_path);
|
|
up->setDisplayDevice();
|
|
// Update the widget
|
|
up->dp.emit();
|
|
}
|
|
void UPower::deviceNotify_cb(gpointer data) {
|
|
UPower* up = static_cast<UPower*>(data);
|
|
// Update the widget
|
|
up->dp.emit();
|
|
}
|
|
|
|
void UPower::removeDevice(const std::string devicePath) { devices.erase(devicePath); }
|
|
|
|
void UPower::addDevice(UpDevice* device) {
|
|
if (device) {
|
|
const gchar* objectPath = up_device_get_object_path(device);
|
|
devices[objectPath] = device;
|
|
g_signal_connect(devices[objectPath], "notify", G_CALLBACK(deviceNotify_cb), this);
|
|
}
|
|
}
|
|
|
|
void UPower::setDisplayDevice() {
|
|
displayDevice = up_client_get_display_device(client);
|
|
g_signal_connect(displayDevice, "notify", G_CALLBACK(deviceNotify_cb), this);
|
|
}
|
|
|
|
/** Removes all devices and adds the current devices */
|
|
void UPower::resetDevices() {
|
|
// Removes all devices
|
|
if (devices.size() > 0) {
|
|
auto it = devices.cbegin();
|
|
while (it != devices.cend()) {
|
|
devices.erase(it++);
|
|
}
|
|
}
|
|
|
|
// Adds all devices
|
|
GPtrArray* newDevices = up_client_get_devices2(client);
|
|
for (guint i = 0; i < newDevices->len; i++) {
|
|
UpDevice* device = (UpDevice*)g_ptr_array_index(newDevices, i);
|
|
if (device) addDevice(device);
|
|
}
|
|
|
|
setDisplayDevice();
|
|
|
|
// Update the widget
|
|
dp.emit();
|
|
}
|
|
|
|
auto UPower::update() -> void {
|
|
printf("UPDATE!\n");
|
|
|
|
UpDeviceKind kind;
|
|
UpDeviceState state;
|
|
double percentage;
|
|
gboolean is_power_supply;
|
|
gboolean is_present;
|
|
gchar* icon_name;
|
|
|
|
g_object_get(displayDevice,
|
|
"kind",
|
|
&kind,
|
|
"state",
|
|
&state,
|
|
"is-present",
|
|
&is_present,
|
|
"power-supply",
|
|
&is_power_supply,
|
|
"percentage",
|
|
&percentage,
|
|
"icon-name",
|
|
&icon_name,
|
|
NULL);
|
|
|
|
printf("ICON: %s\n", icon_name);
|
|
|
|
// Call parent update
|
|
AModule::update();
|
|
}
|
|
|
|
} // namespace waybar::modules
|