prax 0.1.1

a web penetration proxy hosted in neovim
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use std::str::Split;

use hyper::body::Bytes;

use super::Encoding;

#[derive(Debug, Clone, PartialEq)]
pub struct Body(Bytes);

impl From<Vec<u8>> for Body {
    fn from(value: Vec<u8>) -> Self {
        Body(value.into())
    }
}

impl AsRef<[u8]> for Body {
    fn as_ref(&self) -> &[u8] {
        self.0.as_ref()
    }
}

impl Body {
    pub fn lines(&self) -> Option<Split<char>> {
        let s = std::str::from_utf8(&self.0);

        let s = s.ok()?;
        Some(s.split('\n'))
    }

    pub fn hex(&self, buf: &mut Vec<String>) {
        let mut line = String::new();

        for (pos, byte) in self.0.iter().enumerate() {
            let byte = *byte;

            if pos & 0x0f == 0 {
                if !line.is_empty() {
                    buf.push(line);
                    line = String::new();
                }

                line.push_str(&format!("{:08X} ", pos));
            }

            if (pos & 0x0f) == 0x08 {
                line.push(' ');
            }

            line.push_str(&format!(" {:02X}", byte));
        }

        if !line.is_empty() {
            buf.push(line);
        }
    }

    #[allow(dead_code)]
    pub fn encode(&self, encoding: Encoding) -> std::io::Result<Body> {
        Ok(Body(encoding.encode(&self.0)?))
    }

    pub fn decode(&self, encoding: Encoding) -> std::io::Result<Body> {
        Ok(Body(encoding.decode(&self.0)?))
    }
}