mirror of https://github.com/polybar/polybar.git
131 lines
2.5 KiB
C++
131 lines
2.5 KiB
C++
#include "utils/inotify.hpp"
|
|
|
|
#include <unistd.h>
|
|
|
|
#include "errors.hpp"
|
|
#include "utils/memory.hpp"
|
|
|
|
POLYBAR_NS
|
|
|
|
/**
|
|
* Construct inotify watch
|
|
*/
|
|
inotify_watch::inotify_watch(string path) : m_path(move(path)) {}
|
|
|
|
/**
|
|
* Deconstruct inotify watch
|
|
*/
|
|
inotify_watch::~inotify_watch() {
|
|
if (m_wd != -1) {
|
|
inotify_rm_watch(m_fd, m_wd);
|
|
}
|
|
if (m_fd != -1) {
|
|
close(m_fd);
|
|
}
|
|
}
|
|
|
|
inotify_watch::inotify_watch(inotify_watch&& other) noexcept {
|
|
std::swap(m_path, other.m_path);
|
|
std::swap(m_wd, other.m_wd);
|
|
std::swap(m_fd, other.m_fd);
|
|
std::swap(m_mask, other.m_mask);
|
|
}
|
|
|
|
inotify_watch& inotify_watch::operator=(inotify_watch&& other) noexcept {
|
|
std::swap(m_path, other.m_path);
|
|
std::swap(m_wd, other.m_wd);
|
|
std::swap(m_fd, other.m_fd);
|
|
std::swap(m_mask, other.m_mask);
|
|
return *this;
|
|
}
|
|
|
|
/**
|
|
* Attach inotify watch
|
|
*/
|
|
void inotify_watch::attach(int mask) {
|
|
if (m_fd == -1 && (m_fd = inotify_init()) == -1) {
|
|
throw system_error("Failed to allocate inotify fd");
|
|
}
|
|
if ((m_wd = inotify_add_watch(m_fd, m_path.c_str(), mask)) == -1) {
|
|
throw system_error("Failed to attach inotify watch");
|
|
}
|
|
m_mask |= mask;
|
|
}
|
|
|
|
/**
|
|
* Remove inotify watch
|
|
*/
|
|
void inotify_watch::remove(bool force) {
|
|
if (inotify_rm_watch(m_fd, m_wd) == -1 && !force) {
|
|
throw system_error("Failed to remove inotify watch");
|
|
}
|
|
m_wd = -1;
|
|
m_mask = 0;
|
|
}
|
|
|
|
/**
|
|
* Poll the inotify fd for events
|
|
*
|
|
* @brief A wait_ms of -1 blocks until an event is fired
|
|
*/
|
|
bool inotify_watch::poll(int wait_ms) const {
|
|
if (m_fd == -1) {
|
|
return false;
|
|
}
|
|
|
|
struct pollfd fds[1];
|
|
fds[0].fd = m_fd;
|
|
fds[0].events = POLLIN;
|
|
|
|
::poll(fds, 1, wait_ms);
|
|
|
|
return fds[0].revents & POLLIN;
|
|
}
|
|
|
|
/**
|
|
* Get the latest inotify event
|
|
*/
|
|
inotify_event inotify_watch::get_event() const {
|
|
inotify_event event;
|
|
|
|
if (m_fd == -1 || m_wd == -1) {
|
|
return event;
|
|
}
|
|
|
|
event.is_valid = true;
|
|
|
|
char buffer[1024];
|
|
auto bytes = read(m_fd, buffer, 1024);
|
|
auto len = 0;
|
|
|
|
while (len < bytes) {
|
|
auto* e = reinterpret_cast<::inotify_event*>(&buffer[len]);
|
|
|
|
event.filename = e->len ? e->name : m_path;
|
|
event.wd = e->wd;
|
|
event.cookie = e->cookie;
|
|
event.is_dir = e->mask & IN_ISDIR;
|
|
event.mask |= e->mask;
|
|
|
|
len += sizeof(*e) + e->len;
|
|
}
|
|
|
|
return event;
|
|
}
|
|
|
|
/**
|
|
* Get watch file path
|
|
*/
|
|
string inotify_watch::path() const {
|
|
return m_path;
|
|
}
|
|
|
|
/**
|
|
* Get the file descriptor associated with the watch
|
|
*/
|
|
int inotify_watch::get_file_descriptor() const {
|
|
return m_fd;
|
|
}
|
|
|
|
POLYBAR_NS_END
|