waybar/src/modules/battery.cpp

142 lines
4.2 KiB
C++
Raw Normal View History

2018-08-08 23:54:58 +02:00
#include "modules/battery.hpp"
2018-08-20 14:50:45 +02:00
waybar::modules::Battery::Battery(const Json::Value& config)
2018-11-23 11:57:37 +01:00
: ALabel(config, "{capacity}%", 60)
2018-08-08 23:54:58 +02:00
{
try {
2018-10-26 09:27:16 +02:00
if (config_["bat"].isString()) {
2018-10-25 17:30:26 +02:00
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);
}
2018-08-16 14:29:41 +02:00
}
2018-08-08 23:54:58 +02:00
}
} catch (fs::filesystem_error &e) {
2018-08-13 14:05:13 +02:00
throw std::runtime_error(e.what());
2018-08-08 23:54:58 +02:00
}
2018-08-16 14:29:41 +02:00
if (batteries_.empty()) {
2018-10-26 09:27:16 +02:00
if (config_["bat"].isString()) {
2018-10-25 17:30:26 +02:00
throw std::runtime_error("No battery named " + config_["bat"].asString());
}
2018-08-13 14:05:13 +02:00
throw std::runtime_error("No batteries.");
2018-08-16 14:29:41 +02:00
}
2018-08-19 13:39:57 +02:00
fd_ = inotify_init1(IN_CLOEXEC);
if (fd_ == -1) {
2018-08-13 14:05:13 +02:00
throw std::runtime_error("Unable to listen batteries.");
2018-08-16 14:29:41 +02:00
}
2018-09-04 23:50:08 +02:00
for (auto const& bat : batteries_) {
2018-08-19 13:39:57 +02:00
inotify_add_watch(fd_, (bat / "uevent").c_str(), IN_ACCESS);
2018-08-16 14:29:41 +02:00
}
2018-08-20 14:50:45 +02:00
worker();
}
waybar::modules::Battery::~Battery()
{
close(fd_);
}
void waybar::modules::Battery::worker()
{
// Trigger first values
update();
2018-11-23 11:57:37 +01:00
thread_timer_ = [this] {
thread_.sleep_for(interval_);
dp.emit();
};
2018-08-19 13:39:57 +02:00
thread_ = [this] {
2018-08-17 14:24:00 +02:00
struct inotify_event event = {0};
2018-08-19 13:39:57 +02:00
int nbytes = read(fd_, &event, sizeof(event));
2018-08-16 14:29:41 +02:00
if (nbytes != sizeof(event)) {
2018-08-13 14:05:13 +02:00
return;
2018-08-16 14:29:41 +02:00
}
// TODO: don't stop timer for now since there is some bugs :?
2018-11-16 10:02:12 +01:00
// thread_timer_.stop();
2018-08-20 14:50:45 +02:00
dp.emit();
2018-08-08 23:54:58 +02:00
};
}
2018-11-23 11:57:37 +01:00
const std::tuple<uint8_t, std::string> waybar::modules::Battery::getInfos() const
2018-08-08 23:54:58 +02:00
{
try {
2018-08-13 14:05:13 +02:00
uint16_t total = 0;
2018-11-02 22:50:01 +01:00
std::string status = "Unknown";
2018-09-04 23:50:08 +02:00
for (auto const& bat : batteries_) {
2018-08-13 14:05:13 +02:00
uint16_t capacity;
std::string _status;
2018-08-09 11:21:08 +02:00
std::ifstream(bat / "capacity") >> capacity;
std::ifstream(bat / "status") >> _status;
2018-08-16 14:29:41 +02:00
if (_status != "Unknown") {
status = _status;
2018-08-16 14:29:41 +02:00
}
2018-08-13 14:05:13 +02:00
total += capacity;
2018-08-08 23:54:58 +02:00
}
2018-08-16 14:29:41 +02:00
uint16_t capacity = total / batteries_.size();
return {capacity, status};
} catch (const std::exception& e) {
std::cerr << e.what() << std::endl;
return {0, "Unknown"};
}
}
2018-11-23 11:57:37 +01:00
const std::string waybar::modules::Battery::getState(uint8_t capacity) const
{
// Get current state
2018-11-23 11:57:37 +01:00
std::vector<std::pair<std::string, uint8_t>> states;
if (config_["states"].isObject()) {
for (auto it = config_["states"].begin(); it != config_["states"].end(); ++it) {
if (it->isUInt() && it.key().isString()) {
states.push_back({it.key().asString(), it->asUInt()});
}
2018-08-16 14:29:41 +02:00
}
}
// Sort states
std::sort(states.begin(), states.end(), [](auto &a, auto &b) {
return a.second < b.second;
});
2018-11-23 11:57:37 +01:00
std::string valid_state;
for (auto const& state : states) {
2018-11-23 11:57:37 +01:00
if (capacity <= state.second && valid_state.empty()) {
label_.get_style_context()->add_class(state.first);
2018-11-23 11:57:37 +01:00
valid_state = state.first;
2018-08-16 14:29:41 +02:00
} else {
label_.get_style_context()->remove_class(state.first);
2018-08-16 14:29:41 +02:00
}
}
2018-11-23 11:57:37 +01:00
return valid_state;
}
auto waybar::modules::Battery::update() -> void
{
auto [capacity, status] = getInfos();
label_.set_tooltip_text(status);
std::transform(status.begin(), status.end(), status.begin(), ::tolower);
auto format = format_;
2018-11-02 22:08:55 +01:00
auto state = getState(capacity);
label_.get_style_context()->remove_class(old_status_);
label_.get_style_context()->add_class(status);
old_status_ = status;
if (!state.empty() && config_["format-" + status + "-" + state].isString()) {
format = config_["format-" + status + "-" + state].asString();
2018-11-02 22:50:01 +01:00
} else if (config_["format-" + status].isString()) {
format = config_["format-" + status].asString();
2018-11-02 22:50:01 +01:00
} else if (!state.empty() && config_["format-" + state].isString()) {
format = config_["format-" + state].asString();
}
if (format.empty()) {
event_box_.hide();
label_.set_name("");
} else {
event_box_.show();
label_.set_name("battery");
2018-11-21 20:49:09 +01:00
label_.set_markup(fmt::format(format, fmt::arg("capacity", capacity),
fmt::arg("icon", getIcon(capacity))));
2018-08-08 23:54:58 +02:00
}
}