feat: update flag parsing and add HLS and Dash generation

Author: icecreammatt

src/ffmpeg.rs

@@ -0,0 +1,74 @@
+use std::{fs, process::Command};
+
+fn run_command(command: &str) {
+    let cmd = Command::new("sh")
+        .arg("-c")
+        .arg(command)
+        .output()
+        .expect("Error");
+
+    println!("status: {}", cmd.status);
+
+    println!("stdout: {}", String::from_utf8_lossy(&cmd.stdout));
+
+    println!("stderr: {}", String::from_utf8_lossy(&cmd.stderr));
+
+    let result = String::from_utf8_lossy(&cmd.stdout);
+    let result = result.to_string();
+    let result = result.as_str();
+
+    println!("{}", result);
+}
+
+// fn create_preview_image() {}
+
+// fn create_preview_gif() {}
+
+pub fn create_thumbnails(input: String, output: String) {
+    println!("\ncreate_thumbnails");
+
+    let thumbs_path = format!("{}/thumbs", output);
+
+    println!("Path: {}", thumbs_path);
+    if !fs::metadata(&thumbs_path).is_ok() {
+        fs::create_dir_all(&thumbs_path).unwrap();
+    }
+
+    let command = format!(
+        "ffmpeg -i {} -vf \"fps=1/4,scale=320:-1\" {}/img%03d.jpg",
+        input, thumbs_path
+    );
+
+    run_command(&command);
+}
+
+pub fn create_hls_encoding(input: String, output: String) {
+    println!("\npub create_hls_encoding");
+
+    if !fs::metadata(&output).is_ok() {
+        fs::create_dir_all(&output).unwrap();
+    }
+
+    // TODO: Sub add paths
+    // let paths = ["hls", "dash"];
+    // for path in paths {
+    //     let p = format!("{}/{}", path, output);
+    //     if !fs::metadata(&p).is_ok() {
+    //         fs::create_dir_all(&p).unwrap();
+    //     }
+    // }
+
+    let command = format!("ffmpeg -stream_loop 0  -i {} \
+        -map 0 -map 0 -map 0 -c:a aac -c:v h264_videotoolbox -allow_sw 1 \
+        -b:v:0 800k -s:v:0 1280x720 -profile:v:0 main \
+        -b:v:1 500k -s:v:1 640x340  -profile:v:1 main \
+        -b:v:2 300k -s:v:2 320x170  -profile:v:2 baseline \
+        -bf 1 \
+        -keyint_min 120 -g 120 -sc_threshold 0 -b_strategy 0 -ar:a:1 22050 -use_timeline 1 -use_template 1 \
+        -window_size 0 -adaptation_sets \"id=0,streams=v id=1,streams=a\" -hls_playlist 1 -seg_duration 4 \
+        -streaming 0 -f dash \
+        -hls_segment_filename \
+         -hls_playlist_type {}/vod \
+         {}/720p_%03d.m3u8 {}/720p.m3u8", input, output, output, output);
+    run_command(&command);
+}

src/fileio.rs

@@ -3,11 +3,11 @@ use std::fs;
 use std::fs::File;
 use std::io::Write;
 
-fn file_to_hyphen(name: &String) -> String {
+pub fn file_to_hyphen(name: &String) -> String {
     return String::from(name.replace("_", "-").replace(" ", "-").replace(".", "-"));
 }
 
-pub fn write_metadata(data: OutputMetadata) {
+pub fn write_metadata(data: &OutputMetadata) {
     let directory = file_to_hyphen(&data.title);
 
     if !fs::metadata(&directory).is_ok() {

src/main.rs

@@ -1,6 +1,50 @@
 use clap::{App, Arg};
 use std::process;
 
+enum Flags {
+    _Gif,
+    HLS,
+    _Metadata,
+    _PreviewImage,
+    Thumbnails,
+}
+
+// impl fmt::Display for Flags {
+//     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+//         match self {
+//             Flags::_Gif => write!(f, "gif"),
+//             Flags::HLS => write!(f, "HLS"),
+//             Flags::_Metadata => write!(f, "metadata"),
+//             Flags::_PreviewImage => write!(f, "preview_image"),
+//             Flags::Thumbnails => write!(f, "thumbs"),
+//         }
+//     }
+// }
+
+impl ToString for Flags {
+    fn to_string(&self) -> String {
+        self.as_str().to_string()
+    }
+}
+
+impl Flags {
+    fn as_str(&self) -> &'static str {
+        match self {
+            Flags::_Gif => "gif",
+            Flags::HLS => "HLS",
+            Flags::_Metadata => "metadata",
+            Flags::_PreviewImage => "preview_image",
+            Flags::Thumbnails => "thumbs",
+        }
+    }
+}
+
+use crate::{
+    ffmpeg::{create_hls_encoding, create_thumbnails},
+    fileio::file_to_hyphen,
+};
+
+mod ffmpeg;
 mod fileio;
 mod media_info;
 mod metadata;
@@ -15,23 +59,45 @@ fn main() {
                 .short('i')
                 .long("input")
                 .value_name("FILE")
-                .help("Sets the input file to use")
-                .takes_value(true),
+                .help("Sets the input file to use"),
+        )
+        .arg(
+            Arg::with_name(Flags::Thumbnails.as_str())
+                .short('t')
+                .long(Flags::Thumbnails.as_str())
+                .help("Generate thumbnails")
+                .takes_value(false),
+        )
+        .arg(
+            Arg::with_name(Flags::HLS.as_str())
+                .short('h')
+                .long(Flags::HLS.as_str())
+                .help("Generate hls chunks")
+                .takes_value(false),
         )
         .get_matches();
 
     let input = matches.value_of("input").unwrap_or("help");
-    // println!("Input file: {}", input);
-
     if input == "help" {
         println!("Usage: <TODO>");
-
         process::exit(1);
     }
 
     let out = media_info::get_media_info(&input);
-    println!("{:#?}", out);
-    fileio::write_metadata(out);
+    println!("{:#?}", &out);
+    fileio::write_metadata(&out);
+
+    let path = format!("./{}", file_to_hyphen(&out.title));
+
+    let gen_thumbs = matches.is_present(Flags::Thumbnails.as_str());
+    if gen_thumbs {
+        create_thumbnails(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());
+    }
 
     /*
     TODO:
@@ -42,11 +108,11 @@ fn main() {
     [x] Store Metadata in JSON
 
     [ ] FFMPEG to generate media
-        [ ] thumbnails
+        [x] thumbnails
         [ ] gif
         [ ] preview image
-        [ ] HLS
-        [ ] DASH
+        [x] HLS
+        [x] DASH
         [ ] Metadata
     [ ] Upload metadata to database