{ lib, config, ... }:

{
  options.nvidia.enable = lib.mkEnableOption "NVIDIA graphics";

  config = lib.mkIf config.nvidia.enable {
    nixpkgs.allowedUnfree = [
      "nvidia-x11"
      "nvidia-persistenced"
      "nvidia-settings"
    ];

    hardware.graphics.enable = true;

    hardware.nvidia = {
      modesetting.enable = true;
      powerManagement.enable = true;
      powerManagement.finegrained = false;
      open = true;
      nvidiaSettings = true;
      package = config.boot.kernelPackages.nvidiaPackages.stable;
    };

    services.xserver.videoDrivers = [ "nvidia" ];
  };
}