7155cf813d
making a twitterqueuehandler.
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's not? --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;
|
|
}
|
|
|