From d88a8102bf3df4d74193928cad77a7c49b4f3016 Mon Sep 17 00:00:00 2001
From: SinTan1729 <sayantan.santra689@gmail.com>
Date: Thu, 8 Dec 2022 19:46:45 -0600
Subject: [PATCH] Further refactor movie processing

---
 src/main.rs | 90 ++++++++++++++++++++++++++---------------------------
 1 file changed, 45 insertions(+), 45 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index c5b576d..603572a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -98,54 +98,12 @@ fn main() {
         }
 
         // Process the filename for movie entries
-        let (mut extension, movie_list) = process_movie(&filename, &tmdb, pattern);
-
-        // If nothing is found, skip
-        if movie_list.len() == 0 {
-            eprintln!("Could not find any entries matching {}", filename);
-            continue;
-        }
-
-        // Choose from the possible entries
-        let mut menu = youchoose::Menu::new(movie_list.iter())
-            .preview(display)
-            .preview_label(filename.to_string());
-        let choice = menu.show()[0];
-
-        // Handle the case for subtitle files
-        if ["srt", "ssa"].contains(&extension.as_str()) {
-            let languages = Vec::from(["en", "hi", "bn", "de", "fr", "sp", "ja", "n/a"]);
-            let mut lang_menu = youchoose::Menu::new(languages.iter());
-            let lang_choice = lang_menu.show()[0];
-            if languages[lang_choice] != "none" {
-                extension = format!("{}.{}", languages[lang_choice], extension);
-            }
-        }
-
-        // Create the new name
-        let mut new_name_vec = vec![
-            movie_list[choice].rename_format(pattern.to_string()),
-            extension,
-        ];
-        new_name_vec.retain(|x| !x.is_empty());
-        let new_name = new_name_vec.join(".");
-
-        // Process the renaming
-        if filename == new_name {
-            println!("{} already has correct name.", filename);
-        } else {
-            println!("{} -> {}", filename, new_name);
-            // Only do the rename of --dry-run isn't passed
-            if settings["dry_run"] == false {
-                println!("Renaming...");
-                fs::rename(filename, new_name).expect("Unable to rename file.");
-            }
-        }
+        process_file(&filename, &tmdb, pattern, settings["dry_run"]);
     }
 }
 
 // Function to process movie entries
-fn process_movie(filename: &String, tmdb: &TMDb, pattern: &str) -> (String, Vec<MovieEntry>) {
+fn process_file(filename: &String, tmdb: &TMDb, pattern: &str, dry_run: bool) {
     // Parse the filename for metadata
     let metadata = Metadata::from(filename.as_str()).expect("Could not parse filename");
     // Search using the TMDb API
@@ -184,7 +142,49 @@ fn process_movie(filename: &String, tmdb: &TMDb, pattern: &str) -> (String, Vec<
         }
         movie_list.push(movie_details);
     }
-    (metadata.extension().unwrap_or("").to_string(), movie_list)
+
+    // If nothing is found, skip
+    if movie_list.len() == 0 {
+        eprintln!("Could not find any entries matching {}", filename);
+        return;
+    }
+
+    // Choose from the possible entries
+    let mut menu = youchoose::Menu::new(movie_list.iter())
+        .preview(display)
+        .preview_label(filename.to_string());
+    let choice = menu.show()[0];
+
+    let mut extension = metadata.extension().unwrap_or("").to_string();
+    // Handle the case for subtitle files
+    if ["srt", "ssa"].contains(&extension.as_str()) {
+        let languages = Vec::from(["en", "hi", "bn", "de", "fr", "sp", "ja", "n/a"]);
+        let mut lang_menu = youchoose::Menu::new(languages.iter());
+        let lang_choice = lang_menu.show()[0];
+        if languages[lang_choice] != "none" {
+            extension = format!("{}.{}", languages[lang_choice], extension);
+        }
+    }
+
+    // Create the new name
+    let mut new_name_vec = vec![
+        movie_list[choice].rename_format(pattern.to_string()),
+        extension,
+    ];
+    new_name_vec.retain(|x| !x.is_empty());
+    let new_name = new_name_vec.join(".");
+
+    // Process the renaming
+    if *filename == new_name {
+        println!("{} already has correct name.", filename);
+    } else {
+        println!("{} -> {}", filename, new_name);
+        // Only do the rename of --dry-run isn't passed
+        if dry_run == false {
+            println!("Renaming...");
+            fs::rename(filename, new_name).expect("Unable to rename file.");
+        }
+    }
 }
 
 // Display function for preview in menu