oxen-cli 0.50.4

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

use async_trait::async_trait;
use clap::Command;

use liboxen::error::OxenError;
use liboxen::model::LocalRepository;
use liboxen::repositories;

use std::path::PathBuf;

use crate::cmd::{RunCmd, restore::restore_args};

pub const NAME: &str = "restore";
pub struct RemoteModeRestoreCmd;

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

    fn args(&self) -> Command {
        restore_args()
    }

    // TODO: Support multiple paths
    async fn run(&self, args: &clap::ArgMatches) -> Result<(), anyhow::Error> {
        let paths: Vec<PathBuf> = args
            .get_many::<String>("paths")
            .expect("Must supply paths")
            .map(|p| -> Result<PathBuf, OxenError> {
                let path = PathBuf::from(p);
                Ok(path)
            })
            .collect::<Result<Vec<PathBuf>, OxenError>>()?;

        let repo = LocalRepository::from_current_dir()?;

        let revision = if let Some(source) = args.get_one::<String>("source") {
            source.to_string()
        } else {
            let head_commit = repositories::commits::head_commit(&repo)?;
            head_commit.id
        };

        repositories::remote_mode::restore(&repo, &paths, &revision).await?;

        Ok(())
    }
}