hyaenidae/src/nav.rs

use crate::{
    admin::Admin, extensions::ProfileExt, middleware::UserProfile,
    notifications::total_for_profile, ActixLoader, State,
};
use actix_web::{
    dev::Payload,
    web::{Data, Query},
    FromRequest, HttpRequest,
};
use futures::future::LocalBoxFuture;
use hyaenidae_accounts::LogoutState;
use hyaenidae_profiles::store::Profile;
use hyaenidae_toolkit::Button;
use i18n_embed_fl::fl;

impl FromRequest for NavState {
    type Config = ();
    type Error = actix_web::Error;
    type Future = LocalBoxFuture<'static, Result<Self, Self::Error>>;

    fn from_request(req: &HttpRequest, _: &mut Payload) -> Self::Future {
        let profile = Option::<UserProfile>::extract(req);
        let logout = Option::<LogoutState>::extract(req);
        let query = Option::<Query<Vec<(String, String)>>>::extract(req);
        let admin = Option::<Admin>::extract(req);
        let path = req.uri().path().to_owned();
        let state = Data::<State>::extract(req);

        Box::pin(async move {
            let profile = profile.await?.map(|p| p.0);
            let logout_state = logout.await?;
            let query = query.await?;
            let admin = admin.await?;
            let state = state.await?;

            let mut is_open = false;
            let href = if let Some(query) = query {
                let query = query
                    .into_inner()
                    .into_iter()
                    .filter_map(|(key, value)| {
                        if key == "show_nav" {
                            is_open = true;
                            None
                        } else {
                            Some(format!("{}={}", key, value))
                        }
                    })
                    .collect::<Vec<String>>()
                    .join("&");

                let query = if is_open {
                    query
                } else {
                    format!("{}&show_nav=true", query)
                };

                format!("{}?{}", path, query)
            } else {
                format!("{}?show_nav=true", path)
            };

            let notification_count = if let Some(profile) = &profile {
                total_for_profile(profile.id(), &state).await.ok()
            } else {
                None
            };

            Ok(NavState {
                profile,
                notification_count,
                logout_state,
                admin,
                href,
                is_open,
                dark: false,
            })
        })
    }
}

pub struct NavState {
    profile: Option<Profile>,
    notification_count: Option<u64>,
    logout_state: Option<LogoutState>,
    admin: Option<Admin>,
    href: String,
    is_open: bool,
    dark: bool,
}

impl NavState {
    pub(crate) fn href(&self) -> &str {
        &self.href
    }

    fn submission_button(&self, loader: &ActixLoader) -> Button {
        Button::primary(&fl!(loader, "nav-submission-button")).href("/submissions/create")
    }

    fn nav_button(&self, loader: &ActixLoader) -> Button {
        Button::secondary(&fl!(loader, "nav-text"))
            .href(&self.href)
            .class("nav-link")
    }

    fn browse_button(&self, loader: &ActixLoader) -> Button {
        Button::secondary(&fl!(loader, "nav-browse-button")).href("/browse")
    }

    fn profile_button(&self, loader: &ActixLoader) -> Button {
        if let Some(profile) = self.profile.as_ref() {
            Button::secondary(&fl!(loader, "nav-profile-button")).href(&profile.view_path())
        } else {
            Button::secondary(&fl!(loader, "nav-switch-profile-button")).href("/profiles/change")
        }
    }

    fn notifications_button(&self, loader: &ActixLoader) -> Button {
        Button::secondary(&fl!(loader, "nav-notifications-button")).href("/notifications")
    }

    fn admin_button(&self, loader: &ActixLoader) -> Button {
        Button::secondary(&fl!(loader, "nav-admin-button")).href("/admin")
    }

    fn account_button(&self, loader: &ActixLoader) -> Button {
        Button::secondary(&fl!(loader, "nav-account-button")).href("/session/account")
    }

    fn login_button(&self, loader: &ActixLoader) -> Button {
        Button::primary_outline(&fl!(loader, "nav-login-button")).href("/session/auth/login")
    }

    fn register_button(&self, loader: &ActixLoader) -> Button {
        Button::primary_outline(&fl!(loader, "nav-register-button")).href("/session/auth/register")
    }

    fn logout_button(&self, logout_state: &LogoutState, loader: &ActixLoader) -> Button {
        Button::primary_outline(&fl!(loader, "nav-logout-button")).form(&logout_state.logout_path())
    }

    pub(crate) fn buttons(&self, loader: &ActixLoader) -> Vec<Button> {
        let mut nav = vec![];

        if let Some(logout_state) = &self.logout_state {
            nav.push(self.submission_button(loader));
            nav.push(self.browse_button(loader));
            nav.push(self.profile_button(loader));

            if let Some(count) = self.notification_count {
                if count > 0 {
                    nav.push(self.notifications_button(loader));
                }
            }

            nav.push(self.account_button(loader));

            if self.admin.is_some() {
                nav.push(self.admin_button(loader));
            }
            nav.push(self.logout_button(logout_state, loader));
        } else {
            nav.push(self.login_button(loader));
            nav.push(self.register_button(loader));
        }

        nav
    }

    pub(crate) fn narrow_buttons(&self, loader: &ActixLoader) -> Vec<Button> {
        let mut narrow_nav = vec![];

        if self.logout_state.is_some() {
            narrow_nav.push(self.submission_button(loader));
            narrow_nav.push(self.nav_button(loader));
        } else {
            narrow_nav.push(self.nav_button(loader));
        }

        narrow_nav
    }

    pub(crate) fn site_buttons(&self, loader: &ActixLoader) -> Vec<Button> {
        let mut site_nav = vec![];

        if self.logout_state.is_some() {
            site_nav.push(self.submission_button(loader));
            site_nav.push(self.browse_button(loader));
            site_nav.push(self.profile_button(loader));

            if let Some(count) = self.notification_count {
                if count > 0 {
                    site_nav.push(self.notifications_button(loader));
                }
            }

            if self.admin.is_some() {
                site_nav.push(self.admin_button(loader));
            }
        }

        site_nav
    }

    pub(crate) fn account_buttons(&self, loader: &ActixLoader) -> Vec<Button> {
        let mut account_nav = vec![];

        if let Some(logout_state) = &self.logout_state {
            account_nav.push(self.logout_button(logout_state, loader));
        } else {
            account_nav.push(self.login_button(loader));
            account_nav.push(self.register_button(loader));
        }

        account_nav
    }

    pub(crate) fn class_string(&self) -> &str {
        if self.is_open {
            "nav-open"
        } else {
            "nav-closed"
        }
    }

    pub(crate) fn dark(&self) -> bool {
        self.dark
    }
}