use axum::extract::FromRef;
use sqlx::sqlite::SqlitePool;

#[cfg(test)]
use crate::user::app::Users;
use crate::{
    boot::app::Boot,
    conversation::app::Conversations,
    event::{self, app::Events},
    invite::app::Invites,
    login::app::Logins,
    message::app::Messages,
    push::app::Push,
    setup::app::Setup,
    token::{self, app::Tokens},
    vapid::app::Vapid,
};

#[derive(Clone)]
pub struct App<P> {
    db: SqlitePool,
    webpush: P,
    events: event::Broadcaster,
    token_events: token::Broadcaster,
}

impl<P> App<P> {
    pub fn from(db: SqlitePool, webpush: P) -> Self {
        let events = event::Broadcaster::default();
        let token_events = token::Broadcaster::default();
        Self {
            db,
            webpush,
            events,
            token_events,
        }
    }
}

impl<P> App<P> {
    pub fn boot(&self) -> Boot {
        Boot::new(self.db.clone())
    }

    pub fn conversations(&self) -> Conversations {
        Conversations::new(self.db.clone(), self.events.clone())
    }

    pub fn events(&self) -> Events {
        Events::new(self.db.clone(), self.events.clone())
    }

    pub fn invites(&self) -> Invites {
        Invites::new(self.db.clone(), self.events.clone())
    }

    pub fn logins(&self) -> Logins {
        Logins::new(self.db.clone(), self.token_events.clone())
    }

    pub fn messages(&self) -> Messages {
        Messages::new(self.db.clone(), self.events.clone())
    }

    pub fn push(&self) -> Push<P>
    where
        P: Clone,
    {
        Push::new(self.db.clone(), self.webpush.clone())
    }

    pub fn setup(&self) -> Setup {
        Setup::new(self.db.clone(), self.events.clone())
    }

    pub fn tokens(&self) -> Tokens {
        Tokens::new(self.db.clone(), self.token_events.clone())
    }

    #[cfg(test)]
    pub fn users(&self) -> Users {
        Users::new(self.db.clone(), self.events.clone())
    }

    pub fn vapid(&self) -> Vapid {
        Vapid::new(self.db.clone(), self.events.clone())
    }

    #[cfg(test)]
    pub fn webpush(&self) -> &P {
        &self.webpush
    }
}

impl<P> FromRef<App<P>> for Boot {
    fn from_ref(app: &App<P>) -> Self {
        app.boot()
    }
}

impl<P> FromRef<App<P>> for Conversations {
    fn from_ref(app: &App<P>) -> Self {
        app.conversations()
    }
}

impl<P> FromRef<App<P>> for Invites {
    fn from_ref(app: &App<P>) -> Self {
        app.invites()
    }
}

impl<P> FromRef<App<P>> for Logins {
    fn from_ref(app: &App<P>) -> Self {
        app.logins()
    }
}

impl<P> FromRef<App<P>> for Messages {
    fn from_ref(app: &App<P>) -> Self {
        app.messages()
    }
}

impl<P> FromRef<App<P>> for Push<P>
where
    P: Clone,
{
    fn from_ref(app: &App<P>) -> Self {
        app.push()
    }
}

impl<P> FromRef<App<P>> for Setup {
    fn from_ref(app: &App<P>) -> Self {
        app.setup()
    }
}

impl<P> FromRef<App<P>> for Tokens {
    fn from_ref(app: &App<P>) -> Self {
        app.tokens()
    }
}

impl<P> FromRef<App<P>> for Vapid {
    fn from_ref(app: &App<P>) -> Self {
        app.vapid()
    }
}