force.lua

local wildlife = reqscript('fix/wildlife')

local function findCiv(civ)
 if civ == 'player' then return df.historical_entity.find(df.global.plotinfo.civ_id) end
 if tonumber(civ) then return df.historical_entity.find(tonumber(civ)) end
 civ = string.lower(tostring(civ))
 for _,entity in ipairs(df.global.world.entities.all) do
 if string.lower(entity.entity_raw.code) == civ then return entity end
 end
end

local args = { ... }
if #args < 1 then qerror('missing event type') end
if args[1]:find('help') then
 print(dfhack.script_help())
 return
end

local eventType = args[1]:upper()

-- handle synthetic events
if eventType == 'WILDLIFE' then
 wildlife.free_all_wildlife(args[2] == 'all')
 return
end

-- handle native events
for _, type in ipairs(df.timed_event_type) do
 if type:lower() == args[1]:lower() then
 eventType = type
 end
end
if not df.timed_event_type[eventType] then
 qerror('unknown event type: ' .. args[1])
end
if eventType == 'FeatureAttack' then
 qerror('Event type: FeatureAttack is not currently supported')
end

local civ

if eventType == 'Caravan' or eventType == 'Diplomat' then
 civ = findCiv(args[2] or 'player')
 if not civ then
 qerror('unable to find civilization: '..tostring(civ))
 end
elseif eventType == 'Migrants' then
 civ = findCiv('player')
end

df.global.timed_events:insert('#', {
 new=true,
 type=df.timed_event_type[eventType],
 season=df.global.cur_season,
 season_ticks=df.global.cur_season_tick,
 entity=civ,
 feature_ind=-1,
})