1 files changed, 23 insertions, 4 deletions

diff --git a/src/pre_process.rs b/src/pre_process.rs
index d69cfce..0d8c6be 100644
--- a/src/pre_process.rs
+++ b/src/pre_process.rs
@@ -1,5 +1,7 @@
 use super::regex::Regex;
 
+use std::collections::HashMap;
+
 pub fn strip_irrelevant_content(s: &str) -> String {
     let mut retn = "";
     match s.find(r#"<div class=profile id=profile>"#) {
@@ -17,14 +19,31 @@ pub fn strip_irrelevant_content(s: &str) -> String {
 pub struct Section {
     pub name: String,
     pub re: Regex,
-    pub content: String,
+    pub keys: Vec<String>,
+    pub data: HashMap<String, String>,
+}
+
+impl Section {
+        pub fn new(name: &str, re: &str, groups: Vec<String>) -> Self {
+                Section {
+                        name: name.into(),
+                        re: Regex::new(re).unwrap(),
+                        keys: groups,
+                        data: HashMap::new(),
+                }
+        }
 }
 
 pub fn split_sections(d: &str, s: &mut Vec<Section>) {
     for section in s {
         for m in section.re.captures_iter(d) {
-            assert!(m.len() > 1);
-            section.content = format!("{}", m.at(1).unwrap());
+            assert!(m.len() >= section.keys.len() + 1);
+
+                        let mut idx = 0;
+                        for key in &section.keys {
+                                section.data.insert(key.clone(), m.at(idx + 1).unwrap().into());
+                                idx += 1;
+                        }
         }
     }
-}
\ No newline at end of file
+}