From 104a61a05511d813ace040496268016cc97c88f7 Mon Sep 17 00:00:00 2001
From: Youwen Wu <youwenw@gmail.com>
Date: Tue, 24 Dec 2024 18:19:54 -0800
Subject: [PATCH] feat: manage nvidia graphics as a module

---
 modules/linux/graphics/default.nix | 51 ++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 modules/linux/graphics/default.nix

diff --git a/modules/linux/graphics/default.nix b/modules/linux/graphics/default.nix
new file mode 100644
index 0000000..2b7aa74
--- /dev/null
+++ b/modules/linux/graphics/default.nix
@@ -0,0 +1,51 @@
+{
+  lib,
+  config,
+  pkgs,
+  ...
+}:
+let
+  cfg = config.liminalOS.system.graphics;
+in
+{
+  options.liminalOS.system.graphics = {
+    enable = lib.mkOption {
+      type = lib.types.bool;
+      default = true;
+      description = ''
+        Whether to set up basic graphics settings and options.
+      '';
+    };
+    nvidia = {
+      enable = lib.mkEnableOption "recommended nvidia drivers and configuration";
+      optimus-prime.enable = lib.mkEnableOption "Nvidia OPTIMUS and PRIME";
+      optimus-prime.powerMode = lib.mkOption {
+        type = lib.types.enum [
+          "powersaving"
+          "performance"
+        ];
+        default = "performance";
+        description = ''
+          Whether to use Nvidia OPTIMUS with maximum performance using the discrete graphics card, or to use offload mode to primarily use the integrated GPU and save power.
+        '';
+      };
+      suppressUnfreeWarning = lib.mkOption {
+        type = lib.types.bool;
+        default = false;
+        description = ''
+          Whether to disable the assertion that warns the user if they try to enable proprietary nvidia drivers without setting allowUnfree.
+        '';
+      };
+    };
+  };
+
+  config = lib.mkIf cfg.enable {
+    hardware.nvidia = lib.mkIf cfg.nvidia.enable {
+      modesetting.enable = true;
+      powerManagement.enable = true;
+      powerManagement.finegrained = false;
+      open = true;
+      package = config.boot.kernelPackages.nvidiaPackages.beta;
+    };
+  };
+}