shdp 1.2.0

A new protocol (SHDP)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

use std::{fs::File, io::Read, path::Path};

use bitvec::order::Lsb0;

use crate::protocol::{
    errors::Error,
    managers::{bits::encoder::BitEncoder, event::EventEncoder},
    server::bits::utils::CHARS,
};

pub struct FullFyveResponse {
    encoder: BitEncoder<Lsb0>,
    path: String,
}

impl FullFyveResponse {
    pub fn new(path: String) -> Self {
        if cfg!(feature = "debug") {
            println!(
                "[\x1b[38;5;227mSHDP\x1b[0m] \x1b[38;5;192m0x0004\x1b[0m created ({})",
                path
            );
        }

        FullFyveResponse {
            encoder: BitEncoder::<Lsb0>::new(),
            path,
        }
    }
}

impl EventEncoder<Lsb0> for FullFyveResponse {
    fn encode(&mut self) -> Result<(), Error> {
        let file_name = Path::new(&self.path)
            .file_name()
            .ok_or(Error {
                code: 400,
                message: format!("Invalid file name: {}", self.path),
                kind: crate::protocol::errors::ErrorKind::BadRequest,
            })?
            .to_str()
            .ok_or(Error {
                code: 400,
                message: format!("Invalid file name: {}", self.path),
                kind: crate::protocol::errors::ErrorKind::BadRequest,
            })?
            .to_string();

        self.encoder.add_bytes(file_name.as_bytes())?;
        self.encoder.add_data(0, 8)?;

        let mut file = match File::open(&self.path) {
            Ok(file) => file,
            Err(_) => {
                return Err(Error {
                    code: 404,
                    message: format!("File not found: {}", self.path),
                    kind: crate::protocol::errors::ErrorKind::NotFound,
                });
            }
        };

        let mut content = String::new();
        match file.read_to_string(&mut content) {
            Ok(_) => (),
            Err(_) => {
                return Err(Error {
                    code: 500,
                    message: format!("File read error: {}", self.path),
                    kind: crate::protocol::errors::ErrorKind::InternalServerError,
                });
            }
        }

        for byte in content.as_bytes().iter() {
            self.encoder
                .add_bitvec(CHARS.get(&(byte.to_owned() as char)).unwrap())?;
        }

        Ok(())
    }

    fn get_encoder(&self) -> &BitEncoder<Lsb0> {
        &self.encoder
    }

    fn get_event(&self) -> u16 {
        0x0004
    }
}