1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-16 10:19:27 +02:00
RedMew/map_gen/maps/crash_site/rocket_tanks.lua
James Gillham a127196e9c Updates:
- update .luacheckrc.
- Add stack limits to airstrike and rocket tank market items.
- Refactor rocket_tanks.lua
2020-12-27 14:03:18 +00:00

122 lines
3.9 KiB
Lua

local Event = require 'utils.event'
local Global = require 'utils.global'
local Toast = require 'features.gui.toast'
local Retailer = require 'features.retailer'
local Token = require 'utils.token'
local tank_entities = {}
local tank_research = {interval_level = 1}
Global.register({tank_entities = tank_entities, tank_research = tank_research}, function(tbl)
tank_entities = tbl.tank_entities
tank_research = tbl.tank_research
end)
local function register_tank(entity)
if not entity then
return
end
if not entity.valid then
return
end
if entity.name ~= 'tank' then
return
end
tank_entities[entity.unit_number] = entity
script.register_on_entity_destroyed(entity)
end
Event.add(defines.events.on_entity_destroyed, function(event)
tank_entities[event.unit_number] = nil
end)
Event.add(defines.events.on_robot_built_entity, function(event)
register_tank(event.created_entity)
end)
Event.add(defines.events.on_built_entity, function(event)
register_tank(event.created_entity)
end)
Event.add(defines.events.on_entity_cloned, function(event)
register_tank(event.destination)
end)
local static_entities_to_check = {
'spitter-spawner',
'biter-spawner',
'small-worm-turret',
'medium-worm-turret',
'big-worm-turret',
'behemoth-worm-turret'
}
local on_tick = Token.register(function()
for _, tank in pairs(tank_entities) do
if not tank.valid or not tank.get_driver() then -- only allow tanks to fire rockets when the tank has a driver
goto continue
end
local inv = tank.get_inventory(defines.inventory.car_trunk)
local rocket_count = inv.get_item_count("rocket")
if rocket_count <= 0 then
goto continue
end
local surface = tank.surface
local targets = surface.find_entities_filtered {
position = tank.position,
radius = 30,
force = "enemy",
name = static_entities_to_check,
limit = 1
}
local target = targets[1]
if target then
surface.create_entity {name = "rocket", position = tank.position, target = target, speed = 0.2}
inv.remove({name = "rocket", count = 1})
end
::continue::
end
end)
local rocket_tank_level_intervals = {[2] = 180, [3] = 120, [4] = 60, [5] = 30}
Event.add(Retailer.events.on_market_purchase, function(event)
local market_id = event.group_name
local group_label = Retailer.get_market_group_label(market_id)
if group_label ~= 'Spawn' then
return
end
local item = event.item
if item.type ~= 'rocket_tanks' and event.name ~= 'rocket_tanks_fire_rate' then
return
end
local interval_level = tank_research.interval_level
local max_level = 5
if interval_level < max_level then
tank_research.interval_level = tank_research.interval_level + 1
Toast.toast_all_players(15, {'command_description.crash_site_rocket_tank_upgrade_success', interval_level})
item.name_label = {'command_description.crash_site_rocket_tanks_name_label', (interval_level + 1)}
item.price = (interval_level + 1) * 7500
Retailer.set_item(market_id, item) -- this updates the retailer with the new item values.
end
if interval_level >= 4 then -- update label, set price to 0, disable further purchases
item.price = 0
item.name_label = {'command_description.crash_site_rocket_tanks_name_label', {'command_description.max_level'}}
item.disabled = true
item.disabled_reason = {'command_description.max_level'}
Retailer.set_item(market_id, item)
end
if interval_level > 2 then
Event.remove_removable_nth_tick(rocket_tank_level_intervals[tank_research.interval_level - 1], on_tick)
end
Event.add_removable_nth_tick(rocket_tank_level_intervals[tank_research.interval_level], on_tick)
end)