diff options
-rw-r--r-- | base/src/event.rs | 4 | ||||
-rw-r--r-- | ops/src/convert.rs | 77 |
2 files changed, 79 insertions, 2 deletions
diff --git a/base/src/event.rs b/base/src/event.rs index 3176a9a..5d57023 100644 --- a/base/src/event.rs +++ b/base/src/event.rs @@ -210,8 +210,8 @@ impl<'a> Type<'a> { &TopicChange { .. } => "topic_change", &Kick { .. } => "kick", &Mode { .. } => "mode", - &Connect { .. } => "connect", - &Disconnect { .. } => "disconnect", + &Connect => "connect", + &Disconnect => "disconnect", } } diff --git a/ops/src/convert.rs b/ops/src/convert.rs new file mode 100644 index 0000000..15b4bb2 --- /dev/null +++ b/ops/src/convert.rs @@ -0,0 +1,77 @@ +//! Log format conversion +use regex::Regex; + +use ilc_base::{self, Context, Decode, Encode, Event}; +use std::io::{BufRead, Write}; + +#[derive(Copy, Clone)] +pub enum Subject { + Nick, + Time, + Type, + Text, +} + +pub enum Operator { + Exactly(String), + Contains(String), + Matches(Regex), + Equal(i64), + Greater(i64), + Less(i64), +} + +pub struct Filter(pub Subject, pub Operator); + +impl Filter { + pub fn satisfied_by(&self, e: &Event) -> bool { + use self::Subject::*; + use self::Operator::*; + + match (self.0, &self.1) { + (Nick, &Exactly(ref s)) => e.ty.actor().map_or(false, |e| e == s), + (Nick, &Contains(ref s)) => e.ty.actor().map_or(false, |e| e.contains(s)), + (Nick, &Matches(ref r)) => e.ty.actor().map_or(false, |e| r.is_match(e)), + (Nick, _op) => false, + (Time, &Equal(t)) => e.time.as_timestamp() == t, + (Time, &Greater(t)) => e.time.as_timestamp() > t, + (Time, &Less(t)) => e.time.as_timestamp() < t, + (Time, _op) => false, + (Type, &Exactly(ref s)) => e.ty.type_desc() == s, + (Type, &Contains(ref s)) => e.ty.type_desc().contains(s), + (Type, &Matches(ref r)) => r.is_match(e.ty.type_desc()), + (Type, _op) => false, + (Text, &Exactly(ref s)) => e.ty.text().map_or(false, |t| t == s), + (Text, &Contains(ref s)) => e.ty.text().map_or(false, |t| t.contains(s)), + (Text, &Matches(ref r)) => e.ty.text().map_or(false, |t| r.is_match(t)), + (Text, _op) => false, + } + } +} + +/// Convert from one format to another, not necessarily different, format. In combination with a +/// timezone offset, this can be used to correct the timestamps. +/// Will return `Err` and abort conversion if the decoder yields `Err` or re-encoding fails. +pub fn convert(ctx: &Context, + input: &mut BufRead, + decoder: &mut Decode, + output: &mut Write, + encoder: &Encode, + filter: Option<Filter>, + not: bool) + -> ilc_base::Result<()> { + if let Some(f) = filter { + for e in decoder.decode(&ctx, input) { + let e = try!(e); + if not ^ f.satisfied_by(&e) { + try!(encoder.encode(&ctx, output, &e)) + } + } + } else { + // fast path for filter-less conversion, probably premature + for e in decoder.decode(&ctx, input) { + try!(encoder.encode(&ctx, output, &try!(e))); + } + } + Ok(()) +} |