107 lines
3.2 KiB
C++
107 lines
3.2 KiB
C++
#include "modules/battery.hpp"
|
|
|
|
waybar::modules::Battery::Battery(const Json::Value& config)
|
|
: ALabel(config, "{capacity}%")
|
|
{
|
|
try {
|
|
if (config_["bat"].isString()) {
|
|
auto dir = data_dir_ / config_["bat"].asString();
|
|
if (fs::is_directory(dir) && fs::exists(dir / "capacity")
|
|
&& fs::exists(dir / "status") && fs::exists(dir / "uevent")) {
|
|
batteries_.push_back(dir);
|
|
}
|
|
} else {
|
|
for (auto const& node : fs::directory_iterator(data_dir_)) {
|
|
if (fs::is_directory(node) && fs::exists(node / "capacity")
|
|
&& fs::exists(node / "status") && fs::exists(node / "uevent")) {
|
|
batteries_.push_back(node);
|
|
}
|
|
}
|
|
}
|
|
} catch (fs::filesystem_error &e) {
|
|
throw std::runtime_error(e.what());
|
|
}
|
|
if (batteries_.empty()) {
|
|
if (config_["bat"].isString()) {
|
|
throw std::runtime_error("No battery named " + config_["bat"].asString());
|
|
}
|
|
throw std::runtime_error("No batteries.");
|
|
}
|
|
fd_ = inotify_init1(IN_CLOEXEC);
|
|
if (fd_ == -1) {
|
|
throw std::runtime_error("Unable to listen batteries.");
|
|
}
|
|
for (auto const& bat : batteries_) {
|
|
inotify_add_watch(fd_, (bat / "uevent").c_str(), IN_ACCESS);
|
|
}
|
|
label_.set_name("battery");
|
|
worker();
|
|
}
|
|
|
|
waybar::modules::Battery::~Battery()
|
|
{
|
|
close(fd_);
|
|
}
|
|
|
|
void waybar::modules::Battery::worker()
|
|
{
|
|
// Trigger first values
|
|
update();
|
|
uint32_t interval = config_["interval"].isUInt() ? config_["interval"].asUInt() : 60;
|
|
threadTimer_ = [this, interval] {
|
|
thread_.sleep_for(chrono::seconds(interval));
|
|
dp.emit();
|
|
};
|
|
thread_ = [this] {
|
|
struct inotify_event event = {0};
|
|
int nbytes = read(fd_, &event, sizeof(event));
|
|
if (nbytes != sizeof(event)) {
|
|
return;
|
|
}
|
|
threadTimer_.stop();
|
|
dp.emit();
|
|
};
|
|
}
|
|
|
|
auto waybar::modules::Battery::update() -> void
|
|
{
|
|
try {
|
|
uint16_t total = 0;
|
|
std::string status;
|
|
for (auto const& bat : batteries_) {
|
|
uint16_t capacity;
|
|
std::string _status;
|
|
std::ifstream(bat / "capacity") >> capacity;
|
|
std::ifstream(bat / "status") >> _status;
|
|
if (_status != "Unknown") {
|
|
status = _status;
|
|
}
|
|
total += capacity;
|
|
}
|
|
uint16_t capacity = total / batteries_.size();
|
|
label_.set_text(fmt::format(format_, fmt::arg("capacity", capacity),
|
|
fmt::arg("icon", getIcon(capacity))));
|
|
label_.set_tooltip_text(status);
|
|
bool charging = status == "Charging";
|
|
if (charging) {
|
|
label_.get_style_context()->add_class("charging");
|
|
} else {
|
|
label_.get_style_context()->remove_class("charging");
|
|
}
|
|
auto warning = config_["warning"].isUInt() ? config_["warning"].asUInt() : 30;
|
|
auto critical = config_["critical"].isUInt() ? config_["critical"].asUInt() : 15;
|
|
if (capacity <= critical && !charging) {
|
|
label_.get_style_context()->add_class("critical");
|
|
label_.get_style_context()->remove_class("warning");
|
|
} else if (capacity <= warning && !charging) {
|
|
label_.get_style_context()->add_class("warning");
|
|
label_.get_style_context()->remove_class("critical");
|
|
} else {
|
|
label_.get_style_context()->remove_class("critical");
|
|
label_.get_style_context()->remove_class("warning");
|
|
}
|
|
} catch (const std::exception& e) {
|
|
std::cerr << e.what() << std::endl;
|
|
}
|
|
}
|