# Enables Nvidia GPU support.
{
  pkgs,
  config,
  lib,
  ...
}:
let
  cfg = config.aux.system.gpu.nvidia;
in
{
  options = {
    aux.system.gpu.nvidia = {
      enable = lib.mkEnableOption "Enables Nvidia GPU support.";
      hybrid = {
        enable = lib.mkEnableOption "Enables hybrid GPU support.";
        sync = lib.mkEnableOption "Enables sync mode for faster performance at the cost of higher battery usage.";
        busIDs = {
          nvidia = lib.mkOption {
            description = "The bus ID for your Nvidia GPU.";
            type = lib.types.str;
            example = "PCI:0:2:0";
            default = "";
          };
          intel = lib.mkOption {
            description = "The bus ID for your integrated Intel GPU. If you don't have an Intel GPU, you can leave this blank.";
            type = lib.types.str;
            example = "PCI:14:0:0";
            default = "";
          };
          amd = lib.mkOption {
            description = "The bus ID for your integrated AMD GPU. If you don't have an AMD GPU, you can leave this blank.";
            type = lib.types.str;
            example = "PCI:54:0:0";
            default = "";
          };
        };
      };
    };

  };

  config = lib.mkIf cfg.enable {
    assertions = [
      {
        assertion = (cfg.hybrid.busIDs.nvidia != "");
        message = "You need to define a bus ID for your Nvidia GPU. To learn how to find the bus ID, see https://wiki.nixos.org/wiki/Nvidia#Configuring_Optimus_PRIME:_Bus_ID_Values_.28Mandatory.29.";
      }
      {
        assertion = (cfg.hybrid.busIDs.intel != "" || cfg.hybrid.busIDs.amd != "");
        message = "You need to define a bus ID for your non-Nvidia GPU. To learn how to find your bus ID, see https://wiki.nixos.org/wiki/Nvidia#Configuring_Optimus_PRIME:_Bus_ID_Values_.28Mandatory.29.";
      }
    ];

    aux.system.allowUnfree = true;

    services.xserver.videoDrivers = lib.mkDefault [ "nvidia" ];
    hardware = {
      opengl.extraPackages = with pkgs; [ vaapiVdpau ];
      nvidia = {
        modesetting.enable = true;
        nvidiaSettings = config.aux.system.ui.desktops.enable;
        package = config.boot.kernelPackages.nvidiaPackages.stable;
        prime = lib.mkIf cfg.hybrid.enable {

          offload = lib.mkIf (!cfg.hybrid.sync) {
            enable = true;
            enableOffloadCmd = true; # Provides `nvidia-offload` command.
          };

          sync.enable = cfg.hybrid.sync;

          nvidiaBusId = cfg.hybrid.busIDs.nvidia;
          intelBusId = cfg.hybrid.busIDs.intel;
          amdgpuBusId = cfg.hybrid.busIDs.amd;
        };
      };
    };
  };
}