leigh/neogen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(c) Add support for functions that return pointers

Browse Source
This commit is contained in:
Daniel Mathiot
2021-08-28 15:28:14 +02:00
parent c9ea7ee461
commit 460d022cc1
1 changed files with 27 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lua/neogen/configurations/c.lua
View File

@@ -1,24 +1,30 @@
local c_params = {
retrieve = "first",
node_type = "parameter_list",
subtree = {
{
retrieve = "all",
node_type = "parameter_declaration",
subtree = {
{ retrieve = "first_recursive", node_type = "identifier", extract = true}
},
},
},
}
local c_function_extractor = function(node)
local tree = {
{
retrieve = "first",
node_type = "function_declarator",
subtree = {
{
retrieve = "first",
node_type = "parameter_list",
subtree = {
{
retrieve = "all",
node_type = "parameter_declaration",
subtree = {
{ retrieve = "first_recursive", node_type = "identifier", extract = true}
},
},
},
},
c_params,
},
},
{
retrieve = "first_recursive",
node_type = "function_declarator",
extract = true
},
{
retrieve = "first",
node_type = "compound_statement",
@@ -26,11 +32,19 @@ local c_function_extractor = function(node)
{ retrieve = "first", node_type = "return_statement", extract = true },
},
},
c_params
}
local nodes = neogen.utilities.nodes:matching_nodes_from(node, tree)
local res = neogen.utilities.extractors:extract_from_matched(nodes)
if nodes.function_declarator then
local subnodes = neogen.utilities.nodes:matching_nodes_from(nodes.function_declarator[1], tree)
local subres = neogen.utilities.extractors:extract_from_matched(subnodes)
res = vim.tbl_deep_extend("keep", res, subres)
end
return {
parameters = res.identifier,
return_statement = res.return_statement,