82 lines
2.3 KiB
Lua
82 lines
2.3 KiB
Lua
-------------------------------------------------------------------------------
|
|
-- - Vimwiki extensions library -
|
|
-- Relies on vimwiki-reviews-lua for its vimwiki API
|
|
-------------------------------------------------------------------------------
|
|
local api = require("vimwiki_reviews.vimwiki_api")
|
|
local utils = require("vimwiki_reviews.utils")
|
|
local templates = require("vimwiki_reviews.templates")
|
|
local Path = require("plenary.path")
|
|
local scandir = require("plenary.scandir")
|
|
|
|
local M = {}
|
|
|
|
-- Gets path to any dir of provided vimwiki (by index)
|
|
function M.get_vimwiki_subdir(vimwiki_index, directory)
|
|
vimwiki_index = api.normalize_vimwiki_index(vimwiki_index)
|
|
local vimwiki = vim.g.vimwiki_list[vimwiki_index]
|
|
|
|
return vimwiki.path .. directory .. "/"
|
|
end
|
|
|
|
function M.get_directory_index(vimwiki_index, directory)
|
|
local dir = M.get_vimwiki_subdir(vimwiki_index, directory)
|
|
local path = Path:new(dir):expand()
|
|
|
|
local entries = scandir.scan_dir(path, {
|
|
hidden = false,
|
|
add_dirs = false,
|
|
respect_gitignore = true,
|
|
depth = 1,
|
|
})
|
|
|
|
local index = {}
|
|
|
|
for _, entry in pairs(entries) do
|
|
local filename = utils.get_filename_from_path(entry)
|
|
local ext = api.get_vimwiki_extension(vimwiki_index)
|
|
local noext = string.gsub(filename, ext, "")
|
|
|
|
index[noext] = filename
|
|
end
|
|
|
|
return index
|
|
end
|
|
|
|
-- Open subdirectory index file
|
|
function M.open_subdirectory_index_file(vimwiki_index, subdirectory)
|
|
local dir = M.get_vimwiki_subdir(vimwiki_index, subdirectory)
|
|
local ext = api.get_vimwiki_extension(vimwiki_index)
|
|
local filename = dir .. "index" .. ext
|
|
vim.cmd("edit " .. filename)
|
|
|
|
local index = M.get_directory_index(vimwiki_index, subdirectory)
|
|
|
|
local builder = templates.for_vimwiki(vimwiki_index)
|
|
|
|
local lines = {
|
|
builder.header(1, subdirectory:sub(1, 1):upper() .. subdirectory:sub(2)),
|
|
"",
|
|
}
|
|
|
|
-- Add items
|
|
local items = {}
|
|
for title, fname in pairs(index) do
|
|
if title ~= "index" then
|
|
title = string.gsub(title, "-", " ")
|
|
title = title:sub(1, 1):upper() .. title:sub(2)
|
|
table.insert(items, builder.list_item(builder.link(fname, title)))
|
|
end
|
|
end
|
|
table.sort(items)
|
|
|
|
for _, item in ipairs(items) do
|
|
table.insert(lines, item)
|
|
end
|
|
|
|
local buf = vim.api.nvim_get_current_buf()
|
|
vim.api.nvim_buf_set_lines(buf, 0, -1, false, {}) -- Clear out
|
|
vim.api.nvim_buf_set_lines(buf, 0, -1, false, lines) -- Put new contents
|
|
end
|
|
|
|
return M
|