From cbd26a619ce78e67a8d0a18b739ac3e861e6932e Mon Sep 17 00:00:00 2001
From: Azalea Gui <me@hydev.org>
Date: Fri, 24 Feb 2023 12:02:05 -0500
Subject: [PATCH] [+] Video extensions

---
 backend/src/generator.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/backend/src/generator.rs b/backend/src/generator.rs
index 06cf421..0dfc5a3 100644
--- a/backend/src/generator.rs
+++ b/backend/src/generator.rs
@@ -20,15 +20,20 @@ pub struct Generator {
     pub(crate) thumbnailers: Thumbnailers,
     pub(crate) encoders: Encoders,
     pub(crate) base: PathBuf,
+    video_ext: HashSet<String>
 }
 
 impl Generator {
     pub fn new(base: PathBuf) -> Result<Generator> {
+        let video = vec!["webm", "mkv", "flv", "vob", "ogv", "drc", "avi", "mov", "wmv",
+                         "yuv", "rm", "rmvb", "amv", "mp4", "m4p", "m4v", "mpg", "mpv", "mp2", "mpeg", "mpe", "3gp"];
+
         Ok(Generator {
             mime_db: SharedMimeInfo::new(),
             thumbnailers: Thumbnailers::load_all()?,
             encoders: Encoders::load()?,
-            base: fs::canonicalize(base)?
+            base: fs::canonicalize(base)?,
+            video_ext: HashSet::from_iter(video.into_iter().map(|x| x.to_string())),
         })
     }