aboutsummaryrefslogtreecommitdiff
path: root/src/tw/tweet.rs
blob: 2ca32c3691b73eb15c2c06eea8ced3b6019ee7b3 (plain)
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())
    }
}