dfhack/plugins/lua/autobutcher.lua

81 lines
2.3 KiB
Lua

local _ENV = mkmodule('plugins.autobutcher')
local argparse = require('argparse')
local function is_int(val)
return val and val == math.floor(val)
end
local function is_positive_int(val)
return is_int(val) and val > 0
end
local function check_nonnegative_int(str)
local val = tonumber(str)
if is_positive_int(val) or val == 0 then return val end
qerror('expecting a non-negative integer, but got: '..tostring(str))
end
local function process_args(opts, args)
if args[1] == 'help' then
opts.help = true
return {}
end
return argparse.processArgsGetopt(args, {
{'h', 'help', handler=function() opts.help = true end},
})
end
local function process_races(opts, races, start_idx)
if #races < start_idx then
qerror('missing list of races (or "all" or "new" keywords)')
end
for i=start_idx,#races do
local race = races[i]
if race == 'all' then
opts.races_all = true
elseif race == 'new' then
opts.races_new = true
else
local str = df.new('string')
str.value = race
opts.races:insert('#', str)
end
end
end
function parse_commandline(opts, ...)
local positionals = process_args(opts, {...})
local command = positionals[1]
if command then opts.command = command end
if opts.help or not command or command == 'now' or
command == 'autowatch' or command == 'noautowatch' or
command == 'list' or command == 'list_export' then
return
end
if command == 'watch' or command == 'unwatch' or command == 'forget' then
process_races(opts, positionals, 2)
elseif command == 'target' then
opts.fk = check_nonnegative_int(positionals[2])
opts.mk = check_nonnegative_int(positionals[3])
opts.fa = check_nonnegative_int(positionals[4])
opts.ma = check_nonnegative_int(positionals[5])
process_races(opts, positionals, 6)
elseif command == 'ticks' then
local ticks = tonumber(positionals[2])
if not is_positive_int(ticks) then
qerror('number of ticks must be a positive integer: ' .. ticks)
else
opts.ticks = ticks
end
else
qerror(('unrecognized command: "%s"'):format(command))
end
end
return _ENV