Merge pull request #32 from nekernel-org/dev

0.0.2e3

1 files changed, 45 insertions, 0 deletions

diff --git a/dev/kernel/NeKit/Stream.h b/dev/kernel/NeKit/Stream.h
new file mode 100644
index 00000000..26fc46fc
--- /dev/null
+++ b/dev/kernel/NeKit/Stream.h
@@ -0,0 +1,45 @@
+
+/* -------------------------------------------
+
+  Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <NeKit/Defines.h>
+#include <NeKit/Ref.h>
+
+namespace Kernel {
+template <typename StreamTrait, typename Kind>
+class Stream final {
+ public:
+  explicit Stream(Ref<Stream> ref) : fStream(ref) {}
+
+  ~Stream() = default;
+
+  Stream& operator=(const Stream&) = default;
+  Stream(const Stream&)            = default;
+
+  template <typename Data>
+  friend Stream<StreamTrait, Kind>& operator>>(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
+    Ks.fKind = Ks.fStream->In(Buf);
+    return *Ks;
+  }
+
+  template <typename Data>
+  friend Stream<StreamTrait, Kind>& operator<<(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
+    Ks.fKind = Buf;
+    Ks.fStream->Out(Buf.Leak());
+    return *Ks;
+  }
+
+  Ref<StreamTrait>& AsStreamTrait() { return fStream; }
+
+  Ref<Kind>& AsType() { return fKind; }
+
+ private:
+  Ref<StreamTrait> fStream;
+  Ref<Kind>        fKind;
+};
+}  // namespace Kernel