web: add Download log option

Author: manio

src/web.rs

@@ -1,12 +1,16 @@
 use crate::config::AppConfig;
 use axum::{
+    body::Body,
     extract::State,
+    http::{header, Response, StatusCode},
     response::{Html, IntoResponse},
     routing::get,
     Json, Router,
 };
+use chrono::Local;
 use std::path::PathBuf;
 use std::sync::{Arc, Mutex};
+use tokio::fs;
 
 const TEMPLATE: &str = include_str!("../static/index.html");
 const PICO_CSS: &str = include_str!("../static/pico.min.css");
@@ -21,6 +25,7 @@ pub fn app(state: Arc<AppState>) -> Router {
     Router::new()
         .route("/", get(index))
         .route("/config", get(get_config).post(set_config))
+        .route("/download", get(download_handler))
         .with_state(state)
 }
 
@@ -33,6 +38,32 @@ async fn index() -> impl IntoResponse {
     Html(html)
 }
 
+fn generate_filename() -> String {
+    let now = Local::now();
+    now.format("%Y%m%d%H%M%S_aa-proxy-rs.log").to_string()
+}
+
+async fn download_handler(State(state): State<Arc<AppState>>) -> impl IntoResponse {
+    let file_path = state.config.lock().unwrap().logfile.clone();
+    let filename = generate_filename();
+
+    match fs::read(file_path).await {
+        Ok(content) => Response::builder()
+            .status(StatusCode::OK)
+            .header(header::CONTENT_TYPE, "application/octet-stream")
+            .header(
+                header::CONTENT_DISPOSITION,
+                format!("attachment; filename=\"{}\"", filename),
+            )
+            .body(Body::from(content))
+            .unwrap(),
+        Err(_) => Response::builder()
+            .status(StatusCode::NOT_FOUND)
+            .body(Body::from("Cannot access log file"))
+            .unwrap(),
+    }
+}
+
 async fn get_config(State(state): State<Arc<AppState>>) -> impl IntoResponse {
     let cfg = state.config.lock().unwrap().clone();
     Json(cfg)

static/index.html

@@ -12,10 +12,22 @@
       <small>build: {BUILD_DATE}, git: {GIT_DATE}-{GIT_HASH}</small>
     </div>
     <div style="text-align: center; margin-top: 1em">
-      <h3>🛸 aa-proxy-rs configuration</h3>
+      <h3>🛸 aa-proxy-rs</h3>
     </div>
     <form id="config-form">
       <table>
+        <tr>
+          <td colspan="2" style="color: #fff; background-color: #202632">
+            <strong>🚀 ACTIONS</strong>
+          </td>
+        </tr>
+        <tr>
+          <td colspan="2">
+            <button type="button" onclick="window.location.href='/download'">
+              📥 Download log
+            </button>
+          </td>
+        </tr>
         <tr>
           <td colspan="2" style="color: #fff; background-color: #202632">
             <strong>⚙️ GENERAL OPTIONS</strong>