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 std::sync::Arc;

use actix::*;
use actix_web::*;
use actix_web::http::Method;
use serde_json;
use futures::future::{ok, Future, Either};
use http::header::ACCESS_CONTROL_ALLOW_METHODS;

use utils::SockjsHeaders;
use session::{Message, Session};
use manager::{SessionManager, SessionMessage};

use super::MAXSIZE;


#[allow(non_snake_case)]
pub fn XhrSend<S, SM>(req: HttpRequest<Addr<Syn, SM>>)
                      -> Either<HttpResponse, Box<Future<Item=HttpResponse, Error=Error>>>
    where S: Session, SM: SessionManager<S>
{
    if *req.method() == Method::OPTIONS {
        Either::A(HttpResponse::NoContent()
                  .content_type("application/jsonscript; charset=UTF-8")
                  .header(ACCESS_CONTROL_ALLOW_METHODS, "OPTIONS, POST")
                  .sockjs_cache_headers()
                  .sockjs_cors_headers(req.headers())
                  .sockjs_session_cookie(&req)
                  .finish())
    }
    else if *req.method() != Method::GET && *req.method() != Method::POST {
        Either::A(HttpResponse::Forbidden().reason("Method is not allowed").finish())
    } else {
        Either::B(read(req))
    }
}

pub fn read<S, SM>(req: HttpRequest<Addr<Syn, SM>>)
                   -> Box<Future<Item=HttpResponse, Error=Error>>
    where S: Session, SM: SessionManager<S>
{
    let sid = req.match_info().get("session").unwrap().to_owned();

    Box::new(
        req.clone().body().limit(MAXSIZE)
            .map_err(error::ErrorBadRequest)
            .and_then(move |buf| {
                let sid = Arc::new(sid);

                // empty message
                if buf.is_empty() {
                    return Either::A(ok(
                        HttpResponse::InternalServerError().body("Payload expected.")))
                } else if buf != b"[]".as_ref() {
                    // deserialize json
                    let mut msgs: Vec<String> = match serde_json::from_slice(&buf) {
                        Ok(msgs) => msgs,
                        Err(_) => {
                            return Either::A(ok(
                                HttpResponse::InternalServerError()
                                    .body("Broken JSON encoding.")))
                        }
                    };

                    if !msgs.is_empty() {
                        let last = msgs.pop().unwrap();
                        for msg in msgs {
                            req.state().do_send(
                                SessionMessage {
                                    sid: Arc::clone(&sid),
                                    msg: Message(msg) });
                        }

                        return Either::B(
                            req.state().send(SessionMessage { sid: Arc::clone(&sid),
                                                              msg: Message(last) })
                                .from_err()
                                .and_then(move |res| {
                                    match res {
                                        Ok(_) => {
                                            Ok(HttpResponse::NoContent()
                                               .content_type("text/plain; charset=UTF-8")
                                               .sockjs_no_cache()
                                               .sockjs_cors_headers(req.headers())
                                               .sockjs_session_cookie(&req)
                                               .finish())
                                        },
                                        Err(_) =>
                                            Err(error::ErrorNotFound(())),
                                    }
                                }))
                    }
                }

                Either::A(ok(HttpResponse::NoContent()
                             .content_type("text/plain; charset=UTF-8")
                             .sockjs_no_cache()
                             .sockjs_cors_headers(req.headers())
                             .sockjs_session_cookie(&req)
                             .finish()))
            }))
}