1 files changed, 5 insertions, 141 deletions
diff --git a/src/main.rs b/src/main.rs
index 30534dc..5030857 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,149 +14,13 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-use ical;
-use chrono::{DateTime, TimeZone, Utc};
-use regex::RegexBuilder;
-use std::error::Error;
-use std::fs::File;
-use std::io::BufReader;
-use std::path::{Path, PathBuf};
+mod icaltool;
 
-fn parse_datetime(datetime: Option<String>) -> Option<DateTime<Utc>> {
-    if let Some(mut dt) = datetime {
-        if dt.is_empty() {
-            return None;
-        }
-
-        if !dt.contains('T') {
-            dt += "T000000";
-        }
-        Utc.datetime_from_str(&dt, "%Y%m%dT%H%M%S").ok()
-    }
-    else {
-        None
-    }
-}
-
-fn print_event(event: &ical::parser::ical::component::IcalEvent) {
-    let mut start: Option<DateTime<Utc>> = None;
-    let mut end: Option<DateTime<Utc>> = None;
-    let mut summary = String::new();
-
-    for p in &event.properties {
-        match p.name.as_ref() {
-            "DTSTART" => start = parse_datetime(p.value.clone()),
-            "DTEND"   => end = parse_datetime(p.value.clone()),
-            "SUMMARY" => summary = p.value.clone().unwrap_or_default(),
-            _ => (),
-        }
-    }
-
-    println!("---> {} - {}: {}",
-        start.unwrap_or_else(Utc::now),
-        end.unwrap_or_else(Utc::now),
-        summary);
-}
-
-fn read_icalendar_from_file(filename: &Path) -> std::io::Result<Vec<ical::parser::ical::component::IcalCalendar>> {
-    let buf = BufReader::new(File::open(filename)?);
-    Ok(ical::IcalParser::new(buf)
-        .filter_map(|c| c.ok())
-        .collect())
-}
-
-fn match_event(event: &ical::parser::ical::component::IcalEvent, pattern: &str) -> bool {
-    if pattern == "*" {
-        true
-    }
-    else {
-        let parts: Vec<&str> = pattern.split(':').collect();
-
-        let (key, pat) = if parts.len() == 2 {
-            (parts[0].to_uppercase(), parts[1])
-        } else {
-            (String::from("*"), parts[0])
-        };
-
-        let re = RegexBuilder::new(&pat)
-            .case_insensitive(true)
-            .build()
-            .unwrap();
-
-        event.properties.iter()
-            .filter(|ref p| (&key == "*" || p.name == key) && re.is_match(&p.value.clone().unwrap()))
-            .count() > 0
-    }
-}
-
-#[cfg(test)]
-fn test_match_event_helper(pattern: &str) -> Vec<ical::parser::ical::component::IcalEvent> {
-    let cal = read_icalendar_from_file(&PathBuf::from("test/fixtures/events.ics")).unwrap();
-    cal[0].events.iter()
-        .filter_map(|event| {
-            if match_event(&event, pattern) {
-                Some(event.clone())
-            }
-            else {
-                None
-            }
-        })
-        .collect()
-}
-
-#[test]
-fn match_all_events() {
-    let events:Vec<_> = test_match_event_helper("*");
-    assert_eq!(3, events.len());
-}
-
-#[test]
-fn match_events_by_summary() {
-    let events:Vec<_> = test_match_event_helper("summary:All");
-    assert_eq!(2, events.len());
-}
-
-#[test]
-fn match_events_by_regex() {
-    let events:Vec<_> = test_match_event_helper("summary:^All.+1$");
-    assert_eq!(1, events.len());
-}
-
-#[test]
-fn match_events_regardless_of_case() {
-    let events:Vec<_> = test_match_event_helper("summary:^aLl.+1$");
-    assert_eq!(1, events.len());
-}
-
-struct IcaltoolArgs {
-    filename: PathBuf,
-    pattern: String,
-}
-
-fn parse_args() -> Result<IcaltoolArgs, Box<dyn Error>> {
-    let mut args = std::env::args().skip(1);
-
-    let filename = args.next().ok_or("Missing filename")?.into();
-    let pattern = args.next().ok_or("Missing pattern.")?;
-
-    Ok(IcaltoolArgs{ filename, pattern })
-}
+use icaltool::Icaltool;
 
 fn main() {
-    match parse_args() {
-        Ok(args) => {
-            let calendars = read_icalendar_from_file(&args.filename).unwrap();
-            println!("Found {} calendard in file...", calendars.len());
-
-            for c in calendars {
-                for e in c.events.iter().filter(|e| match_event(&e, &args.pattern)) {
-                    print_event(&e);
-                }
-            }
-        },
-        Err(e) => {
-            println!("[-] Error: {}", e);
-            println!("Usage: icaltool <filename> <patterns>");
-        }
+    if let Err(e) = Icaltool::run() {
+        println!("[-] Error: {}", e);
+        println!("Usage: icaltool <filename> <patterns>");
     }
 }