SortableIterator.php

Go to the documentation of this file.

<?php
 
declare(strict_types=1);
 
namespace ILIAS\Filesystem\Finder\Iterator;
 
use ArrayIterator;
use ILIAS\Filesystem\DTO\Metadata;
use ILIAS\Filesystem\Filesystem;
use InvalidArgumentException;
use IteratorAggregate;
use Traversable;
use Closure;
 
class SortableIterator implements IteratorAggregate
{
    public const SORT_BY_NONE = 0;
    public const SORT_BY_NAME = 1;
    public const SORT_BY_TYPE = 2;
    public const SORT_BY_NAME_NATURAL = 4;
    public const SORT_BY_TIME = 5;
 
    private $sort;
 
    public function __construct(
        private Filesystem $filesystem,
        private Traversable $iterator,
        $sort,
        $reverseOrder = false
    ) {
        $order = $reverseOrder ? -1 : 1;
 
        if (self::SORT_BY_NAME === $sort) {
            $this->sort = static function (Metadata $left, Metadata $right) use ($order): int {
                $leftRealPath = $left->getPath();
                $rightRealPath = $right->getPath();
 
                return $order * strcmp($leftRealPath, $rightRealPath);
            };
        } elseif (self::SORT_BY_NAME_NATURAL === $sort) {
            $this->sort = static function (Metadata $left, Metadata $right) use ($order): int {
                $leftRealPath = $left->getPath();
                $rightRealPath = $right->getPath();
 
                return $order * strnatcmp($leftRealPath, $rightRealPath);
            };
        } elseif (self::SORT_BY_TYPE === $sort) {
            $this->sort = static function (Metadata $left, Metadata $right) use ($order): int {
                if ($left->isDir() && $right->isFile()) {
                    return -$order;
                }
                if ($left->isFile() && $right->isDir()) {
                    return $order;
                }
 
                $leftRealPath = $left->getPath();
                $rightRealPath = $right->getPath();
 
                return $order * strcmp($leftRealPath, $rightRealPath);
            };
        } elseif (self::SORT_BY_TIME === $sort) {
            $this->sort = function (Metadata $left, Metadata $right) use ($order): int {
                $leftTimestamp = $this->filesystem->getTimestamp($left->getPath());
                $rightTimestamp = $this->filesystem->getTimestamp($right->getPath());
 
                return $order * ($leftTimestamp->getTimestamp() - $rightTimestamp->getTimestamp());
            };
        } elseif (self::SORT_BY_NONE === $sort) {
            $this->sort = $order;
        } elseif (is_callable($sort)) {
            $this->sort = $sort;
            if ($reverseOrder) {
                $this->sort = static fn(Metadata $left, Metadata $right): int|float => -$sort($left, $right);
            }
        } else {
            throw new InvalidArgumentException(
                'The SortableIterator takes a PHP callable or a valid built-in sort algorithm as an argument.'
            );
        }
    }
 
    public function getIterator(): \Traversable
    {
        if (1 === $this->sort) {
            return $this->iterator;
        }
 
        $array = iterator_to_array($this->iterator, true);
        if (-1 === $this->sort) {
            $array = array_reverse($array);
        } else {
            uasort($array, $this->sort);
        }
 
        return new ArrayIterator($array);
    }
}