feat: add gif generation

Author: icecreammatt

src/ffmpeg.rs

@@ -20,7 +20,18 @@ fn run_command(command: &str) {
     println!("{}", result);
 }
 
-// fn create_preview_gif() {}
+pub fn create_preview_gif(input: &str, output: &str) {
+    let path = format!("{}/gif", output);
+    if fs::metadata(&path).is_err() {
+        fs::create_dir_all(&path).unwrap();
+    }
+
+    let command = format!(
+    "ffmpeg -ss 30 -t 3 -i {} -vf \"fps=10,scale=320:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse\" -loop 0 {}/{}.gif", input, path, output
+    );
+
+    run_command(&command);
+}
 
 pub fn create_preview_image(input: &str, output: &str) {
     let path = format!("{}/lg", output);

src/main.rs

@@ -1,9 +1,9 @@
-use crate::ffmpeg::create_preview_image;
+use crate::ffmpeg::{create_preview_gif, create_preview_image};
 use clap::{App, Arg};
 use std::process;
 
 enum Flags {
-    _Gif,
+    Gif,
     Hls,
     _Metadata,
     PreviewImage,
@@ -32,7 +32,7 @@ impl ToString for Flags {
 impl Flags {
     fn as_str(&self) -> &'static str {
         match self {
-            Flags::_Gif => "gif",
+            Flags::Gif => "gif",
             Flags::Hls => "HLS",
             Flags::_Metadata => "metadata",
             Flags::PreviewImage => "preview_image",
@@ -85,6 +85,13 @@ fn main() {
                 .help("Generate hls chunks")
                 .takes_value(false),
         )
+        .arg(
+            Arg::with_name(Flags::Gif.as_str())
+                .short('g')
+                .long(Flags::Gif.as_str())
+                .help("Generate gif")
+                .takes_value(false),
+        )
         .get_matches();
 
     let input = matches.value_of("input").unwrap_or(Flags::Help.as_str());
@@ -114,6 +121,11 @@ fn main() {
         create_hls_encoding(&out.file_name, &path);
     }
 
+    let gen_gif = matches.is_present(Flags::Gif.as_str());
+    if gen_gif {
+        create_preview_gif(&out.file_name, &path);
+    }
+
     /*
     TODO:
     [x] Take filename as argument to process