aboutsummaryrefslogtreecommitdiff
path: root/src/tw
diff options
context:
space:
mode:
Diffstat (limited to 'src/tw')
-rw-r--r--src/tw/events.rs122
-rw-r--r--src/tw/mod.rs12
2 files changed, 96 insertions, 38 deletions
diff --git a/src/tw/events.rs b/src/tw/events.rs
index afc7fbb..35167a3 100644
--- a/src/tw/events.rs
+++ b/src/tw/events.rs
@@ -13,39 +13,95 @@ pub enum Event {
}
impl Event {
- pub fn from_json(structure: serde_json::Map<String, serde_json::Value>) -> Option<Event> {
- match &structure["event"].as_str().unwrap() {
- &"follow" => Some(Event::Followed {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"unfollow" => Some(Event::Unfollowed {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"favorite" => Some(Event::Fav {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned(),
- twete_id: structure["target_object"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"unfavorite" => Some(Event::Unfav {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned(),
- twete_id: structure["target_object"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"favorited_retweet" => Some(Event::Fav_RT {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned(),
- twete_id: structure["target_object"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"retweeted_retweet" => Some(Event::RT_RT {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned(),
- twete_id: structure["target_object"]["id_str"].as_str().unwrap().to_owned()
- }),
- &"quoted_tweet" => Some(Event::Quoted {
- user_id: structure["source"]["id_str"].as_str().unwrap().to_owned(),
- twete_id: structure["target_object"]["id_str"].as_str().unwrap().to_owned()
- }),
-// &"list_member_added" =>
-// what about removed?
-// &"blocked" => Blocked { },
-// &"unblocked" => Unblocked { },
- e => { println!("unrecognized event: {}", e); None }
+ fn get_source_id(structure: serde_json::Map<String, serde_json::Value>) -> Result<String, String> {
+ match structure.get("source").and_then(|x| x.get("id_str").and_then(|x| x.as_str())) {
+ Some(id) => Ok(id.to_string()),
+ None => Err("No id_str string at .source.id_str".to_string())
+ }
+ }
+ fn get_source_target_ids(structure: serde_json::Map<String, serde_json::Value>) -> Result<(String, String), String> {
+ match (
+ structure.get("source").and_then(|x| x.get("id_str").and_then(|x| x.as_str())),
+ structure.get("target_obj").and_then(|x| x.get("id_str").and_then(|x| x.as_str()))
+ ) {
+ (Some(source_id), Some(target_id)) => Ok((source_id.to_string(), target_id.to_string())),
+ (None, Some(target_id)) => Err("No id_str string at .source.id_str".to_string()),
+ (Some(target_id), None) => Err("No id_str string at .target_object.id_str".to_string()),
+ (None, None) => Err("No id_str at source or target_object".to_string())
+ }
+ }
+ // maybe type error
+ // better? string is ok
+ // for now..
+ pub fn from_json(structure: serde_json::Map<String, serde_json::Value>) -> Result<Event, String> {
+ match structure.get("event").and_then(|x| x.as_str()).map(|x| x.to_owned()) {
+ Some(event) => {
+ let event_ref: &str = &event;
+ match event_ref {
+ "follow" =>
+ Event::get_source_id(structure)
+ .map(|id_str|
+ Event::Followed {
+ user_id: id_str
+ }
+ ),
+ "unfollow" =>
+ Event::get_source_id(structure)
+ .map(|id_str|
+ Event::Unfollowed {
+ user_id: id_str
+ }
+ ),
+ "favorite" =>
+ Event::get_source_target_ids(structure)
+ .map(|(source_id, target_id)|
+ Event::Fav {
+ user_id: source_id,
+ twete_id: target_id
+ }
+ ),
+ "unfavorite" =>
+ Event::get_source_target_ids(structure)
+ .map(|(source_id, target_id)|
+ Event::Unfav {
+ user_id: source_id,
+ twete_id: target_id
+ }
+ ),
+ "favorited_retweet" =>
+ Event::get_source_target_ids(structure)
+ .map(|(source_id, target_id)|
+ Event::Fav_RT {
+ user_id: source_id,
+ twete_id: target_id
+ }
+ ),
+ "retweeted_retweet" =>
+ Event::get_source_target_ids(structure)
+ .map(|(source_id, target_id)|
+ Event::RT_RT {
+ user_id: source_id,
+ twete_id: target_id
+ }
+ ),
+ "quoted_tweet" =>
+ Event::get_source_target_ids(structure)
+ .map(|(source_id, target_id)|
+ Event::Quoted {
+ user_id: source_id,
+ twete_id: target_id
+ }
+ ),
+ // "list_member_added" =>
+ // what about removed?
+ // "blocked" => Blocked { },
+ // "unblocked" => Unblocked { },
+ e => { println!("unrecognized event: {}", e); Err(e.to_string()) }
+ }
+ },
+ None => {
+ Err("No event in event json...".to_string())
+ }
}
}
}
diff --git a/src/tw/mod.rs b/src/tw/mod.rs
index 5f40c5e..d0bad59 100644
--- a/src/tw/mod.rs
+++ b/src/tw/mod.rs
@@ -257,7 +257,6 @@ fn parse_word_command<'a, 'b>(line: &'b str, commands: &[&'a Command]) -> Option
}
} else if line.starts_with(cmd.keyword) {
if line.find(" ").map(|x| x == cmd.keyword.len()).unwrap_or(false) {
- // let inner_twid = u64::from_str(&linestr.split(" ").collect::<Vec<&str>>()[1]).unwrap();
return Some((line.get((cmd.keyword.len() + 1)..).unwrap().trim(), &cmd));
}
}
@@ -641,10 +640,13 @@ fn handle_twitter_event(
tweeter: &mut TwitterCache,
mut queryer: &mut ::Queryer) {
tweeter.cache_api_event(structure.clone(), &mut queryer);
- if let Some(event) = events::Event::from_json(structure) {
- tweeter.display_info.recv(display::Infos::Event(event));
- } else {
- // ought to handle the None case...
+ match events::Event::from_json(structure) {
+ Ok(event) => {
+ tweeter.display_info.recv(display::Infos::Event(event));
+ },
+ Err(e) => {
+ tweeter.display_info.status(format!("Unknown twitter json: {:?}", e));
+ }
}
}