1 files changed, 33 insertions, 0 deletions
diff --git a/submit/ast/While.java b/submit/ast/While.java
new file mode 100644
index 0000000..a37195e
--- /dev/null
+++ b/submit/ast/While.java
@@ -0,0 +1,33 @@
+/*
+ * Code formatter project
+ * CS 4481
+ */
+package submit.ast;
+
+/**
+ *
+ * @author edwajohn
+ */
+public class While extends AbstractNode implements Statement {
+
+  private final Expression expression;
+  private final Statement statement;
+
+  public While(Expression expression, Statement statement) {
+    this.expression = expression;
+    this.statement = statement;
+  }
+
+  @Override
+  public void toCminus(StringBuilder builder, String prefix) {
+    builder.append(prefix).append("while (");
+    expression.toCminus(builder, prefix);
+    builder.append(")\n");
+    if (statement instanceof CompoundStatement) {
+      statement.toCminus(builder, prefix);
+    } else {
+      statement.toCminus(builder, prefix + " ");
+    }
+
+  }
+}