Make iframes work cross domain (#41)

![Screenshot 2023-10-23 at 16 30
58](https://github.com/mentimeter/linkup/assets/7828615/1019d55c-d80b-4a75-a517-62ddf3c32258)

Host really seemed to be screwing things up, I found it better to remove
the header instead

---------

Co-authored-by: Augusto César <[email protected]>

linkup-cli/src/local_server.rs

@@ -8,7 +8,6 @@ use futures::stream::StreamExt;
 use thiserror::Error;
 
 use linkup::{HeaderMap as LinkupHeaderMap, HeaderName as LinkupHeaderName, *};
-use url::Url;
 
 use crate::LINKUP_LOCALSERVER_PORT;
 
@@ -189,15 +188,12 @@ async fn linkup_request_handler(
         }
     };
 
-    let mut extra_headers = get_additional_headers(&url, &headers, &session_name, &target_service);
-    extra_headers.insert(
-        LinkupHeaderName::Host,
-        Url::parse(&target_service.url).unwrap(),
-    );
+    let extra_headers = get_additional_headers(&url, &headers, &session_name, &target_service);
 
     // Proxy the request using the destination_url and the merged headers
     let client = reqwest::Client::new();
     headers.extend(&extra_headers);
+    headers.remove(LinkupHeaderName::Host);
 
     let response_result = client
         .request(req.method().clone(), &target_service.url)

linkup-cli/src/services/caddy.rs

@@ -23,7 +23,7 @@ pub fn start(local_config: &YamlLocalConfig) -> Result<()> {
     let domains: Vec<String> = local_config
         .top_level_domains()
         .iter()
-        .map(|domain| format!("*.{}", domain))
+        .map(|domain| format!("{domain}, *.{domain}"))
         .collect();
 
     write_caddyfile(&domains)?;

linkup/src/headers.rs

@@ -2,6 +2,7 @@ use std::collections::HashMap;
 
 use unicase::UniCase;
 
+#[derive(Debug)]
 pub struct HeaderMap(HashMap<UniCase<String>, String>);
 
 pub enum HeaderName {
@@ -79,6 +80,10 @@ impl HeaderMap {
         self.0.extend(iter)
     }
 
+    pub fn remove(&mut self, key: impl Into<UniCase<String>>) -> Option<String> {
+        self.0.remove(&key.into())
+    }
+
     #[cfg(feature = "actix")]
     pub fn from_actix_request(req: &actix_web::HttpRequest) -> Self {
         req.headers().into()

linkup/src/lib.rs

@@ -107,7 +107,7 @@ pub fn additional_response_headers() -> HeaderMap {
     headers
 }
 
-#[cfg_attr(test, derive(Debug, PartialEq))]
+#[derive(Debug, PartialEq)]
 pub struct TargetService {
     pub name: String,
     pub url: String,
@@ -287,6 +287,10 @@ mod tests {
                     }
                 ]
             },
+            {
+                "name": "other-frontend",
+                "location": "http://localhost:5000"
+            },
             {
                 "name": "backend",
                 "rewrites": [
@@ -316,6 +320,10 @@ mod tests {
             {
                 "domain": "api.example.com",
                 "default_service": "backend"
+            },
+            {
+                "domain": "other-example.com",
+                "default_service": "other-frontend"
             }
         ]
     }
@@ -592,4 +600,32 @@ mod tests {
         assert_eq!(target.name, "backend");
         assert_eq!(target.url, "http://localhost:8001/user");
     }
+
+    #[tokio::test]
+    async fn test_iframable() {
+        let string_store = MemoryStringStore::new();
+        let sessions = SessionAllocator::new(&string_store);
+
+        let input_config_value: serde_json::Value = serde_json::from_str(CONF_STR).unwrap();
+        let input_config: Session = input_config_value.try_into().unwrap();
+
+        let name = sessions
+            .store_session(input_config, NameKind::Animal, "".to_string())
+            .await
+            .unwrap();
+
+        let mut headers = HeaderMap::new();
+        headers.insert(HeaderName::Referer, format!("{}.example.com", name));
+
+        let (name, config) = sessions
+            .get_request_session("other-example.com", &headers)
+            .await
+            .unwrap();
+
+        let target =
+            get_target_service("http://other-example.com", &headers, &config, &name).unwrap();
+
+        assert_eq!(target.name, "other-frontend");
+        assert_eq!(target.url, "http://localhost:5000/");
+    }
 }