fix disko generation

disko/game-desktop.nix

@@ -1,7 +1,5 @@
-{ disko, ... }:
-{
-  imports = [ disko.nixosModules.disko ];
-  disko.devices.disk.nvme0n1 =
+_: {
+  disk.nvme0n1 =
     let
       baseDevice = "/dev/disk/by-id/nvme-WDC_PC_SN530_SDBPNPZ-512G-1006_21311N802456";
     in
@@ -35,5 +33,4 @@
         };
       };
     };
-  _file = ./disko.nix;
 }

disko/gerg-desktop.nix

@@ -1,9 +1,6 @@
-{ disko, ... }:
-{ lib, ... }:
-{
-  imports = [ disko.nixosModules.disko ];
+lib: {
 
-  disko.devices.disk =
+  disk =
     lib.genAttrs
       [
         "0E"
@@ -53,8 +50,7 @@
           };
         }
       );
-
-  disko.devices.zpool = {
+  zpool = {
     rpool = {
       type = "zpool";
       mode = "mirror";
@@ -108,5 +104,4 @@
       };
     };
   };
-  _file = ./disko.nix;
 }

disko/media-laptop.nix

@@ -1,7 +1,5 @@
-{ disko, ... }:
-{
-  imports = [ disko.nixosModules.disko ];
-  disko.devices.disk.sda =
+_: {
+  disk.sda =
     let
       baseDevice = "/dev/disk/by-id/ata-WDC_WDS240G2G0A-00JH30_180936803144";
     in
@@ -35,5 +33,4 @@
         };
       };
     };
-  _file = ./disko.nix;
 }

lib/default.nix

@@ -1,4 +1,9 @@
-inputs@{ unstable, self, ... }:
+inputs@{
+  unstable,
+  self,
+  disko,
+  ...
+}:
 let
   inherit (unstable) lib;
 
@@ -62,12 +67,23 @@ in
             networking.hostName = name;
             nixpkgs.hostPlatform = system;
           })
+          (lib.optionals (self.diskoConfigurations ? "disko-${name}") [
+            self.diskoConfigurations."disko-${name}"
+            disko.nixosModules.default
+          ])
         ];
       }
     );
   mkDisko =
     names:
-    lib.genAttrs names (name: (import "${self}/hosts/${name}/disko.nix" inputs));
+    lib.listToAttrs (
+      map
+        (name: {
+          name = "disko-${name}";
+          value.disko.devices = import "${self}/disko/${name}.nix" lib;
+        })
+        names
+    );
 
   mkPackages =
     path: pkgs: