stdlib: Implement last() and nth() functions.

[goals.git] / stdlib / prelude.gl

diff --git a/stdlib/prelude.gl b/stdlib/prelude.gl
index e2d0400..13ac311 100644 (file)
--- a/stdlib/prelude.gl
+++ b/stdlib/prelude.gl
@@ -40,6 +40,11 @@ tactic *exists (filename) = @{
 #----------------------------------------------------------------------
 # Basic functions.
 
+function error (msg) = @{
+    echo %msg >&2
+    exit 1
+}
+
 # Wrap list of strings in a call or tactic.
 pure function wrap (wrapper, xs) = @{
     echo '['
@@ -54,6 +59,16 @@ pure function wrap (wrapper, xs) = @{
 #----------------------------------------------------------------------
 # Text functions.
 
+# Filter a list by regexp.
+pure function filter (pat, xs) returning strings = @{
+    for f in %xs; do echo "$f"; done | grep -E -- %pat
+}
+
+# Filter out a list by regexp.
+pure function filter-out (pat, xs) returning strings = @{
+    for f in %xs; do echo "$f"; done | grep -v -E -- %pat
+}
+
 # Head of a list.
 pure function head (xs) returning string = @{
     for f in %xs; do
@@ -67,6 +82,25 @@ pure function join (xs, ys) returning strings = @{
     for f in %xs %ys; do echo "$f"; done
 }
 
+# Last element of a list.
+pure function last (xs) returning string = @{
+    for f in %xs; do
+        r="$f"
+    done
+    echo "$r"
+}
+
+# n'th element of a list.
+pure function nth (n, xs) returning string = @{
+    i=0
+    r=
+    for f in %xs; do
+        if [ $i -eq %n ]; then r="$f"; fi
+        ((++i))
+    done
+    echo "$r"
+}
+
 # Sort + uniq a list.
 pure function sort (xs) returning strings = @{
     for f in %xs; do echo "$f"; done | sort -u
@@ -93,12 +127,12 @@ pure function tail (xs) returning strings = @{
 
 # Base name.
 pure function basename (name) returning string = @{
-    basename %name
+    basename -- %name
 }
 
 # Directory name.
 pure function dirname (name) returning string = @{
-    dirname %name
+    dirname -- %name
 }
 
 # File extension.
@@ -107,6 +141,16 @@ pure function extension (name) returning string = @{
     echo "${name##*.}"
 }
 
+# Read a file.
+function read (filename) returning string = @{
+    cat -- %filename
+}
+
+# Read a file as a list of lines.
+function readlines (filename) returning strings = @{
+    cat -- %filename
+}
+
 # Real path.
 function realpath (filename) returning string = @{
     realpath -- %filename