feat: switch from if else to match case for handling args

This commit is contained in:
newbee1905
2023-02-01 22:31:43 +07:00
parent f435f290ac
commit 652548057b

View File

@@ -20,45 +20,41 @@ fn main() {
let mut provider: String = "gogo".to_string(); let mut provider: String = "gogo".to_string();
let mut cast = (false, "0".to_string()); let mut cast = (false, "0".to_string());
for arg in std::env::args() { for arg in std::env::args() {
if arg == "--help" || arg == "-h" { match &*arg {
help = true; "--help" | "-h" => help = true,
} "--anime" | "-a" => anime = true,
if arg == "--anime" || arg == "-a" { "--provider" | "-r" => {
anime = true; if let Some(arg) = std::env::args().nth(count + 1) {
} //get the next argument and see if it is = to gogo of vrv
if arg == "--provider" || arg == "-r" { if arg == "vrv" {
if let Some(arg) = std::env::args().nth(count + 1) { provider = "vrv".to_string();
//get the next argument and see if it is = to gogo of vrv count += 1;
if arg == "vrv" { } else if arg == "gogo" {
provider = "vrv".to_string(); provider = "gogo".to_string();
count += 1; count += 1;
} else if arg == "gogo" { } else {
provider = "gogo".to_string(); provider = "gogo".to_string();
count += 1; }
} else { } else {
provider = "gogo".to_string(); provider = "vrv".to_string();
} }
} else {
provider = "vrv".to_string();
} }
} "--cast" | "-C" => {
if arg == "--cast" || arg == "-C" { if let Some(arg) = std::env::args().nth(count + 1) {
if let Some(arg) = std::env::args().nth(count + 1) { cast = (true, String::from(arg))
cast = (true, String::from(arg)) } else {
} else { println!("{}", "please provide a ip address".red())
println!("{}", "please provide a ip address".red()) }
} }
} "--ln" | "-l" => ln = true,
"--chapter" | "-c" => {
if arg == "--ln" || arg == "-l" { if let Some(arg) = std::env::args().nth(count + 1) {
ln = true; chapter = arg.parse::<u32>().unwrap();
} } else {
if arg == "--chapter" || arg == "-c" { chapter = 0;
if let Some(arg) = std::env::args().nth(count + 1) { }
chapter = arg.parse::<u32>().unwrap();
} else {
chapter = 0;
} }
&_ => {}
} }
count += 1; count += 1;