Overhaul uucp

1 files changed, 87 insertions, 27 deletions

diff --git a/custom/uucp.nix b/custom/uucp.nix
index f5b89bfe..61e42119 100644
--- a/custom/uucp.nix
+++ b/custom/uucp.nix
@@ -3,23 +3,72 @@
 with lib;
 
 let
-  portSpec = name: ''
-    port ${name}
+  portSpec = name: node: concatStringsSep "\n" (mapAttrsToList (portName: port: ''
+    port ${name}.${portName}
     type pipe
-    protocol ${if builtins.hasAttr name cfg.protocols then cfg.protocols."${name}" else cfg.defaultProtocol}
+    protocol ${node.protocols}
     reliable true
-    command ${pkgs.openssh}/bin/ssh -x -o batchmode=yes ${name}
-  '';
+    command ${pkgs.openssh}/bin/ssh -x -o batchmode=yes ${name}.${portName}
+  '') node.hostnames);
   sysSpec = name: ''
     system ${name}
     time Any
-    port ${name}
     chat ""
-    protocol ${if builtins.hasAttr name cfg.protocols then cfg.protocols."${name}" else cfg.defaultProtocol}
     command-path ${concatStringsSep " " cfg.commandPath}
-    commands ${concatStringsSep " " (if builtins.hasAttr name cfg.commands then unique (cfg.defaultCommands ++ cfg.commands."${name}") else cfg.defaultCommands)}
+    commands ${concatStringsSep " " node.commands}
+    ${concatStringsSep "\nalternate\n" (mapAttrsToList (portName: port: ''
+      port ${name}.${portName}
+    '') node.hostnames)}
+  '';
+  sshConfig = name: node: concatStringsSep "\n" (mapAttrsToList (portName: port: ''
+    Host ${name}.${portName}
+      Hostname ${port}
+      IdentitiesOnly Yes
+      IdentityFile ${cfg.sshKeyDir}/${name}.pub
+  '') node.hostnames);
+  sshKeyGen = name: node: ''
+    if [[ ! -e ${cfg.sshKeyDir}/${name} ]]; then
+      ssh-keygen ${escapeShellArgs node.generateKey} -f ${cfg.sshKeyDir}/${name}
+    fi
+  '';
+  restrictKey = key: ''
+    restrict,command="${config.security.wrapperDir}/uucico" ${key}
   '';
 
+  nodeCfg = {
+    options = {
+      commands = mkOption {
+        type = types.listOf types.string;
+        default = cfg.defaultCommands;
+        description = "Commands to allow for this remote";
+      };
+
+      protocols = mkOption {
+        type = types.string;
+        default = cfg.defaultProtocols;
+        description = "UUCP protocols to use for this remote";
+      };
+
+      publicKeys = mkOption {
+        type = types.listOf types.string;
+        default = [];
+        description = "SSH public keys for this node";
+      };
+
+      generateKey = mkOption {
+        type = types.listOf types.string;
+        default = [ "-t ed25519" ];
+        description = "Arguments to pass to `ssh-keygen` to generate a keypair for communication with this host";
+      };
+
+      hostnames = mkOption {
+        type = types.attrsOf types.string;
+        default = {};
+        description = "";
+      };
+    };
+  };
+
   cfg = config.services.uucp;
 in {
   options = {
@@ -44,14 +93,14 @@ in {
         description = "Overrides for the local uucp linux-user";
       };
 
-      sshConfig = mkOption {
+      extraSSHConfig = mkOption {
         type = types.str;
         default = "";
-        description = "~uucp/.ssh/config";
+        description = "Extra SSH config";
       };
 
       remoteNodes = mkOption {
-        type = types.listOf types.str;
+        type = types.attrsOf (types.submodule nodeCfg);
         default = {};
         description = ''
           Ports to set up
@@ -73,22 +122,22 @@ in {
         description = "Commands allowed for remotes without explicit override";
       };
 
-      commands = mkOption {
-        type = types.attrsOf (types.listOf types.string);
-        default = {};
-        description = "Override commands for specific remotes";
-      };
-
-      defaultProtocol = mkOption {
+      defaultProtocols = mkOption {
         type = types.string;
         default = "e";
         description = "UUCP protocol to use within ssh unless overriden";
       };
 
-      protocols = mkOption {
-        type = types.attrsOf types.string;
-        default = {};
-        description = "UUCP protocols to use for specific remotes";
+      homeDir = mkOption {
+        type = types.path;
+        default = "/var/uucp";
+        description = "Home of the uucp user";
+      };
+
+      sshKeyDir = mkOption {
+        type = types.path;
+        default = "${cfg.homeDir}/.ssh/";
+        description = "Directory to store ssh keypairs";
       };
 
       spoolDir = mkOption {
@@ -184,16 +233,27 @@ in {
       isSystemUser = true;
       isNormalUser = false;
       createHome = true;
-      home = cfg.spoolDir;
+      home = cfg.homeDir;
       description = "User for uucp over ssh";
       useDefaultShell = true;
+      openssh.authorizedKeys.keys = map restrictKey (concat (mapAttrsToList (name: node: node.publicKeys) cfg.remoteNodes));
     } // cfg.sshUser;
 
     system.activationScripts."uucp-sshconfig" = ''
       mkdir -p ${config.users.users."uucp".home}/.ssh
       chown ${config.users.users."uucp".name}:${config.users.users."uucp".group} ${config.users.users."uucp".home}/.ssh
       chmod 700 ${config.users.users."uucp".home}/.ssh
-      ln -fs ${builtins.toFile "ssh-config" cfg.sshConfig} ${config.users.users."uucp".home}/.ssh/config
+      ln -fs ${builtins.toFile "ssh-config" ''
+        ${concatStringsSpec "\n" (mapAttrsToList sshConfig cfg.remoteNodes)}
+      
+        ${cfg.extraSSHConfig}
+      ''} ${config.users.users."uucp".home}/.ssh/config
+
+      mkdir -p ${cfg.sshKeyDir}
+      chown ${config.users.users."uucp".name}:${config.users.users."uucp".group} ${cfg.sshKeyDir}
+      chmod 700 ${config.sshKeyDir}
+
+      ${concatStringsSep "\n" (mapAttrsToList sshKeyGen cfg.remoteNodes)}
     '';
 
     system.activationScripts."uucp-logs" = ''
@@ -213,10 +273,10 @@ in {
         port ssh
         type stdin
         protocol e
-      '' + concatStringsSep "\n" (map portSpec cfg.remoteNodes);
+      '' + concatStringsSep "\n" (mapAttrsToList portSpec cfg.remoteNodes);
     };
     environment.etc."uucp/sys" = {
-      text = cfg.extraSys + "\n" + concatStringsSep "\n" (map sysSpec cfg.remoteNodes);
+      text = cfg.extraSys + "\n" + concatStringsSep "\n" (mapAttrsToList sysSpec cfg.remoteNodes);
     };
 
     security.wrappers = let
@@ -301,7 +361,7 @@ in {
 
         case "''${2}" in
           (?(vpn-)up)
-            ${concatMapStringsSep "\n    " (name: "${pkgs.systemd}/bin/systemctl start uucico@${name}.service") cfg.remoteNodes}
+            ${concatStringsSep "\n    " (mapAttrsToList (name: node: "${pkgs.systemd}/bin/systemctl start uucico@${name}.service") cfg.remoteNodes)}
             ;;
         esac
       '';