Add query for return statement in an if block

2021-08-17 21:46:09 +02:00
parent 4f639a9d3a
commit dc3c7eb4ea
1 changed files with 1 additions and 0 deletions
--- a/lua/neogen.lua
+++ b/lua/neogen.lua
@@ -8,6 +8,7 @@ M.generate = function ()
    (function (parameters) @params)
    (function (return_statement) @return)
    (function_definition (parameters) @params)
+    (function_definition (if_statement (return_statement) @return))
    ]]

    -- Try to find the upper function