Adding authority to Requests for both client and server. (#30)

* Adding authority to Requests for both client and server.

* Removed unnecessary client test for authority

---------

Co-authored-by: Andreas Söderberg <andreas.soderber@nor2.io>

test-programs/src/bin/server_authority.rs

@@ -0,0 +1,16 @@
+use waki::{handler, ErrorCode, Request, Response};
+
+#[handler]
+fn hello(req: Request) -> Result<Response, ErrorCode> {
+    let authority = req.authority();
+
+    match authority {
+        Some(authority) => Response::builder()
+            .body(format!("Hello, {}!", authority.as_str()))
+            .build(),
+        None => Response::builder().body("Hello!").build(),
+    }
+}
+
+// required since this file is built as a `bin`
+fn main() {}

waki/src/request.rs

@@ -10,7 +10,7 @@ use crate::{
 
 use anyhow::{anyhow, Error, Result};
 use http::{
-    uri::{Parts, PathAndQuery},
+    uri::{Authority, Parts, PathAndQuery},
     Uri,
 };
 use std::borrow::Borrow;
@@ -201,6 +201,11 @@ impl Request {
         }
     }
 
+    /// Get the authority of the request.
+    pub fn authority(&self) -> &Option<Authority> {
+        &self.uri.authority
+    }
+
     fn send(self) -> Result<Response> {
         let req = OutgoingRequest::new(self.headers.try_into()?);
         req.set_method(&self.method)

waki/tests/all/server.rs

@@ -88,6 +88,21 @@ async fn status_code() -> Result<()> {
     Ok(())
 }
 
+#[tokio::test(flavor = "multi_thread")]
+async fn authority() -> Result<()> {
+    let req = hyper::Request::builder()
+        .uri("http://127.0.0.1:3000")
+        .body(body::empty())?;
+
+    let resp: http::Response<http_body_util::Collected<bytes::Bytes>> =
+        run_wasi_http(test_programs_artifacts::SERVER_AUTHORITY_COMPONENT, req).await??;
+    let body = resp.into_body().to_bytes();
+    let body = std::str::from_utf8(&body)?;
+    assert_eq!(body, "Hello, 127.0.0.1:3000!");
+
+    Ok(())
+}
+
 mod body {
     use http_body_util::{combinators::BoxBody, BodyExt, Empty, Full};
     use hyper::{body::Bytes, Error};