Adds the ability to post to multiple channels by simply running multiple instances of clbot. We should probably implement support for this in clbot itself, but right now I can't be bothered to write Go. Change-Id: I5cffd0dc10a7f6cc19c37c5834c5610166b4ae23 Reviewed-on: https://cl.tvl.fyi/c/depot/+/1771 Tested-by: BuildkiteCI Reviewed-by: kanepyork <rikingcoding@gmail.com> Reviewed-by: lukegb <lukegb@tvl.fyi>
		
			
				
	
	
		
			75 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			75 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
# Module that configures CLBot, our Gerrit->IRC info bridge.
 | 
						|
{ config, lib, pkgs, ... }:
 | 
						|
 | 
						|
let
 | 
						|
  inherit (builtins) attrValues concatStringsSep mapAttrs readFile;
 | 
						|
  inherit (pkgs) runCommandNoCC;
 | 
						|
 | 
						|
  inherit (lib)
 | 
						|
    listToAttrs
 | 
						|
    mkEnableOption
 | 
						|
    mkIf
 | 
						|
    mkOption
 | 
						|
    removeSuffix
 | 
						|
    types;
 | 
						|
 | 
						|
  description = "Bot to forward CL notifications";
 | 
						|
  cfg = config.services.depot.clbot;
 | 
						|
 | 
						|
  mkFlags = flags:
 | 
						|
    concatStringsSep " "
 | 
						|
      (attrValues (mapAttrs (key: value: "-${key} \"${toString value}\"") flags));
 | 
						|
 | 
						|
  # Escapes a unit name for use in systemd
 | 
						|
  systemdEscape = name: removeSuffix "\n" (readFile (runCommandNoCC "unit-name" {} ''
 | 
						|
    ${pkgs.systemd}/bin/systemd-escape '${name}' >> $out
 | 
						|
  ''));
 | 
						|
 | 
						|
  mkUnit = flags: channel: {
 | 
						|
    name = "clbot-${systemdEscape channel}";
 | 
						|
    value = {
 | 
						|
      description = "${description} to ${channel}";
 | 
						|
      wantedBy = [ "multi-user.target" ];
 | 
						|
 | 
						|
      script = "${config.depot.fun.clbot}/bin/clbot ${mkFlags (cfg.flags // {
 | 
						|
        irc_channel = channel;
 | 
						|
      })} -alsologtostderr";
 | 
						|
 | 
						|
      serviceConfig = {
 | 
						|
        User = "clbot";
 | 
						|
        EnvironmentFile = "/etc/secrets/clbot";
 | 
						|
        Restart = "always";
 | 
						|
      };
 | 
						|
    };
 | 
						|
  };
 | 
						|
in {
 | 
						|
  options.services.depot.clbot = {
 | 
						|
    enable = mkEnableOption description;
 | 
						|
 | 
						|
    flags = mkOption {
 | 
						|
      type = types.attrsOf types.str;
 | 
						|
      description = "Key value pairs for command line flags";
 | 
						|
    };
 | 
						|
 | 
						|
    channels = mkOption {
 | 
						|
      type = with types; listOf str;
 | 
						|
      description = "Channels in which to post (generates one unit per channel)";
 | 
						|
    };
 | 
						|
  };
 | 
						|
 | 
						|
  config = mkIf cfg.enable {
 | 
						|
    # This does not use DynamicUser because we need to make some files
 | 
						|
    # (notably the SSH private key) readable by this user outside of
 | 
						|
    # the module.
 | 
						|
    users = {
 | 
						|
      groups.clbot = {};
 | 
						|
 | 
						|
      users.clbot = {
 | 
						|
        group = "clbot";
 | 
						|
        isNormalUser = false;
 | 
						|
      };
 | 
						|
    };
 | 
						|
 | 
						|
    systemd.services = listToAttrs (map (mkUnit cfg.flags) cfg.channels);
 | 
						|
  };
 | 
						|
}
 |