add single line comments

This fixes #7 Signed-off-by: Flipez <code@brauser.io>
Flipez · Sep 29, 2021 · f52fce8 · f52fce8
1 parent ef25787
commit f52fce8
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/lexer/lexer.go b/lexer/lexer.go
@@ -54,7 +54,12 @@ func (l *Lexer) NextToken() token.Token {
 			tok = newToken(token.BANG, l.ch)
 		}
 	case '/':
-		tok = newToken(token.SLASH, l.ch)
+		if l.peekChar() == '/' {
+			l.skipComment()
+			tok = l.NextToken()
+		} else {
+			tok = newToken(token.SLASH, l.ch)
+		}
 	case '*':
 		tok = newToken(token.ASTERISK, l.ch)
 	case '<':
@@ -146,6 +151,13 @@ func (l *Lexer) skipWhitespace() {
 	}
 }
 
+func (l *Lexer) skipComment() {
+	for l.ch != '\n' && l.ch != 0 {
+		l.readChar()
+	}
+	l.skipWhitespace()
+}
+
 func isLetter(ch byte) bool {
 	return 'a' <= ch && ch <= 'z' || 'A' <= ch && ch <= 'Z' || ch == '_'
 }