1use std::convert::TryFrom;
2use std::fmt;
3use std::future::Future;
4use std::time::Duration;
5
6use serde::Serialize;
7#[cfg(feature = "json")]
8use serde_json;
9
10use super::body::Body;
11use super::client::{Client, Pending};
12#[cfg(feature = "multipart")]
13use super::multipart;
14use super::response::Response;
15#[cfg(feature = "multipart")]
16use crate::header::CONTENT_LENGTH;
17use crate::header::{HeaderMap, HeaderName, HeaderValue, CONTENT_TYPE};
18use crate::{Method, Url};
19use http::{request::Parts, Request as HttpRequest, Version};
20
21pub struct Request {
23 method: Method,
24 url: Url,
25 headers: HeaderMap,
26 body: Option<Body>,
27 timeout: Option<Duration>,
28 version: Version,
29}
30
31#[must_use = "RequestBuilder does nothing until you 'send' it"]
35pub struct RequestBuilder {
36 client: Client,
37 request: crate::Result<Request>,
38}
39
40impl Request {
41 #[inline]
43 pub fn new(method: Method, url: Url) -> Self {
44 Request {
45 method,
46 url,
47 headers: HeaderMap::new(),
48 body: None,
49 timeout: None,
50 version: Version::default(),
51 }
52 }
53
54 #[inline]
56 pub fn method(&self) -> &Method {
57 &self.method
58 }
59
60 #[inline]
62 pub fn method_mut(&mut self) -> &mut Method {
63 &mut self.method
64 }
65
66 #[inline]
68 pub fn url(&self) -> &Url {
69 &self.url
70 }
71
72 #[inline]
74 pub fn url_mut(&mut self) -> &mut Url {
75 &mut self.url
76 }
77
78 #[inline]
80 pub fn headers(&self) -> &HeaderMap {
81 &self.headers
82 }
83
84 #[inline]
86 pub fn headers_mut(&mut self) -> &mut HeaderMap {
87 &mut self.headers
88 }
89
90 #[inline]
92 pub fn body(&self) -> Option<&Body> {
93 self.body.as_ref()
94 }
95
96 #[inline]
98 pub fn body_mut(&mut self) -> &mut Option<Body> {
99 &mut self.body
100 }
101
102 #[inline]
104 pub fn timeout(&self) -> Option<&Duration> {
105 self.timeout.as_ref()
106 }
107
108 #[inline]
110 pub fn timeout_mut(&mut self) -> &mut Option<Duration> {
111 &mut self.timeout
112 }
113
114 #[inline]
116 pub fn version(&self) -> Version {
117 self.version
118 }
119
120 #[inline]
122 pub fn version_mut(&mut self) -> &mut Version {
123 &mut self.version
124 }
125
126 pub fn try_clone(&self) -> Option<Request> {
130 let body = match self.body.as_ref() {
131 Some(body) => Some(body.try_clone()?),
132 None => None,
133 };
134 let mut req = Request::new(self.method().clone(), self.url().clone());
135 *req.timeout_mut() = self.timeout().copied();
136 *req.headers_mut() = self.headers().clone();
137 *req.version_mut() = self.version();
138 req.body = body;
139 Some(req)
140 }
141
142 pub(super) fn pieces(
143 self,
144 ) -> (
145 Method,
146 Url,
147 HeaderMap,
148 Option<Body>,
149 Option<Duration>,
150 Version,
151 ) {
152 (
153 self.method,
154 self.url,
155 self.headers,
156 self.body,
157 self.timeout,
158 self.version,
159 )
160 }
161}
162
163impl RequestBuilder {
164 pub(super) fn new(client: Client, request: crate::Result<Request>) -> RequestBuilder {
165 let mut builder = RequestBuilder { client, request };
166
167 let auth = builder
168 .request
169 .as_mut()
170 .ok()
171 .and_then(|req| extract_authority(&mut req.url));
172
173 if let Some((username, password)) = auth {
174 builder.basic_auth(username, password)
175 } else {
176 builder
177 }
178 }
179
180 pub fn from_parts(client: Client, request: Request) -> RequestBuilder {
182 RequestBuilder {
183 client,
184 request: crate::Result::Ok(request),
185 }
186 }
187
188 pub fn header<K, V>(self, key: K, value: V) -> RequestBuilder
190 where
191 HeaderName: TryFrom<K>,
192 <HeaderName as TryFrom<K>>::Error: Into<http::Error>,
193 HeaderValue: TryFrom<V>,
194 <HeaderValue as TryFrom<V>>::Error: Into<http::Error>,
195 {
196 self.header_sensitive(key, value, false)
197 }
198
199 fn header_sensitive<K, V>(mut self, key: K, value: V, sensitive: bool) -> RequestBuilder
201 where
202 HeaderName: TryFrom<K>,
203 <HeaderName as TryFrom<K>>::Error: Into<http::Error>,
204 HeaderValue: TryFrom<V>,
205 <HeaderValue as TryFrom<V>>::Error: Into<http::Error>,
206 {
207 let mut error = None;
208 if let Ok(ref mut req) = self.request {
209 match <HeaderName as TryFrom<K>>::try_from(key) {
210 Ok(key) => match <HeaderValue as TryFrom<V>>::try_from(value) {
211 Ok(mut value) => {
212 if sensitive {
216 value.set_sensitive(true);
217 }
218 req.headers_mut().append(key, value);
219 }
220 Err(e) => error = Some(crate::error::builder(e.into())),
221 },
222 Err(e) => error = Some(crate::error::builder(e.into())),
223 };
224 }
225 if let Some(err) = error {
226 self.request = Err(err);
227 }
228 self
229 }
230
231 pub fn headers(mut self, headers: crate::header::HeaderMap) -> RequestBuilder {
235 if let Ok(ref mut req) = self.request {
236 crate::util::replace_headers(req.headers_mut(), headers);
237 }
238 self
239 }
240
241 pub fn basic_auth<U, P>(self, username: U, password: Option<P>) -> RequestBuilder
256 where
257 U: fmt::Display,
258 P: fmt::Display,
259 {
260 let header_value = crate::util::basic_auth(username, password);
261 self.header_sensitive(crate::header::AUTHORIZATION, header_value, true)
262 }
263
264 pub fn bearer_auth<T>(self, token: T) -> RequestBuilder
266 where
267 T: fmt::Display,
268 {
269 let header_value = format!("Bearer {token}");
270 self.header_sensitive(crate::header::AUTHORIZATION, header_value, true)
271 }
272
273 pub fn body<T: Into<Body>>(mut self, body: T) -> RequestBuilder {
275 if let Ok(ref mut req) = self.request {
276 *req.body_mut() = Some(body.into());
277 }
278 self
279 }
280
281 pub fn timeout(mut self, timeout: Duration) -> RequestBuilder {
287 if let Ok(ref mut req) = self.request {
288 *req.timeout_mut() = Some(timeout);
289 }
290 self
291 }
292
293 #[cfg(feature = "multipart")]
316 #[cfg_attr(docsrs, doc(cfg(feature = "multipart")))]
317 pub fn multipart(self, mut multipart: multipart::Form) -> RequestBuilder {
318 let mut builder = self.header(
319 CONTENT_TYPE,
320 format!("multipart/form-data; boundary={}", multipart.boundary()).as_str(),
321 );
322
323 builder = match multipart.compute_length() {
324 Some(length) => builder.header(CONTENT_LENGTH, length),
325 None => builder,
326 };
327
328 if let Ok(ref mut req) = builder.request {
329 *req.body_mut() = Some(multipart.stream())
330 }
331 builder
332 }
333
334 pub fn query<T: Serialize + ?Sized>(mut self, query: &T) -> RequestBuilder {
353 let mut error = None;
354 if let Ok(ref mut req) = self.request {
355 let url = req.url_mut();
356 let mut pairs = url.query_pairs_mut();
357 let serializer = serde_urlencoded::Serializer::new(&mut pairs);
358
359 if let Err(err) = query.serialize(serializer) {
360 error = Some(crate::error::builder(err));
361 }
362 }
363 if let Ok(ref mut req) = self.request {
364 if let Some("") = req.url().query() {
365 req.url_mut().set_query(None);
366 }
367 }
368 if let Some(err) = error {
369 self.request = Err(err);
370 }
371 self
372 }
373
374 pub fn version(mut self, version: Version) -> RequestBuilder {
376 if let Ok(ref mut req) = self.request {
377 req.version = version;
378 }
379 self
380 }
381
382 pub fn form<T: Serialize + ?Sized>(mut self, form: &T) -> RequestBuilder {
410 let mut error = None;
411 if let Ok(ref mut req) = self.request {
412 match serde_urlencoded::to_string(form) {
413 Ok(body) => {
414 req.headers_mut()
415 .entry(CONTENT_TYPE)
416 .or_insert(HeaderValue::from_static(
417 "application/x-www-form-urlencoded",
418 ));
419 *req.body_mut() = Some(body.into());
420 }
421 Err(err) => error = Some(crate::error::builder(err)),
422 }
423 }
424 if let Some(err) = error {
425 self.request = Err(err);
426 }
427 self
428 }
429
430 #[cfg(feature = "json")]
441 #[cfg_attr(docsrs, doc(cfg(feature = "json")))]
442 pub fn json<T: Serialize + ?Sized>(mut self, json: &T) -> RequestBuilder {
443 let mut error = None;
444 if let Ok(ref mut req) = self.request {
445 match serde_json::to_vec(json) {
446 Ok(body) => {
447 if !req.headers().contains_key(CONTENT_TYPE) {
448 req.headers_mut()
449 .insert(CONTENT_TYPE, HeaderValue::from_static("application/json"));
450 }
451 *req.body_mut() = Some(body.into());
452 }
453 Err(err) => error = Some(crate::error::builder(err)),
454 }
455 }
456 if let Some(err) = error {
457 self.request = Err(err);
458 }
459 self
460 }
461
462 #[doc(hidden)]
470 #[cfg_attr(target_arch = "wasm32", deprecated)]
471 pub fn fetch_mode_no_cors(self) -> RequestBuilder {
472 self
473 }
474
475 pub fn build(self) -> crate::Result<Request> {
478 self.request
479 }
480
481 pub fn build_split(self) -> (Client, crate::Result<Request>) {
487 (self.client, self.request)
488 }
489
490 pub fn send(self) -> impl Future<Output = Result<Response, crate::Error>> {
512 match self.request {
513 Ok(req) => self.client.execute_request(req),
514 Err(err) => Pending::new_err(err),
515 }
516 }
517
518 pub fn try_clone(&self) -> Option<RequestBuilder> {
538 self.request
539 .as_ref()
540 .ok()
541 .and_then(|req| req.try_clone())
542 .map(|req| RequestBuilder {
543 client: self.client.clone(),
544 request: Ok(req),
545 })
546 }
547}
548
549impl fmt::Debug for Request {
550 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
551 fmt_request_fields(&mut f.debug_struct("Request"), self).finish()
552 }
553}
554
555impl fmt::Debug for RequestBuilder {
556 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
557 let mut builder = f.debug_struct("RequestBuilder");
558 match self.request {
559 Ok(ref req) => fmt_request_fields(&mut builder, req).finish(),
560 Err(ref err) => builder.field("error", err).finish(),
561 }
562 }
563}
564
565fn fmt_request_fields<'a, 'b>(
566 f: &'a mut fmt::DebugStruct<'a, 'b>,
567 req: &Request,
568) -> &'a mut fmt::DebugStruct<'a, 'b> {
569 f.field("method", &req.method)
570 .field("url", &req.url)
571 .field("headers", &req.headers)
572}
573
574pub(crate) fn extract_authority(url: &mut Url) -> Option<(String, Option<String>)> {
577 use percent_encoding::percent_decode;
578
579 if url.has_authority() {
580 let username: String = percent_decode(url.username().as_bytes())
581 .decode_utf8()
582 .ok()?
583 .into();
584 let password = url.password().and_then(|pass| {
585 percent_decode(pass.as_bytes())
586 .decode_utf8()
587 .ok()
588 .map(String::from)
589 });
590 if !username.is_empty() || password.is_some() {
591 url.set_username("")
592 .expect("has_authority means set_username shouldn't fail");
593 url.set_password(None)
594 .expect("has_authority means set_password shouldn't fail");
595 return Some((username, password));
596 }
597 }
598
599 None
600}
601
602impl<T> TryFrom<HttpRequest<T>> for Request
603where
604 T: Into<Body>,
605{
606 type Error = crate::Error;
607
608 fn try_from(req: HttpRequest<T>) -> crate::Result<Self> {
609 let (parts, body) = req.into_parts();
610 let Parts {
611 method,
612 uri,
613 headers,
614 version,
615 ..
616 } = parts;
617 let url = Url::parse(&uri.to_string()).map_err(crate::error::builder)?;
618 Ok(Request {
619 method,
620 url,
621 headers,
622 body: Some(body.into()),
623 timeout: None,
624 version,
625 })
626 }
627}
628
629impl TryFrom<Request> for HttpRequest<Body> {
630 type Error = crate::Error;
631
632 fn try_from(req: Request) -> crate::Result<Self> {
633 let Request {
634 method,
635 url,
636 headers,
637 body,
638 version,
639 ..
640 } = req;
641
642 let mut req = HttpRequest::builder()
643 .version(version)
644 .method(method)
645 .uri(url.as_str())
646 .body(body.unwrap_or_else(Body::empty))
647 .map_err(crate::error::builder)?;
648
649 *req.headers_mut() = headers;
650 Ok(req)
651 }
652}
653
654#[cfg(test)]
655mod tests {
656 #![cfg(not(feature = "rustls-tls-manual-roots-no-provider"))]
657
658 use super::{Client, HttpRequest, Request, RequestBuilder, Version};
659 use crate::Method;
660 use serde::Serialize;
661 use std::collections::BTreeMap;
662 use std::convert::TryFrom;
663
664 #[test]
665 fn add_query_append() {
666 let client = Client::new();
667 let some_url = "https://google.com/";
668 let r = client.get(some_url);
669
670 let r = r.query(&[("foo", "bar")]);
671 let r = r.query(&[("qux", 3)]);
672
673 let req = r.build().expect("request is valid");
674 assert_eq!(req.url().query(), Some("foo=bar&qux=3"));
675 }
676
677 #[test]
678 fn add_query_append_same() {
679 let client = Client::new();
680 let some_url = "https://google.com/";
681 let r = client.get(some_url);
682
683 let r = r.query(&[("foo", "a"), ("foo", "b")]);
684
685 let req = r.build().expect("request is valid");
686 assert_eq!(req.url().query(), Some("foo=a&foo=b"));
687 }
688
689 #[test]
690 fn add_query_struct() {
691 #[derive(Serialize)]
692 struct Params {
693 foo: String,
694 qux: i32,
695 }
696
697 let client = Client::new();
698 let some_url = "https://google.com/";
699 let r = client.get(some_url);
700
701 let params = Params {
702 foo: "bar".into(),
703 qux: 3,
704 };
705
706 let r = r.query(¶ms);
707
708 let req = r.build().expect("request is valid");
709 assert_eq!(req.url().query(), Some("foo=bar&qux=3"));
710 }
711
712 #[test]
713 fn add_query_map() {
714 let mut params = BTreeMap::new();
715 params.insert("foo", "bar");
716 params.insert("qux", "three");
717
718 let client = Client::new();
719 let some_url = "https://google.com/";
720 let r = client.get(some_url);
721
722 let r = r.query(¶ms);
723
724 let req = r.build().expect("request is valid");
725 assert_eq!(req.url().query(), Some("foo=bar&qux=three"));
726 }
727
728 #[test]
729 fn test_replace_headers() {
730 use http::HeaderMap;
731
732 let mut headers = HeaderMap::new();
733 headers.insert("foo", "bar".parse().unwrap());
734 headers.append("foo", "baz".parse().unwrap());
735
736 let client = Client::new();
737 let req = client
738 .get("https://hyper.rs")
739 .header("im-a", "keeper")
740 .header("foo", "pop me")
741 .headers(headers)
742 .build()
743 .expect("request build");
744
745 assert_eq!(req.headers()["im-a"], "keeper");
746
747 let foo = req.headers().get_all("foo").iter().collect::<Vec<_>>();
748 assert_eq!(foo.len(), 2);
749 assert_eq!(foo[0], "bar");
750 assert_eq!(foo[1], "baz");
751 }
752
753 #[test]
754 fn normalize_empty_query() {
755 let client = Client::new();
756 let some_url = "https://google.com/";
757 let empty_query: &[(&str, &str)] = &[];
758
759 let req = client
760 .get(some_url)
761 .query(empty_query)
762 .build()
763 .expect("request build");
764
765 assert_eq!(req.url().query(), None);
766 assert_eq!(req.url().as_str(), "https://google.com/");
767 }
768
769 #[test]
770 fn try_clone_reusable() {
771 let client = Client::new();
772 let builder = client
773 .post("http://httpbin.org/post")
774 .header("foo", "bar")
775 .body("from a &str!");
776 let req = builder
777 .try_clone()
778 .expect("clone successful")
779 .build()
780 .expect("request is valid");
781 assert_eq!(req.url().as_str(), "http://httpbin.org/post");
782 assert_eq!(req.method(), Method::POST);
783 assert_eq!(req.headers()["foo"], "bar");
784 }
785
786 #[test]
787 fn try_clone_no_body() {
788 let client = Client::new();
789 let builder = client.get("http://httpbin.org/get");
790 let req = builder
791 .try_clone()
792 .expect("clone successful")
793 .build()
794 .expect("request is valid");
795 assert_eq!(req.url().as_str(), "http://httpbin.org/get");
796 assert_eq!(req.method(), Method::GET);
797 assert!(req.body().is_none());
798 }
799
800 #[test]
801 #[cfg(feature = "stream")]
802 fn try_clone_stream() {
803 let chunks: Vec<Result<_, ::std::io::Error>> = vec![Ok("hello"), Ok(" "), Ok("world")];
804 let stream = futures_util::stream::iter(chunks);
805 let client = Client::new();
806 let builder = client
807 .get("http://httpbin.org/get")
808 .body(super::Body::wrap_stream(stream));
809 let clone = builder.try_clone();
810 assert!(clone.is_none());
811 }
812
813 #[test]
814 fn convert_url_authority_into_basic_auth() {
815 let client = Client::new();
816 let some_url = "https://Aladdin:open sesame@localhost/";
817
818 let req = client.get(some_url).build().expect("request build");
819
820 assert_eq!(req.url().as_str(), "https://localhost/");
821 assert_eq!(
822 req.headers()["authorization"],
823 "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=="
824 );
825 }
826
827 #[test]
828 fn test_basic_auth_sensitive_header() {
829 let client = Client::new();
830 let some_url = "https://localhost/";
831
832 let req = client
833 .get(some_url)
834 .basic_auth("Aladdin", Some("open sesame"))
835 .build()
836 .expect("request build");
837
838 assert_eq!(req.url().as_str(), "https://localhost/");
839 assert_eq!(
840 req.headers()["authorization"],
841 "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=="
842 );
843 assert!(req.headers()["authorization"].is_sensitive());
844 }
845
846 #[test]
847 fn test_bearer_auth_sensitive_header() {
848 let client = Client::new();
849 let some_url = "https://localhost/";
850
851 let req = client
852 .get(some_url)
853 .bearer_auth("Hold my bear")
854 .build()
855 .expect("request build");
856
857 assert_eq!(req.url().as_str(), "https://localhost/");
858 assert_eq!(req.headers()["authorization"], "Bearer Hold my bear");
859 assert!(req.headers()["authorization"].is_sensitive());
860 }
861
862 #[test]
863 fn test_explicit_sensitive_header() {
864 let client = Client::new();
865 let some_url = "https://localhost/";
866
867 let mut header = http::HeaderValue::from_static("in plain sight");
868 header.set_sensitive(true);
869
870 let req = client
871 .get(some_url)
872 .header("hiding", header)
873 .build()
874 .expect("request build");
875
876 assert_eq!(req.url().as_str(), "https://localhost/");
877 assert_eq!(req.headers()["hiding"], "in plain sight");
878 assert!(req.headers()["hiding"].is_sensitive());
879 }
880
881 #[test]
882 fn convert_from_http_request() {
883 let http_request = HttpRequest::builder()
884 .method("GET")
885 .uri("http://localhost/")
886 .header("User-Agent", "my-awesome-agent/1.0")
887 .body("test test test")
888 .unwrap();
889 let req: Request = Request::try_from(http_request).unwrap();
890 assert!(req.body().is_some());
891 let test_data = b"test test test";
892 assert_eq!(req.body().unwrap().as_bytes(), Some(&test_data[..]));
893 let headers = req.headers();
894 assert_eq!(headers.get("User-Agent").unwrap(), "my-awesome-agent/1.0");
895 assert_eq!(req.method(), Method::GET);
896 assert_eq!(req.url().as_str(), "http://localhost/");
897 }
898
899 #[test]
900 fn set_http_request_version() {
901 let http_request = HttpRequest::builder()
902 .method("GET")
903 .uri("http://localhost/")
904 .header("User-Agent", "my-awesome-agent/1.0")
905 .version(Version::HTTP_11)
906 .body("test test test")
907 .unwrap();
908 let req: Request = Request::try_from(http_request).unwrap();
909 assert!(req.body().is_some());
910 let test_data = b"test test test";
911 assert_eq!(req.body().unwrap().as_bytes(), Some(&test_data[..]));
912 let headers = req.headers();
913 assert_eq!(headers.get("User-Agent").unwrap(), "my-awesome-agent/1.0");
914 assert_eq!(req.method(), Method::GET);
915 assert_eq!(req.url().as_str(), "http://localhost/");
916 assert_eq!(req.version(), Version::HTTP_11);
917 }
918
919 #[test]
920 fn builder_split_reassemble() {
921 let builder = {
922 let client = Client::new();
923 client.get("http://example.com")
924 };
925 let (client, inner) = builder.build_split();
926 let request = inner.unwrap();
927 let builder = RequestBuilder::from_parts(client, request);
928 builder.build().unwrap();
929 }
930
931 }