szurubooru/src/Models/Entities/UserEntity.php

<?php
class UserEntity extends AbstractEntity
{
	public $name;
	public $passSalt;
	public $passHash;
	public $staffConfirmed;
	public $emailUnconfirmed;
	public $emailConfirmed;
	public $joinDate;
	public $lastLoginDate;
	public $accessRank;
	public $settings;
	public $banned;

	public function getAvatarUrl($size = 32)
	{
		$subject = !empty($this->emailConfirmed)
			? $this->emailConfirmed
			: $this->passSalt . $this->name;
		$hash = md5(strtolower(trim($subject)));
		$url = 'http://www.gravatar.com/avatar/' . $hash . '?s=' . $size . '&d=retro';
		return $url;
	}

	public function getSetting($key)
	{
		$settings = json_decode($this->settings, true);
		return isset($settings[$key])
			? $settings[$key]
			: null;
	}

	public function setSetting($key, $value)
	{
		$settings = json_decode($this->settings, true);
		$settings[$key] = $value;
		$settings = json_encode($settings);
		if (strlen($settings) > 200)
			throw new SimpleException('Too much data');
		$this->settings = $settings;
	}

	public function hasEnabledSafety($safety)
	{
		$all = $this->getSetting(UserModel::SETTING_SAFETY);
		if (!$all)
			return $safety == PostSafety::Safe;
		return $all & PostSafety::toFlag($safety);
	}

	public function enableSafety($safety, $enabled)
	{
		$all = $this->getSetting(UserModel::SETTING_SAFETY);
		if (!$all)
			$all = PostSafety::toFlag(PostSafety::Safe);

		$new = $all;
		if (!$enabled)
		{
			$new &= ~PostSafety::toFlag($safety);
			if (!$new)
				$new = PostSafety::toFlag(PostSafety::Safe);
		}
		else
		{
			$new |= PostSafety::toFlag($safety);
		}

		$this->setSetting(UserModel::SETTING_SAFETY, $new);
	}

	public function hasEnabledHidingDislikedPosts()
	{
		$ret = $this->getSetting(UserModel::SETTING_HIDE_DISLIKED_POSTS);
		if ($ret === null)
			$ret = !\Chibi\Registry::getConfig()->browsing->showDislikedPostsDefault;
		return $ret;
	}

	public function enableHidingDislikedPosts($enabled)
	{
		$this->setSetting(UserModel::SETTING_HIDE_DISLIKED_POSTS, $enabled ? 1 : 0);
	}

	public function hasEnabledPostTagTitles()
	{
		$ret = $this->getSetting(UserModel::SETTING_POST_TAG_TITLES);
		if ($ret === null)
			$ret = \Chibi\Registry::getConfig()->browsing->showPostTagTitlesDefault;
		return $ret;
	}

	public function enablePostTagTitles($enabled)
	{
		$this->setSetting(UserModel::SETTING_POST_TAG_TITLES, $enabled ? 1 : 0);
	}

	public function hasEnabledEndlessScrolling()
	{
		$ret = $this->getSetting(UserModel::SETTING_ENDLESS_SCROLLING);
		if ($ret === null)
			$ret = \Chibi\Registry::getConfig()->browsing->endlessScrollingDefault;
		return $ret;
	}

	public function enableEndlessScrolling($enabled)
	{
		$this->setSetting(UserModel::SETTING_ENDLESS_SCROLLING, $enabled ? 1 : 0);
	}

	public function hasFavorited($post)
	{
		$stmt = new SqlSelectStatement();
		$stmt->setColumn(new SqlAliasOperator(new SqlCountOperator('1'), 'count'));
		$stmt->setTable('favoritee');
		$stmt->setCriterion((new SqlConjunction)
			->add(new SqlEqualsOperator('user_id', new SqlBinding($this->id)))
			->add(new SqlEqualsOperator('post_id', new SqlBinding($post->id))));
		return Database::fetchOne($stmt)['count'] == 1;
	}

	public function getScore($post)
	{
		$stmt = new SqlSelectStatement();
		$stmt->setColumn('score');
		$stmt->setTable('post_score');
		$stmt->setCriterion((new SqlConjunction)
			->add(new SqlEqualsOperator('user_id', new SqlBinding($this->id)))
			->add(new SqlEqualsOperator('post_id', new SqlBinding($post->id))));
		$row = Database::fetchOne($stmt);
		if ($row)
			return intval($row['score']);
		return null;
	}

	public function getFavoriteCount()
	{
		$stmt = new SqlSelectStatement();
		$stmt->setColumn(new SqlAliasOperator(new SqlCountOperator('1'), 'count'));
		$stmt->setTable('favoritee');
		$stmt->setCriterion(new SqlEqualsOperator('user_id', new SqlBinding($this->id)));
		return Database::fetchOne($stmt)['count'];
	}

	public function getCommentCount()
	{
		$stmt = new SqlSelectStatement();
		$stmt->setColumn(new SqlAliasOperator(new SqlCountOperator('1'), 'count'));
		$stmt->setTable('comment');
		$stmt->setCriterion(new SqlEqualsOperator('commenter_id', new SqlBinding($this->id)));
		return Database::fetchOne($stmt)['count'];
	}

	public function getPostCount()
	{
		$stmt = new SqlSelectStatement();
		$stmt->setColumn(new SqlAliasOperator(new SqlCountOperator('1'), 'count'));
		$stmt->setTable('post');
		$stmt->setCriterion(new SqlEqualsOperator('uploader_id', new SqlBinding($this->id)));
		return Database::fetchOne($stmt)['count'];
	}
}