waybar/src/modules/custom.cpp

165 lines
3.8 KiB
C++
Raw Normal View History

#include "modules/custom.hpp"
2018-12-18 17:30:54 +01:00
waybar::modules::Custom::Custom(const std::string& name,
2018-08-20 14:50:45 +02:00
const Json::Value& config)
2018-12-08 12:58:47 +01:00
: ALabel(config, "{}"), name_(name), fp_(nullptr)
{
2018-12-18 17:30:54 +01:00
label_.set_name("custom-" + name_);
if (config_["exec"].isString()) {
if (interval_.count() > 0) {
delayWorker();
} else {
continuousWorker();
}
2018-09-18 23:15:37 +02:00
}
2018-12-18 17:30:54 +01:00
dp.emit();
2018-08-20 14:50:45 +02:00
}
2018-12-08 12:58:47 +01:00
waybar::modules::Custom::~Custom()
{
if (fp_) {
pclose(fp_);
fp_ = nullptr;
}
}
2018-09-18 23:15:37 +02:00
void waybar::modules::Custom::delayWorker()
2018-08-20 14:50:45 +02:00
{
2018-11-23 11:57:37 +01:00
thread_ = [this] {
2018-08-18 17:27:40 +02:00
bool can_update = true;
2018-10-26 09:27:16 +02:00
if (config_["exec-if"].isString()) {
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;
2018-12-18 17:30:54 +01:00
event_box_.hide();
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-11-23 11:57:37 +01:00
thread_.sleep_for(interval_);
};
}
2018-09-18 23:15:37 +02:00
void waybar::modules::Custom::continuousWorker()
{
auto cmd = config_["exec"].asString();
2018-12-08 12:58:47 +01:00
fp_ = popen(cmd.c_str(), "r");
if (!fp_) {
2018-09-18 23:15:37 +02:00
throw std::runtime_error("Unable to open " + cmd);
}
2018-12-08 12:58:47 +01:00
thread_ = [this] {
2018-09-18 23:15:37 +02:00
char* buff = nullptr;
size_t len = 0;
2018-12-08 12:58:47 +01:00
if (getline(&buff, &len, fp_) == -1) {
if (fp_) {
pclose(fp_);
fp_ = nullptr;
}
2018-09-18 23:15:37 +02:00
thread_.stop();
output_ = { 1, "" };
2018-12-03 11:13:56 +01:00
std::cerr << name_ + " just stopped, is it endless?" << std::endl;
2018-09-18 23:15:37 +02:00
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 (config_["exec"].isString() && (output_.out.empty() || output_.exit_code != 0)) {
2018-12-18 17:30:54 +01:00
event_box_.hide();
} else {
if (config_["return-type"].asString() == "json") {
parseOutputJson();
} else {
parseOutputRaw();
}
auto str = fmt::format(format_, text_,
fmt::arg("icon", getIcon(percentage_)),
fmt::arg("percentage", percentage_));
2018-11-21 20:49:09 +01:00
label_.set_markup(str);
if (text_ == tooltip_) {
label_.set_tooltip_text(str);
} else {
label_.set_tooltip_text(tooltip_);
}
if (class_ != "") {
if (prevclass_ != "") {
label_.get_style_context()->remove_class(prevclass_);
}
label_.get_style_context()->add_class(class_);
prevclass_ = class_;
} else {
label_.get_style_context()->remove_class(prevclass_);
prevclass_ = "";
}
2018-12-18 17:30:54 +01:00
event_box_.show();
}
}
void waybar::modules::Custom::parseOutputRaw()
{
std::istringstream output(output_.out);
std::string line;
int i = 0;
while (getline(output, line)) {
if (i == 0) {
text_ = line;
tooltip_ = line;
class_ = "";
} else if (i == 1) {
tooltip_ = line;
} else if (i == 2) {
class_ = line;
} else {
break;
}
i++;
}
}
bool waybar::modules::Custom::isInteger(const std::string& n)
{
if (std::isdigit(n[0]) || (n.size() > 1 && (n[0] == '-' || n[0] == '+'))) {
for (std::string::size_type i{ 1 }; i < n.size(); ++i) {
if (!std::isdigit(n[i])) {
return false;
}
}
return true;
}
return false;
}
void waybar::modules::Custom::parseOutputJson()
{
std::istringstream output(output_.out);
std::string line;
while (getline(output, line)) {
auto parsed = parser_.parse(line);
text_ = parsed["text"].asString();
tooltip_ = parsed["tooltip"].asString();
class_ = parsed["class"].asString();
if (!parsed["percentage"].asString().empty() && isInteger(parsed["percentage"].asString())) {
percentage_ = std::stoi(parsed["percentage"].asString(), nullptr);
} else {
percentage_ = 0;
}
break;
}
}