Allow filtering on media type

Author: StirlingMouse

mlm_db/src/impls/categories.rs

@@ -1,3 +1,5 @@
+use std::str::FromStr;
+
 use crate::{Category, MainCat, MediaType, OldMainCat};
 
 impl MediaType {
@@ -59,6 +61,29 @@ impl std::fmt::Display for MediaType {
     }
 }
 
+impl FromStr for MediaType {
+    type Err = String;
+
+    fn from_str(value: &str) -> Result<Self, Self::Err> {
+        match value.to_lowercase().as_str() {
+            "audiobook" => Ok(MediaType::Audiobook),
+            "ebook" => Ok(MediaType::Ebook),
+            "musicology" => Ok(MediaType::Musicology),
+            "radio" => Ok(MediaType::Radio),
+            "manga" => Ok(MediaType::Manga),
+            "comic book / graphic novel" => Ok(MediaType::ComicBook),
+            "comic book" => Ok(MediaType::ComicBook),
+            "comics" => Ok(MediaType::ComicBook),
+            "graphic novel" => Ok(MediaType::ComicBook),
+            "periodical_ebook" => Ok(MediaType::PeriodicalEbook),
+            "periodical ebook" => Ok(MediaType::PeriodicalEbook),
+            "periodical_audiobook" => Ok(MediaType::PeriodicalAudiobook),
+            "periodical audiobook" => Ok(MediaType::PeriodicalAudiobook),
+            _ => Err(format!("Unknown media type: {}", value)),
+        }
+    }
+}
+
 impl From<OldMainCat> for MediaType {
     fn from(value: OldMainCat) -> Self {
         match value {

server/src/config.rs

@@ -1,7 +1,7 @@
 use std::{collections::BTreeMap, path::PathBuf};
 
 use mlm_db::{
-    Flags, Language, OldDbMainCat, Size,
+    Flags, Language, MediaType, OldDbMainCat, Size,
     impls::{parse, parse_opt, parse_vec},
 };
 use mlm_mam::{
@@ -212,6 +212,9 @@ pub struct TorrentFilter {
     #[serde(default)]
     pub name: Option<String>,
 
+    #[serde(default)]
+    #[serde(deserialize_with = "parse_vec")]
+    pub media_type: Vec<MediaType>,
     #[serde(default)]
     pub categories: Categories,
     #[serde(default)]

server/src/config_impl.rs

@@ -28,6 +28,13 @@ impl Config {
 
 impl TorrentFilter {
     pub fn matches(&self, torrent: &MaMTorrent) -> bool {
+        if !self.media_type.is_empty()
+            && MediaType::from_id(torrent.mediatype)
+                .is_none_or(|media_type| !self.media_type.contains(&media_type))
+        {
+            return false;
+        }
+
         if !self.categories.matches(torrent.category) {
             return false;
         }