nixos-config/components/gui/suspend.nix

{
  pkgs,
  config,
  lib,
  ...
}:
with lib;
{
  options.components.gui.suspend.enable = mkOption {
    type = lib.types.bool;
    default = config.components.gui.enable;
  };

  config = mkIf (config.components.gui.suspend.enable) {

    systemd.services.screenlock = {
      before = [ "sleep.target" ];
      requiredBy = [ "sleep.target" ];
      environment =
        let
          display = if (config.services.xserver.display != null) then config.services.xserver.display else 0;
        in
        {
          DISPLAY = ":${toString display}";
        };
      script = ''
        ${pkgs.xlockmore}/bin/xlock -mode life1d -size 1 &
        sleep 1
      '';
      serviceConfig = {
        SyslogIdentifier = "screenlock";
        #Type = "simple";
        Type = "forking";
        User = config.users.users.mainUser.name;
      };
    };

    services.logind.lidSwitch = "suspend";

  };
}