object/hash: add get method with default if key not exists

Flipez · Dec 4, 2022 · b741848 · b741848
1 parent 917b487
commit b741848
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/object/hash.go b/object/hash.go
@@ -122,6 +122,25 @@ func init() {
 				return NewArray(values)
 			},
 		},
+		"get": ObjectMethod{
+			Layout: MethodLayout{
+				ArgPattern: Args(
+					Arg(ANY_OBJ...),
+					Arg(ANY_OBJ...),
+				),
+				ReturnPattern: Args(
+					Arg(ANY_OBJ...),
+				),
+			},
+			method: func(o Object, args []Object, _ Environment) Object {
+				h := o.(*Hash)
+				k := args[0].(Hashable)
+				if pair, ok := h.Pairs[k.HashKey()]; ok {
+					return pair.Value
+				}
+				return args[1]
+			},
+		},
 	}
 }
 

diff --git a/object/hash_test.go b/object/hash_test.go
@@ -31,6 +31,8 @@ func TestHashObjectMethods(t *testing.T) {
 		{`{"a": "b"}.to_json()`, `{"a":"b"}`},
 		{`{1: "b"}.to_json()`, `{"1":"b"}`},
 		{`{true: "b"}.to_json()`, `{"true":"b"}`},
+		{`{"a": 1, "b": 2}.get("a", 10)`, `1`},
+		{`{"a": 1, "b": 2}.get("c", 10)`, `10`},
 	}
 
 	testInput(t, tests)