[Linter] Support std.member and std.repeat.

Author: sbarzowski

linter/internal/types/stdlib.go

@@ -114,13 +114,17 @@ func prepareStdlib(g *typeGraph) {
 
 		"makeArray":     g.newSimpleFuncType(anyArrayType, "sz", "func"),
 		"count":         g.newSimpleFuncType(numberType, "arr", "x"),
+		"member":        g.newSimpleFuncType(numberType, "arr", "x"),
 		"find":          g.newSimpleFuncType(arrayOfNumber, "value", "arr"),
 		"map":           g.newSimpleFuncType(anyArrayType, "func", "arr"),
 		"mapWithIndex":  g.newSimpleFuncType(anyArrayType, "func", "arr"),
 		"filterMap":     g.newSimpleFuncType(anyArrayType, "filter_func", "map_func", "arr"),
+		"flatMap":       g.newSimpleFuncType(anyArrayType, "func", "arr"),
 		"filter":        g.newSimpleFuncType(anyArrayType, "func", "arr"),
 		"foldl":         g.newSimpleFuncType(anyType, "func", "arr", "init"),
 		"foldr":         g.newSimpleFuncType(anyType, "func", "arr", "init"),
+		"repeat":        g.newSimpleFuncType(anyArrayType, "what", "count"),
+		"slice":         g.newSimpleFuncType(arrayOfString, "indexable", "index", "end", "step"),
 		"range":         g.newSimpleFuncType(arrayOfNumber, "from", "to"),
 		"join":          g.newSimpleFuncType(stringOrArray, "sep", "arr"),
 		"lines":         g.newSimpleFuncType(arrayOfString, "arr"),
@@ -155,9 +159,7 @@ func prepareStdlib(g *typeGraph) {
 		"manifestJson":     g.newSimpleFuncType(stringType, "value"),
 		"objectHasEx":      g.newSimpleFuncType(boolType, "obj", "fname", "hidden"),
 		"objectFieldsEx":   g.newSimpleFuncType(arrayOfString, "obj", "hidden"),
-		"flatMap":          g.newSimpleFuncType(anyArrayType, "func", "arr"),
 		"modulo":           g.newSimpleFuncType(numberType, "x", "y"),
-		"slice":            g.newSimpleFuncType(arrayOfString, "indexable", "index", "end", "step"),
 		"primitiveEquals":  g.newSimpleFuncType(boolType, "x", "y"),
 		"mod":              g.newSimpleFuncType(stringOrNumber, "a", "b"),
 		"native":           g.newSimpleFuncType(anyFunctionType, "x"),

testdata/stdlib_smoke_test.golden

@@ -53,6 +53,14 @@
       0,
       5
    ],
+   "flatMap": [
+      2,
+      3,
+      4,
+      6,
+      6,
+      9
+   ],
    "flattenArrays": [
       1,
       2,
@@ -118,6 +126,7 @@
    },
    "max": 3,
    "md5": "1bc29b36f623ba82aaf6724fd3b16718",
+   "member": true,
    "mergePatch": { },
    "min": 2,
    "objectFields": [ ],
@@ -145,6 +154,7 @@
       4,
       5
    ],
+   "repeat": "foofoofoo",
    "set": [
       [
          1,
@@ -197,6 +207,7 @@
    ],
    "sign": 1,
    "sin": -0.95892427466313845,
+   "slice": "o",
    "sort": [
       [
          1,

testdata/stdlib_smoke_test.jsonnet

@@ -101,12 +101,16 @@
     makeArray: std.makeArray(sz=5, func=function(i) i),
     count: std.count(arr=["a", "b", "c", "c"], x="b"),
     find: std.find(value=42, arr=[1, 2, 42, 3, 42]),
+    member: std.member(arr=[1, 2, 3], x=2),
     map: std.map(func=function(x) -x, arr=[1, 2, 3]),
     mapWithIndex: std.mapWithIndex(func=function(i, x) i + x, arr=[3, 2, 1]),
     filterMap: std.filterMap(filter_func=function(x) x % 2 == 0, map_func=function(x) x * 2, arr=[1, 2, 3, 4]),
+    flatMap: std.flatMap(func=function(x) [x*2, x*3], arr=[1,2,3]),
     filter: std.filter(func=function(x) x % 2 == 0, arr=[1, 2, 3, 4]),
     foldl: std.foldl(func=function(x, y) x + y, arr=[[1], [2], [3]], init=[0]),
     foldr: std.foldr(func=function(x, y) x + y, arr=[[1], [2], [3]], init=[4]),
+    repeat: std.repeat(what="foo", count=3),
+    slice: std.slice(indexable="foobar", index=1, end=2, step=1),
     range: std.range(from=1, to=5),
     join:  std.join(sep=",", arr=["a", "b", "c"]),
     lines: std.lines(arr=["a", "b", "c"]),