86 lines
2.4 KiB
Lua
86 lines
2.4 KiB
Lua
|
|
||
|
|
local syndromeUtil = require 'syndromeUtil'
|
||
|
|
local utils = require 'utils'
|
||
|
|
|
||
|
|
mode = mode or utils.invert({
|
||
|
|
'add',
|
||
|
|
'erase',
|
||
|
|
'eraseAll'
|
||
|
|
})
|
||
|
|
|
||
|
|
local args = {...}
|
||
|
|
local i = 1
|
||
|
|
local syndrome
|
||
|
|
local resetPolicy = syndromeUtil.ResetPolicy.NewInstance
|
||
|
|
local currentMode = mode.add
|
||
|
|
local target
|
||
|
|
local skipImmunities = false
|
||
|
|
while i <= #args do
|
||
|
|
if args[i] == '-help' then
|
||
|
|
print('add-syndrome usage:')
|
||
|
|
print(' -help')
|
||
|
|
print(' print this help message')
|
||
|
|
print(' -syndrome [name]')
|
||
|
|
print(' select a syndrome by name')
|
||
|
|
print(' -resetPolicy DoNothing')
|
||
|
|
print(' if the target already has an instance of the syndrome, do nothing')
|
||
|
|
print(' -resetPolicy ResetDuration')
|
||
|
|
print(' if the target already has an instance of the syndrome, reset the duration to maximum')
|
||
|
|
print(' -resetPolicy AddDuration')
|
||
|
|
print(' if the target already has an instance of the syndrome, add the maximum duration to the time remaining')
|
||
|
|
print(' -resetPolicy NewInstance')
|
||
|
|
print(' if the target already has an instance of the syndrome, add a new instance of the syndrome')
|
||
|
|
print(' -erase')
|
||
|
|
print(' instead of adding a syndrome, erase one')
|
||
|
|
print(' -eraseAll')
|
||
|
|
print(' instead of adding a syndrome, erase every instance of it')
|
||
|
|
print(' -target [id]')
|
||
|
|
print(' set the target unit for infection')
|
||
|
|
print(' -skipImmunities')
|
||
|
|
print(' don\'t check syn_class immune/affected stuff.')
|
||
|
|
return
|
||
|
|
elseif args[i] == '-syndrome' then
|
||
|
|
local syn_name = args[i+1]
|
||
|
|
for _,syn in ipairs(df.global.world.raws.syndromes.all) do
|
||
|
|
if syn.syn_name == syn_name then
|
||
|
|
syndrome = syn
|
||
|
|
break
|
||
|
|
end
|
||
|
|
end
|
||
|
|
i = i+2
|
||
|
|
elseif args[i] == '-resetPolicy' then
|
||
|
|
resetPolicy = syndromeUtil.ResetPolicy[args[i+1]]
|
||
|
|
if not resetPolicy then
|
||
|
|
qerror('Invalid arguments to add-syndrome.')
|
||
|
|
end
|
||
|
|
i = i+2
|
||
|
|
elseif args[i] == '-erase' then
|
||
|
|
currentMode = mode.erase
|
||
|
|
i = i+1
|
||
|
|
elseif args[i] == '-eraseAll' then
|
||
|
|
currentMode = mode.eraseAll
|
||
|
|
i = i+1
|
||
|
|
elseif args[i] == '-add' then
|
||
|
|
currentMode = mode.add
|
||
|
|
i = i+1
|
||
|
|
elseif args[i] == '-target' then
|
||
|
|
target = df.unit.find(tonumber(args[i+1]))
|
||
|
|
if not target then
|
||
|
|
qerror('Invalid unit id argument to add-syndrome.')
|
||
|
|
end
|
||
|
|
i = i+2
|
||
|
|
elseif args[i] == '-skipImmunities' then
|
||
|
|
skipImmunities = true
|
||
|
|
i = i+1
|
||
|
|
else
|
||
|
|
qerror('Invalid arguments to add-syndrome.')
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
if skipImmunities then
|
||
|
|
syndromeUtil.infectWithSyndrome(target,syndrome,resetPolicy)
|
||
|
|
else
|
||
|
|
syndromeUtil.infectWithSyndromeIfValidTarget(target,syndrome,resetPolicy)
|
||
|
|
end
|
||
|
|
|