133 lines
3.3 KiB
Lua
133 lines
3.3 KiB
Lua
local M = {}
|
|
|
|
local MATH_NODES = {
|
|
displayed_equation = true,
|
|
inline_formula = true,
|
|
math_environment = true,
|
|
}
|
|
|
|
M.in_env_md = function(env)
|
|
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
|
|
local node = vim.treesitter.get_node({ bufnr = 0, pos = { row - 1, col } })
|
|
local bufnr = vim.api.nvim_get_current_buf()
|
|
while node do
|
|
if node:type() == "generic_environment" then
|
|
local begin = node:child(0)
|
|
local name = begin:field("name")
|
|
if name[1] and vim.treesitter.get_node_text(name[1], bufnr, nil) == "{" .. env .. "}" then
|
|
return true
|
|
end
|
|
end
|
|
node = node:parent()
|
|
end
|
|
return false
|
|
end
|
|
|
|
M.in_env = function(env)
|
|
local pos = vim.fn["vimtex#env#is_inside"](env)
|
|
return pos[1] ~= 0 or pos[2] ~= 0
|
|
end
|
|
|
|
-- For markdown
|
|
M.in_mathzone_md = function()
|
|
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
|
|
local node = vim.treesitter.get_node({ bufnr = 0, pos = { row - 1, col } })
|
|
while node do
|
|
if MATH_NODES[node:type()] then
|
|
return true
|
|
end
|
|
node = node:parent()
|
|
end
|
|
return false
|
|
end
|
|
M.in_text_md = function()
|
|
return not M.in_mathzone_md()
|
|
end
|
|
|
|
-- For typst
|
|
M.in_mathzone_typ = function()
|
|
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
|
|
local node = vim.treesitter.get_node({ bufnr = 0, pos = { row - 1, col } })
|
|
while node do
|
|
if node:type() == "math" then
|
|
return true
|
|
end
|
|
node = node:parent()
|
|
end
|
|
return false
|
|
end
|
|
|
|
M.in_mathzone_tex = function()
|
|
local row, col = unpack(vim.api.nvim_win_get_cursor(0))
|
|
local node = vim.treesitter.get_node({ bufnr = 0, pos = { row - 1, col } })
|
|
while node do
|
|
if MATH_NODES[node:type()] then
|
|
return true
|
|
end
|
|
node = node:parent()
|
|
end
|
|
|
|
return false
|
|
end
|
|
|
|
M.in_mathzone = function()
|
|
local ft = vim.bo.filetype
|
|
if ft == "tex" then
|
|
return M.in_mathzone_tex()
|
|
elseif ft == "markdown" then
|
|
return M.in_mathzone_md()
|
|
elseif ft == "typst" then
|
|
return M.in_mathzone_typ()
|
|
end
|
|
end
|
|
|
|
M.in_text = function()
|
|
return not M.in_mathzone()
|
|
end
|
|
|
|
M.in_item = function()
|
|
return M.in_env("itemize") or M.in_env("enumerate")
|
|
end
|
|
M.in_bib = function()
|
|
return M.in_env("thebibliography")
|
|
end
|
|
M.in_tikz = function()
|
|
return M.in_env("tikzpicture")
|
|
end
|
|
M.in_quantikz = function()
|
|
return M.in_env("quantikz")
|
|
end
|
|
M.in_algo = function()
|
|
return M.in_env("algorithmic")
|
|
end
|
|
|
|
-- M.clean = function()
|
|
-- local current_dir = vim.fn.expand("%:p:h")
|
|
-- local file_types = { "aux", "log", "out", "fls", "fdb_latexmk", "bcf", "run.xml", "toc", "DS_Store", "bak*", "dvi" }
|
|
-- for _, file_type in ipairs(file_types) do
|
|
-- local command = "rm " .. current_dir .. "/*." .. file_type
|
|
-- vim.api.nvim_call_function("system", { command })
|
|
-- end
|
|
-- end
|
|
--
|
|
-- M.format = function()
|
|
-- local current_file = vim.fn.expand("%:p")
|
|
-- local latexindent = "latexindent -g /dev/null " .. current_file .. " -wd -l ~/Documents/Latex/latexindent.yaml"
|
|
-- local build = "pdflatex " .. current_file
|
|
-- vim.api.nvim_call_function("system", { build })
|
|
-- vim.cmd("w")
|
|
-- M.clean()
|
|
-- vim.api.nvim_call_function("system", { latexindent })
|
|
-- vim.cmd("e")
|
|
-- vim.cmd("normal! zz")
|
|
-- -- vim.cmd("TexlabForward")
|
|
-- end
|
|
--
|
|
-- M.sympy_calc = function()
|
|
-- local selected_text = vim.fn.getreg("v")
|
|
-- print(selected_text)
|
|
-- vim.api.nvim_out_write(selected_text)
|
|
-- end
|
|
|
|
return M
|