From cc2ef4188dfa07e146929a1cdfb6b5a6cb99e48a Mon Sep 17 00:00:00 2001
From: Larpon <Larpon@users.noreply.github.com>
Date: Fri, 22 Oct 2021 07:16:47 +0200
Subject: [PATCH] toml: support parsing (+/-)nan and (+/-)inf (#12256)

---
 vlib/toml/scanner/scanner.v                 | 51 +++++++++++++++++----
 vlib/toml/tests/burntsushi.toml-test_test.v |  1 -
 2 files changed, 42 insertions(+), 10 deletions(-)

diff --git a/vlib/toml/scanner/scanner.v b/vlib/toml/scanner/scanner.v
index c0494c1950..e2ae4c1a09 100644
--- a/vlib/toml/scanner/scanner.v
+++ b/vlib/toml/scanner/scanner.v
@@ -95,15 +95,18 @@ pub fn (mut s Scanner) scan() ?token.Token {
 		is_sign := byte_c in [`+`, `-`]
 		is_signed_number := is_sign && byte(s.at()).is_digit() && !byte(s.peek(-1)).is_digit()
 
-		// TODO (+/-)nan & (+/-)inf
-		/*
-		mut is_nan := s.peek(1) == `n` && s.peek(2) == `a` && s.peek(3) == `n`
-		mut is_inf := s.peek(1) == `i` && s.peek(2) == `n` && s.peek(3) == `f`
-		if is_nan || is_inf {
-			util.printdbg(@MOD + '.' + @STRUCT + '.' + @FN, 'identified a special number "$key" ($key.len)')
-			return s.new_token(.number, key, key.len)
+		// (+/-)nan & (+/-)inf
+		is_nan := byte_c == `n` && s.at() == `a` && s.peek(1) == `n` && s.peek(2) == `\n`
+		is_inf := byte_c == `i` && s.at() == `n` && s.peek(1) == `f` && s.peek(2) == `\n`
+		is_signed_nan := is_sign && s.at() == `n` && s.peek(1) == `a` && s.peek(2) == `n`
+			&& s.peek(3) == `\n`
+		is_signed_inf := is_sign && s.at() == `i` && s.peek(1) == `n` && s.peek(2) == `f`
+			&& s.peek(3) == `\n`
+		if is_nan || is_inf || is_signed_nan || is_signed_inf {
+			num := s.extract_nan_or_inf_number() ?
+			util.printdbg(@MOD + '.' + @STRUCT + '.' + @FN, 'identified a special number "$num" ($num.len)')
+			return s.new_token(.number, num, num.len)
 		}
-		*/
 
 		is_digit := byte_c.is_digit()
 		if is_digit || is_signed_number {
@@ -515,7 +518,7 @@ fn (mut s Scanner) handle_escapes(quote byte, is_multiline bool) (string, int) {
 }
 
 // extract_number collects and returns a string containing
-// any bytes recognized as a TOML number.
+// any bytes recognized as a TOML number except for "(+/-)nan" and "(+/-)inf".
 // TOML numbers can include digits 0-9 and `_`.
 [direct_array_access; inline]
 fn (mut s Scanner) extract_number() ?string {
@@ -553,6 +556,36 @@ fn (mut s Scanner) extract_number() ?string {
 	return key
 }
 
+// extract_nan_or_inf_number collects and returns a string containing
+// any bytes recognized as infinity or not-a-number TOML numbers.
+[direct_array_access; inline]
+fn (mut s Scanner) extract_nan_or_inf_number() ?string {
+	// extract_nan_or_inf_number is called when the scanner has already identified that
+	// +/- or 'nan'/'inf' bytes is up but we rewind it to start at the correct position
+	s.pos--
+	s.col--
+	start := s.pos
+
+	mut c := s.at()
+	if c !in [`+`, `-`, `n`, `i`] {
+		return error(@MOD + '.' + @STRUCT + '.' + @FN +
+			' ${byte(c).ascii_str()} is not a number at ${s.excerpt(s.pos, 10)}')
+	}
+	s.pos++
+	s.col++
+	for s.pos < s.text.len {
+		c = s.at()
+		if c !in [`n`, `a`, `i`, `f`] {
+			break
+		}
+		s.pos++
+		s.col++
+	}
+	key := s.text[start..s.pos]
+	util.printdbg(@MOD + '.' + @STRUCT + '.' + @FN, 'identified special number "$key" in range [$start .. $s.pos]')
+	return key
+}
+
 // excerpt returns a string excerpt of the input text centered
 // at `pos`. The `margin` argument defines how many chacters
 // on each side of `pos` is returned
diff --git a/vlib/toml/tests/burntsushi.toml-test_test.v b/vlib/toml/tests/burntsushi.toml-test_test.v
index 1d3a873733..e68b70bbae 100644
--- a/vlib/toml/tests/burntsushi.toml-test_test.v
+++ b/vlib/toml/tests/burntsushi.toml-test_test.v
@@ -9,7 +9,6 @@ import toml
 // TODO Goal: make parsing AND value retrieval of all of https://github.com/BurntSushi/toml-test/test/ pass
 const (
 	valid_exceptions   = [
-		'float/inf-and-nan.toml',
 		'table/array-table-array.toml',
 	]
 	invalid_exceptions = [