lumni 0.0.5

Portable file-utility for object-stores
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

use std::collections::HashMap;
use std::env;

use clap::{Arg, Command};
use tokio::runtime::Builder;
use crate::EnvironmentConfig;

use crate::subcommands::cp::*;
use crate::subcommands::ls::*;
use crate::subcommands::query::*;
use crate::subcommands::request::*;

const PROGRAM_NAME: &str = "lumni";

pub fn run_cli(args: Vec<String>) {
    env_logger::init();
    let app = Command::new(PROGRAM_NAME)
        .version(env!("CARGO_PKG_VERSION"))
        .arg_required_else_help(true)
        .about(format!(
            "List objects in an S3 bucket\n\nExample:\n {} ls \
             s3://bucket-name/ --max-files 100",
            PROGRAM_NAME
        ))
        .arg(
            Arg::new("region")
                .long("region")
                .short('r')
                .help("Region to use"),
        )
        .subcommand(request_subcommand()) // "-X/--request [GET,PUT]"
        .subcommand(query_subcommand()) // "-Q/--query [SELECT,DESCRIBE]"
        .subcommand(ls_subcommand()) // "ls [URI]"
        .subcommand(cp_subcommand()); // "cp" [SOURCE] [TARGET]

    let matches = app.try_get_matches_from(args).unwrap_or_else(|e| {
        e.exit();
    });

    let mut config = create_initial_config(&matches);
    let rt = Builder::new_current_thread().enable_all().build().unwrap();

    match matches.subcommand() {
        Some(("-X", matches)) => {
            rt.block_on(handle_request(matches, &mut config));
        }
        Some(("-Q", matches)) => {
            rt.block_on(handle_query(matches, &mut config));
        }
        Some(("ls", matches)) => {
            rt.block_on(handle_ls(matches, &mut config));
        }
        Some(("cp", matches)) => {
            rt.block_on(handle_cp(matches, &mut config));
        }
        _ => {
            eprintln!("No valid subcommand provided");
        }
    }
}

fn create_initial_config(matches: &clap::ArgMatches) -> EnvironmentConfig {
    let mut config_hashmap = HashMap::new();
    if let Some(region) = matches.get_one::<String>("region") {
        config_hashmap.insert("region".to_string(), region.to_string());
    }

    // Create a Config instance
    EnvironmentConfig::new(config_hashmap)
}