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

(typescript) Add support for typescript

Browse Source
This commit is contained in:
Daniel Mathiot
2021-09-29 15:01:45 +02:00
parent 19d5cd8c01
commit 91c10304d4
2 changed files with 97 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lua/neogen.lua
View File

@@ -127,6 +127,7 @@ neogen.setup = function(opts)
lua = require("neogen.configurations.lua"),
python = require("neogen.configurations.python"),
javascript = require("neogen.configurations.javascript"),
typescript = require("neogen.configurations.typescript"),
c = require("neogen.configurations.c"),
go = require("neogen.configurations.go"),
},

96
lua/neogen/configurations/typescript.lua Normal file
View File

@@ -0,0 +1,96 @@
local function_tree = {
{
retrieve = "first",
node_type = "formal_parameters",
subtree = {
{
retrieve = "all",
node_type = "required_parameter",
subtree = {
{ retrieve = "all", node_type = "identifier", extract = true },
}
},
{
retrieve = "all",
node_type = "optional_parameter",
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 = {
func = { "function_declaration", "expression_statement", "variable_declaration" },
class = { "function_declaration", "expression_statement", "variable_declaration", "class_declaration" },
},
data = {
func = {
["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|variable_declaration"] = {
["1"] = {
extract = function(node)
local results = {}
local tree = { { retrieve = "all", node_type = "function", subtree = 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,
},
},
},
class = {
["function_declaration|class_declaration|expression_statement|variable_declaration"] = {
["0"] = {
extract = function(_)
local results = {}
results.class_tag = { "" }
return results
end,
},
},
},
},
template = {
annotation_convention = "jsdoc",
use_default_comment = false,
jsdoc = {
{ nil, "/* $1 */", { no_results = true } },
{ nil, "/**" },
{ "class_tag", " * @classdesc $1", { before_first_item = { " * ", " * @class" }, type = { "class" } } },
{ "parameters", " * @param {any} %s $1" },
{ "return_statement", " * @returns {$1|any}" },
{ nil, " */" },
},
},
}