X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=classes%2FAvatar.php;h=55abc81b33b3850891b6c3ca6ca80ceddd3efdfd;hb=7f3611c51c53acb46e17189194f0ed82beee7914;hp=8d6424e8b2d1f53c5ccec912404ccde6ee7b26d9;hpb=71d5990ea3fd7864c3595ea738f35c613d52b62f;p=quix0rs-gnu-social.git diff --git a/classes/Avatar.php b/classes/Avatar.php index 8d6424e8b2..55abc81b33 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 */ @@ -20,30 +20,141 @@ class Avatar extends Memcached_DataObject 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' => 255, 'description' => 'local filename, if local'), + 'url' => array('type' => 'varchar', 'length' => 255, '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) { @@ -82,21 +193,40 @@ class Avatar extends Memcached_DataObject $server = common_config('site', 'server'); } - // XXX: protocol + $ssl = common_config('avatar', 'ssl'); - return 'http://'.$server.$path.$filename; + if (is_null($ssl)) { // null -> guess + if (common_config('site', 'ssl') == 'always' && + !common_config('avatar', 'server')) { + $ssl = true; + } else { + $ssl = false; + } + } + + $protocol = ($ssl) ? 'https' : 'http'; + + return $protocol.'://'.$server.$path.$filename; } 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', @@ -104,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; + } }