#include <unistd.h>
-void Poller::add_fd(int fd, std::function<int(int)> handler) {
- pfds.emplace_back(pollfd{.fd = fd, .events = POLLIN, .revents = 0});
- handlers.emplace_back(std::move(handler));
-}
-
-int Poller::check_events() {
- int ret = 0;
- if ((ret = poll(this->pfds.data(), this->pfds.size(), -1)) != -1 &&
- errno != EINTR) {
- for (unsigned i = 0; i < pfds.size(); i++) {
- if ((pfds[i].revents & POLLIN) != 0) {
- if (handlers[i](pfds[i].fd) == -1) {
- return -1;
- }
- pfds[i].revents = 0;
- pfds[i].events = POLLIN;
- }
- }
- }
- return ret;
-}
-
unique_fd::~unique_fd() {
if (this->fd != -1) {
close(this->fd);
}
};
-// _ _ ____ _ _
-// ___| |_ _ __ _ _ ___| |_ | _ \ ___ | | | ___ _ __
-// / __| __| '__| | | |/ __| __| | |_) / _ \| | |/ _ \ '__|
-// \__ \ |_| | | |_| | (__| |_ | __/ (_) | | | __/ |
-// |___/\__|_| \__,_|\___|\__| |_| \___/|_|_|\___|_|
-//
-struct Poller {
- std::vector<std::function<int(int)>> handlers;
- std::vector<struct pollfd> pfds;
-
- Poller() = default;
- void add_fd(int fd, std::function<int(int)> handler);
- int check_events();
-};
-
#endif // !WM_UTIL_HPP