diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/bounding_box.h | 88 | ||||
-rw-r--r-- | src/client.cpp | 1066 | ||||
-rw-r--r-- | src/client.h | 129 | ||||
-rw-r--r-- | src/codes.cpp | 33 | ||||
-rw-r--r-- | src/codes.h | 193 | ||||
-rw-r--r-- | src/configuration.cpp | 101 | ||||
-rw-r--r-- | src/configuration.h | 101 | ||||
-rw-r--r-- | src/direct_message.h | 5 | ||||
-rw-r--r-- | src/list.cpp | 5 | ||||
-rw-r--r-- | src/list.h | 6 | ||||
-rw-r--r-- | src/notification.cpp | 269 | ||||
-rw-r--r-- | src/notification.h | 213 | ||||
-rw-r--r-- | src/stream.cpp | 291 | ||||
-rw-r--r-- | src/stream.h | 67 | ||||
-rw-r--r-- | src/tweet.cpp | 141 | ||||
-rw-r--r-- | src/tweet.h | 77 | ||||
-rw-r--r-- | src/twitter.h | 1 | ||||
-rw-r--r-- | src/user.cpp | 48 | ||||
-rw-r--r-- | src/user.h | 44 | ||||
-rw-r--r-- | src/util.h | 6 |
20 files changed, 1540 insertions, 1344 deletions
diff --git a/src/bounding_box.h b/src/bounding_box.h new file mode 100644 index 0000000..25c7790 --- /dev/null +++ b/src/bounding_box.h | |||
@@ -0,0 +1,88 @@ | |||
1 | #ifndef BOUNDING_BOX_H_75D2077D | ||
2 | #define BOUNDING_BOX_H_75D2077D | ||
3 | |||
4 | namespace twitter { | ||
5 | |||
6 | class coordinate { | ||
7 | public: | ||
8 | |||
9 | coordinate(int degrees, int minutes = 0, int seconds = 0) : | ||
10 | _degrees(degrees), | ||
11 | _minutes(minutes), | ||
12 | _seconds(seconds) | ||
13 | { | ||
14 | } | ||
15 | |||
16 | int getDegrees() const | ||
17 | { | ||
18 | return _degrees; | ||
19 | } | ||
20 | |||
21 | int getMinutes() const | ||
22 | { | ||
23 | return _minutes; | ||
24 | } | ||
25 | |||
26 | int getSeconds() const | ||
27 | { | ||
28 | return _seconds; | ||
29 | } | ||
30 | |||
31 | operator double() const | ||
32 | { | ||
33 | return (double)_degrees + ((double)_minutes / (double)60) + ((double)_seconds / (double)3600); | ||
34 | } | ||
35 | |||
36 | private: | ||
37 | |||
38 | int _degrees; | ||
39 | int _minutes; | ||
40 | int _seconds; | ||
41 | }; | ||
42 | |||
43 | class bounding_box { | ||
44 | public: | ||
45 | |||
46 | bounding_box( | ||
47 | coordinate south_west_long, | ||
48 | coordinate south_west_lat, | ||
49 | coordinate north_east_long, | ||
50 | coordinate north_east_lat) : | ||
51 | _south_west_long(south_west_long), | ||
52 | _south_west_lat(south_west_lat), | ||
53 | _north_east_long(north_east_long), | ||
54 | _north_east_lat(north_east_lat) | ||
55 | { | ||
56 | } | ||
57 | |||
58 | const coordinate& getSouthWestLongitude() const | ||
59 | { | ||
60 | return _south_west_long; | ||
61 | } | ||
62 | |||
63 | const coordinate& getSouthWestLatitude() const | ||
64 | { | ||
65 | return _south_west_lat; | ||
66 | } | ||
67 | |||
68 | const coordinate& getNorthEastLongitude() const | ||
69 | { | ||
70 | return _north_east_long; | ||
71 | } | ||
72 | |||
73 | const coordinate& getNorthEastLatitude() const | ||
74 | { | ||
75 | return _north_east_lat; | ||
76 | } | ||
77 | |||
78 | private: | ||
79 | |||
80 | coordinate _south_west_long; | ||
81 | coordinate _south_west_lat; | ||
82 | coordinate _north_east_long; | ||
83 | coordinate _north_east_lat; | ||
84 | }; | ||
85 | |||
86 | } | ||
87 | |||
88 | #endif /* end of include guard: BOUNDING_BOX_H_75D2077D */ | ||
diff --git a/src/client.cpp b/src/client.cpp index 2c655e2..f8908fd 100644 --- a/src/client.cpp +++ b/src/client.cpp | |||
@@ -5,10 +5,8 @@ | |||
5 | #include <liboauthcpp/liboauthcpp.h> | 5 | #include <liboauthcpp/liboauthcpp.h> |
6 | #include <curl_easy.h> | 6 | #include <curl_easy.h> |
7 | #include <curl_header.h> | 7 | #include <curl_header.h> |
8 | #include "util.h" | ||
9 | #include <json.hpp> | 8 | #include <json.hpp> |
10 | 9 | #include <thread> | |
11 | using nlohmann::json; | ||
12 | 10 | ||
13 | // These are here for debugging curl stuff | 11 | // These are here for debugging curl stuff |
14 | 12 | ||
@@ -84,99 +82,315 @@ int my_trace(CURL *handle, curl_infotype type, | |||
84 | 82 | ||
85 | namespace twitter { | 83 | namespace twitter { |
86 | 84 | ||
87 | int client_stream_progress_callback_wrapper(void* cdata, curl_off_t, curl_off_t, curl_off_t, curl_off_t) | 85 | class request |
88 | { | 86 | { |
89 | return static_cast<client::stream*>(cdata)->progress(); | 87 | public: |
90 | } | 88 | |
89 | explicit request(std::string url) try | ||
90 | : _ios(_output), _conn(_ios) | ||
91 | { | ||
92 | _conn.add<CURLOPT_URL>(url.c_str()); | ||
93 | } catch (const curl::curl_easy_exception& error) | ||
94 | { | ||
95 | error.print_traceback(); | ||
96 | |||
97 | assert(false); | ||
98 | } | ||
99 | |||
100 | std::string perform() | ||
101 | { | ||
102 | try | ||
103 | { | ||
104 | _conn.perform(); | ||
105 | } catch (const curl::curl_easy_exception& error) | ||
106 | { | ||
107 | std::throw_with_nested(connection_error()); | ||
108 | } | ||
109 | |||
110 | int response_code = _conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
111 | std::string result = _output.str(); | ||
112 | |||
113 | if (response_code / 100 != 2) | ||
114 | { | ||
115 | nlohmann::json response_json; | ||
116 | |||
117 | try | ||
118 | { | ||
119 | response_json = nlohmann::json::parse(result); | ||
120 | } catch (const std::invalid_argument& e) | ||
121 | { | ||
122 | std::throw_with_nested(invalid_response(result)); | ||
123 | } | ||
124 | |||
125 | for (nlohmann::json& error : response_json["errors"]) | ||
126 | { | ||
127 | int error_code; | ||
128 | std::string error_message; | ||
129 | |||
130 | try | ||
131 | { | ||
132 | error_code = error["code"].get<int>(); | ||
133 | error_message = error["message"].get<std::string>(); | ||
134 | } catch (const std::domain_error& e) | ||
135 | { | ||
136 | std::throw_with_nested(invalid_response(result)); | ||
137 | } | ||
138 | |||
139 | switch (error_code) | ||
140 | { | ||
141 | case 32: | ||
142 | case 135: | ||
143 | case 215: | ||
144 | throw bad_auth(error_message); | ||
145 | |||
146 | case 44: | ||
147 | throw invalid_media(error_message); | ||
148 | |||
149 | case 64: | ||
150 | throw account_suspended(error_message); | ||
151 | |||
152 | case 88: | ||
153 | throw rate_limit_exceeded(error_message); | ||
154 | |||
155 | case 89: | ||
156 | throw bad_token(error_message); | ||
157 | |||
158 | case 130: | ||
159 | throw server_overloaded(error_message); | ||
160 | |||
161 | case 131: | ||
162 | throw server_error(error_message); | ||
163 | |||
164 | case 185: | ||
165 | throw update_limit_exceeded(error_message); | ||
166 | |||
167 | case 186: | ||
168 | throw bad_length(error_message); | ||
169 | |||
170 | case 187: | ||
171 | throw duplicate_status(error_message); | ||
172 | |||
173 | case 226: | ||
174 | throw suspected_spam(error_message); | ||
175 | |||
176 | case 261: | ||
177 | throw write_restricted(error_message); | ||
178 | } | ||
179 | } | ||
180 | |||
181 | if (response_code == 429) | ||
182 | { | ||
183 | throw rate_limit_exceeded("HTTP 429 Too Many Requests"); | ||
184 | } else if (response_code == 500) | ||
185 | { | ||
186 | throw server_error("HTTP 500 Internal Server Error"); | ||
187 | } else if (response_code == 502) | ||
188 | { | ||
189 | throw server_unavailable("HTTP 502 Bad Gateway"); | ||
190 | } else if (response_code == 503) | ||
191 | { | ||
192 | throw server_overloaded("HTTP 503 Service Unavailable"); | ||
193 | } else if (response_code == 504) | ||
194 | { | ||
195 | throw server_timeout("HTTP 504 Gateway Timeout"); | ||
196 | } | ||
197 | |||
198 | throw unknown_error(response_code, result); | ||
199 | } | ||
200 | |||
201 | return result; | ||
202 | } | ||
203 | |||
204 | private: | ||
205 | |||
206 | std::ostringstream _output; | ||
207 | curl::curl_ios<std::ostringstream> _ios; | ||
208 | |||
209 | protected: | ||
210 | |||
211 | curl::curl_easy _conn; | ||
212 | }; | ||
91 | 213 | ||
92 | size_t client_stream_write_callback_wrapper(void* ptr, size_t size, size_t nmemb, void* cdata) | 214 | class get : public request |
93 | { | 215 | { |
94 | return static_cast<client::stream*>(cdata)->write(static_cast<char*>(ptr), size, nmemb); | 216 | public: |
95 | } | 217 | |
218 | get(const OAuth::Client& oauth_client, std::string url) try | ||
219 | : request(url) | ||
220 | { | ||
221 | std::string oauth_header = oauth_client.getFormattedHttpHeader(OAuth::Http::Get, url, ""); | ||
222 | if (!oauth_header.empty()) | ||
223 | { | ||
224 | _headers.add(std::move(oauth_header)); | ||
225 | } | ||
96 | 226 | ||
97 | client::client(const auth& _arg) | 227 | _conn.add<CURLOPT_HTTPHEADER>(_headers.get()); |
228 | } catch (const OAuth::ParseError& error) | ||
229 | { | ||
230 | std::cout << "Error generating OAuth header:" << std::endl; | ||
231 | std::cout << error.what() << std::endl; | ||
232 | std::cout << "This is likely due to a malformed URL." << std::endl; | ||
233 | |||
234 | assert(false); | ||
235 | } catch (const curl::curl_easy_exception& error) | ||
236 | { | ||
237 | error.print_traceback(); | ||
238 | |||
239 | assert(false); | ||
240 | } | ||
241 | |||
242 | private: | ||
243 | |||
244 | curl::curl_header _headers; | ||
245 | }; | ||
246 | |||
247 | class post : public request | ||
98 | { | 248 | { |
99 | _oauth_consumer = new OAuth::Consumer(_arg.getConsumerKey(), _arg.getConsumerSecret()); | 249 | public: |
100 | _oauth_token = new OAuth::Token(_arg.getAccessKey(), _arg.getAccessSecret()); | 250 | |
101 | _oauth_client = new OAuth::Client(_oauth_consumer, _oauth_token); | 251 | post(const OAuth::Client& oauth_client, std::string url, std::string datastr) try |
102 | 252 | : request(url) | |
103 | std::string url = "https://api.twitter.com/1.1/account/verify_credentials.json"; | ||
104 | long response_code; | ||
105 | std::string response_data; | ||
106 | if (performGet(url, response_code, response_data) && (response_code == 200)) | ||
107 | { | 253 | { |
108 | try { | 254 | std::string oauth_header = oauth_client.getFormattedHttpHeader(OAuth::Http::Post, url, datastr); |
109 | _current_user = user(response_data); | 255 | if (!oauth_header.empty()) |
110 | } catch (std::invalid_argument e) | ||
111 | { | 256 | { |
112 | // Ignore | 257 | _headers.add(std::move(oauth_header)); |
113 | } | 258 | } |
259 | |||
260 | _conn.add<CURLOPT_HTTPHEADER>(_headers.get()); | ||
261 | _conn.add<CURLOPT_COPYPOSTFIELDS>(datastr.c_str()); | ||
262 | } catch (const OAuth::ParseError& error) | ||
263 | { | ||
264 | std::cout << "Error generating OAuth header:" << std::endl; | ||
265 | std::cout << error.what() << std::endl; | ||
266 | std::cout << "This is likely due to a malformed URL." << std::endl; | ||
267 | |||
268 | assert(false); | ||
269 | } catch (const curl::curl_easy_exception& error) | ||
270 | { | ||
271 | error.print_traceback(); | ||
272 | |||
273 | assert(false); | ||
114 | } | 274 | } |
115 | } | 275 | |
276 | private: | ||
277 | |||
278 | curl::curl_header _headers; | ||
279 | }; | ||
280 | |||
281 | class multipost : public request | ||
282 | { | ||
283 | public: | ||
284 | |||
285 | multipost(const OAuth::Client& oauth_client, std::string url, const curl_httppost* fields) try | ||
286 | : request(url) | ||
287 | { | ||
288 | std::string oauth_header = oauth_client.getFormattedHttpHeader(OAuth::Http::Post, url, ""); | ||
289 | if (!oauth_header.empty()) | ||
290 | { | ||
291 | _headers.add(std::move(oauth_header)); | ||
292 | } | ||
293 | |||
294 | _conn.add<CURLOPT_HTTPHEADER>(_headers.get()); | ||
295 | _conn.add<CURLOPT_HTTPPOST>(fields); | ||
296 | } catch (const OAuth::ParseError& error) | ||
297 | { | ||
298 | std::cout << "Error generating OAuth header:" << std::endl; | ||
299 | std::cout << error.what() << std::endl; | ||
300 | std::cout << "This is likely due to a malformed URL." << std::endl; | ||
301 | |||
302 | assert(false); | ||
303 | } catch (const curl::curl_easy_exception& error) | ||
304 | { | ||
305 | error.print_traceback(); | ||
116 | 306 | ||
117 | client::~client() | 307 | assert(false); |
308 | } | ||
309 | |||
310 | private: | ||
311 | |||
312 | curl::curl_header _headers; | ||
313 | }; | ||
314 | |||
315 | client::client(const auth& _arg) | ||
118 | { | 316 | { |
119 | delete _oauth_client; | 317 | _oauth_consumer = |
120 | delete _oauth_token; | 318 | make_unique<OAuth::Consumer>( |
121 | delete _oauth_consumer; | 319 | _arg.getConsumerKey(), |
320 | _arg.getConsumerSecret()); | ||
321 | |||
322 | _oauth_token = | ||
323 | make_unique<OAuth::Token>( | ||
324 | _arg.getAccessKey(), | ||
325 | _arg.getAccessSecret()); | ||
326 | |||
327 | _oauth_client = | ||
328 | make_unique<OAuth::Client>( | ||
329 | _oauth_consumer.get(), | ||
330 | _oauth_token.get()); | ||
331 | |||
332 | _current_user = | ||
333 | make_unique<user>(*this, | ||
334 | get(*_oauth_client, | ||
335 | "https://api.twitter.com/1.1/account/verify_credentials.json") | ||
336 | .perform()); | ||
122 | } | 337 | } |
123 | 338 | ||
124 | response client::updateStatus(std::string msg, tweet& result, tweet in_response_to, std::list<long> media_ids) | 339 | client::~client() = default; |
340 | |||
341 | tweet client::updateStatus(std::string msg, std::list<long> media_ids) const | ||
125 | { | 342 | { |
126 | std::stringstream datastrstream; | 343 | std::stringstream datastrstream; |
127 | datastrstream << "status=" << OAuth::PercentEncode(msg); | 344 | datastrstream << "status=" << OAuth::PercentEncode(msg); |
128 | 345 | ||
129 | if (in_response_to) | ||
130 | { | ||
131 | datastrstream << "&in_reply_to_status_id="; | ||
132 | datastrstream << in_response_to.getID(); | ||
133 | } | ||
134 | |||
135 | if (!media_ids.empty()) | 346 | if (!media_ids.empty()) |
136 | { | 347 | { |
137 | datastrstream << "&media_ids="; | 348 | datastrstream << "&media_ids="; |
138 | datastrstream << twitter::implode(std::begin(media_ids), std::end(media_ids), ","); | 349 | datastrstream << twitter::implode(std::begin(media_ids), std::end(media_ids), ","); |
139 | } | 350 | } |
140 | 351 | ||
141 | std::string datastr = datastrstream.str(); | 352 | return tweet(*this, |
142 | std::string url = "https://api.twitter.com/1.1/statuses/update.json"; | 353 | post(*_oauth_client, |
354 | "https://api.twitter.com/1.1/statuses/update.json", | ||
355 | datastrstream.str()) | ||
356 | .perform()); | ||
357 | } | ||
358 | |||
359 | tweet client::replyToTweet(std::string msg, tweet_id in_response_to, std::list<long> media_ids) const | ||
360 | { | ||
361 | std::stringstream datastrstream; | ||
362 | datastrstream << "status=" << OAuth::PercentEncode(msg); | ||
363 | datastrstream << "&in_reply_to_status_id="; | ||
364 | datastrstream << in_response_to; | ||
143 | 365 | ||
144 | long response_code; | 366 | if (!media_ids.empty()) |
145 | std::string response_data; | ||
146 | if (!performPost(url, datastr, response_code, response_data)) | ||
147 | { | 367 | { |
148 | return response::curl_error; | 368 | datastrstream << "&media_ids="; |
369 | datastrstream << twitter::implode(std::begin(media_ids), std::end(media_ids), ","); | ||
149 | } | 370 | } |
150 | 371 | ||
151 | if (response_code == 200) | 372 | return tweet(*this, |
152 | { | 373 | post(*_oauth_client, |
153 | try { | 374 | "https://api.twitter.com/1.1/statuses/update.json", |
154 | result = tweet(response_data); | 375 | datastrstream.str()) |
155 | return response::ok; | 376 | .perform()); |
156 | } catch (std::invalid_argument e) | ||
157 | { | ||
158 | return response::invalid_response; | ||
159 | } | ||
160 | } else { | ||
161 | return codeForError(response_code, response_data); | ||
162 | } | ||
163 | } | 377 | } |
164 | 378 | ||
165 | response client::uploadMedia(std::string media_type, const char* data, long data_length, long& media_id) | 379 | long client::uploadMedia(std::string media_type, const char* data, long data_length) const try |
166 | { | 380 | { |
167 | curl::curl_form form; | 381 | curl::curl_form form; |
382 | std::string str_data_length = std::to_string(data_length); | ||
168 | 383 | ||
169 | curl::curl_pair<CURLformoption, std::string> command_name(CURLFORM_COPYNAME, "command"); | 384 | curl::curl_pair<CURLformoption, std::string> command_name(CURLFORM_COPYNAME, "command"); |
170 | curl::curl_pair<CURLformoption, std::string> command_cont(CURLFORM_COPYCONTENTS, "INIT"); | 385 | curl::curl_pair<CURLformoption, std::string> command_cont(CURLFORM_COPYCONTENTS, "INIT"); |
171 | curl::curl_pair<CURLformoption, std::string> bytes_name(CURLFORM_COPYNAME, "total_bytes"); | 386 | curl::curl_pair<CURLformoption, std::string> bytes_name(CURLFORM_COPYNAME, "total_bytes"); |
172 | std::string str_data_length = std::to_string(data_length); | ||
173 | curl::curl_pair<CURLformoption, std::string> bytes_cont(CURLFORM_COPYCONTENTS, str_data_length); | 387 | curl::curl_pair<CURLformoption, std::string> bytes_cont(CURLFORM_COPYCONTENTS, str_data_length); |
174 | curl::curl_pair<CURLformoption, std::string> type_name(CURLFORM_COPYNAME, "media_type"); | 388 | curl::curl_pair<CURLformoption, std::string> type_name(CURLFORM_COPYNAME, "media_type"); |
175 | curl::curl_pair<CURLformoption, std::string> type_cont(CURLFORM_COPYCONTENTS, media_type); | 389 | curl::curl_pair<CURLformoption, std::string> type_cont(CURLFORM_COPYCONTENTS, media_type); |
176 | form.add(command_name, command_cont); | 390 | form.add(command_name, command_cont); |
177 | form.add(bytes_name, bytes_cont); | 391 | form.add(bytes_name, bytes_cont); |
178 | form.add(type_name, type_cont); | 392 | form.add(type_name, type_cont); |
179 | 393 | ||
180 | if (media_type == "image/gif") | 394 | if (media_type == "image/gif") |
181 | { | 395 | { |
182 | curl::curl_pair<CURLformoption, std::string> category_name(CURLFORM_COPYNAME, "media_category"); | 396 | curl::curl_pair<CURLformoption, std::string> category_name(CURLFORM_COPYNAME, "media_category"); |
@@ -184,212 +398,109 @@ namespace twitter { | |||
184 | form.add(category_name, category_cont); | 398 | form.add(category_name, category_cont); |
185 | } | 399 | } |
186 | 400 | ||
187 | long response_code; | 401 | std::string init_response = |
188 | std::string response_data; | 402 | multipost(*_oauth_client, |
189 | if (!performMultiPost("https://upload.twitter.com/1.1/media/upload.json", form.get(), response_code, response_data)) | 403 | "https://upload.twitter.com/1.1/media/upload.json", |
190 | { | 404 | form.get()) |
191 | return response::curl_error; | 405 | .perform(); |
192 | } | 406 | |
407 | long media_id; | ||
193 | 408 | ||
194 | if (response_code / 100 != 2) | 409 | try |
195 | { | 410 | { |
196 | return codeForError(response_code, response_data); | 411 | nlohmann::json response_json = nlohmann::json::parse(init_response); |
197 | } | 412 | media_id = response_json["media_id"].get<long>(); |
198 | 413 | } catch (const std::invalid_argument& error) | |
199 | json response_json; | 414 | { |
200 | try { | 415 | std::throw_with_nested(invalid_response(init_response)); |
201 | response_json = json::parse(response_data); | 416 | } catch (const std::domain_error& error) |
202 | } catch (std::invalid_argument e) | ||
203 | { | 417 | { |
204 | return response::invalid_response; | 418 | std::throw_with_nested(invalid_response(init_response)); |
205 | } | 419 | } |
206 | |||
207 | media_id = response_json["media_id"].get<long>(); | ||
208 | 420 | ||
421 | // TODO: Currently have to use the C libcurl API to create this form because it uses a buffer and | ||
422 | // libcurlcpp currently messes that up. | ||
209 | curl_httppost* append_form_post = nullptr; | 423 | curl_httppost* append_form_post = nullptr; |
210 | curl_httppost* append_form_last = nullptr; | 424 | curl_httppost* append_form_last = nullptr; |
211 | curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "command", CURLFORM_COPYCONTENTS, "APPEND", CURLFORM_END); | 425 | if ( curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "command", CURLFORM_COPYCONTENTS, "APPEND", CURLFORM_END) |
212 | curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "media_id", CURLFORM_COPYCONTENTS, std::to_string(media_id).c_str(), CURLFORM_END); | 426 | || curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "media_id", CURLFORM_COPYCONTENTS, std::to_string(media_id).c_str(), CURLFORM_END) |
213 | curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "media", CURLFORM_BUFFER, "media", CURLFORM_BUFFERPTR, data, CURLFORM_BUFFERLENGTH, data_length, CURLFORM_CONTENTTYPE, "application/octet-stream", CURLFORM_END); | 427 | || curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "media", CURLFORM_BUFFER, "media", CURLFORM_BUFFERPTR, data, CURLFORM_BUFFERLENGTH, data_length, CURLFORM_CONTENTTYPE, "application/octet-stream", CURLFORM_END) |
214 | curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "segment_index", CURLFORM_COPYCONTENTS, std::to_string(0).c_str(), CURLFORM_END); | 428 | || curl_formadd(&append_form_post, &append_form_last, CURLFORM_COPYNAME, "segment_index", CURLFORM_COPYCONTENTS, std::to_string(0).c_str(), CURLFORM_END)) |
215 | if (!performMultiPost("https://upload.twitter.com/1.1/media/upload.json", append_form_post, response_code, response_data)) | ||
216 | { | 429 | { |
217 | return response::curl_error; | 430 | assert(false); |
218 | } | 431 | } |
219 | 432 | ||
220 | curl_formfree(append_form_post); | 433 | multipost(*_oauth_client, "https://upload.twitter.com/1.1/media/upload.json", append_form_post).perform(); |
221 | 434 | ||
222 | if (response_code / 100 != 2) | 435 | curl_formfree(append_form_post); |
223 | { | ||
224 | return codeForError(response_code, response_data); | ||
225 | } | ||
226 | 436 | ||
227 | curl::curl_form finalize_form; | 437 | curl::curl_form finalize_form; |
438 | std::string str_media_id = std::to_string(media_id); | ||
439 | |||
228 | curl::curl_pair<CURLformoption, std::string> command3_name(CURLFORM_COPYNAME, "command"); | 440 | curl::curl_pair<CURLformoption, std::string> command3_name(CURLFORM_COPYNAME, "command"); |
229 | curl::curl_pair<CURLformoption, std::string> command3_cont(CURLFORM_COPYCONTENTS, "FINALIZE"); | 441 | curl::curl_pair<CURLformoption, std::string> command3_cont(CURLFORM_COPYCONTENTS, "FINALIZE"); |
230 | curl::curl_pair<CURLformoption, std::string> media_id_name(CURLFORM_COPYNAME, "media_id"); | 442 | curl::curl_pair<CURLformoption, std::string> media_id_name(CURLFORM_COPYNAME, "media_id"); |
231 | std::string str_media_id = std::to_string(media_id); | ||
232 | curl::curl_pair<CURLformoption, std::string> media_id_cont(CURLFORM_COPYCONTENTS, str_media_id); | 443 | curl::curl_pair<CURLformoption, std::string> media_id_cont(CURLFORM_COPYCONTENTS, str_media_id); |
233 | finalize_form.add(command3_name, command3_cont); | 444 | finalize_form.add(command3_name, command3_cont); |
234 | finalize_form.add(media_id_name, media_id_cont); | 445 | finalize_form.add(media_id_name, media_id_cont); |
235 | 446 | ||
236 | if (!performMultiPost("https://upload.twitter.com/1.1/media/upload.json", finalize_form.get(), response_code, response_data)) | 447 | std::string finalize_response = |
237 | { | 448 | multipost(*_oauth_client, |
238 | return response::curl_error; | 449 | "https://upload.twitter.com/1.1/media/upload.json", |
239 | } | 450 | finalize_form.get()) |
451 | .perform(); | ||
452 | |||
453 | nlohmann::json finalize_json; | ||
240 | 454 | ||
241 | if (response_code / 100 != 2) | 455 | try |
242 | { | 456 | { |
243 | return codeForError(response_code, response_data); | 457 | finalize_json = nlohmann::json::parse(finalize_response); |
244 | } | 458 | } catch (const std::invalid_argument& error) |
245 | |||
246 | try { | ||
247 | response_json = json::parse(response_data); | ||
248 | } catch (std::invalid_argument e) | ||
249 | { | 459 | { |
250 | return response::invalid_response; | 460 | std::throw_with_nested(invalid_response(finalize_response)); |
251 | } | 461 | } |
252 | 462 | ||
253 | if (response_json.find("processing_info") != response_json.end()) | 463 | if (finalize_json.find("processing_info") != finalize_json.end()) |
254 | { | 464 | { |
255 | std::stringstream datastr; | 465 | std::stringstream datastr; |
256 | datastr << "https://upload.twitter.com/1.1/media/upload.json?command=STATUS&media_id=" << media_id; | 466 | datastr << "https://upload.twitter.com/1.1/media/upload.json?command=STATUS&media_id=" << media_id; |
257 | 467 | ||
258 | for (;;) | 468 | for (;;) |
259 | { | 469 | { |
260 | if (!performGet(datastr.str(), response_code, response_data)) | 470 | std::string status_response = get(*_oauth_client, datastr.str()).perform(); |
261 | { | ||
262 | return response::curl_error; | ||
263 | } | ||
264 | 471 | ||
265 | if (response_code / 100 != 2) | 472 | try |
266 | { | 473 | { |
267 | return codeForError(response_code, response_data); | 474 | nlohmann::json status_json = nlohmann::json::parse(status_response); |
268 | } | 475 | std::string state = status_json["processing_info"]["state"].get<std::string>(); |
269 | 476 | ||
270 | try { | 477 | if (state == "succeeded") |
271 | response_json = json::parse(response_data); | 478 | { |
272 | } catch (std::invalid_argument e) | 479 | break; |
480 | } | ||
481 | |||
482 | int ttw = status_json["processing_info"]["check_after_secs"].get<int>(); | ||
483 | std::this_thread::sleep_for(std::chrono::seconds(ttw)); | ||
484 | } catch (const std::invalid_argument& error) | ||
273 | { | 485 | { |
274 | return response::invalid_response; | 486 | std::throw_with_nested(invalid_response(status_response)); |
275 | } | 487 | } catch (const std::domain_error& error) |
276 | |||
277 | if (response_json["processing_info"]["state"] == "succeeded") | ||
278 | { | 488 | { |
279 | break; | 489 | std::throw_with_nested(invalid_response(status_response)); |
280 | } | 490 | } |
281 | |||
282 | int ttw = response_json["processing_info"]["check_after_secs"].get<int>(); | ||
283 | std::this_thread::sleep_for(std::chrono::seconds(ttw)); | ||
284 | } | 491 | } |
285 | } | 492 | } |
286 | 493 | ||
287 | return response::ok; | 494 | return media_id; |
288 | } | 495 | } catch (const curl::curl_exception& error) |
289 | |||
290 | response client::follow(user_id toFollow) | ||
291 | { | 496 | { |
292 | std::stringstream datastrstream; | 497 | error.print_traceback(); |
293 | datastrstream << "follow=true&user_id="; | ||
294 | datastrstream << toFollow; | ||
295 | 498 | ||
296 | std::string datastr = datastrstream.str(); | 499 | assert(false); |
297 | std::string url = "https://api.twitter.com/1.1/friendships/create.json"; | ||
298 | |||
299 | long response_code; | ||
300 | std::string response_data; | ||
301 | if (!performPost(url, datastr, response_code, response_data)) | ||
302 | { | ||
303 | return response::curl_error; | ||
304 | } | ||
305 | |||
306 | if (response_code == 200) | ||
307 | { | ||
308 | return response::ok; | ||
309 | } else { | ||
310 | return codeForError(response_code, response_data); | ||
311 | } | ||
312 | } | ||
313 | |||
314 | response client::follow(user toFollow) | ||
315 | { | ||
316 | return follow(toFollow.getID()); | ||
317 | } | ||
318 | |||
319 | response client::unfollow(user_id toUnfollow) | ||
320 | { | ||
321 | std::stringstream datastrstream; | ||
322 | datastrstream << "user_id="; | ||
323 | datastrstream << toUnfollow; | ||
324 | |||
325 | std::string datastr = datastrstream.str(); | ||
326 | std::string url = "https://api.twitter.com/1.1/friendships/destroy.json"; | ||
327 | |||
328 | long response_code; | ||
329 | std::string response_data; | ||
330 | if (!performPost(url, datastr, response_code, response_data)) | ||
331 | { | ||
332 | return response::curl_error; | ||
333 | } | ||
334 | |||
335 | if (response_code == 200) | ||
336 | { | ||
337 | return response::ok; | ||
338 | } else { | ||
339 | return codeForError(response_code, response_data); | ||
340 | } | ||
341 | } | ||
342 | |||
343 | response client::unfollow(user toUnfollow) | ||
344 | { | ||
345 | return unfollow(toUnfollow.getID()); | ||
346 | } | ||
347 | |||
348 | response client::getUser(user& result) | ||
349 | { | ||
350 | if (!_current_user) | ||
351 | { | ||
352 | std::string url = "https://api.twitter.com/1.1/account/verify_credentials.json"; | ||
353 | long response_code; | ||
354 | std::string response_data; | ||
355 | if (performGet(url, response_code, response_data) && (response_code == 200)) | ||
356 | { | ||
357 | try { | ||
358 | _current_user = user(response_data); | ||
359 | } catch (std::invalid_argument e) | ||
360 | { | ||
361 | return response::invalid_response; | ||
362 | } | ||
363 | } | ||
364 | } | ||
365 | |||
366 | result = _current_user; | ||
367 | return response::ok; | ||
368 | } | ||
369 | |||
370 | configuration client::getConfiguration() | ||
371 | { | ||
372 | if (!_configuration || (difftime(time(NULL), _last_configuration_update) > 60*60*24)) | ||
373 | { | ||
374 | long response_code; | ||
375 | std::string response_data; | ||
376 | if (performGet("https://api.twitter.com/1.1/help/configuration.json", response_code, response_data)) | ||
377 | { | ||
378 | _configuration = configuration(response_data); | ||
379 | _last_configuration_update = time(NULL); | ||
380 | } | ||
381 | } | ||
382 | |||
383 | return _configuration; | ||
384 | } | 500 | } |
385 | 501 | ||
386 | response client::getFriends(std::set<user_id>& _ret) | 502 | std::set<user_id> client::getFriends(user_id id) const |
387 | { | 503 | { |
388 | if (!_current_user) | ||
389 | { | ||
390 | return response::unknown_error; | ||
391 | } | ||
392 | |||
393 | long long cursor = -1; | 504 | long long cursor = -1; |
394 | std::set<user_id> result; | 505 | std::set<user_id> result; |
395 | 506 | ||
@@ -397,48 +508,33 @@ namespace twitter { | |||
397 | { | 508 | { |
398 | std::stringstream urlstream; | 509 | std::stringstream urlstream; |
399 | urlstream << "https://api.twitter.com/1.1/friends/ids.json?user_id="; | 510 | urlstream << "https://api.twitter.com/1.1/friends/ids.json?user_id="; |
400 | urlstream << _current_user.getID(); | 511 | urlstream << id; |
401 | urlstream << "&cursor="; | 512 | urlstream << "&cursor="; |
402 | urlstream << cursor; | 513 | urlstream << cursor; |
403 | 514 | ||
404 | std::string url = urlstream.str(); | 515 | std::string url = urlstream.str(); |
405 | 516 | std::string response_data = get(*_oauth_client, url).perform(); | |
406 | long response_code; | 517 | |
407 | std::string response_data; | 518 | try |
408 | if (!performGet(url, response_code, response_data)) | ||
409 | { | ||
410 | return response::curl_error; | ||
411 | } | ||
412 | |||
413 | if (response_code == 200) | ||
414 | { | 519 | { |
415 | json rjs; | 520 | nlohmann::json rjs = nlohmann::json::parse(response_data); |
416 | try { | ||
417 | rjs = json::parse(response_data); | ||
418 | } catch (std::invalid_argument e) | ||
419 | { | ||
420 | return response::invalid_response; | ||
421 | } | ||
422 | 521 | ||
423 | cursor = rjs.at("next_cursor"); | 522 | cursor = rjs["next_cursor"].get<long long>(); |
424 | result.insert(std::begin(rjs.at("ids")), std::end(rjs.at("ids"))); | 523 | result.insert(std::begin(rjs["ids"]), std::end(rjs["ids"])); |
425 | } else { | 524 | } catch (const std::invalid_argument& error) |
426 | return codeForError(response_code, response_data); | 525 | { |
526 | std::throw_with_nested(invalid_response(response_data)); | ||
527 | } catch (const std::domain_error& error) | ||
528 | { | ||
529 | std::throw_with_nested(invalid_response(response_data)); | ||
427 | } | 530 | } |
428 | } | 531 | } |
429 | 532 | ||
430 | _ret = result; | 533 | return result; |
431 | |||
432 | return response::ok; | ||
433 | } | 534 | } |
434 | 535 | ||
435 | response client::getFollowers(std::set<user_id>& _ret) | 536 | std::set<user_id> client::getFollowers(user_id id) const |
436 | { | 537 | { |
437 | if (!_current_user) | ||
438 | { | ||
439 | return response::unknown_error; | ||
440 | } | ||
441 | |||
442 | long long cursor = -1; | 538 | long long cursor = -1; |
443 | std::set<user_id> result; | 539 | std::set<user_id> result; |
444 | 540 | ||
@@ -446,474 +542,68 @@ namespace twitter { | |||
446 | { | 542 | { |
447 | std::stringstream urlstream; | 543 | std::stringstream urlstream; |
448 | urlstream << "https://api.twitter.com/1.1/followers/ids.json?user_id="; | 544 | urlstream << "https://api.twitter.com/1.1/followers/ids.json?user_id="; |
449 | urlstream << _current_user.getID(); | 545 | urlstream << id; |
450 | urlstream << "&cursor="; | 546 | urlstream << "&cursor="; |
451 | urlstream << cursor; | 547 | urlstream << cursor; |
452 | 548 | ||
453 | std::string url = urlstream.str(); | 549 | std::string url = urlstream.str(); |
454 | 550 | std::string response_data = get(*_oauth_client, url).perform(); | |
455 | long response_code; | 551 | |
456 | std::string response_data; | 552 | try |
457 | if (!performGet(url, response_code, response_data)) | ||
458 | { | ||
459 | return response::curl_error; | ||
460 | } | ||
461 | |||
462 | if (response_code == 200) | ||
463 | { | 553 | { |
464 | json rjs; | 554 | nlohmann::json rjs = nlohmann::json::parse(response_data); |
465 | try { | ||
466 | rjs = json::parse(response_data); | ||
467 | } catch (std::invalid_argument e) | ||
468 | { | ||
469 | return response::invalid_response; | ||
470 | } | ||
471 | 555 | ||
472 | cursor = rjs.at("next_cursor"); | 556 | cursor = rjs["next_cursor"].get<long long>(); |
473 | result.insert(std::begin(rjs.at("ids")), std::end(rjs.at("ids"))); | 557 | result.insert(std::begin(rjs["ids"]), std::end(rjs["ids"])); |
474 | } else { | 558 | } catch (const std::invalid_argument& error) |
475 | return codeForError(response_code, response_data); | 559 | { |
476 | } | 560 | std::throw_with_nested(invalid_response(response_data)); |
477 | } | 561 | } catch (const std::domain_error& error) |
478 | |||
479 | _ret = result; | ||
480 | |||
481 | return response::ok; | ||
482 | } | ||
483 | |||
484 | void client::setUserStreamNotifyCallback(stream::notify_callback callback) | ||
485 | { | ||
486 | _user_stream.setNotifyCallback(callback); | ||
487 | } | ||
488 | |||
489 | void client::setUserStreamReceiveAllReplies(bool _arg) | ||
490 | { | ||
491 | _user_stream.setReceiveAllReplies(_arg); | ||
492 | } | ||
493 | |||
494 | void client::startUserStream() | ||
495 | { | ||
496 | _user_stream.start(); | ||
497 | } | ||
498 | |||
499 | void client::stopUserStream() | ||
500 | { | ||
501 | _user_stream.stop(); | ||
502 | } | ||
503 | |||
504 | std::string client::generateReplyPrefill(tweet _tweet) const | ||
505 | { | ||
506 | std::ostringstream output; | ||
507 | output << "@" << _tweet.getAuthor().getScreenName() << " "; | ||
508 | |||
509 | for (auto mention : _tweet.getMentions()) | ||
510 | { | ||
511 | if ((mention.first != _tweet.getAuthor().getID()) && (mention.first != _current_user.getID())) | ||
512 | { | 562 | { |
513 | output << "@" << mention.second << " "; | 563 | std::throw_with_nested(invalid_response(response_data)); |
514 | } | 564 | } |
515 | } | 565 | } |
516 | 566 | ||
517 | return output.str(); | 567 | return result; |
518 | } | ||
519 | |||
520 | bool client::performGet(std::string url, long& response_code, std::string& result) | ||
521 | { | ||
522 | std::ostringstream output; | ||
523 | curl::curl_ios<std::ostringstream> ios(output); | ||
524 | curl::curl_easy conn(ios); | ||
525 | |||
526 | curl::curl_header headers; | ||
527 | std::string oauth_header = _oauth_client->getFormattedHttpHeader(OAuth::Http::Get, url, ""); | ||
528 | if (!oauth_header.empty()) | ||
529 | { | ||
530 | headers.add(oauth_header); | ||
531 | } | ||
532 | |||
533 | try { | ||
534 | //conn.add<CURLOPT_VERBOSE>(1); | ||
535 | //conn.add<CURLOPT_DEBUGFUNCTION>(my_trace); | ||
536 | conn.add<CURLOPT_URL>(url.c_str()); | ||
537 | conn.add<CURLOPT_HTTPHEADER>(headers.get()); | ||
538 | |||
539 | conn.perform(); | ||
540 | } catch (curl::curl_easy_exception error) | ||
541 | { | ||
542 | error.print_traceback(); | ||
543 | |||
544 | return false; | ||
545 | } | ||
546 | |||
547 | response_code = conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
548 | result = output.str(); | ||
549 | |||
550 | return true; | ||
551 | } | ||
552 | |||
553 | bool client::performPost(std::string url, std::string datastr, long& response_code, std::string& result) | ||
554 | { | ||
555 | std::ostringstream output; | ||
556 | curl::curl_ios<std::ostringstream> ios(output); | ||
557 | curl::curl_easy conn(ios); | ||
558 | |||
559 | curl::curl_header headers; | ||
560 | std::string oauth_header = _oauth_client->getFormattedHttpHeader(OAuth::Http::Post, url, datastr); | ||
561 | if (!oauth_header.empty()) | ||
562 | { | ||
563 | headers.add(oauth_header); | ||
564 | } | ||
565 | |||
566 | try { | ||
567 | //conn.add<CURLOPT_VERBOSE>(1); | ||
568 | //conn.add<CURLOPT_DEBUGFUNCTION>(my_trace); | ||
569 | conn.add<CURLOPT_URL>(url.c_str()); | ||
570 | conn.add<CURLOPT_COPYPOSTFIELDS>(datastr.c_str()); | ||
571 | conn.add<CURLOPT_HTTPHEADER>(headers.get()); | ||
572 | |||
573 | conn.perform(); | ||
574 | } catch (curl::curl_easy_exception error) | ||
575 | { | ||
576 | error.print_traceback(); | ||
577 | |||
578 | return false; | ||
579 | } | ||
580 | |||
581 | response_code = conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
582 | result = output.str(); | ||
583 | |||
584 | return true; | ||
585 | } | ||
586 | |||
587 | bool client::performMultiPost(std::string url, const curl_httppost* fields, long& response_code, std::string& result) | ||
588 | { | ||
589 | std::ostringstream output; | ||
590 | curl::curl_ios<std::ostringstream> ios(output); | ||
591 | curl::curl_easy conn(ios); | ||
592 | |||
593 | curl::curl_header headers; | ||
594 | std::string oauth_header = _oauth_client->getFormattedHttpHeader(OAuth::Http::Post, url, ""); | ||
595 | if (!oauth_header.empty()) | ||
596 | { | ||
597 | headers.add(oauth_header); | ||
598 | } | ||
599 | |||
600 | try { | ||
601 | //conn.add<CURLOPT_VERBOSE>(1); | ||
602 | //conn.add<CURLOPT_DEBUGFUNCTION>(my_trace); | ||
603 | conn.add<CURLOPT_HTTPHEADER>(headers.get()); | ||
604 | conn.add<CURLOPT_URL>(url.c_str()); | ||
605 | conn.add<CURLOPT_HTTPPOST>(fields); | ||
606 | |||
607 | conn.perform(); | ||
608 | } catch (curl::curl_easy_exception error) | ||
609 | { | ||
610 | error.print_traceback(); | ||
611 | |||
612 | return false; | ||
613 | } | ||
614 | |||
615 | response_code = conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
616 | result = output.str(); | ||
617 | |||
618 | return true; | ||
619 | } | ||
620 | |||
621 | response client::codeForError(int response_code, std::string response_data) const | ||
622 | { | ||
623 | json response_json; | ||
624 | try { | ||
625 | response_json = json::parse(response_data); | ||
626 | } catch (std::invalid_argument e) | ||
627 | { | ||
628 | return response::invalid_response; | ||
629 | } | ||
630 | |||
631 | std::set<int> error_codes; | ||
632 | if (response_json.find("errors") != response_json.end()) | ||
633 | { | ||
634 | std::transform(std::begin(response_json["errors"]), std::end(response_json["errors"]), std::inserter(error_codes, std::begin(error_codes)), [] (const json& error) { | ||
635 | return error["code"].get<int>(); | ||
636 | }); | ||
637 | } | ||
638 | |||
639 | if (error_codes.count(32) == 1 || error_codes.count(135) == 1 || error_codes.count(215) == 1) | ||
640 | { | ||
641 | return response::bad_auth; | ||
642 | } else if (error_codes.count(64) == 1) | ||
643 | { | ||
644 | return response::suspended; | ||
645 | } else if (error_codes.count(88) == 1 || error_codes.count(185) == 1) | ||
646 | { | ||
647 | return response::limited; | ||
648 | } else if (error_codes.count(89) == 1) | ||
649 | { | ||
650 | return response::bad_token; | ||
651 | } else if (error_codes.count(130) == 1) | ||
652 | { | ||
653 | return response::server_overloaded; | ||
654 | } else if (error_codes.count(131) == 1) | ||
655 | { | ||
656 | return response::server_error; | ||
657 | } else if (error_codes.count(186) == 1) | ||
658 | { | ||
659 | return response::bad_length; | ||
660 | } else if (error_codes.count(187) == 1) | ||
661 | { | ||
662 | return response::duplicate_status; | ||
663 | } else if (error_codes.count(226) == 1) | ||
664 | { | ||
665 | return response::suspected_spam; | ||
666 | } else if (error_codes.count(261) == 1) | ||
667 | { | ||
668 | return response::write_restricted; | ||
669 | } else if (error_codes.count(44) == 1) | ||
670 | { | ||
671 | return response::invalid_media; | ||
672 | } else if (response_code == 429) | ||
673 | { | ||
674 | return response::limited; | ||
675 | } else if (response_code == 500) | ||
676 | { | ||
677 | return response::server_error; | ||
678 | } else if (response_code == 502) | ||
679 | { | ||
680 | return response::server_unavailable; | ||
681 | } else if (response_code == 503) | ||
682 | { | ||
683 | return response::server_overloaded; | ||
684 | } else if (response_code == 504) | ||
685 | { | ||
686 | return response::server_timeout; | ||
687 | } else { | ||
688 | return response::unknown_error; | ||
689 | } | ||
690 | } | ||
691 | |||
692 | client::stream::stream(client& _client) : _client(_client) | ||
693 | { | ||
694 | |||
695 | } | ||
696 | |||
697 | bool client::stream::isRunning() const | ||
698 | { | ||
699 | return _thread.joinable(); | ||
700 | } | 568 | } |
701 | 569 | ||
702 | void client::stream::setNotifyCallback(notify_callback _n) | 570 | void client::follow(user_id toFollow) const |
703 | { | 571 | { |
704 | std::lock_guard<std::mutex> _running_lock(_running_mutex); | 572 | std::stringstream datastrstream; |
573 | datastrstream << "follow=true&user_id="; | ||
574 | datastrstream << toFollow; | ||
705 | 575 | ||
706 | if (!_thread.joinable()) | 576 | post(*_oauth_client, "https://api.twitter.com/1.1/friendships/create.json", datastrstream.str()).perform(); |
707 | { | ||
708 | _notify = _n; | ||
709 | } | ||
710 | } | 577 | } |
711 | 578 | ||
712 | void client::stream::setReceiveAllReplies(bool _arg) | 579 | void client::unfollow(user_id toUnfollow) const |
713 | { | 580 | { |
714 | std::lock_guard<std::mutex> _running_lock(_running_mutex); | 581 | std::stringstream datastrstream; |
715 | 582 | datastrstream << "user_id="; | |
716 | if (!_thread.joinable()) | 583 | datastrstream << toUnfollow; |
717 | { | ||
718 | _receive_all_replies = _arg; | ||
719 | } | ||
720 | } | ||
721 | |||
722 | void client::stream::start() | ||
723 | { | ||
724 | std::lock_guard<std::mutex> _running_lock(_running_mutex); | ||
725 | 584 | ||
726 | if (!_thread.joinable()) | 585 | post(*_oauth_client, "https://api.twitter.com/1.1/friendships/destroy.json", datastrstream.str()).perform(); |
727 | { | ||
728 | _thread = std::thread(&stream::run, this); | ||
729 | } | ||
730 | } | 586 | } |
731 | |||
732 | void client::stream::stop() | ||
733 | { | ||
734 | std::lock_guard<std::mutex> _running_lock(_running_mutex); | ||
735 | |||
736 | if (_thread.joinable()) | ||
737 | { | ||
738 | _stop = true; | ||
739 | _thread.join(); | ||
740 | _stop = false; | ||
741 | } | ||
742 | } | ||
743 | |||
744 | void client::stream::run() | ||
745 | { | ||
746 | std::ostringstream urlstr; | ||
747 | urlstr << "https://userstream.twitter.com/1.1/user.json"; | ||
748 | |||
749 | if (_receive_all_replies) | ||
750 | { | ||
751 | urlstr << "?replies=all"; | ||
752 | } | ||
753 | |||
754 | std::string url = urlstr.str(); | ||
755 | |||
756 | _backoff_type = backoff::none; | ||
757 | _backoff_amount = std::chrono::milliseconds(0); | ||
758 | for (;;) | ||
759 | { | ||
760 | curl::curl_easy conn; | ||
761 | curl::curl_header headers; | ||
762 | std::string oauth_header = _client._oauth_client->getFormattedHttpHeader(OAuth::Http::Get, url, ""); | ||
763 | if (!oauth_header.empty()) | ||
764 | { | ||
765 | headers.add(oauth_header); | ||
766 | } | ||
767 | |||
768 | conn.add<CURLOPT_WRITEFUNCTION>(client_stream_write_callback_wrapper); | ||
769 | conn.add<CURLOPT_WRITEDATA>(this); | ||
770 | conn.add<CURLOPT_HEADERFUNCTION>(nullptr); | ||
771 | conn.add<CURLOPT_HEADERDATA>(nullptr); | ||
772 | conn.add<CURLOPT_XFERINFOFUNCTION>(client_stream_progress_callback_wrapper); | ||
773 | conn.add<CURLOPT_XFERINFODATA>(this); | ||
774 | conn.add<CURLOPT_NOPROGRESS>(0); | ||
775 | //conn.add<CURLOPT_VERBOSE>(1); | ||
776 | //conn.add<CURLOPT_DEBUGFUNCTION>(my_trace); | ||
777 | conn.add<CURLOPT_URL>(url.c_str()); | ||
778 | conn.add<CURLOPT_HTTPHEADER>(headers.get()); | ||
779 | |||
780 | bool failure = false; | ||
781 | try { | ||
782 | conn.perform(); | ||
783 | } catch (curl::curl_easy_exception error) | ||
784 | { | ||
785 | failure = true; | ||
786 | if ((error.get_code() == CURLE_ABORTED_BY_CALLBACK) && _stop) | ||
787 | { | ||
788 | break; | ||
789 | } else { | ||
790 | if (_backoff_type == backoff::none) | ||
791 | { | ||
792 | _established = false; | ||
793 | _backoff_type = backoff::network; | ||
794 | _backoff_amount = std::chrono::milliseconds(0); | ||
795 | } | ||
796 | } | ||
797 | } | ||
798 | |||
799 | if (!failure) | ||
800 | { | ||
801 | long response_code = conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
802 | if (response_code == 420) | ||
803 | { | ||
804 | if (_backoff_type == backoff::none) | ||
805 | { | ||
806 | _established = false; | ||
807 | _backoff_type = backoff::rate_limit; | ||
808 | _backoff_amount = std::chrono::minutes(1); | ||
809 | } | ||
810 | } else if (response_code != 200) | ||
811 | { | ||
812 | if (_backoff_type == backoff::none) | ||
813 | { | ||
814 | _established = false; | ||
815 | _backoff_type = backoff::http; | ||
816 | _backoff_amount = std::chrono::seconds(5); | ||
817 | } | ||
818 | } else { | ||
819 | if (_backoff_type == backoff::none) | ||
820 | { | ||
821 | _established = false; | ||
822 | _backoff_type = backoff::network; | ||
823 | _backoff_amount = std::chrono::milliseconds(0); | ||
824 | } | ||
825 | } | ||
826 | } | ||
827 | |||
828 | std::this_thread::sleep_for(_backoff_amount); | ||
829 | 587 | ||
830 | switch (_backoff_type) | 588 | const user& client::getUser() const |
831 | { | ||
832 | case backoff::network: | ||
833 | { | ||
834 | if (_backoff_amount < std::chrono::seconds(16)) | ||
835 | { | ||
836 | _backoff_amount += std::chrono::milliseconds(250); | ||
837 | } | ||
838 | |||
839 | break; | ||
840 | } | ||
841 | |||
842 | case backoff::http: | ||
843 | { | ||
844 | if (_backoff_amount < std::chrono::seconds(320)) | ||
845 | { | ||
846 | _backoff_amount *= 2; | ||
847 | } | ||
848 | |||
849 | break; | ||
850 | } | ||
851 | |||
852 | case backoff::rate_limit: | ||
853 | { | ||
854 | _backoff_amount *= 2; | ||
855 | |||
856 | break; | ||
857 | } | ||
858 | } | ||
859 | } | ||
860 | } | ||
861 | |||
862 | size_t client::stream::write(char* ptr, size_t size, size_t nmemb) | ||
863 | { | 589 | { |
864 | for (size_t i = 0; i < size*nmemb; i++) | 590 | return *_current_user; |
865 | { | ||
866 | if (ptr[i] == '\r') | ||
867 | { | ||
868 | i++; // Skip the \n | ||
869 | |||
870 | if (!_buffer.empty()) | ||
871 | { | ||
872 | notification n(_buffer, _client._current_user); | ||
873 | if (n.getType() == notification::type::friends) | ||
874 | { | ||
875 | _established = true; | ||
876 | _backoff_type = backoff::none; | ||
877 | _backoff_amount = std::chrono::milliseconds(0); | ||
878 | } | ||
879 | |||
880 | if (_notify) | ||
881 | { | ||
882 | _notify(n); | ||
883 | } | ||
884 | |||
885 | _buffer = ""; | ||
886 | } | ||
887 | } else { | ||
888 | _buffer.push_back(ptr[i]); | ||
889 | } | ||
890 | } | ||
891 | |||
892 | { | ||
893 | std::lock_guard<std::mutex> _stall_lock(_stall_mutex); | ||
894 | time(&_last_write); | ||
895 | } | ||
896 | |||
897 | return size*nmemb; | ||
898 | } | 591 | } |
899 | 592 | ||
900 | int client::stream::progress() | 593 | const configuration& client::getConfiguration() const |
901 | { | 594 | { |
902 | if (_stop) | 595 | if (!_configuration || (difftime(time(NULL), _last_configuration_update) > 60*60*24)) |
903 | { | ||
904 | return 1; | ||
905 | } | ||
906 | |||
907 | if (_established) | ||
908 | { | 596 | { |
909 | std::lock_guard<std::mutex> _stall_lock(_stall_mutex); | 597 | _configuration = |
910 | if (difftime(time(NULL), _last_write) >= 90) | 598 | make_unique<configuration>( |
911 | { | 599 | get(*_oauth_client, |
912 | return 1; | 600 | "https://api.twitter.com/1.1/help/configuration.json") |
913 | } | 601 | .perform()); |
602 | |||
603 | _last_configuration_update = time(NULL); | ||
914 | } | 604 | } |
915 | 605 | ||
916 | return 0; | 606 | return *_configuration; |
917 | } | 607 | } |
918 | 608 | ||
919 | }; | 609 | }; |
diff --git a/src/client.h b/src/client.h index 6963412..37081ff 100644 --- a/src/client.h +++ b/src/client.h | |||
@@ -1,17 +1,15 @@ | |||
1 | #ifndef TWITTER_H_ABFF6A12 | 1 | #ifndef TWITTER_H_ABFF6A12 |
2 | #define TWITTER_H_ABFF6A12 | 2 | #define TWITTER_H_ABFF6A12 |
3 | 3 | ||
4 | #include "codes.h" | ||
5 | #include "tweet.h" | ||
6 | #include "auth.h" | ||
7 | #include <list> | 4 | #include <list> |
8 | #include <thread> | ||
9 | #include <mutex> | ||
10 | #include "notification.h" | ||
11 | #include <set> | 5 | #include <set> |
12 | #include <ctime> | 6 | #include <ctime> |
13 | #include <chrono> | 7 | #include <memory> |
8 | #include "codes.h" | ||
9 | #include "tweet.h" | ||
10 | #include "auth.h" | ||
14 | #include "configuration.h" | 11 | #include "configuration.h" |
12 | #include "util.h" | ||
15 | 13 | ||
16 | namespace OAuth { | 14 | namespace OAuth { |
17 | class Consumer; | 15 | class Consumer; |
@@ -19,96 +17,39 @@ namespace OAuth { | |||
19 | class Client; | 17 | class Client; |
20 | }; | 18 | }; |
21 | 19 | ||
22 | class curl_httppost; | ||
23 | |||
24 | namespace twitter { | 20 | namespace twitter { |
25 | 21 | ||
26 | class client { | 22 | class client { |
27 | public: | 23 | public: |
28 | class stream { | 24 | |
29 | public: | 25 | client(const auth& _auth); |
30 | typedef std::function<void(notification _notification)> notify_callback; | 26 | ~client(); |
31 | 27 | ||
32 | stream(client& _client); | 28 | tweet updateStatus(std::string msg, std::list<long> media_ids = {}) const; |
33 | 29 | long uploadMedia(std::string media_type, const char* data, long data_length) const; | |
34 | void setNotifyCallback(notify_callback _n); | 30 | |
35 | void setReceiveAllReplies(bool _arg); | 31 | tweet replyToTweet(std::string msg, tweet_id in_response_to, std::list<long> media_ids = {}) const; |
36 | 32 | std::set<user_id> getFriends(user_id id) const; | |
37 | bool isRunning() const; | 33 | std::set<user_id> getFollowers(user_id id) const; |
38 | void start(); | 34 | void follow(user_id toFollow) const; |
39 | void stop(); | 35 | void unfollow(user_id toUnfollow) const; |
40 | 36 | ||
41 | int progress(); | 37 | const user& getUser() const; |
42 | size_t write(char* ptr, size_t size, size_t nmemb); | 38 | |
43 | 39 | const configuration& getConfiguration() const; | |
44 | private: | 40 | |
45 | enum class backoff { | 41 | private: |
46 | none, | 42 | |
47 | network, | 43 | friend class stream; |
48 | http, | 44 | |
49 | rate_limit | 45 | std::unique_ptr<OAuth::Consumer> _oauth_consumer; |
50 | }; | 46 | std::unique_ptr<OAuth::Token> _oauth_token; |
51 | 47 | std::unique_ptr<OAuth::Client> _oauth_client; | |
52 | void run(); | 48 | |
53 | 49 | std::unique_ptr<user> _current_user; | |
54 | client& _client; | 50 | |
55 | notify_callback _notify; | 51 | mutable std::unique_ptr<configuration> _configuration; |
56 | bool _stop = false; | 52 | mutable time_t _last_configuration_update; |
57 | std::thread _thread; | ||
58 | std::mutex _running_mutex; | ||
59 | std::mutex _stall_mutex; | ||
60 | std::string _buffer; | ||
61 | time_t _last_write; | ||
62 | bool _established = false; | ||
63 | backoff _backoff_type = backoff::none; | ||
64 | std::chrono::milliseconds _backoff_amount; | ||
65 | bool _receive_all_replies = false; | ||
66 | }; | ||
67 | |||
68 | client(const auth& _auth); | ||
69 | ~client(); | ||
70 | |||
71 | response updateStatus(std::string msg, tweet& result, tweet in_response_to = tweet(), std::list<long> media_ids = {}); | ||
72 | response uploadMedia(std::string media_type, const char* data, long data_length, long& media_id); | ||
73 | |||
74 | response follow(user_id toFollow); | ||
75 | response follow(user toFollow); | ||
76 | |||
77 | response unfollow(user_id toUnfollow); | ||
78 | response unfollow(user toUnfollow); | ||
79 | |||
80 | response getFriends(std::set<user_id>& result); | ||
81 | response getFollowers(std::set<user_id>& result); | ||
82 | |||
83 | response getUser(user& result); | ||
84 | |||
85 | configuration getConfiguration(); | ||
86 | |||
87 | // NOTE: stream setting function calls will fail silently when stream is running | ||
88 | void setUserStreamNotifyCallback(stream::notify_callback callback); | ||
89 | void setUserStreamReceiveAllReplies(bool _arg); | ||
90 | void startUserStream(); | ||
91 | void stopUserStream(); | ||
92 | |||
93 | std::string generateReplyPrefill(tweet t) const; | ||
94 | |||
95 | private: | ||
96 | friend class stream; | ||
97 | |||
98 | OAuth::Consumer* _oauth_consumer; | ||
99 | OAuth::Token* _oauth_token; | ||
100 | OAuth::Client* _oauth_client; | ||
101 | |||
102 | user _current_user; | ||
103 | stream _user_stream{*this}; | ||
104 | |||
105 | configuration _configuration; | ||
106 | time_t _last_configuration_update; | ||
107 | |||
108 | bool performGet(std::string url, long& response_code, std::string& result); | ||
109 | bool performPost(std::string url, std::string dataStr, long& response_code, std::string& result); | ||
110 | bool performMultiPost(std::string url, const curl_httppost* fields, long& response_code, std::string& result); | ||
111 | response codeForError(int httpcode, std::string errors) const; | ||
112 | }; | 53 | }; |
113 | 54 | ||
114 | }; | 55 | }; |
diff --git a/src/codes.cpp b/src/codes.cpp index 9639d5d..7805c7a 100644 --- a/src/codes.cpp +++ b/src/codes.cpp | |||
@@ -1,24 +1,17 @@ | |||
1 | #include "codes.h" | 1 | #include "codes.h" |
2 | #include <sstream> | ||
2 | 3 | ||
3 | std::ostream& operator<<(std::ostream& os, twitter::response r) | 4 | namespace twitter { |
4 | { | 5 | |
5 | switch (r) | 6 | const char* invalid_response::WHAT_TEXT = "Invalid response data received from Twitter"; |
7 | const char* connection_error::WHAT_TEXT = "Error connecting to Twitter"; | ||
8 | |||
9 | std::string unknown_error::generateMessage(int response_code) | ||
6 | { | 10 | { |
7 | case twitter::response::ok: return os << "OK"; | 11 | std::ostringstream msgbuilder; |
8 | case twitter::response::curl_error: return os << "Curl Error"; | 12 | msgbuilder << "Unknown error (HTTP " << response_code << ")"; |
9 | case twitter::response::bad_auth: return os << "Bad Auth"; | 13 | |
10 | case twitter::response::limited: return os << "Rate Limit Exceeded"; | 14 | return msgbuilder.str(); |
11 | case twitter::response::server_error: return os << "Twitter Server Error"; | ||
12 | case twitter::response::server_unavailable: return os << "Twitter Is Down"; | ||
13 | case twitter::response::server_overloaded: return os << "Twitter Is Over Capacity"; | ||
14 | case twitter::response::server_timeout: return os << "Twitter Connection Timed Out"; | ||
15 | case twitter::response::suspended: return os << "Authenticated User Is Suspended"; | ||
16 | case twitter::response::bad_token: return os << "Invalid Or Expired Access Token"; | ||
17 | case twitter::response::duplicate_status: return os << "Duplicate Status"; | ||
18 | case twitter::response::suspected_spam: return os << "Request Looks Automated"; | ||
19 | case twitter::response::write_restricted: return os << "Cannot Perform Write"; | ||
20 | case twitter::response::bad_length: return os << "Message Body Too Long"; | ||
21 | case twitter::response::unknown_error: return os << "Unknown Error"; | ||
22 | case twitter::response::invalid_media: return os << "Invalid Media"; | ||
23 | } | 15 | } |
24 | } | 16 | |
17 | }; | ||
diff --git a/src/codes.h b/src/codes.h index 6767805..baf4f8f 100644 --- a/src/codes.h +++ b/src/codes.h | |||
@@ -1,32 +1,183 @@ | |||
1 | #ifndef CODES_H_05838D39 | 1 | #ifndef CODES_H_05838D39 |
2 | #define CODES_H_05838D39 | 2 | #define CODES_H_05838D39 |
3 | 3 | ||
4 | #include <ostream> | 4 | #include <stdexcept> |
5 | #include <string> | ||
5 | 6 | ||
6 | namespace twitter { | 7 | namespace twitter { |
7 | 8 | ||
8 | enum class response { | 9 | class twitter_error : public std::runtime_error { |
9 | ok, | 10 | public: |
10 | curl_error, | 11 | |
11 | bad_auth, | 12 | using std::runtime_error::runtime_error; |
12 | limited, | 13 | }; |
13 | server_error, | 14 | |
14 | server_unavailable, | 15 | class bad_auth : public twitter_error { |
15 | server_overloaded, | 16 | public: |
16 | server_timeout, | 17 | |
17 | suspended, | 18 | using twitter_error::twitter_error; |
18 | bad_token, | 19 | }; |
19 | duplicate_status, | 20 | |
20 | suspected_spam, | 21 | class invalid_response : public twitter_error { |
21 | write_restricted, | 22 | public: |
22 | bad_length, | 23 | |
23 | unknown_error, | 24 | static const char* WHAT_TEXT; |
24 | invalid_media, | 25 | |
25 | invalid_response | 26 | explicit invalid_response(std::string response) noexcept |
27 | : twitter_error(WHAT_TEXT), _response(std::move(response)) | ||
28 | { | ||
29 | } | ||
30 | |||
31 | const std::string& getResponse() const noexcept | ||
32 | { | ||
33 | return _response; | ||
34 | } | ||
35 | |||
36 | private: | ||
37 | |||
38 | std::string _response; | ||
39 | }; | ||
40 | |||
41 | class account_suspended : public twitter_error { | ||
42 | public: | ||
43 | |||
44 | using twitter_error::twitter_error; | ||
45 | }; | ||
46 | |||
47 | class rate_limit_exceeded : public twitter_error { | ||
48 | public: | ||
49 | |||
50 | using twitter_error::twitter_error; | ||
51 | }; | ||
52 | |||
53 | class update_limit_exceeded : public twitter_error { | ||
54 | public: | ||
55 | |||
56 | using twitter_error::twitter_error; | ||
57 | }; | ||
58 | |||
59 | class bad_token : public twitter_error { | ||
60 | public: | ||
61 | |||
62 | using twitter_error::twitter_error; | ||
63 | }; | ||
64 | |||
65 | class server_overloaded : public twitter_error { | ||
66 | public: | ||
67 | |||
68 | using twitter_error::twitter_error; | ||
69 | }; | ||
70 | |||
71 | class server_error : public twitter_error { | ||
72 | public: | ||
73 | |||
74 | using twitter_error::twitter_error; | ||
75 | }; | ||
76 | |||
77 | class bad_length : public twitter_error { | ||
78 | public: | ||
79 | |||
80 | using twitter_error::twitter_error; | ||
81 | }; | ||
82 | |||
83 | class duplicate_status : public twitter_error { | ||
84 | public: | ||
85 | |||
86 | using twitter_error::twitter_error; | ||
87 | }; | ||
88 | |||
89 | class suspected_spam : public twitter_error { | ||
90 | public: | ||
91 | |||
92 | using twitter_error::twitter_error; | ||
93 | }; | ||
94 | |||
95 | class write_restricted : public twitter_error { | ||
96 | public: | ||
97 | |||
98 | using twitter_error::twitter_error; | ||
99 | }; | ||
100 | |||
101 | class invalid_media : public twitter_error { | ||
102 | public: | ||
103 | |||
104 | using twitter_error::twitter_error; | ||
105 | }; | ||
106 | |||
107 | class server_unavailable : public twitter_error { | ||
108 | public: | ||
109 | |||
110 | using twitter_error::twitter_error; | ||
111 | }; | ||
112 | |||
113 | class server_timeout : public twitter_error { | ||
114 | public: | ||
115 | |||
116 | using twitter_error::twitter_error; | ||
117 | }; | ||
118 | |||
119 | class unknown_error : public twitter_error { | ||
120 | public: | ||
121 | |||
122 | unknown_error(int response_code, std::string response_data) | ||
123 | : twitter_error(generateMessage(response_code)), | ||
124 | _response_code(response_code), | ||
125 | _response_data(std::move(response_data)) | ||
126 | { | ||
127 | } | ||
128 | |||
129 | int getResponseCode() const noexcept | ||
130 | { | ||
131 | return _response_code; | ||
132 | } | ||
133 | |||
134 | const std::string& getResponse() const noexcept | ||
135 | { | ||
136 | return _response_data; | ||
137 | } | ||
138 | |||
139 | private: | ||
140 | |||
141 | static std::string generateMessage(int response_code); | ||
142 | |||
143 | int _response_code; | ||
144 | std::string _response_data; | ||
145 | }; | ||
146 | |||
147 | class connection_error : public twitter_error { | ||
148 | public: | ||
149 | |||
150 | static const char* WHAT_TEXT; | ||
151 | |||
152 | connection_error() noexcept : twitter_error(WHAT_TEXT) | ||
153 | { | ||
154 | } | ||
155 | }; | ||
156 | |||
157 | class invalid_member : public std::domain_error { | ||
158 | public: | ||
159 | |||
160 | using std::domain_error::domain_error; | ||
161 | }; | ||
162 | |||
163 | class malformed_object : public invalid_response { | ||
164 | public: | ||
165 | |||
166 | malformed_object(std::string type, std::string data) noexcept | ||
167 | : invalid_response(std::move(data)), _type(std::move(type)) | ||
168 | { | ||
169 | } | ||
170 | |||
171 | const std::string& getType() const noexcept | ||
172 | { | ||
173 | return _type; | ||
174 | } | ||
175 | |||
176 | private: | ||
177 | |||
178 | std::string _type; | ||
26 | }; | 179 | }; |
27 | 180 | ||
28 | }; | 181 | }; |
29 | 182 | ||
30 | std::ostream& operator<<(std::ostream& os, twitter::response r); | ||
31 | |||
32 | #endif /* end of include guard: CODES_H_05838D39 */ | 183 | #endif /* end of include guard: CODES_H_05838D39 */ |
diff --git a/src/configuration.cpp b/src/configuration.cpp index 63464ed..0010a40 100644 --- a/src/configuration.cpp +++ b/src/configuration.cpp | |||
@@ -1,35 +1,27 @@ | |||
1 | #include "configuration.h" | 1 | #include "configuration.h" |
2 | #include <json.hpp> | 2 | #include <json.hpp> |
3 | #include <cassert> | 3 | #include <cassert> |
4 | 4 | #include "codes.h" | |
5 | using nlohmann::json; | ||
6 | 5 | ||
7 | namespace twitter { | 6 | namespace twitter { |
8 | 7 | ||
9 | configuration::configuration() | 8 | configuration::configuration(std::string data) try |
10 | { | 9 | { |
11 | _valid = false; | 10 | auto json_data = nlohmann::json::parse(data); |
12 | } | ||
13 | |||
14 | configuration::configuration(std::string data) | ||
15 | { | ||
16 | _valid = true; | ||
17 | |||
18 | auto _data = json::parse(data); | ||
19 | 11 | ||
20 | _characters_reserved_per_media = _data.at("characters_reserved_per_media"); | 12 | _characters_reserved_per_media = json_data["characters_reserved_per_media"].get<size_t>(); |
21 | _dm_text_character_limit = _data.at("dm_text_character_limit"); | 13 | _dm_text_character_limit = json_data["dm_text_character_limit"].get<size_t>(); |
22 | _max_media_per_upload = _data.at("max_media_per_upload"); | 14 | _max_media_per_upload = json_data["max_media_per_upload"].get<size_t>(); |
23 | _photo_size_limit = _data.at("photo_size_limit"); | 15 | _photo_size_limit = json_data["photo_size_limit"].get<size_t>(); |
24 | _short_url_length = _data.at("short_url_length"); | 16 | _short_url_length = json_data["short_url_length"].get<size_t>(); |
25 | _short_https_url_length = _data.at("short_url_length_https"); | 17 | _short_https_url_length = json_data["short_url_length_https"].get<size_t>(); |
26 | 18 | ||
27 | for (json::iterator sizedata = _data.at("photo_sizes").begin(); sizedata != _data.at("photo_sizes").end(); ++sizedata) | 19 | for (auto sizedata = std::begin(json_data["photo_sizes"]); sizedata != std::end(json_data["photo_sizes"]); ++sizedata) |
28 | { | 20 | { |
29 | photosize size; | 21 | photosize size; |
30 | size.height = sizedata.value().at("h"); | 22 | size.height = sizedata.value()["h"].get<size_t>(); |
31 | size.width = sizedata.value().at("w"); | 23 | size.width = sizedata.value()["w"].get<size_t>(); |
32 | if (sizedata.value().at("resize") == "fit") | 24 | if (sizedata.value()["resize"].get<std::string>() == "fit") |
33 | { | 25 | { |
34 | size.resize = resizetype::fit; | 26 | size.resize = resizetype::fit; |
35 | } else { | 27 | } else { |
@@ -39,71 +31,16 @@ namespace twitter { | |||
39 | _photo_sizes[sizedata.key()] = size; | 31 | _photo_sizes[sizedata.key()] = size; |
40 | } | 32 | } |
41 | 33 | ||
42 | for (auto path : _data.at("non_username_paths")) | 34 | for (auto path : json_data["non_username_paths"]) |
43 | { | 35 | { |
44 | _non_username_paths.insert(path.get<std::string>()); | 36 | _non_username_paths.insert(path.get<std::string>()); |
45 | } | 37 | } |
46 | } | 38 | } catch (const std::invalid_argument& error) |
47 | |||
48 | size_t configuration::getCharactersReservedPerMedia() const | ||
49 | { | ||
50 | assert(_valid); | ||
51 | |||
52 | return _characters_reserved_per_media; | ||
53 | } | ||
54 | |||
55 | size_t configuration::getDirectMessageCharacterLimit() const | ||
56 | { | ||
57 | assert(_valid); | ||
58 | |||
59 | return _dm_text_character_limit; | ||
60 | } | ||
61 | |||
62 | size_t configuration::getMaxMediaPerUpload() const | ||
63 | { | ||
64 | assert(_valid); | ||
65 | |||
66 | return _max_media_per_upload; | ||
67 | } | ||
68 | |||
69 | size_t configuration::getPhotoSizeLimit() const | ||
70 | { | ||
71 | assert(_valid); | ||
72 | |||
73 | return _photo_size_limit; | ||
74 | } | ||
75 | |||
76 | std::map<std::string, configuration::photosize> configuration::getPhotoSizes() const | ||
77 | { | ||
78 | assert(_valid); | ||
79 | |||
80 | return _photo_sizes; | ||
81 | } | ||
82 | |||
83 | size_t configuration::getShortUrlLength() const | ||
84 | { | ||
85 | assert(_valid); | ||
86 | |||
87 | return _short_url_length; | ||
88 | } | ||
89 | |||
90 | size_t configuration::getShortHttpsUrlLength() const | ||
91 | { | ||
92 | assert(_valid); | ||
93 | |||
94 | return _short_https_url_length; | ||
95 | } | ||
96 | |||
97 | std::set<std::string> configuration::getNonUsernamePaths() const | ||
98 | { | 39 | { |
99 | assert(_valid); | 40 | std::throw_with_nested(malformed_object("configuration", data)); |
100 | 41 | } catch (const std::domain_error& error) | |
101 | return _non_username_paths; | ||
102 | } | ||
103 | |||
104 | configuration::operator bool() const | ||
105 | { | 42 | { |
106 | return _valid; | 43 | std::throw_with_nested(malformed_object("configuration", data)); |
107 | } | 44 | } |
108 | 45 | ||
109 | }; | 46 | }; |
diff --git a/src/configuration.h b/src/configuration.h index a15be3d..543e306 100644 --- a/src/configuration.h +++ b/src/configuration.h | |||
@@ -8,43 +8,70 @@ | |||
8 | namespace twitter { | 8 | namespace twitter { |
9 | 9 | ||
10 | class configuration { | 10 | class configuration { |
11 | public: | 11 | public: |
12 | enum class resizetype { | 12 | enum class resizetype { |
13 | fit, | 13 | fit, |
14 | crop | 14 | crop |
15 | }; | 15 | }; |
16 | 16 | ||
17 | struct photosize { | 17 | struct photosize { |
18 | size_t height; | 18 | size_t height; |
19 | size_t width; | 19 | size_t width; |
20 | resizetype resize; | 20 | resizetype resize; |
21 | }; | 21 | }; |
22 | 22 | ||
23 | configuration(); | 23 | explicit configuration(std::string data); |
24 | configuration(std::string data); | 24 | |
25 | 25 | size_t getCharactersReservedPerMedia() const | |
26 | size_t getCharactersReservedPerMedia() const; | 26 | { |
27 | size_t getDirectMessageCharacterLimit() const; | 27 | return _characters_reserved_per_media; |
28 | size_t getMaxMediaPerUpload() const; | 28 | } |
29 | size_t getPhotoSizeLimit() const; | 29 | |
30 | std::map<std::string, photosize> getPhotoSizes() const; | 30 | size_t getDirectMessageCharacterLimit() const |
31 | size_t getShortUrlLength() const; | 31 | { |
32 | size_t getShortHttpsUrlLength() const; | 32 | return _dm_text_character_limit; |
33 | std::set<std::string> getNonUsernamePaths() const; | 33 | } |
34 | 34 | ||
35 | operator bool() const; | 35 | size_t getMaxMediaPerUpload() const |
36 | 36 | { | |
37 | private: | 37 | return _max_media_per_upload; |
38 | bool _valid = false; | 38 | } |
39 | 39 | ||
40 | size_t _characters_reserved_per_media; | 40 | size_t getPhotoSizeLimit() const |
41 | size_t _dm_text_character_limit; | 41 | { |
42 | size_t _max_media_per_upload; | 42 | return _photo_size_limit; |
43 | size_t _photo_size_limit; | 43 | } |
44 | std::map<std::string, photosize> _photo_sizes; | 44 | |
45 | size_t _short_url_length; | 45 | const std::map<std::string, photosize>& getPhotoSizes() const |
46 | size_t _short_https_url_length; | 46 | { |
47 | std::set<std::string> _non_username_paths; | 47 | return _photo_sizes; |
48 | } | ||
49 | |||
50 | size_t getShortUrlLength() const | ||
51 | { | ||
52 | return _short_url_length; | ||
53 | } | ||
54 | |||
55 | size_t getShortHttpsUrlLength() const | ||
56 | { | ||
57 | return _short_https_url_length; | ||
58 | } | ||
59 | |||
60 | const std::set<std::string>& getNonUsernamePaths() const | ||
61 | { | ||
62 | return _non_username_paths; | ||
63 | } | ||
64 | |||
65 | private: | ||
66 | |||
67 | size_t _characters_reserved_per_media; | ||
68 | size_t _dm_text_character_limit; | ||
69 | size_t _max_media_per_upload; | ||
70 | size_t _photo_size_limit; | ||
71 | std::map<std::string, photosize> _photo_sizes; | ||
72 | size_t _short_url_length; | ||
73 | size_t _short_https_url_length; | ||
74 | std::set<std::string> _non_username_paths; | ||
48 | }; | 75 | }; |
49 | 76 | ||
50 | }; | 77 | }; |
diff --git a/src/direct_message.h b/src/direct_message.h index 4b5e285..12e254c 100644 --- a/src/direct_message.h +++ b/src/direct_message.h | |||
@@ -6,8 +6,9 @@ | |||
6 | namespace twitter { | 6 | namespace twitter { |
7 | 7 | ||
8 | class direct_message { | 8 | class direct_message { |
9 | public: | 9 | public: |
10 | direct_message(std::string data); | 10 | |
11 | explicit direct_message(std::string data); | ||
11 | }; | 12 | }; |
12 | 13 | ||
13 | }; | 14 | }; |
diff --git a/src/list.cpp b/src/list.cpp index 49405d0..d2b61cc 100644 --- a/src/list.cpp +++ b/src/list.cpp | |||
@@ -2,11 +2,6 @@ | |||
2 | 2 | ||
3 | namespace twitter { | 3 | namespace twitter { |
4 | 4 | ||
5 | list::list() | ||
6 | { | ||
7 | |||
8 | } | ||
9 | |||
10 | list::list(std::string data) | 5 | list::list(std::string data) |
11 | { | 6 | { |
12 | 7 | ||
diff --git a/src/list.h b/src/list.h index 3876f2d..69085b4 100644 --- a/src/list.h +++ b/src/list.h | |||
@@ -6,9 +6,9 @@ | |||
6 | namespace twitter { | 6 | namespace twitter { |
7 | 7 | ||
8 | class list { | 8 | class list { |
9 | public: | 9 | public: |
10 | list(); | 10 | |
11 | list(std::string data); | 11 | explicit list(std::string data); |
12 | }; | 12 | }; |
13 | 13 | ||
14 | }; | 14 | }; |
diff --git a/src/notification.cpp b/src/notification.cpp index 0397ce5..3269a90 100644 --- a/src/notification.cpp +++ b/src/notification.cpp | |||
@@ -2,8 +2,8 @@ | |||
2 | #include <cassert> | 2 | #include <cassert> |
3 | #include <new> | 3 | #include <new> |
4 | #include <json.hpp> | 4 | #include <json.hpp> |
5 | 5 | #include "codes.h" | |
6 | using nlohmann::json; | 6 | #include "client.h" |
7 | 7 | ||
8 | namespace twitter { | 8 | namespace twitter { |
9 | 9 | ||
@@ -12,26 +12,32 @@ namespace twitter { | |||
12 | return _type; | 12 | return _type; |
13 | } | 13 | } |
14 | 14 | ||
15 | notification::notification() : _type(type::invalid) | 15 | notification::notification(const client& tclient, std::string data) |
16 | { | 16 | { |
17 | const user& current_user = tclient.getUser(); | ||
17 | 18 | ||
18 | } | 19 | nlohmann::json json; |
19 | |||
20 | notification::notification(std::string data, const user& current_user) | ||
21 | { | ||
22 | try { | ||
23 | auto _data = json::parse(data); | ||
24 | 20 | ||
25 | if (_data.find("in_reply_to_status_id") != _data.end()) | 21 | try |
22 | { | ||
23 | json = nlohmann::json::parse(data); | ||
24 | } catch (const std::invalid_argument& error) | ||
25 | { | ||
26 | std::throw_with_nested(invalid_response(data)); | ||
27 | } | ||
28 | |||
29 | try | ||
30 | { | ||
31 | if (!json["in_reply_to_status_id"].is_null()) | ||
26 | { | 32 | { |
27 | _type = type::tweet; | 33 | _type = type::tweet; |
28 | 34 | ||
29 | new(&_tweet) tweet(data); | 35 | new(&_tweet) tweet(tclient, data); |
30 | } else if (_data.find("event") != _data.end()) | 36 | } else if (!json["event"].is_null()) |
31 | { | 37 | { |
32 | std::string event = _data.at("event"); | 38 | std::string event = json["event"]; |
33 | user source(_data.at("source").dump()); | 39 | user source(tclient, json["source"].dump()); |
34 | user target(_data.at("target").dump()); | 40 | user target(tclient, json["target"].dump()); |
35 | 41 | ||
36 | if (event == "user_update") | 42 | if (event == "user_update") |
37 | { | 43 | { |
@@ -50,7 +56,7 @@ namespace twitter { | |||
50 | new(&_user) user(target); | 56 | new(&_user) user(target); |
51 | } else if (event == "favorite") | 57 | } else if (event == "favorite") |
52 | { | 58 | { |
53 | new(&_user_and_tweet._tweet) tweet(_data.at("target_object").dump()); | 59 | new(&_user_and_tweet._tweet) tweet(tclient, json["target_object"].dump()); |
54 | 60 | ||
55 | if (current_user == source) | 61 | if (current_user == source) |
56 | { | 62 | { |
@@ -64,7 +70,7 @@ namespace twitter { | |||
64 | } | 70 | } |
65 | } else if (event == "unfavorite") | 71 | } else if (event == "unfavorite") |
66 | { | 72 | { |
67 | new(&_user_and_tweet._tweet) tweet(_data.at("target_object").dump()); | 73 | new(&_user_and_tweet._tweet) tweet(tclient, json["target_object"].dump()); |
68 | 74 | ||
69 | if (current_user == source) | 75 | if (current_user == source) |
70 | { | 76 | { |
@@ -97,20 +103,20 @@ namespace twitter { | |||
97 | { | 103 | { |
98 | _type = type::list_created; | 104 | _type = type::list_created; |
99 | 105 | ||
100 | new(&_list) list(_data.at("target_object").dump()); | 106 | new(&_list) list(json["target_object"].dump()); |
101 | } else if (event == "list_destroyed") | 107 | } else if (event == "list_destroyed") |
102 | { | 108 | { |
103 | _type = type::list_destroyed; | 109 | _type = type::list_destroyed; |
104 | 110 | ||
105 | new(&_list) list(_data.at("target_object").dump()); | 111 | new(&_list) list(json["target_object"].dump()); |
106 | } else if (event == "list_updated") | 112 | } else if (event == "list_updated") |
107 | { | 113 | { |
108 | _type = type::list_updated; | 114 | _type = type::list_updated; |
109 | 115 | ||
110 | new(&_list) list(_data.at("target_object").dump()); | 116 | new(&_list) list(json["target_object"].dump()); |
111 | } else if (event == "list_member_added") | 117 | } else if (event == "list_member_added") |
112 | { | 118 | { |
113 | new(&_user_and_list._list) list(_data.at("target_object").dump()); | 119 | new(&_user_and_list._list) list(json["target_object"].dump()); |
114 | 120 | ||
115 | if (current_user == source) | 121 | if (current_user == source) |
116 | { | 122 | { |
@@ -124,7 +130,7 @@ namespace twitter { | |||
124 | } | 130 | } |
125 | } else if (event == "list_member_removed") | 131 | } else if (event == "list_member_removed") |
126 | { | 132 | { |
127 | new(&_user_and_list._list) list(_data.at("target_object").dump()); | 133 | new(&_user_and_list._list) list(json["target_object"].dump()); |
128 | 134 | ||
129 | if (current_user == source) | 135 | if (current_user == source) |
130 | { | 136 | { |
@@ -138,7 +144,7 @@ namespace twitter { | |||
138 | } | 144 | } |
139 | } else if (event == "list_member_subscribe") | 145 | } else if (event == "list_member_subscribe") |
140 | { | 146 | { |
141 | new(&_user_and_list._list) list(_data.at("target_object").dump()); | 147 | new(&_user_and_list._list) list(json["target_object"].dump()); |
142 | 148 | ||
143 | if (current_user == source) | 149 | if (current_user == source) |
144 | { | 150 | { |
@@ -152,7 +158,7 @@ namespace twitter { | |||
152 | } | 158 | } |
153 | } else if (event == "list_member_unsubscribe") | 159 | } else if (event == "list_member_unsubscribe") |
154 | { | 160 | { |
155 | new(&_user_and_list._list) list(_data.at("target_object").dump()); | 161 | new(&_user_and_list._list) list(json["target_object"].dump()); |
156 | 162 | ||
157 | if (current_user == source) | 163 | if (current_user == source) |
158 | { | 164 | { |
@@ -169,66 +175,67 @@ namespace twitter { | |||
169 | _type = type::quoted; | 175 | _type = type::quoted; |
170 | 176 | ||
171 | new(&_user_and_tweet._user) user(source); | 177 | new(&_user_and_tweet._user) user(source); |
172 | new(&_user_and_tweet._tweet) tweet(_data.at("target_object").dump()); | 178 | new(&_user_and_tweet._tweet) tweet(tclient, json["target_object"].dump()); |
173 | } | 179 | } |
174 | } else if (_data.find("warning") != _data.end()) | 180 | } else if (!json["warning"].is_null()) |
175 | { | 181 | { |
176 | new(&_warning) std::string(_data.at("warning").at("message").get<std::string>()); | 182 | new(&_warning) std::string(json["warning"]["message"].get<std::string>()); |
177 | 183 | ||
178 | if (_data.at("warning").at("code") == "FALLING_BEHIND") | 184 | auto warning_code = json["warning"]["code"].get<std::string>(); |
185 | if (warning_code == "FALLING_BEHIND") | ||
179 | { | 186 | { |
180 | _type = type::stall; | 187 | _type = type::stall; |
181 | } else if (_data.at("warning").at("code") == "FOLLOWS_OVER_LIMIT") | 188 | } else if (warning_code == "FOLLOWS_OVER_LIMIT") |
182 | { | 189 | { |
183 | _type = type::follow_limit; | 190 | _type = type::follow_limit; |
184 | } else { | 191 | } else { |
185 | _type = type::unknown_warning; | 192 | _type = type::unknown_warning; |
186 | } | 193 | } |
187 | } else if (_data.find("delete") != _data.end()) | 194 | } else if (!json["delete"].is_null()) |
188 | { | 195 | { |
189 | _type = type::deletion; | 196 | _type = type::deletion; |
190 | 197 | ||
191 | _user_id_and_tweet_id._tweet_id = _data.at("delete").at("status").at("id"); | 198 | _user_id_and_tweet_id._tweet_id = json["delete"]["status"]["id"].get<tweet_id>(); |
192 | _user_id_and_tweet_id._user_id = _data.at("delete").at("status").at("user_id"); | 199 | _user_id_and_tweet_id._user_id = json["delete"]["status"]["user_id"].get<user_id>(); |
193 | } else if (_data.find("scrub_geo") != _data.end()) | 200 | } else if (!json["scrub_geo"].is_null()) |
194 | { | 201 | { |
195 | _type = type::scrub_location; | 202 | _type = type::scrub_location; |
196 | 203 | ||
197 | _user_id_and_tweet_id._tweet_id = _data.at("scrub_geo").at("up_to_status_id"); | 204 | _user_id_and_tweet_id._tweet_id = json["scrub_geo"]["up_to_status_id"].get<tweet_id>(); |
198 | _user_id_and_tweet_id._user_id = _data.at("scrub_geo").at("user_id"); | 205 | _user_id_and_tweet_id._user_id = json["scrub_geo"]["user_id"].get<user_id>(); |
199 | } else if (_data.find("limit") != _data.end()) | 206 | } else if (!json["limit"].is_null()) |
200 | { | 207 | { |
201 | _type = type::limit; | 208 | _type = type::limit; |
202 | 209 | ||
203 | _limit = _data.at("limit").at("track"); | 210 | _limit = json["limit"]["track"].get<int>(); |
204 | } else if (_data.find("status_withheld") != _data.end()) | 211 | } else if (!json["status_withheld"].is_null()) |
205 | { | 212 | { |
206 | _type = type::withhold_status; | 213 | _type = type::withhold_status; |
207 | 214 | ||
208 | _withhold_status._user_id = _data.at("status_withheld").at("user_id"); | 215 | _withhold_status._user_id = json["status_withheld"]["user_id"].get<user_id>(); |
209 | _withhold_status._tweet_id = _data.at("status_withheld").at("id"); | 216 | _withhold_status._tweet_id = json["status_withheld"]["id"].get<tweet_id>(); |
210 | 217 | ||
211 | new(&_withhold_status._countries) std::vector<std::string>(); | 218 | new(&_withhold_status._countries) std::vector<std::string>(); |
212 | for (auto s : _data.at("status_withheld").at("withheld_in_countries")) | 219 | for (auto s : json["status_withheld"]["withheld_in_countries"]) |
213 | { | 220 | { |
214 | _withhold_status._countries.push_back(s); | 221 | _withhold_status._countries.push_back(s); |
215 | } | 222 | } |
216 | } else if (_data.find("user_withheld") != _data.end()) | 223 | } else if (!json["user_withheld"].is_null()) |
217 | { | 224 | { |
218 | _type = type::withhold_user; | 225 | _type = type::withhold_user; |
219 | 226 | ||
220 | _withhold_user._user_id = _data.at("user_withheld").at("id"); | 227 | _withhold_user._user_id = json["user_withheld"]["id"].get<user_id>(); |
221 | 228 | ||
222 | new(&_withhold_user._countries) std::vector<std::string>(); | 229 | new(&_withhold_user._countries) std::vector<std::string>(); |
223 | for (auto s : _data.at("user_withheld").at("withheld_in_countries")) | 230 | for (auto s : json["user_withheld"]["withheld_in_countries"]) |
224 | { | 231 | { |
225 | _withhold_user._countries.push_back(s); | 232 | _withhold_user._countries.push_back(s); |
226 | } | 233 | } |
227 | } else if (_data.find("disconnect") != _data.end()) | 234 | } else if (!json["disconnect"].is_null()) |
228 | { | 235 | { |
229 | _type = type::disconnect; | 236 | _type = type::disconnect; |
230 | 237 | ||
231 | switch (_data.at("disconnect").at("code").get<int>()) | 238 | switch (json["disconnect"]["code"].get<int>()) |
232 | { | 239 | { |
233 | case 1: _disconnect = disconnect_code::shutdown; break; | 240 | case 1: _disconnect = disconnect_code::shutdown; break; |
234 | case 2: _disconnect = disconnect_code::duplicate; break; | 241 | case 2: _disconnect = disconnect_code::duplicate; break; |
@@ -242,26 +249,26 @@ namespace twitter { | |||
242 | case 12: _disconnect = disconnect_code::load; break; | 249 | case 12: _disconnect = disconnect_code::load; break; |
243 | default: _disconnect = disconnect_code::unknown; | 250 | default: _disconnect = disconnect_code::unknown; |
244 | } | 251 | } |
245 | } else if (_data.find("friends") != _data.end()) | 252 | } else if (!json["friends"].is_null()) |
246 | { | 253 | { |
247 | _type = type::friends; | 254 | _type = type::friends; |
248 | 255 | ||
249 | new(&_friends) std::set<user_id>(_data.at("friends").begin(), _data.at("friends").end()); | 256 | new(&_friends) std::set<user_id>(std::begin(json["friends"]), std::end(json["friends"])); |
250 | } else if (_data.find("direct_message") != _data.end()) | 257 | } else if (!json["direct_message"].is_null()) |
251 | { | 258 | { |
252 | _type = type::direct; | 259 | _type = type::direct; |
253 | 260 | ||
254 | new(&_direct_message) direct_message(_data.at("direct_message").dump()); | 261 | new(&_direct_message) direct_message(json["direct_message"].dump()); |
255 | } else { | 262 | } else { |
256 | _type = type::unknown; | 263 | _type = type::unknown; |
257 | } | 264 | } |
258 | } catch (std::invalid_argument e) | 265 | } catch (const std::domain_error& error) |
259 | { | 266 | { |
260 | _type = type::invalid; | 267 | std::throw_with_nested(invalid_response(data)); |
261 | } | 268 | } |
262 | } | 269 | } |
263 | 270 | ||
264 | notification::notification(const notification& other) | 271 | notification::notification(notification&& other) |
265 | { | 272 | { |
266 | _type = other._type; | 273 | _type = other._type; |
267 | 274 | ||
@@ -269,11 +276,11 @@ namespace twitter { | |||
269 | { | 276 | { |
270 | case type::tweet: | 277 | case type::tweet: |
271 | { | 278 | { |
272 | new(&_tweet) tweet(other._tweet); | 279 | new(&_tweet) tweet(std::move(other._tweet)); |
273 | 280 | ||
274 | break; | 281 | break; |
275 | } | 282 | } |
276 | 283 | ||
277 | case type::update_user: | 284 | case type::update_user: |
278 | case type::block: | 285 | case type::block: |
279 | case type::unblock: | 286 | case type::unblock: |
@@ -281,32 +288,32 @@ namespace twitter { | |||
281 | case type::followed: | 288 | case type::followed: |
282 | case type::unfollow: | 289 | case type::unfollow: |
283 | { | 290 | { |
284 | new(&_user) user(other._user); | 291 | new(&_user) user(std::move(other._user)); |
285 | 292 | ||
286 | break; | 293 | break; |
287 | } | 294 | } |
288 | 295 | ||
289 | case type::favorite: | 296 | case type::favorite: |
290 | case type::favorited: | 297 | case type::favorited: |
291 | case type::unfavorite: | 298 | case type::unfavorite: |
292 | case type::unfavorited: | 299 | case type::unfavorited: |
293 | case type::quoted: | 300 | case type::quoted: |
294 | { | 301 | { |
295 | new(&_user_and_tweet._user) user(other._user_and_tweet._user); | 302 | new(&_user_and_tweet._user) user(std::move(other._user_and_tweet._user)); |
296 | new(&_user_and_tweet._tweet) tweet(other._user_and_tweet._tweet); | 303 | new(&_user_and_tweet._tweet) tweet(std::move(other._user_and_tweet._tweet)); |
297 | 304 | ||
298 | break; | 305 | break; |
299 | } | 306 | } |
300 | 307 | ||
301 | case type::list_created: | 308 | case type::list_created: |
302 | case type::list_destroyed: | 309 | case type::list_destroyed: |
303 | case type::list_updated: | 310 | case type::list_updated: |
304 | { | 311 | { |
305 | new(&_list) list(other._list); | 312 | new(&_list) list(std::move(other._list)); |
306 | 313 | ||
307 | break; | 314 | break; |
308 | } | 315 | } |
309 | 316 | ||
310 | case type::list_add: | 317 | case type::list_add: |
311 | case type::list_added: | 318 | case type::list_added: |
312 | case type::list_remove: | 319 | case type::list_remove: |
@@ -316,78 +323,84 @@ namespace twitter { | |||
316 | case type::list_unsubscribe: | 323 | case type::list_unsubscribe: |
317 | case type::list_unsubscribed: | 324 | case type::list_unsubscribed: |
318 | { | 325 | { |
319 | new(&_user_and_list._user) user(other._user_and_list._user); | 326 | new(&_user_and_list._user) user(std::move(other._user_and_list._user)); |
320 | new(&_user_and_list._list) list(other._user_and_list._list); | 327 | new(&_user_and_list._list) list(std::move(other._user_and_list._list)); |
321 | 328 | ||
322 | break; | 329 | break; |
323 | } | 330 | } |
324 | 331 | ||
325 | case type::stall: | 332 | case type::stall: |
326 | case type::follow_limit: | 333 | case type::follow_limit: |
327 | case type::unknown_warning: | 334 | case type::unknown_warning: |
328 | { | 335 | { |
329 | new(&_warning) std::string(other._warning); | 336 | new(&_warning) std::string(std::move(other._warning)); |
330 | 337 | ||
331 | break; | 338 | break; |
332 | } | 339 | } |
333 | 340 | ||
334 | case type::deletion: | 341 | case type::deletion: |
335 | case type::scrub_location: | 342 | case type::scrub_location: |
336 | { | 343 | { |
337 | _user_id_and_tweet_id._user_id = other._user_id_and_tweet_id._user_id; | 344 | _user_id_and_tweet_id._user_id = other._user_id_and_tweet_id._user_id; |
338 | _user_id_and_tweet_id._tweet_id = other._user_id_and_tweet_id._tweet_id; | 345 | _user_id_and_tweet_id._tweet_id = other._user_id_and_tweet_id._tweet_id; |
339 | 346 | ||
340 | break; | 347 | break; |
341 | } | 348 | } |
342 | 349 | ||
343 | case type::limit: | 350 | case type::limit: |
344 | { | 351 | { |
345 | _limit = other._limit; | 352 | _limit = other._limit; |
346 | 353 | ||
347 | break; | 354 | break; |
348 | } | 355 | } |
349 | 356 | ||
350 | case type::withhold_status: | 357 | case type::withhold_status: |
351 | { | 358 | { |
352 | _withhold_status._user_id = other._withhold_status._user_id; | 359 | _withhold_status._user_id = other._withhold_status._user_id; |
353 | _withhold_status._tweet_id = other._withhold_status._tweet_id; | 360 | _withhold_status._tweet_id = other._withhold_status._tweet_id; |
354 | new(&_withhold_status._countries) std::vector<std::string>(other._withhold_status._countries); | 361 | new(&_withhold_status._countries) std::vector<std::string>(std::move(other._withhold_status._countries)); |
355 | 362 | ||
356 | break; | 363 | break; |
357 | } | 364 | } |
358 | 365 | ||
359 | case type::withhold_user: | 366 | case type::withhold_user: |
360 | { | 367 | { |
361 | _withhold_user._user_id = other._withhold_user._user_id; | 368 | _withhold_user._user_id = other._withhold_user._user_id; |
362 | new(&_withhold_user._countries) std::vector<std::string>(other._withhold_user._countries); | 369 | new(&_withhold_user._countries) std::vector<std::string>(std::move(other._withhold_user._countries)); |
363 | 370 | ||
364 | break; | 371 | break; |
365 | } | 372 | } |
366 | 373 | ||
367 | case type::disconnect: | 374 | case type::disconnect: |
368 | { | 375 | { |
369 | _disconnect = other._disconnect; | 376 | _disconnect = other._disconnect; |
370 | 377 | ||
371 | break; | 378 | break; |
372 | } | 379 | } |
373 | 380 | ||
374 | case type::friends: | 381 | case type::friends: |
375 | { | 382 | { |
376 | new(&_friends) std::set<user_id>(other._friends); | 383 | new(&_friends) std::set<user_id>(std::move(other._friends)); |
377 | 384 | ||
378 | break; | 385 | break; |
379 | } | 386 | } |
380 | 387 | ||
381 | case type::direct: | 388 | case type::direct: |
382 | { | 389 | { |
383 | new(&_direct_message) direct_message(other._direct_message); | 390 | new(&_direct_message) direct_message(std::move(other._direct_message)); |
384 | 391 | ||
392 | break; | ||
393 | } | ||
394 | |||
395 | case type::unknown: | ||
396 | case type::invalid: | ||
397 | { | ||
385 | break; | 398 | break; |
386 | } | 399 | } |
387 | } | 400 | } |
388 | } | 401 | } |
389 | 402 | ||
390 | notification& notification::operator=(const notification& other) | 403 | notification& notification::operator=(notification&& other) |
391 | { | 404 | { |
392 | this->~notification(); | 405 | this->~notification(); |
393 | 406 | ||
@@ -397,7 +410,7 @@ namespace twitter { | |||
397 | { | 410 | { |
398 | case type::tweet: | 411 | case type::tweet: |
399 | { | 412 | { |
400 | new(&_tweet) tweet(other._tweet); | 413 | new(&_tweet) tweet(std::move(other._tweet)); |
401 | 414 | ||
402 | break; | 415 | break; |
403 | } | 416 | } |
@@ -409,7 +422,7 @@ namespace twitter { | |||
409 | case type::followed: | 422 | case type::followed: |
410 | case type::unfollow: | 423 | case type::unfollow: |
411 | { | 424 | { |
412 | new(&_user) user(other._user); | 425 | new(&_user) user(std::move(other._user)); |
413 | 426 | ||
414 | break; | 427 | break; |
415 | } | 428 | } |
@@ -420,8 +433,8 @@ namespace twitter { | |||
420 | case type::unfavorited: | 433 | case type::unfavorited: |
421 | case type::quoted: | 434 | case type::quoted: |
422 | { | 435 | { |
423 | new(&_user_and_tweet._user) user(other._user_and_tweet._user); | 436 | new(&_user_and_tweet._user) user(std::move(other._user_and_tweet._user)); |
424 | new(&_user_and_tweet._tweet) tweet(other._user_and_tweet._tweet); | 437 | new(&_user_and_tweet._tweet) tweet(std::move(other._user_and_tweet._tweet)); |
425 | 438 | ||
426 | break; | 439 | break; |
427 | } | 440 | } |
@@ -430,7 +443,7 @@ namespace twitter { | |||
430 | case type::list_destroyed: | 443 | case type::list_destroyed: |
431 | case type::list_updated: | 444 | case type::list_updated: |
432 | { | 445 | { |
433 | new(&_list) list(other._list); | 446 | new(&_list) list(std::move(other._list)); |
434 | 447 | ||
435 | break; | 448 | break; |
436 | } | 449 | } |
@@ -444,8 +457,8 @@ namespace twitter { | |||
444 | case type::list_unsubscribe: | 457 | case type::list_unsubscribe: |
445 | case type::list_unsubscribed: | 458 | case type::list_unsubscribed: |
446 | { | 459 | { |
447 | new(&_user_and_list._user) user(other._user_and_list._user); | 460 | new(&_user_and_list._user) user(std::move(other._user_and_list._user)); |
448 | new(&_user_and_list._list) list(other._user_and_list._list); | 461 | new(&_user_and_list._list) list(std::move(other._user_and_list._list)); |
449 | 462 | ||
450 | break; | 463 | break; |
451 | } | 464 | } |
@@ -454,7 +467,7 @@ namespace twitter { | |||
454 | case type::follow_limit: | 467 | case type::follow_limit: |
455 | case type::unknown_warning: | 468 | case type::unknown_warning: |
456 | { | 469 | { |
457 | new(&_warning) std::string(other._warning); | 470 | new(&_warning) std::string(std::move(other._warning)); |
458 | 471 | ||
459 | break; | 472 | break; |
460 | } | 473 | } |
@@ -479,7 +492,7 @@ namespace twitter { | |||
479 | { | 492 | { |
480 | _withhold_status._user_id = other._withhold_status._user_id; | 493 | _withhold_status._user_id = other._withhold_status._user_id; |
481 | _withhold_status._tweet_id = other._withhold_status._tweet_id; | 494 | _withhold_status._tweet_id = other._withhold_status._tweet_id; |
482 | new(&_withhold_status._countries) std::vector<std::string>(other._withhold_status._countries); | 495 | new(&_withhold_status._countries) std::vector<std::string>(std::move(other._withhold_status._countries)); |
483 | 496 | ||
484 | break; | 497 | break; |
485 | } | 498 | } |
@@ -487,7 +500,7 @@ namespace twitter { | |||
487 | case type::withhold_user: | 500 | case type::withhold_user: |
488 | { | 501 | { |
489 | _withhold_user._user_id = other._withhold_user._user_id; | 502 | _withhold_user._user_id = other._withhold_user._user_id; |
490 | new(&_withhold_user._countries) std::vector<std::string>(other._withhold_user._countries); | 503 | new(&_withhold_user._countries) std::vector<std::string>(std::move(other._withhold_user._countries)); |
491 | 504 | ||
492 | break; | 505 | break; |
493 | } | 506 | } |
@@ -501,17 +514,23 @@ namespace twitter { | |||
501 | 514 | ||
502 | case type::friends: | 515 | case type::friends: |
503 | { | 516 | { |
504 | new(&_friends) std::set<user_id>(other._friends); | 517 | new(&_friends) std::set<user_id>(std::move(other._friends)); |
505 | 518 | ||
506 | break; | 519 | break; |
507 | } | 520 | } |
508 | 521 | ||
509 | case type::direct: | 522 | case type::direct: |
510 | { | 523 | { |
511 | new(&_direct_message) direct_message(other._direct_message); | 524 | new(&_direct_message) direct_message(std::move(other._direct_message)); |
512 | 525 | ||
513 | break; | 526 | break; |
514 | } | 527 | } |
528 | |||
529 | case type::invalid: | ||
530 | case type::unknown: | ||
531 | { | ||
532 | break; | ||
533 | } | ||
515 | } | 534 | } |
516 | 535 | ||
517 | return *this; | 536 | return *this; |
@@ -578,6 +597,7 @@ namespace twitter { | |||
578 | 597 | ||
579 | case type::stall: | 598 | case type::stall: |
580 | case type::follow_limit: | 599 | case type::follow_limit: |
600 | case type::unknown_warning: | ||
581 | { | 601 | { |
582 | using string_type = std::string; | 602 | using string_type = std::string; |
583 | _warning.~string_type(); | 603 | _warning.~string_type(); |
@@ -615,10 +635,20 @@ namespace twitter { | |||
615 | 635 | ||
616 | break; | 636 | break; |
617 | } | 637 | } |
638 | |||
639 | case type::deletion: | ||
640 | case type::scrub_location: | ||
641 | case type::limit: | ||
642 | case type::disconnect: | ||
643 | case type::unknown: | ||
644 | case type::invalid: | ||
645 | { | ||
646 | break; | ||
647 | } | ||
618 | } | 648 | } |
619 | } | 649 | } |
620 | 650 | ||
621 | tweet notification::getTweet() const | 651 | const tweet& notification::getTweet() const |
622 | { | 652 | { |
623 | switch (_type) | 653 | switch (_type) |
624 | { | 654 | { |
@@ -639,13 +669,11 @@ namespace twitter { | |||
639 | default: | 669 | default: |
640 | { | 670 | { |
641 | assert(false); | 671 | assert(false); |
642 | |||
643 | return tweet(); | ||
644 | } | 672 | } |
645 | } | 673 | } |
646 | } | 674 | } |
647 | 675 | ||
648 | user notification::getUser() const | 676 | const user& notification::getUser() const |
649 | { | 677 | { |
650 | switch (_type) | 678 | switch (_type) |
651 | { | 679 | { |
@@ -683,13 +711,11 @@ namespace twitter { | |||
683 | default: | 711 | default: |
684 | { | 712 | { |
685 | assert(false); | 713 | assert(false); |
686 | |||
687 | return user(); | ||
688 | } | 714 | } |
689 | } | 715 | } |
690 | } | 716 | } |
691 | 717 | ||
692 | list notification::getList() const | 718 | const list& notification::getList() const |
693 | { | 719 | { |
694 | switch (_type) | 720 | switch (_type) |
695 | { | 721 | { |
@@ -715,8 +741,6 @@ namespace twitter { | |||
715 | default: | 741 | default: |
716 | { | 742 | { |
717 | assert(false); | 743 | assert(false); |
718 | |||
719 | return list(); | ||
720 | } | 744 | } |
721 | } | 745 | } |
722 | } | 746 | } |
@@ -739,8 +763,6 @@ namespace twitter { | |||
739 | default: | 763 | default: |
740 | { | 764 | { |
741 | assert(false); | 765 | assert(false); |
742 | |||
743 | return 0; | ||
744 | } | 766 | } |
745 | } | 767 | } |
746 | } | 768 | } |
@@ -768,13 +790,11 @@ namespace twitter { | |||
768 | default: | 790 | default: |
769 | { | 791 | { |
770 | assert(false); | 792 | assert(false); |
771 | |||
772 | return 0; | ||
773 | } | 793 | } |
774 | } | 794 | } |
775 | } | 795 | } |
776 | 796 | ||
777 | std::vector<std::string> notification::getCountries() const | 797 | const std::vector<std::string>& notification::getCountries() const |
778 | { | 798 | { |
779 | switch (_type) | 799 | switch (_type) |
780 | { | 800 | { |
@@ -791,8 +811,6 @@ namespace twitter { | |||
791 | default: | 811 | default: |
792 | { | 812 | { |
793 | assert(false); | 813 | assert(false); |
794 | |||
795 | return std::vector<std::string>(); | ||
796 | } | 814 | } |
797 | } | 815 | } |
798 | } | 816 | } |
@@ -804,14 +822,14 @@ namespace twitter { | |||
804 | return _disconnect; | 822 | return _disconnect; |
805 | } | 823 | } |
806 | 824 | ||
807 | std::set<user_id> notification::getFriends() const | 825 | const std::set<user_id>& notification::getFriends() const |
808 | { | 826 | { |
809 | assert(_type == type::friends); | 827 | assert(_type == type::friends); |
810 | 828 | ||
811 | return _friends; | 829 | return _friends; |
812 | } | 830 | } |
813 | 831 | ||
814 | direct_message notification::getDirectMessage() const | 832 | const direct_message& notification::getDirectMessage() const |
815 | { | 833 | { |
816 | assert(_type == type::direct); | 834 | assert(_type == type::direct); |
817 | 835 | ||
@@ -825,7 +843,7 @@ namespace twitter { | |||
825 | return _limit; | 843 | return _limit; |
826 | } | 844 | } |
827 | 845 | ||
828 | std::string notification::getWarning() const | 846 | const std::string& notification::getWarning() const |
829 | { | 847 | { |
830 | switch (_type) | 848 | switch (_type) |
831 | { | 849 | { |
@@ -839,15 +857,8 @@ namespace twitter { | |||
839 | default: | 857 | default: |
840 | { | 858 | { |
841 | assert(false); | 859 | assert(false); |
842 | |||
843 | return ""; | ||
844 | } | 860 | } |
845 | } | 861 | } |
846 | } | 862 | } |
847 | 863 | ||
848 | notification::operator bool() const | ||
849 | { | ||
850 | return _type != type::invalid; | ||
851 | } | ||
852 | |||
853 | }; | 864 | }; |
diff --git a/src/notification.h b/src/notification.h index da83b0f..b5ecd49 100644 --- a/src/notification.h +++ b/src/notification.h | |||
@@ -11,7 +11,10 @@ | |||
11 | 11 | ||
12 | namespace twitter { | 12 | namespace twitter { |
13 | 13 | ||
14 | enum class disconnect_code { | 14 | class client; |
15 | |||
16 | enum class disconnect_code | ||
17 | { | ||
15 | shutdown, | 18 | shutdown, |
16 | duplicate, | 19 | duplicate, |
17 | stall, | 20 | stall, |
@@ -26,118 +29,118 @@ namespace twitter { | |||
26 | }; | 29 | }; |
27 | 30 | ||
28 | class notification { | 31 | class notification { |
29 | public: | 32 | public: |
30 | enum class type { | 33 | enum class type { |
31 | // Tweet object | 34 | // Tweet object |
32 | tweet, | 35 | tweet, |
33 | 36 | ||
34 | // User object | 37 | // User object |
35 | update_user, | 38 | update_user, |
36 | block, | 39 | block, |
37 | unblock, | 40 | unblock, |
38 | follow, | 41 | follow, |
39 | followed, | 42 | followed, |
40 | unfollow, | 43 | unfollow, |
41 | 44 | ||
42 | // User and tweet | 45 | // User and tweet |
43 | favorite, | 46 | favorite, |
44 | favorited, | 47 | favorited, |
45 | unfavorite, | 48 | unfavorite, |
46 | unfavorited, | 49 | unfavorited, |
47 | quoted, | 50 | quoted, |
48 | |||
49 | // List | ||
50 | list_created, | ||
51 | list_destroyed, | ||
52 | list_updated, | ||
53 | |||
54 | // User and list | ||
55 | list_add, | ||
56 | list_added, | ||
57 | list_remove, | ||
58 | list_removed, | ||
59 | list_subscribe, | ||
60 | list_subscribed, | ||
61 | list_unsubscribe, | ||
62 | list_unsubscribed, | ||
63 | |||
64 | // Warning | ||
65 | stall, | ||
66 | follow_limit, | ||
67 | unknown_warning, | ||
68 | |||
69 | // User ID and tweet ID | ||
70 | deletion, | ||
71 | scrub_location, | ||
72 | |||
73 | // Special | ||
74 | limit, | ||
75 | withhold_status, | ||
76 | withhold_user, | ||
77 | disconnect, | ||
78 | friends, | ||
79 | direct, | ||
80 | |||
81 | // Nothing | ||
82 | unknown, | ||
83 | invalid | ||
84 | }; | ||
85 | 51 | ||
86 | type getType() const; | 52 | // List |
53 | list_created, | ||
54 | list_destroyed, | ||
55 | list_updated, | ||
87 | 56 | ||
88 | notification(); | 57 | // User and list |
89 | notification(std::string data, const user& current_user); | 58 | list_add, |
90 | notification(const notification& other); | 59 | list_added, |
91 | notification& operator=(const notification& other); | 60 | list_remove, |
92 | ~notification(); | 61 | list_removed, |
62 | list_subscribe, | ||
63 | list_subscribed, | ||
64 | list_unsubscribe, | ||
65 | list_unsubscribed, | ||
93 | 66 | ||
94 | tweet getTweet() const; | 67 | // Warning |
95 | user getUser() const; | 68 | stall, |
96 | list getList() const; | 69 | follow_limit, |
97 | tweet_id getTweetID() const; | 70 | unknown_warning, |
98 | user_id getUserID() const; | ||
99 | std::vector<std::string> getCountries() const; | ||
100 | disconnect_code getDisconnectCode() const; | ||
101 | std::set<user_id> getFriends() const; | ||
102 | direct_message getDirectMessage() const; | ||
103 | int getLimit() const; | ||
104 | std::string getWarning() const; | ||
105 | 71 | ||
106 | operator bool() const; | 72 | // User ID and tweet ID |
73 | deletion, | ||
74 | scrub_location, | ||
107 | 75 | ||
108 | private: | 76 | // Special |
109 | union { | 77 | limit, |
78 | withhold_status, | ||
79 | withhold_user, | ||
80 | disconnect, | ||
81 | friends, | ||
82 | direct, | ||
83 | |||
84 | // Nothing | ||
85 | unknown, | ||
86 | invalid | ||
87 | }; | ||
88 | |||
89 | type getType() const; | ||
90 | |||
91 | notification(const client& tclient, std::string data); | ||
92 | notification(notification&& other); | ||
93 | notification& operator=(notification&& other); | ||
94 | ~notification(); | ||
95 | |||
96 | notification(const notification& other) = delete; | ||
97 | notification& operator=(const notification& other) = delete; | ||
98 | |||
99 | const tweet& getTweet() const; | ||
100 | const user& getUser() const; | ||
101 | const list& getList() const; | ||
102 | tweet_id getTweetID() const; | ||
103 | user_id getUserID() const; | ||
104 | const std::vector<std::string>& getCountries() const; | ||
105 | disconnect_code getDisconnectCode() const; | ||
106 | const std::set<user_id>& getFriends() const; | ||
107 | const direct_message& getDirectMessage() const; | ||
108 | int getLimit() const; | ||
109 | const std::string& getWarning() const; | ||
110 | |||
111 | private: | ||
112 | union { | ||
113 | tweet _tweet; | ||
114 | user _user; | ||
115 | list _list; | ||
116 | struct { | ||
117 | user _user; | ||
110 | tweet _tweet; | 118 | tweet _tweet; |
119 | } _user_and_tweet; | ||
120 | struct { | ||
111 | user _user; | 121 | user _user; |
112 | list _list; | 122 | list _list; |
113 | struct { | 123 | } _user_and_list; |
114 | user _user; | 124 | std::string _warning; |
115 | tweet _tweet; | 125 | struct { |
116 | } _user_and_tweet; | 126 | user_id _user_id; |
117 | struct { | 127 | tweet_id _tweet_id; |
118 | user _user; | 128 | } _user_id_and_tweet_id; |
119 | list _list; | 129 | int _limit; |
120 | } _user_and_list; | 130 | struct { |
121 | std::string _warning; | 131 | user_id _user_id; |
122 | struct { | 132 | tweet_id _tweet_id; |
123 | user_id _user_id; | 133 | std::vector<std::string> _countries; |
124 | tweet_id _tweet_id; | 134 | } _withhold_status; |
125 | } _user_id_and_tweet_id; | 135 | struct { |
126 | int _limit; | 136 | user_id _user_id; |
127 | struct { | 137 | std::vector<std::string> _countries; |
128 | user_id _user_id; | 138 | } _withhold_user; |
129 | tweet_id _tweet_id; | 139 | disconnect_code _disconnect; |
130 | std::vector<std::string> _countries; | 140 | std::set<user_id> _friends; |
131 | } _withhold_status; | 141 | direct_message _direct_message; |
132 | struct { | 142 | }; |
133 | user_id _user_id; | 143 | type _type; |
134 | std::vector<std::string> _countries; | ||
135 | } _withhold_user; | ||
136 | disconnect_code _disconnect; | ||
137 | std::set<user_id> _friends; | ||
138 | direct_message _direct_message; | ||
139 | }; | ||
140 | type _type; | ||
141 | }; | 144 | }; |
142 | 145 | ||
143 | }; | 146 | }; |
diff --git a/src/stream.cpp b/src/stream.cpp new file mode 100644 index 0000000..d2b3afe --- /dev/null +++ b/src/stream.cpp | |||
@@ -0,0 +1,291 @@ | |||
1 | #include "stream.h" | ||
2 | #include <liboauthcpp/liboauthcpp.h> | ||
3 | #include <curl_easy.h> | ||
4 | #include <curl_header.h> | ||
5 | #include "util.h" | ||
6 | #include "notification.h" | ||
7 | #include "client.h" | ||
8 | |||
9 | namespace twitter { | ||
10 | |||
11 | stream::stream( | ||
12 | const client& tclient, | ||
13 | notify_callback callback, | ||
14 | bool with_followings, | ||
15 | bool receive_all_replies, | ||
16 | std::list<std::string> track, | ||
17 | std::list<bounding_box> locations) : | ||
18 | _client(tclient), | ||
19 | _notify(callback), | ||
20 | _thread(&stream::run, this, generateUrl(with_followings, receive_all_replies, track, locations)) | ||
21 | { | ||
22 | } | ||
23 | |||
24 | stream::~stream() | ||
25 | { | ||
26 | if (_thread.joinable()) | ||
27 | { | ||
28 | _stop = true; | ||
29 | _thread.join(); | ||
30 | } | ||
31 | } | ||
32 | |||
33 | std::string stream::generateUrl( | ||
34 | bool with_followings, | ||
35 | bool receive_all_replies, | ||
36 | std::list<std::string> track, | ||
37 | std::list<bounding_box> locations) | ||
38 | { | ||
39 | std::list<std::string> arguments; | ||
40 | |||
41 | if (receive_all_replies) | ||
42 | { | ||
43 | arguments.push_back("replies=all"); | ||
44 | } | ||
45 | |||
46 | if (!with_followings) | ||
47 | { | ||
48 | arguments.push_back("with=user"); | ||
49 | } | ||
50 | |||
51 | if (!track.empty()) | ||
52 | { | ||
53 | std::ostringstream trackstr; | ||
54 | trackstr << "track="; | ||
55 | |||
56 | for (auto it = std::begin(track); it != std::end(track); it++) | ||
57 | { | ||
58 | if (it != std::begin(track)) | ||
59 | { | ||
60 | trackstr << ","; | ||
61 | } | ||
62 | |||
63 | trackstr << OAuth::HttpEncodeQueryValue(*it); | ||
64 | } | ||
65 | |||
66 | arguments.push_back(trackstr.str()); | ||
67 | } | ||
68 | |||
69 | if (!locations.empty()) | ||
70 | { | ||
71 | std::ostringstream localstr; | ||
72 | localstr << "locations="; | ||
73 | |||
74 | for (auto it = std::begin(locations); it != std::end(locations); it++) | ||
75 | { | ||
76 | if (it != std::begin(locations)) | ||
77 | { | ||
78 | localstr << ","; | ||
79 | } | ||
80 | |||
81 | localstr << (double)it->getSouthWestLongitude() << ","; | ||
82 | localstr << (double)it->getSouthWestLatitude() << ","; | ||
83 | localstr << (double)it->getNorthEastLongitude() << ","; | ||
84 | localstr << (double)it->getNorthEastLatitude(); | ||
85 | } | ||
86 | |||
87 | arguments.push_back(localstr.str()); | ||
88 | } | ||
89 | |||
90 | std::ostringstream urlstr; | ||
91 | urlstr << "https://userstream.twitter.com/1.1/user.json"; | ||
92 | |||
93 | if (!arguments.empty()) | ||
94 | { | ||
95 | urlstr << "?"; | ||
96 | urlstr << implode(std::begin(arguments), std::end(arguments), "&"); | ||
97 | } | ||
98 | |||
99 | return urlstr.str(); | ||
100 | } | ||
101 | |||
102 | void stream::run(std::string url) | ||
103 | { | ||
104 | _backoff_type = backoff::none; | ||
105 | _backoff_amount = std::chrono::milliseconds(0); | ||
106 | for (;;) | ||
107 | { | ||
108 | curl::curl_ios<stream> ios(this, [] (void* contents, size_t size, size_t nmemb, void* userp) { | ||
109 | return static_cast<stream*>(userp)->write(static_cast<char*>(contents), size, nmemb); | ||
110 | }); | ||
111 | |||
112 | curl::curl_easy conn(ios); | ||
113 | curl::curl_header headers; | ||
114 | std::string oauth_header; | ||
115 | |||
116 | try | ||
117 | { | ||
118 | oauth_header = _client._oauth_client->getFormattedHttpHeader(OAuth::Http::Get, url, ""); | ||
119 | |||
120 | if (!oauth_header.empty()) | ||
121 | { | ||
122 | headers.add(oauth_header); | ||
123 | } | ||
124 | } catch (const OAuth::ParseError& error) | ||
125 | { | ||
126 | std::cout << "Error generating OAuth header:" << std::endl; | ||
127 | std::cout << error.what() << std::endl; | ||
128 | std::cout << "This is likely due to a malformed URL." << std::endl; | ||
129 | |||
130 | assert(false); | ||
131 | } | ||
132 | |||
133 | try | ||
134 | { | ||
135 | conn.add<CURLOPT_HEADERFUNCTION>(nullptr); | ||
136 | conn.add<CURLOPT_HEADERDATA>(nullptr); | ||
137 | conn.add<CURLOPT_XFERINFOFUNCTION>([] (void* cdata, curl_off_t, curl_off_t, curl_off_t, curl_off_t) { | ||
138 | return static_cast<stream*>(cdata)->progress(); | ||
139 | }); | ||
140 | conn.add<CURLOPT_XFERINFODATA>(this); | ||
141 | conn.add<CURLOPT_NOPROGRESS>(0); | ||
142 | //conn.add<CURLOPT_VERBOSE>(1); | ||
143 | //conn.add<CURLOPT_DEBUGFUNCTION>(my_trace); | ||
144 | conn.add<CURLOPT_URL>(url.c_str()); | ||
145 | conn.add<CURLOPT_HTTPHEADER>(headers.get()); | ||
146 | } catch (const curl::curl_exception& error) | ||
147 | { | ||
148 | error.print_traceback(); | ||
149 | |||
150 | assert(false); | ||
151 | } | ||
152 | |||
153 | bool failure = false; | ||
154 | try | ||
155 | { | ||
156 | conn.perform(); | ||
157 | } catch (const curl::curl_easy_exception& error) | ||
158 | { | ||
159 | failure = true; | ||
160 | if ((error.get_code() == CURLE_ABORTED_BY_CALLBACK) && _stop) | ||
161 | { | ||
162 | break; | ||
163 | } else { | ||
164 | if (_backoff_type == backoff::none) | ||
165 | { | ||
166 | _established = false; | ||
167 | _backoff_type = backoff::network; | ||
168 | _backoff_amount = std::chrono::milliseconds(0); | ||
169 | } | ||
170 | } | ||
171 | } | ||
172 | |||
173 | if (!failure) | ||
174 | { | ||
175 | long response_code = conn.get_info<CURLINFO_RESPONSE_CODE>().get(); | ||
176 | if (response_code == 420) | ||
177 | { | ||
178 | if (_backoff_type == backoff::none) | ||
179 | { | ||
180 | _established = false; | ||
181 | _backoff_type = backoff::rate_limit; | ||
182 | _backoff_amount = std::chrono::minutes(1); | ||
183 | } | ||
184 | } else if (response_code != 200) | ||
185 | { | ||
186 | if (_backoff_type == backoff::none) | ||
187 | { | ||
188 | _established = false; | ||
189 | _backoff_type = backoff::http; | ||
190 | _backoff_amount = std::chrono::seconds(5); | ||
191 | } | ||
192 | } else { | ||
193 | if (_backoff_type == backoff::none) | ||
194 | { | ||
195 | _established = false; | ||
196 | _backoff_type = backoff::network; | ||
197 | _backoff_amount = std::chrono::milliseconds(0); | ||
198 | } | ||
199 | } | ||
200 | } | ||
201 | |||
202 | std::this_thread::sleep_for(_backoff_amount); | ||
203 | |||
204 | switch (_backoff_type) | ||
205 | { | ||
206 | case backoff::network: | ||
207 | { | ||
208 | if (_backoff_amount < std::chrono::seconds(16)) | ||
209 | { | ||
210 | _backoff_amount += std::chrono::milliseconds(250); | ||
211 | } | ||
212 | |||
213 | break; | ||
214 | } | ||
215 | |||
216 | case backoff::http: | ||
217 | { | ||
218 | if (_backoff_amount < std::chrono::seconds(320)) | ||
219 | { | ||
220 | _backoff_amount *= 2; | ||
221 | } | ||
222 | |||
223 | break; | ||
224 | } | ||
225 | |||
226 | case backoff::rate_limit: | ||
227 | { | ||
228 | _backoff_amount *= 2; | ||
229 | |||
230 | break; | ||
231 | } | ||
232 | |||
233 | case backoff::none: | ||
234 | { | ||
235 | break; | ||
236 | } | ||
237 | } | ||
238 | } | ||
239 | } | ||
240 | |||
241 | size_t stream::write(char* ptr, size_t size, size_t nmemb) | ||
242 | { | ||
243 | for (size_t i = 0; i < size*nmemb; i++) | ||
244 | { | ||
245 | if (ptr[i] == '\r') | ||
246 | { | ||
247 | i++; // Skip the \n | ||
248 | |||
249 | if (!_buffer.empty()) | ||
250 | { | ||
251 | notification n(_client, _buffer); | ||
252 | if (n.getType() == notification::type::friends) | ||
253 | { | ||
254 | _established = true; | ||
255 | _backoff_type = backoff::none; | ||
256 | _backoff_amount = std::chrono::milliseconds(0); | ||
257 | } | ||
258 | |||
259 | _notify(n); | ||
260 | |||
261 | _buffer = ""; | ||
262 | } | ||
263 | } else { | ||
264 | _buffer.push_back(ptr[i]); | ||
265 | } | ||
266 | } | ||
267 | |||
268 | time(&_last_write); | ||
269 | |||
270 | return size*nmemb; | ||
271 | } | ||
272 | |||
273 | int stream::progress() | ||
274 | { | ||
275 | if (_stop) | ||
276 | { | ||
277 | return 1; | ||
278 | } | ||
279 | |||
280 | if (_established) | ||
281 | { | ||
282 | if (difftime(time(NULL), _last_write) >= 90) | ||
283 | { | ||
284 | return 1; | ||
285 | } | ||
286 | } | ||
287 | |||
288 | return 0; | ||
289 | } | ||
290 | |||
291 | } | ||
diff --git a/src/stream.h b/src/stream.h new file mode 100644 index 0000000..3391912 --- /dev/null +++ b/src/stream.h | |||
@@ -0,0 +1,67 @@ | |||
1 | #ifndef STREAM_H_E9146952 | ||
2 | #define STREAM_H_E9146952 | ||
3 | |||
4 | #include <functional> | ||
5 | #include <list> | ||
6 | #include <string> | ||
7 | #include <chrono> | ||
8 | #include <thread> | ||
9 | #include "bounding_box.h" | ||
10 | |||
11 | namespace twitter { | ||
12 | |||
13 | class client; | ||
14 | class notification; | ||
15 | |||
16 | class stream { | ||
17 | public: | ||
18 | |||
19 | typedef std::function<void(const notification& _notification)> notify_callback; | ||
20 | |||
21 | stream( | ||
22 | const client& tclient, | ||
23 | notify_callback callback, | ||
24 | bool with_followings = true, | ||
25 | bool receive_all_replies = false, | ||
26 | std::list<std::string> track = {}, | ||
27 | std::list<bounding_box> locations = {}); | ||
28 | |||
29 | ~stream(); | ||
30 | |||
31 | stream(const stream& other) = delete; | ||
32 | stream(stream&& other) = delete; | ||
33 | stream& operator=(const stream& other) = delete; | ||
34 | stream& operator=(stream&& other) = delete; | ||
35 | |||
36 | private: | ||
37 | enum class backoff { | ||
38 | none, | ||
39 | network, | ||
40 | http, | ||
41 | rate_limit | ||
42 | }; | ||
43 | |||
44 | static std::string generateUrl( | ||
45 | bool with_followings, | ||
46 | bool receive_all_replies, | ||
47 | std::list<std::string> track, | ||
48 | std::list<bounding_box> locations); | ||
49 | |||
50 | void run(std::string url); | ||
51 | int progress(); | ||
52 | size_t write(char* ptr, size_t size, size_t nmemb); | ||
53 | |||
54 | const client& _client; | ||
55 | notify_callback _notify; | ||
56 | bool _stop = false; | ||
57 | std::string _buffer; | ||
58 | time_t _last_write; | ||
59 | bool _established = false; | ||
60 | backoff _backoff_type = backoff::none; | ||
61 | std::chrono::milliseconds _backoff_amount; | ||
62 | std::thread _thread; | ||
63 | }; | ||
64 | |||
65 | } | ||
66 | |||
67 | #endif /* end of include guard: STREAM_H_E9146952 */ | ||
diff --git a/src/tweet.cpp b/src/tweet.cpp index 193df6e..4abe2f8 100644 --- a/src/tweet.cpp +++ b/src/tweet.cpp | |||
@@ -1,139 +1,80 @@ | |||
1 | #include "tweet.h" | 1 | #include "tweet.h" |
2 | #include <json.hpp> | 2 | #include <json.hpp> |
3 | #include <cassert> | 3 | #include "util.h" |
4 | 4 | #include "codes.h" | |
5 | using nlohmann::json; | 5 | #include "client.h" |
6 | 6 | ||
7 | namespace twitter { | 7 | namespace twitter { |
8 | 8 | ||
9 | tweet::tweet() : _valid(false) | 9 | tweet::tweet(const client& tclient, std::string data) try |
10 | { | 10 | : _client(tclient) |
11 | |||
12 | } | ||
13 | |||
14 | tweet::tweet(std::string data) : _valid(true) | ||
15 | { | 11 | { |
16 | auto _data = json::parse(data); | 12 | auto json = nlohmann::json::parse(data); |
17 | _id = _data.at("id"); | 13 | _id = json["id"].get<tweet_id>(); |
18 | _text = _data.at("text"); | 14 | _text = json["text"].get<std::string>(); |
19 | _author = user(_data.at("user").dump()); | 15 | _author = make_unique<user>(_client, json["user"].dump()); |
20 | 16 | ||
21 | if (_data.find("retweeted_status") != _data.end()) | 17 | if (!json["retweeted_status"].is_null()) |
22 | { | 18 | { |
23 | _is_retweet = true; | 19 | _is_retweet = true; |
24 | 20 | ||
25 | std::string retweet = _data.at("retweeted_status").dump(); | 21 | _retweeted_status = make_unique<tweet>(_client, json["retweeted_status"].dump()); |
26 | _retweeted_status = new tweet(retweet); | ||
27 | } | 22 | } |
28 | 23 | ||
29 | if (_data.find("entities") != _data.end()) | 24 | if (!json["entities"].is_null()) |
30 | { | 25 | { |
31 | auto _entities = _data.at("entities"); | 26 | auto entities = json["entities"]; |
32 | if (_entities.find("user_mentions") != _entities.end()) | 27 | if (!entities["user_mentions"].is_null()) |
33 | { | 28 | { |
34 | for (auto _mention : _entities.at("user_mentions")) | 29 | for (auto mention : entities["user_mentions"]) |
35 | { | 30 | { |
36 | _mentions.push_back(std::make_pair(_mention.at("id"), _mention.at("screen_name").get<std::string>())); | 31 | _mentions.push_back(std::make_pair(mention["id"].get<user_id>(), mention["screen_name"].get<std::string>())); |
37 | } | 32 | } |
38 | } | 33 | } |
39 | } | 34 | } |
40 | } | 35 | } catch (const std::invalid_argument& error) |
41 | |||
42 | tweet::tweet(const tweet& other) | ||
43 | { | 36 | { |
44 | _valid = other._valid; | 37 | std::throw_with_nested(malformed_object("tweet", data)); |
45 | _id = other._id; | 38 | } catch (const std::domain_error& error) |
46 | _text = other._text; | ||
47 | _author = other._author; | ||
48 | _is_retweet = other._is_retweet; | ||
49 | |||
50 | if (_is_retweet) | ||
51 | { | ||
52 | _retweeted_status = new tweet(*other._retweeted_status); | ||
53 | } | ||
54 | |||
55 | _mentions = other._mentions; | ||
56 | } | ||
57 | |||
58 | tweet::tweet(tweet&& other) : tweet() | ||
59 | { | 39 | { |
60 | swap(*this, other); | 40 | std::throw_with_nested(malformed_object("tweet", data)); |
61 | } | 41 | } |
62 | 42 | ||
63 | tweet::~tweet() | 43 | std::string tweet::generateReplyPrefill() const |
64 | { | 44 | { |
65 | if (_is_retweet) | 45 | std::ostringstream output; |
46 | output << "@" << _author->getScreenName() << " "; | ||
47 | |||
48 | for (auto mention : _mentions) | ||
66 | { | 49 | { |
67 | delete _retweeted_status; | 50 | if ((mention.first != _author->getID()) && (mention.first != _client.getUser().getID())) |
51 | { | ||
52 | output << "@" << mention.second << " "; | ||
53 | } | ||
68 | } | 54 | } |
69 | } | ||
70 | |||
71 | tweet& tweet::operator=(tweet other) | ||
72 | { | ||
73 | swap(*this, other); | ||
74 | 55 | ||
75 | return *this; | 56 | return output.str(); |
76 | } | ||
77 | |||
78 | void swap(tweet& first, tweet& second) | ||
79 | { | ||
80 | std::swap(first._valid, second._valid); | ||
81 | std::swap(first._id, second._id); | ||
82 | std::swap(first._text, second._text); | ||
83 | std::swap(first._author, second._author); | ||
84 | std::swap(first._is_retweet, second._is_retweet); | ||
85 | std::swap(first._retweeted_status, second._retweeted_status); | ||
86 | std::swap(first._mentions, second._mentions); | ||
87 | } | 57 | } |
88 | 58 | ||
89 | tweet_id tweet::getID() const | 59 | tweet tweet::reply(std::string message, std::list<long> media_ids) const |
90 | { | 60 | { |
91 | assert(_valid); | 61 | return _client.replyToTweet(message, _id, media_ids); |
92 | |||
93 | return _id; | ||
94 | } | ||
95 | |||
96 | std::string tweet::getText() const | ||
97 | { | ||
98 | assert(_valid); | ||
99 | |||
100 | return _text; | ||
101 | } | 62 | } |
102 | 63 | ||
103 | const user& tweet::getAuthor() const | 64 | bool tweet::isMyTweet() const |
104 | { | 65 | { |
105 | assert(_valid); | 66 | return *_author == _client.getUser(); |
106 | |||
107 | return _author; | ||
108 | } | ||
109 | |||
110 | bool tweet::isRetweet() const | ||
111 | { | ||
112 | assert(_valid); | ||
113 | |||
114 | return _is_retweet; | ||
115 | } | ||
116 | |||
117 | tweet tweet::getRetweet() const | ||
118 | { | ||
119 | assert(_valid && _is_retweet); | ||
120 | |||
121 | return *_retweeted_status; | ||
122 | } | ||
123 | |||
124 | std::vector<std::pair<user_id, std::string>> tweet::getMentions() const | ||
125 | { | ||
126 | return _mentions; | ||
127 | } | 67 | } |
128 | 68 | ||
129 | std::string tweet::getURL() const | 69 | std::string tweet::getURL() const |
130 | { | 70 | { |
131 | return "https://twitter.com/" + _author.getScreenName() + "/statuses/" + std::to_string(_id); | 71 | std::ostringstream urlstr; |
132 | } | 72 | urlstr << "https://twitter.com"; |
133 | 73 | urlstr << _author->getScreenName(); | |
134 | tweet::operator bool() const | 74 | urlstr << "/statuses/"; |
135 | { | 75 | urlstr << _id; |
136 | return _valid; | 76 | |
77 | return urlstr.str(); | ||
137 | } | 78 | } |
138 | 79 | ||
139 | }; | 80 | }; |
diff --git a/src/tweet.h b/src/tweet.h index 74d73d2..a29e45c 100644 --- a/src/tweet.h +++ b/src/tweet.h | |||
@@ -2,42 +2,77 @@ | |||
2 | #define TWEET_H_CE980721 | 2 | #define TWEET_H_CE980721 |
3 | 3 | ||
4 | #include <string> | 4 | #include <string> |
5 | #include "user.h" | ||
6 | #include <vector> | 5 | #include <vector> |
7 | #include <utility> | 6 | #include <utility> |
7 | #include <cassert> | ||
8 | #include <list> | ||
9 | #include "user.h" | ||
8 | 10 | ||
9 | namespace twitter { | 11 | namespace twitter { |
10 | 12 | ||
13 | class client; | ||
14 | |||
11 | typedef unsigned long long tweet_id; | 15 | typedef unsigned long long tweet_id; |
12 | 16 | ||
13 | class tweet { | 17 | class tweet { |
14 | public: | 18 | public: |
15 | tweet(); | ||
16 | tweet(std::string data); | ||
17 | tweet(const tweet& other); | ||
18 | tweet(tweet&& other); | ||
19 | ~tweet(); | ||
20 | |||
21 | tweet& operator=(tweet other); | ||
22 | friend void swap(tweet& first, tweet& second); | ||
23 | |||
24 | tweet_id getID() const; | ||
25 | std::string getText() const; | ||
26 | const user& getAuthor() const; | ||
27 | bool isRetweet() const; | ||
28 | tweet getRetweet() const; | ||
29 | std::vector<std::pair<user_id, std::string>> getMentions() const; | ||
30 | std::string getURL() const; | ||
31 | 19 | ||
32 | operator bool() const; | 20 | tweet(const client& tclient, std::string data); |
21 | |||
22 | tweet(const tweet& other) = delete; | ||
23 | tweet& operator=(const tweet& other) = delete; | ||
24 | |||
25 | tweet(tweet&& other) = default; | ||
26 | tweet& operator=(tweet&& other) = default; | ||
27 | |||
28 | tweet_id getID() const | ||
29 | { | ||
30 | return _id; | ||
31 | } | ||
32 | |||
33 | std::string getText() const | ||
34 | { | ||
35 | return _text; | ||
36 | } | ||
37 | |||
38 | const user& getAuthor() const | ||
39 | { | ||
40 | return *_author; | ||
41 | } | ||
42 | |||
43 | bool isRetweet() const | ||
44 | { | ||
45 | return _is_retweet; | ||
46 | } | ||
47 | |||
48 | const tweet& getRetweet() const | ||
49 | { | ||
50 | assert(_is_retweet); | ||
51 | |||
52 | return *_retweeted_status; | ||
53 | } | ||
54 | |||
55 | const std::vector<std::pair<user_id, std::string>>& getMentions() const | ||
56 | { | ||
57 | return _mentions; | ||
58 | } | ||
59 | |||
60 | std::string generateReplyPrefill() const; | ||
61 | |||
62 | tweet reply(std::string message, std::list<long> media_ids = {}) const; | ||
63 | |||
64 | bool isMyTweet() const; | ||
65 | |||
66 | std::string getURL() const; | ||
33 | 67 | ||
34 | private: | 68 | private: |
35 | bool _valid; | 69 | |
70 | const client& _client; | ||
36 | tweet_id _id; | 71 | tweet_id _id; |
37 | std::string _text; | 72 | std::string _text; |
38 | user _author; | 73 | std::unique_ptr<user> _author; |
39 | bool _is_retweet = false; | 74 | bool _is_retweet = false; |
40 | tweet* _retweeted_status = nullptr; | 75 | std::unique_ptr<tweet> _retweeted_status; |
41 | std::vector<std::pair<user_id, std::string>> _mentions; | 76 | std::vector<std::pair<user_id, std::string>> _mentions; |
42 | }; | 77 | }; |
43 | 78 | ||
diff --git a/src/twitter.h b/src/twitter.h index 90925df..1ba4394 100644 --- a/src/twitter.h +++ b/src/twitter.h | |||
@@ -11,6 +11,7 @@ namespace twitter { | |||
11 | #include "util.h" | 11 | #include "util.h" |
12 | #include "auth.h" | 12 | #include "auth.h" |
13 | #include "client.h" | 13 | #include "client.h" |
14 | #include "stream.h" | ||
14 | #include "tweet.h" | 15 | #include "tweet.h" |
15 | #include "user.h" | 16 | #include "user.h" |
16 | #include "notification.h" | 17 | #include "notification.h" |
diff --git a/src/user.cpp b/src/user.cpp index 0fa1e39..0b6e93a 100644 --- a/src/user.cpp +++ b/src/user.cpp | |||
@@ -1,51 +1,43 @@ | |||
1 | #include "user.h" | 1 | #include "user.h" |
2 | #include <json.hpp> | 2 | #include <json.hpp> |
3 | 3 | #include "codes.h" | |
4 | using nlohmann::json; | 4 | #include "client.h" |
5 | 5 | ||
6 | namespace twitter { | 6 | namespace twitter { |
7 | 7 | ||
8 | user::user() : _valid(false) | 8 | user::user(const client& tclient, std::string data) try |
9 | { | 9 | : _client(tclient) |
10 | |||
11 | } | ||
12 | |||
13 | user::user(std::string data) : _valid(true) | ||
14 | { | 10 | { |
15 | auto _data = json::parse(data); | 11 | auto json = nlohmann::json::parse(data); |
16 | _id = _data.at("id"); | 12 | _id = json["id"].get<user_id>(); |
17 | _screen_name = _data.at("screen_name"); | 13 | _screen_name = json["screen_name"].get<std::string>(); |
18 | _name = _data.at("name"); | 14 | _name = json["name"].get<std::string>(); |
19 | } | 15 | } catch (const std::invalid_argument& error) |
20 | |||
21 | user_id user::getID() const | ||
22 | { | 16 | { |
23 | return _id; | 17 | std::throw_with_nested(malformed_object("user", data)); |
24 | } | 18 | } catch (const std::domain_error& error) |
25 | |||
26 | std::string user::getScreenName() const | ||
27 | { | 19 | { |
28 | return _screen_name; | 20 | std::throw_with_nested(malformed_object("user", data)); |
29 | } | 21 | } |
30 | 22 | ||
31 | std::string user::getName() const | 23 | std::set<user_id> user::getFriends() const |
32 | { | 24 | { |
33 | return _name; | 25 | return _client.getFriends(_id); |
34 | } | 26 | } |
35 | 27 | ||
36 | user::operator bool() const | 28 | std::set<user_id> user::getFollowers() const |
37 | { | 29 | { |
38 | return _valid; | 30 | return _client.getFollowers(_id); |
39 | } | 31 | } |
40 | 32 | ||
41 | bool user::operator==(const user& other) const | 33 | void user::follow() const |
42 | { | 34 | { |
43 | return _id == other._id; | 35 | _client.follow(_id); |
44 | } | 36 | } |
45 | 37 | ||
46 | bool user::operator!=(const user& other) const | 38 | void user::unfollow() const |
47 | { | 39 | { |
48 | return _id != other._id; | 40 | _client.unfollow(_id); |
49 | } | 41 | } |
50 | 42 | ||
51 | }; | 43 | }; |
diff --git a/src/user.h b/src/user.h index 1d8be99..f08840b 100644 --- a/src/user.h +++ b/src/user.h | |||
@@ -2,26 +2,52 @@ | |||
2 | #define USER_H_BF3AB38C | 2 | #define USER_H_BF3AB38C |
3 | 3 | ||
4 | #include <string> | 4 | #include <string> |
5 | #include <set> | ||
5 | 6 | ||
6 | namespace twitter { | 7 | namespace twitter { |
7 | 8 | ||
9 | class client; | ||
10 | |||
8 | typedef unsigned long long user_id; | 11 | typedef unsigned long long user_id; |
9 | 12 | ||
10 | class user { | 13 | class user { |
11 | public: | 14 | public: |
12 | user(); | ||
13 | user(std::string data); | ||
14 | 15 | ||
15 | user_id getID() const; | 16 | user(const client& tclient, std::string data); |
16 | std::string getScreenName() const; | 17 | |
17 | std::string getName() const; | 18 | user_id getID() const |
19 | { | ||
20 | return _id; | ||
21 | } | ||
22 | |||
23 | std::string getScreenName() const | ||
24 | { | ||
25 | return _screen_name; | ||
26 | } | ||
27 | |||
28 | std::string getName() const | ||
29 | { | ||
30 | return _name; | ||
31 | } | ||
18 | 32 | ||
19 | operator bool() const; | 33 | bool operator==(const user& other) const |
20 | bool operator==(const user& other) const; | 34 | { |
21 | bool operator!=(const user& other) const; | 35 | return _id == other._id; |
36 | } | ||
37 | |||
38 | bool operator!=(const user& other) const | ||
39 | { | ||
40 | return _id != other._id; | ||
41 | } | ||
42 | |||
43 | std::set<user_id> getFriends() const; | ||
44 | std::set<user_id> getFollowers() const; | ||
45 | void follow() const; | ||
46 | void unfollow() const; | ||
22 | 47 | ||
23 | private: | 48 | private: |
24 | bool _valid = false; | 49 | |
50 | const client& _client; | ||
25 | user_id _id; | 51 | user_id _id; |
26 | std::string _screen_name; | 52 | std::string _screen_name; |
27 | std::string _name; | 53 | std::string _name; |
diff --git a/src/util.h b/src/util.h index e1c9370..2f4b9a3 100644 --- a/src/util.h +++ b/src/util.h | |||
@@ -24,6 +24,12 @@ namespace twitter { | |||
24 | return result.str(); | 24 | return result.str(); |
25 | } | 25 | } |
26 | 26 | ||
27 | template<typename T, typename... Args> | ||
28 | std::unique_ptr<T> make_unique(Args&&... args) | ||
29 | { | ||
30 | return std::unique_ptr<T>(new T(std::forward<Args>(args)...)); | ||
31 | } | ||
32 | |||
27 | }; | 33 | }; |
28 | 34 | ||
29 | #endif /* end of include guard: UTIL_H_440DEAA0 */ | 35 | #endif /* end of include guard: UTIL_H_440DEAA0 */ |