neogen/lua/neogen/configurations/lua.lua

local ts_utils = require("nvim-treesitter.ts_utils")

return {
    -- Search for these nodes
    parent = { "function", "local_function", "local_variable_declaration", "field", "variable_declaration" },

    -- Traverse down these nodes and extract the information as necessary
    data = {
        ["function|local_function"] = {
            -- Get second child from the parent node
            ["2"] = {
                -- It has to be of type "parameters"
                match = "parameters",

                extract = function(node)
                    local regular_params = neogen.utility:extract_children("identifier")(node)
                    local varargs = neogen.utility:extract_children("spread")(node)

                    return {
                        parameters = regular_params,
                        vararg = varargs,
                    }
                end,
            },
        },
        ["local_variable_declaration|field|variable_declaration"] = {
            ["2"] = {
                match = "function_definition",

                extract = function(node)
                    local regular_params = neogen.utility:extract_children_from({
                        [1] = "extract",
                    }, "identifier")(node)

                    local varargs = neogen.utility:extract_children_from({
                        [1] = "extract",
                    }, "spread")(node)

                    local return_statement = neogen.utility:extract_children("return_statement")(node)

                    return {
                        parameters = regular_params,
                        vararg = varargs,
                        return_statement = return_statement
                    }
                end,
            },
        },
    },

    -- Custom lua locator that escapes from comments
    locator = require("neogen.locators.lua"),

    -- Use default granulator and generator
    granulator = nil,
    generator = nil,

    template = {
        { nil, "- " },
        { "parameters", "- @param %s any" },
        { "vararg", "- @vararg any" },
        { "return_statement", "- @return any" }
    },
}