live-server.nvim/lua/live-server.lua
2023-01-09 15:58:39 -06:00

92 lines
2 KiB
Lua

local M = {}
local function log(message, level)
vim.notify(
string.format('live-server.nvim: %s', message),
vim.log.levels[level]
)
end
local job_cache = {}
local function find_cached_dir(dir)
local cur = dir
while not job_cache[cur] do
if cur == '/' then
return
end
cur = vim.fn.fnamemodify(cur, ':h')
end
return cur
end
M.config = {
-- let live-server handle the defaults
args = {},
}
M.setup = function(user_config)
M.config = vim.tbl_deep_extend('force', M.config, user_config or {})
if not vim.fn.executable 'live-server' then
log(
'live-server is not executable. Ensure the npm module is properly installed',
vim.log.levels.ERROR
)
return
end
vim.api.nvim_create_user_command('LiveServerStart', M.start, {})
vim.api.nvim_create_user_command('LiveServerStop', M.stop, {})
end
M.start = function()
local dir = vim.fn.expand '%:p:h'
local cached_dir = find_cached_dir(dir)
if cached_dir then
log('live-server already running', 'INFO')
return
end
local cmd = { 'live-server', cached_dir }
vim.list_extend(cmd, M.config.args)
local job_id = vim.fn.jobstart(cmd, {
on_stderr = function(_, data)
if not data or data[1] == '' then
return
end
-- Remove color from error
log(data[1]:match '.-m(.-)\27', 'ERROR')
end,
on_exit = function(_, exit_code)
job_cache[dir] = nil
if exit_code == 0 then
return
end
log(string.format('stopped with code %s', exit_code), 'INFO')
end,
})
log('live-server started', 'INFO')
job_cache[dir] = job_id
end
M.stop = function()
local dir = vim.fn.expand '%:p:h'
local cached_dir = find_cached_dir(dir)
if cached_dir then
vim.fn.jobstop(job_cache[cached_dir])
job_cache[cached_dir] = nil
end
end
return M