43 lines
1.2 KiB
Rust
43 lines
1.2 KiB
Rust
use saturn_patch::{SaturnDisc, DESIRED_SATURN_DISC};
|
|
use std::env;
|
|
use std::ffi::OsString;
|
|
|
|
use anyhow::Result;
|
|
|
|
fn main() {
|
|
let mut args = env::args_os().peekable();
|
|
|
|
// let's go by name of executable to decide what operation to run
|
|
let mut patch = args.next().map_or(true, |exe| !exe.to_string_lossy().to_ascii_lowercase().contains("unpatch"));
|
|
|
|
// but also support a single arg in the first position, if people wish
|
|
if args.peek().map_or(false, |arg| arg.eq("-u") || arg.eq("--unpatch")) {
|
|
args.next(); // just skip the next one
|
|
patch = false;
|
|
}
|
|
|
|
let saturn_disc = if patch {
|
|
match SaturnDisc::from_env_args() {
|
|
Ok(sat_p) => sat_p,
|
|
Err(err) => {
|
|
eprintln!("ERROR: {}", err);
|
|
std::process::exit(2);
|
|
}
|
|
}
|
|
} else {
|
|
DESIRED_SATURN_DISC
|
|
};
|
|
|
|
let patch: Box<dyn Fn(&OsString) -> Result<()>> = if patch { Box::new(|f| saturn_disc.patch(f)) } else { Box::new(|f| SaturnDisc::unpatch(f)) };
|
|
|
|
let mut exit_code = 0;
|
|
|
|
for file_name in args {
|
|
if let Err(err) = patch(&file_name) {
|
|
eprintln!("ERROR: {}", err);
|
|
exit_code = 1;
|
|
}
|
|
}
|
|
std::process::exit(exit_code);
|
|
}
|