Add support for custom functions

2019-07-06 00:20:36 +02:00
parent 2ecc86a9b2
commit f11b3b67f2
8 changed files with 168 additions and 6 deletions
--- a/smnp/newast/node/statement.py
+++ b/smnp/newast/node/statement.py
@@ -7,9 +7,13 @@ class StatementNode(Node):
    @classmethod
    def _parse(cls, input):
        from smnp.newast.node.block import BlockNode
+        from smnp.newast.node.function import FunctionDefinition
        from smnp.newast.node.expression import ExpressionNode
+        from smnp.newast.node.ret import ReturnNode

        return Parser.oneOf(
            BlockNode.parse,
+            FunctionDefinition.parse,
+            ReturnNode.parse,
            ExpressionNode.parse
        )(input)