feat: adds modulus babyyy

Tezzish · Sep 27, 2024 · 2f30cf0 · 2f30cf0
1 parent 5472228
commit 2f30cf0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/lexer/lexer.go b/lexer/lexer.go
@@ -66,6 +66,8 @@ func (l *Lexer) NextToken() token.Token {
 		tok = newToken(token.ASTERISK, l.character)
 	case '/':
 		tok = newToken(token.SLASH, l.character)
+	case '%':
+		tok = newToken(token.MODULUS, l.character)
 	case '<':
 		tok = newToken(token.LT, l.character)
 	case '>':

diff --git a/parser/parser.go b/parser/parser.go
@@ -24,6 +24,7 @@ var precedences = map[token.TokenType]int{
 	token.NEQ:      EQUALS,
 	token.LT:       LESSGREATER,
 	token.GT:       LESSGREATER,
+	token.MODULUS:  PRODUCT,
 	token.PLUS:     SUM,
 	token.MINUS:    SUM,
 	token.SLASH:    PRODUCT,
@@ -62,6 +63,7 @@ func New(l *lexer.Lexer) *Parser {
 	p.registerPrefix(token.IF, p.parseIfExpression)
 	p.registerPrefix(token.FUNCTION, p.parseFunctionLiteral)
 	p.infixParseFns = make(map[token.TokenType]infixParseFn)
+	p.registerInfix(token.MODULUS, p.parseInfixExpression)
 	p.registerInfix(token.PLUS, p.parseInfixExpression)
 	p.registerInfix(token.MINUS, p.parseInfixExpression)
 	p.registerInfix(token.SLASH, p.parseInfixExpression)

diff --git a/token/token.go b/token/token.go
@@ -20,6 +20,7 @@ const (
 	BANG     = "!"
 	ASTERISK = "*"
 	SLASH    = "/"
+	MODULUS  = "%"
 
 	LT = "<"
 	GT = ">"