normogen/backend/src/main.rs

//! Binary entrypoint: loads config, connects to MongoDB, wires services into
//! [`config::AppState`], builds the router via [`app::build_app`], and serves.
//!
//! All module wiring and route/middleware composition lives in the library
//! (`lib.rs` + `app.rs`) so integration tests can reuse them.

use std::sync::Arc;

use normogen_backend::{
    app::build_app,
    auth::{JwtService, TokenVersionCache},
    config::{self, Config, Environment},
    db,
    models::{health_stats::HealthStatisticsRepository, refresh_token::RefreshTokenRepository},
    security, services,
};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    eprintln!("NORMOGEN BACKEND STARTING...");
    eprintln!("Loading environment variables...");

    match dotenv::dotenv() {
        Ok(path) => eprintln!("Loaded .env from: {:?}", path),
        Err(e) => eprintln!("No .env file found (this is OK in Docker): {}", e),
    }

    eprintln!("Initializing logging...");
    tracing_subscriber::fmt::init();

    tracing::info!("Starting Normogen backend server");

    // Load configuration
    let config = Config::from_env()?;
    if config.environment == Environment::Production {
        eprintln!("Running in PRODUCTION mode");
    } else {
        eprintln!("Running in DEVELOPMENT mode");
    }
    eprintln!("Configuration loaded successfully");

    // Connect to MongoDB
    tracing::info!("Connecting to MongoDB at {}", config.database.uri);
    eprintln!("Connecting to MongoDB...");
    let db = match db::MongoDb::new(&config.database.uri, &config.database.database).await {
        Ok(db) => {
            tracing::info!(
                "Connected to MongoDB database: {}",
                config.database.database
            );
            eprintln!("MongoDB connection successful");
            db
        }
        Err(e) => {
            eprintln!("FATAL: Failed to connect to MongoDB: {}", e);
            return Err(e);
        }
    };

    match db.health_check().await {
        Ok(_) => {
            tracing::info!("MongoDB health check: OK");
            eprintln!("MongoDB health check: OK");
        }
        Err(e) => {
            tracing::warn!("MongoDB health check failed: {}", e);
            eprintln!("WARNING: MongoDB health check failed: {}", e);
        }
    }

    // Create JWT service
    let jwt_service = JwtService::new(config.jwt.clone());

    // Get the underlying MongoDB database for security services
    let database = db.get_database();

    // Ensure collections/indexes exist (best-effort; failures are logged, not
    // fatal). Includes the refresh_tokens tokenHash-unique + expiresAt TTL
    // indexes and the previously-never-run users.email unique index.
    if let Err(e) = db::DatabaseInitializer::new(database.clone())
        .initialize()
        .await
    {
        tracing::warn!("Database index initialization skipped: {}", e);
    }

    // Short-TTL cache of token_version, so the JWT middleware can reject stale
    // access tokens without a Mongo lookup per request.
    let token_version_cache = Arc::new(TokenVersionCache::with_default_ttl());

    // Persisted refresh tokens (hashed) for rotation/revocation.
    let refresh_token_repo = Arc::new(RefreshTokenRepository::new(&database));

    // Initialize security services (Phase 2.6)
    let audit_logger = security::AuditLogger::new(&database);
    let session_manager = security::SessionManager::new(&database);

    // Create account lockout service with reasonable defaults
    let user_collection = database.collection("users");
    let account_lockout = security::AccountLockout::new(
        user_collection,
        5,    // max_attempts
        15,   // base_duration_minutes
        1440, // max_duration_minutes (24 hours)
    );

    // Initialize health stats repository (Phase 2.7) - using Database pattern
    let health_stats_repo = HealthStatisticsRepository::new(&database);

    // Initialize interaction service (Phase 2.8)
    let interaction_service = Arc::new(services::InteractionService::new());
    eprintln!("Interaction service initialized (Phase 2.8)");

    // Create application state
    let state = config::AppState {
        db,
        jwt_service,
        config: config.clone(),
        audit_logger: Some(audit_logger),
        session_manager: Some(session_manager),
        account_lockout: Some(account_lockout),
        health_stats_repo: Some(health_stats_repo),
        mongo_client: None,
        interaction_service: Some(interaction_service),
        token_version_cache,
        refresh_token_repo: Some(refresh_token_repo),
    };

    eprintln!("Building router with security middleware...");
    let app = build_app(state);

    let addr = format!("{}:{}", config.server.host, config.server.port);
    eprintln!("Binding to {}...", addr);
    let listener = tokio::net::TcpListener::bind(&addr).await?;
    eprintln!("Server listening on {}", &addr);
    tracing::info!("Server listening on {}", &addr);

    // into_make_service_with_connect_info makes ConnectInfo<SocketAddr> available
    // to the client_ip_middleware so we can fall back to the peer address when no
    // proxy header (X-Forwarded-For / X-Real-IP) is present.
    axum::serve(
        listener,
        app.into_make_service_with_connect_info::<std::net::SocketAddr>(),
    )
    .await?;

    Ok(())
}