From 8b1fd1048d08ccb3f73825032221001b7484ad3c Mon Sep 17 00:00:00 2001
From: NullBite <me@nullbite.com>
Date: Fri, 1 Mar 2024 15:48:26 -0500
Subject: [PATCH] multimedia: load nvidia_uvm if NVIDIA is enabled

---
 system/package-sets/multimedia.nix | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/system/package-sets/multimedia.nix b/system/package-sets/multimedia.nix
index ab6f5fc..bf37498 100644
--- a/system/package-sets/multimedia.nix
+++ b/system/package-sets/multimedia.nix
@@ -1,7 +1,8 @@
 { config, lib, pkgs, ...}:
 let
   cfg = config.nixfiles.packageSets.multimedia;
-  inherit (lib) optionals mkEnableOption mkIf;
+  inherit (lib) optional optionals mkEnableOption mkIf;
+  nvidiaEnabled = (lib.elem "nvidia" config.services.xserver.videoDrivers);
 in
 {
   options.nixfiles.packageSets.multimedia = {
@@ -13,10 +14,14 @@ in
       gimp-with-plugins
       krita
       inkscape
+      obs-studio
     ] ++ [
       yt-dlp
       imagemagick
       ffmpeg
     ];
+
+    # needed for NVENC to work in OBS Studio and FFmpeg
+    boot.kernelModules = optional nvidiaEnabled "nvidia_uvm";
   };
 }