mirror of
https://github.com/Respect/Validation.git
synced 2026-03-15 14:55:44 +01:00
99 lines
3 KiB
PHP
99 lines
3 KiB
PHP
<?php
|
|
|
|
/*
|
|
* Copyright (c) Alexandre Gomes Gaigalas <alganet@gmail.com>
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Respect\Validation\Rules;
|
|
|
|
use Attribute;
|
|
use Psr\Http\Message\StreamInterface;
|
|
use Psr\Http\Message\UploadedFileInterface;
|
|
use Respect\Validation\Exceptions\InvalidRuleConstructorException;
|
|
use Respect\Validation\Message\Template;
|
|
use Respect\Validation\Result;
|
|
use Respect\Validation\Rule;
|
|
use Respect\Validation\Rules\Core\Wrapper;
|
|
use SplFileInfo;
|
|
|
|
use function filesize;
|
|
use function is_string;
|
|
use function ucfirst;
|
|
|
|
#[Attribute(Attribute::TARGET_PROPERTY | Attribute::IS_REPEATABLE)]
|
|
#[Template(
|
|
'The size in {{unit|trans}} of',
|
|
'The size in {{unit|trans}} of',
|
|
Size::TEMPLATE_STANDARD,
|
|
)]
|
|
#[Template(
|
|
'{{name}} must be a filename or an instance of SplFileInfo or a PSR-7 interface',
|
|
'{{name}} must not be a filename or an instance of SplFileInfo or a PSR-7 interface',
|
|
self::TEMPLATE_WRONG_TYPE,
|
|
)]
|
|
final class Size extends Wrapper
|
|
{
|
|
public const string TEMPLATE_WRONG_TYPE = '__wrong_type__';
|
|
|
|
private const array DATA_STORAGE_UNITS = [
|
|
'B' => ['name' => 'bytes', 'bytes' => 1],
|
|
'KB' => ['name' => 'kilobytes', 'bytes' => 1024],
|
|
'MB' => ['name' => 'megabytes', 'bytes' => 1024 ** 2],
|
|
'GB' => ['name' => 'gigabytes', 'bytes' => 1024 ** 3],
|
|
'TB' => ['name' => 'terabytes', 'bytes' => 1024 ** 4],
|
|
'PB' => ['name' => 'petabytes', 'bytes' => 1024 ** 5],
|
|
'EB' => ['name' => 'exabytes', 'bytes' => 1024 ** 6],
|
|
'ZB' => ['name' => 'zettabytes', 'bytes' => 1024 ** 7],
|
|
'YB' => ['name' => 'yottabytes', 'bytes' => 1024 ** 8],
|
|
];
|
|
|
|
/** @param "B"|"KB"|"MB"|"GB"|"TB"|"PB"|"EB"|"ZB"|"YB" $unit */
|
|
public function __construct(
|
|
private readonly string $unit,
|
|
Rule $rule,
|
|
) {
|
|
if (!isset(self::DATA_STORAGE_UNITS[$unit])) {
|
|
throw new InvalidRuleConstructorException('"%s" is not a recognized data storage unit.', $unit);
|
|
}
|
|
|
|
parent::__construct($rule);
|
|
}
|
|
|
|
public function evaluate(mixed $input): Result
|
|
{
|
|
$size = $this->getSize($input);
|
|
if ($size === null) {
|
|
return Result::failed($input, $this, [], self::TEMPLATE_WRONG_TYPE)
|
|
->withId('size' . ucfirst($this->rule->evaluate($input)->id));
|
|
}
|
|
|
|
$result = $this->rule->evaluate($this->getSize($input) / self::DATA_STORAGE_UNITS[$this->unit]['bytes']);
|
|
$parameters = ['unit' => self::DATA_STORAGE_UNITS[$this->unit]['name']];
|
|
|
|
return Result::fromAdjacent($input, 'size', $this, $result, $parameters);
|
|
}
|
|
|
|
private function getSize(mixed $input): int|null
|
|
{
|
|
if (is_string($input)) {
|
|
return (int) filesize($input);
|
|
}
|
|
|
|
if ($input instanceof SplFileInfo) {
|
|
return $input->getSize();
|
|
}
|
|
|
|
if ($input instanceof UploadedFileInterface) {
|
|
return $input->getSize();
|
|
}
|
|
|
|
if ($input instanceof StreamInterface) {
|
|
return $input->getSize();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|