From ec6d572d94619ee696286173ccb69a84481e6d79 Mon Sep 17 00:00:00 2001
From: cnst <adamhilmersson@gmail.com>
Date: Tue, 20 Aug 2024 21:23:31 +0200
Subject: [PATCH] cleaned up nvidia code

---
 hosts/toothpc/modules.nix                           | 2 +-
 system/modules/hardware/graphics/nvidia/default.nix | 8 +++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/hosts/toothpc/modules.nix b/hosts/toothpc/modules.nix
index 3c32d4ab..5cb7de19 100644
--- a/hosts/toothpc/modules.nix
+++ b/hosts/toothpc/modules.nix
@@ -20,7 +20,7 @@
         amd.enable = false;
         nvidia = {
           enable = true;
-          package = "stable"; # set to stable or beta depending on preference
+          package = "production"; # set to beta/stable/production depending on preference
         };
       };
     };
diff --git a/system/modules/hardware/graphics/nvidia/default.nix b/system/modules/hardware/graphics/nvidia/default.nix
index a9bde556..693d970a 100644
--- a/system/modules/hardware/graphics/nvidia/default.nix
+++ b/system/modules/hardware/graphics/nvidia/default.nix
@@ -24,9 +24,9 @@ in {
     modules.hardware.graphics.nvidia = {
       enable = mkEnableOption "Enables NVidia graphics";
       package = mkOption {
-        type = types.enum ["stable" "beta"];
+        type = types.enum ["stable" "beta" "production"]; # Added "production" here
         default = "stable";
-        description = "Choose between the stable or beta NVidia driver package";
+        description = "Choose between the stable, beta, or production NVidia driver package";
       };
     };
   };
@@ -54,7 +54,9 @@ in {
         package =
           if cfg.package == "beta"
           then config.boot.kernelPackages.nvidiaPackages.beta
-          else config.boot.kernelPackages.nvidiaPackages.production;
+          else if cfg.package == "production"
+          then config.boot.kernelPackages.nvidiaPackages.production
+          else config.boot.kernelPackages.nvidiaPackages.stable;
         modesetting.enable = true;
         powerManagement = {
           enable = false;