82 lines
1.6 KiB
Lua
82 lines
1.6 KiB
Lua
local core = require "core"
|
|
local command = {}
|
|
|
|
command.map = {}
|
|
|
|
local always_true = function() return true end
|
|
|
|
|
|
function command.add(predicate, map)
|
|
predicate = predicate or always_true
|
|
if type(predicate) == "string" then
|
|
predicate = require(predicate)
|
|
end
|
|
if type(predicate) == "table" then
|
|
local class = predicate
|
|
predicate = function() return core.active_view:is(class) end
|
|
end
|
|
for name, fn in pairs(map) do
|
|
assert(not command.map[name], "command already exists: " .. name)
|
|
command.map[name] = { predicate = predicate, perform = fn }
|
|
end
|
|
end
|
|
|
|
|
|
local function capitalize_first(str)
|
|
return str:sub(1, 1):upper() .. str:sub(2)
|
|
end
|
|
|
|
function command.prettify_name(name)
|
|
return name:gsub(":", ": "):gsub("-", " "):gsub("%S+", capitalize_first)
|
|
end
|
|
|
|
|
|
function command.get_all_valid()
|
|
local res = {}
|
|
for name, cmd in pairs(command.map) do
|
|
if cmd.predicate() then
|
|
table.insert(res, name)
|
|
end
|
|
end
|
|
return res
|
|
end
|
|
|
|
|
|
local function perform(name)
|
|
local cmd = command.map[name]
|
|
if cmd and cmd.predicate() then
|
|
cmd.perform()
|
|
return true
|
|
end
|
|
return false
|
|
end
|
|
|
|
|
|
function command.perform(...)
|
|
local ok, res = core.try(perform, ...)
|
|
return not ok or res
|
|
end
|
|
|
|
|
|
function command.perform_many(n, name)
|
|
local cmd = command.map[name]
|
|
if not cmd or not cmd.predicate() then return false end
|
|
pcall(function()
|
|
for i = 1, n do
|
|
cmd.perform()
|
|
end
|
|
end)
|
|
return true
|
|
end
|
|
|
|
|
|
function command.add_defaults()
|
|
local reg = { "core", "root", "command", "doc", "findreplace", "files" }
|
|
for _, name in ipairs(reg) do
|
|
require("core.commands." .. name)
|
|
end
|
|
end
|
|
|
|
|
|
return command
|