51 lines
1.3 KiB
C++
51 lines
1.3 KiB
C++
|
#include "modules/custom.hpp"
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
waybar::modules::Custom::Custom(std::string name, Json::Value config)
|
||
|
|
: _name(name), _config(config)
|
||
|
|
{
|
||
|
|
_label.get_style_context()->add_class("custom-" + name);
|
||
|
|
if (!_config["exec"]) {
|
||
|
|
std::cerr << name + " has no exec path." << std::endl;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
_thread = [this] {
|
||
|
|
update();
|
||
|
|
int interval = _config["interval"] ? _config["inveral"].asInt() : 30;
|
||
|
|
_thread.sleep_for(chrono::seconds(interval));
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
auto waybar::modules::Custom::update() -> void
|
||
|
|
{
|
||
|
|
std::array<char, 128> buffer;
|
||
|
|
std::string output;
|
||
|
|
std::shared_ptr<FILE> fp(popen(_config["exec"].asCString(), "r"), pclose);
|
||
|
|
if (!fp) {
|
||
|
|
std::cerr << _name + " can't exec " + _config["exec"].asString() << std::endl;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
while (!feof(fp.get())) {
|
||
|
|
if (fgets(buffer.data(), 128, fp.get()) != nullptr)
|
||
|
|
output += buffer.data();
|
||
|
|
}
|
||
|
|
|
||
|
|
// Remove last newline
|
||
|
|
if (!output.empty() && output[output.length()-1] == '\n') {
|
||
|
|
output.erase(output.length()-1);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Hide label if output is empty
|
||
|
|
if (output.empty())
|
||
|
|
_label.hide();
|
||
|
|
else {
|
||
|
|
auto format = _config["format"] ? _config["format"].asString() : "{}";
|
||
|
|
_label.set_text(fmt::format(format, output));
|
||
|
|
_label.show();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
waybar::modules::Custom::operator Gtk::Widget &() {
|
||
|
|
return _label;
|
||
|
|
}
|