lite-xl/data/plugins/treeview.lua

261 lines
6.4 KiB
Lua
Raw Normal View History

2019-12-28 12:16:32 +01:00
local core = require "core"
local common = require "core.common"
local command = require "core.command"
local config = require "core.config"
local keymap = require "core.keymap"
local style = require "core.style"
local View = require "core.view"
config.treeview_size = 200 * SCALE
2019-12-28 12:16:32 +01:00
local function get_depth(filename)
local n = 1
2019-12-28 12:16:32 +01:00
for sep in filename:gmatch("[\\/]") do
n = n + 1
end
return n
end
local TreeView = View:extend()
2019-12-28 12:16:32 +01:00
function TreeView:new()
TreeView.super.new(self)
self.scrollable = true
self.visible = true
self.init_size = true
2019-12-28 12:16:32 +01:00
self.cache = {}
self.last = {}
2019-12-28 12:16:32 +01:00
end
function TreeView:get_cached(item, dirname)
local dir_cache = self.cache[dirname]
if not dir_cache then
dir_cache = {}
self.cache[dirname] = dir_cache
end
local t = dir_cache[item.filename]
2019-12-28 12:16:32 +01:00
if not t then
t = {}
local basename = item.filename:match("[^\\/]+$")
if item.topdir then
2020-12-27 11:32:52 +01:00
t.filename = basename
t.expanded = true
t.depth = 0
t.abs_filename = dirname
else
t.filename = item.filename
t.depth = get_depth(item.filename)
t.abs_filename = dirname .. PATHSEP .. item.filename
2020-12-27 11:32:52 +01:00
end
t.name = basename
2019-12-28 12:16:32 +01:00
t.type = item.type
dir_cache[item.filename] = t
2019-12-28 12:16:32 +01:00
end
return t
end
function TreeView:get_name()
return "Project"
end
function TreeView:get_item_height()
return style.font:get_height() + style.padding.y
end
function TreeView:check_cache()
-- invalidate cache's skip values if project_files has changed
for i = 1, #core.project_directories do
local dir = core.project_directories[i]
local last_files = self.last[dir.name]
if not last_files then
self.last[dir.name] = dir.files
else
if dir.files ~= last_files then
for _, v in pairs(self.cache[dir.name]) do
v.skip = nil
end
self.last[dir.name] = dir.files
end
2019-12-28 12:16:32 +01:00
end
end
end
function TreeView:each_item()
return coroutine.wrap(function()
self:check_cache()
2020-12-30 16:15:33 +01:00
local count_lines = 0
2019-12-28 12:16:32 +01:00
local ox, oy = self:get_content_offset()
local y = oy + style.padding.y
local w = self.size.x
local h = self:get_item_height()
for k = 1, #core.project_directories do
local dir = core.project_directories[k]
local dir_cached = self:get_cached(dir.item, dir.name)
coroutine.yield(dir_cached, ox, y, w, h)
2020-12-30 16:15:33 +01:00
count_lines = count_lines + 1
2019-12-28 12:16:32 +01:00
y = y + h
local i = 1
2020-12-27 11:32:52 +01:00
while i <= #dir.files and dir_cached.expanded do
local item = dir.files[i]
local cached = self:get_cached(item, dir.name)
coroutine.yield(cached, ox, y, w, h)
2020-12-30 16:15:33 +01:00
count_lines = count_lines + 1
y = y + h
i = i + 1
if not cached.expanded then
if cached.skip then
i = cached.skip
else
local depth = cached.depth
while i <= #dir.files do
if get_depth(dir.files[i].filename) <= depth then break end
i = i + 1
end
cached.skip = i
2019-12-28 12:16:32 +01:00
end
end
end -- while files
end -- for directories
2020-12-30 16:15:33 +01:00
self.count_lines = count_lines
2019-12-28 12:16:32 +01:00
end)
end
2020-12-30 16:15:33 +01:00
function TreeView:on_mouse_moved(px, py, ...)
TreeView.super.on_mouse_moved(self, px, py, ...)
if self.dragging_scrollbar then return end
2019-12-28 12:16:32 +01:00
self.hovered_item = nil
for item, x,y,w,h in self:each_item() do
if px > x and py > y and px <= x + w and py <= y + h then
self.hovered_item = item
break
end
end
end
local function create_directory_in(item)
local path = item.abs_filename
local basename = path:match("[^\\/]+$")
core.command_view:enter("Create directory in " .. path, function(text)
local dirname = path .. PATHSEP .. text
local success, err = system.mkdir(dirname)
if not success then
core.error("cannot create directory %q: %s", dirname, err)
end
item.expanded = true
core.reschedule_project_scan()
end)
end
2020-12-30 16:15:33 +01:00
function TreeView:on_mouse_pressed(button, x, y, clicks)
local caught = TreeView.super.on_mouse_pressed(self, button, x, y, clicks)
if caught then
return
end
2019-12-28 12:16:32 +01:00
if not self.hovered_item then
return
elseif self.hovered_item.type == "dir" then
if keymap.modkeys["ctrl"] and button == "left" then
create_directory_in(self.hovered_item)
else
self.hovered_item.expanded = not self.hovered_item.expanded
end
2019-12-28 12:16:32 +01:00
else
core.try(function()
core.root_view:open_doc(core.open_doc(self.hovered_item.abs_filename))
2019-12-28 12:16:32 +01:00
end)
end
end
function TreeView:update()
-- update width
local dest = self.visible and config.treeview_size or 0
if self.init_size then
self.size.x = dest
self.init_size = false
else
self:move_towards(self.size, "x", dest)
end
2019-12-28 12:16:32 +01:00
TreeView.super.update(self)
end
2020-12-30 16:15:33 +01:00
function TreeView:get_scrollable_size()
return self.count_lines and self:get_item_height() * (self.count_lines + 1) or math.huge
end
2019-12-28 12:16:32 +01:00
function TreeView:draw()
self:draw_background(style.background2)
local icon_width = style.icon_font:get_width("D")
local spacing = style.font:get_width(" ") * 2
local doc = core.active_view.doc
local active_filename = doc and system.absolute_path(doc.filename or "")
for item, x,y,w,h in self:each_item() do
local color = style.text
-- highlight active_view doc
if item.abs_filename == active_filename then
color = style.accent
end
-- hovered item background
if item == self.hovered_item then
renderer.draw_rect(x, y, w, h, style.line_highlight)
color = style.accent
end
-- icons
x = x + item.depth * style.padding.x + style.padding.x
2019-12-28 12:16:32 +01:00
if item.type == "dir" then
local icon1 = item.expanded and "-" or "+"
2019-12-28 12:16:32 +01:00
local icon2 = item.expanded and "D" or "d"
common.draw_text(style.icon_font, color, icon1, nil, x, y, 0, h)
x = x + style.padding.x
common.draw_text(style.icon_font, color, icon2, nil, x, y, 0, h)
x = x + icon_width
else
x = x + style.padding.x
common.draw_text(style.icon_font, color, "f", nil, x, y, 0, h)
x = x + icon_width
end
-- text
x = x + spacing
x = common.draw_text(style.font, color, item.name, nil, x, y, 0, h)
end
2020-12-30 16:15:33 +01:00
self:draw_scrollbar()
2019-12-28 12:16:32 +01:00
end
-- init
local view = TreeView()
local node = core.root_view:get_active_node()
node:split("left", view, true)
-- register commands and keymap
command.add(nil, {
["treeview:toggle"] = function()
view.visible = not view.visible
end,
})
keymap.add { ["ctrl+\\"] = "treeview:toggle" }