mirror of
https://github.com/Refactorio/RedMew.git
synced 2024-12-16 10:19:27 +02:00
a127196e9c
- update .luacheckrc. - Add stack limits to airstrike and rocket tank market items. - Refactor rocket_tanks.lua
122 lines
3.9 KiB
Lua
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)
|