<?php

declare(strict_types=1);

namespace Kreait\Firebase\Database\Reference;

use Kreait\Firebase\Exception\InvalidArgumentException;
use Psr\Http\Message\UriInterface;

use function explode;
use function mb_strlen;
use function mb_substr_count;
use function preg_match;
use function preg_quote;
use function rawurldecode;
use function sprintf;
use function trim;

class Validator
{
    public const MAX_DEPTH = 32;
    public const MAX_KEY_SIZE = 768;
    public const INVALID_KEY_CHARS = '.$#[]';

    /**
     * Checks the reference URI for invalid properties.
     *
     * @throws InvalidArgumentException
     */
    public function validateUri(UriInterface $uri): void
    {
        $this->validatePath($uri->getPath());
    }

    /**
     * Checks a reference path for invalid properties.
     *
     * @throws InvalidArgumentException
     */
    public function validatePath(string $path): void
    {
        $path = trim($path, '/');

        $this->validateDepth($path);

        foreach (explode('/', $path) as $key) {
            $this->validateKeySize($key);
            $this->validateChars($key);
        }
    }

    private function validateDepth(string $path): void
    {
        $depth = mb_substr_count($path, '/') + 1;

        if ($depth > self::MAX_DEPTH) {
            throw new InvalidArgumentException(sprintf(
                'A reference location must not more than %d levels deep, "%s" has %d.',
                self::MAX_DEPTH,
                $path,
                $depth,
            ));
        }
    }

    private function validateKeySize(string $key): void
    {
        if (($length = mb_strlen($key, '8bit')) > self::MAX_KEY_SIZE) {
            throw new InvalidArgumentException(sprintf(
                'A reference\'s child key must not be larger than %d bytes, "%s" has a size of %d bytes.',
                self::MAX_KEY_SIZE,
                $key,
                $length,
            ));
        }
    }

    private function validateChars(string $key): void
    {
        $key = rawurldecode($key);

        $pattern = sprintf('/[%s]/', preg_quote(self::INVALID_KEY_CHARS, '/'));

        if (preg_match($pattern, $key)) {
            throw new InvalidArgumentException(sprintf(
                'The child key "%s" contains one of the following invalid characters: "%s"',
                $key,
                self::INVALID_KEY_CHARS,
            ));
        }
    }
}