95 lines
3.2 KiB
PHP
95 lines
3.2 KiB
PHP
<?php
|
|
|
|
namespace WP_Ultimo\Dependencies\Amp\Process;
|
|
|
|
use WP_Ultimo\Dependencies\Amp\ByteStream\ClosedException;
|
|
use WP_Ultimo\Dependencies\Amp\ByteStream\OutputStream;
|
|
use WP_Ultimo\Dependencies\Amp\ByteStream\ResourceOutputStream;
|
|
use WP_Ultimo\Dependencies\Amp\ByteStream\StreamException;
|
|
use WP_Ultimo\Dependencies\Amp\Deferred;
|
|
use WP_Ultimo\Dependencies\Amp\Failure;
|
|
use WP_Ultimo\Dependencies\Amp\Promise;
|
|
final class ProcessOutputStream implements OutputStream
|
|
{
|
|
/** @var \SplQueue */
|
|
private $queuedWrites;
|
|
/** @var bool */
|
|
private $shouldClose = \false;
|
|
/** @var ResourceOutputStream */
|
|
private $resourceStream;
|
|
/** @var StreamException|null */
|
|
private $error;
|
|
public function __construct(Promise $resourceStreamPromise)
|
|
{
|
|
$this->queuedWrites = new \SplQueue();
|
|
$resourceStreamPromise->onResolve(function ($error, $resourceStream) {
|
|
if ($error) {
|
|
$this->error = new StreamException("Failed to launch process", 0, $error);
|
|
while (!$this->queuedWrites->isEmpty()) {
|
|
list(, $deferred) = $this->queuedWrites->shift();
|
|
$deferred->fail($this->error);
|
|
}
|
|
return;
|
|
}
|
|
while (!$this->queuedWrites->isEmpty()) {
|
|
/**
|
|
* @var string $data
|
|
* @var \Amp\Deferred $deferred
|
|
*/
|
|
list($data, $deferred) = $this->queuedWrites->shift();
|
|
$deferred->resolve($resourceStream->write($data));
|
|
}
|
|
$this->resourceStream = $resourceStream;
|
|
if ($this->shouldClose) {
|
|
$this->resourceStream->close();
|
|
}
|
|
});
|
|
}
|
|
/** @inheritdoc */
|
|
public function write(string $data) : Promise
|
|
{
|
|
if ($this->resourceStream) {
|
|
return $this->resourceStream->write($data);
|
|
}
|
|
if ($this->error) {
|
|
return new Failure($this->error);
|
|
}
|
|
if ($this->shouldClose) {
|
|
throw new ClosedException("Stream has already been closed.");
|
|
}
|
|
$deferred = new Deferred();
|
|
$this->queuedWrites->push([$data, $deferred]);
|
|
return $deferred->promise();
|
|
}
|
|
/** @inheritdoc */
|
|
public function end(string $finalData = "") : Promise
|
|
{
|
|
if ($this->resourceStream) {
|
|
return $this->resourceStream->end($finalData);
|
|
}
|
|
if ($this->error) {
|
|
return new Failure($this->error);
|
|
}
|
|
if ($this->shouldClose) {
|
|
throw new ClosedException("Stream has already been closed.");
|
|
}
|
|
$deferred = new Deferred();
|
|
$this->queuedWrites->push([$finalData, $deferred]);
|
|
$this->shouldClose = \true;
|
|
return $deferred->promise();
|
|
}
|
|
public function close()
|
|
{
|
|
$this->shouldClose = \true;
|
|
if ($this->resourceStream) {
|
|
$this->resourceStream->close();
|
|
} elseif (!$this->queuedWrites->isEmpty()) {
|
|
$error = new ClosedException("Stream closed.");
|
|
do {
|
|
list(, $deferred) = $this->queuedWrites->shift();
|
|
$deferred->fail($error);
|
|
} while (!$this->queuedWrites->isEmpty());
|
|
}
|
|
}
|
|
}
|