From 249abe9c33dfd08afa6b09f3e53be91b0f445561 Mon Sep 17 00:00:00 2001
From: Mel <einebeere@gmail.com>
Date: Sat, 23 Oct 2021 22:26:58 +0200
Subject: Allow _ in identifiers.

---
 src/lex/lexer.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/lex/lexer.rs')

diff --git a/src/lex/lexer.rs b/src/lex/lexer.rs
index 60301c3..41384a9 100644
--- a/src/lex/lexer.rs
+++ b/src/lex/lexer.rs
@@ -32,7 +32,7 @@ impl Iterator for Lexer<'_> {
 
         let token = if c.is_numeric() {
             self.number()
-        } else if c.is_alphabetic() {
+        } else if c.is_alphabetic() || c == '_' {
             self.identifier()
         } else if c == '"' {
             self.str()
@@ -180,7 +180,7 @@ impl<'s> Lexer<'s> {
 
         let mut buffer = String::new();
 
-        while self.peek().map_or(false, |c| c.is_alphabetic()) {
+        while self.peek().map_or(false, |c| c.is_alphabetic() || c == '_') {
             let c = self.advance().unwrap();
             buffer.push(c);
         }
-- 
cgit 1.4.1