dioxus-mirror/packages/cli/src/main.rs

use std::path::PathBuf;

use anyhow::anyhow;
use clap::Parser;
use dioxus_cli::*;

#[cfg(feature = "plugin")]
use dioxus_cli::plugin::PluginManager;

use Commands::*;

fn get_bin(bin: Option<String>) -> Result<PathBuf> {
    let metadata = cargo_metadata::MetadataCommand::new()
        .exec()
        .map_err(Error::CargoMetadata)?;
    let package = if let Some(bin) = bin {
        metadata
            .workspace_packages()
            .into_iter()
            .find(|p| p.name == bin)
            .ok_or(format!("no such package: {}", bin))
            .map_err(Error::CargoError)?
    } else {
        metadata
            .root_package()
            .ok_or("no root package?".into())
            .map_err(Error::CargoError)?
    };

    let crate_dir = package
        .manifest_path
        .parent()
        .ok_or("couldn't take parent dir".into())
        .map_err(Error::CargoError)?;

    Ok(crate_dir.into())
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let args = Cli::parse();

    set_up_logging();

    let bin = get_bin(args.bin);

    if let Ok(bin) = &bin {
        let _dioxus_config = DioxusConfig::load(Some(bin.clone()))
        .map_err(|e| anyhow!("Failed to load Dioxus config because: {e}"))?
        .unwrap_or_else(|| {
            log::warn!("You appear to be creating a Dioxus project from scratch; we will use the default config");
            DioxusConfig::default()
        });

        #[cfg(feature = "plugin")]
        PluginManager::init(_dioxus_config.plugin)
            .map_err(|e| anyhow!("🚫 Plugin system initialization failed: {e}"))?;
    }

    match args.action {
        Translate(opts) => opts
            .translate()
            .map_err(|e| anyhow!("🚫 Translation of HTML into RSX failed: {}", e)),

        Build(opts) if bin.is_ok() => opts
            .build(Some(bin.unwrap().clone()))
            .map_err(|e| anyhow!("🚫 Building project failed: {}", e)),

        Clean(opts) if bin.is_ok() => opts
            .clean(Some(bin.unwrap().clone()))
            .map_err(|e| anyhow!("🚫 Cleaning project failed: {}", e)),

        Serve(opts) if bin.is_ok() => opts
            .serve(Some(bin.unwrap().clone()))
            .await
            .map_err(|e| anyhow!("🚫 Serving project failed: {}", e)),

        Create(opts) => opts
            .create()
            .map_err(|e| anyhow!("🚫 Creating new project failed: {}", e)),

        Config(opts) => opts
            .config()
            .map_err(|e| anyhow!("🚫 Configuring new project failed: {}", e)),

        Bundle(opts) if bin.is_ok() => opts
            .bundle(Some(bin.unwrap().clone()))
            .map_err(|e| anyhow!("🚫 Bundling project failed: {}", e)),

        #[cfg(feature = "plugin")]
        Plugin(opts) => opts
            .plugin()
            .await
            .map_err(|e| anyhow!("🚫 Error with plugin: {}", e)),

        Autoformat(opts) => opts
            .autoformat()
            .await
            .map_err(|e| anyhow!("🚫 Error autoformatting RSX: {}", e)),

        Check(opts) => opts
            .check()
            .await
            .map_err(|e| anyhow!("🚫 Error checking RSX: {}", e)),

        Version(opt) => {
            let version = opt.version();
            println!("{}", version);

            Ok(())
        }
        _ => Err(anyhow::anyhow!(bin.unwrap_err())),
    }
}