start-os/backend/src/bin/embassy-init.rs

use std::path::{Path, PathBuf};
use std::sync::Arc;
use std::time::Duration;

use embassy::context::rpc::RpcContextConfig;
use embassy::context::{DiagnosticContext, InstallContext, SetupContext};
use embassy::disk::fsck::RepairStrategy;
use embassy::disk::main::DEFAULT_PASSWORD;
use embassy::disk::REPAIR_DISK_PATH;
use embassy::hostname::get_current_ip;
use embassy::init::STANDBY_MODE_PATH;
use embassy::net::embassy_service_http_server::EmbassyServiceHTTPServer;
#[cfg(feature = "avahi")]
use embassy::net::mdns::MdnsController;
use embassy::net::net_utils::ResourceFqdn;
use embassy::net::static_server::{
    diag_ui_file_router, install_ui_file_router, setup_ui_file_router,
};
use embassy::shutdown::Shutdown;
use embassy::sound::CHIME;
use embassy::util::logger::EmbassyLogger;
use embassy::util::Invoke;
use embassy::{Error, ErrorKind, ResultExt};
use tokio::process::Command;
use tracing::instrument;

#[instrument]
async fn setup_or_init(cfg_path: Option<PathBuf>) -> Result<(), Error> {
    if tokio::fs::metadata("/cdrom").await.is_ok() {
        #[cfg(feature = "avahi")]
        let _mdns = MdnsController::init().await?;

        let ctx = InstallContext::init(cfg_path).await?;

        let embassy_ip = get_current_ip(ctx.ethernet_interface.to_owned()).await?;
        let embassy_ip_fqdn: ResourceFqdn = embassy_ip.parse()?;
        let embassy_fqdn: ResourceFqdn = "pureos.local".parse()?;

        let localhost_fqdn = ResourceFqdn::LocalHost;

        let install_ui_handler = install_ui_file_router(ctx.clone()).await?;

        let mut install_http_server =
            EmbassyServiceHTTPServer::new([0, 0, 0, 0].into(), 80, None).await?;
        install_http_server
            .add_svc_handler_mapping(embassy_ip_fqdn, install_ui_handler.clone())
            .await?;
        install_http_server
            .add_svc_handler_mapping(embassy_fqdn, install_ui_handler.clone())
            .await?;

        install_http_server
            .add_svc_handler_mapping(localhost_fqdn, install_ui_handler.clone())
            .await?;

        tokio::time::sleep(Duration::from_secs(1)).await; // let the record state that I hate this
        CHIME.play().await?;

        ctx.shutdown
            .subscribe()
            .recv()
            .await
            .expect("context dropped");
        install_http_server.shutdown.send(()).unwrap();
        Command::new("reboot")
            .invoke(embassy::ErrorKind::Unknown)
            .await?;
    } else if tokio::fs::metadata("/media/embassy/config/disk.guid")
        .await
        .is_err()
    {
        #[cfg(feature = "avahi")]
        let _mdns = MdnsController::init().await?;

        let ctx = SetupContext::init(cfg_path).await?;

        let embassy_ip = get_current_ip(ctx.ethernet_interface.to_owned()).await?;
        let embassy_ip_fqdn: ResourceFqdn = embassy_ip.parse()?;
        let embassy_fqdn: ResourceFqdn = "embassy.local".parse()?;
        let localhost_fqdn = ResourceFqdn::LocalHost;

        let setup_ui_handler = setup_ui_file_router(ctx.clone()).await?;

        let mut setup_http_server =
            EmbassyServiceHTTPServer::new([0, 0, 0, 0].into(), 80, None).await?;
        setup_http_server
            .add_svc_handler_mapping(embassy_ip_fqdn, setup_ui_handler.clone())
            .await?;
        setup_http_server
            .add_svc_handler_mapping(embassy_fqdn, setup_ui_handler.clone())
            .await?;

        setup_http_server
            .add_svc_handler_mapping(localhost_fqdn, setup_ui_handler.clone())
            .await?;

        tokio::time::sleep(Duration::from_secs(1)).await; // let the record state that I hate this
        CHIME.play().await?;
        ctx.shutdown
            .subscribe()
            .recv()
            .await
            .expect("context dropped");
        setup_http_server.shutdown.send(()).unwrap();
    } else {
        let cfg = RpcContextConfig::load(cfg_path).await?;
        let guid_string = tokio::fs::read_to_string("/media/embassy/config/disk.guid") // unique identifier for volume group - keeps track of the disk that goes with your embassy
            .await?;
        let guid = guid_string.trim();
        let requires_reboot = embassy::disk::main::import(
            guid,
            cfg.datadir(),
            if tokio::fs::metadata(REPAIR_DISK_PATH).await.is_ok() {
                RepairStrategy::Aggressive
            } else {
                RepairStrategy::Preen
            },
            DEFAULT_PASSWORD,
        )
        .await?;
        if tokio::fs::metadata(REPAIR_DISK_PATH).await.is_ok() {
            tokio::fs::remove_file(REPAIR_DISK_PATH)
                .await
                .with_ctx(|_| (embassy::ErrorKind::Filesystem, REPAIR_DISK_PATH))?;
        }
        if requires_reboot.0 {
            embassy::disk::main::export(guid, cfg.datadir()).await?;
            Command::new("reboot")
                .invoke(embassy::ErrorKind::Unknown)
                .await?;
        }
        tracing::info!("Loaded Disk");
        embassy::init::init(&cfg).await?;
    }

    Ok(())
}

async fn run_script_if_exists<P: AsRef<Path>>(path: P) {
    let script = path.as_ref();
    if script.exists() {
        match Command::new("/bin/bash").arg(script).spawn() {
            Ok(mut c) => {
                if let Err(e) = c.wait().await {
                    tracing::error!("Error Running {}: {}", script.display(), e);
                    tracing::debug!("{:?}", e);
                }
            }
            Err(e) => {
                tracing::error!("Error Running {}: {}", script.display(), e);
                tracing::debug!("{:?}", e);
            }
        }
    }
}

#[instrument]
async fn inner_main(cfg_path: Option<PathBuf>) -> Result<Option<Shutdown>, Error> {
    if tokio::fs::metadata(STANDBY_MODE_PATH).await.is_ok() {
        tokio::fs::remove_file(STANDBY_MODE_PATH).await?;
        Command::new("sync").invoke(ErrorKind::Filesystem).await?;
        embassy::sound::SHUTDOWN.play().await?;
        futures::future::pending::<()>().await;
    }

    embassy::sound::BEP.play().await?;

    run_script_if_exists("/media/embassy/config/preinit.sh").await;

    let res = if let Err(e) = setup_or_init(cfg_path.clone()).await {
        async move {
            tracing::error!("{}", e.source);
            tracing::debug!("{}", e.source);
            embassy::sound::BEETHOVEN.play().await?;
            #[cfg(feature = "avahi")]
            let _mdns = MdnsController::init().await?;

            let ctx = DiagnosticContext::init(
                cfg_path,
                if tokio::fs::metadata("/media/embassy/config/disk.guid")
                    .await
                    .is_ok()
                {
                    Some(Arc::new(
                        tokio::fs::read_to_string("/media/embassy/config/disk.guid") // unique identifier for volume group - keeps track of the disk that goes with your embassy
                            .await?
                            .trim()
                            .to_owned(),
                    ))
                } else {
                    None
                },
                e,
            )
            .await?;

            let embassy_ip = get_current_ip(ctx.ethernet_interface.to_owned()).await?;
            let embassy_ip_fqdn: ResourceFqdn = embassy_ip.parse()?;
            let embassy_fqdn: ResourceFqdn = "embassy.local".parse()?;

            let localhost_fqdn = ResourceFqdn::LocalHost;

            let diag_ui_handler = diag_ui_file_router(ctx.clone()).await?;

            let mut diag_http_server =
                EmbassyServiceHTTPServer::new([0, 0, 0, 0].into(), 80, None).await?;
            diag_http_server
                .add_svc_handler_mapping(embassy_ip_fqdn, diag_ui_handler.clone())
                .await?;
            diag_http_server
                .add_svc_handler_mapping(embassy_fqdn, diag_ui_handler.clone())
                .await?;

            diag_http_server
                .add_svc_handler_mapping(localhost_fqdn, diag_ui_handler.clone())
                .await?;

            let shutdown = ctx.shutdown.subscribe().recv().await.unwrap();
            diag_http_server.shutdown.send(()).unwrap();
            Ok(shutdown)
        }
        .await
    } else {
        Ok(None)
    };

    run_script_if_exists("/media/embassy/config/postinit.sh").await;

    res
}

fn main() {
    let matches = clap::App::new("embassyd")
        .arg(
            clap::Arg::with_name("config")
                .short('c')
                .long("config")
                .takes_value(true),
        )
        .get_matches();

    EmbassyLogger::init();

    let cfg_path = matches.value_of("config").map(|p| Path::new(p).to_owned());
    let res = {
        let rt = tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .expect("failed to initialize runtime");
        rt.block_on(inner_main(cfg_path))
    };

    match res {
        Ok(Some(shutdown)) => shutdown.execute(),
        Ok(None) => (),
        Err(e) => {
            eprintln!("{}", e.source);
            tracing::debug!("{:?}", e.source);
            drop(e.source);
            std::process::exit(e.kind as i32)
        }
    }
}