From f7d346778118a0eb1b87d83702bab226b5374d8c Mon Sep 17 00:00:00 2001
From: Georges Dubus <georges.dubus@soundcloud.com>
Date: Wed, 31 Jan 2018 18:00:09 +0100
Subject: [PATCH] More WIP

---
 Nix/Eval.hs | 37 ++++++++++++++++++++-----------------
 1 file changed, 20 insertions(+), 17 deletions(-)

diff --git a/Nix/Eval.hs b/Nix/Eval.hs
index 94c8b43..fa3ec21 100644
--- a/Nix/Eval.hs
+++ b/Nix/Eval.hs
@@ -20,6 +20,8 @@ import           Nix.Atoms
 import           Nix.Expr
 import           Prelude hiding (mapM, sequence)
 
+import Debug.Trace
+
 -- | An 'NValue' is the most reduced form of an 'NExpr' after evaluation
 -- is completed.
 data NValueF m r
@@ -38,7 +40,7 @@ instance Show f => Show (NValueF m f) where
       go (NVStr      text) = showsCon1 "NVStr"      text
       go (NVList     list) = showsCon1 "NVList"     list
       go (NVSet     attrs) = showsCon1 "NVSet"      attrs
-      go (NVFunction r _)  = showsCon1 "NVFunction" (() <$ r)
+      go (NVFunction r _)  = showsCon1 "NVFunction" (() <$ r)  -- Discard the default value, that we don't care about displaying
       go (NVLiteralPath p) = showsCon1 "NVLiteralPath" p
       go (NVEnvPath p)     = showsCon1 "NVEnvPath" p
 
@@ -66,19 +68,23 @@ atomText (NBool b)  = if b then "true" else "false"
 atomText NNull      = "null"
 atomText (NUri uri) = uri
 
-buildArgument :: Params (NValue m) -> NValue m -> ValueSet m
-buildArgument paramSpec arg = either error id $ case paramSpec of
-    Param name -> return $ Map.singleton name arg
-    ParamSet (FixedParamSet s) Nothing -> lookupParamSet s
-    ParamSet (FixedParamSet s) (Just name) ->
-      Map.insert name arg <$> lookupParamSet s
-    ParamSet _ _ -> error "Can't yet handle variadic param sets"
+buildArgument :: ValueSet m -> Params (ValueSet m -> NValue m) -> NValue m -> ValueSet m
+buildArgument env paramSpec arg = do
+    let envAndArgs = env `Map.union` result
+        params = fmap ($ envAndArgs) paramSpec
+        result = case params of
+          Param name -> Map.singleton name arg
+          ParamSet (FixedParamSet s) Nothing -> lookupParamSet s
+          --ParamSet (FixedParamSet s) (Just name) ->
+          --  Map.insert name arg <$> lookupParamSet s
+          ParamSet _ _ -> error "Can't yet handle variadic param sets"
+      in result
   where
-    go env k def = maybe (Left err) return $ Map.lookup k env <|> def
+    go env k def = maybe (error err) id $ (Map.lookup k env) <|> def
       where err = "Could not find " ++ show k
     lookupParamSet s = case arg of
-        Fix (NVSet env) -> Map.traverseWithKey (go env) s
-        _               -> Left "Unexpected function environment"
+        Fix (NVSet env) -> Map.mapWithKey (go env) s
+        _               -> error "Unexpected function environment"
 
 -- | Evaluate an nix expression, with a given ValueSet as environment
 evalExpr :: MonadFix m => NExpr -> ValueSet m -> m (NValue m)
@@ -187,16 +193,13 @@ evalExpr = cata phi
         case fun' of
             Fix (NVFunction argset f) -> do
                 arg <- x env
-                let arg' = buildArgument argset arg
+                arg' <- buildArgument env argset arg
                 f arg'
             _ -> error "Attempt to call non-function"
 
     phi (NAbs a b) = \env -> do
-        -- jww (2014-06-28): arglists should not receive the current
-        -- environment, but rather should recursively view their own arg
-        -- set
-        args <- traverse ($ env) a
-        return $ Fix $ NVFunction args b
+      -- TODO wrong anyway. The abs still captures then env when it is defined
+        return $ Fix $ NVFunction a b
 
 evalString :: Monad m
            => ValueSet m -> NString (ValueSet m -> m (NValue m)) -> m Text