300 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			300 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/*
 | 
						|
 * Laconica - a distributed open-source microblogging tool
 | 
						|
 * Copyright (C) 2008, Controlez-Vous, Inc.
 | 
						|
 *
 | 
						|
 * This program is free software: you can redistribute it and/or modify
 | 
						|
 * it under the terms of the GNU Affero General Public License as published by
 | 
						|
 * the Free Software Foundation, either version 3 of the License, or
 | 
						|
 * (at your option) any later version.
 | 
						|
 *
 | 
						|
 * This program 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 Affero General Public License for more details.
 | 
						|
 *
 | 
						|
 * You should have received a copy of the GNU Affero General Public License
 | 
						|
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
if (!defined('LACONICA')) { exit(1); }
 | 
						|
 | 
						|
define("TWITTER_SERVICE", 1); // Twitter is foreign_service ID 1
 | 
						|
 | 
						|
function get_twitter_data($uri, $screen_name, $password)
 | 
						|
{
 | 
						|
 | 
						|
    $options = array(
 | 
						|
            CURLOPT_USERPWD => sprintf("%s:%s", $screen_name, $password),
 | 
						|
            CURLOPT_RETURNTRANSFER    => true,
 | 
						|
            CURLOPT_FAILONERROR        => true,
 | 
						|
            CURLOPT_HEADER            => false,
 | 
						|
            CURLOPT_FOLLOWLOCATION    => true,
 | 
						|
            CURLOPT_USERAGENT      => "Laconica",
 | 
						|
            CURLOPT_CONNECTTIMEOUT    => 120,
 | 
						|
            CURLOPT_TIMEOUT            => 120,
 | 
						|
            # Twitter is strict about accepting invalid "Expect" headers
 | 
						|
            CURLOPT_HTTPHEADER => array('Expect:')
 | 
						|
    );
 | 
						|
 | 
						|
    $ch = curl_init($uri);
 | 
						|
    curl_setopt_array($ch, $options);
 | 
						|
    $data = curl_exec($ch);
 | 
						|
    $errmsg = curl_error($ch);
 | 
						|
 | 
						|
    if ($errmsg) {
 | 
						|
        common_debug("Twitter bridge - cURL error: $errmsg - trying to load: $uri with user $screen_name.",
 | 
						|
            __FILE__);
 | 
						|
    }
 | 
						|
 | 
						|
    curl_close($ch);
 | 
						|
 | 
						|
    return $data;
 | 
						|
}
 | 
						|
 | 
						|
function twitter_user_info($screen_name, $password)
 | 
						|
{
 | 
						|
 | 
						|
    $uri = "http://twitter.com/users/show/$screen_name.json";
 | 
						|
    $data = get_twitter_data($uri, $screen_name, $password);
 | 
						|
 | 
						|
    if (!$data) {
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    $twit_user = json_decode($data);
 | 
						|
 | 
						|
    if (!$twit_user) {
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    return $twit_user;
 | 
						|
}
 | 
						|
 | 
						|
function update_twitter_user($fuser, $twitter_id, $screen_name)
 | 
						|
{
 | 
						|
 | 
						|
    $original = clone($fuser);
 | 
						|
    $fuser->nickname = $screen_name;
 | 
						|
    $fuser->uri = 'http://twitter.com/' . $screen_name;
 | 
						|
    $result = $fuser->updateKeys($original);
 | 
						|
 | 
						|
    if (!$result) {
 | 
						|
        common_log_db_error($fuser, 'UPDATE', __FILE__);
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
function add_twitter_user($twitter_id, $screen_name)
 | 
						|
{
 | 
						|
 | 
						|
    // Otherwise, create a new Twitter user
 | 
						|
    $fuser = DB_DataObject::factory('foreign_user');
 | 
						|
 | 
						|
    $fuser->nickname = $screen_name;
 | 
						|
    $fuser->uri = 'http://twitter.com/' . $screen_name;
 | 
						|
    $fuser->id = $twitter_id;
 | 
						|
    $fuser->service = TWITTER_SERVICE; // Twitter
 | 
						|
    $fuser->created = common_sql_now();
 | 
						|
    $result = $fuser->insert();
 | 
						|
 | 
						|
    if (!$result) {
 | 
						|
        common_debug("Twitter bridge - failed to add new Twitter user: $twitter_id - $screen_name.");
 | 
						|
        common_log_db_error($fuser, 'INSERT', __FILE__);
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    common_debug("Twitter bridge - Added new Twitter user: $screen_name ($twitter_id).");
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
// Creates or Updates a Twitter user
 | 
						|
function save_twitter_user($twitter_id, $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) {
 | 
						|
 | 
						|
            common_debug('Twitter bridge - Updated nickname (and URI) for Twitter user ' .
 | 
						|
                "$fuser->id to $screen_name, was $fuser->nickname");
 | 
						|
 | 
						|
            return update_twitter_user($fuser, $twitter_id, $screen_name);
 | 
						|
        }
 | 
						|
 | 
						|
    } else {
 | 
						|
        return add_twitter_user($twitter_id, $screen_name);
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
function retreive_twitter_friends($twitter_id, $screen_name, $password)
 | 
						|
{
 | 
						|
 | 
						|
    $uri = "http://twitter.com/statuses/friends/$twitter_id.json?page=";
 | 
						|
    $twitter_user = twitter_user_info($screen_name, $password);
 | 
						|
 | 
						|
    // Calculate how many pages to get...
 | 
						|
    $pages = ceil($twitter_user->friends_count / 100);
 | 
						|
 | 
						|
    if ($pages == 0) {
 | 
						|
        common_debug("Twitter bridge - Twitter user $screen_name has no friends! Lame.");
 | 
						|
    }
 | 
						|
 | 
						|
    $friends = array();
 | 
						|
 | 
						|
    for ($i = 1; $i <= $pages; $i++) {
 | 
						|
 | 
						|
        $data = get_twitter_data($uri . $i, $screen_name, $password);
 | 
						|
 | 
						|
        if (!$data) {
 | 
						|
            return null;
 | 
						|
        }
 | 
						|
 | 
						|
        $more_friends = json_decode($data);
 | 
						|
 | 
						|
        if (!$more_friends) {
 | 
						|
            return null;
 | 
						|
        }
 | 
						|
 | 
						|
         $friends = array_merge($friends, $more_friends);
 | 
						|
    }
 | 
						|
 | 
						|
    return $friends;
 | 
						|
}
 | 
						|
 | 
						|
function save_twitter_friends($user, $twitter_id, $screen_name, $password)
 | 
						|
{
 | 
						|
 | 
						|
    $friends = retreive_twitter_friends($twitter_id, $screen_name, $password);
 | 
						|
 | 
						|
    if (is_null($friends)) {
 | 
						|
        common_debug("Twitter bridge - Couldn't get friends data from Twitter.");
 | 
						|
        return false;
 | 
						|
    }
 | 
						|
 | 
						|
    foreach ($friends as $friend) {
 | 
						|
 | 
						|
        $friend_name = $friend->screen_name;
 | 
						|
        $friend_id = $friend->id;
 | 
						|
 | 
						|
        // Update or create the Foreign_user record
 | 
						|
        if (!save_twitter_user($friend_id, $friend_name)) {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
 | 
						|
        // Check to see if there's a related local user
 | 
						|
        $flink = Foreign_link::getByForeignID($friend_id, 1);
 | 
						|
 | 
						|
        if ($flink) {
 | 
						|
 | 
						|
            // Get associated user and subscribe her
 | 
						|
            $friend_user = User::staticGet('id', $flink->user_id);
 | 
						|
            subs_subscribe_to($user, $friend_user);
 | 
						|
            common_debug("Twitter bridge - subscribed $friend_user->nickname to $user->nickname.");
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
function is_twitter_bound($notice, $flink) {
 | 
						|
 | 
						|
    // Check to see if notice should go to Twitter
 | 
						|
    if (($flink->noticesync & FOREIGN_NOTICE_SEND)) {
 | 
						|
 | 
						|
        // If it's not a Twitter-style reply, or if the user WANTS to send replies.
 | 
						|
        if (!preg_match('/^@[a-zA-Z0-9_]{1,15}\b/u', $notice->content) ||
 | 
						|
            ($flink->noticesync & FOREIGN_NOTICE_SEND_REPLY)) {
 | 
						|
                return true;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    
 | 
						|
    return false;
 | 
						|
}
 | 
						|
 | 
						|
function broadcast_twitter($notice)
 | 
						|
{
 | 
						|
    global $config;
 | 
						|
    $success = true;
 | 
						|
 | 
						|
    $flink = Foreign_link::getByUserID($notice->profile_id, 
 | 
						|
        TWITTER_SERVICE);
 | 
						|
            
 | 
						|
    // XXX: Not sure WHERE to check whether a notice should go to 
 | 
						|
    // Twitter. Should we even put in the queue if it shouldn't? --Zach
 | 
						|
    if (is_twitter_bound($notice, $flink)) {
 | 
						|
 | 
						|
        $fuser = $flink->getForeignUser();
 | 
						|
        $twitter_user = $fuser->nickname;
 | 
						|
        $twitter_password = $flink->credentials;
 | 
						|
        $uri = 'http://www.twitter.com/statuses/update.json';
 | 
						|
 | 
						|
        // XXX: Hack to get around PHP cURL's use of @ being a a meta character
 | 
						|
        $statustxt = preg_replace('/^@/', ' @', $notice->content);
 | 
						|
 | 
						|
        $options = array(
 | 
						|
            CURLOPT_USERPWD        => "$twitter_user:$twitter_password",
 | 
						|
            CURLOPT_POST           => true,
 | 
						|
            CURLOPT_POSTFIELDS     => 
 | 
						|
                array(
 | 
						|
                        'status' => $statustxt,
 | 
						|
                        'source' => $config['integration']['source']
 | 
						|
                     ),
 | 
						|
            CURLOPT_RETURNTRANSFER => true,
 | 
						|
            CURLOPT_FAILONERROR    => true,
 | 
						|
            CURLOPT_HEADER         => false,
 | 
						|
            CURLOPT_FOLLOWLOCATION => true,
 | 
						|
            CURLOPT_USERAGENT      => "Laconica",
 | 
						|
            CURLOPT_CONNECTTIMEOUT => 120,  // XXX: How long should this be?
 | 
						|
            CURLOPT_TIMEOUT        => 120,
 | 
						|
 | 
						|
            # Twitter is strict about accepting invalid "Expect" headers
 | 
						|
            CURLOPT_HTTPHEADER => array('Expect:')
 | 
						|
            );
 | 
						|
 | 
						|
        $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 send notice for $twitter_user.",
 | 
						|
                         __FILE__);
 | 
						|
            $success = false;
 | 
						|
        }
 | 
						|
 | 
						|
        curl_close($ch);
 | 
						|
 | 
						|
        if (!$data) {
 | 
						|
            common_debug("No data returned by Twitter's " .
 | 
						|
                "API trying to send update for $twitter_user",
 | 
						|
                         __FILE__);
 | 
						|
            $success = false;
 | 
						|
        }
 | 
						|
 | 
						|
        // Twitter should return a status
 | 
						|
        $status = json_decode($data);
 | 
						|
 | 
						|
        if (!$status->id) {
 | 
						|
            common_debug("Unexpected data returned by Twitter " .
 | 
						|
                " API trying to send update for $twitter_user",
 | 
						|
                         __FILE__);
 | 
						|
            $success = false;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    
 | 
						|
    return $success;
 | 
						|
}
 | 
						|
 |