1use crate::fields::{
2 media_fields::MediaFields, place_fields::PlaceFields, poll_fields::PollFields,
3 tweet_fields::TweetFields, user_fields::UserFields,
4};
5use crate::responses::{errors::Errors, includes::Includes, tweets::Tweets};
6use crate::{
7 api::{Authentication, TwapiOptions, apply_options, execute_twitter, make_url},
8 error::Error,
9 headers::Headers,
10};
11use itertools::Itertools;
12use reqwest::RequestBuilder;
13use serde::{Deserialize, Serialize};
14use std::collections::HashSet;
15
16const URL: &str = "/2/tweets";
17
18#[derive(Serialize, Deserialize, Debug, Eq, Hash, PartialEq, Clone, Default)]
19pub enum Expansions {
20 #[serde(rename = "article.cover_media")]
21 #[default]
22 ArticleCoverMedia,
23 #[serde(rename = "article.media_entities")]
24 ArticleMediaEntities,
25 #[serde(rename = "attachments.media_keys")]
26 AttachmentsMediaKeys,
27 #[serde(rename = "attachments.media_source_tweet")]
28 AttachmentsMediaSourceTweet,
29 #[serde(rename = "attachments.poll_ids")]
30 AttachmentsPollIds,
31 #[serde(rename = "author_id")]
32 AuthorId,
33 #[serde(rename = "edit_history_tweet_ids")]
34 EditHistoryTweetIds,
35 #[serde(rename = "entities.mentions.username")]
36 EntitiesMentionsUsername,
37 #[serde(rename = "geo.place_id")]
38 GeoPlaceId,
39 #[serde(rename = "in_reply_to_user_id")]
40 InReplyToUserId,
41 #[serde(rename = "entities.note.mentions.username")]
42 EntitiesNoteMentionsUsername,
43 #[serde(rename = "referenced_tweets.id")]
44 ReferencedTweetsId,
45 #[serde(rename = "referenced_tweets.id.author_id")]
46 ReferencedTweetsIdAuthorId,
47}
48
49impl Expansions {
50 pub fn all() -> HashSet<Self> {
51 let mut result = HashSet::new();
52 result.insert(Self::ArticleCoverMedia);
53 result.insert(Self::ArticleMediaEntities);
54 result.insert(Self::AttachmentsMediaKeys);
55 result.insert(Self::AttachmentsMediaSourceTweet);
56 result.insert(Self::AttachmentsPollIds);
57 result.insert(Self::AuthorId);
58 result.insert(Self::EditHistoryTweetIds);
59 result.insert(Self::EntitiesMentionsUsername);
60 result.insert(Self::GeoPlaceId);
61 result.insert(Self::InReplyToUserId);
62 result.insert(Self::EntitiesNoteMentionsUsername);
63 result.insert(Self::ReferencedTweetsId);
64 result.insert(Self::ReferencedTweetsIdAuthorId);
65 result
66 }
67}
68
69impl std::fmt::Display for Expansions {
70 fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
71 match self {
72 Self::ArticleCoverMedia => write!(f, "article.cover_media"),
73 Self::ArticleMediaEntities => write!(f, "article.media_entities"),
74 Self::AttachmentsMediaKeys => write!(f, "attachments.media_keys"),
75 Self::AttachmentsMediaSourceTweet => write!(f, "attachments.media_source_tweet"),
76 Self::AttachmentsPollIds => write!(f, "attachments.poll_ids"),
77 Self::AuthorId => write!(f, "author_id"),
78 Self::EditHistoryTweetIds => write!(f, "edit_history_tweet_ids"),
79 Self::EntitiesMentionsUsername => write!(f, "entities.mentions.username"),
80 Self::GeoPlaceId => write!(f, "geo.place_id"),
81 Self::InReplyToUserId => write!(f, "in_reply_to_user_id"),
82 Self::EntitiesNoteMentionsUsername => write!(f, "entities.note.mentions.username"),
83 Self::ReferencedTweetsId => write!(f, "referenced_tweets.id"),
84 Self::ReferencedTweetsIdAuthorId => write!(f, "referenced_tweets.id.author_id"),
85 }
86 }
87}
88
89#[derive(Debug, Clone, Default)]
90pub struct Api {
91 ids: String,
92 expansions: Option<HashSet<Expansions>>,
93 media_fields: Option<HashSet<MediaFields>>,
94 place_fields: Option<HashSet<PlaceFields>>,
95 poll_fields: Option<HashSet<PollFields>>,
96 tweet_fields: Option<HashSet<TweetFields>>,
97 user_fields: Option<HashSet<UserFields>>,
98 twapi_options: Option<TwapiOptions>,
99}
100
101impl Api {
102 pub fn new(ids: &str) -> Self {
103 Self {
104 ids: ids.to_owned(),
105 ..Default::default()
106 }
107 }
108
109 pub fn all(ids: &str) -> Self {
110 Self {
111 ids: ids.to_owned(),
112 expansions: Some(Expansions::all()),
113 media_fields: Some(MediaFields::all()),
114 place_fields: Some(PlaceFields::all()),
115 poll_fields: Some(PollFields::all()),
116 tweet_fields: Some(TweetFields::organic()),
117 user_fields: Some(UserFields::all()),
118 ..Default::default()
119 }
120 }
121
122 pub fn open(ids: &str) -> Self {
123 Self {
124 ids: ids.to_owned(),
125 expansions: Some(Expansions::all()),
126 media_fields: Some(MediaFields::open()),
127 place_fields: Some(PlaceFields::all()),
128 poll_fields: Some(PollFields::all()),
129 tweet_fields: Some(TweetFields::open()),
130 user_fields: Some(UserFields::all()),
131 ..Default::default()
132 }
133 }
134
135 pub fn expansions(mut self, value: HashSet<Expansions>) -> Self {
136 self.expansions = Some(value);
137 self
138 }
139
140 pub fn media_fields(mut self, value: HashSet<MediaFields>) -> Self {
141 self.media_fields = Some(value);
142 self
143 }
144
145 pub fn place_fields(mut self, value: HashSet<PlaceFields>) -> Self {
146 self.place_fields = Some(value);
147 self
148 }
149
150 pub fn poll_fields(mut self, value: HashSet<PollFields>) -> Self {
151 self.poll_fields = Some(value);
152 self
153 }
154
155 pub fn tweet_fields(mut self, value: HashSet<TweetFields>) -> Self {
156 self.tweet_fields = Some(value);
157 self
158 }
159
160 pub fn user_fields(mut self, value: HashSet<UserFields>) -> Self {
161 self.user_fields = Some(value);
162 self
163 }
164
165 pub fn twapi_options(mut self, value: TwapiOptions) -> Self {
166 self.twapi_options = Some(value);
167 self
168 }
169
170 pub fn build(self, authentication: &impl Authentication) -> RequestBuilder {
171 let mut query_parameters = vec![];
172 query_parameters.push(("ids", self.ids));
173 if let Some(expansions) = self.expansions {
174 query_parameters.push(("expansions", expansions.iter().join(",")));
175 }
176 if let Some(media_fields) = self.media_fields {
177 query_parameters.push(("media.fields", media_fields.iter().join(",")));
178 }
179 if let Some(place_fields) = self.place_fields {
180 query_parameters.push(("place.fields", place_fields.iter().join(",")));
181 }
182 if let Some(poll_fields) = self.poll_fields {
183 query_parameters.push(("poll.fields", poll_fields.iter().join(",")));
184 }
185 if let Some(tweet_fields) = self.tweet_fields {
186 query_parameters.push(("tweet.fields", tweet_fields.iter().join(",")));
187 }
188 if let Some(user_fields) = self.user_fields {
189 query_parameters.push(("user.fields", user_fields.iter().join(",")));
190 }
191 let client = reqwest::Client::new();
192 let url = make_url(&self.twapi_options, URL);
193 let builder = client.get(&url).query(&query_parameters);
194 authentication.execute(
195 apply_options(builder, &self.twapi_options),
196 "GET",
197 &url,
198 &query_parameters
199 .iter()
200 .map(|it| (it.0, it.1.as_str()))
201 .collect::<Vec<_>>(),
202 )
203 }
204
205 pub async fn execute(
206 self,
207 authentication: &impl Authentication,
208 ) -> Result<(Response, Headers), Error> {
209 execute_twitter(self.build(authentication)).await
210 }
211}
212
213#[derive(Serialize, Deserialize, Debug, Clone, Default, PartialEq)]
214pub struct Response {
215 #[serde(skip_serializing_if = "Option::is_none")]
216 pub data: Option<Vec<Tweets>>,
217 #[serde(skip_serializing_if = "Option::is_none")]
218 pub errors: Option<Vec<Errors>>,
219 #[serde(skip_serializing_if = "Option::is_none")]
220 pub includes: Option<Includes>,
221 #[serde(flatten)]
222 pub extra: std::collections::HashMap<String, serde_json::Value>,
223}
224
225impl Response {
226 pub fn is_empty_extra(&self) -> bool {
227 let res = self.extra.is_empty()
228 && self
229 .data
230 .as_ref()
231 .map(|it| it.iter().all(|item| item.is_empty_extra()))
232 .unwrap_or(true)
233 && self
234 .errors
235 .as_ref()
236 .map(|it| it.iter().all(|item| item.is_empty_extra()))
237 .unwrap_or(true)
238 && self
239 .includes
240 .as_ref()
241 .map(|it| it.is_empty_extra())
242 .unwrap_or(true);
243 if !res {
244 println!("Response {:?}", self.extra);
245 }
246 res
247 }
248}