1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-12 10:04:40 +02:00
RedMew/features/autodeconstruct.lua
2019-01-04 15:02:55 -05:00

50 lines
1.5 KiB
Lua

--Author: Valansch
local Event = require 'utils.event'
local Token = require 'utils.token'
local Task = require 'utils.task'
local function is_depleted(drill, entity)
local position = drill.position
local area
if drill.name == 'electric-mining-drill' then
area = {{position.x - 2.5, position.y - 2.5}, {position.x + 2.5, position.y + 2.5}}
else
area = {{position.x - 1, position.y - 1}, {position.x + 1, position.y + 1}}
end
for _, resource in pairs(drill.surface.find_entities_filtered {type = 'resource', area = area}) do
if resource ~= entity and resource.name ~= 'crude-oil' then
return false
end
end
return true
end
local callback =
Token.register(
function(drill)
if drill.valid then
drill.order_deconstruction(drill.force)
end
end
)
local function on_resource_depleted(event)
local entity = event.entity
if entity.name == 'uranium-ore' then
return nil
end
local position = entity.position
local area = {{position.x - 1, position.y - 1}, {position.x + 1, position.y + 1}}
local drills = event.entity.surface.find_entities_filtered {area = area, type = 'mining-drill'}
for _, drill in ipairs(drills) do
if drill.name ~= 'pumpjack' and is_depleted(drill, entity) then
Task.set_timeout_in_ticks(5, callback, drill)
end
end
end
Event.add(defines.events.on_resource_depleted, on_resource_depleted)