leyline-http 0.1.1

HTTP client that mimics browsers on the wire
Documentation 
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::io::Write;

use super::limit::{Cap, Overflow};

pub(super) struct Sink {
    pub(super) buf: Vec<u8>,
    written: usize,
    cap: Cap,
    truncated: bool,
}

impl Sink {
    pub(super) fn new(cap: Cap) -> Self {
        Self {
            buf: Vec::new(),
            written: 0,
            cap,
            truncated: false,
        }
    }

    pub(super) fn truncated(&self) -> bool {
        self.truncated
    }
}

impl Write for Sink {
    fn write(&mut self, data: &[u8]) -> std::io::Result<usize> {
        let room = self.cap.room(self.written);
        self.written = self.written.saturating_add(data.len());
        if data.len() > room {
            if self.cap.overflow == Overflow::Truncate {
                self.buf.extend_from_slice(&data[..room]);
                self.truncated = true;
            }
            return Err(self.cap.limit.into_io());
        }
        self.buf.extend_from_slice(data);
        Ok(data.len())
    }

    fn flush(&mut self) -> std::io::Result<()> {
        Ok(())
    }
}