initialize fave, sub, and membership URIs

[quix0rs-gnu-social.git] / plugins / TwitterBridge / scripts / streamtest.php

diff --git a/plugins/TwitterBridge/scripts/streamtest.php b/plugins/TwitterBridge/scripts/streamtest.php
index 96ff33fa528ad9a4b94aa1d66248f010be770e20..aad15fdeab95a02c0a30851c73dc90dcd9ba267a 100644 (file)
--- a/plugins/TwitterBridge/scripts/streamtest.php
+++ b/plugins/TwitterBridge/scripts/streamtest.php
@@ -28,11 +28,16 @@
 define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../..'));
 
 $shortoptions = 'n:';
-$longoptions = array('nick=');
+$longoptions = array('nick=','import','all','apiroot=');
 
 $helptext = <<<ENDOFHELP
 USAGE: streamtest.php -n <username>
 
+  -n --nick=<username> Local user whose Twitter timeline to watch
+     --import          Experimental: run incoming messages through import
+     --all             Experimental: run multiuser; requires nick be the app owner
+     --apiroot=<url>   Provide alternate streaming API root URL
+
 Attempts a User Stream connection to Twitter as the given user, dumping
 data as it comes.
 
@@ -78,44 +83,107 @@ function homeStreamForUser(User $user)
     return new TwitterUserStream($auth);
 }
 
+function siteStreamForOwner(User $user)
+{
+    // The user we auth as must be the owner of the application.
+    $auth = twitterAuthForUser($user);
+
+    if (have_option('apiroot')) {
+        $stream = new TwitterSiteStream($auth, get_option_value('apiroot'));
+    } else {
+        $stream = new TwitterSiteStream($auth);
+    }
+
+    // Pull Twitter user IDs for all users we want to pull data for
+    $userIds = array();
+
+    $flink = new Foreign_link();
+    $flink->service = TWITTER_SERVICE;
+    $flink->find();
+
+    while ($flink->fetch()) {
+        if (($flink->noticesync & FOREIGN_NOTICE_RECV) ==
+            FOREIGN_NOTICE_RECV) {
+            $userIds[] = $flink->foreign_id;
+        }
+    }
+
+    $stream->followUsers($userIds);
+    return $stream;
+}
+
+
 $user = User::staticGet('nickname', $nickname);
-$stream = homeStreamForUser($user);
-$stream->hookEvent('raw', function($data) {
-    common_log(LOG_INFO, json_encode($data));
+global $myuser;
+$myuser = $user;
+
+if (have_option('all')) {
+    $stream = siteStreamForOwner($user);
+} else {
+    $stream = homeStreamForUser($user);
+}
+
+
+$stream->hookEvent('raw', function($data, $context) {
+    common_log(LOG_INFO, json_encode($data) . ' for ' . json_encode($context));
 });
-$stream->hookEvent('friends', function($data) {
-    printf("Friend list: %s\n", implode(', ', $data));
+$stream->hookEvent('friends', function($data, $context) {
+    printf("Friend list: %s\n", implode(', ', $data->friends));
 });
-$stream->hookEvent('favorite', function($data) {
+$stream->hookEvent('favorite', function($data, $context) {
     printf("%s favorited %s's notice: %s\n",
-            $data['source']['screen_name'],
-            $data['target']['screen_name'],
-            $data['target_object']['text']);
+            $data->source->screen_name,
+            $data->target->screen_name,
+            $data->target_object->text);
+});
+$stream->hookEvent('unfavorite', function($data, $context) {
+    printf("%s unfavorited %s's notice: %s\n",
+            $data->source->screen_name,
+            $data->target->screen_name,
+            $data->target_object->text);
 });
-$stream->hookEvent('follow', function($data) {
+$stream->hookEvent('follow', function($data, $context) {
     printf("%s friended %s\n",
-            $data['source']['screen_name'],
-            $data['target']['screen_name']);
+            $data->source->screen_name,
+            $data->target->screen_name);
 });
-$stream->hookEvent('delete', function($data) {
+$stream->hookEvent('unfollow', function($data, $context) {
+    printf("%s unfriended %s\n",
+            $data->source->screen_name,
+            $data->target->screen_name);
+});
+$stream->hookEvent('delete', function($data, $context) {
     printf("Deleted status notification: %s\n",
-            $data['status']['id']);
+            $data->status->id);
 });
-$stream->hookEvent('scrub_geo', function($data) {
+$stream->hookEvent('scrub_geo', function($data, $context) {
     printf("Req to scrub geo data for user id %s up to status ID %s\n",
-            $data['user_id'],
-            $data['up_to_status_id']);
+            $data->user_id,
+            $data->up_to_status_id);
 });
-$stream->hookEvent('status', function($data) {
+$stream->hookEvent('status', function($data, $context) {
     printf("Received status update from %s: %s\n",
-            $data['user']['screen_name'],
-            $data['text']);
+            $data->user->screen_name,
+            $data->text);
+
+    if (have_option('import')) {
+        $importer = new TwitterImport();
+        printf("\timporting...");
+        $notice = $importer->importStatus($data);
+        if ($notice) {
+            global $myuser;
+            Inbox::insertNotice($myuser->id, $notice->id);
+            printf(" %s\n", $notice->id);
+        } else {
+            printf(" FAIL\n");
+        }
+    }
 });
 $stream->hookEvent('direct_message', function($data) {
     printf("Direct message from %s to %s: %s\n",
-            $data['sender']['screen_name'],
-            $data['recipient']['screen_name'],
-            $data['text']);
+            $data->sender->screen_name,
+            $data->recipient->screen_name,
+            $data->text);
 });
 
 class TwitterManager extends IoManager