summaryrefslogtreecommitdiffhomepage
path: root/dev/kernel/NeKit/Stream.h
blob: dabf0f1998e3355d485fc2c4babb8d61f1de4e57 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

/* -------------------------------------------

  Copyright (C) 2024-2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.

------------------------------------------- */

#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