pondsocket 0.1.4

Channel-based realtime server framework with distributed pubsub support.
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

use pondsocket_common::PondAssigns;
use serde_json::Value;
use std::collections::HashMap;

use crate::errors::{Result, bad_request};
use crate::types::Route;

#[derive(Debug, Clone, Default)]
pub struct IncomingConnection {
    pub id: String,
    pub headers: HashMap<String, String>,
    pub cookies: HashMap<String, String>,
    pub query: HashMap<String, String>,
    pub params: HashMap<String, String>,
    pub address: String,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ConnectionDecision {
    Pending,
    Accepted,
    Declined,
}

#[derive(Debug, Clone)]
pub struct ConnectionContext {
    pub user_id: String,
    pub request: IncomingConnection,
    pub route: Route,
    assigns: PondAssigns,
    decision: ConnectionDecision,
    decline_code: u16,
    decline_message: String,
    pending_reply: Option<(String, Value)>,
}

impl ConnectionContext {
    pub fn new(user_id: impl Into<String>, request: IncomingConnection, route: Route) -> Self {
        Self {
            user_id: user_id.into(),
            request,
            route,
            assigns: PondAssigns::new(),
            decision: ConnectionDecision::Pending,
            decline_code: 0,
            decline_message: String::new(),
            pending_reply: None,
        }
    }

    pub fn accept(&mut self, assigns: PondAssigns) -> Result<&mut Self> {
        self.ensure_pending()?;
        self.decision = ConnectionDecision::Accepted;
        self.assigns.extend(assigns);
        Ok(self)
    }

    pub fn decline(&mut self, code: u16, message: impl Into<String>) -> Result<()> {
        self.ensure_pending()?;
        self.decision = ConnectionDecision::Declined;
        self.decline_code = code;
        self.decline_message = message.into();
        Ok(())
    }

    pub fn reply(&mut self, event: impl Into<String>, payload: Value) -> Result<&mut Self> {
        if self.decision == ConnectionDecision::Pending {
            self.decision = ConnectionDecision::Accepted;
        }
        if self.decision == ConnectionDecision::Declined {
            return Err(bad_request("GATEWAY", "cannot reply after declining"));
        }
        self.pending_reply = Some((event.into(), payload));
        Ok(self)
    }

    pub fn set_assign(&mut self, key: impl Into<String>, value: Value) -> &mut Self {
        self.assigns.insert(key.into(), value);
        self
    }

    pub fn decision(&self) -> ConnectionDecision {
        self.decision
    }

    pub fn is_accepted(&self) -> bool {
        self.decision == ConnectionDecision::Accepted
    }

    pub fn is_declined(&self) -> bool {
        self.decision == ConnectionDecision::Declined
    }

    pub fn decline_info(&self) -> (u16, String) {
        (self.decline_code, self.decline_message.clone())
    }

    pub fn assigns(&self) -> PondAssigns {
        self.assigns.clone()
    }

    pub fn pending_reply(&self) -> Option<(String, Value)> {
        self.pending_reply.clone()
    }

    fn ensure_pending(&self) -> Result<()> {
        if self.decision != ConnectionDecision::Pending {
            return Err(bad_request("GATEWAY", "connection response already sent"));
        }
        Ok(())
    }
}