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

// Copyright Materialize, Inc. and contributors. All rights reserved.
//
// Use of this software is governed by the Business Source License
// included in the LICENSE file.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0.

use anyhow::bail;
use mz_ore::collections::HashSet;
use reqwest::header::CONTENT_TYPE;
use reqwest::Method;

use crate::action::{ControlFlow, State};
use crate::parser::BuiltinCommand;

pub async fn run_request(
    mut cmd: BuiltinCommand,
    _: &mut State,
) -> Result<ControlFlow, anyhow::Error> {
    let url = cmd.args.string("url")?;
    let method: Method = cmd.args.parse("method")?;
    let content_type = cmd.args.opt_string("content-type");
    let further_accepted_status_codes = match cmd.args.opt_string("accept-additional-status-codes")
    {
        Some(s) => s
            .split(',')
            .map(|s| s.parse::<u16>().unwrap())
            .collect::<HashSet<u16>>(),
        None => HashSet::new(),
    };
    let body = cmd.input.join("\n");

    println!("$ http-request {} {}\n{}", method, url, body);

    let client = reqwest::Client::new();

    let mut request = client.request(method, &url).body(body);

    if let Some(value) = &content_type {
        request = request.header(CONTENT_TYPE, value);
    }

    let response = request.send().await?;
    let status = response.status();

    println!("{}\n{}", status, response.text().await?);

    if status.is_success() || further_accepted_status_codes.contains(&status.as_u16()) {
        Ok(ControlFlow::Continue)
    } else {
        bail!("http request returned failing status: {}", status)
    }
}