oxen-cli 0.50.6

Oxen is a fast, unstructured data version control, to help version large machine learning datasets written in Rust.
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

use async_trait::async_trait;
use clap::{Arg, ArgMatches, Command};

use liboxen::error::OxenError;
use liboxen::model::LocalRepository;
use liboxen::repositories;
use liboxen::util;
use std::path::Path;

use crate::cmd::RunCmd;
pub const NAME: &str = "save";
pub struct SaveCmd;

#[async_trait]
impl RunCmd for SaveCmd {
    fn name(&self) -> &str {
        NAME
    }

    fn args(&self) -> Command {
        Command::new(NAME)
            .arg(
                Arg::new("PATH")
                    .help("Path of the local repository to save")
                    .required(true)
                    .index(1), // This represents the position of the argument in the command line command.
            )
            .arg(
                Arg::new("output")
                    .help("Name of the output .tar.gz archive")
                    .short('o')
                    .long("output")
                    .required(true),
            )
    }

    async fn run(&self, args: &ArgMatches) -> Result<(), anyhow::Error> {
        let repo_str = args.get_one::<String>("PATH").expect("Required");
        let output_str = args.get_one::<String>("output").expect("Required");

        let output_path = Path::new(output_str);
        let repo_path = Path::new(repo_str);
        let repo_dir = util::fs::get_repo_root(repo_path)
            .ok_or_else(|| OxenError::local_repo_not_found(repo_path))?;
        let repo = LocalRepository::from_dir(&repo_dir)?;

        repositories::save(&repo, output_path)?;

        Ok(())
    }
}