local invalidunitnames = {
    ["Slicer"] = true,
}

local weapondefstoadd = {
    metalboom = {
        areaOfEffect       = 384,
        craterBoost        = 1,
        craterMult         = 3.5,
        edgeEffectiveness  = 0.4,
        explosionGenerator = "custom:ROACHPLOSION",
        explosionSpeed     = 10000,
        impulseBoost       = 0,
        impulseFactor      = 0.3,
        name               = "Explosion",
        scarIndices        = Shared.SCAR_SET.MEDIUM_AOE,
        soundHit           = "explosion/mini_nuke",
        damage = {
            default = 1200.8,
        },
        customParams = {
            burst = Shared.BURST_UNRELIABLE,
        },
    },
}

local function mergeweapondefs(weapondefs1, weapondefs2)
    local mergeddefs = {}
    for k, v in pairs(weapondefs1) do
        mergeddefs[k] = v
    end
    for k, v in pairs(weapondefs2) do
        if not mergeddefs[k] then
            mergeddefs[k] = v
        end
    end
    return mergeddefs
end

for name, ud in pairs(UnitDefs) do
    if not invalidunitnames[ud.name] then
        if ud.metalcost > 0 then
            Spring.Echo("Processing unit: " .. name)
            Spring.Echo("Updating unit: " .. ud.name)
            Spring.Echo("Existing weapondefs:")
            if ud.weapondefs then
                for k, v in pairs(ud.weapondefs) do
                    Spring.Echo("- " .. k)
                end
            else
                Spring.Echo("- None")
            end

            Spring.Echo("metalcost".. ud.metalcost)

            local areaofeffectbuff = math.floor(ud.metalcost * 1)
            weapondefstoadd.metalboom.areaofeffect = areaofeffectbuff

            local damagebuff = math.floor(ud.metalcost * 0.5)
            weapondefstoadd.metalboom.damage.default = damagebuff

            ud.weapondefs = mergeweapondefs(ud.weapondefs or {}, weapondefstoadd)
            
            Spring.Echo("Updated weapondefs:")
            if ud.weapondefs then
                for k, v in pairs(ud.weapondefs) do
                    Spring.Echo("- " .. k)
                end
            else
                Spring.Echo("- None")
            end

            ud.explodeas = "metalboom"
            ud.selfdestructas = "metalboom"

            Spring.Echo("explodeas for unit " .. ud.name .. ": " .. ud.explodeas)
            Spring.Echo("selfdestructas for unit " .. ud.name .. ": " .. ud.selfdestructas)
            Spring.Echo("damage".. ud.weapondefs.metalboom.damage.default)
            Spring.Echo("aoe".. ud.weapondefs.metalboom.areaofeffect)
        end
    end
end

local invalidUDNames={
	["slicer"]=true
}

local invalidunitnames = {
    --[[
    ["Guardian Commander"] = true,
    ["Strike Commander"] = true,
    ["Recon Commander"] = true,
    ["Engineer Commander"] = true,
    ["Slicer"] = true,
    --]]
}

--- create table of wd for ud
local function MakeWDForUD(ud)
	return {
            areaOfEffect       = math.pow(ud.metalcost,5/7)*8,
            craterBoost        = 1,
            craterMult         = 3.5,
            edgeEffectiveness  = 0.4,
            explosionGenerator = "custom:ROACHPLOSION",
            explosionSpeed     = 10000,
            impulseBoost       = 0,
            impulseFactor      = 0.3,
            name               = "Explosion",
            scarIndices        = Shared.SCAR_SET.MEDIUM_AOE,
            soundHit           = "explosion/mini_nuke",
            damage = {
                default = ud.metalcost*1,
            },
            customParams = {
                burst = Shared.BURST_UNRELIABLE,
            },
        }
end


--- set wds for ud
local function SetWDForUD(ud)
	ud.weapondefs=ud.weapondefs or {}
    ud.weapondefs.metalboom = MakeWDForUD(ud)
end


for name, ud in pairs(UnitDefs) do
	Spring.Echo("Processing unit: " .. name )
    if (
        (not invalidUDNames[name]) and
        (not invalidunitnames[ud.name]) and

        ((
            ud.metalcost > 0
        )) and
        (not
            (ud.selfDestructCountdown and ud.selfDestructCountdown < 1 ) -- don't work dk why
        )
    ) then
        Spring.Echo("Updating unit: " .. ud.name)
        
        

		SetWDForUD(ud)
        
        

        ud.explodeas = "metalboom"
        ud.selfdestructas = "metalboom"
		---ud.customParams and ud.customParams.stats_show_death_explosion
		
		ud.customParams = ud.customParams or {}
		ud.customParams.stats_show_death_explosion=true -- don't work dk why
    end
end

local invalidUDNames={
["terraunit"]=true,
["slicer"]=true
}
local function MakeWDForUD(ud)
return {
areaOfEffect = math.pow(ud.metalcost,5/7)*8,
craterBoost = 1,
craterMult = 3.5,
edgeEffectiveness = 0.4,
explosionGenerator = "custom:ROACHPLOSION",
explosionSpeed = 10000,
impulseBoost = 0,
impulseFactor = 0.3,
name = "Explosion",
scarIndices = Shared.SCAR_SET.MEDIUM_AOE,
soundHit = "explosion/mini_nuke",
damage = {
default = ud.metalcost*1,
},
customParams = {
burst = Shared.BURST_UNRELIABLE,
},
}
end
local function SetWDForUD(ud)
ud.weapondefs=ud.weapondefs or {}
ud.weapondefs.metalboom = MakeWDForUD(ud)
end
for name, ud in pairs(UnitDefs) do
if (
(not invalidUDNames[name]) and
((
ud.metalcost > 0
)) and
(not
(ud.selfDestructCountdown and ud.selfDestructCountdown < 1 )
)
) then
SetWDForUD(ud)
ud.explodeas = "metalboom"
ud.selfdestructas = "metalboom"
ud.customParams = ud.customParams or {}
ud.customParams.stats_show_death_explosion=true
end
end