web-petting/src/main.rs

mod admin;
mod i18n;
mod migrations;
pub mod models;
mod public;
mod telegram;
mod turnstile;
mod tz;

use tracing_subscriber;

use cot::cli::CliMetadata;
use cot::config::{
    DatabaseConfig, MiddlewareConfig, ProjectConfig, SameSite, SessionMiddlewareConfig,
    SessionStoreConfig, SessionStoreTypeConfig,
};
use cot::db::migrations::SyncDynMigration;
use cot::middleware::SessionMiddleware;
use cot::project::{MiddlewareContext, RegisterAppsContext, RootHandler};
use cot::router::Router;
use cot::session::db::SessionApp;
use cot::{App, AppBuilder, Project};

struct PettingApp;

impl App for PettingApp {
    fn name(&self) -> &'static str {
        "web-petting"
    }

    fn migrations(&self) -> Vec<Box<SyncDynMigration>> {
        cot::db::migrations::wrap_migrations(migrations::MIGRATIONS)
    }

    fn router(&self) -> Router {
        admin::admin_router()
    }
}

struct PublicApp;

impl App for PublicApp {
    fn name(&self) -> &'static str {
        "public"
    }

    fn router(&self) -> Router {
        public::public_router()
    }
}

struct PettingProject;

fn parse_bool_env(name: &str) -> Option<bool> {
    let value = std::env::var(name).ok()?;
    match value.trim().to_ascii_lowercase().as_str() {
        "1" | "true" | "yes" | "on" => Some(true),
        "0" | "false" | "no" | "off" => Some(false),
        _ => None,
    }
}

fn debug_enabled(config_name: &str) -> bool {
    parse_bool_env("WEB_PETTING_DEBUG").unwrap_or_else(|| {
        matches!(
            config_name,
            "dev" | "development" | "debug" | "local" | "test"
        )
    })
}

fn database_url() -> String {
    std::env::var("WEB_PETTING_DATABASE_URL")
        .or_else(|_| std::env::var("DATABASE_URL"))
        .unwrap_or_else(|_| "postgresql://postgres:postgres@localhost:5432/web_petting".to_string())
}

impl Project for PettingProject {
    fn cli_metadata(&self) -> CliMetadata {
        cot::cli::metadata!()
    }

    fn config(&self, config_name: &str) -> cot::Result<ProjectConfig> {
        Ok(ProjectConfig::builder()
            .debug(debug_enabled(config_name))
            .database(DatabaseConfig::builder().url(database_url()).build())
            .middlewares(
                MiddlewareConfig::builder()
                    .session(
                        SessionMiddlewareConfig::builder()
                            .secure(false)
                            .same_site(SameSite::Lax)
                            .store(
                                SessionStoreConfig::builder()
                                    .store_type(SessionStoreTypeConfig::Database)
                                    .build(),
                            )
                            .build(),
                    )
                    .build(),
            )
            .build())
    }

    fn register_apps(&self, apps: &mut AppBuilder, _context: &RegisterAppsContext) {
        apps.register(SessionApp::new());
        apps.register_with_views(PublicApp, "");
        apps.register_with_views(PettingApp, "/admin");
    }

    fn middlewares(
        &self,
        handler: cot::project::RootHandlerBuilder,
        context: &MiddlewareContext,
    ) -> RootHandler {
        handler
            .middleware(SessionMiddleware::from_context(context))
            .build()
    }
}

#[cot::main]
fn main() -> impl Project {
    let filter = tracing_subscriber::EnvFilter::try_from_default_env()
        .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new("info"));
    let _ = tracing_subscriber::fmt().with_env_filter(filter).try_init();
    PettingProject
}