iframe-proxy | Sunbelt Computer Software

102 lines (83 loc) · 2.44 KB

-- Blocks most types of visitors (caravans, migrants, etc.)
--@ enable=true
--@ module=true
local argparse = require('argparse')
local repeatutil = require('repeat-util')
local GLOBAL_KEY = 'hermit'
local timed_events = df.global.timed_events
local allowlist = {
    [df.timed_event_type.WildlifeCurious] = true,
    [df.timed_event_type.WildlifeMischievous] = true,
    [df.timed_event_type.WildlifeFlier] = true,
enabled = enabled or false
function isEnabled()
    return enabled
local function persist_state()
    dfhack.persistent.saveSiteData(GLOBAL_KEY, {enabled=enabled})
local function load_state()
    local persisted_data = dfhack.persistent.getSiteData(GLOBAL_KEY, {enabled=false})
    enabled = persisted_data.enabled
function event_loop()
    if not enabled then return end
    local tmp_events = {} --as:df.timed_event[]
    for _, event in pairs(timed_events) do
        table.insert(tmp_events, event)
    timed_events:resize(0)
    for _, event in pairs(tmp_events) do
        if allowlist[event.type] then
            timed_events:insert('#', event)
        else
            event:delete()
        end
    repeatutil.scheduleUnlessAlreadyScheduled(GLOBAL_KEY, 1, 'days', event_loop)
local function print_status()
    print(('hermit is currently %s.'):format(enabled and 'enabled' or 'disabled'))
dfhack.onStateChange[GLOBAL_KEY] = function(sc)
    if sc == SC_MAP_UNLOADED then
        enabled = false
        return
    if sc ~= SC_MAP_LOADED or df.global.gamemode ~= df.game_mode.DWARF then
        return
    load_state()
    event_loop()
if dfhack_flags.module then
if df.global.gamemode ~= df.game_mode.DWARF or not dfhack.isMapLoaded() then
    dfhack.printerr('hermit needs a loaded fortress to work')
local args, opts = {...}, {}
if dfhack_flags and dfhack_flags.enable then
    args = {dfhack_flags.enable_state and 'enable' or 'disable'}
local positionals = argparse.processArgsGetopt(args, {
    {'h', 'help', handler=function() opts.help = true end},
local command = positionals[1]
if command == 'help' or opts.help then
    print(dfhack.script_help())
elseif command == 'enable' then
    enabled = true
    persist_state()
    event_loop()
elseif command == 'disable' then
    enabled = false
    persist_state()
    repeatutil.cancel(GLOBAL_KEY)
elseif not command or command == 'status' then
    print_status()
    print(dfhack.script_help())

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

hermit.lua

Latest commit

History

hermit.lua

File metadata and controls