thumbnail generation

1 files changed, 35 insertions, 0 deletions

diff --git a/transcoder/src/thumbnail.rs b/transcoder/src/thumbnail.rs
new file mode 100644
index 0000000..5baf888
--- /dev/null
+++ b/transcoder/src/thumbnail.rs
@@ -0,0 +1,35 @@
+use crate::LOCAL_IMAGE_TRANSCODING_TASKS;
+use jellybase::cache::async_cache_file;
+use jellycommon::AssetLocation;
+use log::info;
+use std::{path::Path, process::Stdio};
+use tokio::{io::copy, process::Command};
+
+pub async fn create_thumbnail(path: &Path, time: f64) -> anyhow::Result<AssetLocation> {
+    Ok(async_cache_file(
+        &["thumb", path.to_str().unwrap(), &format!("{time}")],
+        move |mut output| async move {
+            let _permit = LOCAL_IMAGE_TRANSCODING_TASKS.acquire().await?;
+            info!("creating thumbnail of {path:?} at {time}s",);
+
+            let mut proc = Command::new("ffmpeg")
+                .stdout(Stdio::piped())
+                .args(&["-ss", &format!("{time}")])
+                .args(&["-f", "matroska", "-i", path.to_str().unwrap()])
+                .args(&["-frames:v", "1"])
+                .args(&["-c:v", "qoi"])
+                .args(&["-f", "image2"])
+                .args(&["-update", "1"])
+                .arg("pipe:1")
+                .spawn()?;
+
+            let mut stdout = proc.stdout.take().unwrap();
+            copy(&mut stdout, &mut output).await?;
+
+            proc.wait().await.unwrap().exit_ok()?;
+            info!("done");
+            Ok(())
+        },
+    )
+    .await?)
+}