From e73ad00f44479e3065ccf12d07ca3ae22ca1f3f0 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Tue, 6 Dec 2022 15:40:52 +0200
Subject: Remove ! from end of builtin top-level functions

---
 src/Builtins.hs | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

(limited to 'src/Builtins.hs')

diff --git a/src/Builtins.hs b/src/Builtins.hs
index b731e66..7ad966a 100644
--- a/src/Builtins.hs
+++ b/src/Builtins.hs
@@ -4,6 +4,7 @@ module Builtins where
 import qualified Data.Map as M
 import qualified Data.Text as T
 import qualified Text.Read as TR
+import qualified Data.List as L
 import Control.Monad.Except
 import Utils
 
@@ -26,6 +27,7 @@ builtinEnv = M.fromList [
     builtinHead,
     builtinTail,
     builtinPrepend,
+    builtinSortByFirst,
     -- string operations
     builtinSubstr,
     builtinStrToVec,
@@ -45,10 +47,10 @@ builtinEnv = M.fromList [
     builtinKind,
     -- reserved keywords
     reservedKeyword "\\",
-    reservedKeyword "let!",
+    reservedKeyword "let",
     reservedKeyword "match",
-    reservedKeyword "env!",
-    reservedKeyword "record!"
+    reservedKeyword "Debug/env",
+    reservedKeyword "record"
     ]
 
 argError1 :: String -> AST -> String
@@ -308,3 +310,21 @@ builtinAppendFile = (name, makeNonsenseAST $ ASTFunction False fn1) where
                     Nothing -> throwL (astPos ast1) $ "failed to append to file: " ++ filePath
             fn2 ast2 = throwL (astPos ast2) $ argError2 name ast1 ast2
     fn1 ast1 = throwL (astPos ast1) $ argError1 name ast1
+
+builtinSortByFirst :: (String, AST)
+builtinSortByFirst = (name, makeNonsenseAST $ ASTFunction True fn1) where
+    name = "sort-by-first"
+    fn1 ast1@AST { astNode = ASTVector elems } = do
+        pairs <- mapM elemToPair elems
+        let sorted = L.sortBy (\(a, _) (b, _) -> compare a b) pairs
+        let sortedASTS = map (\(k, v) -> makeNonsenseAST $
+                ASTVector [makeNonsenseAST $ ASTInteger k, v]) sorted
+        return $ makeNonsenseAST $ ASTVector sortedASTS where
+            itemsToPair [AST { astNode = ASTInteger k }, v] =
+                return $ (k, v)
+            itemsToPair items = throwL (astPos ast1) $
+                "invalid element in vector supplied to sort-by-first: " ++ show items
+            elemToPair AST { astNode = ASTVector items } =
+                itemsToPair items
+            elemToPair ast2 = throwL (astPos ast2) $ argError1 name ast1
+    fn1 ast1 = throwL (astPos ast1) $ argError1 name ast1
-- 
cgit v1.3