add support for fetching media info using ffprobe

Author: neevek

backend/src/main.rs

@@ -14,6 +14,7 @@ use path_dedot::*;
 use serde::Deserialize;
 use std::net::{IpAddr, Ipv6Addr, SocketAddr};
 use std::path::PathBuf;
+use std::process::Command;
 use std::str::FromStr;
 use tokio::io::AsyncWriteExt;
 use tower_http::services::ServeDir;
@@ -78,6 +79,7 @@ async fn main() {
         .route("/api/listing/*path", get(list_files).post(create_dir))
         .route("/api/upload/*path", post(save_request_body))
         .route("/api/delete/*path", post(delete_path))
+        .route("/api/ffprobe/*path", get(ffprobe))
         .nest(
             "/api/static",
             get_service(ServeDir::new(unsafe {
@@ -217,18 +219,15 @@ async fn serve_root() -> impl IntoResponse {
 }
 
 async fn list_files(Path(path): Path<String>) -> impl IntoResponse {
-    let mut abs_path;
-    unsafe {
-        abs_path = SERVE_DIR.as_ref().unwrap().to_path_buf();
-    }
     let path = path.trim_start_matches('/');
-    abs_path.push(path);
+    let parent_dir = unsafe { SERVE_DIR.as_ref().unwrap() };
+    let full_path = parent_dir.join(path.trim_start_matches('/'));
 
-    log::debug!("list files for path: {:?}", abs_path);
+    log::debug!("list files for path: {:?}", full_path);
 
-    if abs_path.is_dir() {
+    if full_path.is_dir() {
         let mut descendants = vec![];
-        for entry in WalkDir::new(&abs_path)
+        for entry in WalkDir::new(&full_path)
             .follow_links(true)
             .max_depth(1)
             .into_iter()
@@ -246,7 +245,7 @@ async fn list_files(Path(path): Path<String>) -> impl IntoResponse {
         };
 
         return (StatusCode::OK, Json(dir_desc)).into_response();
-    } else if abs_path.is_symlink() || abs_path.is_file() {
+    } else if full_path.is_symlink() || full_path.is_file() {
         return Redirect::permanent(format!("/static/{}", path).as_str()).into_response();
     }
 
@@ -287,6 +286,35 @@ fn convert_dir_entry(entry: &walkdir::DirEntry) -> DirEntry {
     }
 }
 
+async fn ffprobe(Path(path): Path<String>) -> impl IntoResponse {
+    let path = path.trim_start_matches('/');
+    let parent_dir = unsafe { SERVE_DIR.as_ref().unwrap() };
+    let full_path = parent_dir.join(path.trim_start_matches('/'));
+
+    if full_path.is_file() {
+        if let Ok(output) = Command::new("ffprobe")
+            .args([
+                "-v",
+                "quiet",
+                "-print_format",
+                "json",
+                "-show_format",
+                "-show_streams",
+            ])
+            .arg(full_path)
+            .output()
+        {
+            let json_str = String::from_utf8(output.stdout).unwrap_or("{}".to_string());
+            return (StatusCode::OK, Json(json_str)).into_response();
+        }
+    }
+
+    let json_resp = Json(JsonResponse::Failed {
+        msg: Some("ffprobe not found".to_string()),
+    });
+    (StatusCode::OK, json_resp).into_response()
+}
+
 #[derive(Debug)]
 struct AppError(String);
 

frontend/src/main.rs

@@ -8,7 +8,7 @@ use fast_qr::{
     QRBuilder, Version, ECL,
 };
 use gloo_net::http::Request;
-use log::info;
+use log::{error, info};
 use reqwest::Url;
 
 fn main() {
@@ -63,6 +63,8 @@ fn Listing(cx: Scope) -> Element {
         });
     }
 
+    let info_state = use_state(&cx, || None as Option<String>);
+
     cx.render(match fut.value() {
         Some(Ok(dir_desc)) => rsx!(
             div {
@@ -76,7 +78,9 @@ fn Listing(cx: Scope) -> Element {
                 create_dir_state: create_dir_state.clone(),
             }
 
-            ListingTable{ dir_desc: dir_desc, cur_url: &url, update_state: update_state },
+            ListingTable{ dir_desc: dir_desc, cur_url: &url, update_state: update_state, info_state: info_state },
+
+            InfoDialog { info_state: info_state }
         ),
         Some(Err(err)) => rsx!("Error: {err}"),
         _ => rsx!("Loading..."),
@@ -244,6 +248,7 @@ fn ListingTable<'a>(cx: Scope<'a, DirDescProps<'a>>) -> Element {
                     entry: entry,
                     cur_path: cur_path,
                     update_state: cx.props.update_state,
+                    info_state: cx.props.info_state,
                     qrcode_state: qrcode_state,
                 }))
 
@@ -288,6 +293,37 @@ fn TableRow<'a>(cx: Scope<'a, DirEntryProps<'a>>) -> Element {
                     img { src:"data:image/x-icon;base64,AAABAAEAEBACAAAAAACwAAAAFgAAACgAAAAQAAAAIAAAAAEAAQAAAAAAQAAAAAAAAAAAAAAAAgAAAAAAAAAAAAAA////AAKnAAB6MgAASlIAAEtCAAB7AAAAAnkAAP/YAACDBQAAUGMAAPy/AAACQAAAel4AAEpSAABK0gAAel4AAAJAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA", alt:"QRCode" }
                 }
 
+                a {
+                    href: "#",
+                    prevent_default: "onclick",
+                    onclick: move |_| {
+                        let path = format!("/api/ffprobe{}/{}", cx.props.cur_path, entry.file_name);
+                        let info_state = cx.props.info_state.clone();
+                        cx.spawn(async move {
+                            let resp = Request::get(path.as_str())
+                                .send()
+                                .await;
+
+                            match resp {
+                                Ok(resp) => {
+                                    let text = resp.text().await.unwrap_or("".to_string());
+                                    if text.is_empty() || text == "\"{\\n\\n}\\n\"" {
+                                        info_state.set(Some("Not Available!".to_string()));
+                                    } else {
+                                        let json_resp: serde_json::Value = serde_json::from_str(text.as_str()).unwrap_or(serde_json::Value::default());
+                                        let json_str = json_resp.to_string().replace("\\n", "\n").replace("\\", "");
+                                        info_state.set(Some(json_str));
+                                    }
+                                }
+                                Err(err) => {
+                                    error!("failed: {}", err);
+                                }
+                            }
+                        });
+                    },
+                    "ℹ️ ",
+                }
+
                 a {
                     href: "{api_link}",
                     if entry.file_type == common::FileType::Directory {
@@ -398,6 +434,61 @@ fn Tooltip<'a>(cx: Scope<'a, TooltipProps<'a>>) -> Element {
 //     }))
 // }
 
+#[inline_props]
+fn InfoDialog<'a>(cx: Scope<'a>, info_state: &'a UseState<Option<String>>) -> Element {
+    cx.render(if let Some(str_info) = info_state.get() {
+        rsx!(div {
+            style: "
+                position: fixed;
+                width: 860px;
+                height: 640px;
+                left: 50%;
+                margin-left: -430px;
+                top: 50%;
+                margin-top: -320px;
+                z-index: 20;
+                border-radius: 5px;
+                border: 2px solid #ccc;
+                background: #eee;
+                overflow: scroll;
+            ",
+            textarea {
+                style: "
+                    width:850px;
+                    height: 580px;
+                    margin:5px;
+                    border: none;
+                    box-sizing:border-box;
+                    resize: none;
+                    border-bottom: 2px solid #ccc;
+                ",
+                disabled: "true",
+                value: "{str_info}",
+            }
+            p {
+                style: "
+                    text-align: center;
+                    width: 100%;
+                    position: absolute;
+                    bottom: 0px;
+                    margin:0px;
+                    box-sizing:border-box;
+                    padding: 10px;
+                ",
+                input {
+                    style: "width: 200px; height: 30px",
+                    prevent_default: "onclick",
+                    r#type: "button",
+                    value: "Close",
+                    onclick: |_| info_state.set(None),
+                }
+            }
+        })
+    } else {
+        rsx!("")
+    })
+}
+
 #[derive(Props)]
 struct TooltipProps<'a> {
     w: i32,
@@ -412,6 +503,7 @@ pub struct DirDescProps<'a> {
     cur_url: &'a Url,
     dir_desc: &'a DirDesc,
     update_state: &'a UseState<bool>,
+    info_state: &'a UseState<Option<String>>,
 }
 
 #[derive(Props)]
@@ -421,6 +513,7 @@ struct DirEntryProps<'a> {
     cur_path: &'a str,
     update_state: &'a UseState<bool>,
     qrcode_state: &'a UseState<Option<QRCodeParams>>,
+    info_state: &'a UseState<Option<String>>,
 }
 
 struct QRCodeParams {