s3cli 0.1.1

CLI-first S3 storage for developers and AI agents
Documentation 
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

use std::io::{self, Write};
use std::path::Path;
use std::sync::Arc;

use crate::config::Config;
use crate::storage::Storage;
use super::CmdResult;

pub async fn pull(
    config: &Config,
    id: String,
    output: Option<String>,
    force: bool,
) -> CmdResult<()> {
    let storage = get_storage(config).await?;

    let data = storage.get(&id).await?;

    let output_path = match output {
        Some(ref p) => p.clone(),
        None => {
            Path::new(&id)
                .file_name()
                .and_then(|n| n.to_str())
                .unwrap_or("output")
                .to_string()
        }
    };

    let path = Path::new(&output_path);

    if path.exists() && !force {
        return Err(super::CommandError::InvalidInput(
            format!("File '{}' already exists. Use --force to overwrite.", output_path)
        ));
    }

    if output.is_none() || output_path == "-" {
        io::stdout().write_all(&data.data)?;
    } else {
        if let Some(parent) = path.parent() {
            if !parent.as_os_str().is_empty() {
                std::fs::create_dir_all(parent)?;
            }
        }
        std::fs::write(&output_path, &data.data)?;
        println!("Saved to {}", output_path);
    }

    Ok(())
}

async fn get_storage(config: &Config) -> CmdResult<Arc<dyn Storage>> {
    use crate::storage::S3Storage;
    
    let storage = S3Storage::new(config).await?;
    Ok(Arc::new(storage))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_output_path_from_id() {
        let id = "uploads/test.png";
        let output = None;
        
        let output_path = output.unwrap_or_else(|| {
            Path::new(id)
                .file_name()
                .and_then(|n| n.to_str())
                .unwrap_or("output")
                .to_string()
        });
        
        assert_eq!(output_path, "test.png");
    }

    #[test]
    fn test_output_path_with_explicit() {
        let id = "uploads/test.png";
        let output = Some("custom/path/file.png".to_string());
        
        let output_path = output.unwrap_or_else(|| {
            Path::new(id)
                .file_name()
                .and_then(|n| n.to_str())
                .unwrap_or("output")
                .to_string()
        });
        
        assert_eq!(output_path, "custom/path/file.png");
    }
}