54 lines
2.0 KiB
Lua
54 lines
2.0 KiB
Lua
-- Disable supermaven in JavaScript/Python buffers whose comments contain a
|
|
-- LeetCode problem URL (solution files), and re-enable it when leaving them.
|
|
--
|
|
-- Note: supermaven's built-in `condition` option is one-way — its BufEnter
|
|
-- listener gates the restart branch on `vim.g.SUPERMAVEN_DISABLED`, which
|
|
-- `api.stop()` itself sets, so once auto-stopped it never comes back. This
|
|
-- autocmd owns both directions instead. `vim.g.supermaven_leetcode_stopped`
|
|
-- distinguishes our auto-stop from a manual :SupermavenStop, which stays off.
|
|
local function is_leetcode_buffer()
|
|
local buf = vim.api.nvim_get_current_buf()
|
|
local ft = vim.bo[buf].filetype
|
|
local name = vim.api.nvim_buf_get_name(buf)
|
|
local comment_pat
|
|
if ft == "javascript" or ft == "javascriptreact" or name:match "%.[mc]?jsx?$" then
|
|
comment_pat = "^%s*[/*]"
|
|
elseif ft == "python" or name:match "%.py$" then
|
|
comment_pat = "^%s*[#\"']"
|
|
else
|
|
return false
|
|
end
|
|
-- The URL lives in the header comment block; scanning the top of the file is enough.
|
|
for _, line in ipairs(vim.api.nvim_buf_get_lines(buf, 0, 200, false)) do
|
|
if line:find "leetcode%.com/problems/" and line:match(comment_pat) then return true end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function sync()
|
|
local api = require "supermaven-nvim.api"
|
|
if is_leetcode_buffer() then
|
|
if api.is_running() then
|
|
api.stop()
|
|
vim.g.supermaven_leetcode_stopped = true
|
|
end
|
|
elseif vim.g.supermaven_leetcode_stopped and not api.is_running() then
|
|
vim.g.supermaven_leetcode_stopped = nil
|
|
api.start()
|
|
end
|
|
end
|
|
|
|
return {
|
|
"supermaven-inc/supermaven-nvim",
|
|
optional = true,
|
|
config = function(_, opts)
|
|
require("supermaven-nvim").setup(opts)
|
|
vim.api.nvim_create_autocmd("BufEnter", {
|
|
group = vim.api.nvim_create_augroup("supermaven_leetcode", { clear = true }),
|
|
desc = "Toggle supermaven off in LeetCode solution buffers",
|
|
callback = sync,
|
|
})
|
|
sync() -- plugin loads on VeryLazy; cover the buffer we're already sitting in
|
|
end,
|
|
}
|