Add function definitions to ast

src/ast/func_def.rs

@@ -0,0 +1,95 @@
+use crate::span::{HasSpan, Span};
+
+use super::{Expr, Ident, Space, TablePattern};
+
+#[derive(Debug, Clone)]
+pub enum FuncDef {
+    /// `function() a
+    ///
+    /// Structure: `function s0 ( s1 ) s2 body`
+    AnonNoArg {
+        s0: Space,
+        s1: Space,
+        s2: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+
+    /// `function(foo) a`
+    ///
+    /// Structure: `function s0 ( s1 arg s2 ) s3 body`
+    AnonArg {
+        s0: Space,
+        s1: Space,
+        arg: Ident,
+        s2: Space,
+        s3: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+
+    /// `function{..} a`
+    ///
+    /// Structure: `function s0 pattern s1 body`
+    AnonDestr {
+        s0: Space,
+        pattern: TablePattern,
+        s1: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+
+    /// `function foo() a`
+    ///
+    /// Structure: `function s0 name s1 ( s2 ) s3 body`
+    NamedNoArg {
+        s0: Space,
+        name: Ident,
+        s1: Space,
+        s2: Space,
+        s3: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+
+    /// `function foo(bar) a`
+    ///
+    /// Structure: `function s0 name s1 ( s2 arg s3 ) s4 body`
+    NamedArg {
+        s0: Space,
+        name: Ident,
+        s1: Space,
+        s2: Space,
+        arg: Ident,
+        s3: Space,
+        s4: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+
+    /// `function foo{..} a`
+    ///
+    /// Structure: `function s0 name s1 pattern s2 body`
+    NamedDestr {
+        s0: Space,
+        name: Ident,
+        s1: Space,
+        pattern: TablePattern,
+        s2: Space,
+        body: Box<Expr>,
+        span: Span,
+    },
+}
+
+impl HasSpan for FuncDef {
+    fn span(&self) -> Span {
+        match self {
+            FuncDef::AnonNoArg { span, .. } => *span,
+            FuncDef::AnonArg { span, .. } => *span,
+            FuncDef::AnonDestr { span, .. } => *span,
+            FuncDef::NamedNoArg { span, .. } => *span,
+            FuncDef::NamedArg { span, .. } => *span,
+            FuncDef::NamedDestr { span, .. } => *span,
+        }
+    }
+}