diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/libexpr/primops.cc | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/libexpr/primops.cc b/src/libexpr/primops.cc index e3dd2db18d83..b0986028ee24 100644 --- a/src/libexpr/primops.cc +++ b/src/libexpr/primops.cc @@ -444,6 +444,23 @@ static Expr primCurrentTime(EvalState & state, const ATermVector & args) } +static Expr primRemoveAttrs(EvalState & state, const ATermVector & args) +{ + ATermMap attrs; + queryAllAttrs(evalExpr(state, args[0]), attrs, true); + + ATermList list; + if (!matchList(evalExpr(state, args[1]), list)) + throw Error("`removeAttrs' expects a list as its second argument"); + + for (ATermIterator i(list); i; ++i) + /* It's not an error for *i not to exist. */ + attrs.remove(evalString(state, *i)); + + return makeAttrs(attrs); +} + + void EvalState::addPrimOps() { addPrimOp("true", 0, primTrue); @@ -460,6 +477,7 @@ void EvalState::addPrimOps() addPrimOp("isNull", 1, primIsNull); addPrimOp("map", 2, primMap); + addPrimOp("removeAttrs", 2, primRemoveAttrs); } |