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

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

pub fn cli() -> clap::Command {
    clap::Command::new("call")
        .about("Invokes a reducer function in a database")
        .arg(Arg::new("database").required(true))
        .arg(Arg::new("function_name").required(true))
        .arg(
            Arg::new("arguments")
                .required(false)
                .help("arguments as a JSON array")
                .default_value("[]"),
        )
        .after_help("Run `spacetime help call` 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 function_name = args.get_one::<String>("function_name").unwrap();
    let arg_json = args.get_one::<String>("arguments").unwrap();

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

    let client = reqwest::Client::new();
    let res = client
        .post(format!(
            "http://{}/database/call/{}/{}",
            config.host, address, function_name
        ))
        .body(arg_json.to_owned())
        .send()
        .await?;

    res.error_for_status()?;

    Ok(())
}