diff options
Diffstat (limited to 'protocols/twitter/twitter.c')
| -rw-r--r-- | protocols/twitter/twitter.c | 553 | 
1 files changed, 553 insertions, 0 deletions
| diff --git a/protocols/twitter/twitter.c b/protocols/twitter/twitter.c new file mode 100644 index 00000000..a2f2325c --- /dev/null +++ b/protocols/twitter/twitter.c @@ -0,0 +1,553 @@ +/***************************************************************************\ +*                                                                           * +*  BitlBee - An IRC to IM gateway                                           * +*  Simple module to facilitate twitter functionality.                       * +*                                                                           * +*  Copyright 2009 Geert Mulders <g.c.w.m.mulders@gmail.com>                 * +*                                                                           * +*  This library is free software; you can redistribute it and/or            * +*  modify it under the terms of the GNU Lesser General Public               * +*  License as published by the Free Software Foundation, version            * +*  2.1.                                                                     * +*                                                                           * +*  This library is distributed in the hope that it will be useful,          * +*  but WITHOUT ANY WARRANTY; without even the implied warranty of           * +*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU        * +*  Lesser General Public License for more details.                          * +*                                                                           * +*  You should have received a copy of the GNU Lesser General Public License * +*  along with this library; if not, write to the Free Software Foundation,  * +*  Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA           * +*                                                                           * +****************************************************************************/ + +#include "nogaim.h" +#include "oauth.h" +#include "twitter.h" +#include "twitter_http.h" +#include "twitter_lib.h" +#include "url.h" + +/** + * Main loop function + */ +gboolean twitter_main_loop(gpointer data, gint fd, b_input_condition cond) +{ +	struct im_connection *ic = data; +	 +	// Check if we are still logged in... +	if (!g_slist_find( twitter_connections, ic )) +		return 0; + +	// Do stuff.. +	twitter_get_home_timeline(ic, -1); + +	// If we are still logged in run this function again after timeout. +	return (ic->flags & OPT_LOGGED_IN) == OPT_LOGGED_IN; +} + +static void twitter_main_loop_start( struct im_connection *ic ) +{ +	struct twitter_data *td = ic->proto_data; +	 +	imcb_log( ic, "Getting initial statuses" ); + +	// Run this once. After this queue the main loop function. +	twitter_main_loop(ic, -1, 0); + +	// Queue the main_loop +	// Save the return value, so we can remove the timeout on logout. +	td->main_loop_id = b_timeout_add(60000, twitter_main_loop, ic); +} + +static void twitter_oauth_start( struct im_connection *ic ); + +void twitter_login_finish( struct im_connection *ic ) +{ +	struct twitter_data *td = ic->proto_data; +	 +	if( set_getbool( &ic->acc->set, "oauth" ) && !td->oauth_info ) +		twitter_oauth_start( ic ); +	else if( g_strcasecmp( set_getstr( &ic->acc->set, "mode" ), "one" ) != 0 && +	         !( td->flags & TWITTER_HAVE_FRIENDS ) ) +	{ +		imcb_log( ic, "Getting contact list" ); +		twitter_get_statuses_friends( ic, -1 ); +	} +	else +		twitter_main_loop_start( ic ); +} + +static const struct oauth_service twitter_oauth = +{ +	"http://api.twitter.com/oauth/request_token", +	"http://api.twitter.com/oauth/access_token", +	"https://api.twitter.com/oauth/authorize", +	.consumer_key = "xsDNKJuNZYkZyMcu914uEA", +	.consumer_secret = "FCxqcr0pXKzsF9ajmP57S3VQ8V6Drk4o2QYtqMcOszo", +}; + +static gboolean twitter_oauth_callback( struct oauth_info *info ); + +static void twitter_oauth_start( struct im_connection *ic ) +{ +	struct twitter_data *td = ic->proto_data; +	 +	imcb_log( ic, "Requesting OAuth request token" ); + +	td->oauth_info = oauth_request_token( &twitter_oauth, twitter_oauth_callback, ic ); +} + +static gboolean twitter_oauth_callback( struct oauth_info *info ) +{ +	struct im_connection *ic = info->data; +	struct twitter_data *td; +	 +	if( !g_slist_find( twitter_connections, ic ) ) +		return FALSE; +	 +	td = ic->proto_data; +	if( info->stage == OAUTH_REQUEST_TOKEN ) +	{ +		char name[strlen(ic->acc->user)+9], *msg; +		 +		if( info->request_token == NULL ) +		{ +			imcb_error( ic, "OAuth error: %s", info->http->status_string ); +			imc_logout( ic, TRUE ); +			return FALSE; +		} +		 +		sprintf( name, "%s_%s", td->prefix, ic->acc->user ); +		msg = g_strdup_printf( "To finish OAuth authentication, please visit " +		                       "%s and respond with the resulting PIN code.", +		                       info->auth_url ); +		imcb_buddy_msg( ic, name, msg, 0, 0 ); +		g_free( msg ); +	} +	else if( info->stage == OAUTH_ACCESS_TOKEN ) +	{ +		if( info->token == NULL || info->token_secret == NULL ) +		{ +			imcb_error( ic, "OAuth error: %s", info->http->status_string ); +			imc_logout( ic, TRUE ); +			return FALSE; +		} +		 +		/* IM mods didn't do this so far and it's ugly but I should +		   be able to get away with it... */ +		g_free( ic->acc->pass ); +		ic->acc->pass = oauth_to_string( info ); +		 +		twitter_login_finish( ic ); +	} +	 +	return TRUE; +} + + +static char *set_eval_mode( set_t *set, char *value ) +{ +	if( g_strcasecmp( value, "one" ) == 0 || +	    g_strcasecmp( value, "many" ) == 0 || +	    g_strcasecmp( value, "chat" ) == 0 ) +		return value; +	else +		return NULL; +} + +static gboolean twitter_length_check( struct im_connection *ic, gchar *msg ) +{ +	int max = set_getint( &ic->acc->set, "message_length" ), len; +	 +	if( max == 0 || ( len = g_utf8_strlen( msg, -1 ) ) <= max ) +		return TRUE; +	 +	imcb_error( ic, "Maximum message length exceeded: %d > %d", len, max ); +	 +	return FALSE; +} + +static void twitter_init( account_t *acc ) +{ +	set_t *s; +	char *def_url; +	char *def_oauth; +	 +	if( strcmp( acc->prpl->name, "twitter" ) == 0 ) +	{ +		def_url = TWITTER_API_URL; +		def_oauth = "true"; +	} +	else /* if( strcmp( acc->prpl->name, "identica" ) == 0 ) */ +	{ +		def_url = IDENTICA_API_URL; +		def_oauth = "false"; +	} +	 +	s = set_add( &acc->set, "auto_reply_timeout", "10800", set_eval_int, acc ); +	 +	s = set_add( &acc->set, "base_url", def_url, NULL, acc ); +	s->flags |= ACC_SET_OFFLINE_ONLY; +	 +	s = set_add( &acc->set, "commands", "true", set_eval_bool, acc ); +	 +	s = set_add( &acc->set, "message_length", "140", set_eval_int, acc ); +	 +	s = set_add( &acc->set, "mode", "one", set_eval_mode, acc ); +	s->flags |= ACC_SET_OFFLINE_ONLY; +	 +	s = set_add( &acc->set, "oauth", def_oauth, set_eval_bool, acc ); +} + +/** + * Login method. Since the twitter API works with seperate HTTP request we  + * only save the user and pass to the twitter_data object. + */ +static void twitter_login( account_t *acc ) +{ +	struct im_connection *ic = imcb_new( acc ); +	struct twitter_data *td; +	char name[strlen(acc->user)+9]; +	url_t url; + +	if( !url_set( &url, set_getstr( &ic->acc->set, "base_url" ) ) || +	    ( url.proto != PROTO_HTTP && url.proto != PROTO_HTTPS ) ) +	{ +		imcb_error( ic, "Incorrect API base URL: %s", set_getstr( &ic->acc->set, "base_url" ) ); +		imc_logout( ic, FALSE ); +		return; +	} +	 +	twitter_connections = g_slist_append( twitter_connections, ic ); +	td = g_new0( struct twitter_data, 1 ); +	ic->proto_data = td; +	 +	td->url_ssl = url.proto == PROTO_HTTPS; +	td->url_port = url.port; +	td->url_host = g_strdup( url.host ); +	if( strcmp( url.file, "/" ) != 0 ) +		td->url_path = g_strdup( url.file ); +	else +		td->url_path = g_strdup( "" ); +	if( g_str_has_suffix( url.host, ".com" ) ) +		td->prefix = g_strndup( url.host, strlen( url.host ) - 4 ); +	else +		td->prefix = g_strdup( url.host ); +	 +	td->user = acc->user; +	if( strstr( acc->pass, "oauth_token=" ) ) +		td->oauth_info = oauth_from_string( acc->pass, &twitter_oauth ); +	 +	sprintf( name, "%s_%s", td->prefix, acc->user ); +	imcb_add_buddy( ic, name, NULL ); +	imcb_buddy_status( ic, name, OPT_LOGGED_IN, NULL, NULL ); +	 +	imcb_log( ic, "Connecting" ); +	 +	twitter_login_finish( ic ); +} + +/** + * Logout method. Just free the twitter_data. + */ +static void twitter_logout( struct im_connection *ic ) +{ +	struct twitter_data *td = ic->proto_data; +	 +	// Set the status to logged out. +	ic->flags = 0; + +	// Remove the main_loop function from the function queue. +	b_event_remove(td->main_loop_id); + +	if(td->home_timeline_gc) +		imcb_chat_free(td->home_timeline_gc); + +	if( td ) +	{ +		oauth_info_free( td->oauth_info ); +		g_free( td->prefix ); +		g_free( td->url_host ); +		g_free( td->url_path ); +		g_free( td->pass ); +		g_free( td ); +	} + +	twitter_connections = g_slist_remove( twitter_connections, ic ); +} + +static void twitter_handle_command( struct im_connection *ic, char *message ); + +/** + * + */ +static int twitter_buddy_msg( struct im_connection *ic, char *who, char *message, int away ) +{ +	struct twitter_data *td = ic->proto_data; +	int plen = strlen( td->prefix ); +	 +	if (g_strncasecmp(who, td->prefix, plen) == 0 && who[plen] == '_' && +	    g_strcasecmp(who + plen + 1, ic->acc->user) == 0) +	{ +		if( set_getbool( &ic->acc->set, "oauth" ) && +		    td->oauth_info && td->oauth_info->token == NULL ) +		{ +			char pin[strlen(message)+1], *s; +			 +			strcpy( pin, message ); +			for( s = pin + sizeof( pin ) - 2; s > pin && isspace( *s ); s -- ) +				*s = '\0'; +			for( s = pin; *s && isspace( *s ); s ++ ) {} +			 +			if( !oauth_access_token( s, td->oauth_info ) ) +			{ +				imcb_error( ic, "OAuth error: %s", "Failed to send access token request" ); +				imc_logout( ic, TRUE ); +				return FALSE; +			} +		} +		else +			twitter_handle_command(ic, message); +	} +	else +	{ +		twitter_direct_messages_new(ic, who, message); +	} +	return( 0 ); +} + +/** + * + */ +static void twitter_set_my_name( struct im_connection *ic, char *info ) +{ +} + +static void twitter_get_info(struct im_connection *ic, char *who)  +{ +} + +static void twitter_add_buddy( struct im_connection *ic, char *who, char *group ) +{ +	twitter_friendships_create_destroy(ic, who, 1); +} + +static void twitter_remove_buddy( struct im_connection *ic, char *who, char *group ) +{ +	twitter_friendships_create_destroy(ic, who, 0); +} + +static void twitter_chat_msg( struct groupchat *c, char *message, int flags ) +{ +	if( c && message ) +		twitter_handle_command( c->ic, message ); +} + +static void twitter_chat_invite( struct groupchat *c, char *who, char *message ) +{ +} + +static void twitter_chat_leave( struct groupchat *c ) +{ +	struct twitter_data *td = c->ic->proto_data; +	 +	if( c != td->home_timeline_gc ) +		return; /* WTF? */ +	 +	/* If the user leaves the channel: Fine. Rejoin him/her once new +	   tweets come in. */ +	imcb_chat_free(td->home_timeline_gc); +	td->home_timeline_gc = NULL; +} + +static struct groupchat *twitter_chat_with( struct im_connection *ic, char *who ) +{ +	return NULL; +} + +static void twitter_keepalive( struct im_connection *ic ) +{ +} + +static void twitter_add_permit( struct im_connection *ic, char *who ) +{ +} + +static void twitter_rem_permit( struct im_connection *ic, char *who ) +{ +} + +static void twitter_add_deny( struct im_connection *ic, char *who ) +{ +} + +static void twitter_rem_deny( struct im_connection *ic, char *who ) +{ +} + +static int twitter_send_typing( struct im_connection *ic, char *who, int typing ) +{ +	return( 1 ); +} + +//static char *twitter_set_display_name( set_t *set, char *value ) +//{ +//	return value; +//} + +static void twitter_buddy_data_add( struct bee_user *bu ) +{ +	bu->data = g_new0( struct twitter_user_data, 1 ); +} + +static void twitter_buddy_data_free( struct bee_user *bu ) +{ +	g_free( bu->data ); +} + +static void twitter_handle_command( struct im_connection *ic, char *message ) +{ +	struct twitter_data *td = ic->proto_data; +	char *cmds, **cmd; +	 +	cmds = g_strdup( message ); +	cmd = split_command_parts( cmds ); +	 +	if( cmd[0] == NULL ) +	{ +		g_free( cmds ); +		return; +	} +	else if( !set_getbool( &ic->acc->set, "commands" ) ) +	{ +		/* Not supporting commands. */ +	} +	else if( g_strcasecmp( cmd[0], "undo" ) == 0 ) +	{ +		guint64 id; +		 +		if( cmd[1] ) +			id = g_ascii_strtoull( cmd[1], NULL, 10 ); +		else +			id = td->last_status_id; +		 +		/* TODO: User feedback. */ +		if( id ) +			twitter_status_destroy( ic, id ); +		 +		g_free( cmds ); +		return; +	} +	else if( g_strcasecmp( cmd[0], "follow" ) == 0 && cmd[1] ) +	{ +		twitter_add_buddy( ic, cmd[1], NULL ); +		g_free( cmds ); +		return; +	} +	else if( g_strcasecmp( cmd[0], "unfollow" ) == 0 && cmd[1] ) +	{ +		twitter_remove_buddy( ic, cmd[1], NULL ); +		g_free( cmds ); +		return; +	} +	else if( g_strcasecmp( cmd[0], "rt" ) == 0 && cmd[1] ) +	{ +		struct twitter_user_data *tud; +		bee_user_t *bu; +		guint64 id; +		 +		if( ( bu = bee_user_by_handle( ic->bee, ic, cmd[1] ) ) && +		    ( tud = bu->data ) && tud->last_id ) +			id = tud->last_id; +		else +			id = g_ascii_strtoull( cmd[1], NULL, 10 ); +		 +		td->last_status_id = 0; +		if( id ) +			twitter_status_retweet( ic, id ); +		 +		g_free( cmds ); +		return; +	} +	else if( g_strcasecmp( cmd[0], "post" ) == 0 ) +	{ +		message += 5; +	} +	 +	{ +		guint64 in_reply_to = 0; +		char *s, *new = NULL; +		bee_user_t *bu; +		 +		if( !twitter_length_check( ic, message ) ) +		{ +			g_free( cmds ); +		  	return; +		} +		 +		s = cmd[0] + strlen( cmd[0] ) - 1; +		if( s > cmd[0] && ( *s == ':' || *s == ',' ) ) +		{ +			*s = '\0'; +			 +			if( ( bu = bee_user_by_handle( ic->bee, ic, cmd[0] ) ) ) +			{ +				struct twitter_user_data *tud = bu->data; +				 +				new = g_strdup_printf( "@%s %s", bu->handle, +				                       message + ( s - cmd[0] ) + 2 ); +				message = new; +				 +				if( time( NULL ) < tud->last_time + +				    set_getint( &ic->acc->set, "auto_reply_timeout" ) ) +					in_reply_to = tud->last_id; +			} +		} +		 +		/* If the user runs undo between this request and its response +		   this would delete the second-last Tweet. Prevent that. */ +		td->last_status_id = 0; +		twitter_post_status( ic, message, in_reply_to ); +		g_free( new ); +	} +	g_free( cmds ); +} + +void twitter_initmodule() +{ +	struct prpl *ret = g_new0(struct prpl, 1); +	 +	ret->name = "twitter"; +	ret->login = twitter_login; +	ret->init = twitter_init; +	ret->logout = twitter_logout; +	ret->buddy_msg = twitter_buddy_msg; +	ret->get_info = twitter_get_info; +	ret->set_my_name = twitter_set_my_name; +	ret->add_buddy = twitter_add_buddy; +	ret->remove_buddy = twitter_remove_buddy; +	ret->chat_msg = twitter_chat_msg; +	ret->chat_invite = twitter_chat_invite; +	ret->chat_leave = twitter_chat_leave; +	ret->chat_with = twitter_chat_with; +	ret->keepalive = twitter_keepalive; +	ret->add_permit = twitter_add_permit; +	ret->rem_permit = twitter_rem_permit; +	ret->add_deny = twitter_add_deny; +	ret->rem_deny = twitter_rem_deny; +	ret->send_typing = twitter_send_typing; +	ret->buddy_data_add = twitter_buddy_data_add; +	ret->buddy_data_free = twitter_buddy_data_free; +	ret->handle_cmp = g_strcasecmp; +	 +	register_protocol(ret); + +	/* And an identi.ca variant: */ +	ret = g_memdup(ret, sizeof(struct prpl)); +	ret->name = "identica"; +	register_protocol(ret); + +	// Initialise the twitter_connections GSList. +	twitter_connections = NULL; +} | 
