shimmie2/ext/report_image/main.php

<?php

class RemoveReportedImageEvent extends Event
{
    /** @var  int */
    public $id;

    public function __construct(int $id)
    {
        $this->id = $id;
    }
}

class AddReportedImageEvent extends Event
{
    /** @var ImageReport */
    public $report;

    public function __construct(ImageReport $report)
    {
        $this->report = $report;
    }
}

class ImageReport
{
    /** @var int  */
    public $user_id;
    /** @var int  */
    public $image_id;
    /** @var string  */
    public $reason;

    public function __construct(int $image_id, int $user_id, string $reason)
    {
        $this->image_id = $image_id;
        $this->user_id = $user_id;
        $this->reason = $reason;
    }
}

class ReportImage extends Extension
{
    public function onPageRequest(PageRequestEvent $event)
    {
        global $page, $user;
        if ($event->page_matches("image_report")) {
            if ($event->get_arg(0) == "add") {
                if (!empty($_POST['image_id']) && !empty($_POST['reason'])) {
                    $image_id = int_escape($_POST['image_id']);
                    send_event(new AddReportedImageEvent(new ImageReport($image_id, $user->id, $_POST['reason'])));
                    $page->set_mode(PageMode::REDIRECT);
                    $page->set_redirect(make_link("post/view/$image_id"));
                } else {
                    $this->theme->display_error(500, "Missing input", "Missing image ID or report reason");
                }
            } elseif ($event->get_arg(0) == "remove") {
                if (!empty($_POST['id'])) {
                    if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
                        send_event(new RemoveReportedImageEvent($_POST['id']));
                        $page->set_mode(PageMode::REDIRECT);
                        $page->set_redirect(make_link("image_report/list"));
                    }
                } else {
                    $this->theme->display_error(500, "Missing input", "Missing image ID");
                }
            } elseif ($event->get_arg(0) == "remove_reports_by" && $user->check_auth_token()) {
                if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
                    $this->delete_reports_by(int_escape($_POST['user_id']));
                    $page->set_mode(PageMode::REDIRECT);
                    $page->set_redirect(make_link());
                }
            } elseif ($event->get_arg(0) == "list") {
                if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
                    $this->theme->display_reported_images($page, $this->get_reported_images());
                }
            }
        }
    }

    public function onAddReportedImage(AddReportedImageEvent $event)
    {
        global $cache, $database;
        log_info("report_image", "Adding report of Image #{$event->report->image_id} with reason '{$event->report->reason}'", null, ["image_id" => $event->report->image_id]);
        $database->Execute(
            "INSERT INTO image_reports(image_id, reporter_id, reason)
				VALUES (?, ?, ?)",
            [$event->report->image_id, $event->report->user_id, $event->report->reason]
        );
        $cache->delete("image-report-count");
    }

    public function onRemoveReportedImage(RemoveReportedImageEvent $event)
    {
        global $cache, $database;
        $database->Execute("DELETE FROM image_reports WHERE id = ?", [$event->id]);
        $cache->delete("image-report-count");
    }

    public function onUserPageBuilding(UserPageBuildingEvent $event)
    {
        global $user;
        if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
            $this->theme->get_nuller($event->display_user);
        }
    }

    public function onDisplayingImage(DisplayingImageEvent $event)
    {
        global $user;
        if ($user->can(Permissions::CREATE_IMAGE_REPORT)) {
            $reps = $this->get_reports($event->image);
            $this->theme->display_image_banner($event->image, $reps);
        }
    }


    public function onPageSubNavBuilding(PageSubNavBuildingEvent $event)
    {
        global $user;
        if ($event->parent==="system") {
            if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
                $count = $this->count_reported_images();
                $h_count = $count > 0 ? " ($count)" : "";

                $event->add_nav_link("image_report", new Link('image_report/list'), "Reported Images$h_count");
            }
        }
    }

    public function onUserBlockBuilding(UserBlockBuildingEvent $event)
    {
        global $user;
        if ($user->can(Permissions::VIEW_IMAGE_REPORT)) {
            $count = $this->count_reported_images();
            $h_count = $count > 0 ? " ($count)" : "";
            $event->add_link("Reported Images$h_count", make_link("image_report/list"));
        }
    }

    public function onImageDeletion(ImageDeletionEvent $event)
    {
        global $cache, $database;
        $database->Execute("DELETE FROM image_reports WHERE image_id = ?", [$event->image->id]);
        $cache->delete("image-report-count");
    }

    public function onUserDeletion(UserDeletionEvent $event)
    {
        $this->delete_reports_by($event->id);
    }

    public function onSetupBuilding(SetupBuildingEvent $event)
    {
        $sb = new SetupBlock("Image Reports");

        $opts = [
            "Reporter Only" => "user",
            "Reason Only" => "reason",
            "Both" => "both",
            "None" => "none",
        ];
        $sb->add_choice_option("report_image_publicity", $opts, "Show publicly: ");

        $event->panel->add_block($sb);
    }

    public function delete_reports_by(int $user_id)
    {
        global $cache, $database;
        $database->execute("DELETE FROM image_reports WHERE reporter_id=?", [$user_id]);
        $cache->delete("image-report-count");
    }

    public function onDatabaseUpgrade(DatabaseUpgradeEvent $event)
    {
        global $database, $config;

        if ($config->get_int("ext_report_image_version") < 1) {
            $database->create_table("image_reports", "
				id SCORE_AIPK,
				image_id INTEGER NOT NULL,
				reporter_id INTEGER NOT NULL,
				reason TEXT NOT NULL,
				FOREIGN KEY (image_id) REFERENCES images(id) ON DELETE CASCADE,
				FOREIGN KEY (reporter_id) REFERENCES users(id) ON DELETE CASCADE
			");
            $config->set_int("ext_report_image_version", 1);
        }
    }

    /**
     * #return ImageReport[]
     */
    public function get_reports(Image $image): array
    {
        global $database;

        $rows = $database->get_all("
			SELECT *
			FROM image_reports
			WHERE image_reports.image_id = :image_id
		", ["image_id" => $image->id]);
        $reps = [];
        foreach ($rows as $row) {
            $reps[] = new ImageReport($row["image_id"], $row["reporter_id"], $row["reason"]);
        }
        return $reps;
    }

    public function get_reported_images(): array
    {
        global $database;

        $all_reports = $database->get_all("
			SELECT image_reports.*, users.name AS reporter_name
			FROM image_reports
			JOIN users ON reporter_id = users.id
			ORDER BY image_reports.id DESC");
        if (is_null($all_reports)) {
            $all_reports = [];
        }

        $reports = [];
        foreach ($all_reports as $report) {
            $image_id = int_escape($report['image_id']);
            $image = Image::by_id($image_id);
            if (is_null($image)) {
                send_event(new RemoveReportedImageEvent($report['id']));
                continue;
            }
            $report['image'] = $image;
            $reports[] = $report;
        }

        return $reports;
    }

    public function count_reported_images(): int
    {
        global $cache, $database;

        $count = $cache->get("image-report-count");
        if (is_null($count) || $count === false) {
            $count = $database->get_one("SELECT count(*) FROM image_reports");
            $cache->set("image-report-count", $count, 600);
        }

        return $count;
    }
}