SinTan1729/movie-rename
1
0
Fork 0
mirror of https://github.com/SinTan1729/movie-rename.git synced 2024-10-16 16:53:53 -05:00
Code Issues Projects Releases Packages Wiki Activity

new: Try to sanitize the filename

Browse source
This commit is contained in:
Sayantan Santra 2023-05-21 00:56:28 -05:00
parent 7cb18202eb
commit 02fea4a71c
Signed by: SinTan1729
GPG key ID: EB3E68BFBA25C85F
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/structs.rs
View file

@ -27,18 +27,32 @@ impl MovieEntry {
// Generate desired filename from movie entry // Generate desired filename from movie entry
pub fn rename_format(&self, mut format: String) -> String { pub fn rename_format(&self, mut format: String) -> String {
format = format.replace("{title}", self.title.as_str()); const PATTERN: &str = "^~#%$*+={}?@'`/\\\"><|:&!";
// Try to sanitize the title to avoid some characters
let mut title = self.title.clone();
title.retain(|c| !PATTERN.contains(c));
title.truncate(159);
format = format.replace("{title}", title.as_str());
if self.year.as_str() != "N/A" { if self.year.as_str() != "N/A" {
format = format.replace("{year}", self.year.as_str()); format = format.replace("{year}", self.year.as_str());
} else { } else {
format = format.replace("{year}", ""); format = format.replace("{year}", "");
} }
if self.director.as_str() != "N/A" { if self.director.as_str() != "N/A" {
format = format.replace("{director}", self.director.as_str()); // Try to sanitize the director's name to avoid some characters
let mut director = self.director.clone();
director.retain(|c| !PATTERN.contains(c));
director.truncate(63);
format = format.replace("{director}", director.as_str());
} else { } else {
format = format.replace("{director}", ""); format = format.replace("{director}", "");
} }
format format
.trim_matches(|c| "- ".contains(c))
.replace("--", "-")
.replace("- ", "")
} }
} }