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

Only compile the regex once

Browse source
This commit is contained in:
Daan Boerlage 2022-08-29 03:00:05 +02:00
parent ca87d5391f
commit 8d9a2329be
Signed by: daan
GPG key ID: FCE070E1E4956606
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/main.rs
View file

@ -5,6 +5,7 @@ mod utils;
use std::fmt::{Display, Formatter}; use std::fmt::{Display, Formatter};
use eyre::Result; use eyre::Result;
use clap::Parser; use clap::Parser;
use regex::Regex;
use walkdir::{DirEntry, WalkDir}; use walkdir::{DirEntry, WalkDir};
use args::Args; use args::Args;
@ -47,9 +48,11 @@ fn main() -> Result<()> {
error: 0, error: 0,
}; };
let regex = Regex::new(SERIES_REGEX).unwrap();
for entry in WalkDir::new(params.path) { for entry in WalkDir::new(params.path) {
if let Ok(file) = entry { if let Ok(file) = entry {
check_file(file, &mut stats, params.no_emoji, params.show_success); check_file(file, &mut stats, &regex, params.no_emoji, params.show_success);
} }
} }
@ -58,11 +61,11 @@ fn main() -> Result<()> {
Ok(()) Ok(())
} }
fn check_file(file: DirEntry, stats: &mut Stats, no_emoji: bool, show_success: bool) -> () { fn check_file(file: DirEntry, stats: &mut Stats, regex: &Regex, no_emoji: bool, show_success: bool) -> () {
let file_type = file.file_type(); let file_type = file.file_type();
if file_type.is_file() { if file_type.is_file() {
let filename = file.file_name().to_str().unwrap(); let filename = file.file_name().to_str().unwrap();
match lint_file_name(&file, filename) { match lint_file_name(&file, filename, regex) {
ComplianceStatus::NotMatched => { ComplianceStatus::NotMatched => {
stats.files += 1; stats.files += 1;
stats.error += 1; stats.error += 1;
@ -87,7 +90,7 @@ fn check_file(file: DirEntry, stats: &mut Stats, no_emoji: bool, show_success: b
} }
} }
fn lint_file_name(file: &DirEntry, filename: &str) -> ComplianceStatus { fn lint_file_name(file: &DirEntry, filename: &str, regex: &Regex) -> ComplianceStatus {
if let Some(ext) = file.path().extension() { if let Some(ext) = file.path().extension() {
if !VIDEO_EXTENSIONS.contains(&ext.to_str().unwrap()) { if !VIDEO_EXTENSIONS.contains(&ext.to_str().unwrap()) {
return ComplianceStatus::Ignored; return ComplianceStatus::Ignored;
@ -96,7 +99,7 @@ fn lint_file_name(file: &DirEntry, filename: &str) -> ComplianceStatus {
return ComplianceStatus::Ignored; return ComplianceStatus::Ignored;
} }
let captures = regex::Regex::new(SERIES_REGEX).unwrap().captures(filename); let captures = regex.captures(filename);
if captures.is_none() { if captures.is_none() {
return ComplianceStatus::NotMatched; return ComplianceStatus::NotMatched;