Merge pull request #25 from WHForks/add-runnables

Add entry point detection
scalameta · Oct 10, 2024 · e3cc843 · e3cc843
2 parents 1e33d7e + 19ec9c7
commit e3cc843
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/languages/scala/runnables.scm b/languages/scala/runnables.scm
@@ -0,0 +1,41 @@
+(
+    (
+        (function_definition
+            (annotation
+                name: (type_identifier) @run
+            )
+            name: _
+            parameters: _
+            body: _
+        ) @_scala_main_function_end
+        (#eq? @run "main")
+    )
+    (#set! tag scala-main)
+)
+
+(
+    (
+        (object_definition
+            extend: (extends_clause
+                type: (type_identifier) @run
+            )
+        ) @_scala_app_object_end
+        (#eq? @run "App")
+    )
+    (#set! tag scala-main)
+)
+
+(
+    (
+        (object_definition
+            name: _
+            body: (template_body
+                (function_definition
+                    name: (identifier) @run
+                ) @_jvm_main_function_end
+            )
+        )
+        (#eq? @run "main")
+    )
+    (#set! tag scala-main)
+)