oxen-cli 0.48.1

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

use async_trait::async_trait;
use clap::{Arg, Command, arg};
use std::collections::HashMap;

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

use crate::cmd::RunCmd;
pub const NAME: &str = "schemas";

pub mod add;
pub use add::SchemasAddCmd;

pub mod list;
pub use list::SchemasListCmd;

pub mod rm;
pub use rm::SchemasRmCmd;

pub mod show;
pub use show::SchemasShowCmd;

pub struct SchemasCmd;

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

    fn args(&self) -> Command {
        // Setups the CLI args for the command
        let mut command = Command::new(NAME)
            .about("Manage schemas that are created from committing tabular data")
            .arg(arg!([SCHEMA_REF] "Name, hash, or path of the schema you want to view in more detail."))
            .arg(
                Arg::new("staged")
                    .long("staged")
                    .help("Show the staged schema")
                    .action(clap::ArgAction::SetTrue),
            )
            .arg(
                Arg::new("flatten")
                    .long("flatten")
                    .help("Print the schema in a flattened format")
                    .action(clap::ArgAction::SetTrue),
            )
        ;

        // These are all the subcommands for the schemas command
        // including `add`, `name`, `list`, and `rm`
        let sub_commands = self.get_subcommands();
        for cmd in sub_commands.values() {
            command = command.subcommand(cmd.args());
        }
        command
    }

    async fn run(&self, args: &clap::ArgMatches) -> Result<(), OxenError> {
        let sub_commands = self.get_subcommands();
        if let Some((name, sub_matches)) = args.subcommand() {
            let Some(cmd) = sub_commands.get(name) else {
                return Err(OxenError::unknown_subcommand("schemas", name));
            };

            // Calling await within an await is making it complain?
            tokio::task::block_in_place(|| {
                tokio::runtime::Handle::current().block_on(cmd.run(sub_matches))
            })?;
        } else if let Some(schema_ref) = args.get_one::<String>("SCHEMA_REF") {
            let repository = LocalRepository::from_current_dir()?;
            let staged = args.get_flag("staged");
            let verbose = !args.get_flag("flatten"); // default to verbose
            let val =
                repositories::data_frames::schemas::show(&repository, schema_ref, staged, verbose)?;
            println!("{val}");
        } else {
            // Fall back to list schemas
            let Some(cmd) = sub_commands.get("list") else {
                return Err(OxenError::unknown_subcommand("schemas", "list"));
            };
            tokio::task::block_in_place(|| {
                tokio::runtime::Handle::current().block_on(cmd.run(args))
            })?;
        }
        Ok(())
    }
}

impl SchemasCmd {
    fn get_subcommands(&self) -> HashMap<String, Box<dyn RunCmd>> {
        let commands: Vec<Box<dyn RunCmd>> = vec![
            Box::new(SchemasAddCmd),
            Box::new(SchemasListCmd),
            Box::new(SchemasRmCmd),
            Box::new(SchemasShowCmd),
        ];
        let mut runners: HashMap<String, Box<dyn RunCmd>> = HashMap::new();
        for cmd in commands {
            runners.insert(cmd.name().to_string(), cmd);
        }
        runners
    }
}