bG9jYWwgZnVuY3Rpb24gc2h1ZmZsZSh0YmwpCiAgICBsb2NhbCBsZW4gPSAjdGJsCiAgICBmb3IgaSA9IGxlbiwgMiwgLTEgZG8KICAgICAgICBsb2NhbCBqID0gbWF0aC5yYW5kb20oaSkKICAgICAgICB0YmxbaV0sIHRibFtqXSA9IHRibFtqXSwgdGJsW2ldCiAgICBlbmQKICAgIHJldHVybiB0YmwKZW5kCgpsb2NhbCBDb21WYWxpZFVuaXROYW1lcyA9IHsKICAgIFsiR3VhcmRpYW4gQ29tbWFuZGVyIl0gPSB0cnVlLAogICAgWyJTdHJpa2UgQ29tbWFuZGVyIl0gPSB0cnVlLAogICAgWyJSZWNvbiBDb21tYW5kZXIiXSA9IHRydWUsCiAgICBbIkVuZ2luZWVyIENvbW1hbmRlciJdID0gdHJ1ZQp9Cgpsb2NhbCBSYW5kb21CdWlsZE9wdGlvbnMgPSB7CiAgICAiY2xvYWtyYWlkIiwgImNsb2Frc2tpcm0iLCAiY2xvYWtyaW90IiwgImNsb2FrYXNzYXVsdCIsICJjbG9ha2FydHkiLCAiY2xvYWtoZWF2eXJhaWQiLCAiY2xvYWtzbmlwZSIsICJjbG9ha2JvbWIiLAogICAgInNoaWVsZHJhaWQiLCAic2hpZWxkc2tpcm0iLCAic2hpZWxkcmlvdCIsICJzaGllbGRhc3NhdWx0IiwgInNoaWVsZHNjb3V0IiwgInNoaWVsZGZlbG9uIiwgInNoaWVsZGJvbWIiLAogICAgInZlaHJhaWQiLCAidmVoc3VwcG9ydCIsICJ2ZWhyaW90IiwgInZlaGFzc2F1bHQiLCAidmVoYXJ0eSIsICJ2ZWhzY291dCIsICJ2ZWhoZWF2eWFydHkiLCAidmVoY2FwdHVyZSIsCiAgICAiaG92ZXJyYWlkIiwgImhvdmVyc2tpcm0iLCAiaG92ZXJyaW90IiwgImhvdmVyYXNzYXVsdCIsICJob3ZlcmFydHkiLCAiaG92ZXJoZWF2eXJhaWQiLCAiaG92ZXJkZXB0aGNoYXJnZSIsCiAgICAiZ3Vuc2hpcHJhaWQiLCAiZ3Vuc2hpcHNraXJtIiwgImd1bnNoaXBib21iIiwgImd1bnNoaXBhc3NhdWx0IiwgImd1bnNoaXBoZWF2eXNraXJtIiwgImd1bnNoaXBlbXAiLCAiZ3Vuc2hpcGtyb3ciLAogICAgImd1bnNoaXBoZWF2eXRyYW5zIiwgIm1hZ3BpZSIsICJib21iZXJyaW90IiwgImJvbWJlcnByZWMiLCAiYm9tYmVyaGVhdnkiLCAiYm9tYmVyZGlzYXJtIiwgIm9kaW4iLAogICAgInBsYW5lZmlnaHRlciIsICJzcGlkZXJzY291dCIsICJzcGlkZXJza2lybSIsICJzcGlkZXJyaW90IiwgInNwaWRlcmFzc2F1bHQiLCAic3BpZGVyZW1wIiwgInNwaWRlcmNyYWJlIiwgInNwaWRlcmFudGloZWF2eSIsCiAgICAianVtcHJhaWQiLCAianVtcHNraXJtIiwgImp1bXBhc3NhdWx0IiwgImp1bXBhcnR5IiwgImp1bXBzY291dCIsICJqdW1wc3VtbyIsICJqdW1wYm9tYiIsICJqdW1wYmxhY2tob2xlIiwKICAgICJ0YW5raGVhdnlyYWlkIiwgInRhbmtyaW90IiwgInRhbmthc3NhdWx0IiwgInRhbmthcnR5IiwgInRhbmtyYWlkIiwgInRhbmtoZWF2eWFzc2F1bHQiLCAidGFua2hlYXZ5YXJ0eSIsCiAgICAiYW1waHJhaWQiLCAiYW1waGZsb2F0ZXIiLCAiYW1waHJpb3QiLCAiYW1waHN1cHBvcnQiLCAiYW1waGltcHVsc2UiLCAiYW1waGFzc2F1bHQiLCAiYW1waGJvbWIiLCAiYW1waGxhdW5jaCIsCiAgICAic2hpcHRvcnByYWlkZXIiLCAic2hpcHNraXJtIiwgInNoaXByaW90IiwgInNoaXBhc3NhdWx0IiwgInNoaXBhcnR5IiwgInNoaXBzY291dCIsICJzdWJyYWlkZXIiCn0KCmxvY2FsIHNodWZmbGVkT3B0aW9ucyA9IHNodWZmbGUoUmFuZG9tQnVpbGRPcHRpb25zKQpsb2NhbCBzZWxlY3RlZFVuaXQgPSBzaHVmZmxlZE9wdGlvbnNbMV0KClNwcmluZy5FY2hvKCJTZWxlY3RlZCBVbml0OiAiIC4uIHNlbGVjdGVkVW5pdCkKCmZvciBuYW1lLCB1ZCBpbiBwYWlycyhVbml0RGVmcykgZG8KICAgIGlmIENvbVZhbGlkVW5pdE5hbWVzW3VkLm5hbWVdIHRoZW4KICAgICAgICB1ZC5jdXN0b21QYXJhbXMgPSB1ZC5jdXN0b21QYXJhbXMgb3Ige30KCiAgICAgICAgdWQuY3VzdG9tUGFyYW1zLmdyZXlfZ29vID0gMgogICAgICAgIHVkLmN1c3RvbVBhcmFtcy5ncmV5X2dvb19kcmFpbiA9IDAuMDEKICAgICAgICB1ZC5jdXN0b21QYXJhbXMuZ3JleV9nb29fY29zdCA9IDAuMDEKICAgICAgICB1ZC5jdXN0b21QYXJhbXMuZ3JleV9nb29fcmFuZ2UgPSAxMDAwMAogICAgICAgIHVkLmN1c3RvbVBhcmFtcy5ncmV5X2dvb19zcGF3biA9IHNlbGVjdGVkVW5pdAogICAgICAgIAogICAgICAgIFNwcmluZy5FY2hvKCJVbml0OiAiIC4uIHVkLm5hbWUpCiAgICAgICAgU3ByaW5nLkVjaG8oImdyZXlfZ29vOiAiIC4uIHRvc3RyaW5nKHVkLmN1c3RvbVBhcmFtcy5ncmV5X2dvbykpCiAgICAgICAgU3ByaW5nLkVjaG8oImdyZXlfZ29vX2RyYWluOiAiIC4uIHRvc3RyaW5nKHVkLmN1c3RvbVBhcmFtcy5ncmV5X2dvb19kcmFpbikpCiAgICAgICAgU3ByaW5nLkVjaG8oImdyZXlfZ29vX2Nvc3Q6ICIgLi4gdG9zdHJpbmcodWQuY3VzdG9tUGFyYW1zLmdyZXlfZ29vX2Nvc3QpKQogICAgICAgIFNwcmluZy5FY2hvKCJncmV5X2dvb19yYW5nZTogIiAuLiB0b3N0cmluZyh1ZC5jdXN0b21QYXJhbXMuZ3JleV9nb29fcmFuZ2UpKQogICAgICAgIFNwcmluZy5FY2hvKCJncmV5X2dvb19zcGF3bjogIiAuLiB0b3N0cmluZyh1ZC5jdXN0b21QYXJhbXMuZ3JleV9nb29fc3Bhd24pKQogICAgZW5kCmVuZAo=

local function shuffle(tbl)
    local len = #tbl
    for i = len, 2, -1 do
        local j = math.random(i)
        tbl[i], tbl[j] = tbl[j], tbl[i]
    end
    return tbl
end

local ComValidUnitNames = {
    ["Guardian Commander"] = true,
    ["Strike Commander"] = true,
    ["Recon Commander"] = true,
    ["Engineer Commander"] = true
}

local RandomBuildOptions = {
    "cloakraid", "cloakskirm", "cloakriot", "cloakassault", "cloakarty", "cloakheavyraid", "cloaksnipe", "cloakbomb",
    "shieldraid", "shieldskirm", "shieldriot", "shieldassault", "shieldscout", "shieldfelon", "shieldbomb",
    "vehraid", "vehsupport", "vehriot", "vehassault", "veharty", "vehscout", "vehheavyarty", "vehcapture",
    "hoverraid", "hoverskirm", "hoverriot", "hoverassault", "hoverarty", "hoverheavyraid", "hoverdepthcharge",
    "gunshipraid", "gunshipskirm", "gunshipbomb", "gunshipassault", "gunshipheavyskirm", "gunshipemp", "gunshipkrow",
    "gunshipheavytrans", "magpie", "bomberriot", "bomberprec", "bomberheavy", "bomberdisarm", "odin",
    "planefighter", "spiderscout", "spiderskirm", "spiderriot", "spiderassault", "spideremp", "spidercrabe", "spiderantiheavy",
    "jumpraid", "jumpskirm", "jumpassault", "jumparty", "jumpscout", "jumpsumo", "jumpbomb", "jumpblackhole",
    "tankheavyraid", "tankriot", "tankassault", "tankarty", "tankraid", "tankheavyassault", "tankheavyarty",
    "amphraid", "amphfloater", "amphriot", "amphsupport", "amphimpulse", "amphassault", "amphbomb", "amphlaunch",
    "shiptorpraider", "shipskirm", "shipriot", "shipassault", "shiparty", "shipscout", "subraider"
}

local shuffledOptions = shuffle(RandomBuildOptions)
local selectedUnit = shuffledOptions[1]

Spring.Echo("Selected Unit: " .. selectedUnit)

for name, ud in pairs(UnitDefs) do
    if ComValidUnitNames[ud.name] then
        ud.customParams = ud.customParams or {}

        ud.customParams.grey_goo = 2
        ud.customParams.grey_goo_drain = 0.01
        ud.customParams.grey_goo_cost = 0.01
        ud.customParams.grey_goo_range = 10000
        ud.customParams.grey_goo_spawn = selectedUnit
        
        Spring.Echo("Unit: " .. ud.name)
        Spring.Echo("grey_goo: " .. tostring(ud.customParams.grey_goo))
        Spring.Echo("grey_goo_drain: " .. tostring(ud.customParams.grey_goo_drain))
        Spring.Echo("grey_goo_cost: " .. tostring(ud.customParams.grey_goo_cost))
        Spring.Echo("grey_goo_range: " .. tostring(ud.customParams.grey_goo_range))
        Spring.Echo("grey_goo_spawn: " .. tostring(ud.customParams.grey_goo_spawn))
    end
end