mirror of
https://github.com/esp-rs/espup.git
synced 2025-09-26 20:30:28 +00:00
5c4f586558
* Added a CompletionShell enum and implemented required methods for it * CompletionsOpts now uses CompletionShell enum to be able to call NuShell as well * CompletionsOpts now uses CompletionShell enum to be able to call NuShell as well * CompletionShell implements the generate method - simplifies the code * Added clap_complete_nushell as a dependency * Updated readme * Added nushell completion * Cargo fmt errors
94 lines
2.8 KiB
Rust
94 lines
2.8 KiB
Rust
use clap::{CommandFactory, Parser};
|
|
use espup::{
|
|
cli::{CompletionsOpts, InstallOpts, UninstallOpts},
|
|
logging::initialize_logger,
|
|
toolchain::{
|
|
InstallMode,
|
|
gcc::uninstall_gcc_toolchains,
|
|
install as toolchain_install,
|
|
llvm::Llvm,
|
|
remove_dir,
|
|
rust::{XtensaRust, get_rustup_home},
|
|
},
|
|
update::check_for_update,
|
|
};
|
|
use log::info;
|
|
use miette::Result;
|
|
use std::{env, io::stdout};
|
|
|
|
#[derive(Parser)]
|
|
#[command(about, version)]
|
|
struct Cli {
|
|
#[command(subcommand)]
|
|
subcommand: SubCommand,
|
|
}
|
|
|
|
#[derive(Parser)]
|
|
pub enum SubCommand {
|
|
/// Generate completions for the given shell.
|
|
Completions(CompletionsOpts),
|
|
/// Installs Espressif Rust ecosystem.
|
|
// We use a Box here to make clippy happy (see https://rust-lang.github.io/rust-clippy/master/index.html#large_enum_variant)
|
|
Install(Box<InstallOpts>),
|
|
/// Uninstalls Espressif Rust ecosystem.
|
|
Uninstall(UninstallOpts),
|
|
/// Updates Xtensa Rust toolchain.
|
|
Update(Box<InstallOpts>),
|
|
}
|
|
|
|
/// Updates Xtensa Rust toolchain.
|
|
async fn completions(args: CompletionsOpts) -> Result<()> {
|
|
initialize_logger(&args.log_level);
|
|
check_for_update(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
|
|
|
|
info!("Generating completions for {} shell", args.shell);
|
|
|
|
args.shell
|
|
.generate(&mut Cli::command(), "espup", &mut stdout());
|
|
|
|
info!("Completions successfully generated!");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Installs or updates the Rust for ESP chips environment
|
|
async fn install(args: InstallOpts, install_mode: InstallMode) -> Result<()> {
|
|
initialize_logger(&args.log_level);
|
|
check_for_update(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
|
|
|
|
toolchain_install(args, install_mode).await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Uninstalls the Rust for ESP chips environment
|
|
async fn uninstall(args: UninstallOpts) -> Result<()> {
|
|
initialize_logger(&args.log_level);
|
|
check_for_update(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
|
|
|
|
info!("Uninstalling the Espressif Rust ecosystem");
|
|
let toolchain_dir = get_rustup_home().join("toolchains").join(args.name);
|
|
|
|
if toolchain_dir.exists() {
|
|
Llvm::uninstall(&toolchain_dir).await?;
|
|
|
|
uninstall_gcc_toolchains(&toolchain_dir, args.crosstool_toolchain_version).await?;
|
|
|
|
XtensaRust::uninstall(&toolchain_dir).await?;
|
|
|
|
remove_dir(&toolchain_dir).await?;
|
|
}
|
|
|
|
info!("Uninstallation successfully completed!");
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
match Cli::parse().subcommand {
|
|
SubCommand::Completions(args) => completions(args).await,
|
|
SubCommand::Install(args) => install(*args, InstallMode::Install).await,
|
|
SubCommand::Update(args) => install(*args, InstallMode::Update).await,
|
|
SubCommand::Uninstall(args) => uninstall(args).await,
|
|
}
|
|
}
|