1 files changed, 46 insertions, 0 deletions
diff --git a/submit/ast/FunDeclaration.java b/submit/ast/FunDeclaration.java
new file mode 100644
index 0000000..1ec2ca6
--- /dev/null
+++ b/submit/ast/FunDeclaration.java
@@ -0,0 +1,46 @@
+/*
+ * Code formatter project
+ * CS 4481
+ */
+package submit.ast;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ *
+ * @author edwajohn
+ */
+public class FunDeclaration extends AbstractNode implements Declaration {
+
+  private final VarType returnType;
+  private final String id;
+  private final ArrayList<Param> params;
+  private final Statement statement;
+
+  public FunDeclaration(VarType returnType, String id, List<Param> params,
+      Statement statement) {
+    this.returnType = returnType;
+    this.id = id;
+    this.params = new ArrayList<>(params);
+    this.statement = statement;
+  }
+
+  public void toCminus(StringBuilder builder, final String prefix) {
+    String rt = (returnType != null) ? returnType.toString() : "void";
+    builder.append("\n").append(rt).append(" ");
+    builder.append(id);
+    builder.append("(");
+
+    for (Param param : params) {
+      param.toCminus(builder, prefix);
+      builder.append(", ");
+    }
+    if (!params.isEmpty()) {
+      builder.delete(builder.length() - 2, builder.length());
+    }
+    builder.append(")\n");
+    statement.toCminus(builder, prefix);
+  }
+
+}