Merge pull request #4 from luizmoitinho/erick

Iniciar desenvolvimento do PrettyPrinter

src/ExpressionLanguageParser.py

@@ -690,13 +690,11 @@ def p_error(p):
 lex.lex()
 arquivo = '''
 <?php
-  do
-  {
-    while(1+1){
-      $valor = $valor2;
+  function add() {
+    while(true) {
+      return 1;
     }
-
-  }while($a = 2);
+  }
 ?>
 '''
 

src/PrettyPrinter.py

@@ -0,0 +1,18 @@
+spaces = '    '
+space_buffer = ''
+
+class PrettyPrinter():
+
+  def incrementTab():
+    global spaces 
+    global space_buffer
+    space_buffer = space_buffer + spaces
+
+
+  def printTab():
+    global space_buffer
+    print(space_buffer, end='')
+
+  def decrementTab():
+    global space_buffer
+    space_buffer = space_buffer[:-len(spaces)]

src/Visitor.py

@@ -1,7 +1,11 @@
+from PrettyPrinter import PrettyPrinter as pp
+
 class Visitor():
 
   def visitMain_MainInner(self, main):
     print('<?php')
+    pp.incrementTab()
+    pp.printTab()
     main.mainInner.accept(self)
     print('?>')
 
@@ -53,7 +57,10 @@ def visitFds_parameter_noParameter(self, fds_parameter):
 
   def visitFds_statements_withStatements(self, fds_statements):
     print('{')
+    pp.incrementTab()
     fds_statements.inner_statement_MUL.accept(self)
+    pp.decrementTab()
+    pp.printTab()
     print('}')
 
   def visitFds_statements_noStatements(self, fds_statements):
@@ -125,23 +132,29 @@ def visitScalar_Token(self, scalar):
     print(scalar.token, end='')
 
   def visitStatement_Expr(self, statement):
+    pp.printTab()
     statement.expr.accept(self)
     print(';')
 
   def visitStatement_Break(self, statement):
+    pp.printTab()
     statement._break.accept(self)
 
   def visitStatement_Continue(self, statement):
+    pp.printTab()
     statement._continue.accept(self)
 
   def visitStatement_Return(self, statement):
+    pp.printTab()
     statement._return.accept(self)
 
   def visitStatement_Exit(self, statement):
+    pp.printTab()
     statement.exit.accept(self)
     print(';')
 
   def visitStatement_While(self, statement):
+    pp.printTab()
     statement.whilee.accept(self)
 
   def visitStatement_Do_While(self, statement):
@@ -150,6 +163,7 @@ def visitStatement_Do_While(self, statement):
 
 
   def visitStatement_Die(self, statement):
+    pp.printTab()
     statement.die.accept(self)
     print(';')