Add local function statement

Author: Azganoth

corpus/functions.txt

@@ -63,3 +63,26 @@ t.a.b.c.f = function() return true end
 
   (variable_declaration (variable_declarator (identifier) (property_identifier) (property_identifier) (property_identifier) (property_identifier)) (function_definition (parameters)
     (return_statement (true)))))
+
+
+============================================
+Local function statements
+============================================
+
+local function f()
+  return 'some string'
+end
+
+local f
+f = function() return 'some string' end
+
+---
+
+(lua
+  (local_function (identifier) (parameters)
+    (return_statement (string)))
+
+  (local_variable_declaration (variable_declarator (identifier)))
+  (variable_declaration (variable_declarator (identifier))
+    (function_definition (parameters)
+      (return_statement (string)))))

grammar.js

@@ -67,7 +67,8 @@ module.exports = grammar({
       $.label_statement,
       $._empty_statement,
 
-      alias($.function_statement, $.function)
+      alias($.function_statement, $.function),
+      alias($.local_function_statement, $.local_function)
     ),
 
     // Declarations
@@ -199,6 +200,13 @@ module.exports = grammar({
       $._function_body
     ),
 
+    local_function_statement: $ => seq(
+      'local',
+      'function',
+      $.identifier,
+      $._function_body
+    ),
+
     function_name: $ => seq(
       $.identifier,
       repeat(seq('.', alias($.identifier, $.property_identifier))),

src/grammar.json

@@ -130,6 +130,15 @@
           },
           "named": true,
           "value": "function"
+        },
+        {
+          "type": "ALIAS",
+          "content": {
+            "type": "SYMBOL",
+            "name": "local_function_statement"
+          },
+          "named": true,
+          "value": "local_function"
         }
       ]
     },
@@ -840,6 +849,27 @@
         }
       ]
     },
+    "local_function_statement": {
+      "type": "SEQ",
+      "members": [
+        {
+          "type": "STRING",
+          "value": "local"
+        },
+        {
+          "type": "STRING",
+          "value": "function"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "identifier"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "_function_body"
+        }
+      ]
+    },
     "function_name": {
       "type": "SEQ",
       "members": [