cba to fix the reqwest get title issue in docker production

src/main.rs

@@ -2,11 +2,11 @@
 mod writer;
 
 use axum::{
-    extract::State,
+    extract::{Query, State},
     http::{HeaderMap, Request, StatusCode},
     middleware,
     middleware::Next,
-    response::{Html, Redirect, Response, Result},
+    response::{Html, IntoResponse, Redirect, Response, Result},
     routing::{get, post},
     Form, Router,
 };
@@ -71,7 +71,7 @@ async fn get_login(State(_state): State<Arc<Mutex<LinkFile>>>) -> Result<Html<St
     Ok(Html(fs::read_to_string("templates/login.html").unwrap()))
 }
 
-async fn get_rss(State(state): State<Arc<Mutex<LinkFile>>>) -> Result<String> {
+async fn get_rss(State(state): State<Arc<Mutex<LinkFile>>>) -> Result<Response> {
     let template = fs::read_to_string("templates/rss.xml").unwrap();
     let linkfile = state.lock().await;
     let xmlfrag = fs::read_to_string(linkfile.get_path()).unwrap();
@@ -85,12 +85,19 @@ async fn get_rss(State(state): State<Arc<Mutex<LinkFile>>>) -> Result<String> {
     env.add_template("rss_xml", &template).unwrap();
     let template = env.get_template("rss_xml").unwrap();
 
-    Ok(template
+    let mut resp = template
         .render(context! {
         xmlfrag => xmlfrag,
         link => link,
         })
-        .unwrap())
+        .unwrap()
+        .into_response();
+
+    resp.headers_mut()
+        .insert("Content-Type", "text/xml".parse().unwrap())
+        .unwrap();
+
+    Ok(resp)
 }
 
 #[derive(Deserialize)]
@@ -149,8 +156,14 @@ async fn post_login(
     Ok((jar, Redirect::to("/")))
 }
 
+#[derive(Deserialize)]
+struct Token {
+    rss_token: Option<String>,
+}
+
 async fn authenticate<B>(
     headers: HeaderMap,
+    Query(token): Query<Token>,
     request: Request<B>,
     next: Next<B>,
 ) -> Result<Response, Redirect> {
@@ -161,5 +174,16 @@ async fn authenticate<B>(
         }
     }
 
+    let rss_token = match env::var("LINKY_TOKEN") {
+        Ok(v) => v,
+        Err(_) => "linkytoken".to_string(),
+    };
+
+    if let Some(t) = token.rss_token {
+        if t == rss_token {
+            return Ok(next.run(request).await);
+        }
+    }
+
     Err(Redirect::to("/login"))
 }

src/writer.rs

@@ -1,5 +1,6 @@
 use anyhow::{Context, Result};
 use minijinja::{context, Environment};
+use quick_xml::escape::escape;
 use reqwest::Client;
 use std::fs::{File, OpenOptions};
 use std::io::Write;
@@ -53,7 +54,7 @@ impl LinkFile {
     pub async fn add_link(&mut self, link: &str) -> Result<()> {
         let title = match get_link_title(&self.client, link).await {
             Ok(s) => s,
-            Err(_) => link.to_string(),
+            Err(_) => escape(link).to_string(),
         };
 
         let mut env = Environment::new();
@@ -64,7 +65,7 @@ impl LinkFile {
             template
                 .render(context! {
                 title => title,
-                link => link,
+                link => escape(link).to_string(),
                 })?
                 .as_bytes(),
         )?;