polybar/include/adapters/script_runner.hpp

60 lines
1.1 KiB
C++

#pragma once
#include <atomic>
#include <chrono>
#include <mutex>
#include "common.hpp"
#include "components/logger.hpp"
POLYBAR_NS
class script_runner {
public:
struct data {
int counter{0};
int pid{-1};
int exit_status{0};
string output;
};
using on_update = std::function<void(const data&)>;
using interval = std::chrono::duration<double>;
script_runner(on_update on_update, const string& exec, const string& exec_if, bool tail, interval interval_success,
interval interval_fail, const vector<pair<string, string>>& env);
bool check_condition() const;
interval process();
void clear_output();
void stop();
bool is_stopping() const;
protected:
bool set_output(string&&);
bool set_exit_status(int);
interval run_tail();
interval run();
private:
const logger& m_log;
const on_update m_on_update;
const string m_exec;
const string m_exec_if;
const bool m_tail;
const interval m_interval_success;
const interval m_interval_fail;
const vector<pair<string, string>> m_env;
data m_data;
std::atomic_bool m_stopping{false};
};
POLYBAR_NS_END