From c135b7222b53ffb9d6c47171c0632246c4d9570c Mon Sep 17 00:00:00 2001
From: Emile Clark-Boman <emile@agribit.com.au>
Date: Wed, 28 Jan 2026 15:39:27 +1000
Subject: [PATCH] add allAttrs function

---
 nt/primitives/std/prim/attrs.nix | 26 ++++++++++++++++++++------
 1 file changed, 20 insertions(+), 6 deletions(-)

diff --git a/nt/primitives/std/prim/attrs.nix b/nt/primitives/std/prim/attrs.nix
index fed98a5..7e8497c 100644
--- a/nt/primitives/std/prim/attrs.nix
+++ b/nt/primitives/std/prim/attrs.nix
@@ -1,6 +1,7 @@
 {this, ...}: let
   inherit
     (builtins)
+    all
     attrNames
     elem
     elemAt
@@ -31,18 +32,16 @@ in rec {
     assert enfIsAttrs xs msg;
       hasAttr name xs || throw "${msg}: missing required attribute \"${name}\"";
 
-  getAttrOr = name: f: xs:
-    if xs ? ${name}
-    then xs.${name}
-    else f xs;
-
-  getAttrDefault = name: default: getAttrOr name (_: default);
+  getAttrOr = name: f: xs: xs.${name} or f xs;
+  getAttrDefault = name: default: xs: xs.${name} or default;
 
   genAttrs = names: f:
     names
     |> map (n: nameValuePair n (f n))
     |> listToAttrs;
 
+  attrsToList = xs: mapAttrs nameValuePair xs;
+
   mergeAttrsList = list: let
     # `binaryMerge start end` merges the elements at indices `index` of `list` such that `start <= index < end`
     # Type: Int -> Int -> Attrs
@@ -85,4 +84,19 @@ in rec {
     |> removeAttrs xs;
 
   nameValuePair = name: value: {inherit name value;};
+
+  # allAttrs :: (String -> Any -> Bool) -> AttrSet -> Bool
+  allAttrs = pred: xs:
+    attrNames xs |> all (name: pred name xs.${name});
+
+  enfAllAttrs = pred: xs: xsName: reason: msg: let
+    err = name: throw "${msg}: attribute \"${xsName}.${name}\" cannot ${reason}";
+  in
+    attrNames xs
+    |> all (name: pred name xs.${name} || err name);
+
+  # XXX: TODO: for every enf* function make an equivalent assert function
+  # XXX: TODO: (these MUST be chainable for the pipe operator)
+  assertAllAttrs = pred: xsName: reason: msg: xs:
+    assert enfAllAttrs pred xs xsName reason msg; xs;
 }