use console::style;
use eyre::{eyre, Report, Result, WrapErr};
use rayon::prelude::*;
use crate::config::Settings;
use crate::plugins;
use crate::toolset::install_state;
use crate::ui::multi_progress_report::MultiProgressReport;
#[derive(Debug, clap::Args)]
#[clap(verbatim_doc_comment, visible_aliases = ["up", "upgrade"], after_long_help = AFTER_LONG_HELP)]
pub struct Update {
#[clap()]
plugin: Option<Vec<String>>,
#[clap(long, short, verbatim_doc_comment)]
jobs: Option<usize>,
}
impl Update {
pub fn run(self) -> Result<()> {
let plugins: Vec<_> = match self.plugin {
Some(plugins) => plugins
.into_iter()
.map(|p| match p.split_once('#') {
Some((p, ref_)) => (p.to_string(), Some(ref_.to_string())),
None => (p, None),
})
.collect(),
None => install_state::list_plugins()?
.keys()
.map(|p| (p.clone(), None))
.collect::<Vec<_>>(),
};
let settings = Settings::try_get()?;
let mpr = MultiProgressReport::get();
let mut errors = rayon::ThreadPoolBuilder::new()
.num_threads(self.jobs.unwrap_or(settings.jobs))
.build()?
.install(|| {
plugins
.into_par_iter()
.map(|(short, ref_)| {
let plugin = plugins::get(&short)?;
let prefix = format!("plugin:{}", style(plugin.name()).blue().for_stderr());
let pr = mpr.add(&prefix);
plugin
.update(&pr, ref_)
.wrap_err_with(|| format!("[{plugin}] plugin update"))?;
Ok(())
})
.filter_map(|r| r.err())
.collect::<Vec<_>>()
});
if errors.is_empty() {
Ok(())
} else if errors.len() == 1 {
Err(errors.pop().unwrap())
} else {
let err = eyre!("{} plugins failed to update", errors.len());
Err(errors
.into_iter()
.fold(err, |report: Report, e| report.wrap_err(e)))
}
}
}
static AFTER_LONG_HELP: &str = color_print::cstr!(
r#"<bold><underline>Examples:</underline></bold>
$ <bold>mise plugins update</bold> # update all plugins
$ <bold>mise plugins update node</bold> # update only node
$ <bold>mise plugins update node#beta</bold> # specify a ref
"#
);