oxen-cli 0.48.3

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
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

use std::path::PathBuf;

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

use liboxen::{api, core::oxenignore, error::OxenError, model::LocalRepository};

use crate::cmd::{RunCmd, add::add_args};
pub const NAME: &str = "add";
pub struct WorkspaceAddCmd;

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

    fn args(&self) -> Command {
        // If in remote repo, workspace id can be found from the config instead
        let is_remote_repo = match LocalRepository::from_current_dir() {
            Ok(repo) => repo.is_remote_mode(),
            Err(_) => false,
        };

        add_args()
            .arg(
                Arg::new("workspace-id")
                    .long("workspace-id")
                    .short('w')
                    .help("The workspace ID of the workspace")
                    .conflicts_with("workspace-name"),
            )
            .arg(
                Arg::new("workspace-name")
                    .long("workspace-name")
                    .short('n')
                    .help("The name of the workspace")
                    .conflicts_with("workspace-id"),
            )
            .group(
                ArgGroup::new("workspace-identifier")
                    .args(["workspace-id", "workspace-name"])
                    .required(!is_remote_repo),
            )
            .arg(
                Arg::new("directory")
                    .long("directory")
                    .short('d')
                    .help("The destination directory to add the workspace to")
                    .default_value("."),
            )
            .arg_required_else_help(true)
    }

    async fn run(&self, args: &ArgMatches) -> Result<(), OxenError> {
        // Parse Args
        let mut paths: Vec<PathBuf> = args
            .get_many::<String>("files")
            .expect("Must supply files")
            .map(PathBuf::from)
            .collect();

        let repository = LocalRepository::from_current_dir()?;

        let (workspace_identifier, directory) = if repository.is_remote_mode() {
            (&repository.workspace_name.clone().unwrap(), ".")
        } else {
            let directory = args.get_one::<String>("directory").unwrap(); // safe to unwrap because we have a default value

            let workspace_name = args.get_one::<String>("workspace-name");
            let workspace_id = args.get_one::<String>("workspace-id");
            match workspace_id {
                Some(id) => (id, directory.as_str()),
                None => {
                    // If no ID is provided, try to get the workspace by name
                    if let Some(name) = workspace_name {
                        (name, directory.as_str())
                    } else {
                        return Err(OxenError::basic_str(
                            "Either workspace-id or workspace-name must be provided.",
                        ));
                    }
                }
            }
        };

        let remote_repo = api::client::repositories::get_default_remote(&repository).await?;

        // Handle .oxenignore filtering
        let ignore = oxenignore::create(&repository);
        if let Some(ignore) = &ignore {
            paths.retain(|path| !ignore.matched(path, path.is_dir()).is_ignore());
        }

        // If no paths left after filtering, return early
        if paths.is_empty() {
            return Err(OxenError::basic_str(
                "No files to add after filtering with .oxenignore.",
            ));
        }

        api::client::workspaces::files::add(
            &remote_repo,
            workspace_identifier,
            directory,
            paths,
            &None,
        )
        .await?;

        Ok(())
    }
}