file/lib/EioHandle.php

<?php

namespace Amp\File;

use Amp\Deferred;
use Amp\Promise;
use Amp\Success;
use function Amp\call;

class EioHandle implements Handle {
    private $poll;
    private $fh;
    private $path;
    private $mode;
    private $size;
    private $position;
    private $queue;
    private $isActive = false;
    private $writable = true;

    public function __construct(Internal\EioPoll $poll, $fh, string $path, string $mode, int $size) {
        $this->poll = $poll;
        $this->fh = $fh;
        $this->path = $path;
        $this->mode = $mode;
        $this->size = $size;
        $this->position = ($mode[0] === "a") ? $size : 0;

        $this->queue = new \SplQueue;
    }

    /**
     * {@inheritdoc}
     */
    public function read(int $length = self::DEFAULT_READ_LENGTH): Promise {
        if ($this->isActive) {
            throw new PendingOperationError;
        }

        $this->isActive = true;

        $remaining = $this->size - $this->position;
        $length = $length > $remaining ? $remaining : $length;

        $deferred = new Deferred;
        $this->poll->listen();

        \eio_read(
            $this->fh,
            $length,
            $this->position,
            \EIO_PRI_DEFAULT,
            [$this, "onRead"],
            $deferred
        );

        return $deferred->promise();
    }

    private function onRead(Deferred $deferred, $result, $req) {
        $this->isActive = false;
        $this->poll->done();
        if ($result === -1) {
            $deferred->fail(new FilesystemException(
                sprintf('Reading from file failed: %s.', \eio_get_last_error($req))
            ));
        } else {
            $this->position += \strlen($result);
            $deferred->resolve($result);
        }
    }

    /**
     * {@inheritdoc}
     */
    public function write(string $data): Promise {
        if ($this->isActive && $this->queue->isEmpty()) {
            throw new PendingOperationError;
        }

        if (!$this->writable) {
            throw new \Error("The file is no longer writable");
        }

        $this->isActive = true;

        if ($this->queue->isEmpty()) {
            $promise = $this->push($data);
        } else {
            $promise = $this->queue->top();
            $promise = call(function () use ($promise, $data) {
                yield $promise;
                return yield $this->push($data);
            });
        }

        $this->queue->push($promise);

        return $promise;
    }

    private function push(string $data): Promise {
        $length = \strlen($data);
        $deferred = new Deferred;
        $this->poll->listen();

        \eio_write(
            $this->fh,
            $data,
            $length,
            $this->position,
            \EIO_PRI_DEFAULT,
            [$this, "onWrite"],
            $deferred
        );

        return $deferred->promise();
    }

    /**
     * {@inheritdoc}
     */
    public function end(string $data = ""): Promise {
        $promise = $this->write($data);
        $this->writable = false;
        $promise->onResolve([$this, "close"]);
        return $promise;
    }

    private function onWrite(Deferred $deferred, $result, $req) {
        $this->poll->done();

        if ($this->queue->isEmpty()) {
            $deferred->fail(new FilesystemException('No pending write, the file may have been closed'));
        }

        $this->queue->shift();
        if ($this->queue->isEmpty()) {
            $this->isActive = false;
        }

        if ($result === -1) {
            $deferred->fail(new FilesystemException(
                sprintf('Writing to the file failed: %s', \eio_get_last_error($req))
            ));
        } else {
            $this->position += $result;
            if ($this->position > $this->size) {
                $this->size = $this->position;
            }

            $deferred->resolve($result);
        }
    }

    /**
     * {@inheritdoc}
     */
    public function close(): Promise {
        $this->poll->listen();
        $deferred = new Deferred;
        \eio_close($this->fh, \EIO_PRI_DEFAULT, [$this, "onClose"], $deferred);

        return $deferred->promise();
    }

    private function onClose($deferred, $result, $req) {
        $this->poll->done();
        if ($result === -1) {
            $deferred->fail(new FilesystemException(
                \eio_get_last_error($req)
            ));
        } else {
            $deferred->resolve();
        }
    }

    /**
     * {@inheritdoc}
     */
    public function seek(int $offset, int $whence = \SEEK_SET): Promise {
        if ($this->isActive) {
            throw new PendingOperationError;
        }

        $offset = (int) $offset;
        switch ($whence) {
            case \SEEK_SET:
                $this->position = $offset;
                break;
            case \SEEK_CUR:
                $this->position = $this->position + $offset;
                break;
            case \SEEK_END:
                $this->position = $this->size + $offset;
                break;
            default:
                throw new \Error(
                    "Invalid whence parameter; SEEK_SET, SEEK_CUR or SEEK_END expected"
                );
        }

        return new Success($this->position);
    }

    /**
     * {@inheritdoc}
     */
    public function tell(): int {
        return $this->position;
    }

    /**
     * {@inheritdoc}
     */
    public function eof(): bool {
        return !$this->queue->isEmpty() ? false : ($this->size <= $this->position);
    }

    /**
     * {@inheritdoc}
     */
    public function path(): string {
        return $this->path;
    }

    /**
     * {@inheritdoc}
     */
    public function mode(): string {
        return $this->mode;
    }
}