source: protocols/twitter/twitter.c @ ba3233c

Last change on this file since ba3233c was ba3233c, checked in by Wilmer van der Gaast <wilmer@…>, at 2010-05-30T10:16:37Z

Merge from Geert: Added ability to follow/unfollow Twitter people. (Unfollow
won't work with mode=one.)

  • Property mode set to 100644
File size: 10.6 KB
Line 
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"
25#include "oauth.h"
26#include "twitter.h"
27#include "twitter_http.h"
28#include "twitter_lib.h"
29#include "url.h"
30
31/**
32 * Main loop function
33 */
34gboolean twitter_main_loop(gpointer data, gint fd, b_input_condition cond)
35{
36        struct im_connection *ic = data;
37       
38        // Check if we are still logged in...
39        if (!g_slist_find( twitter_connections, ic ))
40                return 0;
41
42        // If the user uses multiple private message windows we need to get the
43        // users buddies.
44        if (g_strcasecmp(set_getstr(&ic->acc->set, "mode"), "many") == 0)
45                twitter_get_statuses_friends(ic, -1);
46
47        // Do stuff..
48        twitter_get_home_timeline(ic, -1);
49
50        // If we are still logged in run this function again after timeout.
51        return (ic->flags & OPT_LOGGED_IN) == OPT_LOGGED_IN;
52}
53
54static void twitter_main_loop_start( struct im_connection *ic )
55{
56        struct twitter_data *td = ic->proto_data;
57       
58        imcb_log( ic, "Connecting to Twitter" );
59
60        // Run this once. After this queue the main loop function.
61        twitter_main_loop(ic, -1, 0);
62
63        // Queue the main_loop
64        // Save the return value, so we can remove the timeout on logout.
65        td->main_loop_id = b_timeout_add(60000, twitter_main_loop, ic);
66}
67
68
69static const struct oauth_service twitter_oauth =
70{
71        "http://api.twitter.com/oauth/request_token",
72        "http://api.twitter.com/oauth/access_token",
73        "https://api.twitter.com/oauth/authorize",
74        .consumer_key = "xsDNKJuNZYkZyMcu914uEA",
75        .consumer_secret = "FCxqcr0pXKzsF9ajmP57S3VQ8V6Drk4o2QYtqMcOszo",
76};
77
78static gboolean twitter_oauth_callback( struct oauth_info *info );
79
80static void twitter_oauth_start( struct im_connection *ic )
81{
82        struct twitter_data *td = ic->proto_data;
83       
84        imcb_log( ic, "Requesting OAuth request token" );
85
86        td->oauth_info = oauth_request_token( &twitter_oauth, twitter_oauth_callback, ic );
87}
88
89static gboolean twitter_oauth_callback( struct oauth_info *info )
90{
91        struct im_connection *ic = info->data;
92        struct twitter_data *td;
93       
94        if( !g_slist_find( twitter_connections, ic ) )
95                return FALSE;
96       
97        td = ic->proto_data;
98        if( info->stage == OAUTH_REQUEST_TOKEN )
99        {
100                char name[strlen(ic->acc->user)+9], *msg;
101               
102                if( info->request_token == NULL )
103                {
104                        imcb_error( ic, "OAuth error: %s", info->http->status_string );
105                        imc_logout( ic, TRUE );
106                        return FALSE;
107                }
108               
109                sprintf( name, "twitter_%s", ic->acc->user );
110                msg = g_strdup_printf( "To finish OAuth authentication, please visit "
111                                       "%s and respond with the resulting PIN code.",
112                                       info->auth_url );
113                imcb_buddy_msg( ic, name, msg, 0, 0 );
114                g_free( msg );
115        }
116        else if( info->stage == OAUTH_ACCESS_TOKEN )
117        {
118                if( info->token == NULL || info->token_secret == NULL )
119                {
120                        imcb_error( ic, "OAuth error: %s", info->http->status_string );
121                        imc_logout( ic, TRUE );
122                        return FALSE;
123                }
124               
125                /* IM mods didn't do this so far and it's ugly but I should
126                   be able to get away with it... */
127                g_free( ic->acc->pass );
128                ic->acc->pass = oauth_to_string( info );
129               
130                twitter_main_loop_start( ic );
131        }
132       
133        return TRUE;
134}
135
136
137static char *set_eval_mode( set_t *set, char *value )
138{
139        if( g_strcasecmp( value, "one" ) == 0 ||
140            g_strcasecmp( value, "many" ) == 0 ||
141            g_strcasecmp( value, "chat" ) == 0 )
142                return value;
143        else
144                return NULL;
145}
146
147static gboolean twitter_length_check( struct im_connection *ic, gchar *msg )
148{
149        int max = set_getint( &ic->acc->set, "message_length" ), len;
150       
151        if( max == 0 || ( len = g_utf8_strlen( msg, -1 ) ) <= max )
152                return TRUE;
153       
154        imcb_error( ic, "Maximum message length exceeded: %d > %d", len, max );
155       
156        return FALSE;
157}
158
159static void twitter_init( account_t *acc )
160{
161        set_t *s;
162       
163        s = set_add( &acc->set, "base_url", TWITTER_API_URL, NULL, acc );
164        s->flags |= ACC_SET_OFFLINE_ONLY;
165       
166        s = set_add( &acc->set, "message_length", "140", set_eval_int, acc );
167       
168        s = set_add( &acc->set, "mode", "one", set_eval_mode, acc );
169        s->flags |= ACC_SET_OFFLINE_ONLY;
170       
171        s = set_add( &acc->set, "oauth", "true", set_eval_bool, acc );
172}
173
174/**
175 * Login method. Since the twitter API works with seperate HTTP request we
176 * only save the user and pass to the twitter_data object.
177 */
178static void twitter_login( account_t *acc )
179{
180        struct im_connection *ic = imcb_new( acc );
181        struct twitter_data *td;
182        char name[strlen(acc->user)+9];
183        url_t url;
184
185        if( !url_set( &url, set_getstr( &ic->acc->set, "base_url" ) ) ||
186            ( url.proto != PROTO_HTTP && url.proto != PROTO_HTTPS ) )
187        {
188                imcb_error( ic, "Incorrect API base URL: %s", set_getstr( &ic->acc->set, "base_url" ) );
189                imc_logout( ic, FALSE );
190                return;
191        }
192       
193        twitter_connections = g_slist_append( twitter_connections, ic );
194        td = g_new0( struct twitter_data, 1 );
195        ic->proto_data = td;
196       
197        td->url_ssl = url.proto == PROTO_HTTPS;
198        td->url_port = url.port;
199        td->url_host = g_strdup( url.host );
200        if( strcmp( url.file, "/" ) != 0 )
201                td->url_path = g_strdup( url.file );
202        else
203                td->url_path = g_strdup( "" );
204       
205        td->user = acc->user;
206        if( strstr( acc->pass, "oauth_token=" ) )
207                td->oauth_info = oauth_from_string( acc->pass, &twitter_oauth );
208       
209        sprintf( name, "twitter_%s", acc->user );
210        imcb_add_buddy( ic, name, NULL );
211        imcb_buddy_status( ic, name, OPT_LOGGED_IN, NULL, NULL );
212       
213        if( td->oauth_info || !set_getbool( &acc->set, "oauth" ) )
214                twitter_main_loop_start( ic );
215        else
216                twitter_oauth_start( ic );
217}
218
219/**
220 * Logout method. Just free the twitter_data.
221 */
222static void twitter_logout( struct im_connection *ic )
223{
224        struct twitter_data *td = ic->proto_data;
225       
226        // Set the status to logged out.
227        ic->flags = 0;
228
229        // Remove the main_loop function from the function queue.
230        b_event_remove(td->main_loop_id);
231
232        if(td->home_timeline_gc)
233                imcb_chat_free(td->home_timeline_gc);
234
235        if( td )
236        {
237                oauth_info_free( td->oauth_info );
238                g_free( td->pass );
239                g_free( td );
240        }
241
242        twitter_connections = g_slist_remove( twitter_connections, ic );
243}
244
245/**
246 *
247 */
248static int twitter_buddy_msg( struct im_connection *ic, char *who, char *message, int away )
249{
250        struct twitter_data *td = ic->proto_data;
251       
252        if (g_strncasecmp(who, "twitter_", 8) == 0 &&
253            g_strcasecmp(who + 8, ic->acc->user) == 0)
254        {
255                if( set_getbool( &ic->acc->set, "oauth" ) &&
256                    td->oauth_info && td->oauth_info->token == NULL )
257                {
258                        if( !oauth_access_token( message, td->oauth_info ) )
259                        {
260                                imcb_error( ic, "OAuth error: %s", "Failed to send access token request" );
261                                imc_logout( ic, TRUE );
262                                return FALSE;
263                        }
264                }
265                else if( twitter_length_check(ic, message) )
266                        twitter_post_status(ic, message);
267        }
268        else
269        {
270                twitter_direct_messages_new(ic, who, message);
271        }
272        return( 0 );
273}
274
275/**
276 *
277 */
278static void twitter_set_my_name( struct im_connection *ic, char *info )
279{
280}
281
282static void twitter_get_info(struct im_connection *ic, char *who) 
283{
284}
285
286static void twitter_add_buddy( struct im_connection *ic, char *who, char *group )
287{
288        twitter_friendships_create_destroy(ic, who, 1);
289}
290
291static void twitter_remove_buddy( struct im_connection *ic, char *who, char *group )
292{
293        twitter_friendships_create_destroy(ic, who, 0);
294}
295
296static void twitter_chat_msg( struct groupchat *c, char *message, int flags )
297{
298        if( c && message && twitter_length_check(c->ic, message))
299                twitter_post_status(c->ic, message);
300}
301
302static void twitter_chat_invite( struct groupchat *c, char *who, char *message )
303{
304}
305
306static void twitter_chat_leave( struct groupchat *c )
307{
308        struct twitter_data *td = c->ic->proto_data;
309       
310        if( c != td->home_timeline_gc )
311                return; /* WTF? */
312       
313        /* If the user leaves the channel: Fine. Rejoin him/her once new
314           tweets come in. */
315        imcb_chat_free(td->home_timeline_gc);
316        td->home_timeline_gc = NULL;
317}
318
319static struct groupchat *twitter_chat_with( struct im_connection *ic, char *who )
320{
321        return NULL;
322}
323
324static void twitter_keepalive( struct im_connection *ic )
325{
326}
327
328static void twitter_add_permit( struct im_connection *ic, char *who )
329{
330}
331
332static void twitter_rem_permit( struct im_connection *ic, char *who )
333{
334}
335
336static void twitter_add_deny( struct im_connection *ic, char *who )
337{
338}
339
340static void twitter_rem_deny( struct im_connection *ic, char *who )
341{
342}
343
344static int twitter_send_typing( struct im_connection *ic, char *who, int typing )
345{
346        return( 1 );
347}
348
349//static char *twitter_set_display_name( set_t *set, char *value )
350//{
351//      return value;
352//}
353
354void twitter_initmodule()
355{
356        struct prpl *ret = g_new0(struct prpl, 1);
357       
358        ret->name = "twitter";
359        ret->login = twitter_login;
360        ret->init = twitter_init;
361        ret->logout = twitter_logout;
362        ret->buddy_msg = twitter_buddy_msg;
363        ret->get_info = twitter_get_info;
364        ret->set_my_name = twitter_set_my_name;
365        ret->add_buddy = twitter_add_buddy;
366        ret->remove_buddy = twitter_remove_buddy;
367        ret->chat_msg = twitter_chat_msg;
368        ret->chat_invite = twitter_chat_invite;
369        ret->chat_leave = twitter_chat_leave;
370        ret->chat_with = twitter_chat_with;
371        ret->keepalive = twitter_keepalive;
372        ret->add_permit = twitter_add_permit;
373        ret->rem_permit = twitter_rem_permit;
374        ret->add_deny = twitter_add_deny;
375        ret->rem_deny = twitter_rem_deny;
376        ret->send_typing = twitter_send_typing;
377        ret->handle_cmp = g_strcasecmp;
378
379        register_protocol(ret);
380
381        // Initialise the twitter_connections GSList.
382        twitter_connections = NULL;
383}
384
Note: See TracBrowser for help on using the repository browser.