AppFlowy/frontend/rust-lib/flowy-user/src/services/user/notifier.rs

use crate::entities::{UserProfile, UserStatus};
use flowy_net::entities::NetworkType;
use tokio::sync::{broadcast, mpsc};

pub struct UserNotifier {
    user_status_notifier: broadcast::Sender<UserStatus>,
    network_status_notifier: broadcast::Sender<NetworkType>,
}

impl std::default::Default for UserNotifier {
    fn default() -> Self {
        let (user_status_notifier, _) = broadcast::channel(10);
        let (network_status_notifier, _) = broadcast::channel(10);
        UserNotifier {
            user_status_notifier,
            network_status_notifier,
        }
    }
}

impl UserNotifier {
    pub(crate) fn new() -> Self { UserNotifier::default() }

    pub(crate) fn notify_login(&self, token: &str) {
        let _ = self.user_status_notifier.send(UserStatus::Login {
            token: token.to_owned(),
        });
    }

    pub(crate) fn notify_sign_up(&self, ret: mpsc::Sender<()>, user_profile: &UserProfile) {
        let _ = self.user_status_notifier.send(UserStatus::SignUp {
            profile: user_profile.clone(),
            ret,
        });
    }

    pub(crate) fn notify_logout(&self, token: &str) {
        let _ = self.user_status_notifier.send(UserStatus::Logout {
            token: token.to_owned(),
        });
    }

    pub fn update_network_type(&self, ty: &NetworkType) { let _ = self.network_status_notifier.send(ty.clone()); }

    pub fn user_status_subscribe(&self) -> broadcast::Receiver<UserStatus> { self.user_status_notifier.subscribe() }

    pub fn network_type_subscribe(&self) -> broadcast::Receiver<NetworkType> {
        self.network_status_notifier.subscribe()
    }
}