<?php

/**
 * Part of the Joomla Framework Event Package
 *
 * @copyright  Copyright (C) 2005 - 2021 Open Source Matters, Inc. All rights reserved.
 * @license    GNU General Public License version 2 or later; see LICENSE
 */

namespace Joomla\Event;

/**
 * A class containing an inner listeners priority queue that can be iterated multiple times.
 *
 * @since  1.0
 * @internal
 */
final class ListenersPriorityQueue implements \IteratorAggregate, \Countable
{
    /**
     * The listeners for an event.
     *
     * @var    array
     * @since  2.0.0
     */
    private $listeners = [];

    /**
     * Add a listener with the given priority only if not already present.
     *
     * @param   callable  $callback  A callable function acting as an event listener.
     * @param   integer   $priority  The listener priority.
     *
     * @return  $this
     *
     * @since   1.0
     */
    public function add(callable $callback, int $priority): self
    {
        $this->listeners[$priority][] = $callback;

        return $this;
    }

    /**
     * Remove a listener from the queue.
     *
     * @param   callable  $callback  A callable function acting as an event listener.
     *
     * @return  $this
     *
     * @since   1.0
     */
    public function remove(callable $callback): self
    {
        foreach ($this->listeners as $priority => $listeners) {
            if (($key = array_search($callback, $listeners, true)) !== false) {
                unset($this->listeners[$priority][$key]);
            }
        }

        return $this;
    }

    /**
     * Tell if the listener exists in the queue.
     *
     * @param   callable  $callback  A callable function acting as an event listener.
     *
     * @return  boolean  True if it exists, false otherwise.
     *
     * @since   1.0
     */
    public function has(callable $callback): bool
    {
        foreach ($this->listeners as $priority => $listeners) {
            if (($key = array_search($callback, $listeners, true)) !== false) {
                return true;
            }
        }

        return false;
    }

    /**
     * Get the priority of the given listener.
     *
     * @param   callable  $callback  A callable function acting as an event listener.
     * @param   mixed     $default   The default value to return if the listener doesn't exist.
     *
     * @return  mixed  The listener priority if it exists or the specified default value
     *
     * @since   1.0
     */
    public function getPriority(callable $callback, $default = null)
    {
        foreach ($this->listeners as $priority => $listeners) {
            if (($key = array_search($callback, $listeners, true)) !== false) {
                return $priority;
            }
        }

        return $default;
    }

    /**
     * Get all listeners contained in this queue, sorted according to their priority.
     *
     * @return  callable[]  An array of listeners.
     *
     * @since   1.0
     */
    public function getAll(): array
    {
        if (empty($this->listeners)) {
            return [];
        }

        krsort($this->listeners);

        return \call_user_func_array('array_merge', $this->listeners);
    }

    /**
     * Get the priority queue.
     *
     * @return  \ArrayIterator
     *
     * @since   1.0
     */
    #[\ReturnTypeWillChange]
    public function getIterator()
    {
        return new \ArrayIterator($this->getAll());
    }

    /**
     * Count the number of listeners in the queue.
     *
     * @return  integer  The number of listeners in the queue.
     *
     * @since   1.0
     */
    #[\ReturnTypeWillChange]
    public function count()
    {
        $count = 0;

        foreach ($this->listeners as $priority) {
            $count += \count($priority);
        }

        return $count;
    }
}