httimple 0.1.7

Simple HTTP2 package 😃 
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

use proto::frame::headers::Headers;
use proto::frame::data::Data;

use rustls::ServerSession;
use std::io::Write;
use hpack_codec::Encoder;

///
/// A response to be sent.
/// 


#[derive(Debug)]
pub struct Message {
    pub status: u32,
    pub data: Vec<u8>
}


impl Message {

    ///
    /// Send the response.
    /// 

    pub fn send(&self, session: &mut ServerSession, encoder: &mut Encoder, stream_id: &mut u32) {
        if *stream_id == 0 {
            *stream_id += 1;
        }
        println!("[SENDING] Status: {} Stream: {}", self.status, stream_id);
        if self.status == 200 || self.status == 404 {
            let mut headers = Headers::new(*stream_id);
            headers.end_headers();
            session.write_all(&headers.encode(encoder)).unwrap();
            let mut d = Data::new(*stream_id, self.data.clone());
            session.write_all(&d.as_bytes()).unwrap();
        }
    } 

    ///
    /// Generates a message from the data supplied
    /// 


    pub fn from(data: Vec<u8>) -> Message {
        Message { status: 200, data: data }
    }

    ///
    /// Generates a 404 - PAGE NOT FOUND request.
    /// 


    pub fn not_found() -> Message {
        Message { status: 404, data: String::from("<h1>PAGE NOT FOUND</h1>").as_bytes().to_vec() }
    }

}