<?php

declare(strict_types=1);

namespace Laminas\Dom\Document;

use DOMNode;
use Laminas\Dom\Document;
use Laminas\Dom\DOMXPath;

use function array_merge;
use function count;
use function explode;
use function implode;
use function is_array;
use function is_string;
use function ltrim;
use function mt_rand;
use function preg_replace;
use function preg_replace_callback;
use function preg_split;
use function sprintf;
use function str_replace;
use function strpos;
use function strstr;
use function strtolower;
use function trim;
use function uniqid;

/**
 * Query object executable in a Laminas\Dom\Document
 */
class Query
{
    /**#@+
     * Query types
     */
    public const TYPE_XPATH = 'TYPE_XPATH';
    public const TYPE_CSS   = 'TYPE_CSS';
    /**#@-*/

    /**
     * Perform the query on Document
     *
     * @param  string    $expression CSS selector or XPath query
     * @param  Document  $document   Document to query
     * @param  string    $type       The type of $expression
     * @return NodeList
     */
    public static function execute(
        $expression,
        Document $document,
        $type = self::TYPE_XPATH,
        ?DOMNode $contextNode = null
    ) {
        // Expression check
        if ($type === static::TYPE_CSS) {
            $expression = static::cssToXpath($expression);
        }
        $xpath = new DOMXPath($document->getDomDocument());

        $xpathNamespaces = $document->getXpathNamespaces();
        foreach ($xpathNamespaces as $prefix => $namespaceUri) {
            $xpath->registerNamespace((string) $prefix, $namespaceUri);
        }

        if ($xpathPhpfunctions = $document->getXpathPhpFunctions()) {
            $xpath->registerNamespace('php', 'http://php.net/xpath');
            if ($xpathPhpfunctions === true) {
                $xpath->registerPhpFunctions();
            } else {
                $xpath->registerPhpFunctions($xpathPhpfunctions);
            }
        }

        $nodeList = $xpath->queryWithErrorException($expression, $contextNode);
        return new NodeList($nodeList);
    }

    /**
     * Transform CSS expression to XPath
     *
     * @param  string $path
     * @return string
     */
    public static function cssToXpath($path)
    {
        $path = (string) $path;
        if (strstr($path, ',')) {
            $paths       = explode(',', $path);
            $expressions = [];
            foreach ($paths as $path) {
                $xpath = static::cssToXpath(trim($path));
                if (is_string($xpath)) {
                    $expressions[] = $xpath;
                } elseif (is_array($xpath)) {
                    $expressions = array_merge($expressions, $xpath);
                }
            }
            return implode('|', $expressions);
        }

        do {
            $placeholder = '{' . uniqid((string) mt_rand(), true) . '}';
        } while (strpos($path, $placeholder) !== false);

        // Arbitrary attribute value contains whitespace
        $path = preg_replace_callback(
            '/\[\S+?([\'"])((?!\1|\\\1).*?)\1\]/',
            function ($matches) use ($placeholder) {
                return str_replace($matches[2], preg_replace('/\s+/', $placeholder, $matches[2]), $matches[0]);
            },
            $path
        );

        $paths    = ['//'];
        $path     = preg_replace('|\s*>\s*|', '>', $path);
        $segments = preg_split('/\s+/', $path);
        $segments = str_replace($placeholder, ' ', $segments);

        foreach ($segments as $key => $segment) {
            $pathSegment = static::_tokenize($segment);
            if (0 === $key) {
                if (0 === strpos($pathSegment, '[contains(')) {
                    $paths[0] .= '*' . ltrim($pathSegment, '*');
                } else {
                    $paths[0] .= $pathSegment;
                }
                continue;
            }
            if (0 === strpos($pathSegment, '[contains(')) {
                foreach ($paths as $pathKey => $xpath) {
                    $paths[$pathKey] .= '//*' . ltrim($pathSegment, '*');
                    $paths[]          = $xpath . $pathSegment;
                }
            } else {
                foreach ($paths as $pathKey => $xpath) {
                    $paths[$pathKey] .= '//' . $pathSegment;
                }
            }
        }

        if (1 === count($paths)) {
            return $paths[0];
        }
        return implode('|', $paths);
    }

    // @codingStandardsIgnoreStart
    /**
     * Tokenize CSS expressions to XPath
     *
     * @param  string $expression
     * @return string
     */
    protected static function _tokenize($expression)
    {
        // @codingStandardsIgnoreEnd
        // Child selectors
        $expression = str_replace('>', '/', $expression);

        // IDs
        $expression = preg_replace('|#([a-z][a-z0-9_-]*)|i', '[@id=\'$1\']', $expression);
        $expression = preg_replace('|(?<![a-z0-9_-])(\[@id=)|i', '*$1', $expression);

        // arbitrary attribute strict equality
        $expression = preg_replace_callback(
            '/\[@?([a-z0-9_-]+)=([\'"])((?!\2|\\\2).*?)\2\]/i',
            function ($matches) {
                return sprintf("[@%s='%s']", strtolower($matches[1]), str_replace("'", "\\'", $matches[3]));
            },
            $expression
        );

        // arbitrary attribute contains full word
        $expression = preg_replace_callback(
            '/\[([a-z0-9_-]+)~=([\'"])((?!\2|\\\2).*?)\2\]/i',
            function ($matches) {
                return "[contains(concat(' ', normalize-space(@" . strtolower($matches[1]) . "), ' '), ' "
                     . $matches[3] . " ')]";
            },
            $expression
        );

        // arbitrary attribute contains specified content
        $expression = preg_replace_callback(
            '/\[([a-z0-9_-]+)\*=([\'"])((?!\2|\\\2).*?)\2\]/i',
            function ($matches) {
                return "[contains(@" . strtolower($matches[1]) . ", '"
                     . $matches[3] . "')]";
            },
            $expression
        );

        // Classes
        if (false === strpos($expression, "[@")) {
            $expression = preg_replace(
                '|\.([a-z][a-z0-9_-]*)|i',
                "[contains(concat(' ', normalize-space(@class), ' '), ' \$1 ')]",
                $expression
            );
        }

        /** Laminas-9764 -- remove double asterisk */
        $expression = str_replace('**', '*', $expression);

        return $expression;
    }
}