From 7cff52a24546a6cc67701f2bb481d76c60949b54 Mon Sep 17 00:00:00 2001
From: "Zhiyong (Justin) He" <justin.he814@gmail.com>
Date: Tue, 17 Sep 2024 19:08:35 +0300
Subject: [PATCH] Update flake.nix to allow customised resolution for NixOS

---
 flake.nix | 42 +++++++++++++++++++++++++-----------------
 1 file changed, 25 insertions(+), 17 deletions(-)

diff --git a/flake.nix b/flake.nix
index 0481487..520a36d 100644
--- a/flake.nix
+++ b/flake.nix
@@ -26,7 +26,7 @@
             "2k" = "2560x1440";
             "4k" = "3840x2160";
             "ultrawide2k" = "3440x1440";
-          };
+          } // (if cfg.customResolution != null then { "custom" = cfg.customResolution; } else {});
           grub2-theme = pkgs.stdenv.mkDerivation {
             name = "grub2-theme";
             src = "${self}";
@@ -61,7 +61,9 @@
               fi;
             '';
           };
-          resolution = resolutions."${cfg.screen}";
+          resolution = if cfg.customResolution != null
+            then cfg.customResolution
+            else resolutions."${cfg.screen}";
         in
         rec {
           options = {
@@ -93,9 +95,17 @@
               screen = mkOption {
                 default = "1080p";
                 example = "1080p";
-                type = types.enum [ "1080p" "2k" "4k" "ultrawide" "ultrawide2k" ];
+                type = types.enum ([ "1080p" "2k" "4k" "ultrawide" "ultrawide2k" ] ++ (if cfg.customResolution != null then [ "custom" ] else []));
                 description = ''
-                  The screen resolution to use for grub2.
+                  The screen resolution to use for grub2. Use "custom" if you've set a customResolution.
+                '';
+              };
+              customResolution = mkOption {
+                default = null;
+                example = "1600x900";
+                type = types.nullOr (types.strMatching "[0-9]+x[0-9]+");
+                description = ''
+                  Custom resolution for grub2 theme. Should be in the format "WIDTHxHEIGHT".
                 '';
               };
               splashImage = mkOption {
@@ -138,19 +148,17 @@
             environment.systemPackages = [
               grub2-theme
             ];
-            boot.loader.grub =
-              {
-                theme = "${grub2-theme}/grub/themes/${cfg.theme}";
-                splashImage =
-                  "${grub2-theme}/grub/themes/${cfg.theme}/background.jpg";
-                gfxmodeEfi = "${resolution},auto";
-                gfxmodeBios = "${resolution},auto";
-                extraConfig = ''
-                  insmod gfxterm
-                  insmod png
-                  set icondir=($root)/theme/icons
-                '';
-              };
+            boot.loader.grub = {
+              theme = "${grub2-theme}/grub/themes/${cfg.theme}";
+              splashImage = "${grub2-theme}/grub/themes/${cfg.theme}/background.jpg";
+              gfxmodeEfi = "${resolution},auto";
+              gfxmodeBios = "${resolution},auto";
+              extraConfig = ''
+                insmod gfxterm
+                insmod png
+                set icondir=($root)/theme/icons
+              '';
+            };
           }]);
         };
     };