nixos-config/modules/users/modules/desktop/default.nix

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

with lib;
let
  cfg = config.modules.desktop;
in {
  imports = [
    # Import desktop environment modules
    ./bar/waybar.nix
    ./lock-screen/waylock.nix
    ./window-manager/river.nix
  ];

  options.modules.desktop = {
    wayland = mkEnableOption "wayland";
    initScript = mkOption {
      type = types.lines;
      default = "${pkgs.bash}/bin/bash";
      description = "Bash script to execute after logging in.";
    };
  };

  config = {
    home.packages = optionals cfg.wayland (with pkgs; [
      pkgs.wl-clipboard
      pkgs.wtype
    ]);

    home.file.".initrc" = {
      enable = true;
      executable = true;
      text = ''
        #!${pkgs.bash}/bin/bash

      '' + cfg.initScript;
    };
  };
}