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

(js) Add support for functions in statements

Browse Source
This commit is contained in:
Daniel Mathiot
2021-08-24 12:44:33 +02:00
parent 3a40a1d111
commit 14424a0887
1 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
lua/neogen/configurations/javascript.lua
View File

@@ -1,18 +1,37 @@
local function_tree = {
{ retrieve = "first", node_type = "formal_parameters", subtree = {
{ retrieve = "all", node_type = "identifier", extract = true }
} },
{ retrieve = "first", node_type = "statement_block", subtree = {
{ retrieve = "first", node_type = "return_statement", extract = true }
} }
}
return {
parent = { "function_declaration" },
parent = { "function_declaration", "expression_statement" },
data = {
["function_declaration"] = {
["0"] = {
extract = function (node)
local results = {}
local tree = function_tree
local nodes = neogen.utilities.nodes:matching_nodes_from(node, tree)
local res = neogen.utilities.extractors:extract_from_matched(nodes)
results.parameters = res.identifier
results.return_statement = res.return_statement
return results
end
}
},
["expression_statement"] = {
["0"] = {
extract = function (node)
local results = {}
local tree = {
{ retrieve = "first", node_type = "formal_parameters", subtree = {
{ retrieve = "all", node_type = "identifier", extract = true }
} },
{ retrieve = "first", node_type = "statement_block", subtree = {
{ retrieve = "first", node_type = "return_statement", extract = true }
{ retrieve = "all", node_type = "assignment_expression", subtree = {
{ retrieve = "all", node_type = "function", subtree = function_tree }
} }
}
local nodes = neogen.utilities.nodes:matching_nodes_from(node, tree)