mirror of
https://github.com/Refactorio/RedMew.git
synced 2024-12-14 10:13:13 +02:00
50 lines
1.5 KiB
Lua
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)
|