From 694021a9bd1dd0420a8594ea24218c47fbbab51e Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Wed, 5 Oct 2022 14:48:14 +0300
Subject: Refactor Depth into LContext

---
 src/Utils.hs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'src/Utils.hs')

diff --git a/src/Utils.hs b/src/Utils.hs
index 8ee2056..8dd233f 100644
--- a/src/Utils.hs
+++ b/src/Utils.hs
@@ -26,7 +26,8 @@ type Env = M.Map String AST
 
 data LState = LState {
     stateConfig :: Config,
-    stateEnv :: Env
+    stateEnv :: Env,
+    stateDepth :: Int
 }
 
 type LContext a = StateT LState (ExceptT LException IO) a
@@ -53,6 +54,15 @@ insertEnv k v = do
     env <- getEnv
     putEnv $ M.insert k v env
 
+incrementDepth :: LContext ()
+incrementDepth =
+    modify (\s -> s { stateDepth = stateDepth s + 1 })
+
+getDepth :: LContext Int
+getDepth = do
+    s <- get
+    return $ stateDepth s
+
 data Token = Token {
     tokenContent :: String,
     tokenRow :: Int,
-- 
cgit v1.3