* toFile: maintain the references.

1 files changed, 4 insertions, 10 deletions

diff --git a/src/libexpr/primops.cc b/src/libexpr/primops.cc
index 19d233922..d5708f96b 100644
--- a/src/libexpr/primops.cc
+++ b/src/libexpr/primops.cc
@@ -370,17 +370,11 @@ static Expr primToFile(EvalState & state, const ATermVector & args)
 
     PathSet refs;
 
-#if 0
-    /* !!! */
-    for (ATermIterator i(context); i; ++i) {
-        ATerm s;
-        if (matchPath(*i, s)) {
-            assert(isStorePath(aterm2String(s)));
-            refs.insert(aterm2String(s));
-        }
-        else throw EvalError("in `toFile': the file cannot contain references to derivation outputs");
+    for (PathSet::iterator i = context.begin(); i != context.end(); ++i) {
+        if (isDerivation(*i))
+            throw EvalError("in `toFile': the file cannot refer to derivation outputs");
+        refs.insert(*i);
     }
-#endif    
     
     Path storePath = addTextToStore(name, contents, refs);