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

use crate::config::Config;
use crate::util::spacetime_dns;
use clap::Arg;
use clap::ArgAction::SetTrue;
use clap::ArgMatches;

pub fn cli() -> clap::Command {
    clap::Command::new("describe")
        .about("Describe the structure of a database or entities within it")
        .arg(Arg::new("database").required(true))
        .arg(
            Arg::new("entity_type")
                .required(false)
                .value_parser(["reducer", "table", "repeater"]),
        )
        .arg(Arg::new("entity_name").required(false).requires("entity_type"))
        .arg(Arg::new("brief").long("brief").short('b').action(SetTrue))
        .after_help("Run `spacetime help describe` for more detailed information.\n")
}

pub async fn exec(config: Config, args: &ArgMatches) -> Result<(), anyhow::Error> {
    let database = args.get_one::<String>("database").unwrap();
    let expand = !args.get_flag("brief");
    let entity_name = args.get_one::<String>("entity_name");
    let entity_type = args.get_one::<String>("entity_type");

    let address = if let Ok(address) = spacetime_dns(&config, database).await {
        address
    } else {
        database.to_string()
    };

    let res = match entity_name {
        None => {
            let client = reqwest::Client::new();
            client
                .get(format!("http://{}/database/schema/{}", config.host, address))
                .query(&[("expand", expand)])
                .send()
                .await?
        }
        Some(entity_name) => {
            let entity_type = format!("{}s", entity_type.unwrap());

            let client = reqwest::Client::new();
            client
                .get(format!(
                    "http://{}/database/schema/{}/{}/{}",
                    config.host, address, entity_type, entity_name
                ))
                .query(&[("expand", expand)])
                .send()
                .await?
        }
    };

    let res = res.error_for_status()?;
    let body = res.bytes().await?;
    let str = String::from_utf8(body.to_vec())?;
    println!("{}", str);
    Ok(())
}