mirror of
https://github.com/SpaceVim/SpaceVim.git
synced 2025-02-02 22:30:04 +08:00
99 lines
2.1 KiB
Lua
99 lines
2.1 KiB
Lua
--=============================================================================
|
|
-- sidebar.lua --- sidebar for zettelkasten plugin
|
|
-- Copyright (c) 2019-2024 Wang Shidong & Contributors
|
|
-- Author: Wang Shidong < wsdjeg@outlook.com >
|
|
-- URL: https://spacevim.org
|
|
-- License: GPLv3
|
|
--=============================================================================
|
|
|
|
local browser = require('zettelkasten.browser')
|
|
|
|
local M = {}
|
|
|
|
local folded_keys = {}
|
|
|
|
local function unique_string_table(t)
|
|
local temp = {}
|
|
for _, k in ipairs(t) do
|
|
temp[k] = true
|
|
end
|
|
local rst = {}
|
|
for m, _ in pairs(temp) do
|
|
table.insert(rst, m)
|
|
end
|
|
return rst
|
|
end
|
|
|
|
-- 按照首字母归类
|
|
|
|
local function get_sorted_keys(t)
|
|
local keys = {}
|
|
|
|
for k, _ in pairs(t) do
|
|
table.insert(keys, k)
|
|
end
|
|
return vim.fn.sort(keys)
|
|
end
|
|
|
|
local function sort_tags(tags)
|
|
local atags = {}
|
|
|
|
for _, tag in ipairs(vim.fn.sort(tags)) do
|
|
local k = string.upper(string.sub(tag, 2, 2))
|
|
if atags[k] then
|
|
table.insert(atags[k], tag)
|
|
else
|
|
atags[k] = { tag }
|
|
end
|
|
end
|
|
|
|
local lines = {}
|
|
|
|
-- ▼ functions
|
|
-- ▶ functions
|
|
for _, k in ipairs(get_sorted_keys(atags)) do
|
|
if #atags[k] > 0 then
|
|
if not folded_keys[k] then
|
|
table.insert(lines, '▼ ' .. k)
|
|
for _, t in ipairs(atags[k]) do
|
|
table.insert(lines, ' ' .. t)
|
|
end
|
|
else
|
|
table.insert(lines, '▶ ' .. k)
|
|
end
|
|
end
|
|
end
|
|
return lines
|
|
end
|
|
|
|
local function update_sidebar_context()
|
|
vim.opt_local.modifiable = true
|
|
local lines = {}
|
|
local result = browser.get_tags()
|
|
for _, tag in ipairs(result) do
|
|
table.insert(lines, tag.name)
|
|
end
|
|
vim.api.nvim_buf_set_lines(0, 0, -1, false, sort_tags(unique_string_table(lines)))
|
|
vim.opt_local.buflisted = false
|
|
vim.opt_local.modifiable = false
|
|
end
|
|
|
|
function M.open_tag_tree()
|
|
vim.cmd('30vsplit zk://tags_tree')
|
|
vim.opt_local.filetype = 'zktagstree'
|
|
folded_keys = {}
|
|
update_sidebar_context()
|
|
end
|
|
|
|
function M.toggle_folded_key()
|
|
local k = string.sub(vim.fn.getline('.'), 5, 5)
|
|
if folded_keys[k] then
|
|
folded_keys[k] = false
|
|
else
|
|
folded_keys[k] = true
|
|
end
|
|
update_sidebar_context()
|
|
end
|
|
|
|
return M
|