Waybar/src/modules/battery.cpp

92 lines
2.6 KiB
C++
Raw Normal View History

2018-08-08 21:54:58 +00:00
#include "modules/battery.hpp"
2018-08-09 11:30:11 +00:00
waybar::modules::Battery::Battery(Json::Value config)
: ALabel(std::move(config))
2018-08-08 21:54:58 +00:00
{
try {
2018-08-16 12:29:41 +00:00
for (auto &node : fs::directory_iterator(data_dir_)) {
2018-08-13 15:11:47 +00:00
if (fs::is_directory(node) && fs::exists(node / "capacity")
2018-08-16 12:29:41 +00:00
&& fs::exists(node / "status") && fs::exists(node / "uevent")) {
batteries_.push_back(node);
}
2018-08-08 21:54:58 +00:00
}
} catch (fs::filesystem_error &e) {
2018-08-13 12:05:13 +00:00
throw std::runtime_error(e.what());
2018-08-08 21:54:58 +00:00
}
2018-08-16 12:29:41 +00:00
if (batteries_.empty()) {
2018-08-13 12:05:13 +00:00
throw std::runtime_error("No batteries.");
2018-08-16 12:29:41 +00:00
}
2018-08-19 11:39:57 +00:00
fd_ = inotify_init1(IN_CLOEXEC);
if (fd_ == -1) {
2018-08-13 12:05:13 +00:00
throw std::runtime_error("Unable to listen batteries.");
2018-08-16 12:29:41 +00:00
}
for (auto &bat : batteries_) {
2018-08-19 11:39:57 +00:00
inotify_add_watch(fd_, (bat / "uevent").c_str(), IN_ACCESS);
2018-08-16 12:29:41 +00:00
}
2018-08-19 11:39:57 +00:00
// Trigger first values
update();
2018-08-16 12:29:41 +00:00
label_.set_name("battery");
2018-08-19 11:39:57 +00:00
thread_.sig_update.connect(sigc::mem_fun(*this, &Battery::update));
thread_ = [this] {
2018-08-17 12:24:00 +00:00
struct inotify_event event = {0};
2018-08-19 11:39:57 +00:00
int nbytes = read(fd_, &event, sizeof(event));
2018-08-16 12:29:41 +00:00
if (nbytes != sizeof(event)) {
2018-08-13 12:05:13 +00:00
return;
2018-08-16 12:29:41 +00:00
}
2018-08-19 18:37:33 +00:00
thread_.emit();
2018-08-08 21:54:58 +00:00
};
}
2018-08-19 11:39:57 +00:00
waybar::modules::Battery::~Battery()
{
close(fd_);
}
2018-08-08 21:54:58 +00:00
auto waybar::modules::Battery::update() -> void
{
try {
2018-08-13 12:05:13 +00:00
uint16_t total = 0;
std::string status;
2018-08-16 12:29:41 +00:00
for (auto &bat : batteries_) {
2018-08-13 12:05:13 +00:00
uint16_t capacity;
std::string _status;
2018-08-09 09:21:08 +00:00
std::ifstream(bat / "capacity") >> capacity;
std::ifstream(bat / "status") >> _status;
2018-08-16 12:29:41 +00:00
if (_status != "Unknown") {
status = _status;
2018-08-16 12:29:41 +00:00
}
2018-08-13 12:05:13 +00:00
total += capacity;
2018-08-08 21:54:58 +00:00
}
2018-08-16 12:29:41 +00:00
uint16_t capacity = total / batteries_.size();
auto format = config_["format"]
? config_["format"].asString() : "{capacity}%";
label_.set_text(fmt::format(format, fmt::arg("capacity", capacity),
fmt::arg("icon", getIcon(capacity))));
label_.set_tooltip_text(status);
bool charging = status == "Charging";
2018-08-16 12:29:41 +00:00
if (charging) {
label_.get_style_context()->add_class("charging");
} else {
label_.get_style_context()->remove_class("charging");
}
auto critical = config_["critical"] ? config_["critical"].asUInt() : 15;
if (capacity <= critical && !charging) {
label_.get_style_context()->add_class("warning");
} else {
label_.get_style_context()->remove_class("warning");
}
2018-08-13 12:05:13 +00:00
} catch (const std::exception& e) {
2018-08-08 21:54:58 +00:00
std::cerr << e.what() << std::endl;
}
}
2018-08-16 12:29:41 +00:00
std::string waybar::modules::Battery::getIcon(uint16_t percentage)
2018-08-11 11:15:31 +00:00
{
2018-08-16 12:29:41 +00:00
if (!config_["format-icons"] || !config_["format-icons"].isArray()) {
return "";
}
auto size = config_["format-icons"].size();
2018-08-13 20:33:07 +00:00
auto idx = std::clamp(percentage / (100 / size), 0U, size - 1);
2018-08-16 12:29:41 +00:00
return config_["format-icons"][idx].asString();
2018-08-11 11:15:31 +00:00
}