mirror of
https://github.com/neovim/neovim.git
synced 2025-10-02 16:08:36 +00:00
109 lines
2.5 KiB
Lua
109 lines
2.5 KiB
Lua
local helpers = require('test.functional.helpers')
|
|
local spawn, set_session, nvim, nvim_prog, nvim_command, nvim_eval =
|
|
helpers.spawn, helpers.set_session, helpers.nvim, helpers.nvim_prog,
|
|
helpers.command, helpers.eval
|
|
local write_file, merge_args = helpers.write_file, helpers.merge_args
|
|
|
|
local msgpack = require('MessagePack')
|
|
|
|
local tmpname = os.tmpname()
|
|
local additional_cmd = ''
|
|
|
|
local function nvim_argv()
|
|
local ret
|
|
local nvim_argv = {nvim_prog, '-u', 'NONE', '-i', tmpname, '-N',
|
|
'--cmd', 'set shortmess+=I background=light noswapfile',
|
|
'--cmd', additional_cmd,
|
|
'--embed'}
|
|
if helpers.prepend_argv then
|
|
return merge_args(helpers.prepend_argv, nvim_argv)
|
|
else
|
|
return nvim_argv
|
|
end
|
|
end
|
|
|
|
local session = nil
|
|
|
|
local reset = function()
|
|
if session then
|
|
session:exit(0)
|
|
end
|
|
session = spawn(nvim_argv())
|
|
set_session(session)
|
|
nvim('set_var', 'tmpname', tmpname)
|
|
end
|
|
|
|
local set_additional_cmd = function(s)
|
|
additional_cmd = s
|
|
end
|
|
|
|
local clear = function()
|
|
os.remove(tmpname)
|
|
set_additional_cmd('')
|
|
end
|
|
|
|
local exc_exec = function(cmd)
|
|
nvim_command(([[
|
|
try
|
|
execute "%s"
|
|
catch
|
|
let g:__exception = v:exception
|
|
endtry
|
|
]]):format(cmd:gsub('\n', '\\n'):gsub('[\\"]', '\\%0')))
|
|
local ret = nvim_eval('get(g:, "__exception", 0)')
|
|
nvim_command('unlet! g:__exception')
|
|
return ret
|
|
end
|
|
|
|
local get_shada_rw = function(fname)
|
|
local wshada = function(text)
|
|
write_file(fname, text, true)
|
|
end
|
|
local sdrcmd = function(bang)
|
|
return 'rshada' .. (bang and '!' or '') .. ' ' .. fname
|
|
end
|
|
local clean = function()
|
|
os.remove(fname)
|
|
local i = ('a'):byte()
|
|
while i <= ('z'):byte() do
|
|
if not os.remove(fname .. ('.tmp.%c'):format(i)) then
|
|
break
|
|
end
|
|
i = i + 1
|
|
end
|
|
end
|
|
return wshada, sdrcmd, fname, clean
|
|
end
|
|
|
|
local mpack_keys = {'type', 'timestamp', 'length', 'value'}
|
|
|
|
local read_shada_file = function(fname)
|
|
local fd = io.open(fname, 'r')
|
|
local mstring = fd:read('*a')
|
|
fd:close()
|
|
local unpacker = msgpack.unpacker(mstring)
|
|
local ret = {}
|
|
local cur
|
|
local i = 0
|
|
while true do
|
|
local off, val = unpacker()
|
|
if not off then break end
|
|
if i % 4 == 0 then
|
|
cur = {}
|
|
ret[#ret + 1] = cur
|
|
end
|
|
cur[mpack_keys[(i % 4) + 1]] = val
|
|
i = i + 1
|
|
end
|
|
return ret
|
|
end
|
|
|
|
return {
|
|
reset=reset,
|
|
set_additional_cmd=set_additional_cmd,
|
|
clear=clear,
|
|
exc_exec=exc_exec,
|
|
get_shada_rw=get_shada_rw,
|
|
read_shada_file=read_shada_file,
|
|
}
|