shimmie2/ext/replace_file/main.php

<?php

declare(strict_types=1);

namespace Shimmie2;

class ReplaceFile extends Extension
{
    /** @var ReplaceFileTheme */
    protected Themelet $theme;

    public function onPageRequest(PageRequestEvent $event): void
    {
        global $cache, $page, $user;

        if ($event->page_matches("replace/{image_id}", method: "GET", permission: Permissions::REPLACE_IMAGE)) {
            $image_id = $event->get_iarg('image_id');
            $image = Image::by_id_ex($image_id);
            $this->theme->display_replace_page($page, $image_id);
        }

        if ($event->page_matches("replace/{image_id}", method: "POST", permission: Permissions::REPLACE_IMAGE)) {
            $image_id = $event->get_iarg('image_id');
            $image = Image::by_id_ex($image_id);

            if (empty($event->get_POST("url")) && count($_FILES) == 0) {
                $page->set_mode(PageMode::REDIRECT);
                $page->set_redirect(make_link("replace/$image_id"));
                return;
            }

            if (!empty($event->get_POST("url"))) {
                $tmp_filename = shm_tempnam("transload");
                fetch_url($event->req_POST("url"), $tmp_filename);
                send_event(new ImageReplaceEvent($image, $tmp_filename));
            } elseif (count($_FILES) > 0) {
                send_event(new ImageReplaceEvent($image, $_FILES["data"]['tmp_name']));
            }
            if ($event->get_POST("source")) {
                send_event(new SourceSetEvent($image, $event->req_POST("source")));
            }
            $cache->delete("thumb-block:{$image_id}");
            $page->set_mode(PageMode::REDIRECT);
            $page->set_redirect(make_link("post/view/$image_id"));
        }
    }

    public function onImageAdminBlockBuilding(ImageAdminBlockBuildingEvent $event): void
    {
        global $user;

        /* In the future, could perhaps allow users to replace images that they own as well... */
        if ($user->can(Permissions::REPLACE_IMAGE)) {
            $event->add_button("Replace", "replace/{$event->image->id}");
        }
    }

    public function onImageReplace(ImageReplaceEvent $event): void
    {
        $image = $event->image;

        $duplicate = Image::by_hash($event->new_hash);
        if (!is_null($duplicate) && $duplicate->id != $image->id) {
            throw new ImageReplaceException("A different post >>{$duplicate->id} already has hash {$duplicate->hash}");
        }

        $image->remove_image_only(); // Actually delete the old image file from disk

        $target = warehouse_path(Image::IMAGE_DIR, $event->new_hash);
        try {
            \Safe\copy($event->tmp_filename, $target);
        } catch (\Exception $e) {
            throw new ImageReplaceException("Failed to copy file from uploads ({$event->tmp_filename}) to archive ($target): {$e->getMessage()}");
        }
        unlink($event->tmp_filename);

        // update metadata and save metadata to DB
        $event->image->hash = $event->new_hash;
        $event->image->filesize = \Safe\filesize($target);
        $event->image->set_mime(MimeType::get_for_file($target));
        send_event(new MediaCheckPropertiesEvent($image));
        $image->save_to_db();

        send_event(new ThumbnailGenerationEvent($image));

        log_info("image", "Replaced >>{$image->id} {$event->old_hash} with {$event->new_hash}");
    }
}