From cdc2e8c14824a5d58154c5df42d1bc79495d69ea Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Fri, 14 Mar 2025 13:31:41 +0100
Subject: hosts: :fallback and :none special hosts

---
 src/modules/hosts.rs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'src/modules/hosts.rs')

diff --git a/src/modules/hosts.rs b/src/modules/hosts.rs
index 6bd213a..20c6c85 100644
--- a/src/modules/hosts.rs
+++ b/src/modules/hosts.rs
@@ -30,10 +30,16 @@ impl Node for Hosts {
                 .headers()
                 .get(HOST)
                 .and_then(|e| e.to_str().ok())
-                .map(remove_port)
-                .ok_or(ServiceError::NoHost)?;
+                .map(remove_port);
 
-            let node = self.0.get(host).ok_or(ServiceError::UnknownHost)?;
+            let node = match host {
+                Some(host) => self
+                    .0
+                    .get(host)
+                    .or_else(|| self.0.get(":fallback"))
+                    .ok_or(ServiceError::UnknownHost)?,
+                None => self.0.get(":none").ok_or(ServiceError::NoHost)?,
+            };
 
             node.handle(context, request).await
         })
-- 
cgit v1.2.3-70-g09d2