1 files changed, 22 insertions, 10 deletions
diff --git a/src/main.rs b/src/main.rs
index b55999b..1855ab5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,7 +7,7 @@ use toml;
 #[macro_use]
 extern crate log;
 
-use es_public_proxy::{filter_request, ProxyConfig};
+use es_public_proxy::{filter_request, ProxyConfig, ProxyError};
 
 const CARGO_VERSION: Option<&'static str> = option_env!("CARGO_PKG_VERSION");
 
@@ -16,22 +16,35 @@ async fn upstream_req(
     config: ProxyConfig,
 ) -> Result<Response<Body>, hyper::Error> {
     info!("request: {} {}", req.method(), req.uri());
-    let parsed = filter_request(req, &config).await;
-    let mut resp = match parsed {
-        Ok(upstream_req) => {
-            debug!("sending request...");
-            Client::new().request(upstream_req).await?
-        }
+    let upstream_resp: Result<Response<Body>, ProxyError> = match filter_request(req, &config).await {
+        Ok(parsed_req) => {
+            debug!("forwarding request upstream");
+            Client::new().request(parsed_req)
+                .await
+                .map_err(|e| {
+                    warn!("upstream error: {}", e);
+                    ProxyError::UpstreamError(e.to_string())
+                })
+        },
+        Err(e) => {
+            warn!("blocked: {:?}", e);
+            Err(e)
+        },
+    };
+    let mut resp = match upstream_resp {
+        Ok(resp) => resp,
         Err(other) => Response::builder()
             .status(other.http_status_code())
             .header("Content-Type", "application/json; charset=UTF-8")
             .body(
                 serde_json::to_string(&other.to_json_value())
-                    .unwrap()
+                    .expect("trivial JSON value serialization failed")
                     .into(),
             )
-            .unwrap(),
+            .expect("trivial HTTP response construction failed"),
     };
+
+    // regardless of the response type (error or upstream response), add HTTP headers
     resp.headers_mut()
         .insert("Via", HeaderValue::from_static("1.1 es-public-proxy"));
     if config.enable_cors == Some(true) {
@@ -46,7 +59,6 @@ async fn upstream_req(
             HeaderValue::from_static("DNT,User-Agent,Content-Type"),
         );
     }
-    debug!("resp!");
     Ok(resp)
 }