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                         $this->log(LOG_INFO,
 269                                    sprintf(_("Delivering notice %d to channel (%s, %s, %s) for user '%s'"),
 270                                            $notice->id,
 271                                            $channel->action,
 272                                            $channel->arg1,
 273                                            $channel->arg2,
 274                                            ($profile) ? ($profile->nickname) : "<public>"));
 275                         $timeline = $this->_pathToChannel(array($channel->channel_key));
 276                         $this->_publish($timeline, $json);
 277                     }
 278                 }
 279             }
 280
 281             $this->_disconnect();
 282         }
 283
 284         return true;
 285     }
 286
 287     function onStartShowBody($action)
 288     {
 289         $realtime = $action->boolean('realtime');
 290         if (!$realtime) {
 291             return true;
 292         }
 293
 294         $action->elementStart('body',
 295                               (common_current_user()) ? array('id' => $action->trimmed('action'),
 296                                                               'class' => 'user_in realtime-popup')
 297                               : array('id' => $action->trimmed('action'),
 298                                       'class'=> 'realtime-popup'));
 299
 300         // XXX hack to deal with JS that tries to get the
 301         // root url from page output
 302
 303         $action->elementStart('address');
 304
 305         if (common_config('singleuser', 'enabled')) {
 306             $user = User::singleUser();
 307             $url = common_local_url('showstream', array('nickname' => $user->nickname));
 308         } else {
 309             $url = common_local_url('public');
 310         }
 311
 312         $action->element('a', array('class' => 'url',
 313                                     'href' => $url),
 314                          '');
 315
 316         $action->elementEnd('address');
 317
 318         $action->showContentBlock();
 319         $action->showScripts();
 320         $action->elementEnd('body');
 321         return false; // No default processing
 322     }
 323
 324     function noticeAsJson($notice)
 325     {
 326         // FIXME: this code should be abstracted to a neutral third
 327         // party, like Notice::asJson(). I'm not sure of the ethics
 328         // of refactoring from within a plugin, so I'm just abusing
 329         // the ApiAction method. Don't do this unless you're me!
 330
 331         $act = new ApiAction('/dev/null');
 332
 333         $arr = $act->twitterStatusArray($notice, true);
 334         $arr['url'] = $notice->bestUrl();
 335         $arr['html'] = htmlspecialchars($notice->rendered);
 336         $arr['source'] = htmlspecialchars($arr['source']);
 337         $arr['conversation_url'] = $this->getConversationUrl($notice);
 338
 339         $profile = $notice->getProfile();
 340         $arr['user']['profile_url'] = $profile->profileurl;
 341
 342         // Add needed repeat data
 343
 344         if (!empty($notice->repeat_of)) {
 345             $original = Notice::staticGet('id', $notice->repeat_of);
 346             if (!empty($original)) {
 347                 $arr['retweeted_status']['url'] = $original->bestUrl();
 348                 $arr['retweeted_status']['html'] = htmlspecialchars($original->rendered);
 349                 $arr['retweeted_status']['source'] = htmlspecialchars($original->source);
 350                 $originalProfile = $original->getProfile();
 351                 $arr['retweeted_status']['user']['profile_url'] = $originalProfile->profileurl;
 352                 $arr['retweeted_status']['conversation_url'] = $this->getConversationUrl($original);
 353             }
 354             $original = null;
 355         }
 356
 357         return $arr;
 358     }
 359
 360     function getNoticeTags($notice)
 361     {
 362         $tags = null;
 363
 364         $nt = new Notice_tag();
 365         $nt->notice_id = $notice->id;
 366
 367         if ($nt->find()) {
 368             $tags = array();
 369             while ($nt->fetch()) {
 370                 $tags[] = $nt->tag;
 371             }
 372         }
 373
 374         $nt->free();
 375         $nt = null;
 376
 377         return $tags;
 378     }
 379
 380     function getConversationUrl($notice)
 381     {
 382         $convurl = null;
 383
 384         if ($notice->hasConversation()) {
 385             $conv = Conversation::staticGet(
 386                 'id',
 387                 $notice->conversation
 388             );
 389             $convurl = $conv->uri;
 390
 391             if(empty($convurl)) {
 392                 $msg = sprintf( "Could not find Conversation ID %d to make 'in context'"
 393                     . "link for Notice ID %d.",
 394                     $notice->conversation,
 395                     $notice->id
 396                 );
 397
 398                 common_log(LOG_WARNING, $msg);
 399             } else {
 400                 $convurl .= '#notice-' . $notice->id;
 401             }
 402         }
 403
 404         return $convurl;
 405     }
 406
 407     function _getScripts()
 408     {
 409         if (common_config('site', 'minify')) {
 410             $js = 'realtimeupdate.min.js';
 411         } else {
 412             $js = 'realtimeupdate.js';
 413         }
 414         return array(Plugin::staticPath('Realtime', $js));
 415     }
 416
 417     /**
 418      * Export any i18n messages that need to be loaded at runtime...
 419      *
 420      * @param Action $action
 421      * @param array $messages
 422      *
 423      * @return boolean hook return value
 424      */
 425     function onEndScriptMessages($action, &$messages)
 426     {
 427         // TRANS: Text label for realtime view "play" button, usually replaced by an icon.
 428         $messages['realtime_play'] = _m('BUTTON', 'Play');
 429         // TRANS: Tooltip for realtime view "play" button.
 430         $messages['realtime_play_tooltip'] = _m('TOOLTIP', 'Play');
 431         // TRANS: Text label for realtime view "pause" button
 432         $messages['realtime_pause'] = _m('BUTTON', 'Pause');
 433         // TRANS: Tooltip for realtime view "pause" button
 434         $messages['realtime_pause_tooltip'] = _m('TOOLTIP', 'Pause');
 435         // TRANS: Text label for realtime view "popup" button, usually replaced by an icon.
 436         $messages['realtime_popup'] = _m('BUTTON', 'Pop up');
 437         // TRANS: Tooltip for realtime view "popup" button.
 438         $messages['realtime_popup_tooltip'] = _m('TOOLTIP', 'Pop up in a window');
 439
 440         return true;
 441     }
 442
 443     function _updateInitialize($timeline, $user_id)
 444     {
 445         return "RealtimeUpdate.init($user_id, \"$this->showurl\"); ";
 446     }
 447
 448     function _connect()
 449     {
 450     }
 451
 452     function _publish($timeline, $json)
 453     {
 454     }
 455
 456     function _disconnect()
 457     {
 458     }
 459
 460     function _pathToChannel($path)
 461     {
 462         return '';
 463     }
 464
 465
 466     function _getTimeline($action)
 467     {
 468         $channel = $this->_getChannel($action);
 469         if (empty($channel)) {
 470             return null;
 471         }
 472
 473         return $this->_pathToChannel(array($channel->channel_key));
 474     }
 475
 476     function _getChannel($action)
 477     {
 478         $timeline = null;
 479         $arg1     = null;
 480         $arg2     = null;
 481
 482         $action_name = $action->trimmed('action');
 483
 484         // FIXME: lists
 485         // FIXME: search (!)
 486         // FIXME: profile + tag
 487
 488         switch ($action_name) {
 489          case 'public':
 490             // no arguments
 491             break;
 492          case 'tag':
 493             $tag = $action->trimmed('tag');
 494             if (!empty($tag)) {
 495                 $arg1 = $tag;
 496             } else {
 497                 $this->log(LOG_NOTICE, "Unexpected 'tag' action without tag argument");
 498                 return null;
 499             }
 500             break;
 501          case 'showstream':
 502          case 'all':
 503          case 'replies':
 504          case 'showgroup':
 505             $nickname = common_canonical_nickname($action->trimmed('nickname'));
 506             if (!empty($nickname)) {
 507                 $arg1 = $nickname;
 508             } else {
 509                 $this->log(LOG_NOTICE, "Unexpected $action_name action without nickname argument.");
 510                 return null;
 511             }
 512             break;
 513          default:
 514             return null;
 515         }
 516
 517         $user = common_current_user();
 518
 519         $user_id = (!empty($user)) ? $user->id : null;
 520
 521         $channel = Realtime_channel::getChannel($user_id,
 522                                                 $action_name,
 523                                                 $arg1,
 524                                                 $arg2);
 525
 526         return $channel;
 527     }
 528
 529     function onStartReadWriteTables(&$alwaysRW, &$rwdb)
 530     {
 531         $alwaysRW[] = 'realtime_channel';
 532         return true;
 533     }
 534 }