toml: add conversion of ast inf and nan to Any

Author: larpon

vlib/toml/tests/value_query_test.v

@@ -1,4 +1,5 @@
 import toml
+import strconv
 
 const toml_text = '
 modules = [ "ui", "toml" ]
@@ -15,6 +16,11 @@ colors = [
 	"yellow",
 	[ "transparent" ]
 ]
+
+[values]
+nan = nan
+inf = inf
+minus-inf = -inf
 '
 
 fn test_value_query_in_array() {
@@ -29,4 +35,12 @@ fn test_value_query_in_array() {
 	assert value == 'toml'
 	value = toml_doc.value('errors[11]').default_to('<none>').string()
 	assert value == '<none>'
+
+	value = toml_doc.value('values.nan').string()
+	assert value == 'nan'
+
+	mut value_u64 := toml_doc.value('values.inf').u64()
+	assert value_u64 == strconv.double_plus_infinity
+	value_u64 = toml_doc.value('values.minus-inf').u64()
+	assert value_u64 == strconv.double_minus_infinity
 }

vlib/toml/to/to.v

@@ -12,7 +12,7 @@ type DocOrAny = toml.Any | toml.Doc
 pub fn json(doa DocOrAny) string {
 	match doa {
 		toml.Doc {
-			return any_to_json(doa.ast_to_any(doa.ast.table))
+			return any_to_json(toml.ast_to_any(doa.ast.table))
 		}
 		toml.Any {
 			return any_to_json(doa)

vlib/toml/toml.v

@@ -152,7 +152,7 @@ pub fn parse_dotted_key(key string) ?[]string {
 
 // to_any converts the `Doc` to toml.Any type.
 pub fn (d Doc) to_any() Any {
-	return d.ast_to_any(d.ast.table)
+	return ast_to_any(d.ast.table)
 }
 
 // value queries a value from the TOML document.
@@ -193,17 +193,17 @@ fn (d Doc) value_(value ast.Value, key []string) Any {
 	}
 
 	if key.len <= 1 {
-		return d.ast_to_any(ast_value)
+		return ast_to_any(ast_value)
 	}
 	// `match` isn't currently very suitable for these types of sum type constructs...
 	if ast_value is map[string]ast.Value || ast_value is []ast.Value {
 		return d.value_(ast_value, key[1..])
 	}
-	return d.ast_to_any(value)
+	return ast_to_any(value)
 }
 
 // ast_to_any converts `from` ast.Value to toml.Any value.
-pub fn (d Doc) ast_to_any(value ast.Value) Any {
+pub fn ast_to_any(value ast.Value) Any {
 	match value {
 		ast.Date {
 			return Any(Date{value.text})
@@ -218,11 +218,22 @@ pub fn (d Doc) ast_to_any(value ast.Value) Any {
 			return Any(value.text)
 		}
 		ast.Number {
-			// if value.text.contains('inf') || value.text.contains('nan') {
-			// return Any() // TODO
-			//}
-			if !value.text.starts_with('0x')
-				&& (value.text.contains('.') || value.text.to_lower().contains('e')) {
+			val_text := value.text
+			if val_text == 'inf' || val_text == '+inf' || val_text == '-inf' {
+				// NOTE values taken from strconv
+				if !val_text.starts_with('-') {
+					// strconv.double_plus_infinity
+					return Any(u64(0x7FF0000000000000))
+				} else {
+					// strconv.double_minus_infinity
+					return Any(u64(0xFFF0000000000000))
+				}
+			}
+			if val_text == 'nan' || val_text == '+nan' || val_text == '-nan' {
+				return Any('nan')
+			}
+			if !val_text.starts_with('0x')
+				&& (val_text.contains('.') || val_text.to_lower().contains('e')) {
 				return Any(value.f64())
 			}
 			return Any(value.i64())
@@ -238,7 +249,7 @@ pub fn (d Doc) ast_to_any(value ast.Value) Any {
 			m := (value as map[string]ast.Value)
 			mut am := map[string]Any{}
 			for k, v in m {
-				am[k] = d.ast_to_any(v)
+				am[k] = ast_to_any(v)
 			}
 			return am
 			// return d.get_map_value(m, key_split[1..].join('.'))
@@ -247,7 +258,7 @@ pub fn (d Doc) ast_to_any(value ast.Value) Any {
 			a := (value as []ast.Value)
 			mut aa := []Any{}
 			for val in a {
-				aa << d.ast_to_any(val)
+				aa << ast_to_any(val)
 			}
 			return aa
 		}