render: wip

Author: StunxFS

bsc/lua_ast/__init__.py

@@ -25,6 +25,28 @@ def __init__(self, name, args):
         self.args = args
         self.stmts = []
 
+# Statements
+
+class LuaWhile:
+    def __init__(self, cond, stmts):
+        self.cond = cond
+        self.stmts = stmts
+
+class LuaRepeat:
+    def __init__(self, stmts, cond):
+        self.stmts = stmts
+        self.cond = cond
+
+class LuaIf:
+    def __init__(self, branches):
+        self.branches = branches
+
+class LuaIfBranch:
+    def __init__(self, cond, is_else, stmts):
+        self.cond = cond
+        self.is_else = is_else
+        self.stmts = stmts
+
 class LuaBlock:
     def __init__(self):
         self.chunk = []
@@ -35,9 +57,33 @@ def __init__(self, lefts, op, rights):
         self.op = op
         self.rights = rights
 
+# Expressions
+
+class LuaCallExpr:
+    def __init__(self, left, args):
+        self.left = left
+        self.args = args
+
+class LuaSelector:
+    def __init__(self, left, name):
+        self.left = left
+        self.name = name
+
 class LuaIdent:
     def __init__(self, name):
         self.name = name
 
+class LuaParenExpr:
+    def __init__(self, expr):
+        self.expr = expr
+
+class LuaNumberLit:
+    def __init__(self, value):
+        self.value = value
+
+class LuaBooleanLit:
+    def __init__(self, value):
+        self.value = value
+
 class LuaNil:
     pass

bsc/lua_ast/render.py

@@ -83,10 +83,69 @@ def render_fn_decl(self, decl):
             if i < len(decl.args) - 1:
                 self.write(", ")
         self.writeln(")")
-        #self.indent += 1
-        #self.indent -= 1
+        self.indent += 1
+        self.render_stmts(decl.stmts)
+        self.indent -= 1
         self.writeln("end\n")
 
+    def render_stmts(self, stmts):
+        for stmt in stmts:
+            self.render_stmt(stmt)
+
+    def render_stmt(self, stmt):
+        if isinstance(stmt, LuaWhile):
+            self.write("while ")
+            self.render_expr(stmt.cond)
+            self.writeln(" do")
+            self.indent += 1
+            self.render_stmts(stmt.stmts)
+            self.indent -= 1
+            self.writeln("end")
+        elif isinstance(stmt, LuaRepeat):
+            self.writeln("repeat")
+            self.indent += 1
+            self.render_stmts(stmt.stmts)
+            self.indent -= 1
+            self.write("until ")
+            self.render_expr(stmt.cond)
+            self.writeln()
+        elif isinstance(stmt, LuaIf):
+            for i, branch in enumerate(stmt.branches):
+                if branch.is_else:
+                    self.writeln("else")
+                else:
+                    self.write("if " if i == 0 else "elseif ")
+                    self.render_expr(branch.cond)
+                self.indent += 1
+                self.render_stmts(branch.stmts)
+                self.indent -= 1
+            self.writeln("end")
+
+    def render_expr(self, expr):
+        if isinstance(expr, LuaParenExpr):
+            self.write("(")
+            self.render_expr(expr.expr)
+            self.write(")")
+        elif isinstance(expr, LuaCallExpr):
+            self.render_expr(expr.left)
+            self.write("(")
+            for i, arg in enumerate(expr.args):
+                self.render_expr(arg)
+                if i < len(expr.args) - 1:
+                    self.write(", ")
+            self.write(")")
+        elif isinstance(expr, LuaSelector):
+            self.render_expr(expr.left)
+            self.write(f".{expr.name}")
+        elif isinstance(expr, LuaIdent):
+            self.write(expr.name)
+        elif isinstance(expr, (LuaBooleanLit, LuaNumberLit)):
+            self.write(expr.value)
+        elif isinstance(expr, LuaNil):
+            self.write("nil")
+
+    ## Utils
+
     def write(self, s):
         if self.indent > 0 and self.empty_line:
             self.lua_file.write("\t" * self.indent)