From 4724f9a8f57cfcce89097a2744fbdcb6e1849dc4 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Fri, 8 May 2020 08:25:56 +0800 Subject: [PATCH] ref main.rs --- src/main.rs | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/src/main.rs b/src/main.rs index 90f1827..09c79e1 100644 --- a/src/main.rs +++ b/src/main.rs @@ -18,13 +18,13 @@ use vm::*; // https://p.rogram.me/grassmudhorse.js/grassmudhorse.js fn main() { let is_help = env::args().any(|a| a == "-h" || a == "--help"); + let is_debug = env::args().any(|a| a == "-d" || a == "--debug"); + let arg_file = env::args().filter(|a| !a.starts_with("-")).nth(1); + if is_help { println!("grassmudhorse.rs v0.1"); return; } - let is_debug = env::args().any(|a| a == "-d" || a == "--debug"); - - let arg_file = env::args().filter(|a| !a.starts_with("-")).nth(1); let input = match arg_file { Some(f) => match read_to_string(&f) { @@ -37,16 +37,13 @@ fn main() { let mut buff = String::new(); use std::io::Read; match std::io::stdin().lock().read_to_string(&mut buff) { - Ok(_) => buff, - Err(err) => { + Ok(_) => buff, Err(err) => { println!("[ERROR] Read stdin, error: {}", err); return; } } }, }; - - let instructions = match parse_lang(&input) { Ok(i) => i, Err(err) => {