242/376: Add primop ‘attrValues’

[Ludovic Courtès]

civodul pushed a commit to tag 1.8
in repository guix.

commit d4fcbe1687c3d212966e43118341622cf69612f3
Author: Eelco Dolstra <address@hidden>
Date:   Sat Oct 4 16:41:24 2014 +0200

    Add primop ‘attrValues’
---
 src/libexpr/primops.cc             |   24 +++++++++++++++++++++++-
 tests/lang/eval-okay-attrnames.nix |    2 +-
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/libexpr/primops.cc b/src/libexpr/primops.cc
index 072866a..fb37e21 100644
--- a/src/libexpr/primops.cc
+++ b/src/libexpr/primops.cc
@@ -974,7 +974,28 @@ static void prim_attrNames(EvalState & state, const Pos & 
pos, Value * * args, V
     for (auto & i : *args[0]->attrs)
         mkString(*(v.list.elems[n++] = state.allocValue()), i.name);
 
-    std::sort(v.list.elems, v.list.elems + n, [](Value * v1, Value * v2) { 
return strcmp(v1->string.s, v2->string.s) < 0; });
+    std::sort(v.list.elems, v.list.elems + n,
+        [](Value * v1, Value * v2) { return strcmp(v1->string.s, v2->string.s) 
< 0; });
+}
+
+
+/* Return the values of the attributes in a set as a list, in the same
+   order as attrNames. */
+static void prim_attrValues(EvalState & state, const Pos & pos, Value * * 
args, Value & v)
+{
+    state.forceAttrs(*args[0], pos);
+
+    state.mkList(v, args[0]->attrs->size());
+
+    unsigned int n = 0;
+    for (auto & i : *args[0]->attrs)
+        v.list.elems[n++] = (Value *) &i;
+
+    std::sort(v.list.elems, v.list.elems + n,
+        [](Value * v1, Value * v2) { return (string) ((Attr *) v1)->name < 
(string) ((Attr *) v2)->name; });
+
+    for (unsigned int i = 0; i < n; ++i)
+        v.list.elems[i] = ((Attr *) v.list.elems[i])->value;
 }
 
 
@@ -1501,6 +1522,7 @@ void EvalState::createBaseEnv()
 
     // Sets
     addPrimOp("__attrNames", 1, prim_attrNames);
+    addPrimOp("__attrValues", 1, prim_attrValues);
     addPrimOp("__getAttr", 2, prim_getAttr);
     addPrimOp("__unsafeGetAttrPos", 2, prim_unsafeGetAttrPos);
     addPrimOp("__hasAttr", 2, prim_hasAttr);
diff --git a/tests/lang/eval-okay-attrnames.nix 
b/tests/lang/eval-okay-attrnames.nix
index 978138f..e5b26e9 100644
--- a/tests/lang/eval-okay-attrnames.nix
+++ b/tests/lang/eval-okay-attrnames.nix
@@ -8,4 +8,4 @@ let
 
   values = map (name: builtins.getAttr name attrs) names;
 
-in concat values
+in assert values == builtins.attrValues attrs; concat values