Reserve inf, +inf, -inf, nan as keywords

Needed to represent literals as f64::INFINITY, f64::NEG_INFINITY and f64::NAN.

1 files changed, 5 insertions, 1 deletions

diff --git a/compiler/src/ast/parser.rs b/compiler/src/ast/parser.rs
index 89a1280..6b11cc6 100644
--- a/compiler/src/ast/parser.rs
+++ b/compiler/src/ast/parser.rs
@@ -7,7 +7,8 @@ use crate::{
 };
 
 fn parse_number(number: &str) -> Result<Atom, Error> {
-    let is_float = number.bytes().any(|b| matches!(b, b'.' | b'e' | b'E'));
+    let is_float = number.bytes().any(|b| matches!(b, b'.' | b'e' | b'E'))
+        || matches!(number, "inf" | "+inf" | "-inf" | "nan");
 
     let atom = if is_float {
         Atom::Float(number.parse()?)
@@ -52,6 +53,9 @@ fn parse_symbol(symbol: &str) -> Atom {
         "true" => Atom::Bool(true),
         "false" => Atom::Bool(false),
         "nil" => Atom::Nil,
+        "inf" | "+inf" => Atom::Float(f64::INFINITY),
+        "-inf" => Atom::Float(f64::NEG_INFINITY),
+        "nan" => Atom::Float(f64::NAN),
         _ => Atom::Symbol(symbol.into()),
     }
 }