lekoala / silverstripe-devtoolkit

src/Tasks/RehashImagesTask.php

<?php

namespace LeKoala\DevToolkit\Tasks;

use LeKoala\DevToolkit\BuildTaskTools;
use SilverStripe\ORM\DB;
use SilverStripe\Assets\Image;
use SilverStripe\Dev\BuildTask;
use SilverStripe\Versioned\Versioned;

The type SilverStripe\Versioned\Versioned was not found. Maybe you did not declare it correctly or list all dependencies?

use SilverStripe\Assets\Storage\Sha1FileHashingService;

/**
 * In case images are resized by an external utility
 * file hash is not valid anymore
 *
 * We need a way to rehash them
 */
class RehashImagesTask extends BuildTask
{
    use BuildTaskTools;

    protected $title = "Rehash Images";
    private static $segment = 'RehashImagesTask';

The private property $segment is not used, and could be removed.

    public function run($request)
    {
        $this->request = $request;
        $this->addOption("go", "Set this to 1 to proceed", 0);

        $options = $this->askOptions();

        $go = $options['go'];

        $service = new Sha1FileHashingService();

        $images = Image::get();
        $tofix = $todelete = 0;
        foreach ($images as $image) {
            $hash = $image->getHash();

            $stream  = $image->getStream();
            if (!$stream) {
                $filename = $image->getFilename();
                $location = ASSETS_PATH . '/' . $filename;
                if (!is_file($location)) {
                    $todelete++;
                    if ($go) {
                        $this->message("Deleted file " . $image->ID . " since the file was not found");
                        $image->delete();
                    } else {
                        $this->message("Could not read file at $location. It would be deleted by this task.", "bad");
                    }
                    continue;
                }
                $stream = fopen($location, 'rb');
            }

            $fullhash = $service->computeFromStream($stream);

            if ($hash != $fullhash) {
                $tofix++;
                if ($go) {
                    DB::query("UPDATE File SET FileHash = '" . $fullhash . "' WHERE ID = " . $image->ID);
                    DB::query("UPDATE File_Live SET FileHash = '" . $fullhash . "' WHERE ID = " . $image->ID);
                    $this->message($image->ID . " has been fixed");
                } else {
                    $this->message($image->ID . " hash mismatch : $hash vs $fullhash. This task can fix this hash.");
                }
            }
        }

        if (!$tofix && !$todelete) {
            $this->message("All files are good", "good");
        }
    }
}