From 8a1ab709a47f0896cb5b47521e31c8c27f88b2b3 Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <e.dolstra@tudelft.nl>
Date: Fri, 22 Sep 2006 14:31:55 +0000
Subject: [PATCH] * New builtin functions builtins.{hasAttr, getAttr} to check
 for   attribute existence and to return an attribute from an attribute   set,
 respectively.  Example: `hasAttr "foo" {foo = 1;}'.  They   differ from the
 `?' and `.' operators in that the attribute name is   an arbitrary
 expression.  (NIX-61)

---
 src/libexpr/primops.cc          | 16 ++++++++++++++++
 tests/lang/eval-okay-attrs2.exp |  1 +
 tests/lang/eval-okay-attrs2.nix | 10 ++++++++++
 3 files changed, 27 insertions(+)
 create mode 100644 tests/lang/eval-okay-attrs2.exp
 create mode 100644 tests/lang/eval-okay-attrs2.nix

diff --git a/src/libexpr/primops.cc b/src/libexpr/primops.cc
index 93a699b41..f08ad0464 100644
--- a/src/libexpr/primops.cc
+++ b/src/libexpr/primops.cc
@@ -703,6 +703,20 @@ static Expr primCurrentTime(EvalState & state, const ATermVector & args)
 }
 
 
+static Expr primGetAttr(EvalState & state, const ATermVector & args)
+{
+    string attr = evalString(state, args[0]);
+    return evalExpr(state, makeSelect(args[1], toATerm(attr)));
+}
+
+
+static Expr primHasAttr(EvalState & state, const ATermVector & args)
+{
+    string attr = evalString(state, args[0]);
+    return evalExpr(state, makeOpHasAttr(args[1], toATerm(attr)));
+}
+
+
 static Expr primRemoveAttrs(EvalState & state, const ATermVector & args)
 {
     ATermMap attrs(128); /* !!! */
@@ -749,6 +763,8 @@ void EvalState::addPrimOps()
     addPrimOp("abort", 1, primAbort);
 
     addPrimOp("map", 2, primMap);
+    addPrimOp("__getAttr", 2, primGetAttr);
+    addPrimOp("__hasAttr", 2, primHasAttr);
     addPrimOp("removeAttrs", 2, primRemoveAttrs);
     addPrimOp("relativise", 2, primRelativise);
 }
diff --git a/tests/lang/eval-okay-attrs2.exp b/tests/lang/eval-okay-attrs2.exp
new file mode 100644
index 000000000..add36384d
--- /dev/null
+++ b/tests/lang/eval-okay-attrs2.exp
@@ -0,0 +1 @@
+Int(987)
\ No newline at end of file
diff --git a/tests/lang/eval-okay-attrs2.nix b/tests/lang/eval-okay-attrs2.nix
new file mode 100644
index 000000000..9e06b83ac
--- /dev/null
+++ b/tests/lang/eval-okay-attrs2.nix
@@ -0,0 +1,10 @@
+let {
+  as = { x = 123; y = 456; } // { z = 789; } // { z = 987; };
+
+  A = "a";
+  Z = "z";
+
+  body = if builtins.hasAttr A as
+         then builtins.getAttr A as
+         else assert builtins.hasAttr Z as; builtins.getAttr Z as;
+}