fakestream.php can now take --all option to pull the latest messages from multiple locally-authed accounts when generating simulated sitestreams info

This commit is contained in:
Brion Vibber 2010-11-02 16:43:01 -07:00
parent 445b306b54
commit a2f0f68d75

View File

@ -48,6 +48,8 @@ if (have_option('n')) {
$nickname = get_option_value('n');
} else if (have_option('nick')) {
$nickname = get_option_value('nickname');
} else if (have_option('all')) {
$nickname = null;
} else {
show_help($helptext);
exit(0);
@ -82,16 +84,41 @@ function twitterAuthForUser(User $user)
*/
function dumpMessage($flink, $data)
{
$msg->for_user = $flink->foreign_id;
$msg->message = $data;
$msg = prepMessage($flink, $data);
print json_encode($msg) . "\r\n";
}
if (have_option('all')) {
throw new Exception('--all not yet implemented');
function prepMessage($flink, $data)
{
$msg->for_user = $flink->foreign_id;
$msg->message = $data;
return $msg;
}
if (have_option('all')) {
$users = array();
$flink = new Foreign_link();
$flink->service = TWITTER_SERVICE;
$flink->find();
while ($flink->fetch()) {
if (($flink->noticesync & FOREIGN_NOTICE_RECV) ==
FOREIGN_NOTICE_RECV) {
$users[] = $flink->user_id;
}
}
} else {
$user = User::staticGet('nickname', $nickname);
$users = array($user->id);
}
$output = array();
foreach ($users as $id) {
$user = User::staticGet('id', $id);
if (!$user) {
throw new Exception("No user for id $id");
}
$auth = twitterAuthForUser($user);
$flink = Foreign_link::getByUserID($user->id,
TWITTER_SERVICE);
@ -101,6 +128,20 @@ dumpMessage($flink, $friends);
$timeline = $auth->statusesHomeTimeline();
foreach ($timeline as $status) {
dumpMessage($flink, $status);
$output[] = prepMessage($flink, $status);
}
}
usort($output, function($a, $b) {
if ($a->message->id < $b->message->id) {
return -1;
} else if ($a->message->id == $b->message->id) {
return 0;
} else {
return 1;
}
});
foreach ($output as $msg) {
print json_encode($msg) . "\r\n";
}