infrastructure/modules/system/proxmox/network.nix

{config, lib, inputs, ...}: let
  inherit (inputs.self.lib.lib) unmerged eui64 toHexStringLower mkAlmostOptionDefault mapAlmostOptionDefaults;
  inherit (lib.options) mkOption mkEnableOption;
  inherit (lib.modules) mkIf mkMerge mkDefault mkOptionDefault;
  inherit (lib.attrsets) attrValues;
  inherit (lib.lists) elem findSingle findFirst;
  inherit (lib.strings) hasPrefix removePrefix replaceStrings removeSuffix;
  inherit (lib.trivial) mapNullable;
  cfg = config.proxmox.network;
  internalOffset = 32;
  networkInterfaceModule = { config, name, system, ... }: {
    options = with lib.types; {
      enable = mkEnableOption "network interface" // {
        default = true;
      };
      bridge = mkOption {
        type = str;
        default = "vmbr0";
      };
      id = mkOption {
        type = str;
        default = name;
      };
      name = mkOption {
        type = str;
      };
      macAddress = mkOption {
        type = nullOr str;
        default = null;
      };
      address4 = mkOption {
        type = nullOr (either (enum [ "dhcp" ]) str);
        default = null;
      };
      gateway4 = mkOption {
        type = nullOr str;
        default = null;
      };
      address6 = mkOption {
        type = nullOr (either (enum [ "auto" "dhcp" ]) str);
        default = null;
      };
      gateway6 = mkOption {
        type = nullOr str;
        default = null;
      };
      firewall.enable = mkEnableOption "firewall";
      vm.model = mkOption {
        type = enum [ "virtio" "e1000" "rtl8139" "vmxnet3" ];
        default = "virtio";
      };
      mdns = {
        enable = mkEnableOption "mDNS" // {
          default = config.local.enable && config.id == "net0";
        };
      };
      slaac = {
        postfix = mkOption {
          type = nullOr str;
        };
      };
      internal = {
        enable = mkEnableOption "internal network interface";
      };
      local = {
        enable = mkOption {
          type = bool;
          default = system.proxmox.node.name == "reisen" && config.id == "net0" && config.bridge == "vmbr0";
        };
        address4 = mkOption {
          type = nullOr str;
        };
        address6 = mkOption {
          type = nullOr str;
        };
      };
      networkd = {
        enable = mkEnableOption "systemd.network" // {
          default = true;
        };
        name = mkOption {
          type = str;
          default = config.name;
          description = "network unit name";
        };
        networkSettings = mkOption {
          type = unmerged.types.attrs;
        };
      };
    };
    config = let
      hasAddr4 = ! elem config.address4 [ null "dhcp" ];
      hasAddr6 = ! elem config.address6 [ null "dhcp" "auto" ];
      conf = {
        local = mkIf config.local.enable {
          address4 = mkOptionDefault (if hasAddr4 then config.address4 else null);
          address6 = mkOptionDefault (
            if config.address6 == "auto" && config.slaac.postfix != null then "fd0a::${config.slaac.postfix}"
            else if hasAddr6 then config.address6
            else null
          );
        };
        name = mkMerge [
          (mkIf (hasPrefix "net" config.id && system.proxmox.container.enable) (mkOptionDefault ("eth" + removePrefix "net" config.id)))
          # VMs have names like `ens18` for net0...
        ];
        slaac.postfix = mkOptionDefault (mapNullable eui64 config.macAddress);
        gateway4 = mkMerge [
          (mkIf (system.proxmox.node.name == "reisen" && config.bridge == "vmbr0" && config.address4 != null && config.address4 != "dhcp") (mkAlmostOptionDefault "10.1.1.1"))
        ];
        networkd.name = mkIf config.local.enable (
          mkDefault "_00-local"
        );
        networkd.networkSettings = {
          name = mkAlmostOptionDefault config.name;
          ipv6AcceptRAConfig = mkIf config.local.enable {
            UseDNS = mkOptionDefault false;
            DHCPv6Client = mkOptionDefault false;
          };
          matchConfig = {
            MACAddress = mkIf (config.macAddress != null) (mkOptionDefault config.macAddress);
            Type = mkOptionDefault "ether";
          };
          linkConfig = mkMerge [
            (mkIf config.mdns.enable { Multicast = mkOptionDefault true; })
          ];
          networkConfig = mkMerge [
            (mkIf (config.address6 == "auto") {
              IPv6AcceptRA = true;
            })
            (mkIf config.mdns.enable {
              MulticastDNS = "resolve";
            })
          ];
          address = mkMerge [
            (mkIf (! elem config.address4 [ null "dhcp" ]) [ config.address4 ])
            (mkIf (! elem config.address6 [ null "auto" "dhcp" ]) [ config.address6 ])
          ];
          gateway = mkMerge [
            (mkIf (config.gateway4 != null) [ config.gateway4 ])
            (mkIf (config.gateway6 != null) [ config.gateway6 ])
          ];
          DHCP = mkAlmostOptionDefault (
            if config.address4 == "dhcp" && config.address6 == "dhcp" then "yes"
            else if config.address6 == "dhcp" then "ipv6"
            else if config.address4 == "dhcp" then "ipv4"
            else "no"
          );
        };
      };
      confInternal = let
        index = system.proxmox.vm.id - internalOffset;
      in {
        name = mkIf system.proxmox.container.enable (mkAlmostOptionDefault "eth9");
        bridge = mkAlmostOptionDefault "vmbr9";
        address4 = mkAlmostOptionDefault "10.9.1.${toString index}/24";
        address6 = mkAlmostOptionDefault "fd0c::${toHexStringLower index}/64";
        macAddress = mkIf (system.proxmox.network.interfaces.net0.macAddress or null != null && hasPrefix "BC:24:11:" system.proxmox.network.interfaces.net0.macAddress) (mkAlmostOptionDefault (
          replaceStrings [ "BC:24:11:" ] [ "BC:24:19:" ] system.proxmox.network.interfaces.net0.macAddress
        ));
        networkd.name = mkDefault "_00-int";
        networkd.networkSettings = {
          domains = mkDefault [ ]; # int.${domain}?
          linkConfig.RequiredForOnline = false;
          ipv6AcceptRAConfig = {
            Token = mkOptionDefault "static:::${toHexStringLower index}";
            DHCPv6Client = mkOptionDefault false;
          };
          networkConfig = {
            IPv6PrivacyExtensions = mkOptionDefault "no";
          };
        };
      };
    in mkMerge [
      conf
      (mkIf config.internal.enable confInternal)
    ];
  };
in {
  options.proxmox.network = with lib.types; {
    interfaces = mkOption {
      type = attrsOf (submoduleWith {
        modules = [ networkInterfaceModule ];
        specialArgs = {
          system = config;
        };
      });
      default = { };
    };
    internal = {
      interface = mkOption {
        type = nullOr unspecified;
      };
    };
    local = {
      interface = mkOption {
        type = nullOr unspecified;
      };
    };
  };
  config.proxmox.network = {
    internal = {
      interface = mkOptionDefault (findSingle (interface: interface.internal.enable) null (throw "expected only one internal network interface") (attrValues cfg.interfaces));
    };
    local.interface = mkOptionDefault (findFirst (interface: interface.local.enable) null (attrValues cfg.interfaces));
  };
  config.network.networks = let
    strip4 = mapNullable (removeSuffix "/24");
    strip6 = mapNullable (removeSuffix "/64");
  in {
    int = mkIf (cfg.internal.interface != null) (mapAlmostOptionDefaults {
      inherit (cfg.internal.interface) macAddress;
      address4 = strip4 cfg.internal.interface.address4;
      address6 = strip6 cfg.internal.interface.address6;
    });
    local = mkIf (cfg.local.interface != null) (mapAlmostOptionDefaults {
      inherit (cfg.local.interface) macAddress;
      address4 = strip4 cfg.local.interface.local.address4;
      address6 = strip6 cfg.local.interface.local.address6;
    });
  };
}