From efaed91a6007772acf066e7876c06462f4e68fd4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Roman=20Smr=C5=BE?= <roman.smrz@seznam.cz>
Date: Wed, 10 Aug 2022 20:40:09 +0200
Subject: Let statement

---
 src/Main.hs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/Main.hs')

diff --git a/src/Main.hs b/src/Main.hs
index dc5ffd4..7c6d587 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -288,7 +288,7 @@ expect (SourceLine sline) p expr vars = do
              forM_ vars $ \name -> do
                  cur <- gets (lookup name . tsVars)
                  when (isJust cur) $ do
-                     outLine OutputMatchFail (Just $ procName p) $ T.pack "variable '" `T.append` textVarName name `T.append` T.pack "' already exists on " `T.append` sline
+                     outLine OutputError (Just $ procName p) $ T.pack "variable '" `T.append` textVarName name `T.append` T.pack "' already exists on " `T.append` sline
                      throwError ()
 
              modify $ \s -> s { tsVars = zip vars capture ++ tsVars s }
@@ -336,6 +336,14 @@ runTest out opts test = do
         oldHandler <- liftIO $ installHandler processStatusChanged (CatchInfo sigHandler) Nothing
 
         flip catchError (const $ return ()) $ forM_ (testSteps test) $ \case
+            Let (SourceLine sline) name expr -> do
+                 cur <- gets (lookup name . tsVars)
+                 when (isJust cur) $ do
+                     outLine OutputError Nothing $ T.pack "variable '" `T.append` textVarName name `T.append` T.pack "' already exists on " `T.append` sline
+                     throwError ()
+                 value <- eval expr
+                 modify $ \s -> s { tsVars = (name, value) : tsVars s }
+
             Spawn pname nname -> do
                 node <- getNode net nname
                 void $ spawnOn (Right node) pname Nothing $
-- 
cgit v1.2.3