oxen-cli 0.51.2

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

use async_trait::async_trait;
use clap::{Arg, Command};
use liboxen::model::LocalRepository;
use liboxen::repositories;
use std::path::Path;
use tokio::io::AsyncWriteExt;
use tokio_stream::StreamExt;

use crate::cmd::RunCmd;
use crate::helpers::path_relative_to_repo;

pub const NAME: &str = "cat";
pub struct CatCmd;

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

    fn args(&self) -> Command {
        Command::new(NAME)
            .about("Print the raw bytes of a file at a revision to stdout")
            .arg(
                Arg::new("path")
                    .required(true)
                    .help("Path to the file, relative to the current directory"),
            )
            .arg(
                Arg::new("revision")
                    .long("revision")
                    .short('r')
                    .help("The branch name or commit id to read from. Defaults to HEAD.")
                    .default_value("HEAD")
                    .action(clap::ArgAction::Set),
            )
    }

    async fn run(&self, args: &clap::ArgMatches) -> Result<(), anyhow::Error> {
        let repository = LocalRepository::from_current_dir()?;
        let path = args
            .get_one::<String>("path")
            .ok_or_else(|| anyhow::anyhow!("Must supply a path"))?;
        let revision = args
            .get_one::<String>("revision")
            .ok_or_else(|| anyhow::anyhow!("Must supply a revision"))?;

        let repo_path = path_relative_to_repo(&repository, Path::new(path))?;

        let mut stream = repositories::revisions::get_version_stream_from_revision(
            &repository,
            revision,
            repo_path,
        )
        .await?;

        // Unbounded streamed write: use the large-buffer convention and flush explicitly, since
        // `BufWriter`'s `Drop` does not flush.
        let mut stdout = tokio::io::BufWriter::with_capacity(10 * 1024 * 1024, tokio::io::stdout());
        while let Some(chunk) = stream.next().await {
            stdout.write_all(&chunk?).await?;
        }
        stdout.flush().await?;

        Ok(())
    }
}