diff --git a/src/main.rs b/src/main.rs index 602aac7..c53fbef 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,10 +1,23 @@ use std::fs::File; use std::io::{BufRead, BufReader, Result}; use std::collections::HashMap; +use std::env; use std::fs; +use std::process; fn main() -> Result<()> { - let file = File::open("beets_mv3u_renames")?; + let args: Vec = env::args().collect(); + + if args.len() < 3 { + println!("At least two arguments are required"); + process::exit(1); + } + + let rename_file = &args[1]; + let playlist_file = &args[2]; + println!("{:?}", rename_file); + + let file = File::open(rename_file)?; let mut renames = HashMap::new(); for line in BufReader::new(file).lines() { let line = line?; @@ -14,7 +27,7 @@ fn main() -> Result<()> { let mut outputs = Vec::new(); - let pls_file = File::open("playlist.m3u")?; + let pls_file = File::open(playlist_file)?; for entry in BufReader::new(pls_file).lines() { let entry = entry?; let new = renames.get(&entry).unwrap_or(&entry).to_string();