merino_cli/
main.rs

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
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use anyhow::Result;
use clap::{Parser, Subcommand};
use merino::curated_recommendations::{
    CuratedRecommendationsClient, CuratedRecommendationsRequest,
};

#[derive(Debug, Parser)]
struct Cli {
    /// Optional base host (defaults to prod)
    #[arg(long)]
    base_host: Option<String>,

    /// Required user agent header
    #[arg(long)]
    user_agent: String,

    #[command(subcommand)]
    command: Commands,
}

#[derive(Debug, Subcommand)]
enum Commands {
    Query {
        /// JSON string of type CuratedRecommendationsRequest
        #[clap(long)]
        json: Option<String>,

        /// Path to a JSON file containing the request
        #[clap(long, value_name = "FILE")]
        json_file: Option<std::path::PathBuf>,
    },
}

fn main() -> Result<()> {
    let cli = Cli::parse();

    viaduct_reqwest::use_reqwest_backend();
    let client = CuratedRecommendationsClient::new(cli.base_host.clone(), cli.user_agent.clone())?;

    match cli.command {
        Commands::Query { json, json_file } => {
            let json_data = match (json_file, json) {
                (Some(path), _) => std::fs::read_to_string(path)?,
                (None, Some(raw)) => raw,
                (None, None) => anyhow::bail!("You must provide either --json or --json-file"),
            };

            query_from_json(json_data, &client)?;
        }
    };

    Ok(())
}

fn query_from_json(json: String, client: &CuratedRecommendationsClient) -> Result<()> {
    let request: CuratedRecommendationsRequest = serde_json::from_str(&json)?;
    let response = client.get_curated_recommendations(&request)?;

    println!("{}", serde_json::to_string_pretty(&response)?);
    Ok(())
}