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

use core::fmt::Write;
use heapless::{ArrayLength, Vec};

/// A sink to send HTTP requests to
pub trait Sink {
    fn send(&mut self, data: &[u8]) -> Result<usize, ()>;
}

/// A sink implementation for a buffer.
impl<N> Sink for Vec<u8, N>
where
    N: ArrayLength<u8>,
{
    fn send(&mut self, data: &[u8]) -> Result<usize, ()> {
        self.extend_from_slice(data).map_err(|_| ())?;

        Ok(data.len())
    }
}

struct SinkWrapper<'a>(&'a mut dyn Sink);

impl<'a> Write for SinkWrapper<'a> {
    fn write_str(&mut self, s: &str) -> Result<(), core::fmt::Error> {
        let buffer = s.as_bytes();
        let mut pos = 0usize;

        while pos < buffer.len() {
            match self.0.send(&buffer[pos..]) {
                Ok(len) => pos += len,
                Err(_) => return Err(core::fmt::Error),
            }
        }

        Ok(())
    }
}