From 319a7108bfb1e05ef600529516f575e6802896b5 Mon Sep 17 00:00:00 2001
From: Raamakrishnan <raamakrishnan1997@gmail.com>
Date: Wed, 22 Jan 2020 22:46:56 +0530
Subject: [PATCH] Add data to LineType. Add parsing for filelist and defines

---
 src/lib.rs | 35 +++++++++++++++++++++++++----------
 1 file changed, 25 insertions(+), 10 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 50d5f8e..c288fa4 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,16 +1,27 @@
+use std::collections::HashMap;
+
 #[derive(PartialEq, Debug)]
-pub enum LineType {
-    File,
-    Define,
-    Filelist,
+enum LineType <'a> {
+    File(&'a str),
+    IncDir(Vec<&'a str>),
+    Define(HashMap<&'a str, &'a str>),
+    Filelist(&'a str),
     Unknown,
 }
 
-pub fn parse_line(line: &str) -> LineType {
+fn parse_line<'a>(line: &'a str) -> LineType<'a> {
+    let line = line.trim();
     if line.starts_with("-f ") {
-        return LineType::Filelist;
+        let filelist_name = line.trim_start_matches("-f ");
+        return LineType::Filelist(filelist_name);
     } else if line.starts_with("+define+") {
-        return LineType::Define;
+        let defines = line.trim_start_matches("+define+");
+        let mut define_map = HashMap::new();
+        for define in defines.split("+") {
+            let split: Vec<&str> = define.splitn(2, "=").collect();
+            define_map.insert(split[0], split[1]);
+        }
+        return LineType::Define(define_map);
     } else {
         return LineType::Unknown;
     }
@@ -23,12 +34,16 @@ mod test {
     #[test]
     fn parse_line_filelist() {
         let line = "-f sample/files.f";
-        assert_eq!(parse_line(line), LineType::Filelist);
+        assert_eq!(parse_line(line), LineType::Filelist("sample/files.f"));
     }
 
     #[test]
     fn parse_line_define() {
-        let line = "+define+CONST1=const1+CONST2=const2";
-        assert_eq!(parse_line(line), LineType::Define);
+        let line = "+define+CONST1=const1+CONST2=const2+CONST3=const3=1";
+        let mut define_map = HashMap::new();
+        define_map.insert("CONST1", "const1");
+        define_map.insert("CONST2", "const2");
+        define_map.insert("CONST3", "const3=1");
+        assert_eq!(parse_line(line), LineType::Define(define_map));
     }
 }
\ No newline at end of file