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
92
93
94
95
96
|
use std::{collections::HashMap, str::Split};
#[derive(Debug, Clone)]
pub struct HeaderMap {
map: HashMap<String, usize>,
headers: Vec<Header>
}
impl HeaderMap {
#[allow(dead_code)]
pub fn get(&self, key: &str) -> Option<&Header> {
let Some(index) = self.map.get(key) else {
return None;
};
return Some(&self.headers[index.to_owned()]);
}
pub fn put(&mut self, header: Header) {
if let Some(index) = self.map.get(&header.key) {
self.headers[index.to_owned()] = header;
return
}
let index = self.headers.len();
self.map.insert(header.key.clone(), index);
self.headers.push(header);
}
#[allow(dead_code)]
pub fn del(&mut self, key: &str) -> Option<Header> {
let Some(index) = self.map.get(key) else {
return None
};
let removed = self.headers.remove(index.to_owned());
for i in (index.to_owned())..self.headers.len() {
let key = &self.headers[i].key;
let Some(index) = self.map.get(key) else {
continue;
};
self.map.insert(key.clone(), index - 1);
}
Some(removed)
}
pub fn deserialize(&self) -> String {
let mut string = String::new();
for header in &self.headers {
string += &format!("{}: {}\n", header.key, header.value);
}
string
}
pub fn serialize(lines: &mut Split<char>) -> Self {
let mut headers = Self::new();
loop {
let Some(header) = lines.next() else { break };
if header.trim().len() < 1 { break }
let mut parts = header.split(": ").into_iter();
let Some(key) = parts.next() else { continue };
let Some(value) = parts.next() else { continue };
headers.put(Header::new(key.trim(), value.trim()));
}
headers
}
pub fn new() -> Self {
Self {
map: HashMap::new(),
headers: Vec::new()
}
}
}
#[derive(Debug, Clone)]
pub struct Header {
pub key: String,
pub value: String
}
impl Header {
pub fn new(key: &str, value: &str) -> Self {
return Self {key: key.to_owned(), value: value.to_owned()}
}
}
|
