1 files changed, 17 insertions, 4 deletions

diff --git a/src/main.rs b/src/main.rs
index 4cca6d1..eaec1f4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,18 +1,31 @@
 extern crate telegram_bot;
+use telegram_bot::{Api, ListeningMethod, ListeningAction, Update};
 
-use telegram_bot::{Api, ListeningMethod, ListeningAction, MessageType, Update};
+mod feature;
+use feature::{Feature, FeatureResult};
 
 fn main() {
     let api = Api::from_env("TELEGRAM_BOT_TOKEN").unwrap();
 
     println!("api started: {:?}", api.get_me());
+    let mut features = feature::init();
 
     let mut listener = api.listener(ListeningMethod::LongPoll(None));
     let res = listener.listen(|u| {
         if let Some(m) = u.message {
-            let name = m.from.first_name;
-
-            println!("got message from {}", name);
+            for f in &mut features {
+                match (**f).handle(api.clone(), m.clone()) {
+                    Ok(r) => {
+                        match r {
+                            FeatureResult::Handled => break,
+                            FeatureResult::Skip => continue,
+                        };
+                    }
+                    Err(e) => {
+                        println!("Error in feature {}: {}", f.name(), e);
+                    }
+                };
+            }
         };
         Ok(ListeningAction::Continue)
     });