laravel-websockets/tests/Mocks/PromiseResolver.php

<?php

namespace BlaxSoftware\LaravelWebSockets\Test\Mocks;

use BlaxSoftware\LaravelWebSockets\Helpers;
use Clue\React\Block;
use React\EventLoop\LoopInterface;
use React\Promise\FulfilledPromise;
use React\Promise\PromiseInterface;

class PromiseResolver implements PromiseInterface
{
    /**
     * The promise to resolve.
     *
     * @var \React\Promise\PromiseInterface
     */
    protected $promise;

    /**
     * The loop.
     *
     * @var \React\EventLoop\LoopInterface
     */
    protected $loop;

    /**
     * Initialize the promise resolver.
     *
     * @param  mixed  $promise
     * @param  LoopInterface  $loop
     * @return void
     */
    public function __construct($promise, LoopInterface $loop)
    {
        $this->promise = $promise instanceof PromiseInterface ? $promise : new FulfilledPromise($promise);
        $this->loop = $loop;
    }

    /**
     * Intercept the promise then() and run it in sync.
     *
     * @param  callable|null  $onFulfilled
     * @param  callable|null  $onRejected
     * @param  callable|null  $onProgress
     * @return PromiseInterface
     */
    public function then(callable $onFulfilled = null, callable $onRejected = null, callable $onProgress = null)
    {
        $result = Block\await(
            $this->promise, $this->loop
        );

        if ($onFulfilled !== null) {
            $result = call_user_func($onFulfilled, $result);
        }

        return $result instanceof PromiseInterface
            ? new self($result, $this->loop)
            : new self(Helpers::createFulfilledPromise($result), $this->loop);
    }

    /**
     * Pass the calls to the promise.
     *
     * @param  string  $method
     * @param  array  $args
     * @return mixed
     */
    public function __call($method, $args)
    {
        return call_user_func([$this->promise, $method], $args);
    }
}