lavina/crates/lavina-core/src/lib.rs

//! Domain definitions and implementation of common chat logic.
use anyhow::Result;
use prometheus::Registry as MetricsRegistry;

use crate::dialog::DialogRegistry;
use crate::player::PlayerRegistry;
use crate::repo::Storage;
use crate::room::RoomRegistry;

pub mod auth;
pub mod dialog;
pub mod player;
pub mod prelude;
pub mod repo;
pub mod room;
pub mod terminator;

mod table;

#[derive(Clone)]
pub struct LavinaCore {
    pub players: PlayerRegistry,
    pub rooms: RoomRegistry,
    pub dialogs: DialogRegistry,
}

impl LavinaCore {
    pub async fn new(mut metrics: MetricsRegistry, storage: Storage) -> Result<LavinaCore> {
        // TODO shutdown all services in reverse order on error
        let rooms = RoomRegistry::new(&mut metrics, storage.clone())?;
        let dialogs = DialogRegistry::new(storage.clone());
        let players = PlayerRegistry::empty(rooms.clone(), dialogs.clone(), storage.clone(), &mut metrics)?;
        dialogs.set_players(players.clone()).await;
        Ok(LavinaCore {
            players,
            rooms,
            dialogs,
        })
    }

    pub async fn shutdown(mut self) -> Result<()> {
        self.players.shutdown_all().await?;
        self.dialogs.unset_players().await;
        self.players.shutdown()?;
        self.dialogs.shutdown()?;
        self.rooms.shutdown()?;
        Ok(())
    }
}