From aa161777510728c94e51e9ef1d6df4146d7d952b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Mon, 24 Feb 2025 22:08:35 +0100
Subject: Scalar unification for variables in expansion

---
 src/Parser/Core.hs | 9 +++++++++
 src/Parser/Expr.hs | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/Parser/Core.hs b/src/Parser/Core.hs
index a0ba229..5fb2139 100644
--- a/src/Parser/Core.hs
+++ b/src/Parser/Core.hs
@@ -97,6 +97,15 @@ lookupVarExpr off sline name = do
         ExprTypeVar tvar -> return $ SomeExpr $ DynVariable tvar sline fqn
         ExprTypeFunction args (_ :: Proxy a) -> return $ SomeExpr $ (FunVariable args sline fqn :: Expr (FunctionType a))
 
+lookupScalarVarExpr :: Int -> SourceLine -> VarName -> TestParser SomeExpr
+lookupScalarVarExpr off sline name = do
+    ( fqn, etype ) <- lookupVarType off name
+    case etype of
+        ExprTypePrim (Proxy :: Proxy a) -> return $ SomeExpr $ (Variable sline fqn :: Expr a)
+        ExprTypeVar tvar -> return $ SomeExpr $ DynVariable tvar sline fqn
+        ExprTypeFunction args (pa :: Proxy a) -> do
+            SomeExpr <$> unifyExpr off pa (FunVariable args sline fqn :: Expr (FunctionType a))
+
 unify :: Int -> SomeExprType -> SomeExprType -> TestParser SomeExprType
 unify _ (ExprTypeVar aname) (ExprTypeVar bname) | aname == bname = do
     cur <- gets testTypeUnif
diff --git a/src/Parser/Expr.hs b/src/Parser/Expr.hs
index bc16149..9966d6f 100644
--- a/src/Parser/Expr.hs
+++ b/src/Parser/Expr.hs
@@ -89,7 +89,7 @@ someExpansion = do
         [do off <- stateOffset <$> getParserState
             sline <- getSourceLine
             name <- VarName . TL.toStrict <$> takeWhile1P Nothing (\x -> isAlphaNum x || x == '_')
-            lookupVarExpr off sline name
+            lookupScalarVarExpr off sline name
         , between (char '{') (char '}') someExpr
         ]
 
-- 
cgit v1.2.3