1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
extern crate serde_json;
use chrono::prelude::*;
use std::collections::HashMap;
use tw::user::User;
#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct Tweet {
pub id: String,
pub author_id: String,
pub text: String,
pub created_at: String, // lol
#[serde(default = "Utc::now")]
pub recieved_at: DateTime<Utc>,
#[serde(skip_serializing_if="HashMap::is_empty")]
#[serde(default = "HashMap::default")]
pub urls: HashMap<String, String>,
#[serde(skip_serializing_if="Option::is_none")]
#[serde(default = "Option::default")]
pub quoted_tweet_id: Option<String>,
#[serde(skip_serializing_if="Option::is_none")]
#[serde(default = "Option::default")]
pub rt_tweet: Option<String>,
#[serde(skip_serializing_if="Option::is_none")]
#[serde(default = "Option::default")]
pub reply_to_tweet: Option<String>,
#[serde(skip)]
pub internal_id: u64
}
impl Tweet {
pub fn get_mentions(&self) -> Vec<String> {
self.text.split(&[
',', '.', '/', ';', '\'',
'[', ']', '\\', '~', '!',
'#', '$', '%', '^',
'&', '*', '(', ')', '-',
'=', '{', '}', '|', ':',
'"', '<', '>', '?', '`',
' ' // forgot this initially. awkward.
][..])
.filter(|x| x.starts_with("@") && x.len() > 1 && x.chars().skip(1).all(|c| c != '@'))
// discard @, mentions are just the usernames.
.map(|handle| handle.chars().skip(1).collect())
.collect()
}
pub fn from_api_json(json: serde_json::Value) -> Result<(Tweet, User), String> {
Tweet::from_json(json.clone()).and_then(|tw| {
match json.get("user") {
Some(user_json) =>
User::from_json(user_json.to_owned()).map(|u| (tw, u)),
None =>
Err("No user json".to_owned())
}
})
}
pub fn from_json(json: serde_json::Value) -> Result<Tweet, String> {
if let serde_json::Value::Object(json_map) = json {
let mut url_map: HashMap<String, String> = HashMap::new();
for entry in json_map["entities"]["urls"].as_array().unwrap() {
url_map.insert(entry["url"].as_str().unwrap().to_owned(), entry["expanded_url"].as_str().unwrap().to_owned());
}
let text = ::tw::full_twete_text(&json_map);
let rt_twete = json_map.get("retweeted_status")
.and_then(|x| x.get("id_str"))
.and_then(|x| x.as_str())
.map(|x| x.to_owned());
let reply_to_tweet = json_map.get("in_reply_to_status_id_str")
.and_then(|x| x.as_str())
.map(|x| x.to_owned());
if json_map.contains_key("id_str") &&
json_map.contains_key("user") &&
json_map.contains_key("created_at") {
if let (
Some(id_str),
Some(author_id),
Some(created_at)
) = (
json_map["id_str"].as_str(),
json_map["user"]["id_str"].as_str(),
json_map["created_at"].as_str()
) {
return Ok(Tweet {
id: id_str.to_owned(),
author_id: author_id.to_owned(),
text: text,
created_at: created_at.to_owned(),
recieved_at: Utc::now(),
urls: url_map,
quoted_tweet_id: json_map.get("quoted_status_id_str")
.and_then(|x| x.as_str())
.map(|x| x.to_owned()),
rt_tweet: rt_twete,
reply_to_tweet: reply_to_tweet,
internal_id: 0
})
}
}
}
Err("Invalid tweet json".to_owned())
}
}
|
