slince / upload

src/Constraint/SizeConstraint.php

<?php

/*
 * This file is part of the slince/upload package.
 *
 * (c) Slince <taosikai@yeah.net>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Slince\Upload\Constraint;

use Symfony\Component\HttpFoundation\File\UploadedFile;

class SizeConstraint implements ConstraintInterface
{
    /**
     * Maximum file size
     * @var int
     */
    protected $maxSize;

    /**
     * Minimum file size
     * @var int
     */
    protected $minSize;

    /**
     * Limit file size (use "B", "K", M", or "G")
     * @param int|string|null $minSize
     * @param int|string|null $maxSize
     */
    public function __construct($minSize = null, $maxSize = null)
    {
        if ($minSize !== null) {
            $this->minSize = static::humanReadableToBytes($minSize);
        }
        if ($maxSize !== null) {
            $this->maxSize = static::humanReadableToBytes($maxSize);
        }
    }

    /**
     * {@inheritdoc}
     */
    public function validate(UploadedFile $file)
    {
        $size = method_exists($file, 'getClientSize') ?
            $file->getClientSize() : $file->getSize();

The method Symfony\Component\HttpFoundation\File\UploadedFile::getClientSize() has been deprecated with message: since Symfony 4.1, use getSize() instead.

        return ($this->maxSize === null || $size <= $this->maxSize)
            && ($this->minSize === null || $size >= $this->minSize);
    }

    /**
     * {@inheritdoc}
     */
    public function getErrorMessage(UploadedFile $file)
    {
        $message = null;
        if ($this->minSize && $this->maxSize) {
            $message = sprintf('File size should between %s and %s', $this->minSize, $this->maxSize);
        } elseif ($this->minSize === null) {
            $message = sprintf('File size should less than %s', $this->maxSize);
        } elseif ($this->maxSize === null) {
            $message = sprintf('File size should greater than %s', $this->minSize);
        }
        return $message;
    }

    /**
     * Convert human readable file size (e.g. "10K" or "3M") into bytes
     * @link https://github.com/brandonsavage/Upload/blob/master/src/Upload/File.php#L446
     * @param  string $input
     * @return int
     */
    public static function humanReadableToBytes($input)
    {
        $number = (int)$input;
        $units = array(
            'b' => 1,
            'k' => 1024,
            'm' => 1048576,
            'g' => 1073741824
        );
        $unit = strtolower(substr($input, -1));
        if (isset($units[$unit])) {
            $number = $number * $units[$unit];
        }
        return $number;
    }
}