source: protocols/twitter/twitter.c @ 9ed0081

Last change on this file since 9ed0081 was 29f72b7, checked in by Wilmer van der Gaast <wilmer@…>, at 2012-11-25T17:43:13Z

Finally cleaned up the show-tweet functions.

  • Property mode set to 100644
File size: 21.5 KB
RevLine 
[1b221e0]1/***************************************************************************\
2*                                                                           *
3*  BitlBee - An IRC to IM gateway                                           *
4*  Simple module to facilitate twitter functionality.                       *
5*                                                                           *
[96dd574]6*  Copyright 2009-2010 Geert Mulders <g.c.w.m.mulders@gmail.com>            *
[dff0e0b]7*  Copyright 2010-2012 Wilmer van der Gaast <wilmer@gaast.net>              *
[1b221e0]8*                                                                           *
9*  This library is free software; you can redistribute it and/or            *
10*  modify it under the terms of the GNU Lesser General Public               *
11*  License as published by the Free Software Foundation, version            *
12*  2.1.                                                                     *
13*                                                                           *
14*  This library is distributed in the hope that it will be useful,          *
15*  but WITHOUT ANY WARRANTY; without even the implied warranty of           *
16*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU        *
17*  Lesser General Public License for more details.                          *
18*                                                                           *
19*  You should have received a copy of the GNU Lesser General Public License *
20*  along with this library; if not, write to the Free Software Foundation,  *
21*  Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA           *
22*                                                                           *
23****************************************************************************/
24
25#include "nogaim.h"
[713d611]26#include "oauth.h"
[1b221e0]27#include "twitter.h"
28#include "twitter_http.h"
29#include "twitter_lib.h"
[bb5ce4d1]30#include "url.h"
[1b221e0]31
[9c9a29c]32GSList *twitter_connections = NULL;
[1b221e0]33/**
[62d2cfb]34 * Main loop function
35 */
[1b221e0]36gboolean twitter_main_loop(gpointer data, gint fd, b_input_condition cond)
37{
38        struct im_connection *ic = data;
[5983eca]39
[1b221e0]40        // Check if we are still logged in...
[5983eca]41        if (!g_slist_find(twitter_connections, ic))
[1b221e0]42                return 0;
43
44        // Do stuff..
[2322a9f]45        twitter_get_timeline(ic, -1);
[1b221e0]46
47        // If we are still logged in run this function again after timeout.
48        return (ic->flags & OPT_LOGGED_IN) == OPT_LOGGED_IN;
49}
50
[5983eca]51static void twitter_main_loop_start(struct im_connection *ic)
[713d611]52{
53        struct twitter_data *td = ic->proto_data;
[5983eca]54
[631ec80]55        /* Create the room now that we "logged in". */
56        if (td->flags & TWITTER_MODE_CHAT)
57                twitter_groupchat_init(ic);
58
[5983eca]59        imcb_log(ic, "Getting initial statuses");
[713d611]60
[f26d9a3e]61        // Run this once. After this queue the main loop function (or open the
62        // stream if available).
[713d611]63        twitter_main_loop(ic, -1, 0);
[dff0e0b]64       
65        if (set_getbool(&ic->acc->set, "stream")) {
66                /* That fetch was just to get backlog, the stream will give
67                   us the rest. \o/ */
68                twitter_open_stream(ic);
[f26d9a3e]69               
70                /* Stream sends keepalives (empty lines) or actual data at
71                   least twice a minute. Disconnect if this stops. */
[e132b60]72                ic->flags |= OPT_PONGS;
[dff0e0b]73        } else {
74                /* Not using the streaming API, so keep polling the old-
75                   fashioned way. :-( */
76                td->main_loop_id =
77                    b_timeout_add(set_getint(&ic->acc->set, "fetch_interval") * 1000,
78                                  twitter_main_loop, ic);
79        }
[713d611]80}
81
[631ec80]82struct groupchat *twitter_groupchat_init(struct im_connection *ic)
83{
84        char *name_hint;
85        struct groupchat *gc;
86        struct twitter_data *td = ic->proto_data;
87        GSList *l;
88
89        if (td->timeline_gc)
90                return td->timeline_gc;
91
92        td->timeline_gc = gc = imcb_chat_new(ic, "twitter/timeline");
93
94        name_hint = g_strdup_printf("%s_%s", td->prefix, ic->acc->user);
95        imcb_chat_name_hint(gc, name_hint);
96        g_free(name_hint);
97
98        for (l = ic->bee->users; l; l = l->next) {
99                bee_user_t *bu = l->data;
100                if (bu->ic == ic)
[29f72b7]101                        imcb_chat_add_buddy(gc, bu->handle);
[631ec80]102        }
103        imcb_chat_add_buddy(gc, ic->acc->user);
104       
105        return gc;
106}
107
[5983eca]108static void twitter_oauth_start(struct im_connection *ic);
[d6aa6dd]109
[5983eca]110void twitter_login_finish(struct im_connection *ic)
[d6aa6dd]111{
112        struct twitter_data *td = ic->proto_data;
[5983eca]113
[2322a9f]114        td->flags &= ~TWITTER_DOING_TIMELINE;
115
[5983eca]116        if (set_getbool(&ic->acc->set, "oauth") && !td->oauth_info)
117                twitter_oauth_start(ic);
[7f557d5]118        else if (!(td->flags & TWITTER_MODE_ONE) &&
[631ec80]119                 !(td->flags & TWITTER_HAVE_FRIENDS)) {
[5983eca]120                imcb_log(ic, "Getting contact list");
[de923d5]121                twitter_get_friends_ids(ic, -1);
[5983eca]122        } else
123                twitter_main_loop_start(ic);
[d6aa6dd]124}
[c2ecadc]125
[5983eca]126static const struct oauth_service twitter_oauth = {
[3f808ca]127        "https://api.twitter.com/oauth/request_token",
128        "https://api.twitter.com/oauth/access_token",
[c01bbd1]129        "https://api.twitter.com/oauth/authorize",
[c2ecadc]130        .consumer_key = "xsDNKJuNZYkZyMcu914uEA",
131        .consumer_secret = "FCxqcr0pXKzsF9ajmP57S3VQ8V6Drk4o2QYtqMcOszo",
132};
133
[5983eca]134static const struct oauth_service identica_oauth = {
[3f808ca]135        "https://identi.ca/api/oauth/request_token",
136        "https://identi.ca/api/oauth/access_token",
[ce617f0]137        "https://identi.ca/api/oauth/authorize",
138        .consumer_key = "e147ff789fcbd8a5a07963afbb43f9da",
139        .consumer_secret = "c596267f277457ec0ce1ab7bb788d828",
140};
141
[5983eca]142static gboolean twitter_oauth_callback(struct oauth_info *info);
[713d611]143
[5983eca]144static const struct oauth_service *get_oauth_service(struct im_connection *ic)
[ce617f0]145{
146        struct twitter_data *td = ic->proto_data;
[5983eca]147
148        if (strstr(td->url_host, "identi.ca"))
[ce617f0]149                return &identica_oauth;
150        else
151                return &twitter_oauth;
[5983eca]152
[ce617f0]153        /* Could add more services, or allow configuring your own base URL +
154           API keys. */
155}
156
[5983eca]157static void twitter_oauth_start(struct im_connection *ic)
[713d611]158{
[18dbb20]159        struct twitter_data *td = ic->proto_data;
[c42e8b9]160
[5983eca]161        imcb_log(ic, "Requesting OAuth request token");
162
163        td->oauth_info = oauth_request_token(get_oauth_service(ic), twitter_oauth_callback, ic);
164
[748bcdd]165        /* We need help from the user to complete OAuth login, so don't time
166           out on this login. */
167        ic->flags |= OPT_SLOW_LOGIN;
[713d611]168}
169
[5983eca]170static gboolean twitter_oauth_callback(struct oauth_info *info)
[713d611]171{
172        struct im_connection *ic = info->data;
[18dbb20]173        struct twitter_data *td;
[5983eca]174
175        if (!g_slist_find(twitter_connections, ic))
[18dbb20]176                return FALSE;
[5983eca]177
[18dbb20]178        td = ic->proto_data;
[5983eca]179        if (info->stage == OAUTH_REQUEST_TOKEN) {
180                char name[strlen(ic->acc->user) + 9], *msg;
181
182                if (info->request_token == NULL) {
183                        imcb_error(ic, "OAuth error: %s", twitter_parse_error(info->http));
184                        imc_logout(ic, TRUE);
[18dbb20]185                        return FALSE;
[c42e8b9]186                }
[5983eca]187
188                sprintf(name, "%s_%s", td->prefix, ic->acc->user);
189                msg = g_strdup_printf("To finish OAuth authentication, please visit "
190                                      "%s and respond with the resulting PIN code.",
191                                      info->auth_url);
192                imcb_buddy_msg(ic, name, msg, 0, 0);
193                g_free(msg);
194        } else if (info->stage == OAUTH_ACCESS_TOKEN) {
195                if (info->token == NULL || info->token_secret == NULL) {
196                        imcb_error(ic, "OAuth error: %s", twitter_parse_error(info->http));
197                        imc_logout(ic, TRUE);
[18dbb20]198                        return FALSE;
[5983eca]199                } else {
200                        const char *sn = oauth_params_get(&info->params, "screen_name");
201
202                        if (sn != NULL && ic->acc->prpl->handle_cmp(sn, ic->acc->user) != 0) {
203                                imcb_log(ic, "Warning: You logged in via OAuth as %s "
204                                         "instead of %s.", sn, ic->acc->user);
[93cc86f]205                        }
[de923d5]206                        g_free(td->user);
207                        td->user = g_strdup(sn);
[93cc86f]208                }
[5983eca]209
[288b215]210                /* IM mods didn't do this so far and it's ugly but I should
211                   be able to get away with it... */
[5983eca]212                g_free(ic->acc->pass);
213                ic->acc->pass = oauth_to_string(info);
214
215                twitter_login_finish(ic);
[c42e8b9]216        }
[5983eca]217
[18dbb20]218        return TRUE;
[713d611]219}
220
[c2ecadc]221
[5983eca]222static char *set_eval_mode(set_t * set, char *value)
[e88fbe27]223{
[5983eca]224        if (g_strcasecmp(value, "one") == 0 ||
225            g_strcasecmp(value, "many") == 0 || g_strcasecmp(value, "chat") == 0)
[e88fbe27]226                return value;
227        else
228                return NULL;
229}
[1b221e0]230
[7d27962]231int twitter_url_len_diff(gchar *msg, unsigned int target_len)
232{
233        int url_len_diff = 0;
234
235        static GRegex *regex = NULL;
236        GMatchInfo *match_info;
237
238        if (regex == NULL)
239                regex = g_regex_new("(^|\\s)(http(s)?://[^\\s$]+)", 0, 0, NULL);
240       
241        g_regex_match(regex, msg, 0, &match_info);
242        while (g_match_info_matches(match_info)) {
243                gchar *url = g_match_info_fetch(match_info, 2);
244                url_len_diff += target_len - g_utf8_strlen(url, -1);
245                if (g_match_info_fetch(match_info, 3) != NULL)
246                        url_len_diff += 1;
247                g_free(url);
248                g_match_info_next(match_info, NULL);
249        }
250        g_match_info_free(match_info);
251
252        return url_len_diff;
253}
254
[5983eca]255static gboolean twitter_length_check(struct im_connection *ic, gchar * msg)
[9997691]256{
[5983eca]257        int max = set_getint(&ic->acc->set, "message_length"), len;
[7d27962]258        int target_len = set_getint(&ic->acc->set, "target_url_length");
259        int url_len_diff = 0;
260   
261        if (target_len > 0)
262                url_len_diff = twitter_url_len_diff(msg, target_len);
[5983eca]263
[7d27962]264        if (max == 0 || (len = g_utf8_strlen(msg, -1) + url_len_diff) <= max)
[9997691]265                return TRUE;
[5983eca]266
[96dd574]267        twitter_log(ic, "Maximum message length exceeded: %d > %d", len, max);
[5983eca]268
[9997691]269        return FALSE;
270}
271
[5983eca]272static void twitter_init(account_t * acc)
[1b221e0]273{
[62d2cfb]274        set_t *s;
[ffcdf13]275        char *def_url;
[7d27962]276        char *def_tul;
[5983eca]277
278        if (strcmp(acc->prpl->name, "twitter") == 0) {
[ffcdf13]279                def_url = TWITTER_API_URL;
[7d27962]280                def_tul = "20";
[5983eca]281        } else {                /* if( strcmp( acc->prpl->name, "identica" ) == 0 ) */
[ffcdf13]282                def_url = IDENTICA_API_URL;
[7d27962]283                def_tul = "0";
[ffcdf13]284        }
[5983eca]285
286        s = set_add(&acc->set, "auto_reply_timeout", "10800", set_eval_int, acc);
287
288        s = set_add(&acc->set, "base_url", def_url, NULL, acc);
[bb5ce4d1]289        s->flags |= ACC_SET_OFFLINE_ONLY;
[5983eca]290
291        s = set_add(&acc->set, "commands", "true", set_eval_bool, acc);
292
[2322a9f]293        s = set_add(&acc->set, "fetch_interval", "60", set_eval_int, acc);
294        s->flags |= ACC_SET_OFFLINE_ONLY;
295
296        s = set_add(&acc->set, "fetch_mentions", "true", set_eval_bool, acc);
297
[5983eca]298        s = set_add(&acc->set, "message_length", "140", set_eval_int, acc);
299
[7d27962]300        s = set_add(&acc->set, "target_url_length", def_tul, set_eval_int, acc);
301
[5983eca]302        s = set_add(&acc->set, "mode", "chat", set_eval_mode, acc);
[2abceca]303        s->flags |= ACC_SET_OFFLINE_ONLY;
[5983eca]304
[b3d99e3]305        s = set_add(&acc->set, "oauth", "true", set_eval_oauth, acc);
[ce199b7]306
[e93fa05]307        s = set_add(&acc->set, "show_ids", "true", set_eval_bool, acc);
[5983eca]308
[b5fe39b]309        s = set_add(&acc->set, "show_old_mentions", "20", set_eval_int, acc);
[948abab]310
311        s = set_add(&acc->set, "strip_newlines", "false", set_eval_bool, acc);
[dff0e0b]312       
[f26d9a3e]313        if (strcmp(acc->prpl->name, "twitter") == 0) {
314                s = set_add(&acc->set, "stream", "true", set_eval_bool, acc);
315                s->flags |= ACC_SET_OFFLINE_ONLY;
316        }
[1b221e0]317}
318
319/**
[f26d9a3e]320 * Login method. Since the twitter API works with separate HTTP request we
[1b221e0]321 * only save the user and pass to the twitter_data object.
322 */
[5983eca]323static void twitter_login(account_t * acc)
[1b221e0]324{
[5983eca]325        struct im_connection *ic = imcb_new(acc);
[bb5ce4d1]326        struct twitter_data *td;
[5983eca]327        char name[strlen(acc->user) + 9];
[bb5ce4d1]328        url_t url;
[c0f33f1]329        char *s;
330       
[5983eca]331        if (!url_set(&url, set_getstr(&ic->acc->set, "base_url")) ||
332            (url.proto != PROTO_HTTP && url.proto != PROTO_HTTPS)) {
333                imcb_error(ic, "Incorrect API base URL: %s", set_getstr(&ic->acc->set, "base_url"));
334                imc_logout(ic, FALSE);
[bb5ce4d1]335                return;
336        }
[5983eca]337
[f26d9a3e]338        if (!strstr(url.host, "twitter.com") &&
339            set_getbool(&ic->acc->set, "stream")) {
340                imcb_error(ic, "Warning: The streaming API is only supported by Twitter, "
341                               "and you seem to be connecting to a different service.");
342        }
343
[c0f33f1]344        imcb_log(ic, "Connecting");
345
[5983eca]346        twitter_connections = g_slist_append(twitter_connections, ic);
347        td = g_new0(struct twitter_data, 1);
[713d611]348        ic->proto_data = td;
[de923d5]349        td->user = g_strdup(acc->user);
[5983eca]350
[bb5ce4d1]351        td->url_ssl = url.proto == PROTO_HTTPS;
352        td->url_port = url.port;
[5983eca]353        td->url_host = g_strdup(url.host);
354        if (strcmp(url.file, "/") != 0)
355                td->url_path = g_strdup(url.file);
[c0f33f1]356        else {
[5983eca]357                td->url_path = g_strdup("");
[c0f33f1]358                if (g_str_has_suffix(url.host, "twitter.com"))
359                        /* May fire for people who turned on HTTPS. */
360                        imcb_error(ic, "Warning: Twitter requires a version number in API calls "
361                                       "now. Try resetting the base_url account setting.");
362        }
363       
364        /* Hacky string mangling: Turn identi.ca into identi.ca and api.twitter.com
365           into twitter, and try to be sensible if we get anything else. */
366        td->prefix = g_strdup(url.host);
367        if (g_str_has_suffix(td->prefix, ".com"))
368                td->prefix[strlen(url.host) - 4] = '\0';
[4bc66ae]369        if ((s = strrchr(td->prefix, '.')) && strlen(s) > 4) {
370                /* If we have at least 3 chars after the last dot, cut off the rest.
371                   (mostly a www/api prefix or sth) */
[c0f33f1]372                s = g_strdup(s + 1);
373                g_free(td->prefix);
374                td->prefix = s;
375        }
376       
[5983eca]377        if (strstr(acc->pass, "oauth_token="))
378                td->oauth_info = oauth_from_string(acc->pass, get_oauth_service(ic));
379
380        sprintf(name, "%s_%s", td->prefix, acc->user);
381        imcb_add_buddy(ic, name, NULL);
382        imcb_buddy_status(ic, name, OPT_LOGGED_IN, NULL, NULL);
383
[c9b5817]384        td->log = g_new0(struct twitter_log_data, TWITTER_LOG_LENGTH);
385        td->log_id = -1;
[631ec80]386       
387        s = set_getstr(&ic->acc->set, "mode");
388        if (g_strcasecmp(s, "one") == 0)
389                td->flags |= TWITTER_MODE_ONE;
390        else if (g_strcasecmp(s, "many") == 0)
391                td->flags |= TWITTER_MODE_MANY;
392        else
393                td->flags |= TWITTER_MODE_CHAT;
[5983eca]394
395        twitter_login_finish(ic);
[1b221e0]396}
397
398/**
399 * Logout method. Just free the twitter_data.
400 */
[5983eca]401static void twitter_logout(struct im_connection *ic)
[1b221e0]402{
403        struct twitter_data *td = ic->proto_data;
[5983eca]404
[1b221e0]405        // Set the status to logged out.
[5983eca]406        ic->flags &= ~OPT_LOGGED_IN;
[1b221e0]407
[2abceca]408        // Remove the main_loop function from the function queue.
409        b_event_remove(td->main_loop_id);
410
[2322a9f]411        if (td->timeline_gc)
412                imcb_chat_free(td->timeline_gc);
[1014cab]413
[5983eca]414        if (td) {
[1388d30]415                http_close(td->stream);
[5983eca]416                oauth_info_free(td->oauth_info);
[de923d5]417                g_free(td->user);
[5983eca]418                g_free(td->prefix);
419                g_free(td->url_host);
420                g_free(td->url_path);
421                g_free(td->log);
422                g_free(td);
[1b221e0]423        }
[62d2cfb]424
[5983eca]425        twitter_connections = g_slist_remove(twitter_connections, ic);
[1b221e0]426}
427
[5983eca]428static void twitter_handle_command(struct im_connection *ic, char *message);
[7b87539]429
[1b221e0]430/**
431 *
432 */
[5983eca]433static int twitter_buddy_msg(struct im_connection *ic, char *who, char *message, int away)
[1b221e0]434{
[c42e8b9]435        struct twitter_data *td = ic->proto_data;
[5983eca]436        int plen = strlen(td->prefix);
437
[ffcdf13]438        if (g_strncasecmp(who, td->prefix, plen) == 0 && who[plen] == '_' &&
[5983eca]439            g_strcasecmp(who + plen + 1, ic->acc->user) == 0) {
440                if (set_getbool(&ic->acc->set, "oauth") &&
441                    td->oauth_info && td->oauth_info->token == NULL) {
442                        char pin[strlen(message) + 1], *s;
443
444                        strcpy(pin, message);
445                        for (s = pin + sizeof(pin) - 2; s > pin && isspace(*s); s--)
[64f8c425]446                                *s = '\0';
[5983eca]447                        for (s = pin; *s && isspace(*s); s++) {
448                        }
449
450                        if (!oauth_access_token(s, td->oauth_info)) {
451                                imcb_error(ic, "OAuth error: %s",
452                                           "Failed to send access token request");
453                                imc_logout(ic, TRUE);
[c2ecadc]454                                return FALSE;
455                        }
[5983eca]456                } else
[7b87539]457                        twitter_handle_command(ic, message);
[5983eca]458        } else {
[e88fbe27]459                twitter_direct_messages_new(ic, who, message);
[c42e8b9]460        }
[5983eca]461        return (0);
[1b221e0]462}
463
464/**
465 *
466 */
[5983eca]467static void twitter_set_my_name(struct im_connection *ic, char *info)
[1b221e0]468{
469}
470
[5983eca]471static void twitter_get_info(struct im_connection *ic, char *who)
[1b221e0]472{
473}
474
[5983eca]475static void twitter_add_buddy(struct im_connection *ic, char *who, char *group)
[1b221e0]476{
[7d53efb]477        twitter_friendships_create_destroy(ic, who, 1);
[1b221e0]478}
479
[5983eca]480static void twitter_remove_buddy(struct im_connection *ic, char *who, char *group)
[1b221e0]481{
[7d53efb]482        twitter_friendships_create_destroy(ic, who, 0);
[1b221e0]483}
484
[5983eca]485static void twitter_chat_msg(struct groupchat *c, char *message, int flags)
[1b221e0]486{
[5983eca]487        if (c && message)
488                twitter_handle_command(c->ic, message);
[1b221e0]489}
490
[5983eca]491static void twitter_chat_invite(struct groupchat *c, char *who, char *message)
[1b221e0]492{
493}
494
[5983eca]495static void twitter_chat_leave(struct groupchat *c)
[1b221e0]496{
[16592d8]497        struct twitter_data *td = c->ic->proto_data;
[5983eca]498
[2322a9f]499        if (c != td->timeline_gc)
[5983eca]500                return;         /* WTF? */
501
[16592d8]502        /* If the user leaves the channel: Fine. Rejoin him/her once new
503           tweets come in. */
[2322a9f]504        imcb_chat_free(td->timeline_gc);
505        td->timeline_gc = NULL;
[1b221e0]506}
507
[5983eca]508static void twitter_keepalive(struct im_connection *ic)
[1b221e0]509{
510}
511
[5983eca]512static void twitter_add_permit(struct im_connection *ic, char *who)
[1b221e0]513{
514}
515
[5983eca]516static void twitter_rem_permit(struct im_connection *ic, char *who)
[1b221e0]517{
518}
519
[5983eca]520static void twitter_add_deny(struct im_connection *ic, char *who)
[1b221e0]521{
522}
523
[5983eca]524static void twitter_rem_deny(struct im_connection *ic, char *who)
[1b221e0]525{
526}
527
528//static char *twitter_set_display_name( set_t *set, char *value )
529//{
[5983eca]530//      return value;
[1b221e0]531//}
[7b87539]532
[5983eca]533static void twitter_buddy_data_add(struct bee_user *bu)
[203a2d2]534{
[5983eca]535        bu->data = g_new0(struct twitter_user_data, 1);
[203a2d2]536}
537
[5983eca]538static void twitter_buddy_data_free(struct bee_user *bu)
[203a2d2]539{
[5983eca]540        g_free(bu->data);
[203a2d2]541}
542
[b61c74c]543/** Convert the given bitlbee tweet ID, bitlbee username, or twitter tweet ID
544 *  into a twitter tweet ID.
545 *
546 *  Returns 0 if the user provides garbage.
547 */
548static guint64 twitter_message_id_from_command_arg(struct im_connection *ic, struct twitter_data *td, char *arg) {
549        struct twitter_user_data *tud;
550        bee_user_t *bu;
551        guint64 id = 0;
552        if (g_str_has_prefix(arg, "#") &&
553                sscanf(arg + 1, "%" G_GUINT64_FORMAT, &id) == 1) {
554                if (id < TWITTER_LOG_LENGTH && td->log)
555                        id = td->log[id].id;
556        } else if ((bu = bee_user_by_handle(ic->bee, ic, arg)) &&
557                (tud = bu->data) && tud->last_id)
558                id = tud->last_id;
559        else if (sscanf(arg, "%" G_GUINT64_FORMAT, &id) == 1){
560                if (id < TWITTER_LOG_LENGTH && td->log)
561                        id = td->log[id].id;
562        }
563        return id;
564}
565
[5983eca]566static void twitter_handle_command(struct im_connection *ic, char *message)
[7b87539]567{
568        struct twitter_data *td = ic->proto_data;
[15bc063]569        char *cmds, **cmd, *new = NULL;
570        guint64 in_reply_to = 0;
[5983eca]571
572        cmds = g_strdup(message);
573        cmd = split_command_parts(cmds);
574
575        if (cmd[0] == NULL) {
576                g_free(cmds);
[7b87539]577                return;
[5983eca]578        } else if (!set_getbool(&ic->acc->set, "commands")) {
[7b87539]579                /* Not supporting commands. */
[5983eca]580        } else if (g_strcasecmp(cmd[0], "undo") == 0) {
[7b87539]581                guint64 id;
[5983eca]582
[aa2f575]583                if (cmd[1] == NULL)
584                        twitter_status_destroy(ic, td->last_status_id);
585                else if (sscanf(cmd[1], "%" G_GUINT64_FORMAT, &id) == 1) {
586                        if (id < TWITTER_LOG_LENGTH && td->log)
587                                id = td->log[id].id;
588                       
[5983eca]589                        twitter_status_destroy(ic, id);
[aa2f575]590                } else
[96dd574]591                        twitter_log(ic, "Could not undo last action");
[5983eca]592
[b61c74c]593                g_free(cmds);
594                return;
595        } else if (g_strcasecmp(cmd[0], "favourite") == 0 && cmd[1]) {
596                guint64 id;
597                if ((id = twitter_message_id_from_command_arg(ic, td, cmd[1]))) {
598                        twitter_favourite_tweet(ic, id);
599                } else {
[96dd574]600                        twitter_log(ic, "Please provide a message ID or username.");
[b61c74c]601                }
[5983eca]602                g_free(cmds);
[7b87539]603                return;
[5983eca]604        } else if (g_strcasecmp(cmd[0], "follow") == 0 && cmd[1]) {
605                twitter_add_buddy(ic, cmd[1], NULL);
606                g_free(cmds);
[b890626]607                return;
[5983eca]608        } else if (g_strcasecmp(cmd[0], "unfollow") == 0 && cmd[1]) {
609                twitter_remove_buddy(ic, cmd[1], NULL);
610                g_free(cmds);
[b890626]611                return;
[d18dee42]612        } else if ((g_strcasecmp(cmd[0], "report") == 0 ||
613                    g_strcasecmp(cmd[0], "spam") == 0) && cmd[1]) {
614                char * screen_name;
615                guint64 id;
[7de784c]616                screen_name = cmd[1];
[d18dee42]617                /* Report nominally works on users but look up the user who
618                   posted the given ID if the user wants to do it that way */
619                if (g_str_has_prefix(cmd[1], "#") &&
620                    sscanf(cmd[1] + 1, "%" G_GUINT64_FORMAT, &id) == 1) {
621                        if (id < TWITTER_LOG_LENGTH && td->log) {
622                                if (g_slist_find(ic->bee->users, td->log[id].bu)) {
623                                        screen_name = td->log[id].bu->handle;
624                                }
625                        }
626                }
627                twitter_report_spam(ic, screen_name);
628                g_free(cmds);
629                return;
[5983eca]630        } else if (g_strcasecmp(cmd[0], "rt") == 0 && cmd[1]) {
[b61c74c]631                guint64 id = twitter_message_id_from_command_arg(ic, td, cmd[1]);
[5983eca]632
[b890626]633                td->last_status_id = 0;
[5983eca]634                if (id)
635                        twitter_status_retweet(ic, id);
[665c24f]636                else
[96dd574]637                        twitter_log(ic, "User `%s' does not exist or didn't "
[5983eca]638                                    "post any statuses recently", cmd[1]);
639
640                g_free(cmds);
[b890626]641                return;
[5983eca]642        } else if (g_strcasecmp(cmd[0], "reply") == 0 && cmd[1] && cmd[2]) {
[15bc063]643                struct twitter_user_data *tud;
644                bee_user_t *bu = NULL;
645                guint64 id = 0;
[5983eca]646
[aa2f575]647                if (g_str_has_prefix(cmd[1], "#") &&
648                    sscanf(cmd[1] + 1, "%" G_GUINT64_FORMAT, &id) == 1 &&
649                    (id < TWITTER_LOG_LENGTH) && td->log) {
650                        bu = td->log[id].bu;
651                        if (g_slist_find(ic->bee->users, bu))
652                                id = td->log[id].id;
653                        else
654                                bu = NULL;
655                } else if ((bu = bee_user_by_handle(ic->bee, ic, cmd[1])) &&
[5983eca]656                    (tud = bu->data) && tud->last_id) {
[15bc063]657                        id = tud->last_id;
[7ceb6b2]658                } else if (sscanf(cmd[1], "%" G_GUINT64_FORMAT, &id) == 1 &&
659                           (id < TWITTER_LOG_LENGTH) && td->log) {
[15bc063]660                        bu = td->log[id].bu;
[5983eca]661                        if (g_slist_find(ic->bee->users, bu))
[15bc063]662                                id = td->log[id].id;
663                        else
664                                bu = NULL;
665                }
[aa2f575]666
[5983eca]667                if (!id || !bu) {
[96dd574]668                        twitter_log(ic, "User `%s' does not exist or didn't "
[5983eca]669                                    "post any statuses recently", cmd[1]);
[d3bfe3d]670                        g_free(cmds);
[15bc063]671                        return;
672                }
[5983eca]673                message = new = g_strdup_printf("@%s %s", bu->handle, message + (cmd[2] - cmd[0]));
[15bc063]674                in_reply_to = id;
[5983eca]675        } else if (g_strcasecmp(cmd[0], "post") == 0) {
[7b87539]676                message += 5;
677        }
[5983eca]678
[7b87539]679        {
[15bc063]680                char *s;
[7b87539]681                bee_user_t *bu;
[5983eca]682
683                if (!twitter_length_check(ic, message)) {
684                        g_free(new);
685                        g_free(cmds);
686                        return;
[7b87539]687                }
[5983eca]688
689                s = cmd[0] + strlen(cmd[0]) - 1;
690                if (!new && s > cmd[0] && (*s == ':' || *s == ',')) {
[7b87539]691                        *s = '\0';
[5983eca]692
693                        if ((bu = bee_user_by_handle(ic->bee, ic, cmd[0]))) {
[b890626]694                                struct twitter_user_data *tud = bu->data;
[5983eca]695
696                                new = g_strdup_printf("@%s %s", bu->handle,
697                                                      message + (s - cmd[0]) + 2);
[7b87539]698                                message = new;
[5983eca]699
700                                if (time(NULL) < tud->last_time +
701                                    set_getint(&ic->acc->set, "auto_reply_timeout"))
[b890626]702                                        in_reply_to = tud->last_id;
[7b87539]703                        }
704                }
[5983eca]705
[b890626]706                /* If the user runs undo between this request and its response
707                   this would delete the second-last Tweet. Prevent that. */
708                td->last_status_id = 0;
[5983eca]709                twitter_post_status(ic, message, in_reply_to);
710                g_free(new);
[7b87539]711        }
[5983eca]712        g_free(cmds);
[7b87539]713}
[1b221e0]714
[96dd574]715void twitter_log(struct im_connection *ic, char *format, ... )
716{
717        struct twitter_data *td = ic->proto_data;
718        va_list params;
719        char *text;
720       
721        va_start(params, format);
722        text = g_strdup_vprintf(format, params);
723        va_end(params);
724       
725        if (td->timeline_gc)
726                imcb_chat_log(td->timeline_gc, "%s", text);
727        else
728                imcb_log(ic, "%s", text);
729       
730        g_free(text);
731}
732
733
[1b221e0]734void twitter_initmodule()
735{
736        struct prpl *ret = g_new0(struct prpl, 1);
[5983eca]737
[1dd3470]738        ret->options = OPT_NOOTR;
[1b221e0]739        ret->name = "twitter";
740        ret->login = twitter_login;
741        ret->init = twitter_init;
742        ret->logout = twitter_logout;
743        ret->buddy_msg = twitter_buddy_msg;
744        ret->get_info = twitter_get_info;
745        ret->set_my_name = twitter_set_my_name;
746        ret->add_buddy = twitter_add_buddy;
747        ret->remove_buddy = twitter_remove_buddy;
748        ret->chat_msg = twitter_chat_msg;
749        ret->chat_invite = twitter_chat_invite;
750        ret->chat_leave = twitter_chat_leave;
751        ret->keepalive = twitter_keepalive;
752        ret->add_permit = twitter_add_permit;
753        ret->rem_permit = twitter_rem_permit;
754        ret->add_deny = twitter_add_deny;
755        ret->rem_deny = twitter_rem_deny;
[203a2d2]756        ret->buddy_data_add = twitter_buddy_data_add;
757        ret->buddy_data_free = twitter_buddy_data_free;
[1b221e0]758        ret->handle_cmp = g_strcasecmp;
[5983eca]759
[ffcdf13]760        register_protocol(ret);
[1b221e0]761
[ffcdf13]762        /* And an identi.ca variant: */
763        ret = g_memdup(ret, sizeof(struct prpl));
764        ret->name = "identica";
[1b221e0]765        register_protocol(ret);
766}
Note: See TracBrowser for help on using the repository browser.