titanite 0.3.0

Client/Server Library for Gemini protocol with Titan support
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

pub mod default;
pub mod sensitive;

pub use default::Default;
pub use sensitive::Sensitive;

use anyhow::{bail, Result};

/// [Input expected](https://geminiprotocol.net/docs/protocol-specification.gmi#input-expected)
pub enum Input {
    Default(Default),
    Sensitive(Sensitive),
}

impl Input {
    pub fn from_bytes(buffer: &[u8]) -> Result<Self> {
        if buffer.first().is_none_or(|b| *b != b'1') {
            bail!("Unexpected first byte")
        }
        match buffer.get(1) {
            Some(byte) => Ok(match byte {
                b'0' => Self::Default(Default::from_bytes(buffer)?),
                b'1' => Self::Sensitive(Sensitive::from_bytes(buffer)?),
                b => bail!("Unexpected second byte: {b}"),
            }),
            None => bail!("Invalid request"),
        }
    }
    pub fn into_bytes(self) -> Vec<u8> {
        match self {
            Self::Default(this) => this.into_bytes(),
            Self::Sensitive(this) => this.into_bytes(),
        }
    }
}

#[test]
fn test() {
    // 10
    let request = format!("10 message\r\n");
    let source = Input::from_bytes(request.as_bytes()).unwrap();

    match source {
        Input::Default(ref this) => assert_eq!(this.message, Some("message".to_string())),
        _ => panic!(),
    }
    assert_eq!(source.into_bytes(), request.as_bytes());

    // 11
    let request = format!("11 message\r\n");
    let source = Input::from_bytes(request.as_bytes()).unwrap();

    match source {
        Input::Sensitive(ref this) => {
            assert_eq!(this.message, Some("message".to_string()))
        }
        _ => panic!(),
    }
    assert_eq!(source.into_bytes(), request.as_bytes());
}