fix: make interface edge case its own capture group

Jay-Madden · Jay-Madden · commit 9ba7dedbb705 · 2024-08-20T13:42:50.000-05:00
diff --git a/lua/auto-fix-return/lib.lua b/lua/auto-fix-return/lib.lua
@@ -56,7 +56,7 @@ M.wrap_golang_return = function()
          ;;   (ERROR)))) ; [16, 11] - [16, 12]
          ;; We need to handle this case specifically so we anchor it to the ancestor node for interface
          (
-           (ERROR)? @error_end (#has-parent? @error_end interface_type)
+           (ERROR)? @error_interface_end (#has-parent? @error_interface_end interface_type)
          )
       )  
       ;; This is a weird edgecase in regards to handling multi returns, an in progress multireturn on a top level function is
@@ -103,7 +103,7 @@ M.wrap_golang_return = function()
     elseif capture_name == "result" and final_end_row ~= 0 then
       final_end_col = end_col
       final_end_row = end_row
-    elseif capture_name == "error_end" then
+    elseif capture_name == "error_end" or capture_name == "error_interface_end" then
       final_end_col = end_col
       final_end_row = end_row
     end

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ M.wrap_golang_return = function()`
`56`	`56`	`;; (ERROR)))) ; [16, 11] - [16, 12]`
`57`	`57`	`;; We need to handle this case specifically so we anchor it to the ancestor node for interface`
`58`	`58`	`(`
`59`		`- (ERROR)? @error_end (#has-parent? @error_end interface_type)`
	`59`	`+ (ERROR)? @error_interface_end (#has-parent? @error_interface_end interface_type)`
`60`	`60`	`)`
`61`	`61`	`)`
`62`	`62`	`;; This is a weird edgecase in regards to handling multi returns, an in progress multireturn on a top level function is`
`@@ -103,7 +103,7 @@ M.wrap_golang_return = function()`
`103`	`103`	`elseif capture_name == "result" and final_end_row ~= 0 then`
`104`	`104`	`final_end_col = end_col`
`105`	`105`	`final_end_row = end_row`
`106`		`- elseif capture_name == "error_end" then`
	`106`	`+ elseif capture_name == "error_end" or capture_name == "error_interface_end" then`
`107`	`107`	`final_end_col = end_col`
`108`	`108`	`final_end_row = end_row`
`109`	`109`	`end`