Scrawler\Validator\Storage\Whitelist

Complexity

Total Complexity

9

Size/Duplication

Total Lines	90
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	validate()	0	9	5
A	allowedExtensions()	0	5	1
A	video()	0	18	1
A	pdf()	0	12	1
A	allowedMimeTypes()	0	5	1

<?php

/*
 * This file is part of the Scrawler package.
 *
 * (c) Pranjal Pandey <its.pranjalpandey@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Scrawler\Validator\Storage;

use Symfony\Component\HttpFoundation\File\File;
use Symfony\Component\HttpFoundation\File\UploadedFile;

class Whitelist extends AbstractValidator
{
    /**
     * @var array<string>
     */
    protected array $allowedMimeTypes = [];
    /**
     * @var array<string>
     */
    protected array $allowedExtensions = [];

    /**
     * Set the allowed mime types.
     *
     * @param array<string> $allowedMimeTypes
     */
    public function allowedMimeTypes(array $allowedMimeTypes): self
    {
        $this->allowedMimeTypes = $allowedMimeTypes;

        return $this;
    }

    /**
     * Set the allowed extensions.
     *
     * @param array<string> $allowedExtensions
     */
    public function allowedExtensions(array $allowedExtensions): self
    {
        $this->allowedExtensions = $allowedExtensions;

        return $this;
    }

    /**
     * Validate the uploaded file.
     *
     * @throws \Scrawler\Exception\FileValidationException
     */
    #[\Override]
    public function validate(UploadedFile|File $file): void
    {
        if ([] !== $this->allowedMimeTypes && !\in_array($file->getMimeType(), $this->allowedMimeTypes, true)) {
            throw new \Scrawler\Exception\FileValidationException('Invalid file type.');
        }

        if ([] !== $this->allowedExtensions && !\in_array($file->guessExtension(), $this->allowedExtensions, true)) {
            throw new \Scrawler\Exception\FileValidationException('Invalid file extension.');
        }
    }

    /**
     * Pdf whitelist for pdf with 5MB max file size.
     */
    public static function pdf(): self
    {
        $whitelist = new self();
        $whitelist->allowedMimeTypes([
            'application/pdf',
        ]);
        $whitelist->allowedExtensions([
            'pdf',
        ]);
        $whitelist->maxSize(1024 * 1024 * 5);

        return $whitelist;
    }

    /**
     * Video whitelist for mp4, mov, mpeg, webm with 20MB max file size.
     */
    public static function video(): self
    {
        $whitelist = new self();
        $whitelist->allowedMimeTypes([
            'video/mp4',
            'video/quicktime',
            'video/mpeg',
            'video/webm',
        ]);
        $whitelist->allowedExtensions([
            'mp4',
            'mov',
            'mpeg',
            'webm',
        ]);
        $whitelist->maxSize(1024 * 1024 * 20);

        return $whitelist;
    }
}