Fix critical list evaluation bug and REPL updates

src/Eval.hs

@@ -23,7 +23,7 @@ evalSingle env term = case term of
   SVar name ->
     case Map.lookup name env of
       Just value -> Map.insert "__result" value env
-      Nothing -> error $ "Variable " ++ name ++ " not defined"
+      Nothing -> errorWithoutStackTrace $ "Variable " ++ name ++ " not defined"
   _ ->
     let result = evalAST env term
     in Map.insert "__result" result env
@@ -46,19 +46,18 @@ evalAST :: Map String T -> TricuAST -> T
 evalAST env term = case term of
   SVar name -> case Map.lookup name env of
     Just value -> value
-    Nothing -> error $ "Variable " ++ name ++ " not defined"
+    Nothing -> errorWithoutStackTrace $ "Variable " ++ name ++ " not defined"
   TLeaf -> Leaf
   TStem t -> Stem (evalAST env t)
   TFork t1 t2 -> Fork (evalAST env t1) (evalAST env t2)
   SApp t1 t2 -> apply (evalAST env t1) (evalAST env t2)
   SStr str -> ofString str
   SInt num -> ofNumber num
-  SList elems -> ofList (map (evalAST Map.empty) elems)
+  SList elems -> ofList (map (evalAST env) elems)
   SEmpty -> Leaf
   SFunc name args body ->
-    error $ "Unexpected function definition " ++ name
-    ++ " in evalAST; define via evalSingle."
-  SLambda {} -> error "Internal error: SLambda found in evalAST after elimination."
+    errorWithoutStackTrace $ "Unexpected function definition " ++ name
+  SLambda {} -> errorWithoutStackTrace "Internal error: SLambda found in evalAST after elimination."
 
 eliminateLambda :: TricuAST -> TricuAST
 eliminateLambda (SLambda (v:vs) body)
@@ -115,4 +114,4 @@ tS = SApp (SApp TLeaf (SApp TLeaf (SApp (SApp TLeaf TLeaf) TLeaf))) TLeaf
 result :: Map String T -> T
 result r = case Map.lookup "__result" r of
   Just a -> a
-  Nothing -> error "No __result field found in provided environment"
+  Nothing -> errorWithoutStackTrace "No __result field found in provided environment"

src/FileEval.hs

@@ -15,7 +15,7 @@ evaluateFileResult filePath = do
   let finalEnv = evalTricu Map.empty asts
   case Map.lookup "__result" finalEnv of
     Just finalResult -> return finalResult
-    Nothing -> error "No result found in final environment"
+    Nothing -> errorWithoutStackTrace "No expressions to evaluate found"
 
 evaluateFile :: FilePath -> IO Env
 evaluateFile filePath = do

src/Lexer.hs

@@ -90,5 +90,5 @@ tricuLexer = do
 
 lexTricu :: String -> [LToken]
 lexTricu input = case runParser tricuLexer "" input of
-  Left err -> error $ "Lexical error:\n" ++ errorBundlePretty err
+  Left err -> errorWithoutStackTrace $ "Lexical error:\n" ++ errorBundlePretty err
   Right tokens -> tokens

src/Main.hs

@@ -6,7 +6,7 @@ import Parser                 (parseTricu)
 import REPL
 import Research
 
-import Control.Monad 					(foldM)
+import Control.Monad          (foldM)
 import Control.Monad.IO.Class (liftIO)
 import Text.Megaparsec        (runParser)
 import System.Console.CmdArgs
@@ -61,7 +61,7 @@ main = do
       putStrLn "Welcome to the tricu REPL"
       putStrLn "You can exit with `CTRL+D` or the `:_exit` command.`"
       library <- liftIO $ evaluateFile "./lib/base.tri"
-      repl library
+      repl $ Map.delete "__result" library
     Evaluate { file = filePaths, form = form } -> do
       result <- case filePaths of
         [] -> do

src/REPL.hs

@@ -8,7 +8,8 @@ import Research
 
 import Control.Exception         (SomeException, catch)
 import Control.Monad.IO.Class    (liftIO)
-import Data.List                 (intercalate)
+import Data.Char                 (isSpace)
+import Data.List                 (dropWhile, dropWhileEnd, intercalate)
 import System.Console.Haskeline
 
 import qualified Data.Map as Map
@@ -20,23 +21,27 @@ repl env = runInputT defaultSettings (loop env)
     loop env = do
       minput <- getInputLine "tricu < "
       case minput of
-        Nothing -> outputStrLn "Goodbye!"
-        Just ":_exit" -> outputStrLn "Goodbye!"
-        Just ":_load" -> do
-          path <- getInputLine "File path to load < "
-          case path of
-            Nothing -> do
-              outputStrLn "No input received; stopping import."
-              loop env
-            Just path -> do
-              loadedEnv <- liftIO $ evaluateFile path
-              loop $ Map.union loadedEnv env
-        Just "" -> do
-          outputStrLn ""
-          loop env
-        Just input -> do
-          newEnv <- liftIO $ (processInput env input `catch` errorHandler env)
-          loop newEnv
+        Nothing -> outputStrLn "Exiting tricu"
+        Just s -> case strip s of 
+          "!exit" -> outputStrLn "Exiting tricu"
+          "!load" -> do
+            path <- getInputLine "File path to load < "
+            case path of
+              Nothing -> do
+                outputStrLn "No input received; stopping import."
+                loop env
+              Just path -> do
+                loadedEnv <- liftIO $ evaluateFileWithContext env (strip path)
+                loop $ Map.delete "__result" (Map.union loadedEnv env)
+          "" -> do
+            outputStrLn ""
+            loop env
+          input -> do
+            case (take 2 input) of 
+              "--" -> loop env
+              _ -> do
+                newEnv <- liftIO $ (processInput env input `catch` errorHandler env)
+                loop newEnv
   
     processInput :: Env -> String -> IO Env
     processInput env input = do
@@ -44,8 +49,7 @@ repl env = runInputT defaultSettings (loop env)
           newEnv = evalTricu env asts
       case Map.lookup "__result" newEnv of
         Just r -> do
-          putStrLn $ "tricu > " ++ show r
-          putStrLn $ "READ -: \"" ++ decodeResult r ++ "\""
+          putStrLn $ "tricu > " ++ decodeResult r
         Nothing -> return ()
       return newEnv
     
@@ -53,12 +57,13 @@ repl env = runInputT defaultSettings (loop env)
     errorHandler env e = do
       putStrLn $ "Error: " ++ show e
       return env
+    
+    strip :: String -> String
+    strip = dropWhileEnd isSpace . dropWhile isSpace
 
 decodeResult :: T -> String
 decodeResult tc = case toNumber tc of
   Right num -> show num
   Left _ -> case toString tc of
-    Right str -> str
-    Left _ -> case toList tc of
-      Right list -> "[" ++ intercalate ", " (map decodeResult list) ++ "]"
-      Left _ -> ""
+    Right str -> "\"" ++ str ++ "\""
+    Left _ -> formatResult TreeCalculus tc