From b95aa94cae21bc4828a934b14ebe1f69a58e7958 Mon Sep 17 00:00:00 2001
From: Gabriel Arazas <foodogsquared@foodogsquared.one>
Date: Tue, 27 Feb 2024 21:10:06 +0800
Subject: [PATCH] flake-parts/setups/nixos: add custom nixos-generators format

---
 modules/flake-parts/setups/nixos.nix | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/modules/flake-parts/setups/nixos.nix b/modules/flake-parts/setups/nixos.nix
index e46260ee..e9d96bca 100644
--- a/modules/flake-parts/setups/nixos.nix
+++ b/modules/flake-parts/setups/nixos.nix
@@ -39,6 +39,22 @@ let
       system = null;
     };
 
+  # The nixos-generators modules set as well as our custom-made ones.
+  nixosGeneratorsModulesSet =
+    let
+      importNixosGeneratorModule = (_: modulePath: {
+        imports = [
+          modulePath
+          "${inputs.nixos-generators}/format-module.nix"
+        ];
+      });
+
+      customFormats = lib.mapAttrs importNixosGeneratorModule {
+        install-iso-graphical = ../../nixos-generators/install-iso-graphical.nix;
+      };
+    in
+    inputs.nixos-generators.nixosModules // customFormats;
+
   # A very very thin wrapper around `mkHost` to build with the given format.
   mkImage =
     { system
@@ -48,7 +64,7 @@ let
     }:
     let
       extraModules' =
-        extraModules ++ [ inputs.nixos-generators.nixosModules.${format} ];
+        extraModules ++ [ nixosGeneratorsModulesSet.${format} ];
       image = mkHost {
         inherit nixpkgsBranch system;
         extraModules = extraModules';