feat: add preview image generation

Author: icecreammatt

src/ffmpeg.rs

@@ -20,10 +20,22 @@ fn run_command(command: &str) {
     println!("{}", result);
 }
 
-// fn create_preview_image() {}
-
 // fn create_preview_gif() {}
 
+pub fn create_preview_image(input: String, output: String) {
+    let path = format!("{}/lg", output);
+    if !fs::metadata(&path).is_ok() {
+        fs::create_dir_all(&path).unwrap();
+    }
+
+    let command = format!(
+        "ffmpeg -i {} -vf scale=iw*sar:ih,setsar=1 -ss 00:00:05 -t 1 -vframes 1 {}/{}.jpg",
+        input, path, output
+    );
+
+    run_command(&command);
+}
+
 pub fn create_thumbnails(input: String, output: String) {
     println!("\ncreate_thumbnails");
 

src/main.rs

@@ -1,11 +1,12 @@
+use crate::ffmpeg::create_preview_image;
 use clap::{App, Arg};
 use std::process;
 
 enum Flags {
     _Gif,
     HLS,
     _Metadata,
-    _PreviewImage,
+    PreviewImage,
     Thumbnails,
 }
 
@@ -33,7 +34,7 @@ impl Flags {
             Flags::_Gif => "gif",
             Flags::HLS => "HLS",
             Flags::_Metadata => "metadata",
-            Flags::_PreviewImage => "preview_image",
+            Flags::PreviewImage => "preview_image",
             Flags::Thumbnails => "thumbs",
         }
     }
@@ -68,6 +69,13 @@ fn main() {
                 .help("Generate thumbnails")
                 .takes_value(false),
         )
+        .arg(
+            Arg::with_name(Flags::PreviewImage.as_str())
+                .short('p')
+                .long(Flags::PreviewImage.as_str())
+                .help("Generate preview")
+                .takes_value(false),
+        )
         .arg(
             Arg::with_name(Flags::HLS.as_str())
                 .short('h')
@@ -94,6 +102,11 @@ fn main() {
         create_thumbnails(out.file_name.clone(), path.clone());
     }
 
+    let gen_preview_image = matches.is_present(Flags::PreviewImage.as_str());
+    if gen_preview_image {
+        create_preview_image(out.file_name.clone(), path.clone());
+    }
+
     let gen_hls = matches.is_present(Flags::HLS.as_str());
     if gen_hls {
         create_hls_encoding(out.file_name.clone(), path.clone());
@@ -110,7 +123,7 @@ fn main() {
     [ ] FFMPEG to generate media
         [x] thumbnails
         [ ] gif
-        [ ] preview image
+        [x] preview image
         [x] HLS
         [x] DASH
         [ ] Metadata