c4f3725af7
This is a first iteration of refactoring the codebase, and generating annotations along the way. Added "variable_declaration" as parent node for lua
63 lines
1.9 KiB
Lua
63 lines
1.9 KiB
Lua
local ok, ts_utils = pcall(require, "nvim-treesitter.ts_utils")
|
|
assert(ok, "neogen requires nvim-treesitter to operate :(")
|
|
|
|
neogen = { }
|
|
|
|
-- TODO: Move code here
|
|
neogen.generate = function(searcher, generator) end
|
|
|
|
neogen.auto_generate = function(custom_template)
|
|
vim.treesitter.get_parser(0):for_each_tree(function(tree, language_tree)
|
|
local searcher = neogen.configuration.languages[language_tree:lang()]
|
|
|
|
if searcher then
|
|
searcher.locator = searcher.locator or neogen.default_locator
|
|
searcher.granulator = searcher.granulator or neogen.default_granulator
|
|
searcher.generator = searcher.generator or neogen.default_generator
|
|
|
|
local located_parent_node = searcher.locator({
|
|
root = tree:root(),
|
|
current = ts_utils.get_node_at_cursor(0),
|
|
}, searcher.parent)
|
|
|
|
if not located_parent_node then
|
|
return
|
|
end
|
|
|
|
local data = searcher.granulator(located_parent_node, searcher.data)
|
|
|
|
if data and not vim.tbl_isempty(data) then
|
|
local to_place, content = searcher.generator(
|
|
located_parent_node,
|
|
data,
|
|
custom_template or searcher.template
|
|
)
|
|
|
|
vim.fn.append(to_place, content)
|
|
end
|
|
end
|
|
end)
|
|
end
|
|
|
|
function neogen.generate_command()
|
|
vim.api.nvim_command('command! -range -bar Neogen lua require("neogen").auto_generate()')
|
|
end
|
|
|
|
neogen.setup = function(opts)
|
|
neogen.configuration = vim.tbl_deep_extend("keep", opts or {}, {
|
|
-- DEFAULT CONFIGURATION
|
|
languages = {
|
|
lua = require('neogen.configurations.lua'),
|
|
},
|
|
})
|
|
|
|
neogen.generate_command()
|
|
end
|
|
|
|
require('neogen.utility')
|
|
require('neogen.locators.default')
|
|
require('neogen.granulators.default')
|
|
require('neogen.generators.default')
|
|
|
|
return neogen
|