jonas/typstar
1
0
Fork 0
mirror of https://github.com/Ascyii/typstar.git synced 2026-01-01 05:24:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: apply snippets to math wrapped in brackets

Browse Source
This commit is contained in:
arne314
2024-11-22 23:29:27 +01:00
parent d2d91f7a01
commit b03004d44c
1 changed files with 51 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
lua/typstar/snippets/visual.lua
View File

@@ -1,6 +1,7 @@
local ls = require('luasnip')
local i = ls.insert_node
local d = ls.dynamic_node
local f = ls.function_node
local helper = require('typstar.autosnippets')
local math = helper.in_math
@@ -9,28 +10,60 @@ local cap = helper.cap
local get_visual = helper.get_visual
local snippets = {}
local operations = {
{ 'vi', '1/(', ')' },
{ 'rb', '(', ')' },
{ 'sq', '[', ']' },
{ 'abs', '|', '|' },
{ 'ul', 'underline(', ')' },
{ 'ol', 'overline(', ')' },
{ 'ht', 'hat(', ')' },
{ 'br', 'macron(', ')' },
{ 'dt', 'dot(', ')' },
{ 'ci', 'circle(', ')' },
{ 'td', 'tilde(', ')' },
{ 'nr', 'norm(', ')' },
{ 'vv', 'vec(', ')' },
{ 'rt', 'sqrt(', ')' },
local operations = { -- boolean denotes whether an additional layer of () brackets should be removed
{ 'vi', '1/(', ')', true },
{ 'bb', '(', ')', false },
{ 'sq', '[', ']', true },
{ 'abs', '|', '|', false },
{ 'ul', 'underline(', ')', false },
{ 'ol', 'overline(', ')', false },
{ 'ub', 'underbrace(', ')', false },
{ 'ob', 'overbrace(', ')', false },
{ 'ht', 'hat(', ')', false },
{ 'br', 'macron(', ')', false },
{ 'dt', 'dot(', ')', false },
{ 'ci', 'circle(', ')', false },
{ 'td', 'tilde(', ')', false },
{ 'nr', 'norm(', ')', false },
{ 'vv', 'vec(', ')', false },
{ 'rt', 'sqrt(', ')', false },
}
local wrap_brackets = function(args, snippet, val)
local captured = snippet.captures[1]
local bracket_types = { [')'] = '(', [']'] = '[', ['}'] = '{' }
local closing_bracket = captured:sub(-1, -1)
local opening_bracket = bracket_types[closing_bracket]
if opening_bracket == nil then
return captured
end
local n_brackets = 0
local char
for i = #captured, 1, -1 do
char = captured:sub(i, i)
if char == closing_bracket then
n_brackets = n_brackets + 1
elseif char == opening_bracket then
n_brackets = n_brackets - 1
end
if n_brackets == 0 then
local remove_additional = val[4] and opening_bracket == '('
return captured:sub(1, i - 1) .. val[2]
.. captured:sub(i + (remove_additional and 1 or 0), -(remove_additional and 2 or 1)) .. val[3]
end
end
return captured
end
for _, val in pairs(operations) do
table.insert(snippets, snip(val[1], val[2] .. '<>' .. val[3], { d(1, get_visual) }, math, 1200))
table.insert(snippets, snip(val[1], val[2] .. '<>' .. val[3], { d(1, get_visual), extra_node }, math))
table.insert(snippets, snip('%s' .. val[1], val[2] .. '<>' .. val[3], { i(1, '1'), extra_node }, math))
table.insert(snippets,
snip('(%s)([^%s]*)' .. val[1], '<>' .. val[2] .. '<>' .. val[3], { cap(1), cap(2) }, math, 1100))
table.insert(snippets, snip('%s' .. val[1], val[2] .. '<>' .. val[3], { i(1, '1') }, math))
snip('(.*[%)|%]|%}])' .. val[1], '<>', { f(wrap_brackets, {}, { user_args = { val } }), nil }, math, 1100))
end
return {