55 lines
1.6 KiB
Haskell
55 lines
1.6 KiB
Haskell
module REPL where
|
|
|
|
import Eval
|
|
import Lexer
|
|
import Parser
|
|
import Research
|
|
|
|
import Control.Exception (SomeException, catch)
|
|
import Control.Monad.IO.Class (liftIO)
|
|
import Data.List (intercalate)
|
|
import System.Console.Haskeline
|
|
|
|
import qualified Data.Map as Map
|
|
|
|
repl :: Map.Map String T -> IO ()
|
|
repl env = runInputT defaultSettings (loop env)
|
|
where
|
|
loop :: Map.Map String T -> InputT IO ()
|
|
loop env = do
|
|
minput <- getInputLine "tricu < "
|
|
case minput of
|
|
Nothing -> outputStrLn "Goodbye!"
|
|
Just ":_exit" -> outputStrLn "Goodbye!"
|
|
Just "" -> do
|
|
outputStrLn ""
|
|
loop env
|
|
Just input -> do
|
|
newEnv <- liftIO $ (processInput env input `catch` errorHandler env)
|
|
loop newEnv
|
|
|
|
processInput :: Map.Map String T -> String -> IO (Map.Map String T)
|
|
processInput env input = do
|
|
let clearEnv = Map.delete "__result" env
|
|
newEnv = evalSingle clearEnv (parseSingle input)
|
|
case Map.lookup "__result" newEnv of
|
|
Just r -> do
|
|
putStrLn $ "tricu > " ++ show r
|
|
putStrLn $ "READ -: \"" ++ decodeResult r ++ "\""
|
|
Nothing -> return ()
|
|
return newEnv
|
|
|
|
errorHandler :: Map.Map String T -> SomeException -> IO (Map.Map String T)
|
|
errorHandler env e = do
|
|
putStrLn $ "Error: " ++ show e
|
|
return env
|
|
|
|
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 _ -> ""
|