source: protocols/twitter/twitter.c @ 7b87539

Last change on this file since 7b87539 was 7b87539, checked in by Wilmer van der Gaast <wilmer@…>, at 2010-08-07T19:39:01Z

Add commands to the Twitter module, starting with undo (which deletes
either your most recent tweet, or a specific id (pass it as an argument)).

  • Property mode set to 100644
File size: 13.0 KB
RevLine 
[1b221e0]1/***************************************************************************\
2*                                                                           *
3*  BitlBee - An IRC to IM gateway                                           *
4*  Simple module to facilitate twitter functionality.                       *
5*                                                                           *
6*  Copyright 2009 Geert Mulders <g.c.w.m.mulders@gmail.com>                 *
7*                                                                           *
8*  This library is free software; you can redistribute it and/or            *
9*  modify it under the terms of the GNU Lesser General Public               *
10*  License as published by the Free Software Foundation, version            *
11*  2.1.                                                                     *
12*                                                                           *
13*  This library is distributed in the hope that it will be useful,          *
14*  but WITHOUT ANY WARRANTY; without even the implied warranty of           *
15*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU        *
16*  Lesser General Public License for more details.                          *
17*                                                                           *
18*  You should have received a copy of the GNU Lesser General Public License *
19*  along with this library; if not, write to the Free Software Foundation,  *
20*  Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA           *
21*                                                                           *
22****************************************************************************/
23
24#include "nogaim.h"
[713d611]25#include "oauth.h"
[1b221e0]26#include "twitter.h"
27#include "twitter_http.h"
28#include "twitter_lib.h"
[bb5ce4d1]29#include "url.h"
[1b221e0]30
31/**
[62d2cfb]32 * Main loop function
33 */
[1b221e0]34gboolean twitter_main_loop(gpointer data, gint fd, b_input_condition cond)
35{
36        struct im_connection *ic = data;
[3e69802]37       
[1b221e0]38        // Check if we are still logged in...
[3bd4a93]39        if (!g_slist_find( twitter_connections, ic ))
[1b221e0]40                return 0;
41
42        // Do stuff..
43        twitter_get_home_timeline(ic, -1);
44
45        // If we are still logged in run this function again after timeout.
46        return (ic->flags & OPT_LOGGED_IN) == OPT_LOGGED_IN;
47}
48
[713d611]49static void twitter_main_loop_start( struct im_connection *ic )
50{
51        struct twitter_data *td = ic->proto_data;
52       
[d6aa6dd]53        imcb_log( ic, "Getting initial statuses" );
[713d611]54
55        // Run this once. After this queue the main loop function.
56        twitter_main_loop(ic, -1, 0);
57
58        // Queue the main_loop
59        // Save the return value, so we can remove the timeout on logout.
60        td->main_loop_id = b_timeout_add(60000, twitter_main_loop, ic);
61}
62
[d6aa6dd]63static void twitter_oauth_start( struct im_connection *ic );
64
65void twitter_login_finish( struct im_connection *ic )
66{
67        struct twitter_data *td = ic->proto_data;
68       
69        if( set_getbool( &ic->acc->set, "oauth" ) && !td->oauth_info )
70                twitter_oauth_start( ic );
71        else if( g_strcasecmp( set_getstr( &ic->acc->set, "mode" ), "one" ) != 0 &&
72                 !( td->flags & TWITTER_HAVE_FRIENDS ) )
73        {
74                imcb_log( ic, "Getting contact list" );
75                twitter_get_statuses_friends( ic, -1 );
76        }
77        else
78                twitter_main_loop_start( ic );
79}
[c2ecadc]80
[3b878a1]81static const struct oauth_service twitter_oauth =
[c2ecadc]82{
83        "http://api.twitter.com/oauth/request_token",
84        "http://api.twitter.com/oauth/access_token",
[c01bbd1]85        "https://api.twitter.com/oauth/authorize",
[c2ecadc]86        .consumer_key = "xsDNKJuNZYkZyMcu914uEA",
87        .consumer_secret = "FCxqcr0pXKzsF9ajmP57S3VQ8V6Drk4o2QYtqMcOszo",
88};
89
[18dbb20]90static gboolean twitter_oauth_callback( struct oauth_info *info );
[713d611]91
92static void twitter_oauth_start( struct im_connection *ic )
93{
[18dbb20]94        struct twitter_data *td = ic->proto_data;
95       
[c42e8b9]96        imcb_log( ic, "Requesting OAuth request token" );
97
[c2ecadc]98        td->oauth_info = oauth_request_token( &twitter_oauth, twitter_oauth_callback, ic );
[713d611]99}
100
[18dbb20]101static gboolean twitter_oauth_callback( struct oauth_info *info )
[713d611]102{
103        struct im_connection *ic = info->data;
[18dbb20]104        struct twitter_data *td;
[713d611]105       
[18dbb20]106        if( !g_slist_find( twitter_connections, ic ) )
107                return FALSE;
108       
109        td = ic->proto_data;
[c42e8b9]110        if( info->stage == OAUTH_REQUEST_TOKEN )
[713d611]111        {
112                char name[strlen(ic->acc->user)+9], *msg;
113               
[c42e8b9]114                if( info->request_token == NULL )
115                {
116                        imcb_error( ic, "OAuth error: %s", info->http->status_string );
117                        imc_logout( ic, TRUE );
[18dbb20]118                        return FALSE;
[c42e8b9]119                }
120               
[ffcdf13]121                sprintf( name, "%s_%s", td->prefix, ic->acc->user );
[713d611]122                msg = g_strdup_printf( "To finish OAuth authentication, please visit "
[c2ecadc]123                                       "%s and respond with the resulting PIN code.",
124                                       info->auth_url );
[713d611]125                imcb_buddy_msg( ic, name, msg, 0, 0 );
126                g_free( msg );
127        }
[c42e8b9]128        else if( info->stage == OAUTH_ACCESS_TOKEN )
129        {
[3b878a1]130                if( info->token == NULL || info->token_secret == NULL )
[c42e8b9]131                {
132                        imcb_error( ic, "OAuth error: %s", info->http->status_string );
133                        imc_logout( ic, TRUE );
[18dbb20]134                        return FALSE;
[c42e8b9]135                }
136               
[288b215]137                /* IM mods didn't do this so far and it's ugly but I should
138                   be able to get away with it... */
139                g_free( ic->acc->pass );
[f4b0911]140                ic->acc->pass = oauth_to_string( info );
[288b215]141               
[d6aa6dd]142                twitter_login_finish( ic );
[c42e8b9]143        }
[18dbb20]144       
145        return TRUE;
[713d611]146}
147
[c2ecadc]148
[e88fbe27]149static char *set_eval_mode( set_t *set, char *value )
150{
151        if( g_strcasecmp( value, "one" ) == 0 ||
152            g_strcasecmp( value, "many" ) == 0 ||
[db57e7c]153            g_strcasecmp( value, "chat" ) == 0 )
[e88fbe27]154                return value;
155        else
156                return NULL;
157}
[1b221e0]158
[9997691]159static gboolean twitter_length_check( struct im_connection *ic, gchar *msg )
160{
161        int max = set_getint( &ic->acc->set, "message_length" ), len;
162       
163        if( max == 0 || ( len = g_utf8_strlen( msg, -1 ) ) <= max )
164                return TRUE;
165       
166        imcb_error( ic, "Maximum message length exceeded: %d > %d", len, max );
167       
168        return FALSE;
169}
170
[1b221e0]171static void twitter_init( account_t *acc )
172{
[62d2cfb]173        set_t *s;
[ffcdf13]174        char *def_url;
175        char *def_oauth;
[e88fbe27]176       
[ffcdf13]177        if( strcmp( acc->prpl->name, "twitter" ) == 0 )
178        {
179                def_url = TWITTER_API_URL;
180                def_oauth = "true";
181        }
182        else /* if( strcmp( acc->prpl->name, "identica" ) == 0 ) */
183        {
184                def_url = IDENTICA_API_URL;
185                def_oauth = "false";
186        }
187       
188        s = set_add( &acc->set, "base_url", def_url, NULL, acc );
[bb5ce4d1]189        s->flags |= ACC_SET_OFFLINE_ONLY;
190       
[7b87539]191        s = set_add( &acc->set, "commands", "true", set_eval_bool, acc );
192       
[9997691]193        s = set_add( &acc->set, "message_length", "140", set_eval_int, acc );
194       
[e88fbe27]195        s = set_add( &acc->set, "mode", "one", set_eval_mode, acc );
[2abceca]196        s->flags |= ACC_SET_OFFLINE_ONLY;
[713d611]197       
[ffcdf13]198        s = set_add( &acc->set, "oauth", def_oauth, set_eval_bool, acc );
[1b221e0]199}
200
201/**
202 * Login method. Since the twitter API works with seperate HTTP request we
203 * only save the user and pass to the twitter_data object.
204 */
205static void twitter_login( account_t *acc )
206{
207        struct im_connection *ic = imcb_new( acc );
[bb5ce4d1]208        struct twitter_data *td;
[e88fbe27]209        char name[strlen(acc->user)+9];
[bb5ce4d1]210        url_t url;
[62d2cfb]211
[bb5ce4d1]212        if( !url_set( &url, set_getstr( &ic->acc->set, "base_url" ) ) ||
213            ( url.proto != PROTO_HTTP && url.proto != PROTO_HTTPS ) )
214        {
215                imcb_error( ic, "Incorrect API base URL: %s", set_getstr( &ic->acc->set, "base_url" ) );
216                imc_logout( ic, FALSE );
217                return;
218        }
219       
[d569019]220        twitter_connections = g_slist_append( twitter_connections, ic );
[bb5ce4d1]221        td = g_new0( struct twitter_data, 1 );
[713d611]222        ic->proto_data = td;
223       
[bb5ce4d1]224        td->url_ssl = url.proto == PROTO_HTTPS;
225        td->url_port = url.port;
226        td->url_host = g_strdup( url.host );
227        if( strcmp( url.file, "/" ) != 0 )
228                td->url_path = g_strdup( url.file );
229        else
230                td->url_path = g_strdup( "" );
[ffcdf13]231        if( g_str_has_suffix( url.host, ".com" ) )
232                td->prefix = g_strndup( url.host, strlen( url.host ) - 4 );
233        else
234                td->prefix = g_strdup( url.host );
[bb5ce4d1]235       
[1b221e0]236        td->user = acc->user;
[bb5ce4d1]237        if( strstr( acc->pass, "oauth_token=" ) )
[f4b0911]238                td->oauth_info = oauth_from_string( acc->pass, &twitter_oauth );
[e88fbe27]239       
[ffcdf13]240        sprintf( name, "%s_%s", td->prefix, acc->user );
[e88fbe27]241        imcb_add_buddy( ic, name, NULL );
242        imcb_buddy_status( ic, name, OPT_LOGGED_IN, NULL, NULL );
[713d611]243       
[d6aa6dd]244        imcb_log( ic, "Connecting" );
245       
246        twitter_login_finish( ic );
[1b221e0]247}
248
249/**
250 * Logout method. Just free the twitter_data.
251 */
252static void twitter_logout( struct im_connection *ic )
253{
254        struct twitter_data *td = ic->proto_data;
255       
256        // Set the status to logged out.
257        ic->flags = 0;
258
[2abceca]259        // Remove the main_loop function from the function queue.
260        b_event_remove(td->main_loop_id);
261
[37d84b3]262        if(td->home_timeline_gc)
263                imcb_chat_free(td->home_timeline_gc);
[1014cab]264
[1b221e0]265        if( td )
266        {
[18dbb20]267                oauth_info_free( td->oauth_info );
[ffcdf13]268                g_free( td->prefix );
[04a927c]269                g_free( td->url_host );
270                g_free( td->url_path );
[508c340]271                g_free( td->pass );
[1b221e0]272                g_free( td );
273        }
[62d2cfb]274
275        twitter_connections = g_slist_remove( twitter_connections, ic );
[1b221e0]276}
277
[7b87539]278static void twitter_handle_command( struct im_connection *ic, char *message );
279
[1b221e0]280/**
281 *
282 */
283static int twitter_buddy_msg( struct im_connection *ic, char *who, char *message, int away )
284{
[c42e8b9]285        struct twitter_data *td = ic->proto_data;
[ffcdf13]286        int plen = strlen( td->prefix );
[c42e8b9]287       
[ffcdf13]288        if (g_strncasecmp(who, td->prefix, plen) == 0 && who[plen] == '_' &&
289            g_strcasecmp(who + plen + 1, ic->acc->user) == 0)
[c42e8b9]290        {
[c2ecadc]291                if( set_getbool( &ic->acc->set, "oauth" ) &&
292                    td->oauth_info && td->oauth_info->token == NULL )
[18dbb20]293                {
[64f8c425]294                        char pin[strlen(message)+1], *s;
295                       
296                        strcpy( pin, message );
297                        for( s = pin + sizeof( pin ) - 2; s > pin && isspace( *s ); s -- )
298                                *s = '\0';
299                        for( s = pin; *s && isspace( *s ); s ++ ) {}
300                       
301                        if( !oauth_access_token( s, td->oauth_info ) )
[c2ecadc]302                        {
303                                imcb_error( ic, "OAuth error: %s", "Failed to send access token request" );
304                                imc_logout( ic, TRUE );
305                                return FALSE;
306                        }
[18dbb20]307                }
[7b87539]308                else
309                        twitter_handle_command(ic, message);
[c42e8b9]310        }
[e88fbe27]311        else
[c42e8b9]312        {
[e88fbe27]313                twitter_direct_messages_new(ic, who, message);
[c42e8b9]314        }
[1b221e0]315        return( 0 );
316}
317
318/**
319 *
320 */
321static void twitter_set_my_name( struct im_connection *ic, char *info )
322{
323}
324
325static void twitter_get_info(struct im_connection *ic, char *who) 
326{
327}
328
329static void twitter_add_buddy( struct im_connection *ic, char *who, char *group )
330{
[7d53efb]331        twitter_friendships_create_destroy(ic, who, 1);
[1b221e0]332}
333
334static void twitter_remove_buddy( struct im_connection *ic, char *who, char *group )
335{
[7d53efb]336        twitter_friendships_create_destroy(ic, who, 0);
[1b221e0]337}
338
339static void twitter_chat_msg( struct groupchat *c, char *message, int flags )
340{
[7b87539]341        if( c && message )
342                twitter_handle_command( c->ic, message );
[1b221e0]343}
344
345static void twitter_chat_invite( struct groupchat *c, char *who, char *message )
346{
347}
348
349static void twitter_chat_leave( struct groupchat *c )
350{
[16592d8]351        struct twitter_data *td = c->ic->proto_data;
352       
353        if( c != td->home_timeline_gc )
354                return; /* WTF? */
355       
356        /* If the user leaves the channel: Fine. Rejoin him/her once new
357           tweets come in. */
358        imcb_chat_free(td->home_timeline_gc);
359        td->home_timeline_gc = NULL;
[1b221e0]360}
361
362static struct groupchat *twitter_chat_with( struct im_connection *ic, char *who )
363{
364        return NULL;
365}
366
367static void twitter_keepalive( struct im_connection *ic )
368{
369}
370
371static void twitter_add_permit( struct im_connection *ic, char *who )
372{
373}
374
375static void twitter_rem_permit( struct im_connection *ic, char *who )
376{
377}
378
379static void twitter_add_deny( struct im_connection *ic, char *who )
380{
381}
382
383static void twitter_rem_deny( struct im_connection *ic, char *who )
384{
385}
386
387static int twitter_send_typing( struct im_connection *ic, char *who, int typing )
388{
389        return( 1 );
390}
391
392//static char *twitter_set_display_name( set_t *set, char *value )
393//{
394//      return value;
395//}
[7b87539]396
397static void twitter_handle_command( struct im_connection *ic, char *message )
398{
399        struct twitter_data *td = ic->proto_data;
400        char *cmds, **cmd;
401       
402        cmds = g_strdup( message );
403        cmd = split_command_parts( cmds );
404       
405        if( cmd[0] == NULL )
406        {
407                g_free( cmds );
408                return;
409        }
410        else if( !set_getbool( &ic->set, "commands" ) )
411        {
412                /* Not supporting commands. */
413        }
414        else if( g_strcasecmp( cmd[0], "undo" ) == 0 )
415        {
416                guint64 id;
417               
418                if( cmd[1] )
419                {
420                        char *end = NULL;
421                       
422                        id = g_ascii_strtoull( cmd[1], &end, 10 );
423                        if( end == NULL )
424                                id = 0;
425                }
426                else
427                        id = td->last_status_id;
428               
429                /* TODO: User feedback. */
430                if( id )
431                        twitter_status_destroy( ic, id );
432               
433                g_free( cmds );
434                return;
435        }
436        else if( g_strcasecmp( cmd[0], "post" ) == 0 )
437        {
438                message += 5;
439        }
440       
441        {
442                char *s, *new = NULL;
443                bee_user_t *bu;
444               
445                if( !twitter_length_check( ic, message ) )
446                {
447                        g_free( cmds );
448                        return;
449                }
450               
451                s = cmd[0] + strlen( cmd[0] ) - 1;
452                if( s > cmd[0] && ( *s == ':' || *s == ',' ) )
453                {
454                        *s = '\0';
455                       
456                        if( ( bu = bee_user_by_handle( ic->bee, ic, cmd[0] ) ) )
457                        {
458                                new = g_strdup_printf( "@%s %s", bu->handle,
459                                                       message + ( s - cmd[0] ) + 2 );
460                                message = new;
461                        }
462                }
463               
464                twitter_post_status( ic, message );
465                g_free( new );
466        }
467        g_free( cmds );
468}
[1b221e0]469
470void twitter_initmodule()
471{
472        struct prpl *ret = g_new0(struct prpl, 1);
473       
474        ret->name = "twitter";
475        ret->login = twitter_login;
476        ret->init = twitter_init;
477        ret->logout = twitter_logout;
478        ret->buddy_msg = twitter_buddy_msg;
479        ret->get_info = twitter_get_info;
480        ret->set_my_name = twitter_set_my_name;
481        ret->add_buddy = twitter_add_buddy;
482        ret->remove_buddy = twitter_remove_buddy;
483        ret->chat_msg = twitter_chat_msg;
484        ret->chat_invite = twitter_chat_invite;
485        ret->chat_leave = twitter_chat_leave;
486        ret->chat_with = twitter_chat_with;
487        ret->keepalive = twitter_keepalive;
488        ret->add_permit = twitter_add_permit;
489        ret->rem_permit = twitter_rem_permit;
490        ret->add_deny = twitter_add_deny;
491        ret->rem_deny = twitter_rem_deny;
492        ret->send_typing = twitter_send_typing;
493        ret->handle_cmp = g_strcasecmp;
[ffcdf13]494       
495        register_protocol(ret);
[1b221e0]496
[ffcdf13]497        /* And an identi.ca variant: */
498        ret = g_memdup(ret, sizeof(struct prpl));
499        ret->name = "identica";
[1b221e0]500        register_protocol(ret);
[62d2cfb]501
502        // Initialise the twitter_connections GSList.
503        twitter_connections = NULL;
[1b221e0]504}
Note: See TracBrowser for help on using the repository browser.