Add 'import' statements

smnp/newast/node/imports.py

@@ -0,0 +1,76 @@
+from smnp.newast.node.identifier import IdentifierNode
+from smnp.newast.node.model import Node
+from smnp.newast.node.none import NoneNode
+from smnp.newast.node.string import StringLiteralNode
+from smnp.newast.node.type import TypeNode
+from smnp.newast.parser import Parser
+from smnp.token.type import TokenType
+
+
+class ImportNode(Node):
+    def __init__(self, pos):
+        super().__init__(pos)
+        self.children = [NoneNode(), NoneNode(), NoneNode()]
+
+    @property
+    def source(self):
+        return self[0]
+
+    @source.setter
+    def source(self, value):
+        self[0] = value
+
+    @property
+    def type(self):
+        return self[1]
+
+    @type.setter
+    def type(self, value):
+        self[1] = value
+
+    @property
+    def variable(self):
+        return self[2]
+
+    @variable.setter
+    def variable(self, value):
+        self[2] = value
+
+    @classmethod
+    def _parse(cls, input):
+        return Parser.oneOf(
+            cls._literalImportParser(),
+            cls._fileImportParser()
+        )(input)
+
+    @classmethod
+    def _literalImportParser(cls):
+        def createNode(importKeyword, type, fromKeyword, source, asKeyword, variable):
+            node = ImportNode(importKeyword.pos)
+            node.source = source
+            node.type = type
+            node.variable = variable
+            return node
+
+        return Parser.allOf(
+            Parser.terminalParser(TokenType.IMPORT),
+            TypeNode.parse,
+            Parser.terminalParser(TokenType.FROM),
+            StringLiteralNode._literalParser(),
+            Parser.terminalParser(TokenType.AS),
+            IdentifierNode.identifierParser(),
+            createNode=createNode
+        )
+
+    @classmethod
+    def _fileImportParser(cls):
+        def createNode(importKeyword, source):
+            node = ImportNode(importKeyword.pos)
+            node.source = source
+            return node
+
+        return Parser.allOf(
+            Parser.terminalParser(TokenType.IMPORT),
+            StringLiteralNode._literalParser(),
+            createNode=createNode
+        )