Merge pull request #160 from brandonspark/toplevel-methods

Author: aryx

grammar.js

@@ -76,7 +76,12 @@ module.exports = grammar({
   word: $ => $.identifier,
 
   rules: {
-    program: $ => repeat($.statement),
+    program: $ => repeat($._toplevel_statement),
+
+    _toplevel_statement: $ => choice(
+      $.statement,
+      $.method_declaration,
+    ),
 
     // Literals
 

src/grammar.json

@@ -6,9 +6,22 @@
       "type": "REPEAT",
       "content": {
         "type": "SYMBOL",
-        "name": "statement"
+        "name": "_toplevel_statement"
       }
     },
+    "_toplevel_statement": {
+      "type": "CHOICE",
+      "members": [
+        {
+          "type": "SYMBOL",
+          "name": "statement"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "method_declaration"
+        }
+      ]
+    },
     "_literal": {
       "type": "CHOICE",
       "members": [

src/node-types.json

@@ -2941,6 +2941,10 @@
       "multiple": true,
       "required": false,
       "types": [
+        {
+          "type": "method_declaration",
+          "named": true
+        },
         {
           "type": "statement",
           "named": true