1 files changed, 108 insertions, 0 deletions
diff --git a/vendor/ipl/scheduler/src/Common/Promises.php b/vendor/ipl/scheduler/src/Common/Promises.php
new file mode 100644
index 0000000..b896627
--- /dev/null
+++ b/vendor/ipl/scheduler/src/Common/Promises.php
@@ -0,0 +1,108 @@
+<?php
+
+namespace ipl\Scheduler\Common;
+
+use ArrayObject;
+use InvalidArgumentException;
+use Ramsey\Uuid\UuidInterface;
+use React\Promise\PromiseInterface;
+use SplObjectStorage;
+
+trait Promises
+{
+    /** @var SplObjectStorage<UuidInterface, ArrayObject<int, PromiseInterface>> */
+    protected $promises;
+
+    /**
+     * Add the given promise for the specified UUID
+     *
+     * **Example Usage:**
+     *
+     * ```php
+     * $promise = work();
+     * $promises->addPromise($uuid, $promise);
+     * ```
+     *
+     * @param UuidInterface $uuid
+     * @param PromiseInterface $promise
+     *
+     * @return $this
+     */
+    protected function addPromise(UuidInterface $uuid, PromiseInterface $promise): self
+    {
+        if (! $this->promises->contains($uuid)) {
+            $this->promises->attach($uuid, new ArrayObject());
+        }
+
+        $this->promises[$uuid][] = $promise;
+
+        return $this;
+    }
+
+    /**
+     * Remove the given promise for the specified UUID
+     *
+     * **Example Usage:**
+     *
+     * ```php
+     * $promise->always(function () use ($uuid, $promise) {
+     *     $promises->removePromise($uuid, $promise);
+     * })
+     * ```
+     *
+     * @param UuidInterface $uuid
+     * @param PromiseInterface $promise
+     *
+     * @return $this
+     *
+     * @throws InvalidArgumentException If the given UUID doesn't have any registered promises or when the specified
+     *                                  UUID promises doesn't contain the provided promise
+     */
+    protected function removePromise(UuidInterface $uuid, PromiseInterface $promise): self
+    {
+        if (! $this->promises->contains($uuid)) {
+            throw new InvalidArgumentException(
+                sprintf('There are no registered promises for UUID %s', $uuid->toString())
+            );
+        }
+
+        foreach ($this->promises[$uuid] as $k => $v) {
+            if ($v === $promise) {
+                unset($this->promises[$uuid][$k]);
+
+                return $this;
+            }
+        }
+
+        throw new InvalidArgumentException(
+            sprintf('There is no such promise for UUID %s', $uuid->toString())
+        );
+    }
+
+    /**
+     * Detach and return promises for the given UUID, if any
+     *
+     * **Example Usage:**
+     *
+     * ```php
+     * foreach ($promises->detachPromises($uuid) as $promise) {
+     *     $promise->cancel();
+     * }
+     * ```
+     *
+     * @param UuidInterface $uuid
+     *
+     * @return PromiseInterface[]
+     */
+    protected function detachPromises(UuidInterface $uuid): array
+    {
+        if (! $this->promises->contains($uuid)) {
+            return [];
+        }
+
+        $promises = $this->promises[$uuid];
+        $this->promises->detach($uuid);
+
+        return $promises->getArrayCopy();
+    }
+}