diff options
Diffstat (limited to 'src/Builtins.hs')
| -rw-r--r-- | src/Builtins.hs | 87 |
1 files changed, 45 insertions, 42 deletions
diff --git a/src/Builtins.hs b/src/Builtins.hs index f8d1567..ebaad42 100644 --- a/src/Builtins.hs +++ b/src/Builtins.hs @@ -37,13 +37,16 @@ builtinEnv = M.fromList [ ] argError1 :: String -> AST -> String -argError1 fn arg = pos arg ++ ": invalid argument to " ++ fn ++ ": " ++ show arg +argError1 fn arg1 = + "invalid argument to " ++ fn ++ ": " ++ show arg1 -argError2 :: (Show a1, Show a2) => String -> a1 -> a2 -> String -argError2 fn arg1 arg2 = "invalid arguments to " ++ fn ++ ": " ++ show arg1 ++ ", " ++ show arg2 +argError2 :: String -> AST -> AST -> String +argError2 fn arg1 arg2 = + "invalid arguments to " ++ fn ++ ": " ++ show arg1 ++ ", " ++ show arg2 -argError3 :: (Show a1, Show a2, Show a3) => String -> a1 -> a2 -> a3 -> String -argError3 fn arg1 arg2 arg3 = "invalid arguments to " ++ fn ++ ": " ++ show arg1 ++ ", " ++ show arg2 ++ ", " ++ show arg3 +argError3 :: String -> AST -> AST -> AST -> String +argError3 fn arg1 arg2 arg3 = + "invalid arguments to " ++ fn ++ ": " ++ show arg1 ++ ", " ++ show arg2 ++ ", " ++ show arg3 -- BUILTINS @@ -54,13 +57,13 @@ builtinAdd2 = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTInteger b } = return $ makeNonsenseAST $ ASTInteger $ a + b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 fn1 _ ast1@AST { astNode = ASTDouble a } = return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTDouble b } = return $ makeNonsenseAST $ ASTDouble $ a + b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinSubtract2 :: (String, AST) builtinSubtract2 = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -69,13 +72,13 @@ builtinSubtract2 = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTInteger b } = return $ makeNonsenseAST $ ASTInteger $ a - b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 fn1 _ ast1@AST { astNode = ASTDouble a } = return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTDouble b } = return $ makeNonsenseAST $ ASTDouble $ a - b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinMultiply2 :: (String, AST) builtinMultiply2 = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -84,30 +87,30 @@ builtinMultiply2 = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTInteger b } = return $ makeNonsenseAST $ ASTInteger $ a * b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 fn1 _ ast1@AST { astNode = ASTDouble a } = return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTDouble b } = return $ makeNonsenseAST $ ASTDouble $ a * b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinDivide2 :: (String, AST) builtinDivide2 = (name, makeNonsenseAST $ ASTFunction fn1) where name = "/" fn1 _ ast1@AST { astNode = ASTInteger a } = return $ makeNonsenseAST $ ASTFunction $ fn2 where - fn2 _ AST { astNode = ASTInteger b } = - do when (b == 0) $ throwL $ "division by zero" + fn2 _ ast2@AST { astNode = ASTInteger b } = + do when (b == 0) $ throwL (astPos ast2) $ "division by zero" return $ makeNonsenseAST $ ASTInteger $ a `div` b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 fn1 _ ast1@AST { astNode = ASTDouble a } = return $ makeNonsenseAST $ ASTFunction $ fn2 where - fn2 _ AST { astNode = ASTDouble b } = - do when (b == 0) $ throwL $ "division by zero" + fn2 _ ast2@AST { astNode = ASTDouble b } = + do when (b == 0) $ throwL (astPos ast2) $ "division by zero" return $ makeNonsenseAST $ ASTDouble $ a / b - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinEq2 :: (String, AST) builtinEq2 = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -127,13 +130,13 @@ builtinFloor :: (String, AST) builtinFloor = (name, makeNonsenseAST $ ASTFunction fn1) where name = "floor" fn1 _ AST { astNode = ASTDouble dbl } = return $ makeNonsenseAST $ ASTInteger $ floor dbl - fn1 _ ast = throwL $ argError1 name ast + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinToDouble :: (String, AST) builtinToDouble = (name, makeNonsenseAST $ ASTFunction fn1) where name = "to-double" fn1 _ AST { astNode = ASTInteger int } = return $ makeNonsenseAST $ ASTDouble $ fromIntegral int - fn1 _ ast = throwL $ argError1 name ast + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinFmt :: (String, AST) builtinFmt = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -143,8 +146,8 @@ builtinFmt = (name, makeNonsenseAST $ ASTFunction fn1) where fn2 _ AST { astNode = ASTVector replacements } = return $ makeNonsenseAST $ ASTString $ T.unpack $ replaceAll (0 :: Int) replacements (T.pack str) - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 replaceAll _ [] text = text replaceAll n (x:xs) text = let text' = T.replace (T.pack $ "{" ++ show n ++ "}") (T.pack $ show x) text @@ -153,18 +156,18 @@ builtinFmt = (name, makeNonsenseAST $ ASTFunction fn1) where builtinHead :: (String, AST) builtinHead = (name, makeNonsenseAST $ ASTFunction fn1) where name = "head" - fn1 _ AST { astNode = ASTVector vec } = - do when (length vec == 0) $ throwL $ name ++ " of empty vector" + fn1 _ ast1@AST { astNode = ASTVector vec } = + do when (length vec == 0) $ throwL (astPos ast1) $ name ++ " of empty vector" return $ head vec - fn1 _ ast = throwL $ argError1 name ast + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinTail :: (String, AST) builtinTail = (name, makeNonsenseAST $ ASTFunction fn1) where name = "tail" - fn1 _ AST { astNode = ASTVector vec } = - do when (length vec == 0) $ throwL $ name ++ " of empty vector" + fn1 _ ast1@AST { astNode = ASTVector vec } = + do when (length vec == 0) $ throwL (astPos ast1) $ name ++ " of empty vector" return $ makeNonsenseAST $ ASTVector $ tail vec - fn1 _ ast = throwL $ argError1 name ast + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinSubstr :: (String, AST) builtinSubstr = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -175,9 +178,9 @@ builtinSubstr = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn3 where fn3 _ AST { astNode = ASTString str } = return $ makeNonsenseAST $ ASTString $ drop at .> take len $ str - fn3 _ ast3 = throwL $ argError3 name ast1 ast2 ast3 - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn3 _ ast3 = throwL (astPos ast3) $ argError3 name ast1 ast2 ast3 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinPrepend :: (String, AST) builtinPrepend = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -186,7 +189,7 @@ builtinPrepend = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTVector vec } = return $ makeNonsenseAST $ ASTVector $ ast1 : vec - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 builtinPrint :: (String, AST) builtinPrint = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -194,7 +197,7 @@ builtinPrint = (name, makeNonsenseAST $ ASTFunction fn1) where fn1 _ AST { astNode = ASTString str } = do liftIO $ putStr $ str return $ makeNonsenseAST ASTUnit - fn1 _ ast = throwL $ argError1 name ast + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinConcat :: (String, AST) builtinConcat = (name, makeNonsenseAST $ ASTFunction fn1) where @@ -203,13 +206,13 @@ builtinConcat = (name, makeNonsenseAST $ ASTFunction fn1) where return $ makeNonsenseAST $ ASTFunction $ fn2 where fn2 _ AST { astNode = ASTString str2 } = return $ makeNonsenseAST $ ASTString $ str1 ++ str2 - fn2 _ ast2 = throwL $ argError2 name ast1 ast2 - fn1 _ ast1 = throwL $ argError1 name ast1 + fn2 _ ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2 + fn1 _ ast1 = throwL (astPos ast1) $ argError1 name ast1 builtinFatal :: (String, AST) builtinFatal = (name, makeNonsenseAST $ ASTFunction fn1) where name = "fatal" - fn1 _ AST { astNode = ASTString str } = - throwL $ str - fn1 _ ast = - throwL $ argError1 name ast + fn1 _ ast1@AST { astNode = ASTString str } = + throwL (astPos ast1) $ str + fn1 _ ast1 = + throwL (astPos ast1) $ argError1 name ast1 |
