]> git.mxchange.org Git - quix0rs-gnu-social.git/blob - plugins/TwitterBridge/scripts/streamtest.php
plugins onAutoload now only overloads if necessary (extlibs etc.)
[quix0rs-gnu-social.git] / plugins / TwitterBridge / scripts / streamtest.php
1 <?php
2 /**
3  * StatusNet, the distributed open-source microblogging tool
4  *
5  * PHP version 5
6  *
7  * LICENCE: This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU Affero General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU Affero General Public License for more details.
16  *
17  * You should have received a copy of the GNU Affero General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  *
20  * @category  Plugin
21  * @package   StatusNet
22  * @author    Brion Vibber <brion@status.net>
23  * @copyright 2010 StatusNet, Inc.
24  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
25  * @link      http://status.net/
26  */
27
28 define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../..'));
29
30 $shortoptions = 'n:';
31 $longoptions = array('nick=','import','all','apiroot=');
32
33 $helptext = <<<ENDOFHELP
34 USAGE: streamtest.php -n <username>
35
36   -n --nick=<username> Local user whose Twitter timeline to watch
37      --import          Experimental: run incoming messages through import
38      --all             Experimental: run multiuser; requires nick be the app owner
39      --apiroot=<url>   Provide alternate streaming API root URL
40
41 Attempts a User Stream connection to Twitter as the given user, dumping
42 data as it comes.
43
44 ENDOFHELP;
45
46 require_once INSTALLDIR.'/scripts/commandline.inc';
47 require_once dirname(dirname(__FILE__)) . '/lib/jsonstreamreader.php';
48 require_once dirname(dirname(__FILE__)) . '/lib/twitterstreamreader.php';
49
50 if (have_option('n')) {
51     $nickname = get_option_value('n');
52 } else if (have_option('nick')) {
53     $nickname = get_option_value('nickname');
54 } else {
55     show_help($helptext);
56     exit(0);
57 }
58
59 /**
60  *
61  * @param User $user 
62  * @return TwitterOAuthClient
63  */
64 function twitterAuthForUser(User $user)
65 {
66     $flink = Foreign_link::getByUserID($user->id,
67                                        TWITTER_SERVICE);
68     if (!$flink) {
69         throw new ServerException("No Twitter config for this user.");
70     }
71
72     $token = TwitterOAuthClient::unpackToken($flink->credentials);
73     if (!$token) {
74         throw new ServerException("No Twitter OAuth credentials for this user.");
75     }
76
77     return new TwitterOAuthClient($token->key, $token->secret);
78 }
79
80 function homeStreamForUser(User $user)
81 {
82     $auth = twitterAuthForUser($user);
83     return new TwitterUserStream($auth);
84 }
85
86 function siteStreamForOwner(User $user)
87 {
88     // The user we auth as must be the owner of the application.
89     $auth = twitterAuthForUser($user);
90
91     if (have_option('apiroot')) {
92         $stream = new TwitterSiteStream($auth, get_option_value('apiroot'));
93     } else {
94         $stream = new TwitterSiteStream($auth);
95     }
96
97     // Pull Twitter user IDs for all users we want to pull data for
98     $userIds = array();
99
100     $flink = new Foreign_link();
101     $flink->service = TWITTER_SERVICE;
102     $flink->find();
103
104     while ($flink->fetch()) {
105         if (($flink->noticesync & FOREIGN_NOTICE_RECV) ==
106             FOREIGN_NOTICE_RECV) {
107             $userIds[] = $flink->foreign_id;
108         }
109     }
110
111     $stream->followUsers($userIds);
112     return $stream;
113 }
114
115
116 $user = User::getKV('nickname', $nickname);
117 global $myuser;
118 $myuser = $user;
119
120 if (have_option('all')) {
121     $stream = siteStreamForOwner($user);
122 } else {
123     $stream = homeStreamForUser($user);
124 }
125
126
127 $stream->hookEvent('raw', function($data, $context) {
128     common_log(LOG_INFO, json_encode($data) . ' for ' . json_encode($context));
129 });
130 $stream->hookEvent('friends', function($data, $context) {
131     printf("Friend list: %s\n", implode(', ', $data->friends));
132 });
133 $stream->hookEvent('favorite', function($data, $context) {
134     printf("%s favorited %s's notice: %s\n",
135             $data->source->screen_name,
136             $data->target->screen_name,
137             $data->target_object->text);
138 });
139 $stream->hookEvent('unfavorite', function($data, $context) {
140     printf("%s unfavorited %s's notice: %s\n",
141             $data->source->screen_name,
142             $data->target->screen_name,
143             $data->target_object->text);
144 });
145 $stream->hookEvent('follow', function($data, $context) {
146     printf("%s friended %s\n",
147             $data->source->screen_name,
148             $data->target->screen_name);
149 });
150 $stream->hookEvent('unfollow', function($data, $context) {
151     printf("%s unfriended %s\n",
152             $data->source->screen_name,
153             $data->target->screen_name);
154 });
155 $stream->hookEvent('delete', function($data, $context) {
156     printf("Deleted status notification: %s\n",
157             $data->status->id);
158 });
159 $stream->hookEvent('scrub_geo', function($data, $context) {
160     printf("Req to scrub geo data for user id %s up to status ID %s\n",
161             $data->user_id,
162             $data->up_to_status_id);
163 });
164 $stream->hookEvent('status', function($data, $context) {
165     printf("Received status update from %s: %s\n",
166             $data->user->screen_name,
167             $data->text);
168
169     if (have_option('import')) {
170         $importer = new TwitterImport();
171         printf("\timporting...");
172         $notice = $importer->importStatus($data);
173         if ($notice) {
174             global $myuser;
175             Inbox::insertNotice($myuser->id, $notice->id);
176             printf(" %s\n", $notice->id);
177         } else {
178             printf(" FAIL\n");
179         }
180     }
181 });
182 $stream->hookEvent('direct_message', function($data) {
183     printf("Direct message from %s to %s: %s\n",
184             $data->sender->screen_name,
185             $data->recipient->screen_name,
186             $data->text);
187 });
188
189 class TwitterManager extends IoManager
190 {
191     function __construct(TwitterStreamReader $stream)
192     {
193         $this->stream = $stream;
194     }
195
196     function getSockets()
197     {
198         return $this->stream->getSockets();
199     }
200
201     function handleInput($data)
202     {
203         $this->stream->handleInput($data);
204         return true;
205     }
206
207     function start()
208     {
209         $this->stream->connect();
210         return true;
211     }
212
213     function finish()
214     {
215         $this->stream->close();
216         return true;
217     }
218
219     public static function get()
220     {
221         throw new Exception('not a singleton');
222     }
223 }
224
225 class TwitterStreamMaster extends IoMaster
226 {
227     function __construct($id, $ioManager)
228     {
229         parent::__construct($id);
230         $this->ioManager = $ioManager;
231     }
232
233     /**
234      * Initialize IoManagers which are appropriate to this instance.
235      */
236     function initManagers()
237     {
238         $this->instantiate($this->ioManager);
239     }
240 }
241
242 $master = new TwitterStreamMaster('TwitterStream', new TwitterManager($stream));
243 $master->init();
244 $master->service();