rattler-bin 0.1.4

Rust binary for common Conda operations
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use std::{fs, path::PathBuf};

use clap::Parser;
use miette::IntoDiagnostic;
use rattler_conda_types::{menuinst::MenuMode, PackageName, Platform, PrefixRecord};

/// Install menu items for an installed package.
#[derive(Debug, Parser)]
pub struct InstallOpt {
    /// Target prefix to look for the package (defaults to `.prefix`)
    #[clap(long, short, default_value = ".prefix")]
    target_prefix: PathBuf,

    /// Name of the package for which to install menu items
    package_name: PackageName,
}

/// Remove installed menu items for a package.
#[derive(Debug, Parser)]
pub struct RemoveOpt {
    /// Target prefix to look for the package (defaults to `.prefix`)
    #[clap(long, short, default_value = ".prefix")]
    target_prefix: PathBuf,

    /// Name of the package for which to remove menu items
    package_name: PackageName,
}

pub async fn install_menu(opts: InstallOpt) -> miette::Result<()> {
    // Find the prefix record in the target_prefix and call `install_menu` on it
    let records: Vec<PrefixRecord> =
        PrefixRecord::collect_from_prefix(&opts.target_prefix).into_diagnostic()?;

    let record = records
        .iter()
        .find(|r| r.repodata_record.package_record.name == opts.package_name)
        .ok_or_else(|| {
            miette::miette!(
                "Package {} not found in prefix {:?}",
                opts.package_name.as_normalized(),
                opts.target_prefix
            )
        })?;
    let prefix = fs::canonicalize(&opts.target_prefix).into_diagnostic()?;
    rattler_menuinst::install_menuitems_for_record(
        &prefix,
        record,
        Platform::current(),
        MenuMode::User,
    )
    .into_diagnostic()?;

    Ok(())
}

pub async fn remove_menu(opts: RemoveOpt) -> miette::Result<()> {
    // Find the prefix record in the target_prefix and call `remove_menu` on it
    let records: Vec<PrefixRecord> =
        PrefixRecord::collect_from_prefix(&opts.target_prefix).into_diagnostic()?;

    let record = records
        .iter()
        .find(|r| r.repodata_record.package_record.name == opts.package_name)
        .ok_or_else(|| {
            miette::miette!(
                "Package {} not found in prefix {:?}",
                opts.package_name.as_normalized(),
                opts.target_prefix
            )
        })?;

    rattler_menuinst::remove_menu_items(&record.installed_system_menus).into_diagnostic()?;

    Ok(())
}