Use getopts to parse command-line

This commit is contained in:
Matteo Cypriani 2016-12-08 16:58:26 -05:00
parent d5d83ab03e
commit 2b8f4a98d3
4 changed files with 89 additions and 27 deletions

10
Cargo.lock generated
View File

@ -1,4 +1,14 @@
[root] [root]
name = "nrgrip" name = "nrgrip"
version = "0.1.0" version = "0.1.0"
dependencies = [
"getopts 0.2.14 (registry+https://github.com/rust-lang/crates.io-index)",
]
[[package]]
name = "getopts"
version = "0.2.14"
source = "registry+https://github.com/rust-lang/crates.io-index"
[metadata]
"checksum getopts 0.2.14 (registry+https://github.com/rust-lang/crates.io-index)" = "d9047cfbd08a437050b363d35ef160452c5fe8ea5187ae0a624708c91581d685"

View File

@ -9,3 +9,4 @@ name = "nrgrip"
doc = false doc = false
[dependencies] [dependencies]
getopts = "0.2"

View File

@ -24,31 +24,52 @@ use std::env;
use std::fs::File; use std::fs::File;
use std::process; use std::process;
extern crate getopts;
use getopts::Options;
extern crate nrgrip; extern crate nrgrip;
fn exit_usage(prog_name: &String) {
println!("Usage:\n\t{} <image.nrg>", prog_name);
process::exit(1);
}
fn main() { fn main() {
let mut argv = env::args(); let args: Vec<String> = env::args().collect();
let prog_name = &args.first().expect("Can't retrieve program's name");
let prog_name = argv.next().unwrap_or("nrgrip".to_string()); let mut opts = Options::new();
opts.optflag("i", "info",
"display the image's metadata (default action)");
opts.optflag("x", "extract",
"same as --extract-cue --extract-raw");
opts.optflag("c", "extract-cue",
"extract cue sheet from the NRG metadata");
opts.optflag("r", "extract-raw",
"extract the raw audio tracks");
opts.optflag("h", "help",
"print this help message");
let img_path = argv.next().unwrap_or("".to_string()); let options = match opts.parse(&args[1..]) {
if img_path == "" { Ok(m) => m,
exit_usage(&prog_name); Err(err) => panic!(err.to_string()),
};
if options.opt_present("help") {
exit_usage(&prog_name, &opts);
} }
// Get input NRG image name
if options.free.len() != 1 {
// We need exactly one input file!
fail_usage(&prog_name, &opts);
}
let img_path = &options.free[0];
println!("NRG image path: \"{}\"", img_path); println!("NRG image path: \"{}\"", img_path);
// We don't support more than one input file // See what actions are to be taken on that file
if argv.next().is_some() { let action_cue =
exit_usage(&prog_name); options.opt_present("extract-cue") || options.opt_present("extract");
} let action_raw =
options.opt_present("extract-raw") || options.opt_present("extract");
let action_info =
options.opt_present("info") || !(action_cue || action_raw);
// Open the image file // Open the image file
let fd = File::open(&img_path); let fd = File::open(&img_path);
@ -59,7 +80,7 @@ fn main() {
} }
let mut fd = fd.unwrap(); let mut fd = fd.unwrap();
// Read and display the image's metadata // Read the image's metadata
let metadata = nrgrip::metadata::read_nrg_metadata(&mut fd); let metadata = nrgrip::metadata::read_nrg_metadata(&mut fd);
if metadata.is_err() { if metadata.is_err() {
println!("Error reading \"{}\": {}", println!("Error reading \"{}\": {}",
@ -67,20 +88,50 @@ fn main() {
process::exit(1); process::exit(1);
} }
let metadata = metadata.unwrap(); let metadata = metadata.unwrap();
println!("\n{}", metadata);
// Display metadata if requested
if action_info {
println!("\n{}", metadata);
}
// Read and write the cue sheet // Read and write the cue sheet
println!("\nNow extracting cue sheet..."); if action_cue {
if let Err(err) = nrgrip::cue_sheet::write_cue_sheet(&img_path, &metadata) { println!("\nExtracting cue sheet...");
println!("Error writing cue sheet: {}", err.to_string()); if let Err(err) = nrgrip::cue_sheet::write_cue_sheet(&img_path,
process::exit(1); &metadata) {
println!("Error writing cue sheet: {}", err.to_string());
process::exit(1);
}
println!("OK!");
} }
// Extract raw audio data // Extract raw audio data
println!("Now extracting raw audio data..."); if action_raw {
if let Err(err) = nrgrip::raw_audio::extract_nrg_raw_audio( println!("\nExtracting raw audio data...");
&mut fd, &img_path, &metadata) { if let Err(err) = nrgrip::raw_audio::extract_nrg_raw_audio(
println!("Error extracting raw audio data: {}", err.to_string()); &mut fd, &img_path, &metadata) {
println!("Error extracting raw audio data: {}", err.to_string());
}
println!("OK!");
} }
println!("OK!"); }
fn exit_usage(prog_name: &str, opts: &Options) {
print_usage(prog_name, opts);
process::exit(0);
}
fn fail_usage(prog_name: &str, opts: &Options) {
print_usage(prog_name, opts);
process::exit(1);
}
fn print_usage(prog_name: &str, opts: &Options) {
let brief = format!("NRGrip - rip Nero Burning ROM audio images
Usage:
{prog} [-icrx] [options] <image.nrg>
{prog} -h", prog = prog_name);
print!("{}", opts.usage(&brief));
} }

View File

@ -105,7 +105,7 @@ impl fmt::Display for NrgMetadata {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
try!(write!(f, "Image size: {} Bytes\n\ try!(write!(f, "Image size: {} Bytes\n\
NRG format version: {}\n\ NRG format version: {}\n\
First chunk offset: {}", First NRG chunk offset: {}",
self.file_size, self.file_size,
self.nrg_version, self.nrg_version,
self.chunk_offset, self.chunk_offset,