90 lines
2.2 KiB
Lua
90 lines
2.2 KiB
Lua
local M = {}
|
|
|
|
---Get the directory pattern based on OS
|
|
---@return string
|
|
function M.dir_pattern()
|
|
local pattern = "/"
|
|
if vim.fn.has("win32") == 1 then
|
|
pattern = "[\\:]"
|
|
end
|
|
return pattern
|
|
end
|
|
|
|
--- Escape special pattern matching characters in a string
|
|
---@param input string
|
|
---@return string
|
|
function M.escape_dir_pattern(input)
|
|
local magic_chars = { "%", "(", ")", "+", "-", "*", "?", "[", "^", "$" }
|
|
|
|
for _, char in ipairs(magic_chars) do
|
|
input = input:gsub("%" .. char, "%%" .. char)
|
|
end
|
|
|
|
return input
|
|
end
|
|
|
|
---Check if a directory is a subdirectory of another
|
|
---@param parent string
|
|
---@param child string
|
|
---@return boolean
|
|
function M.is_subdirectory(parent, child)
|
|
return vim.startswith(child, parent)
|
|
end
|
|
|
|
---Check if a directory exists in the given table of directories
|
|
---@param dir string The directory to check
|
|
---@param dirs table The table of directories to search in
|
|
---@return boolean
|
|
function M.dirs_match(dir, dirs)
|
|
dir = M.escape_dir_pattern(vim.fn.expand(dir))
|
|
|
|
for _, search in ipairs(dirs) do
|
|
if type(search) == "string" then
|
|
search = M.escape_dir_pattern(vim.fn.expand(search))
|
|
if M.is_subdirectory(search, dir) then
|
|
return true
|
|
end
|
|
elseif type(search) == "table" then
|
|
if search.exact then
|
|
search = M.escape_dir_pattern(vim.fn.expand(search[1]))
|
|
if dir == search then
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
return false
|
|
end
|
|
|
|
---Check if a string matches and entry in a given table
|
|
---@param val string
|
|
---@param tbl table
|
|
---@param callback function
|
|
---@return boolean
|
|
function M.in_table(val, tbl, callback)
|
|
if val == nil then
|
|
return false
|
|
end
|
|
|
|
return tbl
|
|
and next(vim.tbl_filter(function(pattern)
|
|
if pattern.exact then
|
|
pattern = pattern[1]
|
|
-- Stripping off the trailing backslash that a user might put here,
|
|
-- but only if we aren't looking at the root directory
|
|
if pattern:sub(-1) == M.dir_pattern() and pattern:len() > 1 then
|
|
pattern = pattern:sub(1, -2)
|
|
end
|
|
return val == pattern
|
|
else
|
|
if callback and type(callback) == "function" then
|
|
pattern = callback(pattern)
|
|
end
|
|
return val:match(pattern)
|
|
end
|
|
end, tbl))
|
|
end
|
|
|
|
return M
|