#include "machines.hh"
#include <algorithm>
#include "globals.hh"
#include "util.hh"

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") || hasPrefix(storeUri, "/")
              ? 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;")) {
    trim(line);
    line.erase(std::find(line.begin(), line.end(), '#'), line.end());
    if (line.empty()) continue;

    if (line[0] == '@') {
      auto file = trim(std::string(line, 1));
      try {
        parseMachines(readFile(file), machines);
      } catch (const SysError& e) {
        if (e.errNo != ENOENT) throw;
        debug("cannot find machines file '%s'", file);
      }
      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() {
  static auto machines = [&]() {
    Machines machines;
    parseMachines(settings.builders, machines);
    return machines;
  }();
  return machines;
}

}  // namespace nix