1 files changed, 61 insertions, 0 deletions

diff --git a/src/feature/tasterank.rs b/src/feature/tasterank.rs
new file mode 100644
index 0000000..c4a29f9
--- /dev/null
+++ b/src/feature/tasterank.rs
@@ -0,0 +1,61 @@
+use std::option::Option;
+use std::io;
+use std::io::prelude::*;
+use std::fs::File;
+
+use std::ascii::AsciiExt;
+
+extern crate telegram_bot;
+use telegram_bot::{Api, Message, MessageType};
+
+use feature::FeatureResult;
+use feature::Feature;
+
+pub struct Tasterank {
+    ranks: Vec<String>,
+}
+
+impl Feature for Tasterank {
+    fn name(&self) -> &'static str {
+        "Tasterank"
+    }
+    fn init(&mut self) {
+        if let Err(e) = self.update_ranks() {
+            panic!(e);
+        }
+    }
+    fn handle(&mut self, a: Api, m: Message) -> Result<FeatureResult, String> {
+        if let MessageType::Text(s) = m.msg {
+            if !(&s).to_ascii_lowercase().contains("luggas tasterank") {
+                return Ok(FeatureResult::Skip);
+            }
+            let mut msg = String::new();
+            for (i, rank) in self.ranks.iter().enumerate() {
+                msg.push_str(&format!("{}. {}\n", i + 1, rank));
+            }
+            a.send_message(m.chat.id(), msg, None, None, Some(m.message_id), None);
+            return Ok(FeatureResult::Handled);
+        }
+        Ok(FeatureResult::Skip)
+    }
+}
+
+impl Tasterank {
+    pub fn new() -> Tasterank {
+        let mut t = Tasterank { ranks: vec![] };
+        t.init();
+        t
+    }
+    fn update_ranks(&mut self) -> Result<(), io::Error> {
+        let mut f = try!(File::open("data/tasterank.list"));
+        let mut buf = String::new();
+        try!(f.read_to_string(&mut buf));
+        let temp: Vec<&str> = (&buf).split("\n").collect();
+        self.ranks.clear();
+        for r in &temp {
+            self.ranks.push(r.to_string());
+        }
+        self.ranks.pop().unwrap();
+        Ok(())
+    }
+}