classes/Profile.php

   1 <?php
   2 /*
   3  * Laconica - a distributed open-source microblogging tool
   4  * Copyright (C) 2008, Control Yourself, 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 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;                             // varchar(140)  multiple_key
  39     public $location;                        // varchar(255)  multiple_key
  40     public $created;                         // datetime()   not_null
  41     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  42
  43     /* Static get */
  44     function staticGet($k,$v=null)
  45     { return Memcached_DataObject::staticGet('Profile',$k,$v); }
  46
  47     /* the code above is auto generated do not remove the tag below */
  48     ###END_AUTOCODE
  49
  50     function getAvatar($width, $height=null)
  51     {
  52         if (is_null($height)) {
  53             $height = $width;
  54         }
  55         return Avatar::pkeyGet(array('profile_id' => $this->id,
  56                                      'width' => $width,
  57                                      'height' => $height));
  58     }
  59
  60     function getOriginalAvatar()
  61     {
  62         $avatar = DB_DataObject::factory('avatar');
  63         $avatar->profile_id = $this->id;
  64         $avatar->original = true;
  65         if ($avatar->find(true)) {
  66             return $avatar;
  67         } else {
  68             return null;
  69         }
  70     }
  71
  72     function setOriginal($filename)
  73     {
  74         $imagefile = new ImageFile($this->id, Avatar::path($filename));
  75
  76         $avatar = new Avatar();
  77         $avatar->profile_id = $this->id;
  78         $avatar->width = $imagefile->width;
  79         $avatar->height = $imagefile->height;
  80         $avatar->mediatype = image_type_to_mime_type($imagefile->type);
  81         $avatar->filename = $filename;
  82         $avatar->original = true;
  83         $avatar->url = Avatar::url($filename);
  84         $avatar->created = DB_DataObject_Cast::dateTime(); # current time
  85
  86         # XXX: start a transaction here
  87
  88         if (!$this->delete_avatars() || !$avatar->insert()) {
  89             @unlink(Avatar::path($filename));
  90             return null;
  91         }
  92
  93         foreach (array(AVATAR_PROFILE_SIZE, AVATAR_STREAM_SIZE, AVATAR_MINI_SIZE) as $size) {
  94             # We don't do a scaled one if original is our scaled size
  95             if (!($avatar->width == $size && $avatar->height == $size)) {
  96
  97                 $scaled_filename = $imagefile->resize($size);
  98
  99                 //$scaled = DB_DataObject::factory('avatar');
 100                 $scaled = new Avatar();
 101                 $scaled->profile_id = $this->id;
 102                 $scaled->width = $size;
 103                 $scaled->height = $size;
 104                 $scaled->original = false;
 105                 $scaled->mediatype = image_type_to_mime_type($imagefile->type);
 106                 $scaled->filename = $scaled_filename;
 107                 $scaled->url = Avatar::url($scaled_filename);
 108                 $scaled->created = DB_DataObject_Cast::dateTime(); # current time
 109
 110                 if (!$scaled->insert()) {
 111                     return null;
 112                 }
 113             }
 114         }
 115
 116         return $avatar;
 117     }
 118
 119     function delete_avatars($original=true)
 120     {
 121         $avatar = new Avatar();
 122         $avatar->profile_id = $this->id;
 123         $avatar->find();
 124         while ($avatar->fetch()) {
 125             if ($avatar->original) {
 126                 if ($original == false) {
 127                     continue;
 128                 }
 129             }
 130             $avatar->delete();
 131         }
 132         return true;
 133     }
 134
 135     function getBestName()
 136     {
 137         return ($this->fullname) ? $this->fullname : $this->nickname;
 138     }
 139
 140     # Get latest notice on or before date; default now
 141     function getCurrentNotice($dt=null)
 142     {
 143         $notice = new Notice();
 144         $notice->profile_id = $this->id;
 145         if ($dt) {
 146             $notice->whereAdd('created < "' . $dt . '"');
 147         }
 148         $notice->orderBy('created DESC, notice.id DESC');
 149         $notice->limit(1);
 150         if ($notice->find(true)) {
 151             return $notice;
 152         }
 153         return null;
 154     }
 155
 156     function getTaggedNotices($tag, $offset=0, $limit=NOTICES_PER_PAGE, $since_id=0, $max_id=0, $since=null)
 157     {
 158         $ids = Notice::stream(array($this, '_streamTaggedDirect'),
 159                               array($tag),
 160                               'profile:notice_ids_tagged:' . $this->id . ':' . $tag,
 161                               $offset, $limit, $since_id, $max_id, $since);
 162         return Notice::getStreamByIds($ids);
 163     }
 164
 165     function getNotices($offset=0, $limit=NOTICES_PER_PAGE, $since_id=0, $max_id=0, $since=null)
 166     {
 167         // XXX: I'm not sure this is going to be any faster. It probably isn't.
 168         $ids = Notice::stream(array($this, '_streamDirect'),
 169                               array(),
 170                               'profile:notice_ids:' . $this->id,
 171                               $offset, $limit, $since_id, $max_id, $since);
 172
 173         return Notice::getStreamByIds($ids);
 174     }
 175
 176     function _streamTaggedDirect($tag, $offset, $limit, $since_id, $max_id, $since)
 177     {
 178         // XXX It would be nice to do this without a join
 179
 180         $notice = new Notice();
 181
 182         $query =
 183           "select id from notice join notice_tag on id=notice_id where tag='".
 184           $notice->escape($tag) .
 185           "' and profile_id=" . $notice->escape($this->id);
 186
 187         if ($since_id != 0) {
 188             $query .= " and id > $since_id";
 189         }
 190
 191         if ($max_id != 0) {
 192             $query .= " and id < $max_id";
 193         }
 194
 195         if (!is_null($since)) {
 196             $query .= " and created > '" . date('Y-m-d H:i:s', $since) . "'";
 197         }
 198
 199         $query .= ' order by id DESC';
 200
 201         if (!is_null($offset)) {
 202             $query .= " limit $offset, $limit";
 203         }
 204
 205         $notice->query($query);
 206
 207         $ids = array();
 208
 209         while ($notice->fetch()) {
 210             $ids[] = $notice->id;
 211         }
 212
 213         return $ids;
 214     }
 215
 216     function _streamDirect($offset, $limit, $since_id, $max_id, $since = null)
 217     {
 218         $notice = new Notice();
 219
 220         $notice->profile_id = $this->id;
 221
 222         $notice->selectAdd();
 223         $notice->selectAdd('id');
 224
 225         if ($since_id != 0) {
 226             $notice->whereAdd('id > ' . $since_id);
 227         }
 228
 229         if ($max_id != 0) {
 230             $notice->whereAdd('id <= ' . $max_id);
 231         }
 232
 233         if (!is_null($since)) {
 234             $notice->whereAdd('created > \'' . date('Y-m-d H:i:s', $since) . '\'');
 235         }
 236
 237         $notice->orderBy('id DESC');
 238
 239         if (!is_null($offset)) {
 240             $notice->limit($offset, $limit);
 241         }
 242
 243         $ids = array();
 244
 245         if ($notice->find()) {
 246             while ($notice->fetch()) {
 247                 $ids[] = $notice->id;
 248             }
 249         }
 250
 251         return $ids;
 252     }
 253
 254     function isMember($group)
 255     {
 256         $mem = new Group_member();
 257
 258         $mem->group_id = $group->id;
 259         $mem->profile_id = $this->id;
 260
 261         if ($mem->find()) {
 262             return true;
 263         } else {
 264             return false;
 265         }
 266     }
 267
 268     function isAdmin($group)
 269     {
 270         $mem = new Group_member();
 271
 272         $mem->group_id = $group->id;
 273         $mem->profile_id = $this->id;
 274         $mem->is_admin = 1;
 275
 276         if ($mem->find()) {
 277             return true;
 278         } else {
 279             return false;
 280         }
 281     }
 282
 283     function avatarUrl($size=AVATAR_PROFILE_SIZE)
 284     {
 285         $avatar = $this->getAvatar($size);
 286         if ($avatar) {
 287             return $avatar->displayUrl();
 288         } else {
 289             return Avatar::defaultImage($size);
 290         }
 291     }
 292 }