Added splitting at newline

hw9/task1/src/lib.rs

@@ -1,34 +1,40 @@
 pub fn parse(message: &str) -> Command {
     let m : String = String::from(message);
-    let mut s = m.split_whitespace();
-    match s.next() {
-        Some("STAGE") => {
-            let msg = m[6..].trim_left();
+    let mut line = m.lines();
+    match line.next() {
+        Some(x) => {
+            let mut str = x.split_whitespace();
+            match str.next() {
+                Some("STAGE") => {
+                    let msg = m[6..].trim_left();
 
-            /* let mut msg = String::new();
-            while let Some(sub_msg) = s.next() {
-                msg += " ";
-                msg += sub_msg
-            }*/
-            Command::Stage(msg)
-        }
-        Some("CONTROL") => {
-            let cmd = m[8..].trim_left();
+                    /* let mut msg = String::new();
+                    while let Some(sub_msg) = s.next() {
+                        msg += " ";
+                        msg += sub_msg
+                    }*/
+                    Command::Stage(msg.to_string())
+                }
+                Some("CONTROL") => {
+                    let cmd = m[8..].trim_left();
 
-            /*let mut cmd = String::new();
-            while let Some(sub_cmd) = s.next() {
-                cmd += " ";
-                cmd += sub_cmd
-            }*/
-            Command::Control(&cmd)
-        }
-        Some("RETRIEVE") => {
-            Command::Retrieve
-        }
-        Some(_) => {
-            Command::Error
+                    /*let mut cmd = String::new();
+                    while let Some(sub_cmd) = s.next() {
+                        cmd += " ";
+                        cmd += sub_cmd
+                    }*/
+                    Command::Control(cmd.to_string())
+                }
+                Some("RETRIEVE") => {
+                    Command::Retrieve
+                }
+                Some(_) => {
+                    Command::Error
+                }
+                None => panic!("Parse Error")
+            }
         }
-        None => panic!("Parse Error")
+        None => panic!("Parse Error!")
     }
 }