refactor import plugins part 3

1 files changed, 36 insertions, 1 deletions

diff --git a/import/src/plugins/misc.rs b/import/src/plugins/misc.rs
index 6f2c18e..8d7028c 100644
--- a/import/src/plugins/misc.rs
+++ b/import/src/plugins/misc.rs
@@ -4,15 +4,17 @@
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
 use crate::plugins::{ImportContext, ImportPlugin, PluginInfo};
-use anyhow::{Result, bail};
+use anyhow::{Context, Result, bail};
 use jellycache::{HashKey, cache_store};
 use jellycommon::{Asset, NodeID, NodeKind, PictureSlot, Visibility};
 use jellyremuxer::matroska::{AttachedFile, Segment};
 use log::info;
+use regex::Regex;
 use std::{
     fs::{File, read_to_string},
     io::Read,
     path::Path,
+    sync::LazyLock,
 };
 
 pub struct ImageFiles;
@@ -152,3 +154,36 @@ impl ImportPlugin for Children {
         Ok(())
     }
 }
+
+static RE_EPISODE_FILENAME: LazyLock<Regex> =
+    LazyLock::new(|| Regex::new(r#"([sS](?<season>\d+))?([eE](?<episode>\d+))( (.+))?"#).unwrap());
+
+pub struct EpisodeIndex;
+impl ImportPlugin for EpisodeIndex {
+    fn info(&self) -> PluginInfo {
+        PluginInfo {
+            name: "episode-info",
+            handle_media: true,
+            ..Default::default()
+        }
+    }
+    fn media(&self, ct: &ImportContext, node: NodeID, path: &Path, _seg: &Segment) -> Result<()> {
+        let filename = path.file_name().unwrap().to_string_lossy();
+        if let Some(cap) = RE_EPISODE_FILENAME.captures(&filename) {
+            if let Some(episode) = cap.name("episode").map(|m| m.as_str()) {
+                let season = cap.name("season").map(|m| m.as_str());
+                let episode = episode.parse::<usize>().context("parse episode num")?;
+                let season = season
+                    .unwrap_or("1")
+                    .parse::<usize>()
+                    .context("parse season num")?;
+
+                ct.db.update_node_init(node, |node| {
+                    node.index = Some(episode);
+                    Ok(())
+                })?;
+            }
+        }
+        Ok(())
+    }
+}