Support path prefix in reverse proxy scenarios.

Author: MIvanchev

crates/wastebin_server/src/handlers/delete/form.rs

@@ -17,7 +17,7 @@ pub async fn delete(
     async {
         let id = id.parse()?;
         db.delete_for(id, &uids).await?;
-        Ok(Redirect::to("/"))
+        Ok(crate::redirect("/"))
     }
     .await
     .map_err(|err| make_error(err, page.clone(), theme, lang))

crates/wastebin_server/src/handlers/extract.rs

@@ -186,7 +186,7 @@ where
             .and_then(|referer| referer.to_str().ok())
             .map(|referer| {
                 if referer.starts_with('/') && !referer.starts_with("//") {
-                    Redirect::to(referer)
+                    crate::redirect(referer)
                 } else {
                     referer
                         .parse::<url::Url>()
@@ -198,10 +198,10 @@ where
                                 |q| Redirect::to(&format!("{path}?{q}")),
                             )
                         })
-                        .unwrap_or_else(|| Redirect::to("/"))
+                        .unwrap_or_else(|| crate::redirect("/"))
                 }
             })
-            .unwrap_or_else(|| Redirect::to("/"));
+            .unwrap_or_else(|| crate::redirect("/"));
 
         Ok(SafeReferer(redirect))
     }

crates/wastebin_server/src/handlers/html/paste.rs

@@ -90,7 +90,7 @@ pub async fn get<E>(
         }
         let mut cookie = cookie("uid", serialize_uids(&new_uids));
         cookie.set_secure(true);
-        return Ok((jar.add(cookie), Redirect::to(&format!("/{id}"))).into_response());
+        return Ok((jar.add(cookie), crate::redirect(&format!("/{id}"))).into_response());
     }
 
     async {

crates/wastebin_server/src/handlers/insert/form.rs

@@ -87,7 +87,7 @@ pub async fn post<E: std::fmt::Debug>(
         let mut cookie = cookie("uid", serialize_uids(&uids));
         cookie.set_secure(true);
 
-        Ok((jar.add(cookie), Redirect::to(&url)))
+        Ok((jar.add(cookie), crate::redirect(&url)))
     }
     .await
     .map_err(|err| make_error(err, page, theme, lang))

crates/wastebin_server/src/main.rs

@@ -15,7 +15,7 @@ use std::time::Duration;
 use axum::extract::{DefaultBodyLimit, FromRef, Request, State};
 use axum::http::{HeaderName, HeaderValue, StatusCode};
 use axum::middleware::{Next, from_fn, from_fn_with_state};
-use axum::response::{IntoResponse, Response};
+use axum::response::{IntoResponse, Redirect, Response};
 use axum::routing::{Router, get, post};
 use axum_extra::extract::cookie::Key;
 use futures::future::TryFutureExt;
@@ -24,6 +24,7 @@ use http::header::{
     X_XSS_PROTECTION,
 };
 use tokio::net::{TcpListener, UnixListener};
+use tokio::task_local;
 use tower::ServiceBuilder;
 use tower_http::compression::CompressionLayer;
 use tower_http::timeout::TimeoutLayer;
@@ -36,6 +37,10 @@ use crate::handlers::{delete, download, html, insert, raw, robots, theme};
 use crate::i18n::Lang;
 use wastebin_core::db::Database;
 
+task_local! {
+    static PATH_PREFIX: String;
+}
+
 /// Reference counted [`page::Page`] wrapper.
 pub(crate) type Page = Arc<page::Page>;
 
@@ -81,6 +86,26 @@ impl FromRef<AppState> for Cache {
     }
 }
 
+fn path_prefix() -> String {
+    PATH_PREFIX.with(|val| val.to_string())
+}
+
+fn redirect(uri: &str) -> Redirect {
+    let prefixed_uri = path_prefix() + uri;
+    Redirect::to(&prefixed_uri)
+}
+
+async fn path_prefix_layer(req: Request, next: Next) -> Response {
+    let prefix = req
+        .headers()
+        .get("x-forwarded-prefix")
+        .and_then(|val| val.to_str().ok())
+        .unwrap_or("")
+        .to_string();
+
+    PATH_PREFIX.scope(prefix, next.run(req)).await
+}
+
 async fn security_headers_layer(req: Request, next: Next) -> impl IntoResponse {
     // Rendered Markdown may embed remote images via `![](…)`; relax img-src for that route only.
     const CSP_STRICT: HeaderValue = HeaderValue::from_static(
@@ -239,6 +264,7 @@ fn make_app(state: AppState, timeout: Duration, max_body_size: usize) -> Router
                     timeout,
                 ))
                 .layer(from_fn_with_state(state.clone(), handle_service_errors))
+                .layer(from_fn(path_prefix_layer))
                 .layer(from_fn(security_headers_layer)),
         )
         .with_state(state)

crates/wastebin_server/templates/base.html

@@ -1,3 +1,4 @@
+{% let path_prefix = crate::path_prefix() %}
 <!DOCTYPE html>
 <html lang="{{ lang.code() }}">
   <head>
@@ -7,27 +8,27 @@
     <title>{{ page.title }}{% block title_content %}{% endblock %}</title>
     {% match theme %}
       {% when Some with (Theme::Dark) %}
-        <link rel="stylesheet" href="{{ page.assets.css.dark.route() }}">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.dark.route() }}">
       {% when Some with (Theme::Light) %}
-        <link rel="stylesheet" href="{{ page.assets.css.light.route() }}">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.light.route() }}">
       {% when Some with (Theme::System) %}
-        <link rel="stylesheet" href="{{ page.assets.css.dark.route() }}" media="(prefers-color-scheme: dark)">
-        <link rel="stylesheet" href="{{ page.assets.css.light.route() }}" media="(prefers-color-scheme: light)">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.dark.route() }}" media="(prefers-color-scheme: dark)">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.light.route() }}" media="(prefers-color-scheme: light)">
       {% when None %}
-        <link rel="stylesheet" href="{{ page.assets.css.dark.route() }}" media="(prefers-color-scheme: dark)">
-        <link rel="stylesheet" href="{{ page.assets.css.light.route() }}" media="(prefers-color-scheme: light)">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.dark.route() }}" media="(prefers-color-scheme: dark)">
+        <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.light.route() }}" media="(prefers-color-scheme: light)">
     {% endmatch %}
-    <link rel="stylesheet" href="{{ page.assets.css.style.route() }}">
-    <link rel="icon" href="{{ page.assets.favicon.route() }}" type="image/png">
-    <noscript><link rel="stylesheet" href="{{ page.assets.css.no_js.route() }}"></noscript>
+    <link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.style.route() }}">
+    <link rel="icon" href="{{ path_prefix }}{{ page.assets.favicon.route() }}" type="image/png">
+    <noscript><link rel="stylesheet" href="{{ path_prefix }}{{ page.assets.css.no_js.route() }}"></noscript>
     {% block head %}{% endblock %}
   </head>
   <body{% block body_attrs %}{% endblock %}>
     {% block body %}
     <div id="main-container">
       <header>
         <div class="nav-group">
-          <a href="/" class="nav-button" title="{{ lang.t("nav.home") }}" aria-label="{{ lang.t("nav.home") }}">
+          <a href="{{ path_prefix }}/" class="nav-button" title="{{ lang.t("nav.home") }}" aria-label="{{ lang.t("nav.home") }}">
             <svg viewBox="0 0 24 24" width="16" height="16" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="m3 9 9-7 9 7v11a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2z"/><path d="M9 22V12h6v10"/></svg>
           </a>
           {% block title %}{% endblock %}
@@ -37,43 +38,43 @@
           <div class="theme-switcher">
             {% match theme %}
               {% when None %}
-                <a href="/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
+                <a href="{{ path_prefix }}/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 3a6 6 0 0 0 9 9 9 9 0 1 1-9-9Z"/></svg>
                 </a>
-                <a href="/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
+                <a href="{{ path_prefix }}/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="4"/><path d="M12 2v2"/><path d="M12 20v2"/><path d="m4.93 4.93 1.41 1.41"/><path d="m17.66 17.66 1.41 1.41"/><path d="M2 12h2"/><path d="M20 12h2"/><path d="m6.34 17.66-1.41 1.41"/><path d="m19.07 4.93-1.41 1.41"/></svg>
                 </a>
-                <a href="/theme?pref=system" class="theme-opt active" title="{{ lang.t("theme.auto") }}">
+                <a href="{{ path_prefix }}/theme?pref=system" class="theme-opt active" title="{{ lang.t("theme.auto") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect width="20" height="14" x="2" y="3" rx="2"/><path d="M8 21h8"/><path d="M12 17v4"/></svg>
                 </a>
               {% when Some with (Theme::System) %}
-                <a href="/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
+                <a href="{{ path_prefix }}/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 3a6 6 0 0 0 9 9 9 9 0 1 1-9-9Z"/></svg>
                 </a>
-                <a href="/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
+                <a href="{{ path_prefix }}/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="4"/><path d="M12 2v2"/><path d="M12 20v2"/><path d="m4.93 4.93 1.41 1.41"/><path d="m17.66 17.66 1.41 1.41"/><path d="M2 12h2"/><path d="M20 12h2"/><path d="m6.34 17.66-1.41 1.41"/><path d="m19.07 4.93-1.41 1.41"/></svg>
                 </a>
-                <a href="/theme?pref=system" class="theme-opt active" title="{{ lang.t("theme.auto") }}">
+                <a href="{{ path_prefix }}/theme?pref=system" class="theme-opt active" title="{{ lang.t("theme.auto") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect width="20" height="14" x="2" y="3" rx="2"/><path d="M8 21h8"/><path d="M12 17v4"/></svg>
                 </a>
               {% when Some with (Theme::Dark) %}
-                <a href="/theme?pref=dark" class="theme-opt active" title="{{ lang.t("theme.dark") }}">
+                <a href="{{ path_prefix }}/theme?pref=dark" class="theme-opt active" title="{{ lang.t("theme.dark") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 3a6 6 0 0 0 9 9 9 9 0 1 1-9-9Z"/></svg>
                 </a>
-                <a href="/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
+                <a href="{{ path_prefix }}/theme?pref=light" class="theme-opt" title="{{ lang.t("theme.light") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="4"/><path d="M12 2v2"/><path d="M12 20v2"/><path d="m4.93 4.93 1.41 1.41"/><path d="m17.66 17.66 1.41 1.41"/><path d="M2 12h2"/><path d="M20 12h2"/><path d="m6.34 17.66-1.41 1.41"/><path d="m19.07 4.93-1.41 1.41"/></svg>
                 </a>
-                <a href="/theme?pref=system" class="theme-opt" title="{{ lang.t("theme.auto") }}">
+                <a href="{{ path_prefix }}/theme?pref=system" class="theme-opt" title="{{ lang.t("theme.auto") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect width="20" height="14" x="2" y="3" rx="2"/><path d="M8 21h8"/><path d="M12 17v4"/></svg>
                 </a>
               {% when Some with (Theme::Light) %}
-                <a href="/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
+                <a href="{{ path_prefix }}/theme?pref=dark" class="theme-opt" title="{{ lang.t("theme.dark") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 3a6 6 0 0 0 9 9 9 9 0 1 1-9-9Z"/></svg>
                 </a>
-                <a href="/theme?pref=light" class="theme-opt active" title="{{ lang.t("theme.light") }}">
+                <a href="{{ path_prefix }}/theme?pref=light" class="theme-opt active" title="{{ lang.t("theme.light") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="4"/><path d="M12 2v2"/><path d="M12 20v2"/><path d="m4.93 4.93 1.41 1.41"/><path d="m17.66 17.66 1.41 1.41"/><path d="M2 12h2"/><path d="M20 12h2"/><path d="m6.34 17.66-1.41 1.41"/><path d="m19.07 4.93-1.41 1.41"/></svg>
                 </a>
-                <a href="/theme?pref=system" class="theme-opt" title="{{ lang.t("theme.auto") }}">
+                <a href="{{ path_prefix }}/theme?pref=system" class="theme-opt" title="{{ lang.t("theme.auto") }}">
                   <svg viewBox="0 0 24 24" width="12" height="12" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect width="20" height="14" x="2" y="3" rx="2"/><path d="M8 21h8"/><path d="M12 17v4"/></svg>
                 </a>
             {% endmatch %}

crates/wastebin_server/templates/burn-confirmation.html

@@ -9,9 +9,9 @@
     {% if let Some(title) = title %}
     <div class="dialog-section dialog-subtitle">{{ title }}</div>
     {% endif %}
-    <form class="dialog-section dialog-actions" action="/{{ id }}" method="post">
+    <form class="dialog-section dialog-actions" action="{{ path_prefix }}/{{ id }}" method="post">
       <input type="hidden" name="confirm_burn" value="1">
-      <a class="button button-secondary" href="/">{{ lang.t("burn_confirm.cancel") }}</a>
+      <a class="button button-secondary" href="{{ path_prefix }}/">{{ lang.t("burn_confirm.cancel") }}</a>
       <button class="button" type="submit">{{ lang.t("burn_confirm.reveal") }}</button>
     </form>
   </div>

crates/wastebin_server/templates/encrypted.html

@@ -1,15 +1,15 @@
 {% extends "base.html" %}
 
 {% block head %}
-<script defer src="{{ page.assets.password_toggle_js.route() }}"></script>
+<script defer src="{{path_prefix}}{{ page.assets.password_toggle_js.route() }}"></script>
 {% endblock %}
 
 {% block content %}
 <div class="flex-center">
   <div class="dialog">
     <div class="dialog-header">{{ lang.t("encrypted.title") }}</div>
-    <form action="/{{ id }}" method="post">
-      <input type="hidden" name="confirm_burn" value="1">
+    <form action="{{ path_prefix }}/{{ id }}" method="post">
+      <input type="hidden" name="``confirm_burn" value="1">
       <div class="dialog-section">
         <div class="password-input-wrap">
           <input type="password" id="password" name="password" placeholder="{{ lang.t("encrypted.placeholder") }}" autofocus>
@@ -20,7 +20,7 @@
         </div>
       </div>
       <div class="dialog-section dialog-actions">
-        <a class="button button-secondary" href="/">{{ lang.t("encrypted.cancel") }}</a>
+        <a class="button button-secondary" href="{{ path_prefix }}/">{{ lang.t("encrypted.cancel") }}</a>
         <button class="button" type="submit">{{ lang.t("encrypted.decrypt") }}</button>
       </div>
     </form>

crates/wastebin_server/templates/error.html

@@ -7,7 +7,7 @@
       {{ description }}
     </div>
     <div class="dialog-section dialog-actions">
-      <a class="button button-secondary" href="/">{{ lang.t("error.back") }}</a>
+      <a class="button button-secondary" href="{{ path_prefix }}/">{{ lang.t("error.back") }}</a>
     </div>
   </div>
 </div>

crates/wastebin_server/templates/formatted.html

@@ -2,7 +2,7 @@
 
 {% block view_toggle %}
   {% if is_markdown %}
-    <a id="view-toggle" href="/md/{{ key }}" class="nav-button" title="{{ lang.t("nav.rendered") }}" aria-label="{{ lang.t("nav.rendered") }}">
+    <a id="view-toggle" href="{{ path_prefix }}/md/{{ key }}" class="nav-button" title="{{ lang.t("nav.rendered") }}" aria-label="{{ lang.t("nav.rendered") }}">
       <svg viewBox="0 0 24 24" width="16" height="16" fill="currentColor" stroke="none"><path d="M22.269 19.385H1.731A1.73 1.73 0 0 1 0 17.654V6.345a1.73 1.73 0 0 1 1.731-1.73h20.538A1.73 1.73 0 0 1 24 6.345v11.308a1.73 1.73 0 0 1-1.731 1.731zm-16.5-3.462v-4.5l2.308 2.885 2.307-2.885v4.5h2.308V8.078h-2.308l-2.307 2.885-2.308-2.885H3.46v7.847zM21.231 12h-2.308V8.077h-2.307V12h-2.308l3.461 4.039z"/></svg>
     </a>
   {% endif %}
@@ -14,6 +14,6 @@
   </div>
   {% if !is_available %}
   <div id="burn-message" hidden data-message="{{ lang.t("paste.toast.burned") }}"></div>
-  <script defer src="{{ page.assets.burn_js.route()}}"></script>
+  <script defer src="{{ path_prefix }}{{ page.assets.burn_js.route()}}"></script>
   {% endif %}
 {% endblock %}