X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=classes%2FAvatar.php;h=3b3317b3ef5ed9f58e066a6e95f43aac9392f696;hb=95b61a5e1212b5017d425f19a6a5ce08cd629ffc;hp=dbe2cd813847eb585f94999b595e2f495ab0d03b;hpb=d5cbfe8071d56438cfa168dc3db56a959317eae0;p=quix0rs-gnu-social.git diff --git a/classes/Avatar.php b/classes/Avatar.php index dbe2cd8138..3b3317b3ef 100644 --- a/classes/Avatar.php +++ b/classes/Avatar.php @@ -4,7 +4,7 @@ */ require_once INSTALLDIR.'/classes/Memcached_DataObject.php'; -class Avatar extends Memcached_DataObject +class Avatar extends Managed_DataObject { ###START_AUTOCODE /* the code below is auto generated do not remove the above tag */ @@ -15,35 +15,146 @@ class Avatar extends Memcached_DataObject public $width; // int(4) primary_key not_null public $height; // int(4) primary_key not_null public $mediatype; // varchar(32) not_null - public $filename; // varchar(255) - public $url; // varchar(255) unique_key + public $filename; // varchar(191) not 255 because utf8mb4 takes more space + public $url; // varchar(191) unique_key not 255 because utf8mb4 takes more space public $created; // datetime() not_null public $modified; // timestamp() not_null default_CURRENT_TIMESTAMP - /* Static get */ - function staticGet($k,$v=null) - { return Memcached_DataObject::staticGet('Avatar',$k,$v); } - /* the code above is auto generated do not remove the tag below */ ###END_AUTOCODE + + public static function schemaDef() + { + return array( + 'fields' => array( + 'profile_id' => array('type' => 'int', 'not null' => true, 'description' => 'foreign key to profile table'), + 'original' => array('type' => 'int', 'size' => 'tiny', 'default' => 0, 'description' => 'uploaded by user or generated?'), + 'width' => array('type' => 'int', 'not null' => true, 'description' => 'image width'), + 'height' => array('type' => 'int', 'not null' => true, 'description' => 'image height'), + 'mediatype' => array('type' => 'varchar', 'length' => 32, 'not null' => true, 'description' => 'file type'), + 'filename' => array('type' => 'varchar', 'length' => 191, 'description' => 'local filename, if local'), + 'url' => array('type' => 'varchar', 'length' => 191, 'description' => 'avatar location'), + 'created' => array('type' => 'datetime', 'not null' => true, 'description' => 'date this record was created'), + 'modified' => array('type' => 'timestamp', 'not null' => true, 'description' => 'date this record was modified'), + ), + 'primary key' => array('profile_id', 'width', 'height'), + 'unique keys' => array( + 'avatar_url_key' => array('url'), + ), + 'foreign keys' => array( + 'avatar_profile_id_fkey' => array('profile', array('profile_id' => 'id')), + ), + 'indexes' => array( + 'avatar_profile_id_idx' => array('profile_id'), + ), + ); + } - # We clean up the file, too - - function delete() + // We clean up the file, too + function delete($useWhere=false) { $filename = $this->filename; - if (parent::delete()) { + if (file_exists(Avatar::path($filename))) { @unlink(Avatar::path($filename)); } + + return parent::delete($useWhere); + } + + /* + * Deletes all avatars (but may spare the original) from a profile. + * + * @param Profile $target The profile we're deleting avatars of. + * @param boolean $original Whether original should be removed or not. + */ + public static function deleteFromProfile(Profile $target, $original=true) { + try { + $avatars = self::getProfileAvatars($target); + foreach ($avatars as $avatar) { + if ($avatar->original && !$original) { + continue; + } + $avatar->delete(); + } + } catch (NoAvatarException $e) { + // There are no avatars to delete, a sort of success. + } + + return true; + } + + static protected $_avatars = array(); + + /* + * Get an avatar by profile. Currently can't call newSize with $height + */ + public static function byProfile(Profile $target, $width=null, $height=null) + { + $width = intval($width); + $height = !is_null($height) ? intval($height) : null; + if (is_null($height)) { + $height = $width; + } + + $size = "{$width}x{$height}"; + if (!isset(self::$_avatars[$target->id])) { + self::$_avatars[$target->id] = array(); + } elseif (isset(self::$_avatars[$target->id][$size])){ + return self::$_avatars[$target->id][$size]; + } + + $avatar = null; + if (Event::handle('StartProfileGetAvatar', array($target, $width, &$avatar))) { + $avatar = self::pkeyGet( + array( + 'profile_id' => $target->id, + 'width' => $width, + 'height' => $height, + ) + ); + Event::handle('EndProfileGetAvatar', array($target, $width, &$avatar)); + } + + if (is_null($avatar)) { + // Obviously we can't find an avatar, so let's resize the original! + $avatar = Avatar::newSize($target, $width); + } elseif (!($avatar instanceof Avatar)) { + throw new NoAvatarException($target, $avatar); + } + + self::$_avatars[$target->id]["{$avatar->width}x{$avatar->height}"] = $avatar; + return $avatar; } - function pkeyGet($kv) + public static function getUploaded(Profile $target) { - return Memcached_DataObject::pkeyGet('Avatar', $kv); + $avatar = new Avatar(); + $avatar->profile_id = $target->id; + $avatar->original = true; + if (!$avatar->find(true)) { + throw new NoAvatarException($target, $avatar); + } + if (!file_exists(Avatar::path($avatar->filename))) { + // The delete call may be odd for, say, unmounted filesystems + // that cause a file to currently not exist, but actually it does... + $avatar->delete(); + throw new NoAvatarException($target, $avatar); + } + return $avatar; } - // where should the avatar go for this user? + public static function getProfileAvatars(Profile $target) { + $avatar = new Avatar(); + $avatar->profile_id = $target->id; + if (!$avatar->find()) { + throw new NoAvatarException($target, $avatar); + } + return $avatar->fetchAll(); + } + /** + * Where should the avatar go for this user? + */ static function filename($id, $extension, $size=null, $extra=null) { if ($size) { @@ -101,13 +212,21 @@ class Avatar extends Memcached_DataObject function displayUrl() { $server = common_config('avatar', 'server'); - if ($server) { + if ($server && !empty($this->filename)) { return Avatar::url($this->filename); } else { return $this->url; } } + static function urlByProfile(Profile $target, $width=null, $height=null) { + try { + return self::byProfile($target, $width, $height)->displayUrl(); + } catch (Exception $e) { + return self::defaultImage($width); + } + } + static function defaultImage($size) { static $sizenames = array(AVATAR_PROFILE_SIZE => 'profile', @@ -115,4 +234,33 @@ class Avatar extends Memcached_DataObject AVATAR_MINI_SIZE => 'mini'); return Theme::path('default-avatar-'.$sizenames[$size].'.png'); } + + static function newSize(Profile $target, $width) { + $width = intval($width); + if ($width < 1 || $width > common_config('avatar', 'maxsize')) { + // TRANS: An error message when avatar size is unreasonable + throw new Exception(_m('Avatar size too large')); + } + + $original = Avatar::getUploaded($target); + + $imagefile = new ImageFile($target->id, Avatar::path($original->filename)); + $filename = $imagefile->resize($width); + + $scaled = clone($original); + $scaled->original = false; + $scaled->width = $width; + $scaled->height = $width; + $scaled->url = Avatar::url($filename); + $scaled->filename = $filename; + $scaled->created = common_sql_now(); + + if (!$scaled->insert()) { + // TRANS: An error message when unable to insert avatar data into the db + throw new Exception(_m('Could not insert new avatar data to database')); + } + + // Return the new avatar object + return $scaled; + } }