mirror of
https://github.com/NixOS/nix
synced 2025-06-30 15:48:00 +02:00
Factor out MemorySourceAccessor
, implement missing features
The new `MemorySourceAccessor` rather than being a slightly lossy flat map is a complete in-memory model of file system objects. Co-authored-by: Eelco Dolstra <edolstra@gmail.com>
This commit is contained in:
parent
8e222fbb12
commit
9b880e3e29
4 changed files with 205 additions and 39 deletions
124
src/libutil/memory-source-accessor.cc
Normal file
124
src/libutil/memory-source-accessor.cc
Normal file
|
@ -0,0 +1,124 @@
|
|||
#include "memory-source-accessor.hh"
|
||||
|
||||
namespace nix {
|
||||
|
||||
MemorySourceAccessor::File *
|
||||
MemorySourceAccessor::open(const CanonPath & path, std::optional<File> create)
|
||||
{
|
||||
File * cur = &root;
|
||||
|
||||
bool newF = false;
|
||||
|
||||
for (std::string_view name : path)
|
||||
{
|
||||
auto * curDirP = std::get_if<File::Directory>(&cur->raw);
|
||||
if (!curDirP)
|
||||
return nullptr;
|
||||
auto & curDir = *curDirP;
|
||||
|
||||
auto i = curDir.contents.find(name);
|
||||
if (i == curDir.contents.end()) {
|
||||
if (!create)
|
||||
return nullptr;
|
||||
else {
|
||||
newF = true;
|
||||
i = curDir.contents.insert(i, {
|
||||
std::string { name },
|
||||
File::Directory {},
|
||||
});
|
||||
}
|
||||
}
|
||||
cur = &i->second;
|
||||
}
|
||||
|
||||
if (newF && create) *cur = std::move(*create);
|
||||
|
||||
return cur;
|
||||
}
|
||||
|
||||
std::string MemorySourceAccessor::readFile(const CanonPath & path)
|
||||
{
|
||||
auto * f = open(path, std::nullopt);
|
||||
if (!f)
|
||||
throw Error("file '%s' does not exist", path);
|
||||
if (auto * r = std::get_if<File::Regular>(&f->raw))
|
||||
return r->contents;
|
||||
else
|
||||
throw Error("file '%s' is not a regular file", path);
|
||||
}
|
||||
|
||||
bool MemorySourceAccessor::pathExists(const CanonPath & path)
|
||||
{
|
||||
return open(path, std::nullopt);
|
||||
}
|
||||
|
||||
MemorySourceAccessor::Stat MemorySourceAccessor::File::lstat() const
|
||||
{
|
||||
return std::visit(overloaded {
|
||||
[](const Regular & r) {
|
||||
return Stat {
|
||||
.type = tRegular,
|
||||
.fileSize = r.contents.size(),
|
||||
.isExecutable = r.executable,
|
||||
};
|
||||
},
|
||||
[](const Directory &) {
|
||||
return Stat {
|
||||
.type = tDirectory,
|
||||
};
|
||||
},
|
||||
[](const Symlink &) {
|
||||
return Stat {
|
||||
.type = tSymlink,
|
||||
};
|
||||
},
|
||||
}, this->raw);
|
||||
}
|
||||
|
||||
std::optional<MemorySourceAccessor::Stat>
|
||||
MemorySourceAccessor::maybeLstat(const CanonPath & path)
|
||||
{
|
||||
const auto * f = open(path, std::nullopt);
|
||||
return f ? std::optional { f->lstat() } : std::nullopt;
|
||||
}
|
||||
|
||||
MemorySourceAccessor::DirEntries MemorySourceAccessor::readDirectory(const CanonPath & path)
|
||||
{
|
||||
auto * f = open(path, std::nullopt);
|
||||
if (!f)
|
||||
throw Error("file '%s' does not exist", path);
|
||||
if (auto * d = std::get_if<File::Directory>(&f->raw)) {
|
||||
DirEntries res;
|
||||
for (auto & [name, file] : d->contents)
|
||||
res.insert_or_assign(name, file.lstat().type);
|
||||
return res;
|
||||
} else
|
||||
throw Error("file '%s' is not a directory", path);
|
||||
return {};
|
||||
}
|
||||
|
||||
std::string MemorySourceAccessor::readLink(const CanonPath & path)
|
||||
{
|
||||
auto * f = open(path, std::nullopt);
|
||||
if (!f)
|
||||
throw Error("file '%s' does not exist", path);
|
||||
if (auto * s = std::get_if<File::Symlink>(&f->raw))
|
||||
return s->target;
|
||||
else
|
||||
throw Error("file '%s' is not a symbolic link", path);
|
||||
}
|
||||
|
||||
CanonPath MemorySourceAccessor::addFile(CanonPath path, std::string && contents)
|
||||
{
|
||||
auto * f = open(path, File { File::Regular {} });
|
||||
if (!f)
|
||||
throw Error("file '%s' cannot be made because some parent file is not a directory", path);
|
||||
if (auto * r = std::get_if<File::Regular>(&f->raw))
|
||||
r->contents = std::move(contents);
|
||||
else
|
||||
throw Error("file '%s' is not a regular file", path);
|
||||
|
||||
return path;
|
||||
}
|
||||
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue