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

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

use liboxen::model::LocalRepository;
use liboxen::repositories;

use crate::cmd::RunCmd;

pub const NAME: &str = "checkout";
pub struct RemoteModeCheckoutCmd;

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

    fn args(&self) -> Command {
        Command::new(NAME)
            .about("Checks out a branch in the repository")
            .arg(Arg::new("name").help("Name of the branch or commit id to checkout"))
            .arg(
                Arg::new("create")
                    .long("create")
                    .short('b')
                    .value_name("BRANCH_NAME")
                    .num_args(1),
            )
            .group(
                clap::ArgGroup::new("checkout_args")
                    .args(["name", "create"])
                    .required(true),
            )
    }

    async fn run(&self, args: &clap::ArgMatches) -> Result<(), anyhow::Error> {
        let mut repo = LocalRepository::from_current_dir()?;

        // Parse Args
        if let Some(name) = args.get_one::<String>("create") {
            repositories::remote_mode::create_checkout(&mut repo, name).await?
        } else if let Some(name) = args.get_one::<String>("name") {
            repositories::remote_mode::checkout(&mut repo, name).await?;
        }

        Ok(())
    }
}