refactor: Optimized config build function

Author: dev-xero

src/lib.rs

@@ -8,13 +8,22 @@ pub struct Config {
 }
 
 impl Config {
-    pub fn build(args: &[String]) -> Result<Config, &'static str> {
-        if args.len() < 3 {
-            return Err("Not enough arguments, expected format: [search_text] [file_path]");
-        }
+    pub fn build<T>(mut args: T) -> Result<Config, &'static str>
+    where
+        T: Iterator<Item = String>
+    {
+        args.next();
+
+        let query = match args.next() {
+            Some(arg) => arg,
+            None => return Err("Didn't get a query string")
+        };
+
+        let file_path = match args.next() {
+            Some(arg) => arg,
+            None => return Err("Didn't get a file path")
+        };
 
-        let query = args[1].clone();
-        let file_path = args[2].clone();
         let ignore_case = env::var("IGNORE_CASE").is_ok();
 
         Ok(Config {

src/main.rs

@@ -2,8 +2,7 @@ use std::{env, process};
 use rust_grep::Config;
 
 fn main() {
-    let args: Vec<String> = env::args().collect();
-    let config = Config::build(&args)
+    let config = Config::build(env::args())
         .unwrap_or_else(|err| {
             eprintln!("Problem parsing arguments: {}", err);
             process::exit(1);