about summary refs log tree commit diff
path: root/src/libstore/machines.cc
blob: 076c3cab3e90cd344458fece3f32f66f3cd9bb6e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#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 = [&](size_t 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;
}

}