packager/rust/src/main.rs

use std::net::{IpAddr, SocketAddr};
use std::pin::Pin;
use std::process::ExitCode;
use std::str::FromStr;

use packager::{auth, cmd, models, routing, sqlite, telemetry, AppState, ClientState, Error};

use clap::Parser;

struct MainResult(Result<(), Error>);

impl std::process::Termination for MainResult {
    fn report(self) -> std::process::ExitCode {
        match self.0 {
            Ok(_) => ExitCode::SUCCESS,
            Err(e) => {
                eprintln!("Error: {e}");
                ExitCode::FAILURE
            }
        }
    }
}

impl From<Error> for MainResult {
    fn from(error: Error) -> Self {
        Self(Err(error))
    }
}

#[tokio::main]
async fn main() -> MainResult {
    let args = cmd::Args::parse();
    telemetry::init_tracing(
        if args.enable_opentelemetry.into() {
            telemetry::OpenTelemetryConfig::Enabled
        } else {
            telemetry::OpenTelemetryConfig::Disabled
        },
        if args.enable_tokio_console.into() {
            telemetry::TokioConsoleConfig::Enabled
        } else {
            telemetry::TokioConsoleConfig::Disabled
        },
        args,
        |args| -> Pin<Box<dyn std::future::Future<Output = MainResult>>> {
            Box::pin(async move {
                match args.command {
                    cmd::Command::Serve(serve_args) => {
                        if let Err(e) = sqlite::migrate(&args.database_url).await {
                            return <_ as Into<Error>>::into(e).into();
                        }

                        let database_pool =
                            match sqlite::init_database_pool(&args.database_url).await {
                                Ok(pool) => pool,
                                Err(e) => return <_ as Into<Error>>::into(e).into(),
                            };

                        let state = AppState {
                            database_pool,
                            client_state: ClientState::new(),
                            auth_config: if let Some(assume_user) =
                                serve_args.disable_auth_and_assume_user
                            {
                                auth::Config::Disabled { assume_user }
                            } else {
                                auth::Config::Enabled
                            },
                        };

                        // build our application with a route
                        let app = routing::router(state);
                        let app = telemetry::init_request_tracing(app);

                        let addr = SocketAddr::from((
                            IpAddr::from_str(&serve_args.bind)
                                .map_err(|error| {
                                    format!(
                                        "error parsing bind address {}: {}",
                                        &serve_args.bind, error
                                    )
                                })
                                .unwrap(),
                            serve_args.port,
                        ));
                        tracing::debug!("listening on {}", addr);
                        if let Err(e) = axum::Server::try_bind(&addr)
                            .map_err(|error| format!("error binding to {}: {}", addr, error))
                            .unwrap()
                            .serve(app.into_make_service())
                            .await
                        {
                            return <hyper::Error as Into<Error>>::into(e).into();
                        }
                    }
                    cmd::Command::Admin(admin_command) => match admin_command {
                        cmd::Admin::User(cmd) => match cmd {
                            cmd::UserCommand::Create(user) => {
                                let database_pool =
                                    match sqlite::init_database_pool(&args.database_url).await {
                                        Ok(pool) => pool,
                                        Err(e) => return <_ as Into<Error>>::into(e).into(),
                                    };

                                let id = match models::user::create(
                                    &database_pool,
                                    models::user::NewUser {
                                        username: &user.username,
                                        fullname: &user.fullname,
                                    },
                                )
                                .await
                                .map_err(|error| match error {
                                    models::Error::Query(models::QueryError::Duplicate {
                                        description: _,
                                    }) => Error::Command(packager::CommandError::UserExists {
                                        username: user.username.clone(),
                                    }),
                                    _ => Error::Model(error),
                                }) {
                                    Ok(id) => id,
                                    Err(e) => {
                                        return e.into();
                                    }
                                };

                                println!(
                                    "User \"{}\" created successfully (id {})",
                                    &user.username, id
                                )
                            }
                        },
                    },
                    cmd::Command::Migrate => {
                        if let Err(e) = sqlite::migrate(&args.database_url).await {
                            return <_ as Into<Error>>::into(e).into();
                        }

                        println!("Migrations successfully applied");
                    }
                }
                MainResult(Ok(()))
            })
        },
    )
    .await
}