PSR-2. I'm not a huge fan, but ugly consistency beats no consistency...
This commit is contained in:
Shish
@ -19,241 +19,269 @@
|
||||
* </ul>
|
||||
*/
|
||||
|
||||
class RatingSetEvent extends Event {
|
||||
/** @var \Image */
|
||||
public $image;
|
||||
/** @var string */
|
||||
public $rating;
|
||||
class RatingSetEvent extends Event
|
||||
{
|
||||
/** @var \Image */
|
||||
public $image;
|
||||
/** @var string */
|
||||
public $rating;
|
||||
|
||||
public function __construct(Image $image, string $rating) {
|
||||
assert(in_array($rating, array("s", "q", "e", "u")));
|
||||
public function __construct(Image $image, string $rating)
|
||||
{
|
||||
assert(in_array($rating, ["s", "q", "e", "u"]));
|
||||
|
||||
$this->image = $image;
|
||||
$this->rating = $rating;
|
||||
}
|
||||
$this->image = $image;
|
||||
$this->rating = $rating;
|
||||
}
|
||||
}
|
||||
|
||||
class Ratings extends Extension {
|
||||
protected $db_support = ['mysql']; // ?
|
||||
class Ratings extends Extension
|
||||
{
|
||||
protected $db_support = ['mysql']; // ?
|
||||
|
||||
public function get_priority(): int {return 50;}
|
||||
public function get_priority(): int
|
||||
{
|
||||
return 50;
|
||||
}
|
||||
|
||||
public function onInitExt(InitExtEvent $event) {
|
||||
global $config;
|
||||
|
||||
if($config->get_int("ext_ratings2_version") < 2) {
|
||||
$this->install();
|
||||
}
|
||||
public function onInitExt(InitExtEvent $event)
|
||||
{
|
||||
global $config;
|
||||
|
||||
if ($config->get_int("ext_ratings2_version") < 2) {
|
||||
$this->install();
|
||||
}
|
||||
|
||||
$config->set_default_string("ext_rating_anon_privs", 'squ');
|
||||
$config->set_default_string("ext_rating_user_privs", 'sqeu');
|
||||
$config->set_default_string("ext_rating_admin_privs", 'sqeu');
|
||||
}
|
||||
|
||||
public function onSetupBuilding(SetupBuildingEvent $event) {
|
||||
$privs = array();
|
||||
$privs['Safe Only'] = 's';
|
||||
$privs['Safe and Unknown'] = 'su';
|
||||
$privs['Safe and Questionable'] = 'sq';
|
||||
$privs['Safe, Questionable, Unknown'] = 'squ';
|
||||
$privs['All'] = 'sqeu';
|
||||
$config->set_default_string("ext_rating_anon_privs", 'squ');
|
||||
$config->set_default_string("ext_rating_user_privs", 'sqeu');
|
||||
$config->set_default_string("ext_rating_admin_privs", 'sqeu');
|
||||
}
|
||||
|
||||
public function onSetupBuilding(SetupBuildingEvent $event)
|
||||
{
|
||||
$privs = [];
|
||||
$privs['Safe Only'] = 's';
|
||||
$privs['Safe and Unknown'] = 'su';
|
||||
$privs['Safe and Questionable'] = 'sq';
|
||||
$privs['Safe, Questionable, Unknown'] = 'squ';
|
||||
$privs['All'] = 'sqeu';
|
||||
|
||||
$sb = new SetupBlock("Image Ratings");
|
||||
$sb->add_choice_option("ext_rating_anon_privs", $privs, "Anonymous: ");
|
||||
$sb->add_choice_option("ext_rating_user_privs", $privs, "<br>Users: ");
|
||||
$sb->add_choice_option("ext_rating_admin_privs", $privs, "<br>Admins: ");
|
||||
$event->panel->add_block($sb);
|
||||
}
|
||||
|
||||
public function onPostListBuilding(PostListBuildingEvent $event) {
|
||||
global $user;
|
||||
if($user->is_admin() && !empty($event->search_terms)) {
|
||||
$this->theme->display_bulk_rater(Tag::implode($event->search_terms));
|
||||
}
|
||||
}
|
||||
$sb = new SetupBlock("Image Ratings");
|
||||
$sb->add_choice_option("ext_rating_anon_privs", $privs, "Anonymous: ");
|
||||
$sb->add_choice_option("ext_rating_user_privs", $privs, "<br>Users: ");
|
||||
$sb->add_choice_option("ext_rating_admin_privs", $privs, "<br>Admins: ");
|
||||
$event->panel->add_block($sb);
|
||||
}
|
||||
|
||||
public function onPostListBuilding(PostListBuildingEvent $event)
|
||||
{
|
||||
global $user;
|
||||
if ($user->is_admin() && !empty($event->search_terms)) {
|
||||
$this->theme->display_bulk_rater(Tag::implode($event->search_terms));
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public function onDisplayingImage(DisplayingImageEvent $event) {
|
||||
global $user, $page;
|
||||
/**
|
||||
* Deny images upon insufficient permissions.
|
||||
**/
|
||||
$user_view_level = Ratings::get_user_privs($user);
|
||||
$user_view_level = preg_split('//', $user_view_level, -1);
|
||||
if(!in_array($event->image->rating, $user_view_level)) {
|
||||
$page->set_mode("redirect");
|
||||
$page->set_redirect(make_link("post/list"));
|
||||
}
|
||||
}
|
||||
|
||||
public function onRatingSet(RatingSetEvent $event) {
|
||||
if(empty($event->image->rating)){
|
||||
$old_rating = "";
|
||||
}else{
|
||||
$old_rating = $event->image->rating;
|
||||
}
|
||||
$this->set_rating($event->image->id, $event->rating, $old_rating);
|
||||
}
|
||||
|
||||
public function onImageInfoBoxBuilding(ImageInfoBoxBuildingEvent $event) {
|
||||
$event->add_part($this->theme->get_rater_html($event->image->id, $event->image->rating, $this->can_rate()), 80);
|
||||
}
|
||||
|
||||
public function onImageInfoSet(ImageInfoSetEvent $event) {
|
||||
if($this->can_rate() && isset($_POST["rating"])) {
|
||||
$rating = $_POST["rating"];
|
||||
if (Ratings::rating_is_valid($rating)) {
|
||||
send_event(new RatingSetEvent($event->image, $rating));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public function onDisplayingImage(DisplayingImageEvent $event)
|
||||
{
|
||||
global $user, $page;
|
||||
/**
|
||||
* Deny images upon insufficient permissions.
|
||||
**/
|
||||
$user_view_level = Ratings::get_user_privs($user);
|
||||
$user_view_level = preg_split('//', $user_view_level, -1);
|
||||
if (!in_array($event->image->rating, $user_view_level)) {
|
||||
$page->set_mode("redirect");
|
||||
$page->set_redirect(make_link("post/list"));
|
||||
}
|
||||
}
|
||||
|
||||
public function onRatingSet(RatingSetEvent $event)
|
||||
{
|
||||
if (empty($event->image->rating)) {
|
||||
$old_rating = "";
|
||||
} else {
|
||||
$old_rating = $event->image->rating;
|
||||
}
|
||||
$this->set_rating($event->image->id, $event->rating, $old_rating);
|
||||
}
|
||||
|
||||
public function onImageInfoBoxBuilding(ImageInfoBoxBuildingEvent $event)
|
||||
{
|
||||
$event->add_part($this->theme->get_rater_html($event->image->id, $event->image->rating, $this->can_rate()), 80);
|
||||
}
|
||||
|
||||
public function onImageInfoSet(ImageInfoSetEvent $event)
|
||||
{
|
||||
if ($this->can_rate() && isset($_POST["rating"])) {
|
||||
$rating = $_POST["rating"];
|
||||
if (Ratings::rating_is_valid($rating)) {
|
||||
send_event(new RatingSetEvent($event->image, $rating));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public function onParseLinkTemplate(ParseLinkTemplateEvent $event) {
|
||||
$event->replace('$rating', $this->rating_to_human($event->image->rating));
|
||||
}
|
||||
public function onParseLinkTemplate(ParseLinkTemplateEvent $event)
|
||||
{
|
||||
$event->replace('$rating', $this->rating_to_human($event->image->rating));
|
||||
}
|
||||
|
||||
public function onSearchTermParse(SearchTermParseEvent $event) {
|
||||
global $user;
|
||||
|
||||
$matches = array();
|
||||
if(is_null($event->term) && $this->no_rating_query($event->context)) {
|
||||
$set = Ratings::privs_to_sql(Ratings::get_user_privs($user));
|
||||
$event->add_querylet(new Querylet("rating IN ($set)"));
|
||||
}
|
||||
if(preg_match("/^rating[=|:](?:([sqeu]+)|(safe|questionable|explicit|unknown))$/D", strtolower($event->term), $matches)) {
|
||||
$ratings = $matches[1] ? $matches[1] : $matches[2][0];
|
||||
$ratings = array_intersect(str_split($ratings), str_split(Ratings::get_user_privs($user)));
|
||||
$set = "'" . join("', '", $ratings) . "'";
|
||||
$event->add_querylet(new Querylet("rating IN ($set)"));
|
||||
}
|
||||
}
|
||||
public function onSearchTermParse(SearchTermParseEvent $event)
|
||||
{
|
||||
global $user;
|
||||
|
||||
$matches = [];
|
||||
if (is_null($event->term) && $this->no_rating_query($event->context)) {
|
||||
$set = Ratings::privs_to_sql(Ratings::get_user_privs($user));
|
||||
$event->add_querylet(new Querylet("rating IN ($set)"));
|
||||
}
|
||||
if (preg_match("/^rating[=|:](?:([sqeu]+)|(safe|questionable|explicit|unknown))$/D", strtolower($event->term), $matches)) {
|
||||
$ratings = $matches[1] ? $matches[1] : $matches[2][0];
|
||||
$ratings = array_intersect(str_split($ratings), str_split(Ratings::get_user_privs($user)));
|
||||
$set = "'" . join("', '", $ratings) . "'";
|
||||
$event->add_querylet(new Querylet("rating IN ($set)"));
|
||||
}
|
||||
}
|
||||
|
||||
public function onPageRequest(PageRequestEvent $event) {
|
||||
global $user, $page;
|
||||
|
||||
if ($event->page_matches("admin/bulk_rate")) {
|
||||
if(!$user->is_admin()) {
|
||||
throw new PermissionDeniedException();
|
||||
}
|
||||
else {
|
||||
$n = 0;
|
||||
while(true) {
|
||||
$images = Image::find_images($n, 100, Tag::explode($_POST["query"]));
|
||||
if(count($images) == 0) break;
|
||||
|
||||
reset($images); // rewind to first element in array.
|
||||
|
||||
foreach($images as $image) {
|
||||
send_event(new RatingSetEvent($image, $_POST['rating']));
|
||||
}
|
||||
$n += 100;
|
||||
}
|
||||
#$database->execute("
|
||||
# update images set rating=? where images.id in (
|
||||
# select image_id from image_tags join tags
|
||||
# on image_tags.tag_id = tags.id where tags.tag = ?);
|
||||
# ", array($_POST["rating"], $_POST["tag"]));
|
||||
$page->set_mode("redirect");
|
||||
$page->set_redirect(make_link("post/list"));
|
||||
}
|
||||
}
|
||||
}
|
||||
public function onPageRequest(PageRequestEvent $event)
|
||||
{
|
||||
global $user, $page;
|
||||
|
||||
if ($event->page_matches("admin/bulk_rate")) {
|
||||
if (!$user->is_admin()) {
|
||||
throw new PermissionDeniedException();
|
||||
} else {
|
||||
$n = 0;
|
||||
while (true) {
|
||||
$images = Image::find_images($n, 100, Tag::explode($_POST["query"]));
|
||||
if (count($images) == 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
reset($images); // rewind to first element in array.
|
||||
|
||||
foreach ($images as $image) {
|
||||
send_event(new RatingSetEvent($image, $_POST['rating']));
|
||||
}
|
||||
$n += 100;
|
||||
}
|
||||
#$database->execute("
|
||||
# update images set rating=? where images.id in (
|
||||
# select image_id from image_tags join tags
|
||||
# on image_tags.tag_id = tags.id where tags.tag = ?);
|
||||
# ", array($_POST["rating"], $_POST["tag"]));
|
||||
$page->set_mode("redirect");
|
||||
$page->set_redirect(make_link("post/list"));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static function get_user_privs(User $user): string {
|
||||
global $config;
|
||||
public static function get_user_privs(User $user): string
|
||||
{
|
||||
global $config;
|
||||
|
||||
if($user->is_anonymous()) {
|
||||
$sqes = $config->get_string("ext_rating_anon_privs");
|
||||
}
|
||||
else if($user->is_admin()) {
|
||||
$sqes = $config->get_string("ext_rating_admin_privs");
|
||||
}
|
||||
else {
|
||||
$sqes = $config->get_string("ext_rating_user_privs");
|
||||
}
|
||||
return $sqes;
|
||||
}
|
||||
if ($user->is_anonymous()) {
|
||||
$sqes = $config->get_string("ext_rating_anon_privs");
|
||||
} elseif ($user->is_admin()) {
|
||||
$sqes = $config->get_string("ext_rating_admin_privs");
|
||||
} else {
|
||||
$sqes = $config->get_string("ext_rating_user_privs");
|
||||
}
|
||||
return $sqes;
|
||||
}
|
||||
|
||||
public static function privs_to_sql(string $sqes): string {
|
||||
$arr = array();
|
||||
$length = strlen($sqes);
|
||||
for($i=0; $i<$length; $i++) {
|
||||
$arr[] = "'" . $sqes[$i] . "'";
|
||||
}
|
||||
$set = join(', ', $arr);
|
||||
return $set;
|
||||
}
|
||||
public static function privs_to_sql(string $sqes): string
|
||||
{
|
||||
$arr = [];
|
||||
$length = strlen($sqes);
|
||||
for ($i=0; $i<$length; $i++) {
|
||||
$arr[] = "'" . $sqes[$i] . "'";
|
||||
}
|
||||
$set = join(', ', $arr);
|
||||
return $set;
|
||||
}
|
||||
|
||||
public static function rating_to_human(string $rating): string {
|
||||
switch($rating) {
|
||||
case "s": return "Safe";
|
||||
case "q": return "Questionable";
|
||||
case "e": return "Explicit";
|
||||
default: return "Unknown";
|
||||
}
|
||||
}
|
||||
public static function rating_to_human(string $rating): string
|
||||
{
|
||||
switch ($rating) {
|
||||
case "s": return "Safe";
|
||||
case "q": return "Questionable";
|
||||
case "e": return "Explicit";
|
||||
default: return "Unknown";
|
||||
}
|
||||
}
|
||||
|
||||
public static function rating_is_valid(string $rating): bool {
|
||||
switch($rating) {
|
||||
case "s":
|
||||
case "q":
|
||||
case "e":
|
||||
case "u":
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
public static function rating_is_valid(string $rating): bool
|
||||
{
|
||||
switch ($rating) {
|
||||
case "s":
|
||||
case "q":
|
||||
case "e":
|
||||
case "u":
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* FIXME: this is a bit ugly and guessey, should have proper options
|
||||
*/
|
||||
private function can_rate(): bool {
|
||||
global $config, $user;
|
||||
if($user->is_anonymous() && $config->get_string("ext_rating_anon_privs") == "sqeu") return false;
|
||||
if($user->is_admin()) return true;
|
||||
if(!$user->is_anonymous() && $config->get_string("ext_rating_user_privs") == "sqeu") return true;
|
||||
return false;
|
||||
}
|
||||
/**
|
||||
* FIXME: this is a bit ugly and guessey, should have proper options
|
||||
*/
|
||||
private function can_rate(): bool
|
||||
{
|
||||
global $config, $user;
|
||||
if ($user->is_anonymous() && $config->get_string("ext_rating_anon_privs") == "sqeu") {
|
||||
return false;
|
||||
}
|
||||
if ($user->is_admin()) {
|
||||
return true;
|
||||
}
|
||||
if (!$user->is_anonymous() && $config->get_string("ext_rating_user_privs") == "sqeu") {
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* #param string[] $context
|
||||
*/
|
||||
private function no_rating_query(array $context): bool {
|
||||
foreach($context as $term) {
|
||||
if(preg_match("/^rating[=|:]/", $term)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
/**
|
||||
* #param string[] $context
|
||||
*/
|
||||
private function no_rating_query(array $context): bool
|
||||
{
|
||||
foreach ($context as $term) {
|
||||
if (preg_match("/^rating[=|:]/", $term)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
private function install() {
|
||||
global $database, $config;
|
||||
private function install()
|
||||
{
|
||||
global $database, $config;
|
||||
|
||||
if($config->get_int("ext_ratings2_version") < 1) {
|
||||
$database->Execute("ALTER TABLE images ADD COLUMN rating CHAR(1) NOT NULL DEFAULT 'u'");
|
||||
$database->Execute("CREATE INDEX images__rating ON images(rating)");
|
||||
$config->set_int("ext_ratings2_version", 3);
|
||||
}
|
||||
if ($config->get_int("ext_ratings2_version") < 1) {
|
||||
$database->Execute("ALTER TABLE images ADD COLUMN rating CHAR(1) NOT NULL DEFAULT 'u'");
|
||||
$database->Execute("CREATE INDEX images__rating ON images(rating)");
|
||||
$config->set_int("ext_ratings2_version", 3);
|
||||
}
|
||||
|
||||
if($config->get_int("ext_ratings2_version") < 2) {
|
||||
$database->Execute("CREATE INDEX images__rating ON images(rating)");
|
||||
$config->set_int("ext_ratings2_version", 2);
|
||||
}
|
||||
if ($config->get_int("ext_ratings2_version") < 2) {
|
||||
$database->Execute("CREATE INDEX images__rating ON images(rating)");
|
||||
$config->set_int("ext_ratings2_version", 2);
|
||||
}
|
||||
|
||||
if($config->get_int("ext_ratings2_version") < 3) {
|
||||
$database->Execute("ALTER TABLE images CHANGE rating rating CHAR(1) NOT NULL DEFAULT 'u'");
|
||||
$config->set_int("ext_ratings2_version", 3);
|
||||
}
|
||||
}
|
||||
if ($config->get_int("ext_ratings2_version") < 3) {
|
||||
$database->Execute("ALTER TABLE images CHANGE rating rating CHAR(1) NOT NULL DEFAULT 'u'");
|
||||
$config->set_int("ext_ratings2_version", 3);
|
||||
}
|
||||
}
|
||||
|
||||
private function set_rating(int $image_id, string $rating, string $old_rating) {
|
||||
global $database;
|
||||
if($old_rating != $rating){
|
||||
$database->Execute("UPDATE images SET rating=? WHERE id=?", array($rating, $image_id));
|
||||
log_info("rating", "Rating for Image #{$image_id} set to: ".$this->rating_to_human($rating));
|
||||
}
|
||||
}
|
||||
private function set_rating(int $image_id, string $rating, string $old_rating)
|
||||
{
|
||||
global $database;
|
||||
if ($old_rating != $rating) {
|
||||
$database->Execute("UPDATE images SET rating=? WHERE id=?", [$rating, $image_id]);
|
||||
log_info("rating", "Rating for Image #{$image_id} set to: ".$this->rating_to_human($rating));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user