Add unsubscribe endpoint

src/usimp/handler/mod.rs

@@ -2,6 +2,7 @@ mod authenticate;
 mod new_event;
 mod ping;
 mod subscribe;
+mod unsubscribe;
 
 use crate::usimp::*;
 use tokio::sync::mpsc;
@@ -31,6 +32,7 @@ pub async fn endpoint(
         "ping" => input.respond(ping::handle(&input, session).await?, None),
         "authenticate" => input.respond(authenticate::handle(&input, session).await?, None),
         "subscribe" => input.respond(subscribe::handle(&input, session, tx).await?, Some(OutputAction::Subscribe)),
+        "unsubscribe" => input.respond(unsubscribe::handle(&input, session).await?, Some(OutputAction::Unsubscribe)),
         "new_event" => input.respond(new_event::handle(&input, session).await?, None),
         _ => input.new_error(
             ErrorKind::UsimpError,

src/usimp/handler/new_event.rs

@@ -22,14 +22,14 @@ pub async fn handle(input: &InputEnvelope, session: Option<Session>) -> Result<V
 }
 
 async fn new_event(input: Input, session: Option<Session>) -> Result<Output, Error> {
-    let _account = get_account(&session)?;
+    let _account = get_account_opt(&session)?;
     let mut uuids = vec![];
     // TODO check permissions
     for mut event in input.events {
         let uuid = Uuid::new_v4();
         event.id = Some(uuid);
         uuids.push(uuid);
-        subscription::push(&input.room_id, event).await?;
+        subscription::push_room(&input.room_id, event).await?;
     }
     Ok(Output {events: uuids})
 }

src/usimp/handler/subscribe.rs

@@ -35,8 +35,16 @@ async fn subscribe(
     req_nr: Option<u64>,
     tx: Option<mpsc::Sender<WebSocketEnvelope>>,
 ) -> Result<Output, Error> {
-    let account = get_account(&session)?;
+    let session = match session {
-    let mut rx = subscription::subscribe_account(account).await;
+        Some(s) => s,
+        None => return Err(Error::new(
+            ErrorKind::SubscriptionError,
+            ErrorClass::ClientError,
+            None,
+        )),
+    };
+
+    let mut rx = subscription::subscribe_account(&session, req_nr).await?;
     match tx {
         Some(tx) => {
             tokio::spawn(async move {

src/usimp/handler/unsubscribe.rs

@@ -0,0 +1,45 @@
+use crate::usimp::subscription;
+use crate::usimp::*;
+
+use serde::{Deserialize, Serialize};
+use serde_json::{from_value, to_value, Value};
+use tokio::sync::mpsc;
+use crate::websocket::WebSocketEnvelope;
+
+#[derive(Serialize, Deserialize, Clone)]
+struct Input {}
+
+#[derive(Serialize, Deserialize, Clone)]
+struct Output {}
+
+pub async fn handle(
+    input: &InputEnvelope,
+    session: Option<Session>,
+) -> Result<Value, Error> {
+    Ok(to_value(
+        unsubscribe(
+            from_value(input.data.clone())?,
+            session,
+            input.request_nr,
+        ).await?,
+    )?)
+}
+
+async fn unsubscribe(
+    _input: Input,
+    session: Option<Session>,
+    req_nr: Option<u64>,
+) -> Result<Output, Error> {
+    let session = match session {
+        Some(s) => s,
+        None => return Err(Error::new(
+            ErrorKind::SubscriptionError,
+            ErrorClass::ClientError,
+            None,
+        )),
+    };
+
+    subscription::unsubscribe_account(&session, req_nr).await?;
+
+    Ok(Output {})
+}

src/usimp/mod.rs

@@ -93,10 +93,9 @@ pub struct Session {
     account: Option<Account>,
 }
 
-pub fn get_account(session: &Option<Session>) -> Result<&Account, Error> {
+pub fn get_account_opt(session: &Option<Session>) -> Result<&Account, Error> {
     match session {
-        Some(session) => match &session.account {
+        Some(session) => get_account(session),
-            Some(account) => Ok(&account),
         None => {
             return Err(Error::new(
                 ErrorKind::UsimpError,
@@ -104,7 +103,12 @@ pub fn get_account(session: &Option<Session>) -> Result<&Account, Error> {
                 None,
             ))
         }
-        },
+    }
+}
+
+pub fn get_account(session: &Session) -> Result<&Account, Error> {
+    match &session.account {
+        Some(account) => Ok(&account),
         None => {
             return Err(Error::new(
                 ErrorKind::UsimpError,

src/usimp/subscription.rs

@@ -1,36 +1,80 @@
 use crate::database;
 use crate::usimp::*;
-use std::collections::HashMap;
+use std::collections::{HashMap, HashSet};
+use std::ops::Deref;
 use std::sync::Arc;
 use tokio::sync::{mpsc, Mutex};
+use tokio::sync::mpsc::Sender;
 
-static mut ROOMS: Option<Arc<Mutex<HashMap<Uuid, Vec<mpsc::Sender<Event>>>>>> = None;
+#[derive(Clone, Eq, Hash, PartialEq)]
-static mut ACCOUNTS: Option<Arc<Mutex<HashMap<Uuid, Vec<mpsc::Sender<Event>>>>>> = None;
+struct Subscription {
+    session: Uuid,
+    req_nr: Option<u64>,
+}
+
+static mut SUBSCRIPTIONS: Option<Arc<Mutex<HashMap<Subscription, mpsc::Sender<Event>>>>> = None;
+static mut ACCOUNTS: Option<Arc<Mutex<HashMap<Uuid, HashSet<Subscription>>>>> = None;
 
 pub fn init() {
     unsafe {
-        ROOMS = Some(Arc::new(Mutex::new(HashMap::new())));
+        SUBSCRIPTIONS = Some(Arc::new(Mutex::new(HashMap::new())));
         ACCOUNTS = Some(Arc::new(Mutex::new(HashMap::new())));
     }
 }
 
-pub async fn subscribe_account(account: &Account) -> mpsc::Receiver<Event> {
+pub async fn subscribe_account(session: &Session, req_nr: Option<u64>) -> Result<mpsc::Receiver<Event>, Error> {
+    let account = get_account(session)?;
+    let sub = Subscription {session: session.id, req_nr};
+
     let (tx, rx) = mpsc::channel::<Event>(64);
     unsafe {
-        let mut acc = ACCOUNTS.as_ref().unwrap().lock().await;
+        let mut subs = SUBSCRIPTIONS.as_ref().unwrap().lock().await;
-        match acc.get_mut(&account.id) {
+        let mut accs = ACCOUNTS.as_ref().unwrap().lock().await;
-            Some(vec) => {
+        match accs.get_mut(&account.id) {
-                vec.push(tx);
+            Some(set) => {
+                set.insert(sub.clone());
             }
             None => {
-                acc.insert(account.id, vec![tx]);
+                let mut set = HashSet::new();
+                set.insert(sub.clone());
+                accs.insert(account.id, set);
+            }
+        }
+        match subs.get_mut(&sub) {
+            Some(_) => return Err(Error::new(
+                ErrorKind::SubscriptionError,
+                ErrorClass::ClientError,
+                None
+            )),
+            None => {
+                subs.insert(sub.clone(), tx);
             }
         }
     }
-    rx
+
+    Ok(rx)
 }
 
-pub async fn push(room_id: &Uuid, event: Event) -> Result<(), Error> {
+pub async fn unsubscribe_account(session: &Session, req_nr: Option<u64>) -> Result<(), Error> {
+    let account = get_account(session)?;
+    let sub = Subscription {session: session.id, req_nr};
+
+    unsafe {
+        let mut subs = SUBSCRIPTIONS.as_ref().unwrap().lock().await;
+        let mut accs = ACCOUNTS.as_ref().unwrap().lock().await;
+        match accs.get_mut(&account.id) {
+            Some(set) => {
+                set.remove(&sub);
+            }
+            None => {},
+        }
+        subs.remove(&sub);
+    }
+
+    Ok(())
+}
+
+pub async fn push_room(room_id: &Uuid, event: Event) -> Result<(), Error> {
     let backend = database::client().await?;
     let accounts = match backend {
         database::Client::Postgres(client) => {
@@ -50,26 +94,26 @@ pub async fn push(room_id: &Uuid, event: Event) -> Result<(), Error> {
         }
     };
 
-    let mut rooms = unsafe {
+    let mut room: Vec<mpsc::Sender<Event>> = Vec::new();
-        let mut rooms = ROOMS.as_ref().unwrap().lock().await;
-        if let Some(rooms) = rooms.get_mut(room_id) {
-            rooms.clone()
-        } else {
-            Vec::new()
-        }
-    };
 
     for account in accounts {
         unsafe {
-            let mut accounts = ACCOUNTS.as_ref().unwrap().lock().await;
+            let subs = SUBSCRIPTIONS.as_ref().unwrap().lock().await;
-            if let Some(acc) = accounts.get_mut(&account) {
+            let accs = ACCOUNTS.as_ref().unwrap().lock().await;
-                let mut acc = acc.clone();
+            if let Some(acc_subs) = accs.get(&account) {
-                rooms.append(&mut acc);
+                for sub in acc_subs {
+                    match subs.get(sub) {
+                        None => {}
+                        Some(tx) => {
+                            room.push(tx.clone());
+                        }
+                    }
+                }
             }
         }
     }
 
-    for tx in rooms {
+    for tx in room {
         let _res = tx.send(event.clone()).await;
     }