From 18fc627838cee293125d447017ccb3db2257df0b Mon Sep 17 00:00:00 2001
From: przepompownia <przepompownia@users.noreply.github.com>
Date: Wed, 27 Mar 2024 15:30:53 +0100
Subject: [PATCH] fix(php): start class_declaration context from name

Fix method_declaration context

Fix for function_definition context

Add test for functions

Test for class and method
---
 queries/php/context.scm  |  3 ++
 test/test.php            | 12 +++++++
 test/ts_context_spec.lua | 67 ++++++++++++++++++++++++++++++++++++++++
 3 files changed, 82 insertions(+)

diff --git a/queries/php/context.scm b/queries/php/context.scm
index 49e8dce3..984c0c24 100644
--- a/queries/php/context.scm
+++ b/queries/php/context.scm
@@ -1,9 +1,11 @@
 
 (function_definition
+  name: (_) @context.start
   body: (_ (_) @context.end)
 ) @context
 
 (method_declaration
+  name: (_) @context.start
   body: (_ (_) @context.end)
 ) @context
 
@@ -32,6 +34,7 @@
 ) @context
 
 (class_declaration
+  name: (_) @context.start
   body: (_ (_) @context.end)
 ) @context
 
diff --git a/test/test.php b/test/test.php
index 94b5fd33..eba2669a 100644
--- a/test/test.php
+++ b/test/test.php
@@ -2,6 +2,7 @@
 /*
  * comment
  */
+#[ReturnTypeWillChange]
 function foo($a, $b) {
   //loop, between low & high
   while ($a <= $b) {
@@ -70,11 +71,22 @@ function foo($a, $b) {
   return null;
 }
 
+#[Attribute]
 class Fruit {
 
 
 
 
+    #[ReturnTypeWillChange]
+    public function rot(): void
+    {
+
+
+        return;
+    }
+
+
+
  // comment
 
 
diff --git a/test/ts_context_spec.lua b/test/ts_context_spec.lua
index 93ed0e96..ac6645ca 100644
--- a/test/ts_context_spec.lua
+++ b/test/ts_context_spec.lua
@@ -45,6 +45,7 @@ describe('ts_context', function()
         "markdown_inline",
         "html",
         "javascript",
+        "php",
       },
       sync_install = true,
     }
@@ -296,6 +297,72 @@ describe('ts_context', function()
       ]]}
     end)
 
+    it('php', function()
+      cmd('edit test/test.php')
+      exec_lua [[vim.treesitter.start()]]
+
+      feed'7<C-e>'
+      screen:expect{grid=[[
+        {1:function}{2: }{3:foo}{14:(}{3:$a}{14:,}{2: }{3:$b}{14:)}{2: }{14:{}{2:        }|
+        {2:  }{1:while}{2: }{14:(}{3:$a}{2: }{1:<=}{2: }{3:$b}{14:)}{2: }{14:{}{2:          }|
+            {5:$index} {4:=} {5:$low} {4:+} {5:floor}{15:((}{5:$hi}|
+            {8:// comment}                |
+            {5:$indexValue} {4:=} {5:$a}{15:;}         |
+        ^    {4:if} {15:(}{5:$indexValue} {4:===} {5:$a}{15:)} {15:{} |
+              {8:// comment}              |
+                                      |
+                                      |
+              {5:$position} {4:=} {5:$index}{15:;}     |
+              {4:return} {15:(}{9:int}{15:)} {5:$position}{15:;} |
+            {15:}}                         |
+            {4:if} {15:(}{5:$indexValue} {4:<} {5:$key}{15:)} {15:{} |
+              {8:// comment}              |
+                                      |
+                                      |
+      ]]}
+
+      feed'67<C-e>'
+      screen:expect{grid=[[
+        {1:class}{2: }{7:Fruit}{2: }{14:{}{2:                 }|
+                                      |
+                                      |
+                                      |
+                                      |
+        ^    {15:#[}ReturnTypeWillChange{15:]}   |
+            {9:public} {4:function} {5:rot}{15:():} {9:voi}|
+            {15:{}                         |
+                                      |
+                                      |
+                {4:return}{15:;}               |
+            {15:}}                         |
+                                      |
+                                      |
+                                      |
+                                      |
+
+      ]]}
+
+      feed'5<C-e>'
+      screen:expect{grid=[[
+        {1:class}{2: }{7:Fruit}{2: }{14:{}{2:                 }|
+        {2:    }{7:public}{2: }{1:function}{2: }{3:rot}{14:():}{2: }{7:voi}|
+        {2:    }{14:{}{2:                         }|
+                                      |
+                                      |
+        ^        {4:return}{15:;}               |
+            {15:}}                         |
+                                      |
+                                      |
+                                      |
+         {8:// comment}                   |
+                                      |
+                                      |
+                                      |
+                                      |
+                                      |
+      ]]}
+    end)
+
     it('typescript', function()
       cmd('edit test/test.ts')
       exec_lua [[vim.treesitter.start()]]