main.rs

mod app;
mod home;

use crate::app::App;

#[cfg(not(feature = "server"))]
fn main() {
    use shield_bootstrap::BootstrapDioxusStyle;

    dioxus::LaunchBuilder::new()
        .with_context(BootstrapDioxusStyle::default().context())
        .launch(App)
}

#[cfg(feature = "server")]
#[tokio::main]
async fn main() {
    use std::{env, sync::Arc};

    use axum::Router;
    use dioxus::{
        cli_config::fullstack_address_or_localhost,
        prelude::{DioxusRouterExt, *},
    };
    use shield::{ErasedMethod, Method, Shield, ShieldOptions};
    use shield_bootstrap::BootstrapDioxusStyle;
    use shield_dioxus_axum::{AxumDioxusIntegration, ShieldLayer};
    use shield_memory::{MemoryStorage, User};
    use shield_workos::{WorkosMethod, WorkosOauthProvider, WorkosOptions};
    use tokio::net::TcpListener;
    use tower_sessions::{Expiry, MemoryStore, SessionManagerLayer, cookie::time::Duration};
    use tracing::{Level, info};

    // Initialize Dioxus
    dioxus::logger::init(Level::DEBUG).unwrap();
    let addr = fullstack_address_or_localhost();

    // Initialize sessions
    let session_store = MemoryStore::default();
    let session_layer = SessionManagerLayer::new(session_store)
        .with_secure(false)
        .with_expiry(Expiry::OnInactivity(Duration::minutes(10)));

    // Initialize Shield
    let storage = MemoryStorage::new();
    let shield = Shield::new(
        storage.clone(),
        vec![Arc::new(WorkosMethod::from_api_key(
            &env::var("WORKOS_API_KEY").expect("Missing `WORKOS_API_KEY`."),
            &env::var("WORKOS_CLIENT_ID").expect("Missing `WORKOS_CLIENT_ID`."),
            WorkosOptions::builder()
                .oauth_providers(vec![
                    WorkosOauthProvider::AppleOAuth,
                    WorkosOauthProvider::GoogleOAuth,
                    WorkosOauthProvider::MicrosoftOAuth,
                ])
                .redirect_url(format!(
                    "http://localhost:{}/api/auth/workos/sign-in-callback",
                    addr.port()
                ))
                .build(),
        ))],
        ShieldOptions::default(),
    );
    let shield_layer = ShieldLayer::new(shield.clone());

    // Initialize router
    let router = Router::new()
        .serve_dioxus_application(
            ServeConfig::builder()
                .context(AxumDioxusIntegration::<User>::default().context())
                .context(BootstrapDioxusStyle::default().context())
                .build()
                .unwrap(),
            App,
        )
        .layer(shield_layer)
        .layer(session_layer);

    // Start app
    info!("listening on http://{}", &addr);
    let listener = TcpListener::bind(&addr).await.unwrap();
    axum::serve(listener, router.into_make_service())
        .await
        .unwrap();
}