runlight 0.1.3

Privacy friendly web analytics that lives inside your Rust app: mount a route, add a script tag, read your stats at /runlight
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

//! Headers as the Fetch API's `Headers` holds them.

/// Header names matched without regard to case, as the Fetch API's
/// `Headers` are. `get` joins repeated values with `", "`; Set-Cookie is kept
/// apart, since its values may hold commas, and read back with
/// `get_set_cookie`.
#[derive(Clone, Debug, Default, PartialEq, Eq)]
pub struct Headers {
    /// Lowercase name and its values, in the order names were first set.
    values: Vec<(String, Vec<String>)>,
}

impl Headers {
    /// No headers.
    pub fn new() -> Headers {
        Headers::default()
    }

    /// Headers from name and value pairs, each appended.
    pub fn from_pairs<K: AsRef<str>, V: AsRef<str>>(pairs: impl IntoIterator<Item = (K, V)>) -> Headers {
        let mut h = Headers::new();
        for (k, v) in pairs {
            h.append(k.as_ref(), v.as_ref());
        }
        h
    }

    /// `set`, returning the headers, for building them in a line.
    pub fn with(mut self, name: &str, value: impl AsRef<str>) -> Headers {
        self.set(name, value.as_ref());
        self
    }

    /// Every value of the header joined with `", "`, or `None`.
    pub fn get(&self, name: &str) -> Option<String> {
        let name = name.to_ascii_lowercase();
        self.values.iter().find(|(n, _)| *n == name).map(|(_, v)| v.join(", "))
    }

    /// Whether the header is there.
    pub fn has(&self, name: &str) -> bool {
        let name = name.to_ascii_lowercase();
        self.values.iter().any(|(n, _)| *n == name)
    }

    /// Gives the header this one value.
    pub fn set(&mut self, name: &str, value: &str) {
        let name = name.to_ascii_lowercase();
        let value = clean(value);
        match self.values.iter_mut().find(|(n, _)| *n == name) {
            Some(entry) => entry.1 = vec![value],
            None => self.values.push((name, vec![value])),
        }
    }

    /// Adds a value to the header.
    pub fn append(&mut self, name: &str, value: &str) {
        let name = name.to_ascii_lowercase();
        let value = clean(value);
        match self.values.iter_mut().find(|(n, _)| *n == name) {
            Some(entry) => entry.1.push(value),
            None => self.values.push((name, vec![value])),
        }
    }

    /// Removes the header.
    pub fn delete(&mut self, name: &str) {
        let name = name.to_ascii_lowercase();
        self.values.retain(|(n, _)| *n != name);
    }

    /// The Set-Cookie values, each apart.
    pub fn get_set_cookie(&self) -> Vec<String> {
        self.values.iter().find(|(n, _)| n == "set-cookie").map(|(_, v)| v.clone()).unwrap_or_default()
    }

    /// Every name with its values, in the order names were first set.
    pub fn all(&self) -> &[(String, Vec<String>)] {
        &self.values
    }

    /// Name and joined value pairs in name order, as iterating Fetch
    /// `Headers` gives them; each Set-Cookie on its own.
    pub fn entries(&self) -> Vec<(String, String)> {
        let mut names: Vec<&(String, Vec<String>)> = self.values.iter().collect();
        names.sort_by(|a, b| a.0.cmp(&b.0));
        let mut out = Vec::new();
        for (name, values) in names {
            if name == "set-cookie" {
                for v in values {
                    out.push((name.clone(), v.clone()));
                }
            } else {
                out.push((name.clone(), values.join(", ")));
            }
        }
        out
    }

    /// Whether there are none.
    pub fn is_empty(&self) -> bool {
        self.values.is_empty()
    }
}

/// Header values never carry a line break, so nothing a caller passes can add
/// a header of its own; the space Fetch trims goes too.
fn clean(value: &str) -> String {
    let v: String = value.chars().filter(|c| !matches!(c, '\r' | '\n' | '\0')).collect();
    v.trim_matches(|c| c == ' ' || c == '\t').to_string()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn headers_read_as_fetch_reads_them() {
        let mut h = Headers::new();
        h.append("Accept", "a");
        h.append("accept", "b");
        h.append("Set-Cookie", "x=1, y");
        h.append("set-cookie", "z=2");
        h.set("X-Test", " v\r\n ");
        assert_eq!(h.get("ACCEPT").as_deref(), Some("a, b"));
        assert_eq!(h.get_set_cookie(), vec!["x=1, y", "z=2"]);
        assert_eq!(h.get("x-test").as_deref(), Some("v"));
        assert_eq!(h.entries()[0], ("accept".to_string(), "a, b".to_string()));
        h.delete("accept");
        assert!(!h.has("Accept"));
    }
}