(function (parameters) @params)
(function_definition (parameters) @params)
(local_function (parameters) @params)

(function (return_statement) @return)
(function_definition (if_statement (return_statement) @return))
(local_function (return_statement) @return)