http delete now supports pipeline data

This commit is contained in:
Jack Wright 2024-06-28 15:28:30 -07:00
parent a5a804e336
commit 8691a34fc7

View File

@ -1,7 +1,7 @@
use crate::network::http::client::{ use crate::network::http::client::{
check_response_redirection, http_client, http_parse_redirect_mode, http_parse_url, check_response_redirection, http_client, http_parse_redirect_mode, http_parse_url,
request_add_authorization_header, request_add_custom_headers, request_handle_response, request_add_authorization_header, request_add_custom_headers, request_handle_response,
request_set_timeout, send_request, RequestFlags, request_set_timeout, send_request2, HttpBody, RequestFlags,
}; };
use nu_engine::command_prelude::*; use nu_engine::command_prelude::*;
@ -15,7 +15,7 @@ impl Command for SubCommand {
fn signature(&self) -> Signature { fn signature(&self) -> Signature {
Signature::build("http delete") Signature::build("http delete")
.input_output_types(vec![(Type::Nothing, Type::Any)]) .input_output_types(vec![(Type::Any, Type::Any)])
.allow_variants_without_examples(true) .allow_variants_without_examples(true)
.required( .required(
"URL", "URL",
@ -132,6 +132,11 @@ impl Command for SubCommand {
"http delete --content-type application/json --data { field: value } https://www.example.com", "http delete --content-type application/json --data { field: value } https://www.example.com",
result: None, result: None,
}, },
Example {
description: "Perform an HTTP delete with JSON content from a pipeline to example.com",
example: "open foo.json | http delete https://www.example.com",
result: None,
},
] ]
} }
} }
@ -139,7 +144,7 @@ impl Command for SubCommand {
struct Arguments { struct Arguments {
url: Value, url: Value,
headers: Option<Value>, headers: Option<Value>,
data: Option<Value>, data: HttpBody,
content_type: Option<String>, content_type: Option<String>,
raw: bool, raw: bool,
insecure: bool, insecure: bool,
@ -155,13 +160,37 @@ fn run_delete(
engine_state: &EngineState, engine_state: &EngineState,
stack: &mut Stack, stack: &mut Stack,
call: &Call, call: &Call,
_input: PipelineData, input: PipelineData,
) -> Result<PipelineData, ShellError> { ) -> Result<PipelineData, ShellError> {
let (data, maybe_metadata) = call
.opt::<Value>(engine_state, stack, 1)?
.map(|v| (HttpBody::Value(v), None))
.unwrap_or_else(|| match input {
PipelineData::Value(v, metadata) => (HttpBody::Value(v), metadata),
PipelineData::ByteStream(byte_stream, metadata) => {
(HttpBody::ByteStream(byte_stream), metadata)
}
_ => (HttpBody::None, None),
});
let content_type = call
.get_flag(engine_state, stack, "content-type")?
.or_else(|| maybe_metadata.and_then(|m| m.content_type));
if let HttpBody::None = data {
return Err(ShellError::GenericError {
error: "Data must be provided either through pipeline or positional argument".into(),
msg: "".into(),
span: Some(call.head),
help: None,
inner: vec![],
});
}
let args = Arguments { let args = Arguments {
url: call.req(engine_state, stack, 0)?, url: call.req(engine_state, stack, 0)?,
headers: call.get_flag(engine_state, stack, "headers")?, headers: call.get_flag(engine_state, stack, "headers")?,
data: call.get_flag(engine_state, stack, "data")?, data,
content_type: call.get_flag(engine_state, stack, "content-type")?, content_type,
raw: call.has_flag(engine_state, stack, "raw")?, raw: call.has_flag(engine_state, stack, "raw")?,
insecure: call.has_flag(engine_state, stack, "insecure")?, insecure: call.has_flag(engine_state, stack, "insecure")?,
user: call.get_flag(engine_state, stack, "user")?, user: call.get_flag(engine_state, stack, "user")?,
@ -195,7 +224,7 @@ fn helper(
request = request_add_authorization_header(args.user, args.password, request); request = request_add_authorization_header(args.user, args.password, request);
request = request_add_custom_headers(args.headers, request)?; request = request_add_custom_headers(args.headers, request)?;
let response = send_request(request.clone(), args.data, args.content_type, ctrl_c); let response = send_request2(request.clone(), args.data, args.content_type, ctrl_c);
let request_flags = RequestFlags { let request_flags = RequestFlags {
raw: args.raw, raw: args.raw,