home-manager/modules/services/swayidle.nix

{
  config,
  lib,
  pkgs,
  ...
}:

let
  inherit (lib)
    mkOption
    types
    literalExpression
    ;

  cfg = config.services.swayidle;

in
{
  meta.maintainers = [ lib.maintainers.c0deaddict ];

  options.services.swayidle =
    let

      timeoutModule = {
        options = {
          timeout = mkOption {
            type = types.ints.positive;
            example = 60;
            description = "Timeout in seconds.";
          };

          command = mkOption {
            type = types.str;
            description = "Command to run after timeout seconds of inactivity.";
          };

          resumeCommand = mkOption {
            type = with types; nullOr str;
            default = null;
            description = "Command to run when there is activity again.";
          };
        };
      };

      eventsModule = {
        options = {
          before-sleep = mkOption {
            type = types.nullOr types.str;
            default = null;
            description = "Command to run before suspending.";
          };

          after-resume = mkOption {
            type = types.nullOr types.str;
            default = null;
            description = "Command to run after resuming.";
          };

          lock = mkOption {
            type = types.nullOr types.str;
            default = null;
            description = "Command to run when the logind session is locked.";
          };

          unlock = mkOption {
            type = types.nullOr types.str;
            default = null;
            description = "Command to run when the logind session is unlocked.";
          };
        };
      };

    in
    {
      enable = lib.mkEnableOption "idle manager for Wayland";

      package = lib.mkPackageOption pkgs "swayidle" { };

      timeouts = mkOption {
        type = with types; listOf (submodule timeoutModule);
        default = [ ];
        example = literalExpression ''
          [
            { timeout = 60; command = "''${pkgs.swaylock}/bin/swaylock -fF"; }
            { timeout = 90; command = "''${pkgs.systemd}/bin/systemctl suspend"; }
          ]
        '';
        description = "List of commands to run after idle timeout.";
      };

      events = mkOption {
        type =
          with types;
          (coercedTo (listOf attrs)) (
            events:
            lib.warn
              ''
                The syntax of services.swayidle.events has changed. While it
                previously accepted a list of events, it now accepts an attrset
                keyed by the event name.
              ''
              (
                lib.listToAttrs (
                  map (e: {
                    name = e.event;
                    value = e.command;
                  }) events
                )
              )
          ) (submodule eventsModule);
        default = { };
        example = literalExpression ''
          {
            "before-sleep" = "''${pkgs.swaylock}/bin/swaylock -fF";
            "lock" = "lock";
          }
        '';
        description = "Run command on occurrence of a event.";
      };

      extraArgs = mkOption {
        type = with types; listOf str;
        default = [ "-w" ];
        description = "Extra arguments to pass to swayidle.";
      };

      systemdTarget = mkOption {
        type = types.str;
        default = config.wayland.systemd.target;
        defaultText = literalExpression "config.wayland.systemd.target";
        example = "sway-session.target";
        description = ''
          Systemd target to bind to.
        '';
      };

    };

  config = lib.mkIf cfg.enable {
    assertions = [
      (lib.hm.assertions.assertPlatform "services.swayidle" pkgs lib.platforms.linux)
    ];

    home.packages = [ cfg.package ];

    systemd.user.services.swayidle = {
      Unit = {
        Description = "Idle manager for Wayland";
        Documentation = "man:swayidle(1)";
        ConditionEnvironment = "WAYLAND_DISPLAY";
        PartOf = [ cfg.systemdTarget ];
        After = [ cfg.systemdTarget ];
      };

      Service = {
        Type = "simple";
        Restart = "always";
        # swayidle executes commands using "sh -c", so the PATH needs to contain a shell.
        Environment = [ "PATH=${lib.makeBinPath [ pkgs.bash ]}" ];
        ExecStart =
          let
            mkTimeout =
              t:
              [
                "timeout"
                (toString t.timeout)
                t.command
              ]
              ++ lib.optionals (t.resumeCommand != null) [
                "resume"
                t.resumeCommand
              ];

            mkEvent = event: command: [
              event
              command
            ];

            nonemptyEvents = lib.filterAttrs (event: command: command != null) cfg.events;

            args =
              cfg.extraArgs
              ++ (lib.concatMap mkTimeout cfg.timeouts)
              ++ (lib.flatten (lib.mapAttrsToList mkEvent nonemptyEvents));
          in
          "${lib.getExe cfg.package} ${lib.escapeShellArgs args}";
      };

      Install = {
        WantedBy = [ cfg.systemdTarget ];
      };
    };
  };
}