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

use std::ops::Deref;

#[derive(Debug)]
pub struct StringOrBuffer(Vec<u8>);

impl Deref for StringOrBuffer {
  type Target = Vec<u8>;
  fn deref(&self) -> &Vec<u8> {
    &self.0
  }
}

impl StringOrBuffer {
  pub fn into_bytes(self) -> Vec<u8> {
    self.0
  }
}

impl<'de> serde::Deserialize<'de> for StringOrBuffer {
  fn deserialize<D>(deserializer: D) -> Result<StringOrBuffer, D::Error>
  where
    D: serde::Deserializer<'de>,
  {
    StringOrBufferInner::deserialize(deserializer)
      .map(|x| StringOrBuffer(x.into_bytes()))
  }
}

// TODO(@AaronO): explore if we can make this work with ZeroCopyBuf
#[derive(serde::Deserialize)]
#[serde(untagged)]
enum StringOrBufferInner {
  #[serde(with = "serde_bytes")]
  Buffer(Vec<u8>),
  String(String),
}

impl StringOrBufferInner {
  fn into_bytes(self) -> Vec<u8> {
    match self {
      Self::String(s) => s.into_bytes(),
      Self::Buffer(b) => b,
    }
  }
}