Add add/remove_path_prefix fields to domain settings

src/domain/manager.rs

@@ -252,7 +252,12 @@ impl Manager for ManagerImpl {
             return Err(unexpected::Error::from("origin is proxy, can't serve file").into());
         }
 
-        let f = self.origin_store.get_file(&settings.origin_descr, path)?;
+        let path = settings.process_path(path);
+
+        let f = self
+            .origin_store
+            .get_file(&settings.origin_descr, path.as_ref())?;
+
         Ok(f)
     }
 

src/domain/settings.rs

@@ -1,23 +1,12 @@
 use crate::error::unexpected::{self, Mappable};
 use crate::origin;
 
+use std::borrow;
+
 use hex::ToHex;
 use serde::{Deserialize, Serialize};
 use sha2::{Digest, Sha256};
 
-#[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)]
-#[serde(tag = "protocol")]
-pub enum SettingsServeProtocol {
-    #[serde(rename = "http")]
-    Http,
-    #[serde(rename = "https")]
-    Https,
-}
-
-fn default_serve_protocols() -> Vec<SettingsServeProtocol> {
-    vec![SettingsServeProtocol::Http, SettingsServeProtocol::Https]
-}
-
 #[derive(Debug, PartialEq, Eq, Clone, Serialize, Deserialize)]
 /// Defines how a domain will behave when it is accessed. These are configured by the owner of the
 /// domain during setup.
@@ -25,8 +14,8 @@ pub struct Settings {
     #[serde(flatten)]
     pub origin_descr: origin::Descr,
 
-    #[serde(default = "default_serve_protocols")]
+    pub remove_path_prefix: Option<String>,
-    pub serve_protocols: Vec<SettingsServeProtocol>,
+    pub add_path_prefix: Option<String>,
 }
 
 impl Settings {
@@ -35,4 +24,23 @@ impl Settings {
         serde_json::to_writer(&mut h, self).or_unexpected()?;
         Ok(h.finalize().encode_hex::<String>())
     }
+
+    pub fn process_path<'a>(&self, path: &'a str) -> borrow::Cow<'a, str> {
+        let mut path = borrow::Cow::Borrowed(path);
+
+        if let Some(ref remove_path_prefix) = self.remove_path_prefix {
+            if path.starts_with(remove_path_prefix.as_str()) {
+                *path.to_mut() = path.strip_prefix(remove_path_prefix).unwrap().to_string();
+            }
+        }
+
+        if let Some(ref add_path_prefix) = self.add_path_prefix {
+            let mut prefixed_path = String::with_capacity(add_path_prefix.len() + path.len());
+            prefixed_path.push_str(add_path_prefix);
+            prefixed_path.push_str(path.as_ref());
+            *path.to_mut() = prefixed_path;
+        }
+
+        path
+    }
 }

src/domain/store.rs

@@ -181,7 +181,8 @@ mod tests {
                 url: "bar".to_string(),
                 branch_name: "baz".to_string(),
             },
-            serve_protocols: vec![domain::SettingsServeProtocol::Http],
+            remove_path_prefix: None,
+            add_path_prefix: None,
         };
 
         assert!(matches!(
@@ -204,7 +205,8 @@ mod tests {
                 url: "BAR".to_string(),
                 branch_name: "BAZ".to_string(),
             },
-            serve_protocols: vec![],
+            remove_path_prefix: None,
+            add_path_prefix: None,
         };
 
         store.set(&domain, &new_settings).expect("set");

src/origin/proxy.rs

@@ -30,7 +30,7 @@ pub async fn serve_http_request(
     mut req: hyper::Request<hyper::Body>,
     req_is_https: bool,
 ) -> unexpected::Result<hyper::Response<hyper::Body>> {
-    let (proxy_url, request_http_headers) = if let origin::Descr::Proxy {
+    let (url, request_http_headers) = if let origin::Descr::Proxy {
         ref url,
         ref request_http_headers,
     } = settings.origin_descr
@@ -67,12 +67,12 @@ pub async fn serve_http_request(
             .insert("x-forwarded-proto", HeaderValue::from_static("https"));
     }
 
-    match hyper_reverse_proxy::call(client_ip, proxy_url, req).await {
+    match hyper_reverse_proxy::call(client_ip, url, req).await {
         Ok(res) => Ok(res),
         // ProxyError doesn't actually implement Error :facepalm: so we have to format the error
         // manually
         Err(e) => Err(unexpected::Error::from(
-            format!("error while proxying to {proxy_url}: {e:?}").as_str(),
+            format!("error while proxying to {url}: {e:?}").as_str(),
         )),
     }
 }

src/service/http.rs

@@ -177,18 +177,6 @@ impl<'svc> Service {
             }
         };
 
-        let allowed = settings.serve_protocols.iter().any(|p| match p {
-            domain::SettingsServeProtocol::Http => !req_is_https,
-            domain::SettingsServeProtocol::Https => req_is_https,
-        });
-
-        if !allowed {
-            return self.render_error_page(
-                421,
-                "The requested protocol is not supported by this domain",
-            );
-        }
-
         // if the domain is backed by a proxy then that is handled specially.
         if let origin::Descr::Proxy { .. } = settings.origin_descr {
             return origin::proxy::serve_http_request(&settings, client_ip, req, req_is_https)

src/service/util.rs

@@ -1,22 +1,9 @@
 use std::convert::TryFrom;
 
-use serde::{de, Deserialize, Serialize};
+use serde::{Deserialize, Serialize};
 
 use crate::{domain, error::unexpected, origin};
 
-fn deserialize_bool<'de, D>(deserializer: D) -> Result<bool, D::Error>
-where
-    D: de::Deserializer<'de>,
-{
-    let s: &str = de::Deserialize::deserialize(deserializer)?;
-
-    match s {
-        "true" => Ok(true),
-        "false" => Ok(false),
-        _ => Err(de::Error::unknown_variant(s, &["true", "false"])),
-    }
-}
-
 #[derive(Serialize, Deserialize, Default)]
 pub struct FlatDomainSettings {
     domain_setting_origin_descr_kind: String,
@@ -25,14 +12,6 @@ pub struct FlatDomainSettings {
     domain_setting_origin_descr_git_branch_name: Option<String>,
 
     domain_setting_origin_descr_proxy_url: Option<String>,
-
-    #[serde(default)]
-    #[serde(deserialize_with = "deserialize_bool")]
-    domain_setting_serve_protocol_http: bool,
-
-    #[serde(default)]
-    #[serde(deserialize_with = "deserialize_bool")]
-    domain_setting_serve_protocol_https: bool,
 }
 
 impl TryFrom<FlatDomainSettings> for domain::Settings {
@@ -52,19 +31,10 @@ impl TryFrom<FlatDomainSettings> for domain::Settings {
             _ => Err("invalid domain_setting_origin_descr_kind".to_string()),
         }?;
 
-        let mut serve_protocols = Vec::<domain::SettingsServeProtocol>::default();
-
-        if v.domain_setting_serve_protocol_http {
-            serve_protocols.push(domain::SettingsServeProtocol::Http);
-        }
-
-        if v.domain_setting_serve_protocol_https {
-            serve_protocols.push(domain::SettingsServeProtocol::Https);
-        }
-
         Ok(Self {
             origin_descr,
-            serve_protocols,
+            remove_path_prefix: None,
+            add_path_prefix: None,
         })
     }
 }
@@ -88,17 +58,6 @@ impl TryFrom<domain::Settings> for FlatDomainSettings {
             }
         }
 
-        for serve_protocol in v.serve_protocols {
-            match serve_protocol {
-                domain::SettingsServeProtocol::Http => {
-                    res.domain_setting_serve_protocol_http = true
-                }
-                domain::SettingsServeProtocol::Https => {
-                    res.domain_setting_serve_protocol_https = true
-                }
-            }
-        }
-
         Ok(res)
     }
 }