17 lines
434 B
Python
17 lines
434 B
Python
from smnp.ast.node.ret import ReturnNode
|
|
from smnp.ast.parsers.expression import parseExpression
|
|
from smnp.token.type import TokenType
|
|
|
|
|
|
def parseReturn(input, parent):
|
|
if input.current().type == TokenType.RETURN:
|
|
token = input.current()
|
|
input.ahead()
|
|
|
|
expr = parseExpression(input, parent)
|
|
|
|
node = ReturnNode(expr, parent, token.pos)
|
|
expr.parent = node
|
|
|
|
return node
|
|
return None |