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 ($offset>0 && !is_null($limit)){
 314             if (common_config('db','type') == 'pgsql') {
 315                 $qry .= ' LIMIT ' . $limit . ' OFFSET ' . $offset;
 316             } else {
 317                 $qry .= ' LIMIT ' . $offset . ', ' . $limit;
 318             }
 319         }
 320
 321         $profile = new Profile();
 322
 323         $profile->query(sprintf($qry, $this->id));
 324
 325         return $profile;
 326     }
 327
 328     function getSubscribers($offset=0, $limit=null)
 329     {
 330         $qry =
 331           'SELECT profile.* ' .
 332           'FROM profile JOIN subscription ' .
 333           'ON profile.id = subscription.subscriber ' .
 334           'WHERE subscription.subscribed = %d ' .
 335           'AND subscription.subscribed != subscription.subscriber ' .
 336           'ORDER BY subscription.created DESC ';
 337
 338         if ($offset>0 && !is_null($limit)){
 339             if ($offset) {
 340                 if (common_config('db','type') == 'pgsql') {
 341                     $qry .= ' LIMIT ' . $limit . ' OFFSET ' . $offset;
 342                 } else {
 343                     $qry .= ' LIMIT ' . $offset . ', ' . $limit;
 344                 }
 345             }
 346         }
 347
 348         $profile = new Profile();
 349
 350         $cnt = $profile->query(sprintf($qry, $this->id));
 351
 352         return $profile;
 353     }
 354
 355     function subscriptionCount()
 356     {
 357         $c = common_memcache();
 358
 359         if (!empty($c)) {
 360             $cnt = $c->get(common_cache_key('profile:subscription_count:'.$this->id));
 361             if (is_integer($cnt)) {
 362                 return (int) $cnt;
 363             }
 364         }
 365
 366         $sub = new Subscription();
 367         $sub->subscriber = $this->id;
 368
 369         $cnt = (int) $sub->count('distinct subscribed');
 370
 371         $cnt = ($cnt > 0) ? $cnt - 1 : $cnt;
 372
 373         if (!empty($c)) {
 374             $c->set(common_cache_key('profile:subscription_count:'.$this->id), $cnt);
 375         }
 376
 377         return $cnt;
 378     }
 379
 380     function subscriberCount()
 381     {
 382         $c = common_memcache();
 383         if (!empty($c)) {
 384             $cnt = $c->get(common_cache_key('profile:subscriber_count:'.$this->id));
 385             if (is_integer($cnt)) {
 386                 return (int) $cnt;
 387             }
 388         }
 389
 390         $sub = new Subscription();
 391         $sub->subscribed = $this->id;
 392
 393         $cnt = (int) $sub->count('distinct subscriber');
 394
 395         $cnt = ($cnt > 0) ? $cnt - 1 : $cnt;
 396
 397         if (!empty($c)) {
 398             $c->set(common_cache_key('profile:subscriber_count:'.$this->id), $cnt);
 399         }
 400
 401         return $cnt;
 402     }
 403
 404     function faveCount()
 405     {
 406         $c = common_memcache();
 407         if (!empty($c)) {
 408             $cnt = $c->get(common_cache_key('profile:fave_count:'.$this->id));
 409             if (is_integer($cnt)) {
 410                 return (int) $cnt;
 411             }
 412         }
 413
 414         $faves = new Fave();
 415         $faves->user_id = $this->id;
 416         $cnt = (int) $faves->count('distinct notice_id');
 417
 418         if (!empty($c)) {
 419             $c->set(common_cache_key('profile:fave_count:'.$this->id), $cnt);
 420         }
 421
 422         return $cnt;
 423     }
 424
 425     function noticeCount()
 426     {
 427         $c = common_memcache();
 428
 429         if (!empty($c)) {
 430             $cnt = $c->get(common_cache_key('profile:notice_count:'.$this->id));
 431             if (is_integer($cnt)) {
 432                 return (int) $cnt;
 433             }
 434         }
 435
 436         $notices = new Notice();
 437         $notices->profile_id = $this->id;
 438         $cnt = (int) $notices->count('distinct id');
 439
 440         if (!empty($c)) {
 441             $c->set(common_cache_key('profile:notice_count:'.$this->id), $cnt);
 442         }
 443
 444         return $cnt;
 445     }
 446
 447     function blowSubscriberCount()
 448     {
 449         $c = common_memcache();
 450         if (!empty($c)) {
 451             $c->delete(common_cache_key('profile:subscriber_count:'.$this->id));
 452         }
 453     }
 454
 455     function blowSubscriptionCount()
 456     {
 457         $c = common_memcache();
 458         if (!empty($c)) {
 459             $c->delete(common_cache_key('profile:subscription_count:'.$this->id));
 460         }
 461     }
 462
 463     function blowFaveCount()
 464     {
 465         $c = common_memcache();
 466         if (!empty($c)) {
 467             $c->delete(common_cache_key('profile:fave_count:'.$this->id));
 468         }
 469     }
 470
 471     function blowNoticeCount()
 472     {
 473         $c = common_memcache();
 474         if (!empty($c)) {
 475             $c->delete(common_cache_key('profile:notice_count:'.$this->id));
 476         }
 477     }
 478
 479     static function maxBio()
 480     {
 481         $biolimit = common_config('profile', 'biolimit');
 482         // null => use global limit (distinct from 0!)
 483         if (is_null($biolimit)) {
 484             $biolimit = common_config('site', 'textlimit');
 485         }
 486         return $biolimit;
 487     }
 488
 489     static function bioTooLong($bio)
 490     {
 491         $biolimit = self::maxBio();
 492         return ($biolimit > 0 && !empty($bio) && (mb_strlen($bio) > $biolimit));
 493     }
 494
 495     function delete()
 496     {
 497         $this->_deleteNotices();
 498         $this->_deleteSubscriptions();
 499         $this->_deleteMessages();
 500         $this->_deleteTags();
 501         $this->_deleteBlocks();
 502
 503         $related = array('Avatar',
 504                          'Reply',
 505                          'Group_member',
 506                          );
 507
 508         foreach ($related as $cls) {
 509             $inst = new $cls();
 510             $inst->profile_id = $this->id;
 511             $inst->delete();
 512         }
 513
 514         parent::delete();
 515     }
 516
 517     function _deleteNotices()
 518     {
 519         $notice = new Notice();
 520         $notice->profile_id = $this->id;
 521
 522         if ($notice->find()) {
 523             while ($notice->fetch()) {
 524                 $other = clone($notice);
 525                 $other->delete();
 526             }
 527         }
 528     }
 529
 530     function _deleteSubscriptions()
 531     {
 532         $sub = new Subscription();
 533         $sub->subscriber = $this->id;
 534         $sub->delete();
 535
 536         $subd = new Subscription();
 537         $subd->subscribed = $this->id;
 538         $subd->delete();
 539     }
 540
 541     function _deleteMessages()
 542     {
 543         $msg = new Message();
 544         $msg->from_profile = $this->id;
 545         $msg->delete();
 546
 547         $msg = new Message();
 548         $msg->to_profile = $this->id;
 549         $msg->delete();
 550     }
 551
 552     function _deleteTags()
 553     {
 554         $tag = new Profile_tag();
 555         $tag->tagged = $this->id;
 556         $tag->delete();
 557     }
 558
 559     function _deleteBlocks()
 560     {
 561         $block = new Profile_block();
 562         $block->blocked = $this->id;
 563         $block->delete();
 564
 565         $block = new Group_block();
 566         $block->blocked = $this->id;
 567         $block->delete();
 568     }
 569
 570     // XXX: identical to Notice::getLocation.
 571
 572     function getLocation()
 573     {
 574         $location = null;
 575
 576         if (!empty($this->location_id) && !empty($this->location_ns)) {
 577             $location = Location::fromId($this->location_id, $this->location_ns);
 578         }
 579
 580         if (is_null($location)) { // no ID, or Location::fromId() failed
 581             if (!empty($this->lat) && !empty($this->lon)) {
 582                 $location = Location::fromLatLon($this->lat, $this->lon);
 583             }
 584         }
 585
 586         if (is_null($location)) { // still haven't found it!
 587             if (!empty($this->location)) {
 588                 $location = Location::fromName($this->location);
 589             }
 590         }
 591
 592         return $location;
 593     }
 594 }