Merge pull request #1 from tilpner/cows

Update Event API with Option and Cow

1 files changed, 150 insertions, 0 deletions

diff --git a/src/format/energymech.rs b/src/format/energymech.rs
new file mode 100644
index 0000000..e6f7d80
--- /dev/null
+++ b/src/format/energymech.rs
@@ -0,0 +1,150 @@
+// Copyright 2015 Till Höppner
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+use std::io::{ BufRead, Write };
+use std::borrow::{ ToOwned };
+use std::iter::{ Iterator };
+
+use event::{ Event, Type, Time };
+use context::Context;
+use format::{ Encode, Decode, rejoin, strip_one };
+
+use l::LogLevel::Info;
+
+use chrono::*;
+
+pub struct Energymech;
+
+static TIME_FORMAT: &'static str = "%H:%M:%S";
+
+pub struct Iter<'a, R: 'a> where R: BufRead {
+    context: &'a Context,
+    input: R,
+    buffer: String
+}
+
+impl<'a, R: 'a> Iterator for Iter<'a, R> where R: BufRead {
+    type Item = ::Result<Event<'a>>;
+    fn next(&mut self) -> Option<::Result<Event<'a>>> {
+        fn parse_time(context: &Context, time: &str) -> Time {
+            Time::Timestamp(context.timezone.from_local_date(&context.override_date)
+                .and_time(NaiveTime::from_hms(time[1..3].parse::<u32>().unwrap(),
+                                              time[4..6].parse::<u32>().unwrap(),
+                                              time[7..9].parse::<u32>().unwrap()))
+                .single()
+                .expect("Transformed log times can't be represented, due to timezone transitions")
+                .timestamp())
+        }
+
+        loop {
+            self.buffer.clear();
+            match self.input.read_line(&mut self.buffer) {
+                Ok(0) | Err(_) => return None,
+                Ok(_) => ()
+            }
+
+            let mut split_tokens: Vec<char> = Vec::new();
+            let tokens = self.buffer.split( |c: char| {
+                if c.is_whitespace() { split_tokens.push(c); true } else { false }
+            }).collect::<Vec<_>>();
+            if log_enabled!(Info) {
+                info!("Original:  `{}`", self.buffer);
+                info!("Parsing:   {:?}", tokens);
+            }
+            match &tokens[..tokens.len() - 1] {
+                [time, "*", nick, content..] => return Some(Ok(Event {
+                    ty: Type::Action {
+                        from: nick.to_owned().into(),
+                        content: rejoin(content, &split_tokens[3..])
+                    },
+                    time: parse_time(&self.context, time),
+                    channel: None
+                })),
+                [time, "***", old, "is", "now", "known", "as", new] => return Some(Ok(Event {
+                    ty: Type::Nick {
+                        old_nick: old.to_owned().into(),
+                        new_nick: new.to_owned().into()
+                    },
+                    time: parse_time(&self.context, time),
+                    channel: None
+                })),
+                [time, "***", "Joins:", nick, host] => return Some(Ok(Event {
+                    ty: Type::Join {
+                        nick: nick.to_owned().into(),
+                        mask: Some(strip_one(host).into())
+                    },
+                    time: parse_time(&self.context, time),
+                    channel: None
+                })),
+                [time, "***", "Quits:", nick, host, reason..] => return Some(Ok(Event {
+                    ty: Type::Quit {
+                        nick: nick.to_owned().into(),
+                        mask: Some(strip_one(host).into()),
+                        reason: Some(strip_one(&rejoin(reason, &split_tokens[5..])).into())
+                    },
+                    time: parse_time(&self.context, time),
+                    channel: None
+                })),
+                [time, nick, content..]
+                    if nick.starts_with('<') && nick.ends_with('>')
+                    => return Some(Ok(Event {
+                    ty: Type::Msg {
+                        from: strip_one(nick).into(),
+                        content: rejoin(content, &split_tokens[2..])
+                    },
+                    time: parse_time(&self.context, time),
+                    channel: None
+                })),
+                _ => ()
+            }
+        }
+    }
+}
+
+impl<'a, R: 'a> Decode<'a, R, Iter<'a, R>> for Energymech where R: BufRead {
+    fn decode(&'a mut self, context: &'a Context, input: R) -> Iter<R> {
+        Iter {
+            context: context,
+            input: input,
+            buffer: String::new()
+        }
+    }
+}
+
+impl<'a, W> Encode<'a, W> for Energymech where W: Write {
+    fn encode(&'a self, context: &'a Context, mut output: W, event: &'a Event) -> ::Result<()> {
+        match event {
+            &Event { ty: Type::Msg { ref from, ref content }, ref time, .. } => {
+                try!(writeln!(&mut output, "[{}] <{}> {}",
+                    time.with_format(&context.timezone, TIME_FORMAT), from, content))
+            },
+            &Event { ty: Type::Action { ref from, ref content }, ref time, .. } => {
+                try!(writeln!(&mut output, "[{}] * {} {}",
+                    time.with_format(&context.timezone, TIME_FORMAT), from, content))
+            },
+            &Event { ty: Type::Nick { ref old_nick, ref new_nick }, ref time, .. } => {
+                try!(writeln!(&mut output, "[{}] *** {} is now known as {}",
+                    time.with_format(&context.timezone, TIME_FORMAT), old_nick, new_nick))
+            },
+            &Event { ty: Type::Quit { ref nick, ref mask, ref reason }, ref time, .. } => {
+                try!(writeln!(&mut output, "[{}] *** Quits: {} ({}) ({})",
+                    time.with_format(&context.timezone, TIME_FORMAT), nick,
+                    mask.as_ref().expect("Mask not present, but required."),
+                    reason.as_ref().expect("Reason not present, but required.")))
+            },
+            _ => ()
+        }
+        Ok(())
+    }
+}