lib: intro `prompt`, basic Apache 2.0 service to display a prompt
[nit.git] / lib / json / dynamic.nit
index ec3ba25..68567ff 100644 (file)
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Dynamic interface to read Json strings.
+# Dynamic interface to read values from JSON strings
 #
-# `String::to_json_value` returns a `JsonValue` which can be queried
-# to get the underlying Json data. It can also be used as any Json types.
+# Most services are in `JsonValue`, which is created by `Text::to_json_value`.
 module dynamic
 
+import error
 private import static
-import standard
 
+redef class Text
+       # Parse `self` to a `JsonValue`
+       fun to_json_value: JsonValue do return new JsonValue(parse_json)
+end
+
+# Dynamic wrapper of a JSON value, created by `Text::to_json_value`
+#
+# Provides high-level services to explore JSON objects with minimal overhead
+# dealing with static types. Use this class to manipulate the JSON data first
+# and check for errors only before using the resulting data.
+#
+# For example, given:
+# ~~~
+# var json_src = """
+# {
+#   "i": 123,
+#   "m": {
+#          "t": true,
+#          "f": false
+#        },
+#   "a": ["zero", "one", "two"]
+# }"""
+# var json_value = json_src.to_json_value # Parse src to a `JsonValue`
+# ~~~
+#
+# Access array or map values using their indices.
+# ~~~
+# var target_int = json_value["i"]
+# assert target_int.is_int # Check for error and expected type
+# assert target_int.to_i == 123 # Use the value
+# ~~~
+#
+# Use `get` to reach a value nested in multiple objects.
+# ~~~
+# var target_str = json_value.get("a.0")
+# assert target_str.is_string # Check for error and expected type
+# assert target_str.to_s == "zero" # Use the value
+# ~~~
+#
+# This API is useful for scripts and when you need only a few values from a
+# JSON object. To access many values or check the conformity of the JSON
+# beforehand, use the `json` serialization services.
 class JsonValue
+
+       # The wrapped JSON value.
        var value: nullable Object
 
        # Is this value null?
@@ -41,11 +84,17 @@ class JsonValue
 
        # Get this value as a `Int`
        #
-       # require: `self.is_int`
+       # require: `self.is_numeric`
        #
        #     assert "-10".to_json_value.to_i == -10
        #     assert "123".to_json_value.to_i == 123
-       fun to_i: Int do return value.as(Int)
+       #     assert "123.456".to_json_value.to_i == 123
+       fun to_i: Int
+       do
+               var value = value
+               assert value isa Numeric
+               return value.to_i
+       end
 
        # Is this value a float?
        #
@@ -56,11 +105,17 @@ class JsonValue
 
        # Get this value as a `Float`
        #
-       # require: `self.is_float`
+       # require: `self.is_numeric`
        #
        #     assert "0.0".to_json_value.to_f == 0.0
        #     assert "123.456".to_json_value.to_f == 123.456
-       fun to_f: Float do return value.as(Float)
+       #     assert "123".to_json_value.to_f == 123.0
+       fun to_f: Float
+       do
+               var value = value
+               assert value isa Numeric
+               return value.to_f
+       end
 
        # Is the value numeric?
        #
@@ -74,8 +129,8 @@ class JsonValue
        #
        # require: `self.is_numeric`
        #
-       #     assert "1.234".to_json_value.to_numeric = 1.234
-       #     assert "1234".to_json_value.to_numeric = 1234
+       #     assert "1.234".to_json_value.to_numeric == 1.234
+       #     assert "1234".to_json_value.to_numeric == 1234
        fun to_numeric: Numeric
        do
                if is_int then return to_i
@@ -110,12 +165,8 @@ class JsonValue
        #     assert "\"str\"".to_json_value.to_s == "str"
        #     assert "123".to_json_value.to_s == "123"
        #     assert "true".to_json_value.to_s == "true"
-       #     assert "[1, 2, 3]".to_json_value.to_s == "123"
-       redef fun to_s: String
-       do
-               if value == null then return "null"
-               return value.to_s
-       end
+       #     assert "[1, 2, 3]".to_json_value.to_s == "[1,2,3]"
+       redef fun to_s do return (value or else "null").to_s
 
        ### Objects
 
@@ -123,14 +174,14 @@ class JsonValue
        #
        #     assert """{"a": 123}""".to_json_value.is_map
        #     assert not "123".to_json_value.is_map
-       fun is_map: Bool do return value isa HashMap[String, nullable Object]
+       fun is_map: Bool do return value isa MapRead[String, nullable Object]
 
        # Get this value as a `Map[String, JsonValue]`
        #
        # require: `self.is_map`
        fun to_map: Map[String, JsonValue] do
                var value = value
-               assert value isa HashMap[String, nullable Object]
+               assert value isa MapRead[String, nullable Object]
 
                var map = new HashMap[String, JsonValue]
                for k, v in value do map[k] = new JsonValue(v)
@@ -145,7 +196,7 @@ class JsonValue
        #     assert "[1, 2, 3, 4, 5]".to_json_value.is_array
        #     assert "[null, true, false, 0.0, 1, \"str\"]".to_json_value.is_array
        #     assert """["a", "b", "c"]""".to_json_value.is_array
-       fun is_array: Bool do return value isa Array[nullable Object]
+       fun is_array: Bool do return value isa SequenceRead[nullable Object]
 
        # Get this value as an `Array[JsonValue]`
        #
@@ -155,13 +206,29 @@ class JsonValue
        fun to_a: Array[JsonValue]
        do
                var value = value
-               assert value isa Array[nullable Object]
+               assert value isa SequenceRead[nullable Object]
 
                var a = new Array[JsonValue]
                for e in value do a.add(new JsonValue(e))
                return a
        end
 
+       ### Error
+
+       # Is this value an error?
+       #
+       #     assert "[]".to_json_value[0].is_error
+       #     assert "[".to_json_value.is_error
+       #     assert not "[]".to_json_value.is_error
+       fun is_error: Bool do return value isa Error
+
+       # Get this value as a `Error`.
+       #
+       # require: `self.is_error`
+       fun to_error: Error do return value.as(Error)
+
+       ### Children access
+
        # Iterator over the values of the array `self`
        #
        # require: `self.is_array`
@@ -180,47 +247,161 @@ class JsonValue
        #
        #     assert """{"a": 123}""".to_json_value["a"].to_i == 123
        #     assert """{"123": "a"}""".to_json_value[123].to_s == "a"
-       #     assert """{"John Smith": 1980}""".to_json_value[["John ", "Smith"]].to_i == 1980
+       #     assert """{"John Smith": 1980}""".to_json_value["John Smith"].to_i == 1980
+       #     assert """{"a": 123}""".to_json_value["b"].is_error
        #
        #     assert """["a", "b", "c"]""".to_json_value[0].to_s == "a"
-       fun [](key: Object): JsonValue
-       do
+       #     assert """["a", "b", "c"]""".to_json_value[3].is_error
+       fun [](key: Object): JsonValue do
                var value = value
-               if value isa HashMap[String, nullable Object] then
-                       return new JsonValue(value[key.to_s])
-               else if value isa Array[nullable Object] then
-                       assert key isa Int
-                       return new JsonValue(value[key])
-               else abort
+               var result: nullable Object
+               if is_error then
+                       return self
+               else if value isa MapRead[String, nullable Object] then
+                       key = key.to_s
+                       if value.has_key(key) then
+                               result = value[key]
+                       else
+                               result = new JsonKeyError("Key `{key}` not found.", self, key)
+                       end
+               else if value isa SequenceRead[nullable Object] then
+                       if key isa Int then
+                               if key < value.length and key >= 0 then
+                                       result = value[key]
+                               else
+                                       result = new JsonKeyError("Index `{key}` out of bounds.",
+                                                       self, key)
+                               end
+                       else
+                               result = new JsonKeyError("Invalid key type. Expecting `Int`. Got `{key.class_name}`.",
+                                               self, key)
+                       end
+               else
+                       result = new JsonKeyError("Invalid `[]` access on a `{json_type}` JsonValue.",
+                                       self, key)
+               end
+               return new JsonValue(result)
        end
 
-       # Advanced query to get a value within the map `self` or it's children.
+       # Get the value at `query`, a string of map keys and array indices
        #
-       # A query is composed of the keys to each map seperated by '.'.
+       # The `query` is composed of map keys and array indices separated by "." (by default).
+       # The separator can be set with `sep` to any string.
        #
-       # require: `self.is_map`
+       # Given the following JSON object parsed as a `JsonValue`.
+       # ~~~
+       # var jvalue = """
+       # {
+       #   "a": {
+       #          "i": 123,
+       #          "b": true
+       #        },
+       #   "b": ["zero", "one", "two"]
+       # }""".to_json_value
+       # ~~~
+       #
+       # Access a value in maps by its key, starting from the key in the root object.
+       # ~~~
+       # assert jvalue.get("a").is_map
+       # assert jvalue.get("a.i").to_i == 123
+       # assert jvalue.get("a.b").to_bool
+       # ~~~
        #
-       #     assert """{"a": {"t": true, "f": false}}""".to_json_value.get("a").is_map
-       #     assert """{"a": {"t": true, "f": false}}""".to_json_value.get("a.t").to_bool
-       #     assert not """{"a": {"t": true, "f": false}}""".to_json_value.get("a.f").to_bool
-       #     assert """{"a": {"b": {"c": {"d": 123}}}}""".to_json_value.get("a.b.c.d").to_i == 123
-       fun get(query: String): JsonValue
+       # Access an item in an array by its index.
+       # ~~~
+       # assert jvalue.get("b.1").to_s == "one"
+       # ~~~
+       #
+       # Any error at any depth of a query is reported. The client should usually
+       # check for errors before using the returned value.
+       # ~~~
+       # assert jvalue.get("a.b.c").to_error.to_s == "Value at `a.b` is not a map. Got a `map`"
+       # assert jvalue.get("b.3").to_error.to_s == "Index `3` out of bounds at `b`"
+       # ~~~
+       #
+       # Set `sep` to a custom string to access keys containing a dot.
+       # ~~~
+       # jvalue = """
+       # {
+       #   "a.b": { "i": 123 },
+       #   "c/d": [ 456 ]
+       # }""".to_json_value
+       #
+       # assert jvalue.get("a.b/i", sep="/").to_i == 123
+       # assert jvalue.get("c/d:0", sep=":").to_i == 456
+       # ~~~
+       fun get(query: Text, sep: nullable Text): JsonValue
        do
-               var keys = query.split(".")
+               if is_error then return self
+
+               sep = sep or else "."
+               var keys = query.split(sep)
                var value = value
-               for key in keys do
-                       assert value isa HashMap[String, nullable Object]
-                       value = value[key]
+               for i in [0..keys.length[ do
+                       var key = keys[i]
+                       if value isa MapRead[String, nullable Object] then
+                               if value.has_key(key) then
+                                       value = value[key]
+                               else
+                                       var sub_query = sub_query_to_s(keys, i, sep)
+                                       value = new JsonKeyError("Key `{key}` not found.",
+                                                       self, sub_query)
+                                       break
+                               end
+                       else if value isa Sequence[nullable Object] then
+                               if key.is_int then
+                                       var index = key.to_i
+                                       if index < value.length then
+                                               value = value[index]
+                                       else
+                                               var sub_query = sub_query_to_s(keys, i, sep)
+                                               value = new JsonKeyError("Index `{key}` out of bounds at `{sub_query}`",
+                                                               self, sub_query)
+                                               break
+                                       end
+                               end
+                       else
+                               var sub_query = sub_query_to_s(keys, i, sep)
+                               value = new JsonKeyError("Value at `{sub_query}` is not a map. Got a `{json_type}`",
+                                               self, sub_query)
+                               break
+                       end
                end
                return new JsonValue(value)
        end
-end
 
-redef class String
-       # Parse `self` to obtain a `JsonValue`
-       fun to_json_value: JsonValue
+       # Concatenate all keys up to `last` for error reports
+       private fun sub_query_to_s(keys: Array[String], last: Int, sep: Text): String
        do
-               var value = json_to_nit_object
-               return new JsonValue(value)
+               return [for i in [0..last[ do keys[i]].join(sep)
+       end
+
+       # Return a human-readable description of the type.
+       #
+       # For debugging purpose only.
+       private fun json_type: String
+       do
+               if is_array then return "array"
+               if is_bool then return "bool"
+               if is_float then return "float"
+               if is_int then return "int"
+               if is_null then return "null"
+               if is_map then return "map"
+               if is_string then return "string"
+               if is_error then return "error"
+               return "undefined"
        end
 end
+
+# Key access error
+class JsonKeyError
+       super Error
+
+       # The value on which the access was requested
+       var json_value: JsonValue
+
+       # The requested key
+       #
+       # In the case of `JsonValue::get`, the sub-query that failed.
+       var key: Object
+end