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