classes/Notice.php

   1 <?php
   2 /*
   3  * Laconica - a distributed open-source microblogging tool
   4  * Copyright (C) 2008, Controlez-Vous, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 if (!defined('LACONICA')) { exit(1); }
  21
  22 /**
  23  * Table Definition for notice
  24  */
  25 require_once INSTALLDIR.'/classes/Memcached_DataObject.php';
  26
  27 /* We keep the first three 20-notice pages, plus one for pagination check,
  28  * in the memcached cache. */
  29
  30 define('NOTICE_CACHE_WINDOW', 61);
  31
  32 class Notice extends Memcached_DataObject
  33 {
  34     ###START_AUTOCODE
  35     /* the code below is auto generated do not remove the above tag */
  36
  37     public $__table = 'notice';                          // table name
  38     public $id;                              // int(4)  primary_key not_null
  39     public $profile_id;                      // int(4)   not_null
  40     public $uri;                             // varchar(255)  unique_key
  41     public $content;                         // varchar(140)
  42     public $rendered;                        // text()
  43     public $url;                             // varchar(255)
  44     public $created;                         // datetime()   not_null
  45     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  46     public $reply_to;                        // int(4)
  47     public $is_local;                        // tinyint(1)
  48     public $source;                          // varchar(32)
  49
  50     /* Static get */
  51     function staticGet($k,$v=NULL) { return Memcached_DataObject::staticGet('Notice',$k,$v); }
  52
  53     /* the code above is auto generated do not remove the tag below */
  54     ###END_AUTOCODE
  55
  56         function getProfile() {
  57                 return Profile::staticGet($this->profile_id);
  58         }
  59
  60         function delete() {
  61                 $this->blowCaches();
  62                 $this->blowFavesCache();
  63                 parent::delete();
  64         }
  65
  66         function saveTags() {
  67                 /* extract all #hastags */
  68                 $count = preg_match_all('/(?:^|\s)#([A-Za-z0-9_\-\.]{1,64})/', strtolower($this->content), $match);
  69                 if (!$count) {
  70                         return true;
  71                 }
  72
  73                 /* elide characters we don't want in the tag */
  74                 $match[1] = str_replace(array('-', '_', '.'), '', $match[1]);
  75
  76                 /* Add them to the database */
  77                 foreach(array_unique($match[1]) as $hashtag) {
  78                         $tag = DB_DataObject::factory('Notice_tag');
  79                         $tag->notice_id = $this->id;
  80                         $tag->tag = $hashtag;
  81                         $tag->created = $this->created;
  82                         $id = $tag->insert();
  83                         if (!$id) {
  84                                 $last_error = PEAR::getStaticProperty('DB_DataObject','lastError');
  85                                 common_log(LOG_ERR, 'DB error inserting hashtag: ' . $last_error->message);
  86                                 common_server_error(sprintf(_('DB error inserting hashtag: %s'), $last_error->message));
  87                                 return;
  88                         }
  89                 }
  90                 return true;
  91         }
  92
  93         static function saveNew($profile_id, $content, $source=NULL, $is_local=1, $reply_to=NULL, $uri=NULL) {
  94
  95                 $notice = new Notice();
  96                 $notice->profile_id = $profile_id;
  97                 $notice->is_local = $is_local;
  98                 $notice->reply_to = $reply_to;
  99                 $notice->created = common_sql_now();
 100                 $notice->content = $content;
 101                 $notice->rendered = common_render_content($notice->content, $notice);
 102                 $notice->source = $source;
 103                 $notice->uri = $uri;
 104
 105                 $id = $notice->insert();
 106
 107                 if (!$id) {
 108                         common_log_db_error($notice, 'INSERT', __FILE__);
 109                         return _('Problem saving notice.');
 110                 }
 111
 112                 # Update the URI after the notice is in the database
 113                 if (!$uri) {
 114                         $orig = clone($notice);
 115                         $notice->uri = common_notice_uri($notice);
 116
 117                         if (!$notice->update($orig)) {
 118                                 common_log_db_error($notice, 'UPDATE', __FILE__);
 119                                 return _('Problem saving notice.');
 120                         }
 121                 }
 122
 123                 # XXX: do we need to change this for remote users?
 124
 125                 common_save_replies($notice);
 126                 $notice->saveTags();
 127
 128                 # Clear the cache for subscribed users, so they'll update at next request
 129                 # XXX: someone clever could prepend instead of clearing the cache
 130
 131                 if (common_config('memcached', 'enabled')) {
 132                         $notice->blowCaches();
 133                 }
 134
 135                 return $notice;
 136         }
 137
 138         function blowCaches() {
 139                 $this->blowSubsCache();
 140                 $this->blowNoticeCache();
 141                 $this->blowRepliesCache();
 142                 $this->blowPublicCache();
 143         }
 144
 145         function blowSubsCache() {
 146                 $cache = common_memcache();
 147                 if ($cache) {
 148                         $user = new User();
 149
 150                         $user->query('SELECT id ' .
 151                                                  'FROM user JOIN subscription ON user.id = subscription.subscriber ' .
 152                                                  'WHERE subscription.subscribed = ' . $this->profile_id);
 153
 154                         while ($user->fetch()) {
 155                                 $cache->delete(common_cache_key('user:notices_with_friends:' . $user->id));
 156                         }
 157
 158                         $user->free();
 159                         unset($user);
 160                 }
 161         }
 162
 163         function blowNoticeCache() {
 164                 if ($this->is_local) {
 165                         $cache = common_memcache();
 166                         if ($cache) {
 167                                 $cache->delete(common_cache_key('user:notices:'.$this->profile_id));
 168                         }
 169                 }
 170         }
 171
 172         function blowRepliesCache() {
 173                 $cache = common_memcache();
 174                 if ($cache) {
 175                         $reply = new Reply();
 176                         $reply->notice_id = $this->id;
 177                         if ($reply->find()) {
 178                                 while ($reply->fetch()) {
 179                                         $cache->delete(common_cache_key('user:replies:'.$reply->profile_id));
 180                                 }
 181                         }
 182                         $reply->free();
 183                         unset($reply);
 184                 }
 185         }
 186
 187         function blowPublicCache() {
 188                 if ($this->is_local) {
 189                         $cache = common_memcache();
 190                         if ($cache) {
 191                                 $cache->delete(common_cache_key('public'));
 192                         }
 193                 }
 194         }
 195
 196         function blowFavesCache() {
 197                 $cache = common_memcache();
 198                 if ($cache) {
 199                         $fave = new Fave();
 200                         $fave->notice_id = $this->id;
 201                         if ($fave->find()) {
 202                                 while ($fave->fetch()) {
 203                                         $cache->delete(common_cache_key('user:faves:'.$fave->user_id));
 204                                 }
 205                         }
 206                         $fave->free();
 207                         unset($fave);
 208                 }
 209         }
 210
 211         static function getStream($qry, $cachekey, $offset=0, $limit=20) {
 212
 213                 if (common_config('memcached', 'enabled')) {
 214                         return Notice::getCachedStream($qry, $cachekey, $offset, $limit);
 215                 } else {
 216                         return Notice::getStreamDirect($qry, $offset, $limit);
 217                 }
 218
 219         }
 220
 221         static function getStreamDirect($qry, $offset, $limit) {
 222
 223                 $qry .= 'ORDER BY notice.created DESC, notice.id DESC ';
 224
 225                 if(common_config('db','type')=='pgsql') {
 226                         $qry .= 'LIMIT ' . $limit . ' OFFSET ' . $offset;
 227                 } else {
 228                         $qry .= 'LIMIT ' . $offset . ', ' . $limit;
 229                 }
 230
 231                 $notice = new Notice();
 232
 233                 $notice->query($qry);
 234
 235                 return $notice;
 236         }
 237
 238         static function getCachedStream($qry, $cachekey, $offset, $limit) {
 239
 240                 # If outside our cache window, just go to the DB
 241
 242                 if ($offset + $limit > NOTICE_CACHE_WINDOW) {
 243                         common_debug('request is too deep, just getting from DB');
 244                         return Notice::getStreamDirect($qry, $offset, $limit);
 245                 }
 246
 247                 common_debug('CONNECTING TO CACHE');
 248                 # Get the cache; if we can't, just go to the DB
 249
 250                 $cache = common_memcache();
 251
 252
 253                 if (!$cache) {
 254                         common_debug('Failed connecting to cache; just going to db');
 255                         return Notice::getStreamDirect($qry, $offset, $limit);
 256                 }
 257
 258                 common_debug('getting from cache');
 259
 260                 # Get the notices out of the cache
 261
 262                 $notices = $cache->get(common_cache_key($cachekey));
 263
 264                 # On a cache hit, return a DB-object-like wrapper
 265
 266                 if ($notices !== FALSE) {
 267                         common_debug('Notices hit: ' . print_r($notices, TRUE));
 268                         common_debug('Got this many notices: ' . count($notices));
 269                         $wrapper = new NoticeWrapper(array_slice($notices, $offset, $limit));
 270                         return $wrapper;
 271                 }
 272
 273                 common_debug('Getting full window from DB.');
 274                 # Otherwise, get the full cache window out of the DB
 275
 276                 $notice = Notice::getStreamDirect($qry, 0, NOTICE_CACHE_WINDOW);
 277
 278                 common_debug('Got notice: ' . print_r($notice, TRUE));
 279
 280                 # If there are no hits, just return the value
 281
 282                 if (!$notice) {
 283                         return $notice;
 284                 }
 285
 286                 common_debug('Copying notices to an array');
 287
 288                 # Pack results into an array
 289
 290                 $notices = array();
 291
 292                 while ($notice->fetch()) {
 293                         common_debug('Got notice: ' . print_r($notice, TRUE));
 294                         $notices[] = clone($notice);
 295                 }
 296
 297                 common_debug('Array size is: '  . count($notices));
 298
 299                 # Store the array in the cache for next time
 300
 301                 $result = $cache->set(common_cache_key($cachekey), $notices);
 302
 303                 common_debug('memcached result is ' . $result);
 304
 305                 # return a wrapper of the array for use now
 306
 307                 $wrapper = new NoticeWrapper(array_slice($notices, $offset, $limit));
 308
 309                 common_debug('Got wrapper: ' . print_r($wrapper, TRUE));
 310
 311                 return $wrapper;
 312         }
 313
 314         function publicStream($offset=0, $limit=20) {
 315
 316                 $qry = 'SELECT * FROM notice ';
 317
 318                 if (common_config('public', 'localonly')) {
 319                         $qry .= ' WHERE is_local = 1 ';
 320                 }
 321
 322                 return Notice::getStream($qry,
 323                                                                  'public',
 324                                                                  $offset, $limit);
 325         }
 326 }