each{,Default}SystemMap: accept binary functions

Related: #78

Author: ilkecan

default.nix

@@ -125,13 +125,7 @@ let
       # Merge together the outputs for all systems.
       op = attrs: system:
         let
-          ret =
-            let
-              retOrFunc = f system;
-            in
-            if builtins.isFunction retOrFunc
-              then retOrFunc ret
-              else retOrFunc;
+          ret = maybeFix (f system);
 
           op = attrs: key:
             let
@@ -155,7 +149,8 @@ let
   eachDefaultSystemMap = eachSystemMap defaultSystems;
 
   # Builds a map from <attr>=value to <system>.<attr> = value.
-  eachSystemMap = systems: f: builtins.listToAttrs (builtins.map (system: { name = system; value = f system; }) systems);
+  eachSystemMap = systems: f:
+    builtins.listToAttrs (builtins.map (system: { name = system; value = maybeFix (f system); }) systems);
 
   # Nix flakes insists on having a flat attribute set of derivations in
   # various places like the `packages` and `checks` attributes.
@@ -232,5 +227,11 @@ let
       system
       ;
   };
+
+  maybeFix = arg:
+    let
+      fix = f: let x = f x; in x;
+    in
+    if builtins.isFunction arg then fix arg else arg;
 in
 lib