From 19e19dce5a100979fbe13464cf76271f934d96fe Mon Sep 17 00:00:00 2001
From: Robert Hensing <robert@roberthensing.nl>
Date: Fri, 27 May 2022 17:41:07 +0200
Subject: [PATCH] Add withSystem parameter

---
 all-modules.nix        |  1 +
 modules/withSystem.nix | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+)
 create mode 100644 modules/withSystem.nix

diff --git a/all-modules.nix b/all-modules.nix
index ac8a931..d148f9d 100644
--- a/all-modules.nix
+++ b/all-modules.nix
@@ -13,5 +13,6 @@
     ./modules/overlays.nix
     ./modules/packages.nix
     ./modules/perSystem.nix
+    ./modules/withSystem.nix
   ];
 }
diff --git a/modules/withSystem.nix b/modules/withSystem.nix
new file mode 100644
index 0000000..c6b0a16
--- /dev/null
+++ b/modules/withSystem.nix
@@ -0,0 +1,39 @@
+{ config, lib, flake-parts-lib, self, getSystem, ... }:
+let
+  inherit (lib)
+    genAttrs
+    mapAttrs
+    mkOption
+    types
+    ;
+  inherit (flake-parts-lib)
+    mkPerSystemOption
+    ;
+in
+{
+  options = {
+    perSystem = mkPerSystemOption ({ config, options, system, specialArgs, ... }: {
+      _file = ./perSystem.nix;
+      options = {
+        allModuleArgs = mkOption {
+          type = types.lazyAttrsOf (types.raw or types.unspecified);
+          internal = true;
+          readOnly = true;
+          description = "Internal option that exposes _module.args, for use by withSystem.";
+        };
+      };
+      config = {
+        allModuleArgs = config._module.args // specialArgs // { inherit config options; };
+      };
+    });
+  };
+
+  config = {
+    _module.args = {
+      withSystem =
+        system: f:
+        f
+          (getSystem system).allModuleArgs;
+    };
+  };
+}