123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251 |
- <?php
- define('INSTALLDIR', dirname(__DIR__, 3));
- define('PUBLICDIR', INSTALLDIR . DIRECTORY_SEPARATOR . 'public');
- $shortoptions = 'n:';
- $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.
- ENDOFHELP;
- require_once INSTALLDIR . '/scripts/commandline.inc';
- require_once dirname(dirname(__FILE__)) . '/lib/jsonstreamreader.php';
- require_once dirname(dirname(__FILE__)) . '/lib/twitterstreamreader.php';
- if (have_option('n')) {
- $nickname = get_option_value('n');
- } elseif (have_option('nick')) {
- $nickname = get_option_value('nickname');
- } else {
- show_help($helptext);
- exit(0);
- }
- function twitterAuthForUser(User $user)
- {
- $flink = Foreign_link::getByUserID($user->id, TWITTER_SERVICE);
- $token = TwitterOAuthClient::unpackToken($flink->credentials);
- if (!$token) {
- throw new ServerException("No Twitter OAuth credentials for this user.");
- }
- return new TwitterOAuthClient($token->key, $token->secret);
- }
- function homeStreamForUser(User $user)
- {
- $auth = twitterAuthForUser($user);
- return new TwitterUserStream($auth);
- }
- function siteStreamForOwner(User $user)
- {
-
- $auth = twitterAuthForUser($user);
- if (have_option('apiroot')) {
- $stream = new TwitterSiteStream($auth, get_option_value('apiroot'));
- } else {
- $stream = new TwitterSiteStream($auth);
- }
-
- $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::getKV('nickname', $nickname);
- 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, $context) {
- printf("Friend list: %s\n", implode(', ', $data->friends));
- });
- $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
- );
- });
- $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, $context) {
- printf(
- '%s friended %s' . "\n",
- $data->source->screen_name,
- $data->target->screen_name
- );
- });
- $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
- );
- });
- $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
- );
- });
- $stream->hookEvent('status', function ($data, $context) {
- printf(
- 'Received status update from %s: %s' . "\n",
- $data->user->screen_name,
- $data->text
- );
- if (have_option('import')) {
- $importer = new TwitterImport();
- printf("\timporting...");
- $notice = $importer->importStatus($data);
- if (!$notice instanceof Notice) {
- 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
- );
- });
- class TwitterManager extends IoManager
- {
- public function __construct(TwitterStreamReader $stream)
- {
- $this->stream = $stream;
- }
- public function getSockets()
- {
- return $this->stream->getSockets();
- }
- public function handleInput($data)
- {
- $this->stream->handleInput($data);
- return true;
- }
- public function start()
- {
- $this->stream->connect();
- return true;
- }
- public function finish()
- {
- $this->stream->close();
- return true;
- }
- public static function get()
- {
- throw new Exception('not a singleton');
- }
- }
- class TwitterStreamMaster extends IoMaster
- {
- public function __construct($id, $ioManager)
- {
- parent::__construct($id);
- $this->ioManager = $ioManager;
- }
-
- public function initManagers()
- {
- $this->instantiate($this->ioManager);
- }
- }
- $master = new TwitterStreamMaster('TwitterStream', new TwitterManager($stream));
- $master->init();
- $master->service();
|