Anexon
0cbcb93c0a
Add a buffer that invokes a neural completion for the entire buffer contents. * Tech: Split neural prompt and run and add line options * Config: Add initialisation settings for neural buffer * Config: Add completion trigger plug mapping for neural buffer * Fix: prompt ui escape keys * Test: Add filetype in test vimrc for neural buffer tests
58 lines
1.5 KiB
Lua
58 lines
1.5 KiB
Lua
-- External dependencies
|
|
local UI = {}
|
|
local AnimatedSign = {}
|
|
local has_nui, _ = pcall(require, 'nui.input')
|
|
local has_significant, _ = pcall(require, 'significant')
|
|
|
|
if has_nui then
|
|
UI = require('neural.ui')
|
|
end
|
|
|
|
if has_significant then
|
|
AnimatedSign = require('significant')
|
|
end
|
|
|
|
local Neural = {}
|
|
|
|
function Neural.setup(settings)
|
|
-- Call the Vim script function to set up the shared configuration.
|
|
vim.fn['neural#config#Set'](settings)
|
|
end
|
|
|
|
function Neural.prompt()
|
|
local prompt_enabled = vim.g.neural.ui.prompt_enabled
|
|
|
|
if has_nui and (prompt_enabled and prompt_enabled ~= 0) then
|
|
UI.prompt(
|
|
' Neural ',
|
|
function(value)
|
|
if value == nil or value == '' then
|
|
vim.fn['neural#ComplainNoPromptText']()
|
|
else
|
|
vim.fn['neural#Prompt'](value)
|
|
end
|
|
end
|
|
)
|
|
else
|
|
vim.fn.feedkeys(':Neural ')
|
|
end
|
|
end
|
|
|
|
function Neural.start_animated_sign(line)
|
|
local sign_enabled = vim.g.neural and vim.g.neural.ui.animated_sign_enabled
|
|
|
|
if has_significant and (sign_enabled and sign_enabled ~= 0) and line > 0 then
|
|
AnimatedSign.start_animated_sign(line, 'dots', 100)
|
|
end
|
|
end
|
|
|
|
function Neural.stop_animated_sign(line)
|
|
local sign_enabled = vim.g.neural and vim.g.neural.ui.animated_sign_enabled
|
|
|
|
if has_significant and (sign_enabled and sign_enabled ~= 0) and line > 0 then
|
|
AnimatedSign.stop_animated_sign(line, {unplace_sign=true})
|
|
end
|
|
end
|
|
|
|
return Neural
|