Factorio Forums

Code: Select all

function struct_construction_denial(entity, event)
    if not (entity and entity.valid) then return end
    if entity.force.name == "neutral" or entity.force.name == "enemy" or entity.force.get_friend(game.forces["enemy"]) then return end
    if entity.type ~= "entity-ghost" and entity.type ~= "car" then
        local area = {{
            entity.position.x - construction_denial_range,
            entity.position.y - construction_denial_range
        }, {
            entity.position.x + construction_denial_range,
            entity.position.y + construction_denial_range
        }}
        local enemies = entity.surface.find_entities_filtered{
          area= area,
          force="enemy"}
        for _, enemy in pairs(enemies) do
            if (enemy.type == "unit-spawner" or enemy.type == "turret") and Util.vectors_delta_length(entity.position, enemy.position) < construction_denial_range then
                -- deny construction
                if entity.prototype.mineable_properties
                  and entity.prototype.mineable_properties.products
                  and entity.prototype.mineable_properties.products[1]
                  and entity.prototype.mineable_properties.products[1].type == "item" then
                    local stack = {name=entity.prototype.mineable_properties.products[1].name, amount=1}
                    if event and event.name == "on_dolly_moved" and event.start_pos and event.player_index then
                        entity.teleport(event.start_pos)
                        game.players[event.player_index].print{"construction_denied_by_enemy"}
                        return
                    elseif event
                      and event.name == defines.events.on_built_entity
                      and event.player_index
                      and game.players[event.player_index]
                      and game.players[event.player_index].connected
                      and game.players[event.player_index].can_insert(stack) then
                        local player = game.players[event.player_index];
                        player.print{"construction_denied_by_enemy"}
                        player.insert(stack)
                        -- fix foa autofill adding items before removal
                        -- don't do other inventories, fuel includes turret ammo slots for some reason
                        Util.transfer_inventory_loose(entity, player, defines.inventory.fuel)
                    else
                        entity.surface.create_entity{
                          name = "item-on-ground",
                          position = entity.position,
                          stack = stack}
                    end
                end
                if math.random() < 0.05 then
                  local spawn_type = math.random() < 0.25 and "small-spitter" or "small-biter"
                  entity.surface.create_entity{
                    name = spawn_type,
                    position = entity.position,
                    force="enemy"}
                end
                entity.destroy()
                return
            end
        end
    end
end