From 41b8c91c6b4f37c61983d0f8f94877bfcd0eb6d2 Mon Sep 17 00:00:00 2001
From: zach <zach@controlyourself.ca>
Date: Sun, 16 Nov 2008 21:46:24 -0500
Subject: [PATCH] trac685 Twitter bridge - Sync Twitter friends

darcs-hash:20081117024624-462f3-91f148111c2c43ddd302fc63cfa8c8ef14e1f38e.gz
---
 actions/twittersettings.php | 380 ++++++++++++++++++++++++++++--------
 classes/Foreign_link.php    |  20 +-
 lib/util.php                |   2 +-
 3 files changed, 313 insertions(+), 89 deletions(-)

diff --git a/actions/twittersettings.php b/actions/twittersettings.php
index 1115708a5c..bf8827958a 100644
--- a/actions/twittersettings.php
+++ b/actions/twittersettings.php
@@ -21,8 +21,18 @@ if (!defined('LACONICA')) { exit(1); }
 
 require_once(INSTALLDIR.'/lib/settingsaction.php');
 
+define('SUBSCRIPTIONS', 80);
+
 class TwittersettingsAction extends SettingsAction {
 
+	var $twit_id;
+	var $twit_username;
+	var $twit_password;
+	var $friends_count = 0;
+	var $noticesync;
+	var $repliessync;
+	var $friendsync;
+
 	function get_instructions() {
 		return _('Add your Twitter account to automatically send your notices to Twitter, ' .
 			'and subscribe to Twitter friends already here.');
@@ -32,10 +42,10 @@ class TwittersettingsAction extends SettingsAction {
 		$user = common_current_user();
 		$profile = $user->getProfile();
 		$fuser = NULL;
-		$flink = Foreign_link::getForeignLink($user->id, 1); // 1 == Twitter
+		$flink = Foreign_link::getByUserID($user->id, 1); // 1 == Twitter
 
 		if ($flink) {
-			$fuser = Foreign_user::getForeignUser($flink->foreign_id, 1);
+			$fuser = $flink->getForeignUser();
 		}
 
 		$this->form_header(_('Twitter settings'), $msg, $success);
@@ -58,8 +68,6 @@ class TwittersettingsAction extends SettingsAction {
 			common_element_end('p');
 			common_submit('remove', _('Remove'));
 		} else {
-
-			// XXX: Should we make an educated guess as to the twitter acct name? -- Zach
 			common_input('twitter_username', _('Twitter Username'),
 						 ($this->arg('twitter_username')) ? $this->arg('twitter_username') : $profile->nickname,
 						 _('No spaces, please.')); // hey, it's what Twitter says
@@ -76,8 +84,7 @@ class TwittersettingsAction extends SettingsAction {
 						($flink) ? ($flink->noticesync & FOREIGN_NOTICE_SEND_REPLY) : true);
 
 		common_checkbox('friendsync', _('Subscribe to my Twitter friends here.'),
-						($flink) ? ($flink->friendsync & FOREIGN_FRIEND_RECV) : false,
-						NULL, 'true', true);
+						($flink) ? ($flink->friendsync & FOREIGN_FRIEND_RECV) : true);
 
 		if ($flink) {
 			common_submit('save', _('Save'));
@@ -85,9 +92,105 @@ class TwittersettingsAction extends SettingsAction {
 			common_submit('add', _('Add'));
 		}
 
+		
+		$this->show_twitter_subscriptions();
+
 		common_element_end('form');
+		
 		common_show_footer();
 	}
+	
+	function subscribed_twitter_users() {
+
+		$current_user = common_current_user();
+		
+		$qry = 'SELECT user.* ' .
+			'FROM subscription ' . 
+			'JOIN user ON subscription.subscribed = user.id ' .
+			'JOIN foreign_link ON foreign_link.user_id = user.id ' . 
+			'WHERE subscriber = %d ' . 
+			'ORDER BY user.nickname';
+
+		$user = new User();
+		
+		$user->query(sprintf($qry, $current_user->id));
+
+		$users = array();
+
+		while ($user->fetch()) {
+			$users[] = clone($user);
+		}
+		
+		return $users;
+	}
+	
+	
+	function show_twitter_subscriptions() {
+	
+		common_debug('show twitter subs');
+		$friends = $this->subscribed_twitter_users();
+
+		$friends_count = count($friends);
+
+		common_debug("friends count = $friends_count");
+
+		if ($friends_count > 0) {
+
+			common_element('h3', NULL, _('Twitter Friends'));
+			common_element_start('div', array('id' => 'subscriptions'));
+			common_element_start('ul', array('id' => 'subscriptions_avatars'));
+
+			for ($i = 0; $i < min($friends_count, SUBSCRIPTIONS); $i++) {
+
+				$other = Profile::staticGet($friends[$i]->id);
+
+				if (!$other) {
+					common_log_db_error($subs, 'SELECT', __FILE__);
+					continue;
+				}
+				
+				common_element_start('li');
+				common_element_start('a', array('title' => ($other->fullname) ?
+												$other->fullname :
+												$other->nickname,
+												'href' => $other->profileurl,
+												'rel' => 'contact',
+												'class' => 'subscription'));
+				$avatar = $other->getAvatar(AVATAR_MINI_SIZE);
+				common_element('img', array('src' => (($avatar) ? common_avatar_display_url($avatar) :  common_default_avatar(AVATAR_MINI_SIZE)),
+											'width' => AVATAR_MINI_SIZE,
+											'height' => AVATAR_MINI_SIZE,
+											'class' => 'avatar mini',
+											'alt' =>  ($other->fullname) ?
+											$other->fullname :
+											$other->nickname));
+				common_element_end('a');
+				common_element_end('li');
+		
+			}
+
+			common_element_end('ul');
+			common_element_end('div');
+
+		}
+
+		// XXX Figure out a way to show all Twitter friends...
+		
+		/*
+		if ($subs_count > SUBSCRIPTIONS) {
+			common_element_start('p', array('id' => 'subscriptions_viewall'));
+
+			common_element('a', array('href' => common_local_url('subscriptions',
+																 array('nickname' => $profile->nickname)),
+									  'class' => 'moresubscriptions'),
+						   _('All subscriptions'));
+			common_element_end('p');
+		}
+		*/
+
+	}
+	
+	
 
 	function handle_post() {
 
@@ -110,13 +213,14 @@ class TwittersettingsAction extends SettingsAction {
 	}
 
 	function add_twitter_acct() {
-		$twitter_username = $this->trimmed('twitter_username');
-		$twitter_password = $this->trimmed('twitter_password');
-		$noticesync = $this->boolean('noticesync');
-		$replysync = $this->boolean('replysync');
-		$friendsync = $this->boolean('friendsync');
 
-		if (!Validate::string($twitter_username, array('min_length' => 1,
+		$this->twit_username = $this->trimmed('twitter_username');
+		$this->twit_password = $this->trimmed('twitter_password');
+		$this->noticesync = $this->boolean('noticesync');
+		$this->replysync = $this->boolean('replysync');
+		$this->friendsync = $this->boolean('friendsync');
+
+		if (!Validate::string($this->twit_username, array('min_length' => 1,
 													   'max_length' => 15,
 													   'format' => VALIDATE_NUM . VALIDATE_ALPHA . '_'))) {
 			$this->show_form(_('Username must have only numbers, upper- and lowercase letters, and underscore (_). 15 chars max.'));
@@ -124,59 +228,20 @@ class TwittersettingsAction extends SettingsAction {
 		}
 
 		// Verify this is a real Twitter user.
-		if (!$this->verify_credentials($twitter_username, $twitter_password)) {
+		if (!$this->verify_credentials()) {
 			$this->show_form(_('Could not verify your Twitter credentials!'));
 			return;
 		}
 
-		// Now that we have a valid Twitter user, we have to make another api call to
-		// find its Twitter ID.  Dumb, but true.
-		$twitter_id = $this->get_twitter_id($twitter_username, $twitter_password);
-
-		if (!$twitter_id) {
-			$this->show_form(sprintf(_('Unable to retrieve account information for "%s" from Twitter.'), $twitter_username));
+		if (!$this->twitter_user_info()) {
+			$this->show_form(sprintf(_('Unable to retrieve account information for "%s" from Twitter.'),
+				$twitter_username));
 			return;
 		}
 
-		$fuser = null;
-		$result = null;
+		$fuser_id = $this->update_twitter_user($this->twit_id, $this->twit_username);
 
-		// Check to see whether the Twitter user is already in the system,
-		// and update its username and uri if so.
-		$fuser = Foreign_User::getForeignUser($twitter_id, 1);
-
-		if ($fuser) {
-
-			$original = clone($fuser);
-
-			$fuser->nickname = $twitter_username;
-			$fuser->uri = "http://www.twitter.com/$twitter_username";
-
-			$result = $fuser->updateKeys($original);
-
-			if (!$result) {
-				common_log_db_error($fuser, 'UPDATE', __FILE__);
-			}
-
-		} else {
-
-			// Otherwise, add the Twitter user
-			$fuser = DB_DataObject::factory('foreign_user');
-
-			$fuser->nickname = $twitter_username;
-			$fuser->uri = "http://www.twitter.com/$twitter_username";
-			$fuser->id = $twitter_id;
-			$fuser->service = 1; // Twitter
-			$fuser->created = common_sql_now();
-			$result = $fuser->insert();
-
-			if (!$result) {
-				common_log_db_error($fuser, 'INSERT', __FILE__);
-			}
-
-		}
-
-		if (!$result) {
+		if (!$fuser_id) {
 			$this->show_form(_('Unable to save your Twitter settings!'));
 			return;
 		}
@@ -185,12 +250,12 @@ class TwittersettingsAction extends SettingsAction {
 
 		$flink = DB_DataObject::factory('foreign_link');
 		$flink->user_id = $user->id;
-		$flink->foreign_id = $fuser->id;
+		$flink->foreign_id = $fuser_id;
 		$flink->service = 1; // Twitter
-		$flink->credentials = $twitter_password;
+		$flink->credentials = $this->twit_password;
 		$flink->created = common_sql_now();
 
-		$this->set_flags($flink, $noticesync, $replysync, $friendsync);
+		$this->set_flags($flink, $this->noticesync, $this->replysync, $this->friendsync);
 
 		$flink_id = $flink->insert();
 
@@ -200,6 +265,10 @@ class TwittersettingsAction extends SettingsAction {
 			return;
 		}
 
+		if ($this->friendsync) {
+			$this->save_friends();
+		}
+
 		$this->show_form(_('Twitter settings saved.'), true);
 	}
 
@@ -207,7 +276,7 @@ class TwittersettingsAction extends SettingsAction {
 		$user = common_current_user();
 
 		// For now we assume one Twitter acct per Laconica acct
-		$flink = Foreign_link::getForeignLink($user->id, 1);
+		$flink = Foreign_link::getByUserID($user->id, 1);
 		$flink_foreign_id = $this->arg('flink_foreign_id');
 
 		if (!$flink) {
@@ -233,12 +302,12 @@ class TwittersettingsAction extends SettingsAction {
 	}
 
 	function save_preferences() {
-		$noticesync = $this->boolean('noticesync');
-		$friendsync = $this->boolean('friendsync');
-		$replysync = $this->boolean('replysync');
+		$this->noticesync = $this->boolean('noticesync');
+		$this->friendsync = $this->boolean('friendsync');
+		$this->replysync = $this->boolean('replysync');
 
 		$user = common_current_user();
-		$flink = Foreign_link::getForeignLink($user->id, 1);
+		$flink = Foreign_link::getByUserID($user->id, 1);
 
 		if (!$flink) {
 			common_log_db_error($flink, 'SELECT', __FILE__);
@@ -246,12 +315,21 @@ class TwittersettingsAction extends SettingsAction {
 			return;
 		}
 
-		$flink->query('BEGIN');
+		$this->twit_id = $flink->foreign_id;
+		$this->twit_password = $flink->credentials;
+
+		$fuser = $flink->getForeignUser();
+
+		if (!$fuser) {
+			common_log_db_error($fuser, 'SELECT', __FILE__);
+			$this->show_form(_('Couldn\'t save Twitter preferences.'));
+			return;
+		}
+
+		$this->twit_username = $fuser->nickname;
 
 		$original = clone($flink);
-
-		$this->set_flags($flink, $noticesync, $replysync, $friendsync);
-
+		$this->set_flags($flink, $this->noticesync, $this->replysync, $this->friendsync);
 		$result = $flink->update($original);
 
 		if ($result === FALSE) {
@@ -260,33 +338,39 @@ class TwittersettingsAction extends SettingsAction {
 			return;
 		}
 
-		$flink->query('COMMIT');
+		if ($this->friendsync) {
+			$this->save_friends();
+		}
 
 		$this->show_form(_('Twitter preferences saved.'));
-
-		return;
 	}
 
-	function get_twitter_id($user, $password) {
-		$uri = "http://twitter.com/users/show/$user.json";
-		$data = $this->get_twitter_data($uri, $user, $password);
+	function twitter_user_info() {
+		$uri = "http://twitter.com/users/show/$this->twit_username.json";
+		$data = $this->get_twitter_data($uri);
 
 		if (!$data) {
-			return NULL;
+			return false;
 		}
 
-		$user = json_decode($data);
+		$twit_user = json_decode($data);
 
-		if (!$user) {
-			return NULL;
+		if (!$twit_user) {
+			return false;
 		}
 
-		return $user->id;
+		$this->friends_count = $twit_user->friends_count;
+		$this->twit_id = $twit_user->id;
+
+		common_debug("Twitter_id = $this->twit_id");
+		common_debug("Friends_count = $this->friends_count");
+
+		return true;
 	}
 
-	function verify_credentials($user, $password) {
+	function verify_credentials() {
 		$uri = 'http://twitter.com/account/verify_credentials.json';
-		$data = $this->get_twitter_data($uri, $user, $password);
+		$data = $this->get_twitter_data($uri);
 
 		if (!$data) {
 			return false;
@@ -305,10 +389,10 @@ class TwittersettingsAction extends SettingsAction {
 		return false;
 	}
 
-	// PHP's cURL the best thing to use here? -- Zach
-	function get_twitter_data($uri, $user=NULL, $password=NULL) {
+	function get_twitter_data($uri) {
+
 		$options = array(
-				CURLOPT_USERPWD => "$user:$password",
+				CURLOPT_USERPWD => sprintf("%s:%s", $this->twit_username, $this->twit_password),
 				CURLOPT_RETURNTRANSFER	=> true,
 				CURLOPT_FAILONERROR		=> true,
 				CURLOPT_HEADER			=> false,
@@ -318,13 +402,15 @@ class TwittersettingsAction extends SettingsAction {
 				CURLOPT_TIMEOUT			=> 120
 		);
 
+
 		$ch = curl_init($uri);
 	    curl_setopt_array($ch, $options);
 	    $data = curl_exec($ch);
 	    $errmsg = curl_error($ch);
 
 		if ($errmsg) {
-			common_debug("cURL error: $errmsg - trying to load: $uri with user $user.", __FILE__);
+			common_debug("cURL error: $errmsg - trying to load: $uri with user $this->twit_user.",
+				__FILE__);
 		}
 
 		curl_close($ch);
@@ -353,4 +439,126 @@ class TwittersettingsAction extends SettingsAction {
 
 		$flink->profilesync = 0; // XXX: leave as default?
 	}
+
+	function save_friends() {
+
+		$uri = 'http://twitter.com/statuses/friends.json?page=';
+
+		$this->twitter_user_info();
+
+		// Calculate how many pages to get...
+		$pages = ceil($this->friends_count / 100);
+
+		common_debug("number of pages to get: $pages");
+
+		$friends = array();
+
+		for ($i = 1; $i <= $pages; $i++) {
+
+			$data = $this->get_twitter_data($uri . $i);
+
+			common_debug("fetching " . $uri . $i);
+
+			if (!$data) {
+				return false;
+			}
+
+			common_debug("got data");
+		
+			$more_friends = json_decode($data);
+			
+			if (!$more_friends) {
+				return false;
+			}
+
+	 		$friends = array_merge($friends, $more_friends);
+
+		}
+		
+		common_debug("number of friends =" + count($friends));
+
+		$user = common_current_user();
+
+	    foreach ($friends as $friend) {
+		
+			$friend_name = $friend->screen_name;
+			$friend_id = $friend->id;
+
+			// Update or create the Foreign_user record
+			$this->update_twitter_user($friend_id, $friend_name);
+			
+			// Check to see if there's a related local user
+			$flink = Foreign_link::getByForeignID($friend_id, 1);
+						
+			if ($flink) {
+				
+				// Get associated user
+				$friend_user = User::staticGet('id', $flink->user_id);				
+				subs_subscribe_to($user, $friend_user);
+				
+			}
+		}
+		
+	}
+
+	// Creates or Updates a Twitter user
+	function update_twitter_user($twitter_id, $screen_name) {
+
+		$fuser = null;
+
+		$uri = "http://twitter.com/$screen_name";
+
+		// Check to see whether the Twitter user is already in the system,
+		// and update its screen name and uri if so.
+		$fuser = Foreign_User::getForeignUser($twitter_id, 1);
+
+		if ($fuser) {
+
+			// Only update if Twitter screen name has changed
+			if ($fuser->nickname != $screen_name) {
+
+				$original = clone($fuser);
+				$fuser->nickname = $screen_name;
+				$fuser->uri = $uri;
+				$result = $fuser->updateKeys($original);
+
+				if (!$result) {
+					common_log_db_error($fuser, 'UPDATE', __FILE__);
+					return null;
+				}
+
+				common_debug(
+					sprintf('Updated Twitter user %, screen name was: %, now: %s.',
+						$twitter_id, $original->nickname, $screen_name));
+			}
+
+			common_debug("No update for $screen_name needed.");
+
+		} else {
+
+				// Otherwise, create a new Twitter user
+				$fuser = DB_DataObject::factory('foreign_user');
+
+				$fuser->nickname = $screen_name;
+				$fuser->uri = $uri;
+				$fuser->id = $twitter_id;
+				$fuser->service = 1; // Twitter
+				$fuser->created = common_sql_now();
+				$result = $fuser->insert();
+
+				if (!$result) {
+					common_debug("Failed to add new Twitter user: $twitter_id - $screen_name.");
+					common_log_db_error($fuser, 'INSERT', __FILE__);
+					return null;
+				}
+
+				common_debug("Added new Twitter user: $twitter_id - $screen_name.");
+
+				//	common_debug(print_r($friend, true));
+		}
+
+		return $fuser->id;
+
+	}
+
 }
\ No newline at end of file
diff --git a/classes/Foreign_link.php b/classes/Foreign_link.php
index 1ba74e74fa..e3f7bfdd83 100644
--- a/classes/Foreign_link.php
+++ b/classes/Foreign_link.php
@@ -28,7 +28,7 @@ class Foreign_link extends Memcached_DataObject
 
 	// XXX:  This only returns a 1->1 single obj mapping.  Change?  Or make
 	// a getForeignUsers() that returns more than one? --Zach
-	static function getForeignLink($user_id, $service) {
+	static function getByUserID($user_id, $service) {
 		$flink = new Foreign_link();
 		$flink->service = $service;
 		$flink->user_id = $user_id;
@@ -41,11 +41,27 @@ class Foreign_link extends Memcached_DataObject
 		return NULL;		
 	}
 	
+	static function getByForeignID($foreign_id, $service) {
+		$flink = new Foreign_link();
+		$flink->service = $service;
+		$flink->foreign_id = $foreign_id;
+		$flink->limit(1);
+
+		if ($flink->find(TRUE)) {
+			return $flink;
+		}
+
+		return NULL;		
+	}
+	
+	
 	// Convenience method
 	function getForeignUser() {
 		
 		$fuser = new Foreign_user();
-		
+
+		common_debug("service = " . $this->service);
+		common_debug("foreign_id = " . $this->foreign_id);
 		$fuser->service = $this->service;
 		$fuser->id = $this->foreign_id;
 		
diff --git a/lib/util.php b/lib/util.php
index 7dfbc1578e..bf96be264d 100644
--- a/lib/util.php
+++ b/lib/util.php
@@ -1266,7 +1266,7 @@ function common_save_replies($notice) {
 function common_broadcast_notice($notice, $remote=false) {
 
 	// Check to see if notice should go to Twitter
-	$flink = Foreign_link::getForeignLink($notice->profile_id, 1); // 1 == Twitter
+	$flink = Foreign_link::getByUserID($notice->profile_id, 1); // 1 == Twitter
 	if (($flink->noticesync & FOREIGN_NOTICE_SEND) == FOREIGN_NOTICE_SEND) {
 
 		// If it's not a Twitter-style reply, or if the user WANTS to send replies...