128 lines
3.6 KiB
C++
128 lines
3.6 KiB
C++
#include "path-with-outputs.hh"
|
|
#include "store-api.hh"
|
|
#include "nlohmann/json.hpp"
|
|
|
|
#include <regex>
|
|
|
|
namespace nix {
|
|
|
|
std::string StorePathWithOutputs::to_string(const Store & store) const
|
|
{
|
|
return outputs.empty()
|
|
? store.printStorePath(path)
|
|
: store.printStorePath(path) + "!" + concatStringsSep(",", outputs);
|
|
}
|
|
|
|
|
|
DerivedPath StorePathWithOutputs::toDerivedPath() const
|
|
{
|
|
if (!outputs.empty() || path.isDerivation())
|
|
return DerivedPath::Built { path, outputs };
|
|
else
|
|
return DerivedPath::Opaque { path };
|
|
}
|
|
|
|
|
|
std::vector<DerivedPath> toDerivedPaths(const std::vector<StorePathWithOutputs> ss)
|
|
{
|
|
std::vector<DerivedPath> reqs;
|
|
for (auto & s : ss) reqs.push_back(s.toDerivedPath());
|
|
return reqs;
|
|
}
|
|
|
|
|
|
std::variant<StorePathWithOutputs, StorePath> StorePathWithOutputs::tryFromDerivedPath(const DerivedPath & p)
|
|
{
|
|
return std::visit(overloaded {
|
|
[&](const DerivedPath::Opaque & bo) -> std::variant<StorePathWithOutputs, StorePath> {
|
|
if (bo.path.isDerivation()) {
|
|
// drv path gets interpreted as "build", not "get drv file itself"
|
|
return bo.path;
|
|
}
|
|
return StorePathWithOutputs { bo.path };
|
|
},
|
|
[&](const DerivedPath::Built & bfd) -> std::variant<StorePathWithOutputs, StorePath> {
|
|
return StorePathWithOutputs { bfd.drvPath, bfd.outputs };
|
|
},
|
|
}, p.raw());
|
|
}
|
|
|
|
|
|
std::pair<std::string_view, StringSet> parsePathWithOutputs(std::string_view s)
|
|
{
|
|
size_t n = s.find("!");
|
|
return n == s.npos
|
|
? std::make_pair(s, std::set<std::string>())
|
|
: std::make_pair(((std::string_view) s).substr(0, n),
|
|
tokenizeString<std::set<std::string>>(((std::string_view) s).substr(n + 1), ","));
|
|
}
|
|
|
|
|
|
StorePathWithOutputs parsePathWithOutputs(const Store & store, std::string_view pathWithOutputs)
|
|
{
|
|
auto [path, outputs] = parsePathWithOutputs(pathWithOutputs);
|
|
return StorePathWithOutputs { store.parseStorePath(path), std::move(outputs) };
|
|
}
|
|
|
|
|
|
StorePathWithOutputs followLinksToStorePathWithOutputs(const Store & store, std::string_view pathWithOutputs)
|
|
{
|
|
auto [path, outputs] = parsePathWithOutputs(pathWithOutputs);
|
|
return StorePathWithOutputs { store.followLinksToStorePath(path), std::move(outputs) };
|
|
}
|
|
|
|
std::pair<std::string, OutputsSpec> parseOutputsSpec(const std::string & s)
|
|
{
|
|
static std::regex regex(R"((.*)\^((\*)|([a-z]+(,[a-z]+)*)))");
|
|
|
|
std::smatch match;
|
|
if (!std::regex_match(s, match, regex))
|
|
return {s, DefaultOutputs()};
|
|
|
|
if (match[3].matched)
|
|
return {match[1], AllOutputs()};
|
|
|
|
return {match[1], tokenizeString<OutputNames>(match[4].str(), ",")};
|
|
}
|
|
|
|
std::string printOutputsSpec(const OutputsSpec & outputsSpec)
|
|
{
|
|
if (std::get_if<DefaultOutputs>(&outputsSpec))
|
|
return "";
|
|
|
|
if (std::get_if<AllOutputs>(&outputsSpec))
|
|
return "^*";
|
|
|
|
if (auto outputNames = std::get_if<OutputNames>(&outputsSpec))
|
|
return "^" + concatStringsSep(",", *outputNames);
|
|
|
|
assert(false);
|
|
}
|
|
|
|
void to_json(nlohmann::json & json, const OutputsSpec & outputsSpec)
|
|
{
|
|
if (std::get_if<DefaultOutputs>(&outputsSpec))
|
|
json = nullptr;
|
|
|
|
else if (std::get_if<AllOutputs>(&outputsSpec))
|
|
json = std::vector<std::string>({"*"});
|
|
|
|
else if (auto outputNames = std::get_if<OutputNames>(&outputsSpec))
|
|
json = *outputNames;
|
|
}
|
|
|
|
void from_json(const nlohmann::json & json, OutputsSpec & outputsSpec)
|
|
{
|
|
if (json.is_null())
|
|
outputsSpec = DefaultOutputs();
|
|
else {
|
|
auto names = json.get<OutputNames>();
|
|
if (names == OutputNames({"*"}))
|
|
outputsSpec = AllOutputs();
|
|
else
|
|
outputsSpec = names;
|
|
}
|
|
}
|
|
|
|
}
|