From 88cd525dc8ad154104a753181a4491d3dcd73b56 Mon Sep 17 00:00:00 2001
From: patrick96 <p.ziegler96@gmail.com>
Date: Sat, 11 Sep 2021 11:46:50 +0200
Subject: [PATCH] Add wrapper for uv_pipe_t

---
 include/components/eventloop.hpp | 56 +++++++++++++++++++++++++++++---
 1 file changed, 51 insertions(+), 5 deletions(-)

diff --git a/include/components/eventloop.hpp b/include/components/eventloop.hpp
index 1bd92d49..9f3cdf40 100644
--- a/include/components/eventloop.hpp
+++ b/include/components/eventloop.hpp
@@ -31,16 +31,21 @@ struct cb_helper {
   }
 };
 
-template <typename H, typename... Args>
-struct UVHandle {
-  UVHandle(std::function<void(Args...)> fun) {
+template <typename H, typename I, typename... Args>
+struct UVHandleGeneric {
+  UVHandleGeneric(std::function<void(Args...)> fun) {
     handle = std::make_unique<H>();
-    cb = cb_helper<H, Args...>{fun};
+    cb = cb_helper<I, Args...>{fun};
     handle->data = &cb;
   }
 
   std::unique_ptr<H> handle;
-  cb_helper<H, Args...> cb;
+  cb_helper<I, Args...> cb;
+};
+
+template <typename H, typename... Args>
+struct UVHandle : public UVHandleGeneric<H, H, Args...> {
+  UVHandle(std::function<void(Args...)> fun) : UVHandleGeneric<H, H, Args...>(fun) {}
 };
 
 struct SignalHandle : public UVHandle<uv_signal_t, int> {
@@ -75,6 +80,47 @@ struct FSEventHandle : public UVHandle<uv_fs_event_t, const char*, int, int> {
   }
 };
 
+struct PipeHandle : public UVHandleGeneric<uv_pipe_t, uv_stream_t, ssize_t, const uv_buf_t*> {
+  PipeHandle(uv_loop_t* loop, std::function<void(const string)> fun)
+      : UVHandleGeneric([&](ssize_t nread, const uv_buf_t* buf) {
+        if (nread > 0) {
+          string payload = string(buf->base, nread);
+          logger::make().notice("Bytes read: %d: '%s'", nread, payload);
+          fun(std::move(payload));
+        } else if (nread < 0) {
+          if (nread != UV_EOF) {
+            logger::make().err("Read error: %s", uv_err_name(nread));
+            uv_close((uv_handle_t*)handle.get(), nullptr);
+          } else {
+            UV(
+                uv_read_start, (uv_stream_t*)handle.get(),
+                [](uv_handle_t*, size_t, uv_buf_t* buf) {
+                  buf->base = new char[BUFSIZ];
+                  buf->len = BUFSIZ;
+                },
+                &cb.callback);
+          }
+        }
+
+        if (buf->base) {
+          delete[] buf->base;
+        }
+      }) {
+    UV(uv_pipe_init, loop, handle.get(), false);
+  }
+
+  void start(int fd) {
+    UV(uv_pipe_open, handle.get(), fd);
+    UV(
+        uv_read_start, (uv_stream_t*)handle.get(),
+        [](uv_handle_t*, size_t, uv_buf_t* buf) {
+          buf->base = new char[BUFSIZ];
+          buf->len = BUFSIZ;
+        },
+        &cb.callback);
+  }
+};
+
 class eventloop {
  public:
   eventloop();