Pretty print function calls

src/pretty.rs

@@ -1,4 +1,5 @@
 mod basic;
+mod call;
 mod expr;
 mod lit;
 mod program;

src/pretty/call.rs

@@ -0,0 +1,32 @@
+use pretty::{DocAllocator, DocBuilder, Pretty};
+
+use crate::ast::Call;
+
+impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for Call {
+    fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
+        match self {
+            Self::Arg {
+                expr,
+                s0,
+                s1,
+                arg,
+                s2,
+                span: _,
+            } => expr
+                .pretty(allocator)
+                .append(arg.pretty(allocator).parens()),
+            Self::NoArg {
+                expr,
+                s0,
+                s1,
+                span: _,
+            } => expr.pretty(allocator).append(allocator.nil().parens()),
+            Self::Constr {
+                expr,
+                s0,
+                constr,
+                span: _,
+            } => expr.pretty(allocator).append(constr.pretty(allocator)),
+        }
+    }
+}

src/pretty/expr.rs

@@ -26,7 +26,7 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for Expr {
     fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
         match self {
             Self::Lit(lit) => lit.pretty(allocator),
-            Self::Call(call) => allocator.text("<call>"),
+            Self::Call(call) => call.pretty(allocator),
             Self::Field(field) => allocator.text("<field>"),
             Self::Var(var) => allocator.text("<var>"),
             Self::TableConstr(constr) => constr.pretty(allocator),