chc 0.1.2

A simple, fast, and secure HTTP client.
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
91

use std::collections::HashMap;
use std::io::Write;
use std::net::{TcpStream};

pub struct CHC {
    pub stream: TcpStream,
    pub buffer: Vec<u8>,
}

impl CHC {
    pub fn new(stream: TcpStream, buffer: Vec<u8>) -> Self {
        CHC { stream, buffer }
    }

    pub fn respond(&mut self, body: String) {
        self.stream.write(body.as_bytes()).unwrap();
        self.stream.flush().unwrap();
    }

    pub fn headers(&self) -> HashMap<String, String> {
        let mut headers = HashMap::new();
        let mut lines = self.buffer.split(|&b| b == b'\n');
        while let Some(line) = lines.next() {
            if line.is_empty() {
                break;
            } else if line.starts_with(b"Cookie:") {
                continue;
            }

            let mut parts = line.splitn(2, |&b| b == b':');
            let key_final: String;
            let value_final: String;

            if let Some(key) = parts.next() {
                let temp_key = std::str::from_utf8(key).unwrap();
                key_final = temp_key.to_string().trim().to_string();
            } else {
                continue;
            }
            if let Some(value) = parts.next() {
                let temp_value = std::str::from_utf8(value).unwrap();
                value_final = temp_value.to_string().trim().to_string();
            } else {
                continue;
            }

            headers.insert(key_final, value_final);
        }

        headers
    }

    pub fn cookies(&self) -> HashMap<String, String> {
        let mut cookies = HashMap::new();
        let mut lines = self.buffer.split(|&b| b == b'\n');
        while let Some(line) = lines.next() {
            if line.is_empty() {
                break;
            } else if !line.starts_with(b"Cookie:") {
                continue;
            }

            let line = &line[7..];
            // split on ';'
            let mut parts = line.split(|&b| b == b';');

            while let Some(part) = parts.next() {
                let mut parts = part.splitn(2, |&b| b == b'=');
                let key_final: String;
                let value_final: String;

                if let Some(key) = parts.next() {
                    let temp_key = std::str::from_utf8(key).unwrap();
                    key_final = temp_key.to_string().trim().to_string();
                } else {
                    continue;
                }
                if let Some(value) = parts.next() {
                    let temp_value = std::str::from_utf8(value).unwrap();
                    value_final = temp_value.to_string().trim().to_string();
                } else {
                    continue;
                }

                cookies.insert(key_final, value_final);
            }
        }

        cookies
    }
}