163 lines
3.6 KiB
Lua
163 lines
3.6 KiB
Lua
local themes = require('telescope.themes')
|
|
local builtin = require('telescope.builtin')
|
|
local extensions = require('telescope._extensions').manager
|
|
local config = require('telescope.config')
|
|
local command = {}
|
|
|
|
local arg_value = {
|
|
['nil'] = nil,['""'] = '',['"'] = ''
|
|
}
|
|
|
|
local bool_type = {
|
|
['false'] = false,['true'] = true
|
|
}
|
|
|
|
-- convert command line string arguments to
|
|
-- lua number boolean type and nil value
|
|
local function convert_user_opts(user_opts)
|
|
local default_opts = config.values
|
|
|
|
local _switch = {
|
|
['boolean'] = function(key,val)
|
|
if val == 'false' then
|
|
user_opts[key] = false
|
|
return
|
|
end
|
|
user_opts[key] = true
|
|
end,
|
|
['number'] = function(key,val)
|
|
user_opts[key] = tonumber(val)
|
|
end,
|
|
['string'] = function(key,val)
|
|
if arg_value[val] ~= nil then
|
|
user_opts[key] = arg_value[val]
|
|
return
|
|
end
|
|
|
|
if bool_type[val] ~= nil then
|
|
user_opts[key] = bool_type[val]
|
|
end
|
|
end
|
|
}
|
|
|
|
local _switch_metatable = {
|
|
__index = function(_,k)
|
|
print(string.format('Type of %s does not match',k))
|
|
end
|
|
}
|
|
|
|
setmetatable(_switch,_switch_metatable)
|
|
|
|
for key,val in pairs(user_opts) do
|
|
if default_opts[key] ~= nil then
|
|
_switch[type(default_opts[key])](key,val)
|
|
else
|
|
_switch['string'](key,val)
|
|
end
|
|
end
|
|
end
|
|
|
|
-- receive the viml command args
|
|
-- it should be a table value like
|
|
-- {
|
|
-- cmd = 'find_files',
|
|
-- theme = 'dropdown',
|
|
-- extension_type = 'command'
|
|
-- opts = {
|
|
-- cwd = '***',
|
|
-- }
|
|
local function run_command(args)
|
|
local user_opts = args or {}
|
|
if next(user_opts) == nil and not user_opts.cmd then
|
|
print('[Telescope] your command miss args')
|
|
return
|
|
end
|
|
|
|
local cmd = user_opts.cmd
|
|
local opts = user_opts.opts or {}
|
|
local extension_type = user_opts.extension_type or ''
|
|
local theme = user_opts.theme or ''
|
|
|
|
if next(opts) ~= nil then
|
|
convert_user_opts(opts)
|
|
end
|
|
|
|
if string.len(theme) > 0 then
|
|
opts = themes[theme](opts)
|
|
end
|
|
|
|
if string.len(extension_type) > 0 and extension_type ~= '"' then
|
|
extensions[cmd][extension_type](opts)
|
|
return
|
|
end
|
|
|
|
if builtin[cmd] then
|
|
builtin[cmd](opts)
|
|
return
|
|
end
|
|
|
|
if rawget(extensions,cmd) then
|
|
extensions[cmd][cmd](opts)
|
|
end
|
|
end
|
|
|
|
-- @Summary get extensions sub command
|
|
-- register extensions dap gh etc.
|
|
-- input in command line `Telescope gh <TAB>`
|
|
-- Returns a list for each extension.
|
|
function command.get_extensions_subcommand()
|
|
local exts = require('telescope._extensions').manager
|
|
local complete_ext_table = {}
|
|
for cmd,value in pairs(exts) do
|
|
if type(value) == "table" then
|
|
local subcmds = {}
|
|
for key,_ in pairs(value) do
|
|
table.insert(subcmds, key)
|
|
end
|
|
complete_ext_table[cmd] = subcmds
|
|
end
|
|
end
|
|
return complete_ext_table
|
|
end
|
|
|
|
local split_keywords = {
|
|
['find_command'] = true,
|
|
['vimgrep_arguments'] = true,
|
|
['sections'] = true
|
|
}
|
|
|
|
function command.register_keyword(keyword)
|
|
split_keywords[keyword] = true
|
|
end
|
|
|
|
function command.load_command(cmd,...)
|
|
local args = {...}
|
|
if cmd == nil then
|
|
run_command({cmd = 'builtin'})
|
|
return
|
|
end
|
|
|
|
local user_opts = {}
|
|
user_opts['cmd'] = cmd
|
|
user_opts.opts = {}
|
|
|
|
for _,arg in ipairs(args) do
|
|
if arg:find('=',1) == nil then
|
|
user_opts['extension_type'] = arg
|
|
else
|
|
local param = vim.split(arg,'=')
|
|
if param[1] == 'theme' then
|
|
user_opts['theme'] = param[2]
|
|
elseif split_keywords[param[1]] then
|
|
user_opts.opts[param[1]] = vim.split(param[2],',')
|
|
else
|
|
user_opts.opts[param[1]] = param[2]
|
|
end
|
|
end
|
|
end
|
|
|
|
run_command(user_opts)
|
|
end
|
|
|
|
return command
|