2018-08-10 14:26:46 +00:00
|
|
|
#include "modules/custom.hpp"
|
|
|
|
|
2018-08-18 15:27:40 +00:00
|
|
|
waybar::modules::Custom::Custom(std::string name, Json::Value config)
|
|
|
|
: ALabel(std::move(config)), name_(std::move(name))
|
2018-08-10 14:26:46 +00:00
|
|
|
{
|
2018-08-16 12:29:41 +00:00
|
|
|
if (!config_["exec"]) {
|
|
|
|
throw std::runtime_error(name_ + " has no exec path.");
|
|
|
|
}
|
|
|
|
uint32_t interval = config_["interval"] ? config_["inveral"].asUInt() : 30;
|
|
|
|
thread_ = [this, interval] {
|
2018-08-18 15:27:40 +00:00
|
|
|
bool can_update = true;
|
|
|
|
if (config_["exec-if"]) {
|
|
|
|
auto pid = fork();
|
|
|
|
int res = 0;
|
|
|
|
if (pid == 0) {
|
|
|
|
std::istringstream iss(config_["exec-if"].asString());
|
|
|
|
std::vector<char*> av;
|
|
|
|
for (std::string s; iss >> s;) {
|
|
|
|
// Need to copy otherwise values are the same
|
|
|
|
char *str = new char[s.size() + 1];
|
|
|
|
memcpy(str, s.c_str(), s.size() + 1);
|
|
|
|
av.push_back(str);
|
|
|
|
}
|
|
|
|
av.push_back(0);
|
|
|
|
execvp(av.front(), av.data());
|
|
|
|
_exit(127);
|
|
|
|
} else if (pid > 0 && waitpid(pid, &res, 0) != -1
|
|
|
|
&& WEXITSTATUS(res) != 0) {
|
|
|
|
can_update = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (can_update) {
|
|
|
|
Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Custom::update));
|
|
|
|
}
|
2018-08-16 12:29:41 +00:00
|
|
|
thread_.sleep_for(chrono::seconds(interval));
|
2018-08-10 14:26:46 +00:00
|
|
|
};
|
2018-08-18 09:43:48 +00:00
|
|
|
}
|
2018-08-10 14:26:46 +00:00
|
|
|
|
|
|
|
auto waybar::modules::Custom::update() -> void
|
|
|
|
{
|
2018-08-16 12:29:41 +00:00
|
|
|
std::array<char, 128> buffer = {0};
|
2018-08-10 14:26:46 +00:00
|
|
|
std::string output;
|
2018-08-16 12:29:41 +00:00
|
|
|
std::shared_ptr<FILE> fp(popen(config_["exec"].asCString(), "r"), pclose);
|
2018-08-10 14:26:46 +00:00
|
|
|
if (!fp) {
|
2018-08-16 12:29:41 +00:00
|
|
|
std::cerr << name_ + " can't exec " + config_["exec"].asString() << std::endl;
|
2018-08-10 14:26:46 +00:00
|
|
|
return;
|
|
|
|
}
|
2018-08-16 12:29:41 +00:00
|
|
|
|
|
|
|
while (feof(fp.get()) == 0) {
|
|
|
|
if (fgets(buffer.data(), 128, fp.get()) != nullptr) {
|
2018-08-10 14:26:46 +00:00
|
|
|
output += buffer.data();
|
2018-08-16 12:29:41 +00:00
|
|
|
}
|
2018-08-10 14:26:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Remove last newline
|
|
|
|
if (!output.empty() && output[output.length()-1] == '\n') {
|
|
|
|
output.erase(output.length()-1);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Hide label if output is empty
|
2018-08-10 15:05:12 +00:00
|
|
|
if (output.empty()) {
|
2018-08-16 12:29:41 +00:00
|
|
|
label_.hide();
|
2018-08-16 15:59:45 +00:00
|
|
|
label_.set_name("");
|
2018-08-10 15:05:12 +00:00
|
|
|
} else {
|
2018-08-16 12:29:41 +00:00
|
|
|
label_.set_name("custom-" + name_);
|
|
|
|
auto format = config_["format"] ? config_["format"].asString() : "{}";
|
2018-08-16 15:09:51 +00:00
|
|
|
auto str = fmt::format(format, output);
|
|
|
|
label_.set_text(str);
|
|
|
|
label_.set_tooltip_text(str);
|
2018-08-16 12:29:41 +00:00
|
|
|
label_.show();
|
2018-08-10 14:26:46 +00:00
|
|
|
}
|
|
|
|
}
|