nixfiles/system/programs/greetd.nix

{
  pkgs,
  config,
  lib,
  options,
  ...
}: let
  cfg = config.nixfiles.programs.greetd;
  inherit (lib) types optional optionals escapeShellArg escapeShellArgs;
  inherit (types) bool enum nullOr str path listOf;
  inherit (builtins) isNull;
  optionalsSet = val: optionals (!(isNull val));
  optionalSet = val: optional (!(isNull val));
  sessions = config.services.displayManager.sessionData.desktops;
  xsessions = "${sessions}/share/xsessions";
  wayland-sessions = "${sessions}/share/wayland-sessions";

  loginwrap = pkgs.writeShellScriptBin "loginwrap" ''
    exec "$SHELL" -lc 'exec "$@"' "login-wrapper" "$@"
  '';

  mkPresetOption = x:
    lib.mkOption {
      description = "${x} greetd configuration";
      type = bool;
      default = false;
    };
in {
  config = lib.mkIf cfg.enable {
    assertions = lib.optionals cfg.settings.autologin [
      {
        assertion = ! builtins.isNull cfg.settings.autologinUser;
        message = "greetd: Auto-login is enabled but no user is configured";
      }
      {
        assertion = ! builtins.isNull cfg.settings.command;
        message = "greetd: Auto-login is enabled but no login command is configured";
      }
    ];

    environment.systemPackages = [loginwrap];
    services.greetd = {
      enable = true;
      settings = {
        initial_session = lib.mkIf cfg.settings.autologin {
          command = cfg.settings.finalCommand;
          user = cfg.settings.autologinUser;
        };

        default_session = lib.mkMerge [
          # tuigreet configuration
          (lib.mkIf cfg.presets.tuigreet.enable {
            command = let
              st = cfg.settings;
              args =
                [
                  "${pkgs.greetd.tuigreet}/bin/tuigreet"
                  "--asterisks"
                  "--remember"
                  "--remember-session"
                  "--sessions"
                  "${xsessions}:${wayland-sessions}"
                ]
                ++ optionalsSet st.greeting ["--greeting" st.greeting]
                ++ optional st.time "--time"
                ++ optionalsSet st.command ["--cmd" st.finalCommand]
                # i think tuigreet might be outdated on nix. disable this because it's not a valid option
                # ++ optionalsSet st.loginShell [ "--session-wrapper" "loginwrap" ]
                ;
            in
              lib.escapeShellArgs args;
          })
        ];
      };
    };

    # regreet config (it is configured through an upstream module; the only
    # greetd-specific config set is default_session, so we can configure it
    # here instead of above.)
    programs.regreet = let
      # lets us use wlr-randr
      wrapperPackage =
        {inherit (pkgs.greetd.regreet) version;}
        // (pkgs.writeShellScriptBin "regreet-wrapper" ''
          ${cfg.settings.graphicalInit}

          exec ${escapeShellArg (lib.getExe pkgs.greetd.regreet)} "$@"
        '');
    in
      lib.mkIf cfg.presets.regreet.enable {
        enable = lib.mkDefault true;
        package = wrapperPackage;
        settings = {
          background.path = cfg.settings.wallpaper;
          fit = lib.mkDefault "Fill";
          appearance.greeting_msg = cfg.settings.greeting;
        };
      };

    security.pam.services.greetd = {
      kwallet = lib.mkIf config.services.desktopManager.plasma6.enable {
        enable = true;
        forceRun = true;
      };
    };

    systemd.tmpfiles.settings."10-regreet" = let
      defaultConfig = {
        user = "greeter";
        group = config.users.users.${config.services.greetd.settings.default_session.user}.group;
        mode = "0755";
      };
    in
      lib.mkIf config.programs.regreet.enable
      {
        "/var/log/regreet".d = defaultConfig;
        "/var/cache/regreet".d = defaultConfig;
        "/var/lib/regreet".d = defaultConfig;
      };

    # self config
    nixfiles.programs.greetd = {
      presets.${cfg.preset}.enable = true;
      settings.graphicalInit = lib.optionalString (cfg.settings.randr != null) ''
        ${lib.getExe pkgs.wlr-randr} ${escapeShellArgs cfg.settings.randr}
      '';
    };
  };

  options.nixfiles.programs.greetd = {
    enable = lib.mkEnableOption "greetd configuration";

    preset = lib.mkOption {
      description = "greetd configuration to enable (shorthand for presets.<preset>.enable)";
      type = enum (lib.mapAttrsToList (name: value: name) options.nixfiles.programs.greetd.presets);
      default = "regreet";
    };

    settings = {
      greeting = lib.mkOption {
        description = "Greeting to show on the chosen greeter (if configurable)";
        type = nullOr str;
        default = "log in pwease!! uwu";
        example = "something boring";
      };
      finalCommand = lib.mkOption {
        description = "Final version of command";
        type = nullOr str;
        default = let
          st = cfg.settings;
          prevcmd = st.command;
          command-login-wrapped = ["loginwrap"] ++ prevcmd;
          cmd =
            if (builtins.isNull prevcmd)
            then null
            else
              (
                if st.loginShell
                then command-login-wrapped
                else prevcmd
              );
        in
          if builtins.isNull cmd
          then null
          else lib.escapeShellArgs cmd;
        readOnly = true;
      };
      command = lib.mkOption {
        description = "Command to run following successful authentication";
        type = nullOr (listOf str);
        default = null;
        example = ["Hyprland"];
      };

      graphicalInit = lib.mkOption {
        description = "Commands to run upon initialization of a graphical greeter.";
        type = lib.types.lines;
        default = "";
      };

      randr = lib.mkOption {
        description = "Options to pass to wlr-randr";
        type = nullOr (listOf str);
        default = null;
        example = ["--output" "HDMI-A-3" "--off"];
      };

      loginShell = lib.mkOption {
        description = "Wrap in login shell to source .profile/.zshenv/etc. (if configurable).";
        type = bool;
        default = true;
        example = false;
      };
      time = lib.mkOption {
        description = "Whether to show the current time (if configurable)";
        type = bool;
        default = true;
        example = false;
      };
      wallpaper = lib.mkOption {
        description = "Path to custom wallpaper (if configurable)";
        type = nullOr path;
        default = "${pkgs.nixfiles-assets}/share/wallpapers/nixfiles-static/Djayjesse-finding_life.png";
        example = "femboy-bee.png";
      };

      autologin = lib.mkOption {
        description = "Whether to configure auto-login";
        type = bool;
        default = false;
        example = true;
      };
      autologinUser = lib.mkOption {
        description = "User to automatically log in";
        type = nullOr str;
        default = null;
        example = "username";
      };
      autolock = lib.mkOption {
        description = ''
          Whether to indicate to a window manager that it should immediately
          lock. This needs to be implemented on a per window manager basis.
        '';
        type = bool;
        default = false;
        example = true;
      };
    };

    presets.regreet.enable = mkPresetOption "regreet";
    presets.tuigreet.enable = mkPresetOption "tuigreet";
  };
}