data-final-fixes.lua

util = require "data/tf_util/tf_util"
names = require("shared")
local collision_mask_util = require("collision-mask-util")

-- We want the drones to have a unique set of collisions, which is not already accomodated for in the default collision
-- layers. So we need to add a completely new layer.

local drone_collision_mask = "construction_drone"
data:extend({ { type = "collision-layer", name = drone_collision_mask } })


local name = names.units.construction_drone
data.raw.unit[name].collision_mask = { not_colliding_with_itself = true, consider_tile_transitions = true, layers = {} }
data.raw.unit[name].collision_mask.layers[drone_collision_mask] = true

-- Add the new collision layer to all buildings and cliffs, along with all deep water tiles. We use the fact that they
-- have collisions with player and item here as a proxy to identify them. If this ever changes in the future, this might
-- break :^)

for _, prototype in pairs(collision_mask_util.collect_prototypes_with_layer("player")) do
    local mask = collision_mask_util.get_mask(prototype)
    if mask.layers and mask.layers["item"] then
        mask.layers[drone_collision_mask] = true
        prototype.collision_mask = mask
        -- print("proto " .. prototype.name)
        -- print("proto " .. serpent.block(prototype.collision_mask.layers))
    end
end