-- List or manage map features & enable magma furnaces
local help = [[=begin

feature
=======
Enables management of map features.

* Discovering a magma feature (magma pool, volcano, magma sea, or curious
  underground structure) permits magma workshops and furnaces to be built.
* Discovering a cavern layer causes plants (trees, shrubs, and grass) from
  that cavern to grow within your fortress.

Options:

:list:          Lists all map features in your current embark by index.
:magma:         Enable magma furnaces (discovers a random magma feature).
:show X:        Marks the selected map feature as discovered.
:hide X:        Marks the selected map feature as undiscovered.

=end]]

local map_features = df.global.world.features.map_features

function toggle_feature(idx, discovered)
    idx = tonumber(idx)
    if idx < 0 or idx >= #map_features then
        qerror('Invalid feature ID')
    end
    map_features[tonumber(idx)].flags.Discovered = discovered
end

function list_features()
    local name = df.new('string')
    for idx, feat in ipairs(map_features) do
        local tags = ''
        for _, t in pairs({'water', 'magma', 'subterranean', 'chasm', 'layer'}) do
            if feat['is' .. t:sub(1, 1):upper() .. t:sub(2)](feat) then
                tags = tags .. (' [%s]'):format(t)
            end
        end
        feat:getName(name)
        print(('Feature #%i is %s: "%s", type %s%s'):format(
            idx,
            feat.flags.Discovered and 'shown' or 'hidden',
            name.value,
            df.feature_type[feat:getType()],
            tags
        ))
    end
    df.delete(name)
end

function enable_magma_funaces()
    for idx, feat in ipairs(map_features) do
        if tostring(feat):find('magma') ~= nil then
            toggle_feature(idx, true)
            print('Enabled magma furnaces.')
            return
        end
    end
    dfhack.printerr('Could not find a magma-bearing feature.')
end

local args = {...}
if args[1] == 'list' then
    list_features()
elseif args[1] == 'magma' then
    enable_magma_funaces()
elseif args[1] == 'show' then
    toggle_feature(args[2], true)
elseif args[1] == 'hide' then
    toggle_feature(args[2], false)
else
    print((help:gsub('=[a-z]+', '')))
end