Add Ast class (#29)

* chore: add `Ast`class

* chore: PR comments

src/executor.py

@@ -1,6 +1,7 @@
 import inspect
 
 from bindings import Bindings
+from parser import Ast
 
 
 class UnknownSubroutine(Exception):
@@ -12,7 +13,7 @@ class InvalidMemoryAddressException(Exception):
 
 
 class ImperivmExecutor:
-    def __init__(self, ast):
+    def __init__(self, ast: Ast):
         self.subroutines = {}
         self.stack = []
         self.heap = []

src/parser.py

@@ -7,6 +7,17 @@
 import preprocessor
 
 
+class Ast:
+    def __init__(self, tree: tuple):
+        self.tree = tree
+
+    def __iter__(self):
+        return self.tree.__iter__()
+
+    def __repr__(self):
+        return self.tree.__repr__()
+
+
 class ImperivmParser:
     def __init__(
         self,
@@ -18,10 +29,10 @@ def __init__(
         self.visitor = visitor
         self.preprocessor = preprocessor
 
-    def parse(self, program):
+    def parse(self, program: str):
         program = self.preprocessor.process(program)
         tree = self.grammar.parse(program)
-        return self.visitor.visit(tree)
+        return Ast(self.visitor.visit(tree))
 
 
 if __name__ == "__main__":