refactor: refactored ui code

This commit is contained in:
Folke Lemaitre 2022-12-23 10:18:19 +01:00
parent cd023dc709
commit fde5feea6d
No known key found for this signature in database
GPG key ID: 41F8B1FBACAE2040
4 changed files with 238 additions and 184 deletions

View file

@ -2,6 +2,16 @@ local Util = require("lazy.util")
local Render = require("lazy.view.render")
local Config = require("lazy.core.config")
---@class LazyViewState
---@field mode string
---@field plugin? string
---@class LazyView
---@field buf number
---@field win number
---@field render LazyRender
---@field state LazyViewState
---@field win_opts LazyViewWinOpts
local M = {}
M.modes = {
@ -40,29 +50,159 @@ M.modes = {
M.hover = "K"
---@type string?
M.mode = nil
---@type LazyView
M.view = nil
---@param mode? string
function M.show(mode)
if Config.headless then
return
end
M.mode = mode or M.mode or "home"
M.view = M.view or M.create({ mode = mode })
M.view:update(mode)
end
---@param opts? {mode?:string}
function M.create(opts)
require("lazy.view.colors").setup()
opts = opts or {}
local self = setmetatable({}, { __index = M })
if M._buf and vim.api.nvim_buf_is_valid(M._buf) then
-- vim.api.nvim_win_set_cursor(M._win, { 1, 0 })
vim.cmd([[do User LazyRender]])
return
self.state = { mode = "home" }
self:mount()
self.render = Render.new(self)
self.update = Util.throttle(Config.options.ui.throttle, self.update)
self:on_key("q", self.close)
self:on({ "BufDelete", "BufLeave", "BufHidden" }, self.close, { once = true })
self:on("User LazyRender", function()
if not (self.buf and vim.api.nvim_buf_is_valid(self.buf)) then
return true
end
self:update()
end)
-- plugin details
self:on_key("<cr>", function()
local plugin = self.render:get_plugin()
if plugin then
self.state.plugin = self.state.plugin ~= plugin.name and plugin.name or nil
self:update()
end
end)
self:setup_hover()
self:setup_modes()
return self
end
---@param events string|string[]
---@param fn fun(self?):boolean?
---@param opts? table
function M:on(events, fn, opts)
if type(events) == "string" then
events = { events }
end
for _, e in ipairs(events) do
local event, pattern = e:match("(%w+) (%w+)")
event = event or e
vim.api.nvim_create_autocmd(
event,
vim.tbl_extend("force", {
pattern = pattern,
buffer = not pattern and self.buf or nil,
callback = function()
return fn(self)
end,
}, opts or {})
)
end
end
local buf = vim.api.nvim_create_buf(false, false)
M._buf = buf
---@param key string
---@param fn fun(self?)
function M:on_key(key, fn)
vim.keymap.set("n", key, function()
fn(self)
end, {
nowait = true,
buffer = self.buf,
})
end
---@param mode? string
function M:update(mode)
if mode then
self.state.mode = mode
end
if self.buf and vim.api.nvim_buf_is_valid(self.buf) then
vim.bo[self.buf].modifiable = true
self.render:update()
vim.bo[self.buf].modifiable = false
vim.cmd.redraw()
end
end
function M:open_url(path)
local plugin = self.render:get_plugin()
if plugin then
if plugin.url then
local url = plugin.url:gsub("%.git$", "")
Util.open(url .. path)
else
Util.error("No url for " .. plugin.name)
end
end
end
function M:close()
local buf = self.buf
local win = self.win
self.win = nil
self.buf = nil
M.view = nil
vim.diagnostic.reset(Config.ns, buf)
vim.schedule(function()
if win and vim.api.nvim_win_is_valid(win) then
vim.api.nvim_win_close(win, true)
end
if buf and vim.api.nvim_buf_is_valid(buf) then
vim.api.nvim_buf_delete(buf, { force = true })
end
end)
end
function M:focus()
vim.api.nvim_set_current_win(self.win)
-- it seems that setting the current win doesn't work before VimEnter,
-- so do that then
if vim.v.vim_did_enter ~= 1 then
vim.api.nvim_create_autocmd("VimEnter", {
once = true,
callback = function()
if self.win and vim.api.nvim_win_is_valid(self.win) then
pcall(vim.api.nvim_set_current_win, self.win)
end
return true
end,
})
end
end
function M:mount()
self.buf = vim.api.nvim_create_buf(false, false)
local function size(max, value)
return value > 1 and math.min(value, max) or math.floor(max * value)
end
local opts = {
---@class LazyViewWinOpts
self.win_opts = {
relative = "editor",
style = "minimal",
border = Config.options.ui.border,
@ -71,178 +211,87 @@ function M.show(mode)
noautocmd = true,
}
opts.row = (vim.o.lines - opts.height) / 2
opts.col = (vim.o.columns - opts.width) / 2
local win = vim.api.nvim_open_win(buf, true, opts)
M._win = win
vim.api.nvim_set_current_win(win)
self.win_opts.row = (vim.o.lines - self.win_opts.height) / 2
self.win_opts.col = (vim.o.columns - self.win_opts.width) / 2
self.win = vim.api.nvim_open_win(self.buf, true, self.win_opts)
self:focus()
-- it seems that setting the current win doesn't work before VimEnter,
-- so do that then
if vim.v.vim_did_enter ~= 1 then
vim.api.nvim_create_autocmd("VimEnter", {
once = true,
callback = function()
if win and vim.api.nvim_win_is_valid(win) then
pcall(vim.api.nvim_set_current_win, win)
end
end,
})
end
vim.bo[self.buf].buftype = "nofile"
vim.bo[self.buf].filetype = "lazy"
vim.bo[self.buf].bufhidden = "wipe"
vim.wo[self.win].conceallevel = 3
vim.wo[self.win].spell = false
vim.wo[self.win].wrap = true
vim.wo[self.win].winhighlight = "Normal:LazyNormal"
end
vim.bo[buf].buftype = "nofile"
vim.bo[buf].filetype = "lazy"
vim.bo[buf].bufhidden = "wipe"
vim.wo[win].conceallevel = 3
vim.wo[win].spell = false
vim.wo[win].wrap = true
vim.wo[win].winhighlight = "Normal:LazyNormal"
local function close()
M._buf = nil
vim.diagnostic.reset(Config.ns, buf)
vim.schedule(function()
if vim.api.nvim_win_is_valid(win) then
vim.api.nvim_win_close(win, true)
end
if vim.api.nvim_buf_is_valid(buf) then
vim.api.nvim_buf_delete(buf, { force = true })
end
end)
end
vim.keymap.set("n", "q", close, {
nowait = true,
buffer = buf,
})
vim.api.nvim_create_autocmd({ "BufDelete", "BufLeave", "BufHidden" }, {
once = true,
buffer = buf,
callback = close,
})
local render = Render.new(buf, win, 2, opts.width)
local update = Util.throttle(Config.options.ui.throttle, function()
if buf and vim.api.nvim_buf_is_valid(buf) then
vim.bo[buf].modifiable = true
render:update()
vim.bo[buf].modifiable = false
vim.cmd.redraw()
end
end)
local function get_plugin()
local pos = vim.api.nvim_win_get_cursor(win)
return render:get_plugin(pos[1])
end
vim.keymap.set("n", "<cr>", function()
local plugin = get_plugin()
if plugin then
if render._details == plugin.name then
render._details = nil
else
render._details = plugin.name
end
update()
end
end, {
nowait = true,
buffer = buf,
})
local function open(path)
local plugin = get_plugin()
if plugin then
local url = plugin.url:gsub("%.git$", "")
if Util.file_exists(url) then
url = "https://github.com/" .. plugin[1]
end
Util.open(url .. path)
end
end
M.keys(buf, {
function M:setup_hover()
local handlers = {
["%s(" .. string.rep("[a-z0-9]", 7) .. ")%s"] = function(hash)
open("/commit/" .. hash)
self:open_url("/commit/" .. hash)
end,
["%s(" .. string.rep("[a-z0-9]", 7) .. ")$"] = function(hash)
open("/commit/" .. hash)
self:open_url("/commit/" .. hash)
end,
["^(" .. string.rep("[a-z0-9]", 7) .. ")%s"] = function(hash)
open("/commit/" .. hash)
self:open_url("/commit/" .. hash)
end,
["#(%d+)"] = function(issue)
open("/issues/" .. issue)
self:open_url("/issues/" .. issue)
end,
["README.md"] = function()
local plugin = get_plugin()
Util.open(plugin.dir .. "/README.md")
local plugin = self.render:get_plugin()
if plugin then
Util.open(plugin.dir .. "/README.md")
end
end,
["|(%S-)|"] = vim.cmd.help, -- vim help links
["(https?://%S+)"] = function(url)
Util.open(url)
end,
})
}
self:on_key(M.hover, function()
local line = vim.api.nvim_get_current_line()
local pos = vim.api.nvim_win_get_cursor(0)
local col = pos[2] + 1
for pattern, handler in pairs(handlers) do
local from = 1
local to, url
while from do
from, to, url = line:find(pattern, from)
if from and col >= from and col <= to then
return handler(url)
end
if from then
from = to + 1
end
end
end
end)
end
function M:setup_modes()
for _, m in ipairs(M.modes) do
if m.key then
vim.keymap.set("n", m.key, function()
self:on_key(m.key, function()
local Commands = require("lazy.view.commands")
if m.plugin then
local plugin = get_plugin()
local plugin = self.render:get_plugin()
if plugin then
Commands.cmd(m.name, { plugins = { plugin } })
end
else
if M.mode == m.name and m.toggle then
M.mode = nil
return update()
if self.state.mode == m.name and m.toggle then
self.state.mode = "home"
return self:update()
end
Commands.cmd(m.name)
end
end, { buffer = buf })
end)
end
end
vim.api.nvim_create_autocmd("User", {
pattern = "LazyRender",
callback = function()
if not vim.api.nvim_buf_is_valid(buf) then
return true
end
update()
end,
})
update()
end
---@param handlers table<string, fun(str:string)>
function M.keys(buf, handlers)
local function map(lhs)
vim.keymap.set("n", lhs, function()
local line = vim.api.nvim_get_current_line()
local pos = vim.api.nvim_win_get_cursor(0)
local col = pos[2] + 1
for pattern, handler in pairs(handlers) do
local from = 1
local to, url
while from do
from, to, url = line:find(pattern, from)
if from and col >= from and col <= to then
return handler(url)
end
if from then
from = to + 1
end
end
end
end, { buffer = buf, silent = true })
end
map(M.hover)
end
return M