typstar/lua/typstar/autosnippets.lua

local M = {}
local cfg = require('typstar.config').config.snippets
local utils = require('typstar.utils')
local luasnip = require('luasnip')
local fmta = require('luasnip.extras.fmt').fmta
local lsengines = require('luasnip.nodes.util.trig_engines')
local ts = vim.treesitter

local last_keystroke_time = nil
vim.api.nvim_create_autocmd('TextChangedI', {
    callback = function()
        last_keystroke_time = vim.loop.now()
    end,
})
local lexical_result_cache = {}
local ts_markup_query = ts.query.parse('typst', '(text) @markup')
local ts_math_query = ts.query.parse('typst', '(math) @math')
local ts_string_query = ts.query.parse('typst', '(string) @string')

M.in_math = function()
    local cursor = utils.get_cursor_pos()
    return utils.cursor_within_treesitter_query(ts_math_query, 0, cursor)
        and not utils.cursor_within_treesitter_query(ts_string_query, 0, cursor)
end
M.in_markup = function() return utils.cursor_within_treesitter_query(ts_markup_query, 2) end
M.not_in_math = function() return not M.in_math() end
M.not_in_markup = function() return not M.in_markup() end
M.snippets_toggle = true

function M.cap(i)
    return luasnip.function_node(function(_, snip) return snip.captures[i] end)
end

function M.get_visual(args, parent)
    if (#parent.snippet.env.LS_SELECT_RAW > 0) then
        return luasnip.snippet_node(nil, luasnip.insert_node(1, parent.snippet.env.LS_SELECT_RAW))
    else -- If LS_SELECT_RAW is empty, return a blank insert node
        return luasnip.snippet_node(nil, luasnip.insert_node(1))
    end
end

function M.ri(insert_node_id)
    return luasnip.function_node(function(args) return args[1][1] end, insert_node_id)
end

function M.snip(trigger, expand, insert, condition, priority, wordTrig)
    priority = priority or 1000
    return luasnip.snippet(
        {
            trig = trigger,
            trigEngine = M.engine,
            trigEngineOpts = { condition = condition },
            wordTrig = wordTrig,
            priority = priority,
            snippetType = 'autosnippet'
        },
        fmta(expand, { unpack(insert) }),
        {
            condition = function() return M.snippets_toggle end
        }
    )
end

function M.start_snip(trigger, expand, insert, condition, priority)
    return M.snip('^\\s*' .. trigger, expand, insert, condition, priority)
end

function M.engine(trigger, opts)
    local base_engine = lsengines.ecma(trigger, opts)
    local condition = function()
        local cached = lexical_result_cache[opts.condition]
        if cached ~= nil and cached[1] == last_keystroke_time then
            return cached[2]
        end
        local result = opts.condition()
        lexical_result_cache[opts.condition] = { last_keystroke_time, result }
        return result
    end
    return function(line, trig)
        if not M.snippets_toggle or not condition() then
            return nil
        end
        return base_engine(line, trig)
    end
end

function M.toggle_autosnippets()
    M.snippets_toggle = not M.snippets_toggle
    print(string.format('%sabled typstar autosnippets', M.snippets_toggle and 'En' or 'Dis'))
end

function M.setup()
    if cfg.enable then
        local autosnippets = {}
        for _, file in ipairs(cfg.modules) do
            vim.list_extend(
                autosnippets,
                require(('typstar.snippets.%s'):format(file))
            )
        end
        luasnip.add_snippets('typst', autosnippets)
    end
end

return M