feat: Inject <head> if configured response is invalid html

FabianLars · FabianLars · commit b06eb34b4025 · 2023-01-18T14:04:39.000+01:00
diff --git a/src/lib.rs b/src/lib.rs
@@ -112,7 +112,15 @@ fn handle_connection(
     );
     let response = match response {
         Some(s) if s.contains("<head>") => s.replace("<head>", &format!("<head>{}", script)),
-        Some(s) => s.replace("<body>", &format!("<head>{}</head><body>", script)),
+        Some(s) if s.contains("<body>") => {
+            s.replace("<body>", &format!("<head>{}</head><body>", script))
+        }
+        Some(s) => {
+            log::warn!(
+                "`response` does not contain a body or head element. Prepending a head element..."
+            );
+            format!("<head>{}</head>{}", script, s)
+        }
         None => format!(
             "<html><head>{}</head><body>Please return to the app.</body></html>",
             script
