Waybar/src/modules/custom.cpp

147 lines
3.2 KiB
C++
Raw Normal View History

#include "modules/custom.hpp"
waybar::modules::Custom::Custom(const std::string name,
2018-08-20 12:50:45 +00:00
const Json::Value& config)
2018-12-08 11:58:47 +00:00
: ALabel(config, "{}"), name_(name), fp_(nullptr)
{
if (config_["exec"].isString()) {
if (interval_.count() > 0) {
delayWorker();
} else {
continuousWorker();
}
2018-09-18 21:15:37 +00:00
} else {
update();
2018-09-18 21:15:37 +00:00
}
2018-08-20 12:50:45 +00:00
}
2018-12-08 11:58:47 +00:00
waybar::modules::Custom::~Custom()
{
if (fp_) {
pclose(fp_);
fp_ = nullptr;
}
}
2018-09-18 21:15:37 +00:00
void waybar::modules::Custom::delayWorker()
2018-08-20 12:50:45 +00:00
{
2018-11-23 10:57:37 +00:00
thread_ = [this] {
2018-08-18 15:27:40 +00:00
bool can_update = true;
2018-10-26 07:27:16 +00:00
if (config_["exec-if"].isString()) {
auto res = waybar::util::command::exec(config_["exec-if"].asString());
if (res.exit_code != 0) {
2018-08-18 15:27:40 +00:00
can_update = false;
label_.hide();
label_.set_name("");
2018-08-18 15:27:40 +00:00
}
}
if (can_update) {
output_ = waybar::util::command::exec(config_["exec"].asString());
2018-08-20 12:50:45 +00:00
dp.emit();
2018-08-18 15:27:40 +00:00
}
2018-11-23 10:57:37 +00:00
thread_.sleep_for(interval_);
};
}
2018-09-18 21:15:37 +00:00
void waybar::modules::Custom::continuousWorker()
{
auto cmd = config_["exec"].asString();
2018-12-08 11:58:47 +00:00
fp_ = popen(cmd.c_str(), "r");
if (!fp_) {
2018-09-18 21:15:37 +00:00
throw std::runtime_error("Unable to open " + cmd);
}
2018-12-08 11:58:47 +00:00
thread_ = [this] {
2018-09-18 21:15:37 +00:00
char* buff = nullptr;
size_t len = 0;
2018-12-08 11:58:47 +00:00
if (getline(&buff, &len, fp_) == -1) {
pclose(fp_);
fp_ = nullptr;
2018-09-18 21:15:37 +00:00
thread_.stop();
output_ = { 1, "" };
2018-12-03 10:13:56 +00:00
std::cerr << name_ + " just stopped, is it endless?" << std::endl;
2018-09-18 21:15:37 +00: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-08-16 12:29:41 +00:00
label_.hide();
label_.set_name("");
} else {
label_.set_name("custom-" + name_);
if (config_["return-type"].asString() == "json") {
parseOutputJson();
} else {
parseOutputRaw();
}
auto str = fmt::format(format_, text_);
2018-11-21 19:49:09 +00: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-08-16 12:29:41 +00:00
label_.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++;
}
}
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();
break;
}
}