sockjs 0.3.0

SockJS server for Actix
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

use time;
use actix_web::HttpRequest;
use actix_web::http::Cookie;
use actix_web::dev::HttpResponseBuilder;
use http::header::HeaderMap;
use http::header::{EXPIRES, ORIGIN, CACHE_CONTROL,
                   ACCESS_CONTROL_ALLOW_ORIGIN,
                   ACCESS_CONTROL_ALLOW_HEADERS,
                   ACCESS_CONTROL_ALLOW_METHODS,
                   ACCESS_CONTROL_ALLOW_CREDENTIALS,
                   ACCESS_CONTROL_MAX_AGE,
                   ACCESS_CONTROL_REQUEST_HEADERS};

const CACHE_CONTROL_VAL: &str =
    "no-store, no-cache, no-transform, must-revalidate, max-age=0";

#[derive(Serialize)]
pub(crate) struct Info {
    entropy: u32,
    websocket: bool,
    cookie_needed: bool,
    origins: Vec<&'static str>,
}

impl Info {

    pub fn new(entropy: u32, websocket: bool, cookie_needed: bool) -> Info {
        Info {
            entropy,
            websocket,
            cookie_needed,
            origins: vec!["*:*"],
        }
    }
}


pub(crate) trait SockjsHeaders {

    fn sockjs_allow_methods(&mut self) -> &mut Self;

    fn sockjs_no_cache(&mut self) -> &mut Self;

    fn sockjs_cache_headers(&mut self) -> &mut Self;

    fn sockjs_cors_headers(&mut self, headers: &HeaderMap) -> &mut Self;

    fn sockjs_session_cookie<S>(&mut self, req: &HttpRequest<S>) -> &mut Self;

}


impl SockjsHeaders for HttpResponseBuilder {

    fn sockjs_session_cookie<S>(&mut self, req: &HttpRequest<S>) -> &mut Self {
        let builder = if let Some(cookie) = req.cookie("JSESSIONID") {
            Cookie::build("JSESSIONID", cookie.value().to_owned())
        } else {
            Cookie::build("JSESSIONID", "dummy")
        };
        self.cookie(builder.path("/").finish());
        self
    }

    fn sockjs_allow_methods(&mut self) -> &mut Self {
        self.header(ACCESS_CONTROL_ALLOW_METHODS, "OPTIONS, GET")
    }

    fn sockjs_no_cache(&mut self) -> &mut Self {
        self.header(CACHE_CONTROL, CACHE_CONTROL_VAL)
    }

    fn sockjs_cors_headers(&mut self, headers: &HeaderMap) -> &mut Self {
        if let Some(origin) = headers.get(ORIGIN) {
            self.header(ACCESS_CONTROL_ALLOW_ORIGIN, origin.as_ref());
            if origin != "*" {
                self.header(ACCESS_CONTROL_ALLOW_CREDENTIALS, "true");
            }
        } else {
            self.header(ACCESS_CONTROL_ALLOW_ORIGIN, "*");
        }

        if let Some(ac) = headers.get(ACCESS_CONTROL_REQUEST_HEADERS) {
            self.header(ACCESS_CONTROL_ALLOW_HEADERS, ac.as_ref());
        }

        self
    }

    fn sockjs_cache_headers(&mut self) -> &mut Self {
        const TD365_SECONDS: &str = "31536000";
        const TD365_SECONDS_CC: &str  = "max-age=31536000, public";

        let d = time::now() + time::Duration::days(365);

        self.header(CACHE_CONTROL, TD365_SECONDS_CC);
        self.header(ACCESS_CONTROL_MAX_AGE, TD365_SECONDS);
        self.header(EXPIRES, time::strftime("%a, %d %b %Y %H:%M:%S", &d).unwrap().as_str());

        self
    }
}