dimtrovich / validation

src/Spec/FileFactory.php

<?php

/**
 * This file is part of Dimtrovich/Validation.
 *
 * (c) 2023 Dimitri Sitchet Tomkeu <devcode.dst@gmail.com>
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

namespace Dimtrovich\Validation\Spec;

use BlitzPHP\Utilities\Helpers;
use LogicException;

class FileFactory
{
    /**
     * Create a new fake file.
     */
    public function create(string $name, int|string $kilobytes = 0, ?string $mimeType = null): File
    {
        if (is_string($kilobytes)) {
            return $this->createWithContent($name, $kilobytes);
        }

        return Helpers::tap(new File($name, tmpfile()), function ($file) use ($kilobytes, $mimeType) {
            $file->sizeToReport     = $kilobytes * 1024;
            $file->mimeTypeToReport = $mimeType;
        });
    }

    /**
     * Create a new fake file with content.
     */
    public function createWithContent(string $name, string $content): File
    {
        $tmpfile = tmpfile();

        fwrite($tmpfile, $content);

        return Helpers::tap(new File($name, $tmpfile), function ($file) use ($tmpfile) {
            $file->sizeToReport = fstat($tmpfile)['size'];
        });
    }

    /**
     * Create a new fake image.
     *
     * @throws LogicException
     */
    public function image(string $name, int $width = 10, int $height = 10): File
    {
        return new File($name, $this->generateImage(
            $width,
            $height,
            pathinfo($name, PATHINFO_EXTENSION)

It seems like pathinfo($name, Dimtrovich\Validation\Spec\PATHINFO_EXTENSION) can also be of type array; however, parameter $extension of Dimtrovich\Validation\Spec\FileFactory::generateImage() does only seem to accept string, maybe add an additional type check?

        ));
    }

    /**
     * Generate a dummy image of the given width and height.
     *
     * @return resource
     *
     * @throws LogicException
     */
    protected function generateImage(int $width, int $height, string $extension)
    {
        if (! function_exists('imagecreatetruecolor')) {
            throw new LogicException('GD extension is not installed.');
        }

        return Helpers::tap(tmpfile(), function ($temp) use ($width, $height, $extension) {
            ob_start();

            $extension = in_array($extension, ['jpeg', 'png', 'gif', 'webp', 'wbmp', 'bmp'], true)
                ? strtolower($extension)
                : 'jpeg';

            $image = imagecreatetruecolor($width, $height);

            if (! function_exists($functionName = "image{$extension}")) {
                ob_get_clean();

                throw new LogicException("{$functionName} function is not defined and image cannot be generated.");
            }

            $functionName($image);

            fwrite($temp, ob_get_clean());
        });
    }
}