implement std.get function (#202)

databricks · Mar 19, 2024 · 871d462 · 871d462
1 parent f385bca
commit 871d462
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/sjsonnet/src/sjsonnet/Std.scala b/sjsonnet/src/sjsonnet/Std.scala
@@ -1227,7 +1227,19 @@ class Std {
     "asciiUpper" -> AsciiUpper,
     "asciiLower" -> AsciiLower,
     "trace" -> Trace,
-    "extVar" -> ExtVar
+    "extVar" -> ExtVar,
+    builtinWithDefaults("get", "o" -> null, "f" -> null, "default" -> Val.Null(dummyPos), "inc_hidden" -> Val.True(dummyPos)) { (args, pos, ev) =>
+      val obj = args(0).asObj
+      val k = args(1).asString
+      val incHidden = args(3).asBoolean
+      if (incHidden && obj.containsKey(k)) {
+        obj.value(k, pos.noOffset, obj)(ev)
+      } else if (!incHidden && obj.containsVisibleKey(k)) {
+        obj.value(k, pos.noOffset, obj)(ev)
+      } else {
+        args(2)
+      }
+    }
   )
   val Std = Val.Obj.mk(
     null,

diff --git a/sjsonnet/test/src/sjsonnet/Std0150FunctionsTests.scala b/sjsonnet/test/src/sjsonnet/Std0150FunctionsTests.scala
@@ -158,5 +158,13 @@ object Std0150FunctionsTests extends TestSuite {
       eval("""std.reverse([1])""") ==> ujson.Arr(1)
       eval("""std.reverse(["1", true, null])""") ==> ujson.Arr(ujson.Null, true, "1")
     }
+
+    test("get"){
+      eval("""std.get({a: 1}, "a")""") ==> ujson.Num(1)
+      eval("""std.get({a:: 1}, "a")""") ==> ujson.Num(1)
+      eval("""std.get({a: 1}, "b")""") ==> ujson.Null
+      eval("""std.get({a: 1}, "b", default=2)""") ==> ujson.Num(2)
+      eval("""std.get({a:: 1}, "a", inc_hidden=false)""") ==> ujson.Null
+    }
   }
 }