Define some array functions

packages/rose/package.json

@@ -10,6 +10,7 @@
   "exports": {
     ".": "./dist/index.js",
     "./ad": "./dist/ad.js",
+    "./array": "./dist/array.js",
     "./num": "./dist/num.js",
     "./prelude": "./dist/prelude.js"
   },

packages/rose/src/array.rose

@@ -0,0 +1,18 @@
+use { + } from "./num"
+
+val T => init : u32 => []T
+
+pub val T => len : []T => u32
+
+def T => suffix (f : u32 -> T) (n : u32) (xs : []T) (i : u32) : () =
+  if i < n then
+    let _ = xs i = f i
+    in return suffix f n xs (i + 1)
+  else ()
+
+pub def T => new (n : u32) (f : u32 -> T) : []T =
+  let xs = init n
+  let _ = suffix f n xs 0
+  in xs
+
+pub def A B => map (f : A -> B) (xs : []A) : []B = new (len xs) (\i => f (xs i))

packages/rose/src/prelude.rose

@@ -1 +1,2 @@
+pub use { len, map, new } from "./array"
 pub use { +, -, *, / } from "./num"

packages/vscode/syntaxes/rose.tmLanguage.json

@@ -35,11 +35,11 @@
       "patterns": [
         {
           "name": "keyword.control.rose",
-          "match": "\\b(else|from|if|pub|then|use)\\b"
+          "match": "\\b(else|from|if|pub|return|then|use)\\b"
         },
         {
           "name": "keyword.other.rose",
-          "match": "\\b(and|do|in|or)\\b"
+          "match": "\\b(and|in|or)\\b"
         },
         {
           "name": "storage.type.rose",