Evaluate functions in parameters via unifyExpr

1 files changed, 4 insertions, 5 deletions

diff --git a/src/Parser/Expr.hs b/src/Parser/Expr.hs
index d59e0b2..bc16149 100644
--- a/src/Parser/Expr.hs
+++ b/src/Parser/Expr.hs
@@ -391,18 +391,17 @@ recordSelector (SomeExpr expr) = do
 
 checkFunctionArguments :: FunctionArguments SomeArgumentType
                        -> Int -> Maybe ArgumentKeyword -> SomeExpr -> TestParser SomeExpr
-checkFunctionArguments (FunctionArguments argTypes) poff kw expr = do
+checkFunctionArguments (FunctionArguments argTypes) poff kw sexpr@(SomeExpr expr) = do
     case M.lookup kw argTypes of
         Just (SomeArgumentType (_ :: ArgumentType expected)) -> do
-            withRecovery registerParseError $ do
-                void $ unify poff (ExprTypePrim (Proxy @expected)) (someExprType expr)
-            return expr
+            withRecovery (\e -> registerParseError e >> return sexpr) $ do
+                SomeExpr <$> unifyExpr poff (Proxy @expected) expr
         Nothing -> do
             registerParseError $ FancyError poff $ S.singleton $ ErrorFail $ T.unpack $
                 case kw of
                     Just (ArgumentKeyword tkw) -> "unexpected parameter with keyword `" <> tkw <> "'"
                     Nothing                    -> "unexpected parameter"
-            return expr
+            return sexpr
 
 
 functionArguments :: (Int -> Maybe ArgumentKeyword -> a -> TestParser b) -> TestParser a -> TestParser a -> (Int -> Text -> TestParser a) -> TestParser (FunctionArguments b)