polybar/include/utils/file.hpp

118 lines
2.3 KiB
C++
Raw Normal View History

2016-06-15 03:32:35 +00:00
#pragma once
2017-01-11 02:07:28 +00:00
#include <streambuf>
2016-06-15 03:32:35 +00:00
#include "common.hpp"
#include "utils/factory.hpp"
2016-06-15 03:32:35 +00:00
2016-11-19 05:22:44 +00:00
POLYBAR_NS
2016-06-15 03:32:35 +00:00
class file_ptr {
public:
explicit file_ptr(const string& path, const string& mode = "a+");
~file_ptr();
2016-12-26 09:29:32 +00:00
explicit operator bool();
operator bool() const;
2016-06-15 03:32:35 +00:00
2016-12-26 09:29:32 +00:00
explicit operator FILE*();
operator FILE*() const;
2016-06-15 03:32:35 +00:00
2016-12-26 09:29:32 +00:00
explicit operator int();
operator int() const;
2016-12-15 02:30:41 +00:00
2016-12-26 09:29:32 +00:00
private:
FILE* m_ptr = nullptr;
string m_path;
string m_mode;
};
2016-06-15 03:32:35 +00:00
class file_descriptor {
public:
explicit file_descriptor(const string& path, int flags = 0);
explicit file_descriptor(int fd, bool autoclose = true);
~file_descriptor();
2016-12-26 09:29:32 +00:00
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};
2016-12-26 09:29:32 +00:00
};
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);
2017-01-14 08:42:46 +00:00
setp(m_out, m_out + bufsize - 1);
}
2016-12-26 09:29:32 +00:00
~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;
2017-01-14 08:42:46 +00:00
enum { bufsize = 1024 };
char m_out[bufsize];
char m_in[bufsize - 1];
2016-12-26 09:29:32 +00:00
};
2017-01-11 02:07:28 +00:00
template <typename StreamType>
2016-12-26 09:29:32 +00:00
class fd_stream : public StreamType {
public:
using type = fd_stream<StreamType>;
template <typename... Args>
explicit fd_stream(Args&&... args) : StreamType(nullptr), m_buf(forward<Args>(args)...) {
2016-12-26 09:29:32 +00:00
StreamType::rdbuf(&m_buf);
}
explicit operator int() {
return static_cast<const type&>(*this);
}
operator int() const {
return m_buf;
}
2016-06-15 03:32:35 +00:00
protected:
2016-12-26 09:29:32 +00:00
fd_streambuf m_buf;
};
2016-06-15 03:32:35 +00:00
namespace file_util {
2016-11-25 12:55:15 +00:00
bool exists(const string& filename);
string pick(const vector<string>& filenames);
string contents(const string& filename);
bool is_fifo(const string& filename);
vector<string> glob(string pattern);
const string expand(const string& path);
template <typename... Args>
decltype(auto) make_file_descriptor(Args&&... args) {
return factory_util::unique<file_descriptor>(forward<Args>(args)...);
}
2016-06-15 03:32:35 +00:00
}
2016-11-19 05:22:44 +00:00
POLYBAR_NS_END