polybar-dwm/include/utils/file.hpp
Michael Carlberg ccddf052ec refactor(script): Split non-/tail logic
Split the script module into one tailing output module and
one exec once module.

Refs #281
2017-01-01 09:05:18 +01:00

113 lines
2.1 KiB
C++

#pragma once
#include "common.hpp"
#include "utils/factory.hpp"
POLYBAR_NS
class file_ptr {
public:
explicit file_ptr(const string& path, const string& mode = "a+");
~file_ptr();
explicit operator bool();
operator bool() const;
explicit operator FILE*();
operator FILE*() const;
explicit operator int();
operator int() const;
private:
FILE* m_ptr = nullptr;
string m_path;
string m_mode;
};
class file_descriptor {
public:
explicit file_descriptor(const string& path, int flags = 0);
explicit file_descriptor(int fd, bool autoclose = true);
~file_descriptor();
file_descriptor& operator=(const int);
explicit operator bool();
operator bool() const;
explicit operator int();
operator int() const;
protected:
void close();
private:
int m_fd{-1};
bool m_autoclose{true};
};
class fd_streambuf : public std::streambuf {
public:
using traits_type = std::streambuf::traits_type;
template <typename... Args>
explicit fd_streambuf(Args&&... args) : m_fd(forward<Args>(args)...) {
setg(m_in, m_in, m_in);
setp(m_out, m_out + sizeof(m_in));
}
~fd_streambuf();
explicit operator int();
operator int() const;
void open(int fd);
void close();
protected:
int sync();
int overflow(int c);
int underflow();
private:
file_descriptor m_fd;
char m_out[BUFSIZ]{};
char m_in[BUFSIZ - 1]{};
};
template <typename StreamType = std::ostream>
class fd_stream : public StreamType {
public:
using type = fd_stream<StreamType>;
template <typename... Args>
explicit fd_stream(Args&&... args) : m_buf(forward<Args>(args)...) {
StreamType::rdbuf(&m_buf);
}
explicit operator int() {
return static_cast<const type&>(*this);
}
operator int() const {
return m_buf;
}
protected:
fd_streambuf m_buf;
};
namespace file_util {
bool exists(const string& filename);
string pick(const vector<string>& filenames);
string contents(const string& filename);
bool is_fifo(const string& filename);
template <typename... Args>
decltype(auto) make_file_descriptor(Args&&... args) {
return factory_util::unique<file_descriptor>(forward<Args>(args)...);
}
}
POLYBAR_NS_END