From 804a678f696881dcd29f076b6be12aa5500d85fa Mon Sep 17 00:00:00 2001
From: ~karx <karx@tilde.team>
Date: Tue, 16 Mar 2021 12:43:19 -0500
Subject: [PATCH] Add error handling to the importer

---
 src/main.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 614c0c6..905ad49 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -152,8 +152,14 @@ impl Program {
 
         if argument_vec.len() > 1 {
             for name in argument_vec[1..].iter() {
-                let kind = name.chars().nth(0).unwrap();
-                let name_to_import = name.chars().nth(1).unwrap();
+                let kind = match name.chars().nth(0) {
+                    Some(content) => content,
+                    None => panic!("SyntaxError: {}: Invalid syntax", arguments)
+                };
+                let name_to_import = match name.chars().nth(1) {
+                    Some(content) => content,
+                    None => panic!("SyntaxError: {}: Invalid syntax", arguments)
+                };
                 if kind == 'v' {
                     let key = prog.vars.get(&name_to_import);
                     match key {