# SPDX-FileCopyrightText: 2022 Richard Brežák <richard@brezak.sk>
#
# SPDX-License-Identifier: LGPL-3.0-or-later
{
  pkgs,
  lib,
  inputs',
  options,
  config,
  ...
}: let
  inherit
    (lib)
    singleton
    mkForce
    mapAttrs
    mkOverride
    filter
    ;

  kernelPackages = pkgs.linuxKernel.packages.linux_6_8;
in {
  environment.systemPackages = [
    kernelPackages.turbostat
  ];

  hardware.cpu.intel.updateMicrocode = true;

  boot = {
    initrd.systemd.enable = true;
    initrd.availableKernelModules = [
      "xhci_pci"
      "ahci"
      "usb_storage"
      "sr_mod"
      "rtsx_pci_sdmmc"
      "nvme"
      "vmd"
    ];
    initrd.kernelModules = [
      "dm_crypt"
      "encrypted_keys"
      "trusted"
      "nhpoly1305_avx2"
      "nhpoly1305_sse2"
      "nhpoly1305"
      "crypto_null"
      "polyval_clmulni"
      "polyval_generic"
      "gf128mul"
      "dm_mod"
      "crypto_simd"
      "cryptd"
      "adiantum"
      "libpoly1305"
      "libchacha20poly1305"
      "chacha_x86_64"
      "libchacha"
      "poly1305_x86_64"
    ];
    kernelModules = [
      "kvm-intel"
    ];

    kernelParams = [
      "zfs.zfs_arc_max=8589934592"
      "zfs.zfs_arc_sys_free=3221225472"
      "nouveau.config=NvGspRm=1"
      "usbcore.autosuspend=-1"
    ];
    kernelPackages = kernelPackages;

    zfs.package = pkgs.zfs_unstable;
  };

  hardware.enableRedistributableFirmware = true;

  hardware.tuxedo-keyboard.enable = true;

  chaotic.mesa-git = {
    enable = true;
    fallbackSpecialisation = false;
  };

  specialisation.nvidia.configuration = {
    environment.variables.NIXOS_SPECIALISATION = "nvidia";
    boot.loader.grub.enable = mkOverride 0 false;
    boot.kernelPackages = mkForce pkgs.linuxPackages_6_6;
    hardware.nvidia.package = let
      libPathFor = pkgs:
        lib.makeLibraryPath (with pkgs; [
          libdrm
          xorg.libXext
          xorg.libX11
          xorg.libXv
          xorg.libXrandr
          xorg.libxcb
          zlib
          stdenv.cc.cc
          wayland
          # mesa
          libGL
        ]);
    in
      pkgs.linuxPackages_6_6.nvidia_x11_beta.overrideAttrs (old: {
        postUnpack = ''
          set -x
        '';
        libPath = libPathFor pkgs;
        libPath32 = libPathFor pkgs.pkgsi686Linux;
      });
    boot.kernelParams = [
      "nvidia.NVreg_EnableS0ixPowerManagement=1"
      "nvidia.NVreg_S0ixPowerManagementVideoMemoryThreshold=2048"
    ];
    chaotic.mesa-git.enable = mkForce false;

    hardware.nvidia = {
      nvidiaPersistenced = true;

      powerManagement.enable = true;
      powerManagement.finegrained = true;

      prime.reverseSync.enable = false;
      prime.offload.enable = true;

      modesetting.enable = true;
    };

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