SinTan1729/movie-rename
1
0
Fork 0
mirror of https://github.com/SinTan1729/movie-rename.git synced 2024-12-26 12:18:37 -06:00
Code Issues Projects Releases Packages Wiki Activity

Visual improvements

Browse source
This commit is contained in:
Sayantan Santra 2022-12-10 22:23:44 -06:00
parent c52643033f
commit 2d6e063a97
1 changed files with 15 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/functions.rs
View file

@ -31,9 +31,10 @@ pub fn process_file(
} }
None => {} None => {}
} }
println!(" Processing {}...", file_base);
// Parse the filename for metadata // Parse the filename for metadata
let metadata = Metadata::from(file_base.as_str()).expect("Could not parse filename!"); let metadata = Metadata::from(file_base.as_str()).expect(" Could not parse filename!");
// Search using the TMDb API // Search using the TMDb API
let mut search = tmdb.search(); let mut search = tmdb.search();
search.title(metadata.title()); search.title(metadata.title());
@ -47,7 +48,7 @@ pub fn process_file(
if let Ok(search_results) = search.execute() { if let Ok(search_results) = search.execute() {
results = search_results.results; results = search_results.results;
} else { } else {
eprintln!("There was an error while searching {}!", filename); eprintln!(" There was an error while searching {}!", file_base);
} }
let mut movie_list: Vec<MovieEntry> = Vec::new(); let mut movie_list: Vec<MovieEntry> = Vec::new();
@ -73,17 +74,17 @@ pub fn process_file(
// If nothing is found, skip // If nothing is found, skip
if movie_list.len() == 0 { if movie_list.len() == 0 {
eprintln!("Could not find any entries matching {}!", filename); eprintln!(" Could not find any entries matching {}!", file_base);
return ("".to_string(), true); return ("".to_string(), true);
} }
// Choose from the possible entries // Choose from the possible entries
let choice = Select::new( let choice = Select::new(
format!("Possible choices for {}:", file_base).as_str(), format!(" Possible choices for {}:", file_base).as_str(),
movie_list, movie_list,
) )
.prompt() .prompt()
.expect("Invalid choice!"); .expect(" Invalid choice!");
let mut extension = metadata.extension().unwrap_or("").to_string(); let mut extension = metadata.extension().unwrap_or("").to_string();
// Handle the case for subtitle files // Handle the case for subtitle files
@ -93,15 +94,16 @@ pub fn process_file(
let filename_parts: Vec<&str> = filename.rsplit(".").collect(); let filename_parts: Vec<&str> = filename.rsplit(".").collect();
if filename_parts.len() >= 3 && filename_parts[1].len() == 2 { if filename_parts.len() >= 3 && filename_parts[1].len() == 2 {
println!( println!(
"Keeping language {} as detected in the subtitle file's extension...", " Keeping language {} as detected in the subtitle file's extension...",
get_long_lang(filename_parts[1]) get_long_lang(filename_parts[1])
); );
extension = format!("{}.{}", filename_parts[1], extension); extension = format!("{}.{}", filename_parts[1], extension);
} else { } else {
let lang_list = Language::generate_list(); let lang_list = Language::generate_list();
let lang_choice = Select::new("Choose the language for the subtitle file:", lang_list) let lang_choice =
.prompt() Select::new(" Choose the language for the subtitle file:", lang_list)
.expect("Invalid choice!"); .prompt()
.expect(" Invalid choice!");
if lang_choice.short != "none".to_string() { if lang_choice.short != "none".to_string() {
extension = format!("{}.{}", lang_choice.short, extension); extension = format!("{}.{}", lang_choice.short, extension);
} }
@ -122,15 +124,15 @@ pub fn process_file(
// Process the renaming // Process the renaming
if *filename == new_name { if *filename == new_name {
println!("[file] '{}' already has correct name.", file_base); println!(" [file] '{}' already has correct name.", file_base);
} else { } else {
println!("[file] '{}' -> '{}'", file_base, new_name_with_ext); println!(" [file] '{}' -> '{}'", file_base, new_name_with_ext);
// Only do the rename of --dry-run isn't passed // Only do the rename of --dry-run isn't passed
if dry_run == false { if dry_run == false {
if Path::new(new_name.as_str()).is_file() == false { if Path::new(new_name.as_str()).is_file() == false {
fs::rename(filename, new_name.as_str()).expect("Unable to rename file!"); fs::rename(filename, new_name.as_str()).expect(" Unable to rename file!");
} else { } else {
eprintln!("Destination file already exists, skipping..."); eprintln!(" Destination file already exists, skipping...");
} }
} }
} }