mirror of
https://github.com/danog/file.git
synced 2024-11-27 04:14:50 +01:00
525 lines
16 KiB
PHP
525 lines
16 KiB
PHP
<?php
|
|
|
|
namespace Amp\File;
|
|
|
|
use Amp\Success;
|
|
use Amp\Failure;
|
|
use Amp\Deferred;
|
|
use Amp\Coroutine;
|
|
|
|
class UvDriver implements Driver {
|
|
private $driver;
|
|
private $loop;
|
|
private $busy;
|
|
|
|
/**
|
|
* @param \Interop\Async\Loop\Driver $driver
|
|
*/
|
|
public function __construct(\Interop\Async\Loop\Driver $driver) {
|
|
$loop = $driver->getHandle();
|
|
if (!is_resource($loop) || get_resource_type($loop) != "uv_loop") {
|
|
throw new \InvalidArgumentException("Expected a driver whose underlying loop is an uv_loop");
|
|
}
|
|
|
|
$this->driver = $driver;
|
|
$this->loop = $loop;
|
|
|
|
// dummy handle to be able to tell the loop that there is work being done and it shouldn't abort if there are no other watchers at a given moment
|
|
$this->busy = $driver->repeat(PHP_INT_MAX, function(){ });
|
|
$driver->unreference($this->busy);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function open($path, $mode) {
|
|
switch ($mode) {
|
|
case "r": $flags = \UV::O_RDONLY; break;
|
|
case "r+": $flags = \UV::O_RDWR; break;
|
|
case "w": $flags = \UV::O_WRONLY | \UV::O_CREAT; break;
|
|
case "w+": $flags = \UV::O_RDWR | \UV::O_CREAT; break;
|
|
case "a": $flags = \UV::O_WRONLY | \UV::O_CREAT | \UV::O_APPEND; break;
|
|
case "a+": $flags = \UV::O_RDWR | \UV::O_CREAT | \UV::O_APPEND; break;
|
|
case "x": $flags = \UV::O_WRONLY | \UV::O_CREAT | \UV::O_EXCL; break;
|
|
case "x+": $flags = \UV::O_RDWR | \UV::O_CREAT | \UV::O_EXCL; break;
|
|
case "c": $flags = \UV::O_WRONLY | \UV::O_CREAT; break;
|
|
case "c+": $flags = \UV::O_RDWR | \UV::O_CREAT; break;
|
|
default: return new Failure(new FilesystemException(
|
|
"Invalid open mode"
|
|
));
|
|
}
|
|
$chmod = ($flags & \UV::O_CREAT) ? 0644 : 0;
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
$openArr = [$mode, $path, $deferred];
|
|
\uv_fs_open($this->loop, $path, $flags, $chmod, function($fh) use ($openArr) {
|
|
if ($fh) {
|
|
$this->onOpenHandle($fh, $openArr);
|
|
} else {
|
|
$this->driver->unreference($this->busy);
|
|
list( , $path, $deferred) = $openArr;
|
|
$deferred->fail(new FilesystemException(
|
|
"Failed opening file handle to $path"
|
|
));
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
private function onOpenHandle($fh, array $openArr) {
|
|
list($mode) = $openArr;
|
|
if ($mode[0] === "w") {
|
|
\uv_fs_ftruncate($this->loop, $fh, $length = 0, function($fh) use ($openArr) {
|
|
$this->driver->unreference($this->busy);
|
|
if ($fh) {
|
|
$this->finalizeHandle($fh, $size = 0, $openArr);
|
|
} else {
|
|
list( , $path, $deferred) = $openArr;
|
|
$deferred->fail(new FilesystemException(
|
|
"Failed truncating file $path"
|
|
));
|
|
}
|
|
});
|
|
} else {
|
|
\uv_fs_fstat($this->loop, $fh, function($fh, $stat) use ($openArr) {
|
|
$this->driver->unreference($this->busy);
|
|
if ($fh) {
|
|
StatCache::set($openArr[1], $stat);
|
|
$this->finalizeHandle($fh, $stat["size"], $openArr);
|
|
} else {
|
|
list( , $path, $deferred) = $openArr;
|
|
$deferred->fail(new FilesystemException(
|
|
"Failed reading file size from open handle pointing to $path"
|
|
));
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
private function finalizeHandle($fh, $size, array $openArr) {
|
|
list($mode, $path, $deferred) = $openArr;
|
|
$handle = new UvHandle($this->loop, $this->busy, $fh, $path, $mode, $size);
|
|
$deferred->resolve($handle);
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function stat($path) {
|
|
if ($stat = StatCache::get($path)) {
|
|
return new Success($stat);
|
|
}
|
|
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_stat($this->loop, $path, function($fh, $stat) use ($deferred, $path) {
|
|
if (empty($fh)) {
|
|
$stat = null;
|
|
} else {
|
|
StatCache::set($path, $stat);
|
|
}
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve($stat);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function exists($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
$deferred->resolve((bool) $result);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function isdir($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if ($result) {
|
|
$deferred->resolve(!($result["mode"] & \UV::S_IFREG));
|
|
} else {
|
|
$deferred->resolve(false);
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function isfile($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if ($result) {
|
|
$deferred->resolve((bool) ($result["mode"] & \UV::S_IFREG));
|
|
} else {
|
|
$deferred->resolve(false);
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function size($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if (empty($result)) {
|
|
$deferred->fail(new FilesystemException(
|
|
"Specified path does not exist"
|
|
));
|
|
} elseif (($result["mode"] & \UV::S_IFREG)) {
|
|
$deferred->resolve($result["size"]);
|
|
} else {
|
|
$deferred->fail(new FilesystemException(
|
|
"Specified path is not a regular file"
|
|
));
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function mtime($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if ($result) {
|
|
$deferred->resolve($result["mtime"]);
|
|
} else {
|
|
$deferred->fail(new FilesystemException(
|
|
"Specified path does not exist"
|
|
));
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function atime($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if ($result) {
|
|
$deferred->resolve($result["atime"]);
|
|
} else {
|
|
$deferred->fail(new FilesystemException(
|
|
"Specified path does not exist"
|
|
));
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function ctime($path) {
|
|
$deferred = new Deferred;
|
|
$this->stat($path)->when(function ($error, $result) use ($deferred) {
|
|
if ($result) {
|
|
$deferred->resolve($result["ctime"]);
|
|
} else {
|
|
$deferred->fail(new FilesystemException(
|
|
"Specified path does not exist"
|
|
));
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function lstat($path) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_lstat($this->loop, $path, function($fh, $stat) use ($deferred) {
|
|
if (empty($fh)) {
|
|
$stat = null;
|
|
}
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve($stat);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function symlink($target, $link) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
uv_fs_symlink($this->loop, $target, $link, \UV::S_IRWXU | \UV::S_IRUSR, function($fh) use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function rename($from, $to) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_rename($this->loop, $from, $to, function($fh) use ($deferred, $from) {
|
|
$this->driver->unreference($this->busy);
|
|
StatCache::clear($from);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function unlink($path) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_unlink($this->loop, $path, function($fh) use ($deferred, $path) {
|
|
$this->driver->unreference($this->busy);
|
|
StatCache::clear($path);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function mkdir($path, $mode = 0644) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_mkdir($this->loop, $path, $mode, function($fh) use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function rmdir($path) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_rmdir($this->loop, $path, function($fh) use ($deferred, $path) {
|
|
$this->driver->unreference($this->busy);
|
|
StatCache::clear($path);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function scandir($path) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
uv_fs_readdir($this->loop, $path, 0, function($fh, $data) use ($deferred, $path) {
|
|
$this->driver->unreference($this->busy);
|
|
if (empty($fh)) {
|
|
$deferred->fail(new FilesystemException(
|
|
"Failed reading contents from {$path}"
|
|
));
|
|
} else {
|
|
$deferred->resolve($data);
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function chmod($path, $mode) {
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_chmod($this->loop, $path, $mode, function($fh) use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function chown($path, $uid, $gid) {
|
|
// @TODO Return a failure in windows environments
|
|
$this->driver->reference($this->busy);
|
|
$deferred = new Deferred;
|
|
\uv_fs_chown($this->loop, $path, $uid, $gid, function($fh) use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve((bool)$fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function touch($path) {
|
|
$this->driver->reference($this->busy);
|
|
$atime = $mtime = time();
|
|
$deferred = new Deferred;
|
|
\uv_fs_utime($this->loop, $path, $mtime, $atime, function() use ($deferred) {
|
|
// The uv_fs_utime() callback does not receive any args at this time
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve(true);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function get($path) {
|
|
return \Amp\resolve($this->doGet($path), $this->driver);
|
|
}
|
|
|
|
private function doGet($path): \Generator {
|
|
$this->driver->reference($this->busy);
|
|
$awaitable = $this->doFsOpen($path, $flags = \UV::O_RDONLY, $mode = 0);
|
|
if (!$fh = yield $awaitable) {
|
|
$this->driver->unreference($this->busy);
|
|
throw new FilesystemException(
|
|
"Failed opening file handle: {$path}"
|
|
);
|
|
}
|
|
|
|
$deferred = new Deferred;
|
|
$stat = (yield $this->doFsStat($fh));
|
|
if (empty($stat)) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->fail(new FilesystemException(
|
|
"stat operation failed on open file handle"
|
|
));
|
|
} elseif (!$stat["isfile"]) {
|
|
\uv_fs_close($this->loop, $fh, function() use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->fail(new FilesystemException(
|
|
"cannot buffer contents: path is not a file"
|
|
));
|
|
});
|
|
} else {
|
|
$buffer = (yield $this->doFsRead($fh, $offset = 0, $stat["size"]));
|
|
if ($buffer === false ) {
|
|
\uv_fs_close($this->loop, $fh, function() use ($deferred) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->fail(new FilesystemException(
|
|
"read operation failed on open file handle"
|
|
));
|
|
});
|
|
} else {
|
|
\uv_fs_close($this->loop, $fh, function() use ($deferred, $buffer) {
|
|
$this->driver->unreference($this->busy);
|
|
$deferred->resolve($buffer);
|
|
});
|
|
}
|
|
}
|
|
|
|
yield Coroutine::result(yield $deferred->getAwaitable());
|
|
}
|
|
|
|
private function doFsOpen($path, $flags, $mode) {
|
|
$deferred = new Deferred;
|
|
\uv_fs_open($this->loop, $path, $flags, $mode, function($fh) use ($deferred, $path) {
|
|
$deferred->resolve($fh);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
private function doFsStat($fh) {
|
|
$deferred = new Deferred;
|
|
\uv_fs_fstat($this->loop, $fh, function($fh, $stat) use ($deferred) {
|
|
if ($fh) {
|
|
$stat["isdir"] = (bool) ($stat["mode"] & \UV::S_IFDIR);
|
|
$stat["isfile"] = !$stat["isdir"];
|
|
$deferred->resolve($stat);
|
|
} else {
|
|
$deferred->resolve();
|
|
}
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
private function doFsRead($fh, $offset, $len) {
|
|
$deferred = new Deferred;
|
|
\uv_fs_read($this->loop, $fh, $offset, $len, function($fh, $nread, $buffer) use ($deferred) {
|
|
$deferred->resolve($nread < 0 ? false : $buffer);
|
|
});
|
|
|
|
return $deferred->getAwaitable();
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function put($path, $contents) {
|
|
return \Amp\resolve($this->doPut($path, $contents), $this->driver);
|
|
}
|
|
|
|
private function doPut($path, $contents): \Generator {
|
|
$flags = \UV::O_WRONLY | \UV::O_CREAT;
|
|
$mode = \UV::S_IRWXU | \UV::S_IRUSR;
|
|
$this->driver->reference($this->busy);
|
|
$awaitable = $this->doFsOpen($path, $flags, $mode);
|
|
if (!$fh = yield $awaitable) {
|
|
$this->driver->unreference($this->busy);
|
|
throw new FilesystemException(
|
|
"Failed opening write file handle"
|
|
);
|
|
}
|
|
|
|
$deferred = new Deferred;
|
|
$len = strlen($contents);
|
|
\uv_fs_write($this->loop, $fh, $contents, $offset = 0, function($fh, $result) use ($deferred, $len) {
|
|
\uv_fs_close($this->loop, $fh, function() use ($deferred, $result, $len) {
|
|
$this->driver->unreference($this->busy);
|
|
if ($result < 0) {
|
|
$deferred->fail(new FilesystemException(
|
|
\uv_strerror($result)
|
|
));
|
|
} else {
|
|
$deferred->resolve($len);
|
|
}
|
|
});
|
|
});
|
|
|
|
yield Coroutine::result(yield $deferred->getAwaitable());
|
|
}
|
|
}
|