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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use std::path::PathBuf;

use miette::{Context, IntoDiagnostic};
use url::Url;

/// Extract a local or remote conda package.
#[derive(Debug, clap::Parser)]
pub struct Opt {
    /// Path or URL to the conda package archive (.tar.bz2 or .conda)
    #[clap(required = true)]
    package: String,

    /// Destination directory where the package will be extracted
    /// If not specified, extracts to a directory with the same name as the package
    #[clap(short, long)]
    destination: Option<PathBuf>,
}

/// Strips package extensions (.tar.bz2 or .conda) from a filename
fn strip_package_extension(filename: &str) -> String {
    if let Some(stripped) = filename.strip_suffix(".tar.bz2") {
        stripped.to_string()
    } else if let Some(stripped) = filename.strip_suffix(".conda") {
        stripped.to_string()
    } else {
        filename.to_string()
    }
}

/// Determines the destination directory from a URL
fn determine_destination_from_url(url: &Url) -> miette::Result<PathBuf> {
    // Extract filename from URL path
    let filename = url
        .path_segments()
        .and_then(Iterator::last)
        .ok_or_else(|| miette::miette!("Could not extract package name from URL"))?;

    let package_name = strip_package_extension(filename);
    Ok(PathBuf::from(package_name))
}

/// Extracts a conda package from a URL
async fn extract_from_url(
    url: Url,
    destination: Option<PathBuf>,
    package_display: &str,
) -> miette::Result<(PathBuf, rattler_package_streaming::ExtractResult)> {
    let destination = destination.map_or_else(|| determine_destination_from_url(&url), Ok)?;

    println!(
        "Extracting {} to {}",
        package_display,
        destination.display()
    );

    let client = super::client::create_client_with_middleware()?;

    let result =
        rattler_package_streaming::reqwest::tokio::extract(client, url, &destination, None, None)
            .await
            .into_diagnostic()
            .with_context(|| format!("Failed to extract package from URL: {package_display}"))?;

    Ok((destination, result))
}

/// Determines the destination directory from a file path
fn determine_destination_from_path(package_path: &str) -> miette::Result<PathBuf> {
    let path = PathBuf::from(package_path);
    let package_name = path
        .file_stem()
        .and_then(|s| s.to_str())
        .ok_or_else(|| miette::miette!("Invalid package filename"))?
        .to_string();

    Ok(PathBuf::from(package_name))
}

/// Extracts a conda package from a local file path
fn extract_from_path(
    package_path: &str,
    destination: Option<PathBuf>,
) -> miette::Result<(PathBuf, rattler_package_streaming::ExtractResult)> {
    let destination =
        destination.map_or_else(|| determine_destination_from_path(package_path), Ok)?;

    println!("Extracting {} to {}", package_path, destination.display());

    let result = rattler_package_streaming::fs::extract(&PathBuf::from(package_path), &destination)
        .into_diagnostic()
        .with_context(|| format!("Failed to extract package: {package_path}"))?;

    Ok((destination, result))
}

pub async fn extract(opt: Opt) -> miette::Result<()> {
    // Try to parse as URL, otherwise treat as file path
    let (destination, result) = if let Ok(url) = Url::parse(&opt.package) {
        extract_from_url(url, opt.destination, &opt.package).await?
    } else {
        extract_from_path(&opt.package, opt.destination)?
    };

    println!(
        "{} Successfully extracted package",
        console::style("✓").green(),
    );
    println!("  Destination: {}", destination.display());
    println!("  SHA256: {:x}", result.sha256);
    println!("  MD5: {:x}", result.md5);
    println!("  Size: {} bytes", result.total_size);

    Ok(())
}