mirror of
https://github.com/privatevoid-net/nix-super.git
synced 2024-09-21 08:58:04 +03:00
92 lines
2.9 KiB
C++
92 lines
2.9 KiB
C++
#include "machines.hh"
|
||
#include "util.hh"
|
||
#include "globals.hh"
|
||
|
||
#include <algorithm>
|
||
|
||
namespace nix {
|
||
|
||
Machine::Machine(decltype(storeUri) storeUri,
|
||
decltype(systemTypes) systemTypes,
|
||
decltype(sshKey) sshKey,
|
||
decltype(maxJobs) maxJobs,
|
||
decltype(speedFactor) speedFactor,
|
||
decltype(supportedFeatures) supportedFeatures,
|
||
decltype(mandatoryFeatures) mandatoryFeatures,
|
||
decltype(sshPublicHostKey) sshPublicHostKey) :
|
||
storeUri(
|
||
// Backwards compatibility: if the URI is a hostname,
|
||
// prepend ssh://.
|
||
storeUri.find("://") != std::string::npos || hasPrefix(storeUri, "local") || hasPrefix(storeUri, "remote") || hasPrefix(storeUri, "auto")
|
||
? storeUri
|
||
: "ssh://" + storeUri),
|
||
systemTypes(systemTypes),
|
||
sshKey(sshKey),
|
||
maxJobs(maxJobs),
|
||
speedFactor(std::max(1U, speedFactor)),
|
||
supportedFeatures(supportedFeatures),
|
||
mandatoryFeatures(mandatoryFeatures),
|
||
sshPublicHostKey(sshPublicHostKey)
|
||
{}
|
||
|
||
bool Machine::allSupported(const std::set<string> & features) const {
|
||
return std::all_of(features.begin(), features.end(),
|
||
[&](const string & feature) {
|
||
return supportedFeatures.count(feature) ||
|
||
mandatoryFeatures.count(feature);
|
||
});
|
||
}
|
||
|
||
bool Machine::mandatoryMet(const std::set<string> & features) const {
|
||
return std::all_of(mandatoryFeatures.begin(), mandatoryFeatures.end(),
|
||
[&](const string & feature) {
|
||
return features.count(feature);
|
||
});
|
||
}
|
||
|
||
void parseMachines(const std::string & s, Machines & machines)
|
||
{
|
||
for (auto line : tokenizeString<std::vector<string>>(s, "\n;")) {
|
||
chomp(line);
|
||
line.erase(std::find(line.begin(), line.end(), '#'), line.end());
|
||
if (line.empty()) continue;
|
||
auto tokens = tokenizeString<std::vector<string>>(line);
|
||
auto sz = tokens.size();
|
||
if (sz < 1)
|
||
throw FormatError("bad machine specification ‘%s’", line);
|
||
|
||
auto isSet = [&](int n) {
|
||
return tokens.size() > n && tokens[n] != "" && tokens[n] != "-";
|
||
};
|
||
|
||
machines.emplace_back(tokens[0],
|
||
isSet(1) ? tokenizeString<std::vector<string>>(tokens[1], ",") : std::vector<string>{settings.thisSystem},
|
||
isSet(2) ? tokens[2] : "",
|
||
isSet(3) ? std::stoull(tokens[3]) : 1LL,
|
||
isSet(4) ? std::stoull(tokens[4]) : 1LL,
|
||
isSet(5) ? tokenizeString<std::set<string>>(tokens[5], ",") : std::set<string>{},
|
||
isSet(6) ? tokenizeString<std::set<string>>(tokens[6], ",") : std::set<string>{},
|
||
isSet(7) ? tokens[7] : "");
|
||
}
|
||
}
|
||
|
||
Machines getMachines()
|
||
{
|
||
Machines machines;
|
||
|
||
for (auto & file : settings.builderFiles.get()) {
|
||
try {
|
||
parseMachines(readFile(file), machines);
|
||
} catch (const SysError & e) {
|
||
if (e.errNo != ENOENT)
|
||
throw;
|
||
}
|
||
}
|
||
|
||
parseMachines(settings.builders, machines);
|
||
|
||
return machines;
|
||
}
|
||
|
||
}
|