126 lines
2.8 KiB
Lua
126 lines
2.8 KiB
Lua
local vim = vim
|
|
local api = vim.api
|
|
local keymap = vim.keymap
|
|
local opt = vim.opt
|
|
local treesitter = require("nvim-treesitter")
|
|
local textobjects = require("nvim-treesitter-textobjects")
|
|
local select = require("nvim-treesitter-textobjects.select")
|
|
local treesj = require("treesj")
|
|
|
|
local is_nix = require("nixCatsUtils").isNixCats
|
|
|
|
if not is_nix then
|
|
treesitter.install({
|
|
"awk",
|
|
"bash",
|
|
"bibtex",
|
|
"c",
|
|
"cmake",
|
|
"comment",
|
|
"cpp",
|
|
"css",
|
|
"csv",
|
|
"diff",
|
|
"dockerfile",
|
|
"dot",
|
|
"gdscript",
|
|
"gdshader",
|
|
"git_config",
|
|
"git_rebase",
|
|
"gitattributes",
|
|
"gitcommit",
|
|
"gitignore",
|
|
"glsl",
|
|
"gnuplot",
|
|
"go",
|
|
"godot_resource",
|
|
"gpg",
|
|
"graphql",
|
|
"haskell",
|
|
"html",
|
|
"java",
|
|
"javascript",
|
|
"jq",
|
|
"jsdoc",
|
|
"json",
|
|
"jsonc",
|
|
"latex",
|
|
"lua",
|
|
"luadoc",
|
|
"make",
|
|
"python",
|
|
"query",
|
|
"r",
|
|
"racket",
|
|
"readline",
|
|
"regex",
|
|
"requirements",
|
|
"scheme",
|
|
"scss",
|
|
"sql",
|
|
"ssh_config",
|
|
"supercollider",
|
|
"svelte",
|
|
"tmux",
|
|
"toml",
|
|
"tsv",
|
|
"tsx",
|
|
"typescript",
|
|
"udev",
|
|
"vim",
|
|
"vimdoc",
|
|
"xml",
|
|
"yaml",
|
|
"zathurarc",
|
|
})
|
|
end
|
|
|
|
-- https://github.com/MeanderingProgrammer/treesitter-modules.nvim#implementing-yourself
|
|
api.nvim_create_autocmd("FileType", {
|
|
group = api.nvim_create_augroup("treesitter.setup", {}),
|
|
callback = function(args)
|
|
local buf = args.buf
|
|
local filetype = args.match
|
|
local language = vim.treesitter.language.get_lang(filetype) or filetype
|
|
if not (language and vim.treesitter.language.add(language)) then
|
|
return
|
|
end
|
|
-- fold
|
|
vim.wo.foldmethod = "expr"
|
|
vim.wo.foldexpr = "v:lua.vim.treesitter.foldexpr()"
|
|
-- highlight
|
|
vim.treesitter.start(buf, language)
|
|
-- indent
|
|
vim.bo[buf].indentexpr = "v:lua.require'nvim-treesitter'.indentexpr()"
|
|
vim.bo[buf].indentexpr = "v:lua.require'nvim-treesitter'.indentexpr()"
|
|
end,
|
|
})
|
|
|
|
opt.foldmethod = "expr"
|
|
opt.foldexpr = "v:lua.vim.treesitter.foldexpr()"
|
|
opt.foldenable = false
|
|
|
|
-- https://github.com/nvim-treesitter/nvim-treesitter-textobjects/blob/main/README.md#text-objects-select
|
|
textobjects.setup({
|
|
select = {
|
|
lookahead = true,
|
|
},
|
|
})
|
|
|
|
local function select_textobject(capture)
|
|
return function()
|
|
select.select_textobject(capture, "textobjects")
|
|
end
|
|
end
|
|
|
|
keymap.set({ "x", "o" }, "if", select_textobject("@function.inner"), { desc = "Inner function" })
|
|
keymap.set({ "x", "o" }, "af", select_textobject("@function.outer"), { desc = "A function" })
|
|
keymap.set({ "x", "o" }, "ip", select_textobject("@parameter.inner"), { desc = "Inner parameter" })
|
|
keymap.set({ "x", "o" }, "ap", select_textobject("@parameter.outer"), { desc = "A parameter" })
|
|
|
|
treesj.setup({
|
|
use_default_keymaps = false,
|
|
})
|
|
|
|
keymap.set("n", ";", treesj.toggle, { desc = "Toggle join/split (TreeSJ)" })
|