Pretty print function definitions

src/pretty.rs

@@ -2,6 +2,7 @@ mod basic;
 mod call;
 mod expr;
 mod field;
+mod func_def;
 mod lit;
 mod program;
 mod table_constr;

src/pretty/expr.rs

@@ -31,7 +31,7 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for Expr {
             Self::Var(var) => var.pretty(allocator),
             Self::TableConstr(constr) => constr.pretty(allocator),
             Self::TableDestr(destr) => destr.pretty(allocator),
-            Self::FuncDef(def) => allocator.text("<def>"),
+            Self::FuncDef(def) => def.pretty(allocator),
             Self::Paren {
                 s0,
                 inner,

src/pretty/func_def.rs

@@ -0,0 +1,98 @@
+use pretty::{DocAllocator, DocBuilder, Pretty};
+
+use crate::ast::FuncDef;
+
+impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for FuncDef {
+    fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
+        match self {
+            Self::AnonNoArg {
+                s0,
+                s1,
+                s2,
+                body,
+                span: _,
+            } => allocator.text("function() ").append(body.pretty(allocator)),
+
+            Self::AnonArg {
+                s0,
+                s1,
+                arg,
+                s2,
+                s3,
+                body,
+                span: _,
+            } => allocator
+                .text("function")
+                .append(arg.pretty(allocator).parens())
+                .append(allocator.space())
+                .append(body.pretty(allocator)),
+
+            Self::AnonDestr {
+                s0,
+                pattern,
+                s1,
+                body,
+                span: _,
+            } => allocator
+                .text("function")
+                .append(pattern.pretty(allocator))
+                .append(allocator.space())
+                .append(body.pretty(allocator)),
+
+            Self::NamedNoArg {
+                local,
+                s0,
+                name,
+                s1,
+                s2,
+                s3,
+                body,
+                span: _,
+            } => local
+                .map(|s| allocator.text("local "))
+                .unwrap_or_else(|| allocator.nil())
+                .append(allocator.text("function "))
+                .append(name)
+                .append(allocator.text("() "))
+                .append(body.pretty(allocator)),
+
+            Self::NamedArg {
+                local,
+                s0,
+                name,
+                s1,
+                s2,
+                arg,
+                s3,
+                s4,
+                body,
+                span: _,
+            } => local
+                .map(|s| allocator.text("local "))
+                .unwrap_or_else(|| allocator.nil())
+                .append(allocator.text("function "))
+                .append(name)
+                .append(arg.pretty(allocator).parens())
+                .append(allocator.space())
+                .append(body.pretty(allocator)),
+
+            Self::NamedDestr {
+                local,
+                s0,
+                name,
+                s1,
+                pattern,
+                s2,
+                body,
+                span: _,
+            } => local
+                .map(|s| allocator.text("local "))
+                .unwrap_or_else(|| allocator.nil())
+                .append(allocator.text("function "))
+                .append(name)
+                .append(pattern.pretty(allocator))
+                .append(allocator.space())
+                .append(body.pretty(allocator)),
+        }
+    }
+}