classes/Profile.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2008, 2009, StatusNet, 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('STATUSNET') && !defined('LACONICA')) { exit(1); }
  21
  22 /**
  23  * Table Definition for profile
  24  */
  25 require_once INSTALLDIR.'/classes/Memcached_DataObject.php';
  26
  27 class Profile extends Memcached_DataObject
  28 {
  29     ###START_AUTOCODE
  30     /* the code below is auto generated do not remove the above tag */
  31
  32     public $__table = 'profile';                         // table name
  33     public $id;                              // int(4)  primary_key not_null
  34     public $nickname;                        // varchar(64)  multiple_key not_null
  35     public $fullname;                        // varchar(255)  multiple_key
  36     public $profileurl;                      // varchar(255)
  37     public $homepage;                        // varchar(255)  multiple_key
  38     public $bio;                             // text()  multiple_key
  39     public $location;                        // varchar(255)  multiple_key
  40     public $lat;                             // decimal(10,7)
  41     public $lon;                             // decimal(10,7)
  42     public $location_id;                     // int(4)
  43     public $location_ns;                     // int(4)
  44     public $created;                         // datetime()   not_null
  45     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  46
  47     /* Static get */
  48     function staticGet($k,$v=NULL) {
  49         return Memcached_DataObject::staticGet('Profile',$k,$v);
  50     }
  51
  52     /* the code above is auto generated do not remove the tag below */
  53     ###END_AUTOCODE
  54
  55     function getUser()
  56     {
  57         return User::staticGet('id', $this->id);
  58     }
  59
  60     function getAvatar($width, $height=null)
  61     {
  62         if (is_null($height)) {
  63             $height = $width;
  64         }
  65         return Avatar::pkeyGet(array('profile_id' => $this->id,
  66                                      'width' => $width,
  67                                      'height' => $height));
  68     }
  69
  70     function getOriginalAvatar()
  71     {
  72         $avatar = DB_DataObject::factory('avatar');
  73         $avatar->profile_id = $this->id;
  74         $avatar->original = true;
  75         if ($avatar->find(true)) {
  76             return $avatar;
  77         } else {
  78             return null;
  79         }
  80     }
  81
  82     function setOriginal($filename)
  83     {
  84         $imagefile = new ImageFile($this->id, Avatar::path($filename));
  85
  86         $avatar = new Avatar();
  87         $avatar->profile_id = $this->id;
  88         $avatar->width = $imagefile->width;
  89         $avatar->height = $imagefile->height;
  90         $avatar->mediatype = image_type_to_mime_type($imagefile->type);
  91         $avatar->filename = $filename;
  92         $avatar->original = true;
  93         $avatar->url = Avatar::url($filename);
  94         $avatar->created = DB_DataObject_Cast::dateTime(); # current time
  95
  96         # XXX: start a transaction here
  97
  98         if (!$this->delete_avatars() || !$avatar->insert()) {
  99             @unlink(Avatar::path($filename));
 100             return null;
 101         }
 102
 103         foreach (array(AVATAR_PROFILE_SIZE, AVATAR_STREAM_SIZE, AVATAR_MINI_SIZE) as $size) {
 104             # We don't do a scaled one if original is our scaled size
 105             if (!($avatar->width == $size && $avatar->height == $size)) {
 106
 107                 $scaled_filename = $imagefile->resize($size);
 108
 109                 //$scaled = DB_DataObject::factory('avatar');
 110                 $scaled = new Avatar();
 111                 $scaled->profile_id = $this->id;
 112                 $scaled->width = $size;
 113                 $scaled->height = $size;
 114                 $scaled->original = false;
 115                 $scaled->mediatype = image_type_to_mime_type($imagefile->type);
 116                 $scaled->filename = $scaled_filename;
 117                 $scaled->url = Avatar::url($scaled_filename);
 118                 $scaled->created = DB_DataObject_Cast::dateTime(); # current time
 119
 120                 if (!$scaled->insert()) {
 121                     return null;
 122                 }
 123             }
 124         }
 125
 126         return $avatar;
 127     }
 128
 129     function delete_avatars($original=true)
 130     {
 131         $avatar = new Avatar();
 132         $avatar->profile_id = $this->id;
 133         $avatar->find();
 134         while ($avatar->fetch()) {
 135             if ($avatar->original) {
 136                 if ($original == false) {
 137                     continue;
 138                 }
 139             }
 140             $avatar->delete();
 141         }
 142         return true;
 143     }
 144
 145     function getBestName()
 146     {
 147         return ($this->fullname) ? $this->fullname : $this->nickname;
 148     }
 149
 150     # Get latest notice on or before date; default now
 151     function getCurrentNotice($dt=null)
 152     {
 153         $notice = new Notice();
 154         $notice->profile_id = $this->id;
 155         if ($dt) {
 156             $notice->whereAdd('created < "' . $dt . '"');
 157         }
 158         $notice->orderBy('created DESC, notice.id DESC');
 159         $notice->limit(1);
 160         if ($notice->find(true)) {
 161             return $notice;
 162         }
 163         return null;
 164     }
 165
 166     function getTaggedNotices($tag, $offset=0, $limit=NOTICES_PER_PAGE, $since_id=0, $max_id=0, $since=null)
 167     {
 168         $ids = Notice::stream(array($this, '_streamTaggedDirect'),
 169                               array($tag),
 170                               'profile:notice_ids_tagged:' . $this->id . ':' . $tag,
 171                               $offset, $limit, $since_id, $max_id, $since);
 172         return Notice::getStreamByIds($ids);
 173     }
 174
 175     function getNotices($offset=0, $limit=NOTICES_PER_PAGE, $since_id=0, $max_id=0, $since=null)
 176     {
 177         // XXX: I'm not sure this is going to be any faster. It probably isn't.
 178         $ids = Notice::stream(array($this, '_streamDirect'),
 179                               array(),
 180                               'profile:notice_ids:' . $this->id,
 181                               $offset, $limit, $since_id, $max_id, $since);
 182
 183         return Notice::getStreamByIds($ids);
 184     }
 185
 186     function _streamTaggedDirect($tag, $offset, $limit, $since_id, $max_id, $since)
 187     {
 188         // XXX It would be nice to do this without a join
 189
 190         $notice = new Notice();
 191
 192         $query =
 193           "select id from notice join notice_tag on id=notice_id where tag='".
 194           $notice->escape($tag) .
 195           "' and profile_id=" . $notice->escape($this->id);
 196
 197         if ($since_id != 0) {
 198             $query .= " and id > $since_id";
 199         }
 200
 201         if ($max_id != 0) {
 202             $query .= " and id < $max_id";
 203         }
 204
 205         if (!is_null($since)) {
 206             $query .= " and created > '" . date('Y-m-d H:i:s', $since) . "'";
 207         }
 208
 209         $query .= ' order by id DESC';
 210
 211         if (!is_null($offset)) {
 212             $query .= " LIMIT $limit OFFSET $offset";
 213         }
 214
 215         $notice->query($query);
 216
 217         $ids = array();
 218
 219         while ($notice->fetch()) {
 220             $ids[] = $notice->id;
 221         }
 222
 223         return $ids;
 224     }
 225
 226     function _streamDirect($offset, $limit, $since_id, $max_id, $since = null)
 227     {
 228         $notice = new Notice();
 229
 230         $notice->profile_id = $this->id;
 231
 232         $notice->selectAdd();
 233         $notice->selectAdd('id');
 234
 235         if ($since_id != 0) {
 236             $notice->whereAdd('id > ' . $since_id);
 237         }
 238
 239         if ($max_id != 0) {
 240             $notice->whereAdd('id <= ' . $max_id);
 241         }
 242
 243         if (!is_null($since)) {
 244             $notice->whereAdd('created > \'' . date('Y-m-d H:i:s', $since) . '\'');
 245         }
 246
 247         $notice->orderBy('id DESC');
 248
 249         if (!is_null($offset)) {
 250             $notice->limit($offset, $limit);
 251         }
 252
 253         $ids = array();
 254
 255         if ($notice->find()) {
 256             while ($notice->fetch()) {
 257                 $ids[] = $notice->id;
 258             }
 259         }
 260
 261         return $ids;
 262     }
 263
 264     function isMember($group)
 265     {
 266         $mem = new Group_member();
 267
 268         $mem->group_id = $group->id;
 269         $mem->profile_id = $this->id;
 270
 271         if ($mem->find()) {
 272             return true;
 273         } else {
 274             return false;
 275         }
 276     }
 277
 278     function isAdmin($group)
 279     {
 280         $mem = new Group_member();
 281
 282         $mem->group_id = $group->id;
 283         $mem->profile_id = $this->id;
 284         $mem->is_admin = 1;
 285
 286         if ($mem->find()) {
 287             return true;
 288         } else {
 289             return false;
 290         }
 291     }
 292
 293     function avatarUrl($size=AVATAR_PROFILE_SIZE)
 294     {
 295         $avatar = $this->getAvatar($size);
 296         if ($avatar) {
 297             return $avatar->displayUrl();
 298         } else {
 299             return Avatar::defaultImage($size);
 300         }
 301     }
 302
 303     function getSubscriptions($offset=0, $limit=null)
 304     {
 305         $qry =
 306           'SELECT profile.* ' .
 307           'FROM profile JOIN subscription ' .
 308           'ON profile.id = subscription.subscribed ' .
 309           'WHERE subscription.subscriber = %d ' .
 310           'AND subscription.subscribed != subscription.subscriber ' .
 311           'ORDER BY subscription.created DESC ';
 312
 313         if (common_config('db','type') == 'pgsql') {
 314             $qry .= ' LIMIT ' . $limit . ' OFFSET ' . $offset;
 315         } else {
 316             $qry .= ' LIMIT ' . $offset . ', ' . $limit;
 317         }
 318
 319         $profile = new Profile();
 320
 321         $profile->query(sprintf($qry, $this->id));
 322
 323         return $profile;
 324     }
 325
 326     function getSubscribers($offset=0, $limit=null)
 327     {
 328         $qry =
 329           'SELECT profile.* ' .
 330           'FROM profile JOIN subscription ' .
 331           'ON profile.id = subscription.subscriber ' .
 332           'WHERE subscription.subscribed = %d ' .
 333           'AND subscription.subscribed != subscription.subscriber ' .
 334           'ORDER BY subscription.created DESC ';
 335
 336         if ($offset) {
 337             if (common_config('db','type') == 'pgsql') {
 338                 $qry .= ' LIMIT ' . $limit . ' OFFSET ' . $offset;
 339             } else {
 340                 $qry .= ' LIMIT ' . $offset . ', ' . $limit;
 341             }
 342         }
 343
 344         $profile = new Profile();
 345
 346         $cnt = $profile->query(sprintf($qry, $this->id));
 347
 348         return $profile;
 349     }
 350
 351     function subscriptionCount()
 352     {
 353         $c = common_memcache();
 354
 355         if (!empty($c)) {
 356             $cnt = $c->get(common_cache_key('profile:subscription_count:'.$this->id));
 357             if (is_integer($cnt)) {
 358                 return (int) $cnt;
 359             }
 360         }
 361
 362         $sub = new Subscription();
 363         $sub->subscriber = $this->id;
 364
 365         $cnt = (int) $sub->count('distinct subscribed');
 366
 367         $cnt = ($cnt > 0) ? $cnt - 1 : $cnt;
 368
 369         if (!empty($c)) {
 370             $c->set(common_cache_key('profile:subscription_count:'.$this->id), $cnt);
 371         }
 372
 373         return $cnt;
 374     }
 375
 376     function subscriberCount()
 377     {
 378         $c = common_memcache();
 379         if (!empty($c)) {
 380             $cnt = $c->get(common_cache_key('profile:subscriber_count:'.$this->id));
 381             if (is_integer($cnt)) {
 382                 return (int) $cnt;
 383             }
 384         }
 385
 386         $sub = new Subscription();
 387         $sub->subscribed = $this->id;
 388
 389         $cnt = (int) $sub->count('distinct subscriber');
 390
 391         $cnt = ($cnt > 0) ? $cnt - 1 : $cnt;
 392
 393         if (!empty($c)) {
 394             $c->set(common_cache_key('profile:subscriber_count:'.$this->id), $cnt);
 395         }
 396
 397         return $cnt;
 398     }
 399
 400     function faveCount()
 401     {
 402         $c = common_memcache();
 403         if (!empty($c)) {
 404             $cnt = $c->get(common_cache_key('profile:fave_count:'.$this->id));
 405             if (is_integer($cnt)) {
 406                 return (int) $cnt;
 407             }
 408         }
 409
 410         $faves = new Fave();
 411         $faves->user_id = $this->id;
 412         $cnt = (int) $faves->count('distinct notice_id');
 413
 414         if (!empty($c)) {
 415             $c->set(common_cache_key('profile:fave_count:'.$this->id), $cnt);
 416         }
 417
 418         return $cnt;
 419     }
 420
 421     function noticeCount()
 422     {
 423         $c = common_memcache();
 424
 425         if (!empty($c)) {
 426             $cnt = $c->get(common_cache_key('profile:notice_count:'.$this->id));
 427             if (is_integer($cnt)) {
 428                 return (int) $cnt;
 429             }
 430         }
 431
 432         $notices = new Notice();
 433         $notices->profile_id = $this->id;
 434         $cnt = (int) $notices->count('distinct id');
 435
 436         if (!empty($c)) {
 437             $c->set(common_cache_key('profile:notice_count:'.$this->id), $cnt);
 438         }
 439
 440         return $cnt;
 441     }
 442
 443     function blowSubscriberCount()
 444     {
 445         $c = common_memcache();
 446         if (!empty($c)) {
 447             $c->delete(common_cache_key('profile:subscriber_count:'.$this->id));
 448         }
 449     }
 450
 451     function blowSubscriptionCount()
 452     {
 453         $c = common_memcache();
 454         if (!empty($c)) {
 455             $c->delete(common_cache_key('profile:subscription_count:'.$this->id));
 456         }
 457     }
 458
 459     function blowFaveCount()
 460     {
 461         $c = common_memcache();
 462         if (!empty($c)) {
 463             $c->delete(common_cache_key('profile:fave_count:'.$this->id));
 464         }
 465     }
 466
 467     function blowNoticeCount()
 468     {
 469         $c = common_memcache();
 470         if (!empty($c)) {
 471             $c->delete(common_cache_key('profile:notice_count:'.$this->id));
 472         }
 473     }
 474
 475     static function maxBio()
 476     {
 477         $biolimit = common_config('profile', 'biolimit');
 478         // null => use global limit (distinct from 0!)
 479         if (is_null($biolimit)) {
 480             $biolimit = common_config('site', 'textlimit');
 481         }
 482         return $biolimit;
 483     }
 484
 485     static function bioTooLong($bio)
 486     {
 487         $biolimit = self::maxBio();
 488         return ($biolimit > 0 && !empty($bio) && (mb_strlen($bio) > $biolimit));
 489     }
 490
 491     function delete()
 492     {
 493         $this->_deleteNotices();
 494         $this->_deleteSubscriptions();
 495         $this->_deleteMessages();
 496         $this->_deleteTags();
 497         $this->_deleteBlocks();
 498
 499         $related = array('Avatar',
 500                          'Reply',
 501                          'Group_member',
 502                          );
 503
 504         foreach ($related as $cls) {
 505             $inst = new $cls();
 506             $inst->profile_id = $this->id;
 507             $inst->delete();
 508         }
 509
 510         parent::delete();
 511     }
 512
 513     function _deleteNotices()
 514     {
 515         $notice = new Notice();
 516         $notice->profile_id = $this->id;
 517
 518         if ($notice->find()) {
 519             while ($notice->fetch()) {
 520                 $other = clone($notice);
 521                 $other->delete();
 522             }
 523         }
 524     }
 525
 526     function _deleteSubscriptions()
 527     {
 528         $sub = new Subscription();
 529         $sub->subscriber = $this->id;
 530         $sub->delete();
 531
 532         $subd = new Subscription();
 533         $subd->subscribed = $this->id;
 534         $subd->delete();
 535     }
 536
 537     function _deleteMessages()
 538     {
 539         $msg = new Message();
 540         $msg->from_profile = $this->id;
 541         $msg->delete();
 542
 543         $msg = new Message();
 544         $msg->to_profile = $this->id;
 545         $msg->delete();
 546     }
 547
 548     function _deleteTags()
 549     {
 550         $tag = new Profile_tag();
 551         $tag->tagged = $this->id;
 552         $tag->delete();
 553     }
 554
 555     function _deleteBlocks()
 556     {
 557         $block = new Profile_block();
 558         $block->blocked = $this->id;
 559         $block->delete();
 560
 561         $block = new Group_block();
 562         $block->blocked = $this->id;
 563         $block->delete();
 564     }
 565
 566     // XXX: identical to Notice::getLocation.
 567
 568     function getLocation()
 569     {
 570         $location = null;
 571
 572         if (!empty($this->location_id) && !empty($this->location_ns)) {
 573             $location = Location::fromId($this->location_id, $this->location_ns);
 574         }
 575
 576         if (is_null($location)) { // no ID, or Location::fromId() failed
 577             if (!empty($this->lat) && !empty($this->lon)) {
 578                 $location = Location::fromLatLon($this->lat, $this->lon);
 579             }
 580         }
 581
 582         if (is_null($location)) { // still haven't found it!
 583             if (!empty($this->location)) {
 584                 $location = Location::fromName($this->location);
 585             }
 586         }
 587
 588         return $location;
 589     }
 590 }