Add basic frequency counter binary

1 files changed, 53 insertions, 0 deletions

diff --git a/src/freq.rs b/src/freq.rs
new file mode 100644
index 0000000..21c4e62
--- /dev/null
+++ b/src/freq.rs
@@ -0,0 +1,53 @@
+extern crate ilc;
+
+use std::io;
+use std::collections::hash_map::*;
+
+use ilc::log::Event::*;
+use ilc::format::{ self, Decode };
+
+struct Person {
+    lines: u32,
+    words: u32
+}
+
+fn words(s: &str) -> u32 {
+    s.split_whitespace().filter(|s| !s.is_empty()).count() as u32
+}
+
+fn main() {
+    let stdin = io::stdin();
+
+    let mut stats: HashMap<String, Person> = HashMap::new();
+
+    let mut parser = format::weechat3::Weechat3;
+    for e in parser.decode(stdin.lock()) {
+        let m = match e {
+            Ok(m) => m,
+            Err(err) => panic!(err)
+        };
+
+        match m {
+            Msg { ref from, ref content, .. } => {
+                if stats.contains_key(from) {
+                    let p: &mut Person = stats.get_mut(from).unwrap();
+                    p.lines += 1;
+                    p.words += words(content);
+                } else {
+                    stats.insert(from.clone(), Person {
+                        lines: 1,
+                        words: words(content)
+                    });
+                }
+            },
+            _ => ()
+        }
+    }
+
+    let mut stats: Vec<(String, Person)> = stats.into_iter().collect();
+    stats.sort_by(|&(_, ref a), &(_, ref b)| b.words.cmp(&a.words));
+
+    for &(ref name, ref stat) in stats.iter().take(10) {
+        println!("{}:\n\tLines: {}\n\tWords: {}", name, stat.lines, stat.words)
+    }
+}