plugins/Realtime/RealtimePlugin.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * Superclass for plugins that do "real time" updates of timelines using Ajax
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  Plugin
  23  * @package   StatusNet
  24  * @author    Evan Prodromou <evan@status.net>
  25  * @copyright 2009 StatusNet, Inc.
  26  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  27  * @link      http://status.net/
  28  */
  29
  30 if (!defined('STATUSNET') && !defined('LACONICA')) {
  31     exit(1);
  32 }
  33
  34 /**
  35  * Superclass for plugin to do realtime updates
  36  *
  37  * Based on experience with the Comet and Meteor plugins,
  38  * this superclass extracts out some of the common functionality
  39  *
  40  * @category Plugin
  41  * @package  StatusNet
  42  * @author   Evan Prodromou <evan@status.net>
  43  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  44  * @link     http://status.net/
  45  */
  46 class RealtimePlugin extends Plugin
  47 {
  48     protected $showurl = null;
  49
  50     /**
  51      * When it's time to initialize the plugin, calculate and
  52      * pass the URLs we need.
  53      */
  54     function onInitializePlugin()
  55     {
  56         // FIXME: need to find a better way to pass this pattern in
  57         $this->showurl = common_local_url('shownotice',
  58                                             array('notice' => '0000000000'));
  59         return true;
  60     }
  61
  62     function onCheckSchema()
  63     {
  64         $schema = Schema::get();
  65         $schema->ensureTable('realtime_channel', Realtime_channel::schemaDef());
  66         return true;
  67     }
  68
  69     function onAutoload($cls)
  70     {
  71         $dir = dirname(__FILE__);
  72
  73         switch ($cls)
  74         {
  75         case 'KeepalivechannelAction':
  76         case 'ClosechannelAction':
  77             include_once $dir . '/' . strtolower(mb_substr($cls, 0, -6)) . '.php';
  78             return false;
  79         case 'Realtime_channel':
  80             include_once $dir.'/'.$cls.'.php';
  81             return false;
  82         default:
  83             return true;
  84         }
  85     }
  86
  87     /**
  88      * Hook for RouterInitialized event.
  89      *
  90      * @param Net_URL_Mapper $m path-to-action mapper
  91      * @return boolean hook return
  92      */
  93     function onRouterInitialized($m)
  94     {
  95         $m->connect('main/channel/:channelkey/keepalive',
  96                     array('action' => 'keepalivechannel'),
  97                     array('channelkey' => '[a-z0-9]{32}'));
  98         $m->connect('main/channel/:channelkey/close',
  99                     array('action' => 'closechannel'),
 100                     array('channelkey' => '[a-z0-9]{32}'));
 101         return true;
 102     }
 103
 104     function onEndShowScripts($action)
 105     {
 106         $channel = $this->_getChannel($action);
 107
 108         if (empty($channel)) {
 109             return true;
 110         }
 111
 112         $timeline = $this->_pathToChannel(array($channel->channel_key));
 113
 114         // If there's not a timeline on this page,
 115         // just return true
 116
 117         if (empty($timeline)) {
 118             return true;
 119         }
 120
 121         $base = $action->selfUrl();
 122         if (mb_strstr($base, '?')) {
 123             $url = $base . '&realtime=1';
 124         } else {
 125             $url = $base . '?realtime=1';
 126         }
 127
 128         $scripts = $this->_getScripts();
 129
 130         foreach ($scripts as $script) {
 131             $action->script($script);
 132         }
 133
 134         $user = common_current_user();
 135
 136         if (!empty($user->id)) {
 137             $user_id = $user->id;
 138         } else {
 139             $user_id = 0;
 140         }
 141
 142         if ($action->boolean('realtime')) {
 143             $realtimeUI = ' RealtimeUpdate.initPopupWindow();';
 144         }
 145         else {
 146             $pluginPath = common_path('plugins/Realtime/');
 147             $keepalive = common_local_url('keepalivechannel', array('channelkey' => $channel->channel_key));
 148             $close = common_local_url('closechannel', array('channelkey' => $channel->channel_key));
 149             $realtimeUI = ' RealtimeUpdate.initActions('.json_encode($url).', '.json_encode($timeline).', '.json_encode($pluginPath).', '.json_encode($keepalive).', '.json_encode($close).'); ';
 150         }
 151
 152         $script = ' $(document).ready(function() { '.
 153           $realtimeUI.
 154             $this->_updateInitialize($timeline, $user_id).
 155           '}); ';
 156         $action->inlineScript($script);
 157
 158         return true;
 159     }
 160
 161     function onEndShowStatusNetStyles($action)
 162     {
 163         $action->cssLink(Plugin::staticPath('Realtime', 'realtimeupdate.css'),
 164                          null,
 165                          'screen, projection, tv');
 166         return true;
 167     }
 168
 169     function onHandleQueuedNotice($notice)
 170     {
 171         $paths = array();
 172
 173         // Add to the author's timeline
 174
 175         try {
 176             $profile = $notice->getProfile();
 177         } catch (Exception $e) {
 178             $this->log(LOG_ERR, $e->getMessage());
 179             return true;
 180         }
 181
 182         $user = User::staticGet('id', $notice->profile_id);
 183
 184         if (!empty($user)) {
 185             $paths[] = array('showstream', $user->nickname, null);
 186         }
 187
 188         // Add to the public timeline
 189
 190         if ($notice->is_local == Notice::LOCAL_PUBLIC ||
 191             ($notice->is_local == Notice::REMOTE && !common_config('public', 'localonly'))) {
 192             $paths[] = array('public', null, null);
 193         }
 194
 195         // Add to the tags timeline
 196
 197         $tags = $this->getNoticeTags($notice);
 198
 199         if (!empty($tags)) {
 200             foreach ($tags as $tag) {
 201                 $paths[] = array('tag', $tag, null);
 202             }
 203         }
 204
 205         // Add to inbox timelines
 206         // XXX: do a join
 207
 208         $ni = $notice->whoGets();
 209
 210         foreach (array_keys($ni) as $user_id) {
 211             $user = User::staticGet('id', $user_id);
 212             $paths[] = array('all', $user->nickname, null);
 213         }
 214
 215         // Add to the replies timeline
 216
 217         $reply = new Reply();
 218         $reply->notice_id = $notice->id;
 219
 220         if ($reply->find()) {
 221             while ($reply->fetch()) {
 222                 $user = User::staticGet('id', $reply->profile_id);
 223                 if (!empty($user)) {
 224                     $paths[] = array('replies', $user->nickname, null);
 225                 }
 226             }
 227         }
 228
 229         // Add to the group timeline
 230         // XXX: join
 231
 232         $gi = new Group_inbox();
 233         $gi->notice_id = $notice->id;
 234
 235         if ($gi->find()) {
 236             while ($gi->fetch()) {
 237                 $ug = User_group::staticGet('id', $gi->group_id);
 238                 $paths[] = array('showgroup', $ug->nickname, null);
 239             }
 240         }
 241
 242         if (count($paths) > 0) {
 243
 244             $json = $this->noticeAsJson($notice);
 245
 246             $this->_connect();
 247
 248             // XXX: We should probably fan-out here and do a
 249             // new queue item for each path
 250
 251             foreach ($paths as $path) {
 252
 253                 list($action, $arg1, $arg2) = $path;
 254
 255                 $channels = Realtime_channel::getAllChannels($action, $arg1, $arg2);
 256
 257                 foreach ($channels as $channel) {
 258
 259                     // XXX: We should probably fan-out here and do a
 260                     // new queue item for each user/path combo
 261
 262                     if (is_null($channel->user_id)) {
 263                         $profile = null;
 264                     } else {
 265                         $profile = Profile::staticGet('id', $channel->user_id);
 266                     }
 267                     if ($notice->inScope($profile)) {
 268                         $timeline = $this->_pathToChannel(array($channel->channel_key));
 269                         $this->_publish($timeline, $json);
 270                     }
 271                 }
 272             }
 273
 274             $this->_disconnect();
 275         }
 276
 277         return true;
 278     }
 279
 280     function onStartShowBody($action)
 281     {
 282         $realtime = $action->boolean('realtime');
 283         if (!$realtime) {
 284             return true;
 285         }
 286
 287         $action->elementStart('body',
 288                               (common_current_user()) ? array('id' => $action->trimmed('action'),
 289                                                               'class' => 'user_in realtime-popup')
 290                               : array('id' => $action->trimmed('action'),
 291                                       'class'=> 'realtime-popup'));
 292
 293         // XXX hack to deal with JS that tries to get the
 294         // root url from page output
 295
 296         $action->elementStart('address');
 297         $action->element('a', array('class' => 'url',
 298                                   'href' => common_local_url('public')),
 299                          '');
 300         $action->elementEnd('address');
 301
 302         $action->showContentBlock();
 303         $action->showScripts();
 304         $action->elementEnd('body');
 305         return false; // No default processing
 306     }
 307
 308     function noticeAsJson($notice)
 309     {
 310         // FIXME: this code should be abstracted to a neutral third
 311         // party, like Notice::asJson(). I'm not sure of the ethics
 312         // of refactoring from within a plugin, so I'm just abusing
 313         // the ApiAction method. Don't do this unless you're me!
 314
 315         $act = new ApiAction('/dev/null');
 316
 317         $arr = $act->twitterStatusArray($notice, true);
 318         $arr['url'] = $notice->bestUrl();
 319         $arr['html'] = htmlspecialchars($notice->rendered);
 320         $arr['source'] = htmlspecialchars($arr['source']);
 321         $arr['conversation_url'] = $this->getConversationUrl($notice);
 322
 323         $profile = $notice->getProfile();
 324         $arr['user']['profile_url'] = $profile->profileurl;
 325
 326         // Add needed repeat data
 327
 328         if (!empty($notice->repeat_of)) {
 329             $original = Notice::staticGet('id', $notice->repeat_of);
 330             if (!empty($original)) {
 331                 $arr['retweeted_status']['url'] = $original->bestUrl();
 332                 $arr['retweeted_status']['html'] = htmlspecialchars($original->rendered);
 333                 $arr['retweeted_status']['source'] = htmlspecialchars($original->source);
 334                 $originalProfile = $original->getProfile();
 335                 $arr['retweeted_status']['user']['profile_url'] = $originalProfile->profileurl;
 336                 $arr['retweeted_status']['conversation_url'] = $this->getConversationUrl($original);
 337             }
 338             $original = null;
 339         }
 340
 341         return $arr;
 342     }
 343
 344     function getNoticeTags($notice)
 345     {
 346         $tags = null;
 347
 348         $nt = new Notice_tag();
 349         $nt->notice_id = $notice->id;
 350
 351         if ($nt->find()) {
 352             $tags = array();
 353             while ($nt->fetch()) {
 354                 $tags[] = $nt->tag;
 355             }
 356         }
 357
 358         $nt->free();
 359         $nt = null;
 360
 361         return $tags;
 362     }
 363
 364     function getConversationUrl($notice)
 365     {
 366         $convurl = null;
 367
 368         if ($notice->hasConversation()) {
 369             $conv = Conversation::staticGet(
 370                 'id',
 371                 $notice->conversation
 372             );
 373             $convurl = $conv->uri;
 374
 375             if(empty($convurl)) {
 376                 $msg = sprintf( "Could not find Conversation ID %d to make 'in context'"
 377                     . "link for Notice ID %d.",
 378                     $notice->conversation,
 379                     $notice->id
 380                 );
 381
 382                 common_log(LOG_WARNING, $msg);
 383             } else {
 384                 $convurl .= '#notice-' . $notice->id;
 385             }
 386         }
 387
 388         return $convurl;
 389     }
 390
 391     function _getScripts()
 392     {
 393         if (common_config('site', 'minify')) {
 394             $js = 'realtimeupdate.min.js';
 395         } else {
 396             $js = 'realtimeupdate.js';
 397         }
 398         return array(Plugin::staticPath('Realtime', $js));
 399     }
 400
 401     /**
 402      * Export any i18n messages that need to be loaded at runtime...
 403      *
 404      * @param Action $action
 405      * @param array $messages
 406      *
 407      * @return boolean hook return value
 408      */
 409     function onEndScriptMessages($action, &$messages)
 410     {
 411         // TRANS: Text label for realtime view "play" button, usually replaced by an icon.
 412         $messages['realtime_play'] = _m('BUTTON', 'Play');
 413         // TRANS: Tooltip for realtime view "play" button.
 414         $messages['realtime_play_tooltip'] = _m('TOOLTIP', 'Play');
 415         // TRANS: Text label for realtime view "pause" button
 416         $messages['realtime_pause'] = _m('BUTTON', 'Pause');
 417         // TRANS: Tooltip for realtime view "pause" button
 418         $messages['realtime_pause_tooltip'] = _m('TOOLTIP', 'Pause');
 419         // TRANS: Text label for realtime view "popup" button, usually replaced by an icon.
 420         $messages['realtime_popup'] = _m('BUTTON', 'Pop up');
 421         // TRANS: Tooltip for realtime view "popup" button.
 422         $messages['realtime_popup_tooltip'] = _m('TOOLTIP', 'Pop up in a window');
 423
 424         return true;
 425     }
 426
 427     function _updateInitialize($timeline, $user_id)
 428     {
 429         return "RealtimeUpdate.init($user_id, \"$this->showurl\"); ";
 430     }
 431
 432     function _connect()
 433     {
 434     }
 435
 436     function _publish($timeline, $json)
 437     {
 438     }
 439
 440     function _disconnect()
 441     {
 442     }
 443
 444     function _pathToChannel($path)
 445     {
 446         return '';
 447     }
 448
 449
 450     function _getTimeline($action)
 451     {
 452         $channel = $this->_getChannel($action);
 453         if (empty($channel)) {
 454             return null;
 455         }
 456
 457         return $this->_pathToChannel(array($channel->channel_key));
 458     }
 459
 460     function _getChannel($action)
 461     {
 462         $timeline = null;
 463         $arg1     = null;
 464         $arg2     = null;
 465
 466         $action_name = $action->trimmed('action');
 467
 468         // FIXME: lists
 469         // FIXME: search (!)
 470         // FIXME: profile + tag
 471
 472         switch ($action_name) {
 473          case 'public':
 474             // no arguments
 475             break;
 476          case 'tag':
 477             $tag = $action->trimmed('tag');
 478             if (!empty($tag)) {
 479                 $arg1 = $tag;
 480             } else {
 481                 $this->log(LOG_NOTICE, "Unexpected 'tag' action without tag argument");
 482                 return null;
 483             }
 484             break;
 485          case 'showstream':
 486          case 'all':
 487          case 'replies':
 488          case 'showgroup':
 489             $nickname = common_canonical_nickname($action->trimmed('nickname'));
 490             if (!empty($nickname)) {
 491                 $arg1 = $nickname;
 492             } else {
 493                 $this->log(LOG_NOTICE, "Unexpected $action_name action without nickname argument.");
 494                 return null;
 495             }
 496             break;
 497          default:
 498             return null;
 499         }
 500
 501         $user = common_current_user();
 502
 503         $user_id = (!empty($user)) ? $user->id : null;
 504
 505         $channel = Realtime_channel::getChannel($user_id,
 506                                                 $action_name,
 507                                                 $arg1,
 508                                                 $arg2);
 509
 510         return $channel;
 511     }
 512
 513     function onStartReadWriteTables(&$alwaysRW, &$rwdb)
 514     {
 515         $alwaysRW[] = 'realtime_channel';
 516         return true;
 517     }
 518 }