mirror of https://github.com/Hilbis/Hilbish
54 lines
1.2 KiB
Lua
54 lines
1.2 KiB
Lua
|
local function curry(f)
|
||
|
|
return function (x) return function (y) return f(x,y) end end
|
||
|
|
end
|
||
|
|
|
||
|
|
local flags = {}
|
||
|
|
local function flag(f, description)
|
||
|
|
flags[f] = {description}
|
||
|
|
end
|
||
|
|
|
||
|
|
local addflag = curry(flag)
|
||
|
|
|
||
|
|
addflag '-A' 'Ask for password via askpass or $SUDO_ASKPASS'
|
||
|
|
addflag '-B' 'Ring the bell as part of the password prompt.'
|
||
|
|
|
||
|
|
hilbish.complete('command.sudo', function(query, ctx, fields)
|
||
|
|
table.remove(fields, 1)
|
||
|
|
local nonflags = table.filter(fields, function(v)
|
||
|
|
if v == '' then
|
||
|
|
return false
|
||
|
|
end
|
||
|
|
return v:match '^%-' == nil
|
||
|
|
end)
|
||
|
|
|
||
|
|
if #fields == 1 or #nonflags == 0 then
|
||
|
|
-- complete commands or sudo flags
|
||
|
|
if query:match ('^%-') then
|
||
|
|
local compFlags = {}
|
||
|
|
for flg, flgstuff in pairs(flags) do
|
||
|
|
if flg:match('^' .. query) then
|
||
|
|
compFlags[flg] = flgstuff
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
local compGroup = {
|
||
|
|
items = compFlags,
|
||
|
|
type = 'list'
|
||
|
|
}
|
||
|
|
|
||
|
|
return {compGroup}, query
|
||
|
|
end
|
||
|
|
|
||
|
|
local comps, pfx = hilbish.completion.bins(query, ctx, fields)
|
||
|
|
local compGroup = {
|
||
|
|
items = comps,
|
||
|
|
type = 'grid'
|
||
|
|
}
|
||
|
|
|
||
|
|
return {compGroup}, pfx
|
||
|
|
end
|
||
|
|
|
||
|
|
-- otherwise, get command flags
|
||
|
|
return hilbish.completion.call('command.' .. fields[2], query, ctx, fields)
|
||
|
|
end)
|