bitwarden_ipc/traits/
session_repository.rs

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
use std::collections::HashMap;

use tokio::sync::RwLock;

use crate::endpoint::Endpoint;

pub trait SessionRepository {
    type Session;
    type GetError;
    type SaveError;
    type RemoveError;

    fn get(
        &self,
        destination: Endpoint,
    ) -> impl std::future::Future<Output = Result<Option<Self::Session>, Self::GetError>>;
    fn save(
        &self,
        destination: Endpoint,
        session: Self::Session,
    ) -> impl std::future::Future<Output = Result<(), Self::SaveError>>;
    fn remove(
        &self,
        destination: Endpoint,
    ) -> impl std::future::Future<Output = Result<(), Self::RemoveError>>;
}

pub type InMemorySessionRepository<Session> = RwLock<HashMap<Endpoint, Session>>;
impl<Session> SessionRepository for InMemorySessionRepository<Session>
where
    Session: Clone,
{
    type Session = Session;
    type GetError = ();
    type SaveError = ();
    type RemoveError = ();

    async fn get(&self, destination: Endpoint) -> Result<Option<Self::Session>, ()> {
        Ok(self.read().await.get(&destination).cloned())
    }

    async fn save(&self, destination: Endpoint, session: Self::Session) -> Result<(), ()> {
        self.write().await.insert(destination, session);
        Ok(())
    }

    async fn remove(&self, destination: Endpoint) -> Result<(), ()> {
        self.write().await.remove(&destination);
        Ok(())
    }
}