This commit is contained in:
windwp 2021-04-19 17:04:29 +07:00
parent a2955233d7
commit 3a7fc1c5dd
2 changed files with 269 additions and 256 deletions

View File

@ -0,0 +1,13 @@
if _G.__is_log then
return require('plenary.log').new {
plugin = 'nvim-ts-autotag',
level = (_G.__is_log == true and 'debug') or 'warn',
}
else
return{
debug = function() end,
info = function() end,
error = function() end,
}
end

View File

@ -5,36 +5,36 @@ local parsers = require'nvim-treesitter.parsers'
local M = {}
M.tbl_filetypes = {
'html', 'javascript', 'typescript', 'javascriptreact', 'typescriptreact', 'svelte', 'vue', 'tsx', 'jsx'
'html', 'javascript', 'typescript', 'javascriptreact', 'typescriptreact', 'svelte', 'vue', 'tsx', 'jsx'
}
M.tbl_skipTag = {
'area', 'base', 'br', 'col', 'command', 'embed', 'hr', 'img', 'slot',
'input', 'keygen', 'link', 'meta', 'param', 'source', 'track', 'wbr','menuitem'
'area', 'base', 'br', 'col', 'command', 'embed', 'hr', 'img', 'slot',
'input', 'keygen', 'link', 'meta', 'param', 'source', 'track', 'wbr','menuitem'
}
local HTML_TAG = {
start_tag_pattern = 'start_tag',
start_name_tag_pattern = 'tag_name',
end_tag_pattern = "end_tag",
end_name_tag_pattern = "tag_name",
close_tag_pattern = 'erroneous_end_tag',
close_name_tag_pattern = 'erroneous_end_tag_name',
element_tag = 'element',
skip_tag_pattern = {'quoted_attribute_value', 'end_tag'},
start_tag_pattern = 'start_tag',
start_name_tag_pattern = 'tag_name',
end_tag_pattern = "end_tag",
end_name_tag_pattern = "tag_name",
close_tag_pattern = 'erroneous_end_tag',
close_name_tag_pattern = 'erroneous_end_tag_name',
element_tag = 'element',
skip_tag_pattern = {'quoted_attribute_value', 'end_tag'},
}
local ERROR_TAG = "ERROR"
local JSX_TAG = {
start_tag_pattern = 'jsx_opening_element',
start_name_tag_pattern = 'identifier|nested_identifier',
end_tag_pattern = "jsx_closing_element",
end_name_tag_pattern = "identifier",
close_tag_pattern = 'jsx_closing_element',
close_name_tag_pattern = 'identifier|nested_identifier',
element_tag = 'jsx_element',
skip_tag_pattern = {'jsx_closing_element','jsx_expression', 'string', 'jsx_attribute'},
start_tag_pattern = 'jsx_opening_element',
start_name_tag_pattern = 'identifier|nested_identifier',
end_tag_pattern = "jsx_closing_element",
end_name_tag_pattern = "identifier",
close_tag_pattern = 'jsx_closing_element',
close_name_tag_pattern = 'identifier|nested_identifier',
element_tag = 'jsx_element',
skip_tag_pattern = {'jsx_closing_element','jsx_expression', 'string', 'jsx_attribute'},
}
@ -42,313 +42,313 @@ M.enable_rename = true
M.enable_close = true
M.setup = function (opts)
opts = opts or {}
M.tbl_filetypes = opts.filetypes or M.tbl_filetypes
M.tbl_skipTag = opts.skip_tag or M.tbl_skipTag
M.enable_rename = opts.enable_rename or M.enable_rename
M.enable_close = opts.enable_close or M.enable_close
opts = opts or {}
M.tbl_filetypes = opts.filetypes or M.tbl_filetypes
M.tbl_skipTag = opts.skip_tag or M.tbl_skipTag
M.enable_rename = opts.enable_rename or M.enable_rename
M.enable_close = opts.enable_close or M.enable_close
end
local function is_in_table(tbl, val)
if tbl == nil then return false end
for _, value in pairs(tbl) do
if val== value then return true end
end
return false
if tbl == nil then return false end
for _, value in pairs(tbl) do
if val== value then return true end
end
return false
end
M.is_supported = function (lang)
return is_in_table(M.tbl_filetypes,lang)
return is_in_table(M.tbl_filetypes,lang)
end
local function is_jsx()
return is_in_table({
return is_in_table({
'typescriptreact', 'javascriptreact', 'javascript.jsx', 'typescript.tsx', 'javascript', 'typescript'},
vim.bo.filetype)
vim.bo.filetype)
end
local function get_ts_tag()
local ts_tag = HTML_TAG
if is_jsx() then ts_tag = JSX_TAG end
return ts_tag
local ts_tag = HTML_TAG
if is_jsx() then ts_tag = JSX_TAG end
return ts_tag
end
M.on_file_type = function ()
end
local function find_child_match(opts)
local target = opts.target
local pattern = opts.pattern
local skip_tag_pattern = opts.skip_tag_pattern
assert(target ~= nil, "find child target not nil :" .. pattern)
for node in target:iter_children() do
local node_type = node:type()
if node_type ~= nil and
node_type == pattern and
not is_in_table(skip_tag_pattern, node_type)
then
return node
local target = opts.target
local pattern = opts.pattern
local skip_tag_pattern = opts.skip_tag_pattern
assert(target ~= nil, "find child target not nil :" .. pattern)
for node in target:iter_children() do
local node_type = node:type()
if node_type ~= nil and
node_type == pattern and
not is_in_table(skip_tag_pattern, node_type)
then
return node
end
end
end
end
local function find_parent_match(opts)
local target = opts.target
local max_depth = opts.max_depth or 10
local pattern = opts.pattern
local skip_tag_pattern = opts.skip_tag_pattern
assert(target ~= nil, "find parent target not nil :" .. pattern)
local cur_depth = 0
local cur_node = target
local tbl_pattern = vim.split(pattern, "|")
for _,ptn in pairs(tbl_pattern) do
while cur_node ~= nil do
local node_type = cur_node:type()
if is_in_table(skip_tag_pattern, node_type) then return nil end
if node_type ~= nil and node_type == ptn then
return cur_node
elseif cur_depth < max_depth then
cur_depth = cur_depth + 1
cur_node = cur_node:parent()
else
cur_node = nil
local target = opts.target
local max_depth = opts.max_depth or 10
local pattern = opts.pattern
local skip_tag_pattern = opts.skip_tag_pattern
assert(target ~= nil, "find parent target not nil :" .. pattern)
local cur_depth = 0
local cur_node = target
local tbl_pattern = vim.split(pattern, "|")
for _,ptn in pairs(tbl_pattern) do
while cur_node ~= nil do
local node_type = cur_node:type()
if is_in_table(skip_tag_pattern, node_type) then return nil end
if node_type ~= nil and node_type == ptn then
return cur_node
elseif cur_depth < max_depth then
cur_depth = cur_depth + 1
cur_node = cur_node:parent()
else
cur_node = nil
end
end
end
end
return nil end
end
return nil end
local function get_tag_name(node)
local tag_name = nil
if node ~=nil then
tag_name = ts_utils.get_node_text(node)[1]
end
return tag_name
local tag_name = nil
if node ~=nil then
tag_name = ts_utils.get_node_text(node)[1]
end
return tag_name
end
local function find_tag_node(opt)
local target = opt.target or ts_utils.get_node_at_cursor()
local target = opt.target or ts_utils.get_node_at_cursor()
local tag_pattern = opt.tag_pattern
local name_tag_pattern = opt.name_tag_pattern
local skip_tag_pattern = opt.skip_tag_pattern
local find_child = opt.find_child or false
local node
if find_child then
node = find_child_match({
target = target,
pattern = tag_pattern,
skip_tag_pattern = skip_tag_pattern
})
else
node = find_parent_match({
target = target,
pattern = tag_pattern,
skip_tag_pattern = skip_tag_pattern
})
end
if node == nil then return nil end
local name_node = node
local tbl_name_pattern={}
if string.match(name_tag_pattern,"%|") then
tbl_name_pattern = vim.split(name_tag_pattern, '|')
for _, pattern in pairs(tbl_name_pattern) do
name_node = find_child_match({
target = node,
pattern = pattern
})
if name_node then return name_node end
local tag_pattern = opt.tag_pattern
local name_tag_pattern = opt.name_tag_pattern
local skip_tag_pattern = opt.skip_tag_pattern
local find_child = opt.find_child or false
local node
if find_child then
node = find_child_match({
target = target,
pattern = tag_pattern,
skip_tag_pattern = skip_tag_pattern
})
else
node = find_parent_match({
target = target,
pattern = tag_pattern,
skip_tag_pattern = skip_tag_pattern
})
end
if node == nil then return nil end
local name_node = node
local tbl_name_pattern={}
if string.match(name_tag_pattern,"%|") then
tbl_name_pattern = vim.split(name_tag_pattern, '|')
for _, pattern in pairs(tbl_name_pattern) do
name_node = find_child_match({
target = node,
pattern = pattern
})
if name_node then return name_node end
end
end
end
tbl_name_pattern = vim.split(name_tag_pattern, '>')
for _, pattern in pairs(tbl_name_pattern) do
name_node = find_child_match({
target = name_node,
pattern = pattern
})
end
return name_node
tbl_name_pattern = vim.split(name_tag_pattern, '>')
for _, pattern in pairs(tbl_name_pattern) do
name_node = find_child_match({
target = name_node,
pattern = pattern
})
end
return name_node
end
local function find_close_tag_node(opt)
opt.find_child = true
return find_tag_node(opt)
opt.find_child = true
return find_tag_node(opt)
end
local function check_close_tag()
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
skip_tag_pattern = ts_tag.skip_tag_pattern
})
if tag_node ~=nil then
local tag_name = get_tag_name(tag_node)
if tag_name ~= nil and is_in_table(M.tbl_skipTag, tag_name) then
return false
end
-- case 6,9 check close on exist node
local element_node = find_parent_match({
target = tag_node,
pattern = ts_tag.element_tag,
max_depth = 2
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
skip_tag_pattern = ts_tag.skip_tag_pattern
})
if tag_node ~= nil then
local close_tag_node = find_close_tag_node({
target = element_node,
tag_pattern = ts_tag.end_tag_pattern,
name_tag_pattern = ts_tag.end_name_tag_pattern,
})
if close_tag_node ~= nil then
local start_row = tag_node:range()
local close_start_row = close_tag_node:range()
if start_row == close_start_row and tag_name == get_tag_name(close_tag_node) then
return false
if tag_node ~=nil then
local tag_name = get_tag_name(tag_node)
if tag_name ~= nil and is_in_table(M.tbl_skipTag, tag_name) then
return false
end
end
-- case 6,9 check close on exist node
local element_node = find_parent_match({
target = tag_node,
pattern = ts_tag.element_tag,
max_depth = 2
})
if tag_node ~= nil then
local close_tag_node = find_close_tag_node({
target = element_node,
tag_pattern = ts_tag.end_tag_pattern,
name_tag_pattern = ts_tag.end_name_tag_pattern,
})
if close_tag_node ~= nil then
local start_row = tag_node:range()
local close_start_row = close_tag_node:range()
if start_row == close_start_row and tag_name == get_tag_name(close_tag_node) then
return false
end
end
end
return true, tag_name
end
return true, tag_name
end
return false
return false
end
M.close_tag = function ()
parsers.get_parser():parse()
local result, tag_name = check_close_tag()
if result == true and tag_name ~= nil then
vim.cmd(string.format([[normal! a</%s>]],tag_name))
vim.cmd[[normal! F>]]
end
parsers.get_parser():parse()
local result, tag_name = check_close_tag()
if result == true and tag_name ~= nil then
vim.cmd(string.format([[normal! a</%s>]],tag_name))
vim.cmd[[normal! F>]]
end
end
local function replace_text_node(node, tag_name)
if node == nil then return end
local start_row, start_col, end_row, end_col = node:range()
if start_row == end_row then
local line = vim.fn.getline(start_row + 1)
local newline = line:sub(0, start_col) .. tag_name .. line:sub(end_col + 1, string.len(line))
vim.fn.setline(start_row + 1,{newline})
end
if node == nil then return end
local start_row, start_col, end_row, end_col = node:range()
if start_row == end_row then
local line = vim.fn.getline(start_row + 1)
local newline = line:sub(0, start_col) .. tag_name .. line:sub(end_col + 1, string.len(line))
vim.fn.setline(start_row + 1,{newline})
end
end
local function check_tag_correct(node)
if node == nil then return false end
local texts = ts_utils.get_node_text(node)
if string.match(texts[1],"^%<") and string.match(texts[#texts],"%>$") then
return true
end
return false
if node == nil then return false end
local texts = ts_utils.get_node_text(node)
if string.match(texts[1],"^%<") and string.match(texts[#texts],"%>$") then
return true
end
return false
end
local function rename_start_tag()
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
})
if tag_node == nil then return end
if not check_tag_correct(tag_node:parent()) then return end
local tag_name = get_tag_name(tag_node)
local parent_node = tag_node
tag_node = find_parent_match({
target = parent_node,
pattern = ts_tag.element_tag .. "|" .. ERROR_TAG,
max_depth = 2
})
if tag_node == nil then return end
local close_tag_node = find_close_tag_node({
target = tag_node,
tag_pattern = ts_tag.close_tag_pattern,
name_tag_pattern = ts_tag.close_name_tag_pattern,
})
if close_tag_node ~= nil then
local close_tag_name = get_tag_name(close_tag_node)
if tag_name ~=close_tag_name then
replace_text_node(close_tag_node, tag_name)
end
else
close_tag_node = find_child_match({
target = tag_node,
pattern = ERROR_TAG
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
})
if close_tag_node ~=nil then
local close_tag_name = get_tag_name(close_tag_node)
if close_tag_name=='</>' then
replace_text_node(close_tag_node, "</"..tag_name..">")
end
if tag_node == nil then return end
if not check_tag_correct(tag_node:parent()) then return end
local tag_name = get_tag_name(tag_node)
local parent_node = tag_node
tag_node = find_parent_match({
target = parent_node,
pattern = ts_tag.element_tag .. "|" .. ERROR_TAG,
max_depth = 2
})
if tag_node == nil then return end
local close_tag_node = find_close_tag_node({
target = tag_node,
tag_pattern = ts_tag.close_tag_pattern,
name_tag_pattern = ts_tag.close_name_tag_pattern,
})
if close_tag_node ~= nil then
local close_tag_name = get_tag_name(close_tag_node)
if tag_name ~=close_tag_name then
replace_text_node(close_tag_node, tag_name)
end
else
close_tag_node = find_child_match({
target = tag_node,
pattern = ERROR_TAG
})
if close_tag_node ~=nil then
local close_tag_name = get_tag_name(close_tag_node)
if close_tag_name=='</>' then
replace_text_node(close_tag_node, "</"..tag_name..">")
end
end
end
end
end
local function rename_end_tag()
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.close_tag_pattern,
name_tag_pattern = ts_tag.close_name_tag_pattern,
})
local ts_tag = get_ts_tag()
local tag_node = find_tag_node({
tag_pattern = ts_tag.close_tag_pattern,
name_tag_pattern = ts_tag.close_name_tag_pattern,
})
if tag_node == nil then return end
if not check_tag_correct(tag_node:parent()) then return end
local tag_name = get_tag_name(tag_node)
tag_node = find_parent_match({
target = tag_node,
pattern = ts_tag.element_tag,
max_depth = 2
})
if tag_node == nil then return end
local start_tag_node = find_close_tag_node({
target = tag_node,
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
})
if not check_tag_correct(start_tag_node:parent()) then return end
if start_tag_node ~= nil then
local start_tag_name = get_tag_name(start_tag_node)
if tag_name ~= start_tag_name then
replace_text_node(start_tag_node, tag_name)
if tag_node == nil then return end
if not check_tag_correct(tag_node:parent()) then return end
local tag_name = get_tag_name(tag_node)
tag_node = find_parent_match({
target = tag_node,
pattern = ts_tag.element_tag,
max_depth = 2
})
if tag_node == nil then return end
local start_tag_node = find_close_tag_node({
target = tag_node,
tag_pattern = ts_tag.start_tag_pattern,
name_tag_pattern = ts_tag.start_name_tag_pattern,
})
if not check_tag_correct(start_tag_node:parent()) then return end
if start_tag_node ~= nil then
local start_tag_name = get_tag_name(start_tag_node)
if tag_name ~= start_tag_name then
replace_text_node(start_tag_node, tag_name)
end
end
end
end
local function validate_rename()
local cursor = vim.api.nvim_win_get_cursor('.')
local line = vim.fn.getline(cursor[1])
local char = line:sub(cursor[2] + 1, cursor[2] + 1)
-- only rename when last character is a word
if string.match(char,'%w') then
return true
end
return false
local cursor = vim.api.nvim_win_get_cursor('.')
local line = vim.fn.getline(cursor[1])
local char = line:sub(cursor[2] + 1, cursor[2] + 1)
-- only rename when last character is a word
if string.match(char,'%w') then
return true
end
return false
end
M.rename_tag = function ()
if validate_rename() then
parsers.get_parser():parse()
rename_start_tag()
rename_end_tag()
end
if validate_rename() then
parsers.get_parser():parse()
rename_start_tag()
rename_end_tag()
end
end
M.attach = function (bufnr,lang)
M.lang = lang
local config = configs.get_module('autotag')
M.setup(config)
if is_in_table(M.tbl_filetypes, vim.bo.filetype) then
if M.enable_close == true then
vim.cmd[[inoremap <silent> <buffer> > ><c-c>:lua require('nvim-ts-autotag.internal').close_tag()<CR>a]]
M.lang = lang
local config = configs.get_module('autotag')
M.setup(config)
if is_in_table(M.tbl_filetypes, vim.bo.filetype) then
if M.enable_close == true then
vim.cmd[[inoremap <silent> <buffer> > ><c-c>:lua require('nvim-ts-autotag.internal').close_tag()<CR>a]]
end
if M.enable_rename == true then
bufnr = bufnr or vim.api.nvim_get_current_buf()
vim.cmd("augroup nvim_ts_xmltag_" .. bufnr)
vim.cmd[[autocmd!]]
vim.cmd[[autocmd InsertLeave <buffer> call v:lua.require('nvim-ts-autotag.internal').rename_tag() ]]
vim.cmd[[augroup end]]
end
end
if M.enable_rename == true then
bufnr = bufnr or vim.api.nvim_get_current_buf()
vim.cmd("augroup nvim_ts_xmltag_" .. bufnr)
vim.cmd[[autocmd!]]
vim.cmd[[autocmd InsertLeave <buffer> call v:lua.require('nvim-ts-autotag.internal').rename_tag() ]]
vim.cmd[[augroup end]]
end
end
end
M.detach = function ( )