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

(lua) Full specs for type annotations

Browse Source
This commit is contained in:
Daniel Mathiot
2021-08-26 11:03:41 +02:00
parent 93797725e6
commit 48f8f4dd65
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lua/neogen/configurations/lua.lua
View File

@@ -21,13 +21,14 @@ local common_function_extractor = function(node)
} }
end end
local get_identifier_from_var = function (node) local extract_from_var = function (node)
local tree = { local tree = {
{ {
retrieve = "first", retrieve = "first",
node_type = "variable_declarator", node_type = "variable_declarator",
subtree = { subtree = {
{ retrieve = "first", node_type = "identifier", extract = true }, { retrieve = "first", node_type = "identifier", extract = true },
{ retrieve = "first", node_type = "field_expression", extract = true },
}, },
}, },
} }
@@ -66,7 +67,7 @@ return {
["local_variable_declaration|variable_declaration"] = { ["local_variable_declaration|variable_declaration"] = {
["0"] = { ["0"] = {
extract = function(node) extract = function(node)
local res = get_identifier_from_var(node) local res = extract_from_var(node)
return { return {
class_name = res.identifier, class_name = res.identifier,
} }
@@ -78,10 +79,16 @@ return {
["local_variable_declaration|variable_declaration"] = { ["local_variable_declaration|variable_declaration"] = {
["0"] = { ["0"] = {
extract = function(node) extract = function(node)
local res = get_identifier_from_var(node) result = {}
return { local res = extract_from_var(node)
type = res.identifier, result.type = {}
} if res.identifier then
vim.list_extend(result.type, res.identifier)
end
if res.field_expression then
vim.list_extend(result.type, res.field_expression)
end
return result
end, end,
}, },
}, },