Waybar/src/modules/custom.cpp

255 lines
6.8 KiB
C++
Raw Normal View History

#include "modules/custom.hpp"
2020-05-25 07:21:04 +00:00
2019-05-18 23:44:45 +00:00
#include <spdlog/spdlog.h>
#include "util/scope_guard.hpp"
waybar::modules::Custom::Custom(const std::string& name, const std::string& id,
const Json::Value& config)
2022-11-24 11:28:52 +00:00
: ALabel(config, "custom-" + name, id, "{}"),
name_(name),
id_(id),
percentage_(0),
fp_(nullptr),
pid_(-1) {
2020-05-25 07:21:04 +00:00
dp.emit();
2023-09-18 21:56:14 +00:00
if (!config_["signal"].empty() && config_["interval"].empty()) {
waitingWorker();
} else if (interval_.count() > 0) {
2020-05-24 16:27:10 +00:00
delayWorker();
} else if (config_["exec"].isString()) {
continuousWorker();
}
2018-08-20 12:50:45 +00:00
}
2019-04-18 15:52:00 +00:00
waybar::modules::Custom::~Custom() {
if (pid_ != -1) {
2020-05-22 18:57:41 +00:00
killpg(pid_, SIGTERM);
waitpid(pid_, NULL, 0);
pid_ = -1;
2018-12-08 11:58:47 +00:00
}
}
2019-04-18 15:52:00 +00:00
void waybar::modules::Custom::delayWorker() {
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()) {
2020-05-24 16:27:10 +00:00
output_ = util::command::execNoRead(config_["exec-if"].asString());
if (output_.exit_code != 0) {
2018-08-18 15:27:40 +00:00
can_update = false;
2020-05-25 07:21:04 +00:00
dp.emit();
2018-08-18 15:27:40 +00:00
}
}
if (can_update) {
2020-05-24 16:27:10 +00:00
if (config_["exec"].isString()) {
output_ = 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_);
};
}
2019-04-18 15:52:00 +00:00
void waybar::modules::Custom::continuousWorker() {
2018-09-18 21:15:37 +00:00
auto cmd = config_["exec"].asString();
pid_ = -1;
fp_ = util::command::open(cmd, pid_);
2018-12-08 11:58:47 +00:00
if (!fp_) {
2018-09-18 21:15:37 +00:00
throw std::runtime_error("Unable to open " + cmd);
}
2020-05-24 19:33:38 +00:00
thread_ = [this, cmd] {
2022-04-06 06:37:19 +00:00
char* buff = nullptr;
waybar::util::ScopeGuard buff_deleter([buff]() {
if (buff) {
free(buff);
}
});
2018-09-18 21:15:37 +00:00
size_t len = 0;
2018-12-08 11:58:47 +00:00
if (getline(&buff, &len, fp_) == -1) {
int exit_code = 1;
if (fp_) {
exit_code = WEXITSTATUS(util::command::close(fp_, pid_));
fp_ = nullptr;
}
if (exit_code != 0) {
2019-04-18 15:52:00 +00:00
output_ = {exit_code, ""};
dp.emit();
2019-05-18 23:44:45 +00:00
spdlog::error("{} stopped unexpectedly, is it endless?", name_);
}
2020-03-25 21:30:22 +00:00
if (config_["restart-interval"].isUInt()) {
2020-05-24 19:33:38 +00:00
pid_ = -1;
thread_.sleep_for(std::chrono::seconds(config_["restart-interval"].asUInt()));
2020-05-24 19:33:38 +00:00
fp_ = util::command::open(cmd, pid_);
if (!fp_) {
throw std::runtime_error("Unable to open " + cmd);
}
} else {
thread_.stop();
return;
}
2020-05-24 19:33:38 +00:00
} else {
std::string output = buff;
2018-09-18 21:15:37 +00:00
2020-05-24 19:33:38 +00:00
// Remove last newline
if (!output.empty() && output[output.length() - 1] == '\n') {
output.erase(output.length() - 1);
}
output_ = {0, output};
dp.emit();
2018-09-18 21:15:37 +00:00
}
};
}
2023-09-18 21:56:14 +00:00
void waybar::modules::Custom::waitingWorker() {
thread_ = [this] {
bool can_update = true;
if (config_["exec-if"].isString()) {
output_ = util::command::execNoRead(config_["exec-if"].asString());
if (output_.exit_code != 0) {
can_update = false;
dp.emit();
}
}
if (can_update) {
if (config_["exec"].isString()) {
output_ = util::command::exec(config_["exec"].asString());
}
dp.emit();
}
thread_.sleep();
};
}
2019-05-22 10:22:56 +00:00
void waybar::modules::Custom::refresh(int sig) {
2019-04-18 15:52:00 +00:00
if (sig == SIGRTMIN + config_["signal"].asInt()) {
2019-03-18 17:46:44 +00:00
thread_.wake_up();
}
}
void waybar::modules::Custom::handleEvent() {
if (!config_["exec-on-event"].isBool() || config_["exec-on-event"].asBool()) {
thread_.wake_up();
}
}
2019-05-18 14:07:55 +00:00
bool waybar::modules::Custom::handleScroll(GdkEventScroll* e) {
2022-11-24 11:28:52 +00:00
auto ret = ALabel::handleScroll(e);
handleEvent();
2019-05-18 14:07:55 +00:00
return ret;
}
bool waybar::modules::Custom::handleToggle(GdkEventButton* const& e) {
2022-11-24 11:28:52 +00:00
auto ret = ALabel::handleToggle(e);
handleEvent();
2019-05-18 14:07:55 +00:00
return ret;
}
2019-04-18 15:52:00 +00:00
auto waybar::modules::Custom::update() -> void {
// Hide label if output is empty
2020-05-25 07:21:04 +00:00
if ((config_["exec"].isString() || config_["exec-if"].isString()) &&
(output_.out.empty() || output_.exit_code != 0)) {
2018-12-18 16:30:54 +00:00
event_box_.hide();
} else {
if (config_["return-type"].asString() == "json") {
parseOutputJson();
} else {
parseOutputRaw();
}
auto str = fmt::format(fmt::runtime(format_), text_, fmt::arg("alt", alt_),
2019-06-03 07:50:35 +00:00
fmt::arg("icon", getIcon(percentage_, alt_)),
fmt::arg("percentage", percentage_));
if (str.empty()) {
2019-05-26 22:05:21 +00:00
event_box_.hide();
} else {
2022-11-24 11:28:52 +00:00
label_.set_markup(str);
2019-05-26 22:08:16 +00:00
if (tooltipEnabled()) {
if (text_ == tooltip_) {
2022-11-24 11:28:52 +00:00
if (label_.get_tooltip_markup() != str) {
label_.set_tooltip_markup(str);
}
2019-05-26 22:08:16 +00:00
} else {
2022-11-24 11:28:52 +00:00
if (label_.get_tooltip_markup() != tooltip_) {
label_.set_tooltip_markup(tooltip_);
}
2019-05-26 22:08:16 +00:00
}
}
2022-11-24 11:28:52 +00:00
auto classes = label_.get_style_context()->list_classes();
2019-05-26 22:08:16 +00:00
for (auto const& c : classes) {
if (c == id_) continue;
2022-11-24 11:28:52 +00:00
label_.get_style_context()->remove_class(c);
2019-05-26 22:08:16 +00:00
}
for (auto const& c : class_) {
2022-11-24 11:28:52 +00:00
label_.get_style_context()->add_class(c);
2019-05-26 22:08:16 +00:00
}
2022-11-24 11:28:52 +00:00
label_.get_style_context()->add_class("flat");
label_.get_style_context()->add_class("text-button");
2019-05-26 22:05:21 +00:00
event_box_.show();
}
}
2020-04-12 16:30:21 +00:00
// Call parent update
2022-11-24 11:28:52 +00:00
ALabel::update();
}
2019-04-18 15:52:00 +00:00
void waybar::modules::Custom::parseOutputRaw() {
std::istringstream output(output_.out);
2022-04-06 06:37:19 +00:00
std::string line;
int i = 0;
while (getline(output, line)) {
Glib::ustring validated_line = line;
if(!validated_line.validate()) {
validated_line = validated_line.make_valid();
}
if (i == 0) {
2019-03-01 16:02:50 +00:00
if (config_["escape"].isBool() && config_["escape"].asBool()) {
text_ = Glib::Markup::escape_text(validated_line);
2019-03-01 16:02:50 +00:00
} else {
text_ = validated_line;
2019-03-01 16:02:50 +00:00
}
tooltip_ = validated_line;
2019-04-15 08:18:27 +00:00
class_.clear();
} else if (i == 1) {
tooltip_ = validated_line;
} else if (i == 2) {
class_.push_back(validated_line);
} else {
break;
}
i++;
}
}
2019-04-18 15:52:00 +00:00
void waybar::modules::Custom::parseOutputJson() {
std::istringstream output(output_.out);
2022-04-06 06:37:19 +00:00
std::string line;
2019-04-15 08:18:27 +00:00
class_.clear();
while (getline(output, line)) {
auto parsed = parser_.parse(line);
2019-03-01 16:02:50 +00:00
if (config_["escape"].isBool() && config_["escape"].asBool()) {
text_ = Glib::Markup::escape_text(parsed["text"].asString());
} else {
text_ = parsed["text"].asString();
}
if (config_["escape"].isBool() && config_["escape"].asBool()) {
alt_ = Glib::Markup::escape_text(parsed["alt"].asString());
} else {
alt_ = parsed["alt"].asString();
}
tooltip_ = parsed["tooltip"].asString();
2019-04-15 08:18:27 +00:00
if (parsed["class"].isString()) {
class_.push_back(parsed["class"].asString());
} else if (parsed["class"].isArray()) {
2019-04-18 15:52:00 +00:00
for (auto const& c : parsed["class"]) {
2019-04-15 08:18:27 +00:00
class_.push_back(c.asString());
}
}
if (!parsed["percentage"].asString().empty() && parsed["percentage"].isNumeric()) {
2023-01-23 08:25:02 +00:00
percentage_ = (int)lround(parsed["percentage"].asFloat());
} else {
percentage_ = 0;
}
break;
}
}