From 56e6cad1432510fe6e5d370bb30b246066c3afd1 Mon Sep 17 00:00:00 2001
From: lassulus <git@lassul.us>
Date: Sun, 10 Sep 2023 14:48:41 +0200
Subject: [PATCH] module: add images to system.build

---
 module.nix | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/module.nix b/module.nix
index a178d36..7096cf6 100644
--- a/module.nix
+++ b/module.nix
@@ -1,4 +1,4 @@
-{ config, lib, pkgs, extendModules, ... }:
+{ config, lib, pkgs, extendModules, ... }@args:
 let
   diskoLib = import ./lib {
     inherit lib;
@@ -71,6 +71,13 @@ in
       disko = builtins.trace "the .disko output is deprecated, please use .diskoScript instead" (cfg.devices._scripts pkgs).diskoScript;
       diskoNoDeps = builtins.trace "the .diskoNoDeps output is deprecated, please use .diskoScriptNoDeps instead" (cfg.devices._scripts pkgs).diskoScriptNoDeps;
 
+      diskoImages = diskoLib.makeDiskImage {
+        nixosConfig = args;
+      };
+      diskoImageScript = diskoLib.makeDiskImageScript {
+        nixosConfig = args;
+      };
+
       installTest = diskoLib.testLib.makeDiskoTest {
         inherit extendModules pkgs;
         name = "${config.networking.hostName}-disko";