33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from smnp.error.syntax import SyntaxException
|
|
from smnp.newast.node.expression import ExpressionNode
|
|
from smnp.newast.node.extend import ExtendNode
|
|
from smnp.newast.node.function import FunctionDefinitionNode
|
|
from smnp.newast.node.imports import ImportNode
|
|
from smnp.newast.node.model import Node, ParseResult
|
|
from smnp.newast.node.statement import StatementNode
|
|
|
|
from smnp.newast.parser import Parser
|
|
|
|
|
|
class Program(Node):
|
|
def __init__(self):
|
|
super().__init__((-1, -1))
|
|
|
|
@classmethod
|
|
def _parse(cls, input):
|
|
def parseToken(input):
|
|
return Parser.oneOf(
|
|
FunctionDefinitionNode.parse,
|
|
ExtendNode.parse,
|
|
ExpressionNode.parse,
|
|
ImportNode.parse,
|
|
StatementNode.parse,
|
|
exception = SyntaxException(f"Unknown statement: {input.current().pos}")
|
|
)(input)
|
|
|
|
root = Program()
|
|
while input.hasCurrent():
|
|
result = parseToken(input)
|
|
if result.result:
|
|
root.append(result.node)
|
|
return ParseResult.OK(root) |