Function main

vmm/src/main.rs:149–236

()

#[rocket::main]
async fn main() -> Result<()> {
    {
        use tracing_subscriber::{fmt, EnvFilter};
        let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
        fmt().with_env_filter(filter).with_ansi(false).init();
    }

    let args = Args::parse();
    let figment = config::load_config_figment(args.config.as_deref());
    let config = Config::extract_or_default(&figment)?.abs_path()?;

    // Validate host API configuration
    config
        .host_api
        .validate()
        .context("Invalid host_api configuration")?;

    // Handle commands
    match args.command.unwrap_or_default() {
        Command::Run(run_args) => {
            // One-shot VM execution mode
            return one_shot::run_one_shot(
                &run_args.vm_config,
                config,
                run_args.workdir,
                run_args.dry_run,
            )
            .await;
        }
        Command::Serve => {
            // Default server mode - continue to main server logic
        }
    }

    // Register this VMM instance for local discovery
    discovery::cleanup_stale_registrations();
    let listen_address = {
        // Use Rocket's Endpoint type to parse the address exactly as Rocket would,
        // then override the port with the figment's port value (matching Rocket's behavior).
        let endpoint: rocket::listener::Endpoint =
            figment.extract_inner("address").unwrap_or_default();
        match endpoint.tcp() {
            Some(addr) => {
                let port: u16 = figment.extract_inner("port").unwrap_or(addr.port());
                format!("{}:{port}", addr.ip())
            }
            None => endpoint.to_string(),
        }
    };
    let _discovery_reg = discovery::DiscoveryRegistration::register(
        &listen_address,
        args.config.as_deref(),
        &config.image.path,
        &config.run_path,
        &config.node_name,
        &app_version(),
    )
    .context("failed to register VMM instance for discovery")?;