waybar/src/modules/custom.cpp

80 lines
1.9 KiB
C++
Raw Normal View History

#include "modules/custom.hpp"
waybar::modules::Custom::Custom(const std::string name,
2018-08-20 14:50:45 +02:00
const Json::Value& config)
2018-08-27 01:36:25 +02:00
: ALabel(config, "{}"), name_(name)
{
2018-08-16 14:29:41 +02:00
if (!config_["exec"]) {
throw std::runtime_error(name_ + " has no exec path.");
}
2018-09-18 23:15:37 +02:00
if (config_["interval"]) {
delayWorker();
} else {
continuousWorker();
}
2018-08-20 14:50:45 +02:00
}
2018-09-18 23:15:37 +02:00
void waybar::modules::Custom::delayWorker()
2018-08-20 14:50:45 +02:00
{
2018-09-18 23:15:37 +02:00
auto interval = config_["interval"].asUInt();
2018-08-16 14:29:41 +02:00
thread_ = [this, interval] {
2018-08-18 17:27:40 +02:00
bool can_update = true;
if (config_["exec-if"]) {
auto res = waybar::util::command::exec(config_["exec-if"].asString());
if (res.exit_code != 0) {
2018-08-18 17:27:40 +02:00
can_update = false;
label_.hide();
label_.set_name("");
2018-08-18 17:27:40 +02:00
}
}
if (can_update) {
output_ = waybar::util::command::exec(config_["exec"].asString());
2018-08-20 14:50:45 +02:00
dp.emit();
2018-08-18 17:27:40 +02:00
}
2018-08-16 14:29:41 +02:00
thread_.sleep_for(chrono::seconds(interval));
};
}
2018-09-18 23:15:37 +02:00
void waybar::modules::Custom::continuousWorker()
{
auto cmd = config_["exec"].asString();
FILE* fp(popen(cmd.c_str(), "r"));
if (!fp) {
throw std::runtime_error("Unable to open " + cmd);
}
thread_ = [this, fp] {
char* buff = nullptr;
size_t len = 0;
if (getline(&buff, &len, fp) == -1) {
pclose(fp);
2018-09-18 23:15:37 +02:00
thread_.stop();
output_ = { 1, "" };
dp.emit();
return;
}
std::string output = buff;
// Remove last newline
if (!output.empty() && output[output.length()-1] == '\n') {
output.erase(output.length()-1);
}
output_ = { 0, output };
dp.emit();
};
}
auto waybar::modules::Custom::update() -> void
{
// Hide label if output is empty
if (output_.out.empty() || output_.exit_code != 0) {
2018-08-16 14:29:41 +02:00
label_.hide();
label_.set_name("");
} else {
label_.set_name("custom-" + name_);
auto str = fmt::format(format_, output_.out);
2018-08-16 17:09:51 +02:00
label_.set_text(str);
label_.set_tooltip_text(str);
2018-08-16 14:29:41 +02:00
label_.show();
}
}