Merge pull request #95 from NuruProgramming/fh/add-shebang-support

Fh/add shebang support
NuruProgramming · Oct 30, 2024 · f520183 · f520183
2 parents e9c3aec + 5e7a29f
commit f520183
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lexer/lexer.go b/lexer/lexer.go
@@ -168,6 +168,11 @@ func (l *Lexer) NextToken() token.Token {
 		} else {
 			tok = newToken(token.MODULUS, l.line, l.ch)
 		}
+	case rune('#'):
+		if l.peekChar() == rune('!') && l.line == 1 {
+			l.skipSingleLineComment()
+			return l.NextToken()
+		}
 	case 0:
 		tok.Literal = ""
 		tok.Type = token.EOF

diff --git a/token/token.go b/token/token.go
@@ -44,6 +44,7 @@ const (
 	ASTERISK_ASSIGN = "*="
 	SLASH_ASSIGN    = "/="
 	MODULUS_ASSIGN  = "%="
+	SHEBANG         = "#!"
 
 	//Delimiters
 	COMMA     = ","