feature: make possible to set port number in Uri::builder

Author: Dushistov

src/uri/builder.rs

@@ -1,7 +1,10 @@
-use std::convert::{TryFrom, TryInto};
+use std::{
+    convert::{TryFrom, TryInto},
+    fmt::Write,
+};
 
-use super::{Authority, Parts, PathAndQuery, Scheme};
-use crate::Uri;
+use super::{port::Port, Authority, ErrorKind, InvalidUriParts, Parts, PathAndQuery, Scheme};
+use crate::{byte_str::ByteStr, Uri};
 
 /// A builder for `Uri`s.
 ///
@@ -78,6 +81,39 @@ impl Builder {
         })
     }
 
+    /// Set the port number for URI, will be part of `Authority`
+    pub fn port<T, P>(self, port: P) -> Self
+    where
+        P: Into<Port<T>>,
+    {
+        let port: Port<T> = port.into();
+        self.port_u16(port.as_u16())
+    }
+
+    /// Set the port number for URI, will be part of `Authority`
+    pub fn port_u16(self, port: u16) -> Self {
+        self.map(move |mut parts| {
+            let prev_auth = match parts.authority.as_ref() {
+                Some(auth) => {
+                    if auth.port().is_some() {
+                        return Err(InvalidUriParts::from(ErrorKind::InvalidPort).into());
+                    }
+
+                    auth.as_str()
+                }
+                None => "",
+            };
+            // 1 for ':', 5 for port number digists
+            let mut auth = String::with_capacity(prev_auth.len() + 6);
+            auth.push_str(prev_auth);
+            auth.push(':');
+            write!(&mut auth, "{}", port).expect("write to String failed");
+            let data: ByteStr = auth.into();
+            parts.authority = Some(Authority { data });
+            Ok(parts)
+        })
+    }
+
     /// Set the `PathAndQuery` for this URI.
     ///
     /// # Examples

src/uri/tests.rs

@@ -517,3 +517,34 @@ fn test_partial_eq_path_with_terminating_questionmark() {
 
     assert_eq!(uri, a);
 }
+
+#[test]
+fn test_uri_builder() {
+    assert_eq!(
+        "ws://localhost:8000/demo/",
+        Uri::builder()
+            .scheme("ws")
+            .authority("localhost")
+            .port_u16(8000)
+            .path_and_query("/demo/")
+            .build()
+            .unwrap()
+    );
+    assert_eq!(
+        "ws://[2001:db8:1f70::999:de8:7648:6e8]:8000/demo/",
+        Uri::builder()
+            .scheme("ws")
+            .authority("[2001:db8:1f70::999:de8:7648:6e8]")
+            .port_u16(8000)
+            .path_and_query("/demo/")
+            .build()
+            .unwrap()
+    );
+    Uri::builder()
+        .scheme("ws")
+        .authority("localhost:8000")
+        .port_u16(8000)
+        .path_and_query("/demo/")
+        .build()
+        .unwrap_err();
+}