Enable support for custom functions definition

smnp/ast/node/function.py

@@ -1,128 +1,127 @@
-# from smnp.ast.node.block import BlockNode
-# from smnp.ast.node.expression import ExpressionNode
-# from smnp.ast.node.identifier import IdentifierNode
-# from smnp.ast.node.iterable import abstractIterableParser
-# from smnp.ast.node.model import Node
-# from smnp.ast.node.none import NoneNode
-# from smnp.ast.node.statement import StatementNode
-# from smnp.ast.node.type import TypeNode, TypeSpecifier
-# from smnp.ast.parser import Parser
-# from smnp.token.type import TokenType
-#
-#
-# class ArgumentsDeclarationNode(Node):
-#
-#     @classmethod
-#     def _parse(cls, input):
-#         raise RuntimeError("This class is not supposed to be automatically called")
-#
-#
-# class VarargNode(Node):
-#     pass
-#
-#
-# class ArgumentDefinitionNode(ExpressionNode):
-#     def __init__(self, pos):
-#         super().__init__(pos)
-#         self.children.extend([NoneNode(), False])
-#
-#     @property
-#     def type(self):
-#         return self[0]
-#
-#     @type.setter
-#     def type(self, value):
-#         self[0] = value
-#
-#     @property
-#     def variable(self):
-#         return self[1]
-#
-#     @variable.setter
-#     def variable(self, value):
-#         self[1] = value
-#
-#     @property
-#     def vararg(self):
-#         return self[2]
-#
-#     @vararg.setter
-#     def vararg(self, value):
-#         self[2] = value
-#
-#
-#     @classmethod
-#     def parser(cls):
-#         def createNode(type, variable, dots):
-#             node = ArgumentDefinitionNode(type.pos)
-#             node.type = type
-#             node.variable = variable
-#             node.vararg = isinstance(dots, VarargNode)
-#             return node
-#
-#         return Parser.allOf(
-#             Parser.optional(Parser.oneOf(
-#                 TypeNode.parse,
-#                 TypeSpecifier.parse
-#             )),
-#             Parser.doAssert(IdentifierNode.identifierParser(), "variable name"),
-#             Parser.optional(Parser.terminalParser(TokenType.DOTS, lambda val, pos: VarargNode(pos))),
-#             createNode=createNode
-#         )
-#
-#     @classmethod
-#     def _parse(cls, input):
-#         #TODO
-#         raise RuntimeError("Not implemented yet. There is still required work to correctly build AST related to IdentifierNode")
-#
-#
-#
-# class FunctionDefinitionNode(StatementNode):
-#     def __init__(self, pos):
-#         super().__init__(pos)
-#         self.children = [NoneNode(), NoneNode(), NoneNode()]
-#
-#     @property
-#     def name(self):
-#         return self[0]
-#
-#     @name.setter
-#     def name(self, value):
-#         self[0] = value
-#
-#     @property
-#     def arguments(self):
-#         return self[1]
-#
-#     @arguments.setter
-#     def arguments(self, value):
-#         self[1] = value
-#
-#     @property
-#     def body(self):
-#         return self[2]
-#
-#     @body.setter
-#     def body(self, value):
-#         self[2] = value
-#
-#     @classmethod
-#     def _parse(cls, input):
-#         def createNode(function, name, arguments, body):
-#             node = FunctionDefinitionNode(function.pos)
-#             node.name = name
-#             node.arguments = arguments
-#             node.body = body
-#             return node
-#
-#         return Parser.allOf(
-#             Parser.terminalParser(TokenType.FUNCTION),
-#             Parser.doAssert(IdentifierNode.identifierParser(), "function name"),
-#             Parser.doAssert(cls._argumentsDeclarationParser(), "arguments list"),
-#             Parser.doAssert(BlockNode.parse, "function body"),
-#             createNode=createNode
-#         )(input)
-#
-#     @staticmethod
-#     def _argumentsDeclarationParser():
-#         return abstractIterableParser(ArgumentsDeclarationNode, TokenType.OPEN_PAREN, TokenType.CLOSE_PAREN, ArgumentDefinitionNode.parser())
+from smnp.ast.node.block import BlockParser
+from smnp.ast.node.identifier import IdentifierLiteralParser
+from smnp.ast.node.iterable import abstractIterableParser
+from smnp.ast.node.model import Node
+from smnp.ast.node.none import NoneNode
+from smnp.ast.node.type import TypeParser, Type
+from smnp.ast.parser import Parser
+from smnp.token.type import TokenType
+
+
+class ArgumentsDeclaration(Node):
+    pass
+
+
+class Argument(Node):
+
+    def __init__(self, pos):
+        super().__init__(pos)
+        self.children = [NoneNode(), NoneNode(), False]
+
+    @property
+    def type(self):
+        return self[0]
+
+
+    @type.setter
+    def type(self, value):
+        self[0] = value
+
+
+    @property
+    def variable(self):
+        return self[1]
+
+
+    @variable.setter
+    def variable(self, value):
+        self[1] = value
+
+
+    @property
+    def vararg(self):
+        return self[2]
+
+
+    @vararg.setter
+    def vararg(self, value):
+        self[2] = value
+
+
+class VarargNode(Node):
+    pass
+
+
+class FunctionDefinition(Node):
+    def __init__(self, pos):
+        super().__init__(pos)
+        self.children = [NoneNode(), NoneNode(), NoneNode()]
+
+    @property
+    def name(self):
+        return self[0]
+
+    @name.setter
+    def name(self, value):
+        self[0] = value
+
+    @property
+    def arguments(self):
+        return self[1]
+
+    @arguments.setter
+    def arguments(self, value):
+        self[1] = value
+
+    @property
+    def body(self):
+        return self[2]
+
+    @body.setter
+    def body(self, value):
+        self[2] = value
+
+    @classmethod
+    def withValues(cls, name, arguments, body):
+        node = cls(name.pos)
+        node.name = name
+        node.arguments = arguments
+        node.body = body
+        return node
+
+
+def ArgumentParser(input):
+    def createNode(type, variable, vararg):
+        pos = type.pos if isinstance(type, Type) else variable.pos
+        node = Argument(pos)
+        node.type = type
+        node.variable = variable
+        node.vararg = vararg is True
+        return node
+
+    return Parser.allOf(
+        Parser.optional(TypeParser),
+        IdentifierLiteralParser,
+        Parser.optional(Parser.terminalParser(TokenType.DOTS, lambda val, pos: True)),
+        createNode=createNode,
+        name="function argument"
+    )(input)
+
+
+def ArgumentsDeclarationParser(input):
+    return abstractIterableParser(
+        ArgumentsDeclaration,
+        TokenType.OPEN_PAREN,
+        TokenType.CLOSE_PAREN,
+        ArgumentParser
+    )(input)
+
+
+def FunctionDefinitionParser(input):
+    return Parser.allOf(
+        Parser.terminalParser(TokenType.FUNCTION),
+        IdentifierLiteralParser,
+        ArgumentsDeclarationParser,
+        BlockParser,
+        createNode=lambda _, name, args, body: FunctionDefinition.withValues(name, args, body)
+    )(input)

smnp/ast/node/identifier.py

@@ -49,17 +49,21 @@ class Assignment(BinaryOperator):
     pass
 
 
+def IdentifierLiteralParser(input):
+    return Parser.terminalParser(TokenType.IDENTIFIER, createNode=Identifier.withValue)(input)
+
+
 def IdentifierParser(input):
-    identifierLiteralParser = Parser.terminalParser(TokenType.IDENTIFIER, createNode=Identifier.withValue)
+
 
     functionCallParser = Parser.allOf(
-        identifierLiteralParser,
+        IdentifierLiteralParser,
         abstractIterableParser(ArgumentsList, TokenType.OPEN_PAREN, TokenType.CLOSE_PAREN, ExpressionParser),
         createNode=lambda name, arguments: FunctionCall.withChildren(name, arguments)
     )
 
     assignmentParser = Parser.allOf(
-        identifierLiteralParser,
+        IdentifierLiteralParser,
         Parser.terminalParser(TokenType.ASSIGN, createNode=Operator.withValue),
         ExpressionParser,
         createNode=lambda identifier, assign, expr: Assignment.withValues(identifier, assign, expr)
@@ -68,5 +72,5 @@ def IdentifierParser(input):
     return Parser.oneOf(
         assignmentParser,
         functionCallParser,
-        identifierLiteralParser
+        IdentifierLiteralParser
     )(input)

smnp/ast/node/program.py

@@ -1,3 +1,4 @@
+from smnp.ast.node.function import FunctionDefinitionParser
 from smnp.ast.node.imports import ImportParser
 from smnp.ast.node.model import Node, ParseResult
 from smnp.ast.node.statement import StatementParser
@@ -15,6 +16,7 @@ def ProgramParser(input):
             result = Parser.oneOf(
                 # Start Symbol
                 ImportParser,
+                FunctionDefinitionParser,
                 StatementParser,
                 exception=RuntimeError("Nie znam tego wyrazenia")
             )(input)