From 3a61dbca1d47bb5bcc0f4785482ff78864e2f4b0 Mon Sep 17 00:00:00 2001
From: Max <max@privatevoid.net>
Date: Sat, 16 Nov 2024 01:19:37 +0100
Subject: [PATCH] cluster/simulacrum: allow limiting nodes

---
 cluster/lib/service-module.nix         | 8 ++++++--
 cluster/simulacrum/default.nix         | 6 ++++--
 cluster/simulacrum/nowhere/default.nix | 4 +++-
 3 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/cluster/lib/service-module.nix b/cluster/lib/service-module.nix
index ce20019..e96dfa9 100644
--- a/cluster/lib/service-module.nix
+++ b/cluster/lib/service-module.nix
@@ -1,9 +1,8 @@
-{ config, lib, name, ... }:
+{ config, lib, ... }:
 with lib;
 
 let
   filterGroup = group: hostName: builtins.filter (x: x != hostName) group;
-  serviceName = name;
 in
 
 {
@@ -65,6 +64,11 @@ in
         type = types.deferredModule;
         default = {};
       };
+      availableNodes = mkOption {
+        description = "Nodes to include in the scenario.";
+        type = with types; nullOr (listOf str);
+        default = null;
+      };
     };
   };
   config.otherNodes = builtins.mapAttrs (const filterGroup) config.nodes;
diff --git a/cluster/simulacrum/default.nix b/cluster/simulacrum/default.nix
index 26eca7e..00290e0 100644
--- a/cluster/simulacrum/default.nix
+++ b/cluster/simulacrum/default.nix
@@ -22,8 +22,10 @@ let
     };
   };
 
-  nodes = lib.attrNames config.gods.fromLight;
-  nodes' = lib.attrNames (config.gods.fromLight // { nowhere = null; });
+  nodesConfigured = config.cluster.config.services.${service}.simulacrum.availableNodes;
+  availableNodes = if nodesConfigured == null then config.gods.fromLight else lib.genAttrs nodesConfigured (_: throw "unreachable");
+  nodes = lib.attrNames availableNodes;
+  nodes' = lib.attrNames (availableNodes // { nowhere = null; });
   digits = lib.attrsets.listToAttrs (lib.zipListsWith lib.nameValuePair nodes' (lib.range 1 255));
   depot' = extendModules {
     modules = [
diff --git a/cluster/simulacrum/nowhere/default.nix b/cluster/simulacrum/nowhere/default.nix
index 4e3aaaf..65ada22 100644
--- a/cluster/simulacrum/nowhere/default.nix
+++ b/cluster/simulacrum/nowhere/default.nix
@@ -52,6 +52,8 @@ let
   genHostCert = hostname: genCert [ "--hostname=${hostname}" ] { CN = hostname; };
 
   getNodeAddr = node: (builtins.head config.nodes.${node}.networking.interfaces.eth1.ipv4.addresses).address;
+
+  filterActiveNodes = lib.filterAttrs (node: _: config.nodes ? node);
 in
 
 {
@@ -79,7 +81,7 @@ in
         address = hour.interfaces.primary.addrPublic;
         prefixLength = 32;
         via = getNodeAddr name;
-      }) depot.gods.fromLight;
+      }) (filterActiveNodes depot.gods.fromLight);
       nameservers = map (name: depot.hours.${name}.interfaces.primary.addrPublic) cluster.config.services.dns.nodes.authoritative;
     };
     services.nginx = {