Allow underscores in identifiers in scanner

1 files changed, 2 insertions, 2 deletions

diff --git a/pkg/lang/scanner/scanner.go b/pkg/lang/scanner/scanner.go
index 2991083..b01e22f 100644
--- a/pkg/lang/scanner/scanner.go
+++ b/pkg/lang/scanner/scanner.go
@@ -69,7 +69,7 @@ func (s *Scanner) scanToken() (token.Token, error) {
 
 	if c == '"' {
 		return s.scanString()
-	} else if unicode.IsLetter(c) {
+	} else if unicode.IsLetter(c) || c == '_' {
 		return s.scanIdentifierOrKeyword()
 	} else if unicode.IsDigit(c) {
 		return s.scanNumber()
@@ -195,7 +195,7 @@ func (s *Scanner) scanIdentifierOrKeyword() (token.Token, error) {
 			return token.Token{}, err
 		}
 
-		if eof || (!unicode.IsLetter(c) && !unicode.IsDigit(c)) {
+		if eof || !(unicode.IsLetter(c) || unicode.IsDigit(c) || c == '_') {
 			break
 		}