schematools-cli 1.0.2

Tools for codegen, preprocessing and validation of json-schema and openapi spec
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
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use std::{fmt::Display, fs};

use clap::Parser;

#[cfg(feature = "git")]
use schematools::discovery::discover_git;
use schematools::{
    discovery::{Discovery, GitCheckoutType, Registry},
    error::Error,
    hash,
};

#[derive(Clone, Debug, Parser)]
pub struct Opts {
    #[clap(subcommand)]
    command: Command,
}

impl Display for Opts {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self.command {
            Command::Add(_) => write!(f, "add"),
        }
    }
}

#[derive(Clone, Debug, Parser)]
enum Command {
    /// Adds registry
    Add(AddOpts),
}

#[derive(Clone, Debug, Parser)]
struct AddOpts {
    /// Registry name used in codegen name::
    name: String,

    /// Uri of git registry
    uri: String,

    /// Git tag
    #[clap(long)]
    tag: Option<String>,

    /// Git branch
    #[clap(long)]
    branch: Option<String>,

    /// Git revision
    #[clap(long)]
    rev: Option<String>,

    /// Lock checksum
    #[clap(long)]
    lock: Option<String>,

    /// Skip cache during checkout
    #[clap(long)]
    no_cache: bool,
}

impl Opts {
    pub fn run(&self, discovery: &mut Discovery) -> Result<(), Error> {
        match &self.command {
            Command::Add(opts) => {
                log::info!("discovering: {}", opts.uri);

                let registry = if opts.uri.starts_with('.') {
                    add_local_registry(opts)
                } else {
                    #[cfg(not(feature = "git"))]
                    panic!("git feature not built; only local registry supported");
                    #[cfg(feature = "git")]
                    add_git_registry(opts)
                }?;

                if let Some(lock) = &opts.lock {
                    log::info!("calculating registry hash...");

                    let hash = hash::calculate::<sha2::Sha256>(&registry.path)?;
                    let calculated: &[u8] = hash.as_ref();
                    let calculated = calculated
                        .iter()
                        .map(|b| format!("{b:02x}"))
                        .collect::<String>();

                    if !calculated.eq(lock) {
                        return Err(Error::DiscoveryInvalidLock(lock.clone(), calculated));
                    }
                }

                discovery.register(opts.name.clone(), registry);
            }
        }

        Ok(())
    }
}

fn add_local_registry(opts: &AddOpts) -> Result<Registry, Error> {
    let path = fs::canonicalize(&opts.uri).map_err(Error::RegistryLocalIoError)?;

    let md = fs::metadata(&path).map_err(Error::RegistryLocalIoError)?;

    if md.is_dir() {
        Ok(Registry::new(path))
    } else {
        Err(Error::RegistryLocalPathNotDirError(path))
    }
}

#[cfg(feature = "git")]
fn add_git_registry(opts: &AddOpts) -> Result<Registry, Error> {
    let checkout = if let Some(branch) = opts.branch.clone() {
        Ok(GitCheckoutType::Branch(branch))
    } else if let Some(tag) = opts.tag.clone() {
        Ok(GitCheckoutType::Tag(tag))
    } else if let Some(rev) = opts.rev.clone() {
        Ok(GitCheckoutType::Rev(rev))
    } else {
        Err(Error::RegistryMissingRevTagBranch)
    }?;

    let registry = discover_git(&opts.uri, checkout, opts.no_cache)?;

    Ok(registry)
}