daan/media-linter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

WIP: rule collections instead of hardcoded rules

Browse source
This commit is contained in:
Daan Boerlage 2022-09-01 01:49:17 +02:00
parent 978bf187e0
commit 206070ed67
Signed by: daan
GPG key ID: FCE070E1E4956606
1 changed files with 24 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
src/main.rs
View file

@ -5,7 +5,7 @@ mod utils;
use std::fmt::{Display, Formatter};
use eyre::Result;
use clap::Parser;
use regex::Regex;
use regex::{Captures, Regex};
use walkdir::{DirEntry, WalkDir};
use args::Args;
@ -109,6 +109,20 @@ fn check_file(file: DirEntry, stats: &mut Stats, ctx: &mut Context, regex: &Rege
}
}
const SERIES_RULES_WITHOUT_NAME: [fn(&str, &Captures) -> Option<NonCompliantReason>; 3] = [
DashInTitle::check,
EpisodeMarker::check,
HasFluff::check,
];
const SERIES_RULES_WITH_NAME: [fn(&str, &Captures) -> Option<NonCompliantReason>; 5] = [
DashInTitle::check,
EpisodeMarker::check,
HasFluff::check,
HasEpisodeName::check,
MissingNameSeparator::check
];
fn lint_file_name(file: &DirEntry, filename: &str, regex: &Regex, episode_name: bool) -> ComplianceStatus {
if let Some(ext) = file.path().extension() {
if !VIDEO_EXTENSIONS.contains(&ext.to_str().unwrap()) {
@ -126,29 +140,19 @@ fn lint_file_name(file: &DirEntry, filename: &str, regex: &Regex, episode_name:
let captures = captures.unwrap();
if let Some(reason) = DashInTitle::check(filename, &captures) {
return ComplianceStatus::NonCompliant(reason);
if episode_name {
check_rules(SERIES_RULES_WITH_NAME, filename, &captures)
} else {
check_rules(SERIES_RULES_WITHOUT_NAME, filename, &captures)
}
}
if let Some(reason) = EpisodeMarker::check(filename, &captures) {
return ComplianceStatus::NonCompliant(reason);
}
if let Some(reason) = HasFluff::check(filename, &captures) {
return ComplianceStatus::NonCompliant(reason);
}
if let Some(reason) = HasEpisodeName::check(filename, &captures) {
return if episode_name {
ComplianceStatus::NonCompliant(reason)
} else {
ComplianceStatus::Compliant
fn check_rules<const N: usize>(validation_rules: [fn(&str, &Captures) -> Option<NonCompliantReason>; N], filename: &str, captures: &Captures) -> ComplianceStatus {
for rule in validation_rules {
if let Some(reason) = rule(filename, &captures) {
return ComplianceStatus::NonCompliant(reason);
}
}
if let Some(reason) = MissingNameSeparator::check(filename, &captures) {
return ComplianceStatus::NonCompliant(reason);
}
ComplianceStatus::Compliant
}