From 91c10304d4b299494890ab03e4037ef809ab765e Mon Sep 17 00:00:00 2001
From: Daniel Mathiot <d.danymat@gmail.com>
Date: Wed, 29 Sep 2021 15:01:45 +0200
Subject: [PATCH] (typescript) Add support for typescript

---
 lua/neogen.lua                           |  1 +
 lua/neogen/configurations/typescript.lua | 96 ++++++++++++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 lua/neogen/configurations/typescript.lua

diff --git a/lua/neogen.lua b/lua/neogen.lua
index 17ff96b..971c3fe 100644
--- a/lua/neogen.lua
+++ b/lua/neogen.lua
@@ -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"),
         },
diff --git a/lua/neogen/configurations/typescript.lua b/lua/neogen/configurations/typescript.lua
new file mode 100644
index 0000000..c371bb1
--- /dev/null
+++ b/lua/neogen/configurations/typescript.lua
@@ -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, " */" },
+        },
+    },
+}