1097 lines
40 KiB
Nim
1097 lines
40 KiB
Nim
|
# Copyright 2022 Mattia Giambirtone & All Contributors
|
||
|
#
|
||
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
# you may not use this file except in compliance with the License.
|
||
|
# You may obtain a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
# See the License for the specific language governing permissions and
|
||
|
# limitations under the License.
|
||
|
|
||
|
## A recursive-descent top-down parser implementation
|
||
|
|
||
|
import strformat
|
||
|
|
||
|
|
||
|
import meta/token
|
||
|
import meta/ast
|
||
|
import meta/errors
|
||
|
|
||
|
|
||
|
export token, ast, errors
|
||
|
|
||
|
|
||
|
type
|
||
|
|
||
|
LoopContext = enum
|
||
|
Loop, None
|
||
|
|
||
|
Parser* = ref object
|
||
|
## A recursive-descent top-down
|
||
|
## parser implementation
|
||
|
|
||
|
# Index into self.tokens
|
||
|
current: int
|
||
|
# The name of the file being parsed.
|
||
|
# Only meaningful for parse errors
|
||
|
file: string
|
||
|
# The list of tokens representing
|
||
|
# the source code to be parsed.
|
||
|
# In most cases, those will come
|
||
|
# from the builtin lexer, but this
|
||
|
# behavior is not enforced and the
|
||
|
# tokenizer is entirely separate from
|
||
|
# the parser
|
||
|
tokens: seq[Token]
|
||
|
# Little internal attribute that tells
|
||
|
# us if we're inside a loop or not. This
|
||
|
# allows us to detect errors like break
|
||
|
# being used outside loops
|
||
|
currentLoop: LoopContext
|
||
|
# Stores the current function
|
||
|
# being parsed. This is a reference
|
||
|
# to either a FunDecl or LambdaExpr
|
||
|
# AST node and is mostly used to allow
|
||
|
# implicit generators to work. What that
|
||
|
# means is that there is no need for the
|
||
|
# programmer to specifiy a function is a
|
||
|
# generator like in nim, (which uses the
|
||
|
# 'iterator' keyword): any function is
|
||
|
# automatically a generator if it contains
|
||
|
# any number of yield statement(s) or
|
||
|
# yield expression(s). This attribute
|
||
|
# is nil when the parser is at the top-level
|
||
|
# code and is what allows the parser to detect
|
||
|
# errors like return outside functions and attempts
|
||
|
# to declare public names inside them before
|
||
|
# compilation even begins
|
||
|
currentFunction: ASTNode
|
||
|
# Stores the current scope depth (0 = global, > 0 local)
|
||
|
scopeDepth: int
|
||
|
|
||
|
|
||
|
proc initParser*(): Parser =
|
||
|
## Initializes a new Parser object
|
||
|
new(result)
|
||
|
result.current = 0
|
||
|
result.file = ""
|
||
|
result.tokens = @[]
|
||
|
result.currentFunction = nil
|
||
|
result.currentLoop = None
|
||
|
result.scopeDepth = 0
|
||
|
|
||
|
# Public getters for improved error formatting
|
||
|
proc getCurrent*(self: Parser): int = self.current
|
||
|
proc getCurrentToken*(self: Parser): Token = (if self.getCurrent() >= self.tokens.len(): self.tokens[^1] else: self.tokens[self.current - 1])
|
||
|
|
||
|
# Handy templates to make our life easier, thanks nim!
|
||
|
|
||
|
template endOfFile: Token = Token(kind: EndOfFile, lexeme: "", line: -1)
|
||
|
template endOfLine(msg: string) = self.expect(Semicolon, msg)
|
||
|
|
||
|
|
||
|
proc peek(self: Parser, distance: int = 0): Token =
|
||
|
## Peeks at the token at the given distance.
|
||
|
## If the distance is out of bounds, an EOF
|
||
|
## token is returned. A negative distance may
|
||
|
## be used to retrieve previously consumed
|
||
|
## tokens
|
||
|
if self.tokens.high() == -1 or self.current + distance > self.tokens.high() or self.current + distance < 0:
|
||
|
result = endOfFile
|
||
|
else:
|
||
|
result = self.tokens[self.current + distance]
|
||
|
|
||
|
|
||
|
proc done(self: Parser): bool =
|
||
|
## Returns true if we're at the
|
||
|
## end of the file. Note that the
|
||
|
## parser expects an explicit
|
||
|
## EOF token to signal the end
|
||
|
## of the file (unless the token
|
||
|
## list is empty)
|
||
|
result = self.tokens.len() == 0 or self.peek().kind == EndOfFile
|
||
|
|
||
|
|
||
|
proc step(self: Parser, n: int = 1): Token =
|
||
|
## Steps n tokens into the input,
|
||
|
## returning the last consumed one
|
||
|
if self.done():
|
||
|
result = self.peek()
|
||
|
else:
|
||
|
result = self.tokens[self.current]
|
||
|
self.current += 1
|
||
|
|
||
|
|
||
|
proc error(self: Parser, message: string) =
|
||
|
## Raises a formatted ParseError exception
|
||
|
var lexeme = self.getCurrentToken().lexeme
|
||
|
var errorMessage = &"A fatal error occurred while parsing '{self.file}', line {self.peek().line} at '{lexeme}' -> {message}"
|
||
|
raise newException(ParseError, errorMessage)
|
||
|
|
||
|
|
||
|
proc check(self: Parser, kind: TokenType, distance: int = 0): bool =
|
||
|
## Checks if the given token at the given distance
|
||
|
## matches the expected kind and returns a boolean.
|
||
|
## The distance parameter is passed directly to
|
||
|
## self.peek()
|
||
|
self.peek(distance).kind == kind
|
||
|
|
||
|
|
||
|
proc check(self: Parser, kind: openarray[TokenType]): bool =
|
||
|
## Calls self.check() in a loop with each entry of
|
||
|
## the given openarray of token kinds and returns
|
||
|
## at the first match. Note that this assumes
|
||
|
## that only one token may exist at a given
|
||
|
## position
|
||
|
for k in kind:
|
||
|
if self.check(k):
|
||
|
return true
|
||
|
return false
|
||
|
|
||
|
|
||
|
proc match(self: Parser, kind: TokenType, distance: int = 0): bool =
|
||
|
## Behaves like self.check(), except that when a token
|
||
|
## matches it is consumed
|
||
|
if self.check(kind, distance):
|
||
|
discard self.step()
|
||
|
result = true
|
||
|
else:
|
||
|
result = false
|
||
|
|
||
|
|
||
|
proc match(self: Parser, kind: openarray[TokenType]): bool =
|
||
|
## Calls self.match() in a loop with each entry of
|
||
|
## the given openarray of token kinds and returns
|
||
|
## at the first match. Note that this assumes
|
||
|
## that only one token may exist at a given
|
||
|
## position
|
||
|
for k in kind:
|
||
|
if self.match(k):
|
||
|
return true
|
||
|
result = false
|
||
|
|
||
|
|
||
|
proc expect(self: Parser, kind: TokenType, message: string = "") =
|
||
|
## Behaves like self.match(), except that
|
||
|
## when a token doesn't match an error
|
||
|
## is raised. If no error message is
|
||
|
## given, a default one is used
|
||
|
if not self.match(kind):
|
||
|
if message.len() == 0:
|
||
|
self.error(&"expecting token of kind {kind}, found {self.peek().kind} instead")
|
||
|
else:
|
||
|
self.error(message)
|
||
|
|
||
|
|
||
|
proc unnest(self: Parser, node: ASTNode): ASTNode =
|
||
|
## Unpacks an arbitrarily nested grouping expression
|
||
|
var node = node
|
||
|
while node.kind == groupingExpr and GroupingExpr(node).expression != nil:
|
||
|
node = GroupingExpr(node).expression
|
||
|
result = node
|
||
|
|
||
|
|
||
|
# Forward declarations
|
||
|
proc expression(self: Parser): ASTNode
|
||
|
proc expressionStatement(self: Parser): ASTNode
|
||
|
proc statement(self: Parser): ASTNode
|
||
|
proc varDecl(self: Parser, isStatic: bool = true, isPrivate: bool = true): ASTNode
|
||
|
proc funDecl(self: Parser, isAsync: bool = false, isStatic: bool = true, isPrivate: bool = true, isLambda: bool = false): ASTNode
|
||
|
proc declaration(self: Parser): ASTNode
|
||
|
|
||
|
|
||
|
proc primary(self: Parser): ASTNode =
|
||
|
## Parses primary expressions such
|
||
|
## as integer literals and keywords
|
||
|
## that map to builtin types (true, false, etc)
|
||
|
case self.peek().kind:
|
||
|
of True:
|
||
|
result = newTrueExpr(self.step())
|
||
|
of False:
|
||
|
result = newFalseExpr(self.step())
|
||
|
of TokenType.NotANumber:
|
||
|
result = newNanExpr(self.step())
|
||
|
of Nil:
|
||
|
result = newNilExpr(self.step())
|
||
|
of Float:
|
||
|
result = newFloatExpr(self.step())
|
||
|
of Integer:
|
||
|
result = newIntExpr(self.step())
|
||
|
of Identifier:
|
||
|
result = newIdentExpr(self.step())
|
||
|
of LeftParen:
|
||
|
let tok = self.step()
|
||
|
if self.match(RightParen):
|
||
|
# This yields an empty tuple
|
||
|
result = newTupleExpr(@[], tok)
|
||
|
else:
|
||
|
result = self.expression()
|
||
|
if self.match(Comma):
|
||
|
var tupleObject = newTupleExpr(@[result], tok)
|
||
|
while not self.check(RightParen):
|
||
|
tupleObject.members.add(self.expression())
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
result = tupleObject
|
||
|
self.expect(RightParen, "unterminated tuple literal")
|
||
|
else:
|
||
|
self.expect(RightParen, "unterminated parenthesized expression")
|
||
|
result = newGroupingExpr(result, tok)
|
||
|
of LeftBracket:
|
||
|
let tok = self.step()
|
||
|
if self.match(RightBracket):
|
||
|
# This yields an empty list
|
||
|
result = newListExpr(@[], tok)
|
||
|
else:
|
||
|
var listObject = newListExpr(@[], tok)
|
||
|
while not self.check(RightBracket):
|
||
|
listObject.members.add(self.expression())
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
result = listObject
|
||
|
self.expect(RightBracket, "unterminated list literal")
|
||
|
of LeftBrace:
|
||
|
let tok = self.step()
|
||
|
if self.match(RightBrace):
|
||
|
# This yields an empty dictionary, not an empty set!
|
||
|
# For empty sets, there will be a builtin set() type
|
||
|
# that can be instantiated with no arguments
|
||
|
result = newDictExpr(@[], @[], tok)
|
||
|
else:
|
||
|
result = self.expression()
|
||
|
if self.match(Comma) or self.check(RightBrace):
|
||
|
var setObject = newSetExpr(@[result], tok)
|
||
|
while not self.check(RightBrace):
|
||
|
setObject.members.add(self.expression())
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
result = setObject
|
||
|
self.expect(RightBrace, "unterminated set literal")
|
||
|
elif self.match(Colon):
|
||
|
var dictObject = newDictExpr(@[result], @[self.expression()], tok)
|
||
|
if self.match(RightBrace):
|
||
|
return dictObject
|
||
|
if self.match(Comma):
|
||
|
while not self.check(RightBrace):
|
||
|
dictObject.keys.add(self.expression())
|
||
|
self.expect(Colon)
|
||
|
dictObject.values.add(self.expression())
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
self.expect(RightBrace, "unterminated dict literal")
|
||
|
result = dictObject
|
||
|
of Yield:
|
||
|
let tok = self.step()
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'yield' cannot be outside functions")
|
||
|
if self.currentFunction.kind == NodeKind.funDecl:
|
||
|
FunDecl(self.currentFunction).isGenerator = true
|
||
|
else:
|
||
|
LambdaExpr(self.currentFunction).isGenerator = true
|
||
|
if not self.check([RightBrace, RightBracket, RightParen, Comma, Semicolon]):
|
||
|
result = newYieldExpr(self.expression(), tok)
|
||
|
else:
|
||
|
result = newYieldExpr(newNilExpr(Token()), tok)
|
||
|
of Await:
|
||
|
let tok = self.step()
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'await' cannot be used outside functions")
|
||
|
if self.currentFunction.kind == lambdaExpr or not FunDecl(self.currentFunction).isAsync:
|
||
|
self.error("'await' can only be used inside async functions")
|
||
|
result = newAwaitExpr(self.expression(), tok)
|
||
|
of Lambda:
|
||
|
discard self.step()
|
||
|
result = self.funDecl(isLambda=true)
|
||
|
of RightParen, RightBracket, RightBrace:
|
||
|
# This is *technically* unnecessary: the parser would
|
||
|
# throw an error regardless, but it's a little bit nicer
|
||
|
# when the error message is more specific
|
||
|
self.error(&"unmatched '{self.peek().lexeme}'")
|
||
|
of Hex:
|
||
|
result = newHexExpr(self.step())
|
||
|
of Octal:
|
||
|
result = newOctExpr(self.step())
|
||
|
of Binary:
|
||
|
result = newBinExpr(self.step())
|
||
|
of String:
|
||
|
result = newStrExpr(self.step())
|
||
|
of Infinity:
|
||
|
result = newInfExpr(self.step())
|
||
|
else:
|
||
|
self.error("invalid syntax")
|
||
|
|
||
|
|
||
|
proc makeCall(self: Parser, callee: ASTNode): ASTNode =
|
||
|
## Utility function called iteratively by self.call()
|
||
|
## to parse a function-like call
|
||
|
let tok = self.peek(-1)
|
||
|
var argNames: seq[ASTNode] = @[]
|
||
|
var arguments: tuple[positionals: seq[ASTNode], keyword: seq[tuple[name: ASTNode, value: ASTNode]]] = (positionals: @[], keyword: @[])
|
||
|
var argument: ASTNode = nil
|
||
|
var argCount = 0
|
||
|
if not self.check(RightParen):
|
||
|
while true:
|
||
|
if argCount >= 255:
|
||
|
self.error("cannot store more than 255 arguments")
|
||
|
break
|
||
|
argument = self.expression()
|
||
|
if argument.kind == assignExpr:
|
||
|
if AssignExpr(argument).name in argNames:
|
||
|
self.error("duplicate keyword argument in call")
|
||
|
argNames.add(AssignExpr(argument).name)
|
||
|
arguments.keyword.add((name: AssignExpr(argument).name, value: AssignExpr(argument).value))
|
||
|
elif arguments.keyword.len() == 0:
|
||
|
arguments.positionals.add(argument)
|
||
|
else:
|
||
|
self.error("positional arguments cannot follow keyword arguments in call")
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
argCount += 1
|
||
|
self.expect(RightParen)
|
||
|
result = newCallExpr(callee, arguments, tok)
|
||
|
|
||
|
|
||
|
proc call(self: Parser): ASTNode =
|
||
|
## Parses calls, object field
|
||
|
## field accessing and slicing
|
||
|
## expressions
|
||
|
result = self.primary()
|
||
|
while true:
|
||
|
if self.match(LeftParen):
|
||
|
result = self.makeCall(result)
|
||
|
elif self.match(Dot):
|
||
|
self.expect(Identifier, "expecting attribute name after '.'")
|
||
|
result = newGetItemExpr(result, newIdentExpr(self.peek(-1)), self.peek(-1))
|
||
|
elif self.match(LeftBracket):
|
||
|
let tok = self.peek(-1)
|
||
|
var ends: seq[ASTNode] = @[]
|
||
|
while not self.match(RightBracket) and ends.len() < 3:
|
||
|
ends.add(self.expression())
|
||
|
discard self.match(Colon)
|
||
|
if ends.len() < 1:
|
||
|
self.error("invalid syntax")
|
||
|
result = newSliceExpr(result, ends, tok)
|
||
|
else:
|
||
|
break
|
||
|
|
||
|
|
||
|
proc unary(self: Parser): ASTNode =
|
||
|
## Parses unary expressions
|
||
|
if self.match([Minus, Tilde, LogicalNot, Plus]):
|
||
|
result = newUnaryExpr(self.peek(-1), self.unary())
|
||
|
else:
|
||
|
result = self.call()
|
||
|
|
||
|
|
||
|
proc pow(self: Parser): ASTNode =
|
||
|
## Parses exponentiation expressions
|
||
|
result = self.unary()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match(DoubleAsterisk):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.unary()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc mul(self: Parser): ASTNode =
|
||
|
## Parses multiplication and division expressions
|
||
|
result = self.pow()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match([Slash, Percentage, FloorDiv, Asterisk]):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.pow()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc add(self: Parser): ASTNode =
|
||
|
## Parses addition and subtraction expressions
|
||
|
result = self.mul()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match([Plus, Minus]):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.mul()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc comparison(self: Parser): ASTNode =
|
||
|
## Parses comparison expressions
|
||
|
result = self.add()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match([LessThan, GreaterThan, LessOrEqual, GreaterOrEqual, Is, As, Of, IsNot]):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.add()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc equality(self: Parser): ASTNode =
|
||
|
## Parses equality expressions
|
||
|
result = self.comparison()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match([DoubleEqual, NotEqual]):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.comparison()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc logicalAnd(self: Parser): ASTNode =
|
||
|
## Parses logical AND expressions
|
||
|
result = self.equality()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match(LogicalAnd):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.equality()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc logicalOr(self: Parser): ASTNode =
|
||
|
## Parses logical OR expressions
|
||
|
result = self.logicalAnd()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match(LogicalOr):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.logicalAnd()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc bitwiseAnd(self: Parser): ASTNode =
|
||
|
## Parser a & b expressions
|
||
|
result = self.logicalOr()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match(Pipe):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.logicalOr()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc bitwiseOr(self: Parser): ASTNode =
|
||
|
## Parser a | b expressions
|
||
|
result = self.bitwiseAnd()
|
||
|
var operator: Token
|
||
|
var right: ASTNode
|
||
|
while self.match(Ampersand):
|
||
|
operator = self.peek(-1)
|
||
|
right = self.bitwiseAnd()
|
||
|
result = newBinaryExpr(result, operator, right)
|
||
|
|
||
|
|
||
|
proc assignment(self: Parser): ASTNode =
|
||
|
## Parses assignment, the highest-level
|
||
|
## expression (including stuff like a.b = 1).
|
||
|
## Slice assignments are also parsed here
|
||
|
result = self.bitwiseOr()
|
||
|
if self.match([Equal, InplaceAdd, InplaceSub, InplaceDiv, InplaceMod,
|
||
|
InplacePow, InplaceMul, InplaceXor, InplaceAnd, InplaceOr,
|
||
|
InplaceFloorDiv, InplaceRightShift, InplaceLeftShift]):
|
||
|
let tok = self.peek(-1)
|
||
|
var value = self.expression()
|
||
|
if result.kind in {identExpr, sliceExpr}:
|
||
|
result = newAssignExpr(result, value, tok)
|
||
|
elif result.kind == getItemExpr:
|
||
|
result = newSetItemExpr(GetItemExpr(result).obj, GetItemExpr(result).name, value, tok)
|
||
|
else:
|
||
|
self.error("invalid assignment target")
|
||
|
|
||
|
|
||
|
proc delStmt(self: Parser): ASTNode =
|
||
|
## Parses "del" statements,
|
||
|
## which unbind a name from its
|
||
|
## value in the current scope and
|
||
|
## calls its destructor
|
||
|
let tok = self.peek(-1)
|
||
|
var expression = self.expression()
|
||
|
var temp = expression
|
||
|
endOfLIne("missing semicolon after del statement")
|
||
|
if expression.kind == groupingExpr:
|
||
|
# We unpack grouping expressions
|
||
|
temp = self.unnest(temp)
|
||
|
if temp.isLiteral():
|
||
|
self.error("cannot delete a literal")
|
||
|
elif temp.kind in {binaryExpr, unaryExpr}:
|
||
|
self.error("cannot delete operator")
|
||
|
elif temp.kind == callExpr:
|
||
|
self.error("cannot delete function call")
|
||
|
elif temp.kind == assignExpr:
|
||
|
self.error("cannot delete assignment")
|
||
|
else:
|
||
|
result = newDelStmt(expression, tok)
|
||
|
|
||
|
|
||
|
proc assertStmt(self: Parser): ASTNode =
|
||
|
## Parses "assert" statements,
|
||
|
## raise an error if the expression
|
||
|
## fed into them is falsey
|
||
|
let tok = self.peek(-1)
|
||
|
var expression = self.expression()
|
||
|
endOfLine("missing semicolon after assert statement")
|
||
|
result = newAssertStmt(expression, tok)
|
||
|
|
||
|
|
||
|
proc beginScope(self: Parser) =
|
||
|
## Begins a new syntactical scope
|
||
|
inc(self.scopeDepth)
|
||
|
|
||
|
|
||
|
proc endScope(self: Parser) =
|
||
|
## Ends a new syntactical scope
|
||
|
dec(self.scopeDepth)
|
||
|
|
||
|
|
||
|
proc blockStmt(self: Parser): ASTNode =
|
||
|
## Parses block statements. A block
|
||
|
## statement simply opens a new local
|
||
|
## scope
|
||
|
|
||
|
self.beginScope()
|
||
|
let tok = self.peek(-1)
|
||
|
var code: seq[ASTNode] = @[]
|
||
|
while not self.check(RightBrace) and not self.done():
|
||
|
code.add(self.declaration())
|
||
|
self.expect(RightBrace, "unterminated block statement")
|
||
|
result = newBlockStmt(code, tok)
|
||
|
self.endScope()
|
||
|
|
||
|
|
||
|
proc breakStmt(self: Parser): ASTNode =
|
||
|
## Parses break statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentLoop != Loop:
|
||
|
self.error("'break' cannot be used outside loops")
|
||
|
endOfLine("missing semicolon after break statement")
|
||
|
result = newBreakStmt(tok)
|
||
|
|
||
|
|
||
|
proc deferStmt(self: Parser): ASTNode =
|
||
|
## Parses defer statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'defer' cannot be used outside functions")
|
||
|
result = newDeferStmt(self.expression(), tok)
|
||
|
endOfLine("missing semicolon after defer statement")
|
||
|
|
||
|
|
||
|
proc continueStmt(self: Parser): ASTNode =
|
||
|
## Parses continue statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentLoop != Loop:
|
||
|
self.error("'continue' cannot be used outside loops")
|
||
|
endOfLine("missing semicolon after continue statement")
|
||
|
result = newContinueStmt(tok)
|
||
|
|
||
|
|
||
|
proc returnStmt(self: Parser): ASTNode =
|
||
|
## Parses return statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'return' cannot be used outside functions")
|
||
|
var value: ASTNode
|
||
|
if not self.check(Semicolon):
|
||
|
# Since return can be used on its own too
|
||
|
# (in which case it implicitly returns nil),
|
||
|
# we need to check if there's an actual value
|
||
|
# to return or not
|
||
|
value = self.expression()
|
||
|
endOfLine("missing semicolon after return statement")
|
||
|
result = newReturnStmt(value, tok)
|
||
|
|
||
|
|
||
|
proc yieldStmt(self: Parser): ASTNode =
|
||
|
## Parses yield Statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'yield' cannot be outside functions")
|
||
|
if self.currentFunction.kind == NodeKind.funDecl:
|
||
|
FunDecl(self.currentFunction).isGenerator = true
|
||
|
else:
|
||
|
LambdaExpr(self.currentFunction).isGenerator = true
|
||
|
if not self.check(Semicolon):
|
||
|
result = newYieldStmt(self.expression(), tok)
|
||
|
else:
|
||
|
result = newYieldStmt(newNilExpr(Token()), tok)
|
||
|
endOfLine("missing semicolon after yield statement")
|
||
|
|
||
|
|
||
|
proc awaitStmt(self: Parser): ASTNode =
|
||
|
## Parses yield Statements
|
||
|
let tok = self.peek(-1)
|
||
|
if self.currentFunction == nil:
|
||
|
self.error("'await' cannot be used outside functions")
|
||
|
if self.currentFunction.kind == lambdaExpr or not FunDecl(self.currentFunction).isAsync:
|
||
|
self.error("'await' can only be used inside async functions")
|
||
|
result = newAwaitStmt(self.expression(), tok)
|
||
|
endOfLine("missing semicolon after yield statement")
|
||
|
|
||
|
|
||
|
proc raiseStmt(self: Parser): ASTNode =
|
||
|
## Parses raise statements
|
||
|
var exception: ASTNode
|
||
|
let tok = self.peek(-1)
|
||
|
if not self.check(Semicolon):
|
||
|
# Raise can be used on its own, in which
|
||
|
# case it re-raises the last active exception
|
||
|
exception = self.expression()
|
||
|
endOfLine("missing semicolon after raise statement")
|
||
|
result = newRaiseStmt(exception, tok)
|
||
|
|
||
|
|
||
|
proc forEachStmt(self: Parser): ASTNode =
|
||
|
## Parses C#-like foreach loops
|
||
|
let tok = self.peek(-1)
|
||
|
var enclosingLoop = self.currentLoop
|
||
|
self.currentLoop = Loop
|
||
|
self.expect(LeftParen, "expecting '(' after 'foreach'")
|
||
|
self.expect(Identifier)
|
||
|
var identifier = newIdentExpr(self.peek(-1))
|
||
|
self.expect(Colon)
|
||
|
var expression = self.expression()
|
||
|
self.expect(RightParen)
|
||
|
var body = self.statement()
|
||
|
result = newForEachStmt(identifier, expression, body, tok)
|
||
|
self.currentLoop = enclosingLoop
|
||
|
|
||
|
|
||
|
proc importStmt(self: Parser): ASTNode =
|
||
|
## Parses import statements
|
||
|
let tok = self.peek(-1)
|
||
|
self.expect(Identifier, "expecting module name(s) after import statement")
|
||
|
result = newImportStmt(self.expression(), tok)
|
||
|
endOfLine("missing semicolon after import statement")
|
||
|
|
||
|
|
||
|
proc fromStmt(self: Parser): ASTNode =
|
||
|
## Parser from xx import yy statements
|
||
|
let tok = self.peek(-1)
|
||
|
self.expect(Identifier, "expecting module name(s) after import statement")
|
||
|
result = newIdentExpr(self.peek(-1))
|
||
|
var attributes: seq[ASTNode] = @[]
|
||
|
var attribute: ASTNode
|
||
|
self.expect(Import)
|
||
|
self.expect(Identifier)
|
||
|
attribute = newIdentExpr(self.peek(-1))
|
||
|
attributes.add(attribute)
|
||
|
while self.match(Comma):
|
||
|
self.expect(Identifier)
|
||
|
attribute = newIdentExpr(self.peek(-1))
|
||
|
attributes.add(attribute)
|
||
|
# from x import a [, b, c, ...];
|
||
|
endOfLine("missing semicolon after import statement")
|
||
|
result = newFromImportStmt(result, attributes, tok)
|
||
|
|
||
|
|
||
|
proc tryStmt(self: Parser): ASTNode =
|
||
|
## Parses try/except/finally/else blocks
|
||
|
let tok = self.peek(-1)
|
||
|
var body = self.statement()
|
||
|
var handlers: seq[tuple[body, exc, name: ASTNode]] = @[]
|
||
|
var finallyClause: ASTNode
|
||
|
var elseClause: ASTNode
|
||
|
var asName: ASTNode
|
||
|
var excName: ASTNode
|
||
|
var handlerBody: ASTNode
|
||
|
while self.match(Except):
|
||
|
excName = self.expression()
|
||
|
if excName.kind == identExpr:
|
||
|
continue
|
||
|
elif excName.kind == binaryExpr and BinaryExpr(excName).operator.kind == As:
|
||
|
asName = BinaryExpr(excName).b
|
||
|
if BinaryExpr(excName).b.kind != identExpr:
|
||
|
self.error("expecting alias name after 'except ... as'")
|
||
|
excName = BinaryExpr(excName).a
|
||
|
# Note how we don't use elif here: when the if above sets excName to As'
|
||
|
# first operand, that might be a tuple, which we unpack below
|
||
|
if excName.kind == tupleExpr:
|
||
|
# This allows to do except (a, b, c) as SomeError {...}
|
||
|
# TODO: Consider adding the ability to make exc a sequence
|
||
|
# instead of adding the same body with different exception
|
||
|
# types each time
|
||
|
handlerBody = self.statement()
|
||
|
for element in TupleExpr(excName).members:
|
||
|
handlers.add((body: handlerBody, exc: element, name: asName))
|
||
|
continue
|
||
|
else:
|
||
|
excName = nil
|
||
|
handlerBody = self.statement()
|
||
|
handlers.add((body: handlerBody, exc: excName, name: asName))
|
||
|
asName = nil
|
||
|
if self.match(Else):
|
||
|
elseClause = self.statement()
|
||
|
if self.match(Finally):
|
||
|
finallyClause = self.statement()
|
||
|
if handlers.len() == 0 and elseClause == nil and finallyClause == nil:
|
||
|
self.error("expecting 'except', 'finally' or 'else' statements after 'try' block")
|
||
|
for i, handler in handlers:
|
||
|
if handler.exc == nil and i != handlers.high():
|
||
|
self.error("catch-all exception handler with bare 'except' must come last in try statement")
|
||
|
result = newTryStmt(body, handlers, finallyClause, elseClause, tok)
|
||
|
|
||
|
|
||
|
proc whileStmt(self: Parser): ASTNode =
|
||
|
## Parses a C-style while loop statement
|
||
|
let tok = self.peek(-1)
|
||
|
self.beginScope()
|
||
|
var enclosingLoop = self.currentLoop
|
||
|
self.currentLoop = Loop
|
||
|
self.expect(LeftParen, "expecting '(' before while loop condition")
|
||
|
var condition = self.expression()
|
||
|
self.expect(RightParen, "unterminated while loop condition")
|
||
|
result = newWhileStmt(condition, self.statement(), tok)
|
||
|
self.currentLoop = enclosingLoop
|
||
|
self.endScope()
|
||
|
|
||
|
|
||
|
proc forStmt(self: Parser): ASTNode =
|
||
|
## Parses a C-style for loop
|
||
|
self.beginScope()
|
||
|
let tok = self.peek(-1)
|
||
|
var enclosingLoop = self.currentLoop
|
||
|
self.currentLoop = Loop
|
||
|
self.expect(LeftParen, "expecting '(' after 'for'")
|
||
|
var initializer: ASTNode = nil
|
||
|
var condition: ASTNode = nil
|
||
|
var increment: ASTNode = nil
|
||
|
# The code below is not really that illuminating, but
|
||
|
# it's there to disallow weird things like a public for loop
|
||
|
# increment variable which doesn't really make sense, but still
|
||
|
# allow people that like verbosity (for *some* reason) to use
|
||
|
# private static var declarations as well as just private var
|
||
|
# and static var as well as providing decently specific error
|
||
|
# messages
|
||
|
if self.match(Semicolon):
|
||
|
discard
|
||
|
elif self.match(Dynamic):
|
||
|
self.error("dynamic declarations are not allowed in the foor loop initializer")
|
||
|
elif self.match(Public):
|
||
|
self.error("public declarations are not allowed in the for loop initializer")
|
||
|
elif self.match(Static):
|
||
|
self.expect(Var, "expecting 'var' after 'static' in for loop initializer")
|
||
|
initializer = self.varDecl(isStatic=true, isPrivate=true)
|
||
|
elif self.match(Private):
|
||
|
if self.match(Dynamic):
|
||
|
self.error("dynamic declarations are not allowed in the foor loop initializer")
|
||
|
elif self.match(Static):
|
||
|
self.expect(Var, "expecting 'var' after 'static' in for loop initializer")
|
||
|
initializer = self.varDecl(isStatic=true, isPrivate=true)
|
||
|
elif self.match(Var):
|
||
|
initializer = self.varDecl(isStatic=true, isPrivate=true)
|
||
|
elif self.match(Var):
|
||
|
initializer = self.varDecl(isStatic=true, isPrivate=true)
|
||
|
else:
|
||
|
initializer = self.expressionStatement()
|
||
|
if not self.check(Semicolon):
|
||
|
condition = self.expression()
|
||
|
self.expect(Semicolon, "expecting ';' after for loop condition")
|
||
|
if not self.check(RightParen):
|
||
|
increment = self.expression()
|
||
|
self.expect(RightParen, "unterminated for loop increment")
|
||
|
var body = self.statement()
|
||
|
if increment != nil:
|
||
|
# The increment runs after each iteration, so we
|
||
|
# inject it into the block as the last statement
|
||
|
body = newBlockStmt(@[body, newExprStmt(increment, increment.token)], tok)
|
||
|
if condition == nil:
|
||
|
## An empty condition is functionally
|
||
|
## equivalent to "true"
|
||
|
condition = newTrueExpr(Token())
|
||
|
# We can use a while loop, which in this case works just as well
|
||
|
body = newWhileStmt(condition, body, tok)
|
||
|
if initializer != nil:
|
||
|
# Nested blocks, so the initializer is
|
||
|
# only executed once
|
||
|
body = newBlockStmt(@[initializer, body], tok)
|
||
|
# This desgugars the following code:
|
||
|
# for (var i = 0; i < 10; i += 1) {
|
||
|
# print(i);
|
||
|
# }
|
||
|
# To the semantically equivalent snippet
|
||
|
# below:
|
||
|
# {
|
||
|
# private static var i = 0;
|
||
|
# while (i < 10) {
|
||
|
# print(i);
|
||
|
# i += 1;
|
||
|
# }
|
||
|
# }
|
||
|
result = body
|
||
|
self.currentLoop = enclosingLoop
|
||
|
self.endScope()
|
||
|
|
||
|
|
||
|
proc ifStmt(self: Parser): ASTNode =
|
||
|
## Parses if statements
|
||
|
let tok = self.peek(-1)
|
||
|
self.expect(LeftParen, "expecting '(' before if condition")
|
||
|
var condition = self.expression()
|
||
|
self.expect(RightParen, "expecting ')' after if condition")
|
||
|
var thenBranch = self.statement()
|
||
|
var elseBranch: ASTNode = nil
|
||
|
if self.match(Else):
|
||
|
elseBranch = self.statement()
|
||
|
result = newIfStmt(condition, thenBranch, elseBranch, tok)
|
||
|
|
||
|
|
||
|
template checkDecl(self: Parser, isStatic, isPrivate: bool) =
|
||
|
## Handy utility function that avoids us from copy
|
||
|
## pasting the same checks to all declaration handlers
|
||
|
if not isStatic and self.currentFunction != nil:
|
||
|
self.error("dynamic declarations are not allowed inside functions")
|
||
|
if not isStatic and self.scopeDepth > 0:
|
||
|
self.error("dynamic declarations are not allowed inside local scopes")
|
||
|
if not isPrivate and self.currentFunction != nil:
|
||
|
self.error("cannot bind public names inside functions")
|
||
|
if not isPrivate and self.scopeDepth > 0:
|
||
|
self.error("cannot bind public names inside local scopes")
|
||
|
|
||
|
|
||
|
proc varDecl(self: Parser, isStatic: bool = true, isPrivate: bool = true): ASTNode =
|
||
|
## Parses variable declarations
|
||
|
self.checkDecl(isStatic, isPrivate)
|
||
|
var varKind = self.peek(-1)
|
||
|
var keyword = ""
|
||
|
var value: ASTNode
|
||
|
case varKind.kind:
|
||
|
of Const:
|
||
|
# Note that isStatic being false is an error, because constants are replaced at compile-time
|
||
|
if not isStatic:
|
||
|
self.error("constant declarations cannot be dynamic")
|
||
|
keyword = "constant"
|
||
|
else:
|
||
|
keyword = "variable"
|
||
|
self.expect(Identifier, &"expecting {keyword} name after '{varKind.lexeme}'")
|
||
|
var name = newIdentExpr(self.peek(-1))
|
||
|
if self.match(Equal):
|
||
|
value = self.expression()
|
||
|
if varKind.kind == Const and not value.isConst():
|
||
|
self.error("the initializer for constant declarations must be a primitive and constant type")
|
||
|
else:
|
||
|
if varKind.kind == Const:
|
||
|
self.error("constant declaration requires an explicit initializer")
|
||
|
value = newNilExpr(Token())
|
||
|
self.expect(Semicolon, &"expecting semicolon after {keyword} declaration")
|
||
|
case varKind.kind:
|
||
|
of Var:
|
||
|
result = newVarDecl(name, value, isStatic=isStatic, isPrivate=isPrivate, token=varKind, owner=self.file, closedOver=false)
|
||
|
of Const:
|
||
|
result = newVarDecl(name, value, isConst=true, isPrivate=isPrivate, isStatic=true, token=varKind, owner=self.file, closedOver=false)
|
||
|
else:
|
||
|
discard # Unreachable
|
||
|
|
||
|
|
||
|
proc funDecl(self: Parser, isAsync: bool = false, isStatic: bool = true, isPrivate: bool = true, isLambda: bool = false): ASTNode =
|
||
|
## Parses function and lambda declarations. Note that lambdas count as expressions!
|
||
|
self.checkDecl(isStatic, isPrivate)
|
||
|
let tok = self.peek(-1)
|
||
|
var enclosingFunction = self.currentFunction
|
||
|
var arguments: seq[ASTNode] = @[]
|
||
|
var defaults: seq[ASTNode] = @[]
|
||
|
if not isLambda:
|
||
|
self.currentFunction = newFunDecl(nil, arguments, defaults, newBlockStmt(@[], Token()), isAsync=isAsync, isGenerator=false, isStatic=isStatic, isPrivate=isPrivate, token=tok, owner=self.file, closedOver=false)
|
||
|
else:
|
||
|
self.currentFunction = newLambdaExpr(arguments, defaults, newBlockStmt(@[], Token()), isGenerator=false, token=tok)
|
||
|
if not isLambda:
|
||
|
self.expect(Identifier, "expecting function name after 'fun'")
|
||
|
FunDecl(self.currentFunction).name = newIdentExpr(self.peek(-1))
|
||
|
if self.match(LeftBrace):
|
||
|
# Argument-less function
|
||
|
discard
|
||
|
else:
|
||
|
var parameter: IdentExpr
|
||
|
self.expect(LeftParen)
|
||
|
while not self.check(RightParen):
|
||
|
if arguments.len > 255:
|
||
|
self.error("cannot have more than 255 arguments in function declaration")
|
||
|
self.expect(Identifier)
|
||
|
parameter = newIdentExpr(self.peek(-1))
|
||
|
if parameter in arguments:
|
||
|
self.error("duplicate parameter name in function declaration")
|
||
|
arguments.add(parameter)
|
||
|
if self.match(Equal):
|
||
|
defaults.add(self.expression())
|
||
|
elif defaults.len() > 0:
|
||
|
self.error("positional argument(s) cannot follow default argument(s) in function declaration")
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
self.expect(RightParen)
|
||
|
self.expect(LeftBrace)
|
||
|
if not isLambda:
|
||
|
FunDecl(self.currentFunction).body = self.blockStmt()
|
||
|
FunDecl(self.currentFunction).arguments = arguments
|
||
|
else:
|
||
|
LambdaExpr(self.currentFunction).body = self.blockStmt()
|
||
|
LambdaExpr(self.currentFunction).arguments = arguments
|
||
|
result = self.currentFunction
|
||
|
self.currentFunction = enclosingFunction
|
||
|
|
||
|
|
||
|
proc classDecl(self: Parser, isStatic: bool = true, isPrivate: bool = true): ASTNode =
|
||
|
## Parses class declarations
|
||
|
self.checkDecl(isStatic, isPrivate)
|
||
|
let tok = self.peek(-1)
|
||
|
var parents: seq[ASTNode] = @[]
|
||
|
self.expect(Identifier)
|
||
|
var name = newIdentExpr(self.peek(-1))
|
||
|
if self.match(LessThan):
|
||
|
while true:
|
||
|
self.expect(Identifier)
|
||
|
parents.add(newIdentExpr(self.peek(-1)))
|
||
|
if not self.match(Comma):
|
||
|
break
|
||
|
self.expect(LeftBrace)
|
||
|
result = newClassDecl(name, self.blockStmt(), isPrivate=isPrivate, isStatic=isStatic, parents=parents, token=tok, owner=self.file, closedOver=false)
|
||
|
|
||
|
|
||
|
proc expression(self: Parser): ASTNode =
|
||
|
## Parses expressions
|
||
|
result = self.assignment()
|
||
|
|
||
|
|
||
|
proc expressionStatement(self: Parser): ASTNode =
|
||
|
## Parses expression statements, which
|
||
|
## are expressions followed by a semicolon
|
||
|
var expression = self.expression()
|
||
|
endOfLine("missing semicolon after expression")
|
||
|
result = newExprStmt(expression, expression.token)
|
||
|
|
||
|
|
||
|
proc statement(self: Parser): ASTNode =
|
||
|
## Parses statements
|
||
|
case self.peek().kind:
|
||
|
of If:
|
||
|
discard self.step()
|
||
|
result = self.ifStmt()
|
||
|
of Del:
|
||
|
discard self.step()
|
||
|
result = self.delStmt()
|
||
|
of Assert:
|
||
|
discard self.step()
|
||
|
result = self.assertStmt()
|
||
|
of Raise:
|
||
|
discard self.step()
|
||
|
result = self.raiseStmt()
|
||
|
of Break:
|
||
|
discard self.step()
|
||
|
result = self.breakStmt()
|
||
|
of Continue:
|
||
|
discard self.step()
|
||
|
result = self.continueStmt()
|
||
|
of Return:
|
||
|
discard self.step()
|
||
|
result = self.returnStmt()
|
||
|
of Import:
|
||
|
discard self.step()
|
||
|
result = self.importStmt()
|
||
|
of From:
|
||
|
discard self.step()
|
||
|
result = self.fromStmt()
|
||
|
of While:
|
||
|
discard self.step()
|
||
|
result = self.whileStmt()
|
||
|
of For:
|
||
|
discard self.step()
|
||
|
result = self.forStmt()
|
||
|
of Foreach:
|
||
|
discard self.step()
|
||
|
result = self.forEachStmt()
|
||
|
of LeftBrace:
|
||
|
discard self.step()
|
||
|
result = self.blockStmt()
|
||
|
of Yield:
|
||
|
discard self.step()
|
||
|
result = self.yieldStmt()
|
||
|
of Await:
|
||
|
discard self.step()
|
||
|
result = self.awaitStmt()
|
||
|
of Defer:
|
||
|
discard self.step()
|
||
|
result = self.deferStmt()
|
||
|
of Try:
|
||
|
discard self.step()
|
||
|
result = self.tryStmt()
|
||
|
else:
|
||
|
result = self.expressionStatement()
|
||
|
|
||
|
|
||
|
proc declaration(self: Parser): ASTNode =
|
||
|
## Parses declarations
|
||
|
case self.peek().kind:
|
||
|
of Var, Const:
|
||
|
discard self.step()
|
||
|
result = self.varDecl()
|
||
|
of Class:
|
||
|
discard self.step()
|
||
|
result = self.classDecl()
|
||
|
of Fun:
|
||
|
discard self.step()
|
||
|
result = self.funDecl()
|
||
|
of Private, Public:
|
||
|
discard self.step()
|
||
|
var isStatic: bool = true
|
||
|
let isPrivate = if self.peek(-1).kind == Private: true else: false
|
||
|
if self.match(Dynamic):
|
||
|
isStatic = false
|
||
|
elif self.match(Static):
|
||
|
discard # This is just to allow an "explicit" static keyword
|
||
|
if self.match(Async):
|
||
|
result = self.funDecl(isStatic=isStatic, isPrivate=isPrivate, isAsync=true)
|
||
|
else:
|
||
|
case self.peek().kind:
|
||
|
of Var, Const:
|
||
|
discard self.step()
|
||
|
result = self.varDecl(isStatic=isStatic, isPrivate=isPrivate)
|
||
|
of Class:
|
||
|
discard self.step()
|
||
|
result = self.classDecl(isStatic=isStatic, isPrivate=isPrivate)
|
||
|
of Fun:
|
||
|
discard self.step()
|
||
|
result = self.funDecl(isStatic=isStatic, isPrivate=isPrivate)
|
||
|
else:
|
||
|
self.error("expecting declaration")
|
||
|
of Static, Dynamic:
|
||
|
discard self.step()
|
||
|
let isStatic: bool = if self.peek(-1).kind == Static: true else: false
|
||
|
if self.match(Async):
|
||
|
self.expect(Fun)
|
||
|
result = self.funDecl(isStatic=isStatic, isPrivate=true, isAsync=true)
|
||
|
else:
|
||
|
case self.peek().kind:
|
||
|
of Var, Const:
|
||
|
discard self.step()
|
||
|
result = self.varDecl(isStatic=isStatic, isPrivate=true)
|
||
|
of Class:
|
||
|
discard self.step()
|
||
|
result = self.classDecl(isStatic=isStatic, isPrivate=true)
|
||
|
of Fun:
|
||
|
discard self.step()
|
||
|
result = self.funDecl(isStatic=isStatic, isPrivate=true)
|
||
|
else:
|
||
|
self.error("expecting declaration")
|
||
|
of Async:
|
||
|
discard self.step()
|
||
|
self.expect(Fun)
|
||
|
result = self.funDecl(isAsync=true)
|
||
|
|
||
|
else:
|
||
|
result = self.statement()
|
||
|
|
||
|
|
||
|
proc parse*(self: Parser, tokens: seq[Token], file: string): seq[ASTNode] =
|
||
|
## Parses a series of tokens into an AST node
|
||
|
self.tokens = tokens
|
||
|
self.file = file
|
||
|
self.current = 0
|
||
|
self.currentLoop = None
|
||
|
self.currentFunction = nil
|
||
|
self.scopeDepth = 0
|
||
|
while not self.done():
|
||
|
result.add(self.declaration())
|