local randomfactorycloak = {
    "cloakraid", "cloakskirm", "cloakriot", "cloakassault", "cloakarty", "cloakheavyraid", "cloaksnipe", "cloakbomb",
}
local randomfactoryshield = {
    "shieldraid", "shieldskirm", "shieldriot", "shieldassault", "shieldscout", "shieldfelon", "shieldbomb",
}
local randomfactoryveh = {
    "vehraid", "vehsupport", "vehriot", "vehassault", "veharty", "vehscout", "vehheavyarty", "vehcapture", 
}
local randomfactoryhover = {
    "hoverraid", "hoverskirm", "hoverriot", "hoverassault", "hoverarty", "hoverheavyraid", "hoverdepthcharge", 
}
local randomfactorygunship = {
    "gunshipraid", "gunshipskirm", "gunshipbomb", "gunshipassault", "gunshipheavyskirm", "gunshipemp", "gunshipkrow",
}
local randomfactoryplane = {
    "magpie", "bomberriot", "bomberprec", "bomberheavy", "bomberdisarm", "odin", 
}
local randomfactoryspider = {
    "spiderscout", "spiderskirm", "spiderriot", "spiderassault", "spideremp", "spidercrabe", "spiderantiheavy", 
}
local randomfactoryjump = {
    "jumpraid", "jumpskirm", "jumpassault", "jumparty", "jumpscout", "jumpsumo", "jumpbomb", 
}
local randomfactorytank = {
    "tankheavyraid", "tankriot", "tankassault", "tankarty", "tankraid", "tankheavyassault", "tankheavyarty", 
}
local randomfactoryamph = {
    "amphraid", "amphfloater", "amphriot", "amphsupport", "amphimpulse", "amphassault", "amphbomb",
}
local randomfactoryship = {
    "shiptorpraider", "shipskirm", "shipriot", "shipassault", "shiparty", "shipscout", "subraider",
}
local specificfactorycloak = {
    "cloakcon", "cloakaa", "cloakjammer",
}
local specificfactoryshield = {
    "shieldcon", "shieldaa", "shieldshield", "shieldarty",
}
local specificfactoryveh = {
    "vehcon", "vehaa",
}
local specificfactoryhover = {
    "hovercon", "hoveraa",
}
local specificfactorygunship = {
    "gunshipcon", "gunshipaa", "gunshiptrans", "gunshipheavytrans",
}
local specificfactoryplane = {
    "planefighter", "planecon", "planelightscout", "planeheavyfighter", "planescout",
}
local specificfactoryspider = {
    "spidercon", "spideraa",
}
local specificfactoryjump = {
    "jumpcon", "jumpblackhole", "jumpaa",
}
local specificfactorytank = {
    "tankcon", "tankaa",
}
local specificfactoryamph = {
    "amphcon", "amphlaunch", "amphaa", "amphtele", 
}
local specificfactoryship = {
    "shipcon", "shipaa",
}
local validfactories = {
    ["factorycloak"] = true,
    ["factoryshield"] = true,
    ["factoryveh"] = true,
    ["factoryhover"] = true,
    ["factorygunship"] = true,
    ["factoryplane"] = true,
    ["factoryspider"] = true,
    ["factoryjump"] = true,
    ["factorytank"] = true,
    ["factoryamph"] = true,
    ["factoryship"] = true,
}
local comvalidunitnames = {
    ["Guardian Commander"] = true,
    ["Strike Commander"] = true,
    ["Recon Commander"] = true,
    ["Engineer Commander"] = true,
}
local comspecificbuildoptions = {
    "striderhub", "turretlaser", "turretmissile", "turretriot", "turretemp", "turretgauss", "turretheavylaser", "turretaaclose", "turretaalaser", "turretaaflak", "turretaafar", "turretaaheavy", "turretimpulse", "turrettorpedo", "turretheavy", "turretantiheavy", "staticshield", "staticradar", "staticjammer", "staticheavyradar", "staticmissilesilo", "staticantinuke", "staticarty", "staticheavyarty", "staticnuke", "zenith", "raveparty", "mahlazer", "staticmex", "energywind", "energysolar", "energygeo", "energyfusion", "energysingu", "staticstorage", "energypylon", "staticcon", "staticrearm", "striderfunnelweb",
}
local comrandombuildoptions = {
    "factorycloak", "factoryshield", "factoryveh", "factoryhover", "factorygunship", "factoryplane", "factoryspider", "factoryjump", "factorytank", "factoryamph", "factoryship",
}
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 function selectrandomfactory()
    local factories = {}
    for factory, _ in pairs(validfactories) do
        table.insert(factories, factory)
    end
    local randomfactorytype = factories[math.random(#factories)]
    return randomfactorytype
end
local randomfactorytype = selectrandomfactory()
local randomfactorylist = _g["randomfactory" .. randomfactorytype]
local specificfactorylist = _g["specificfactory" .. randomfactorytype]
for name, ud in pairs(UnitDefs) do
    if facvalidunitnames[ud.name] then
        local shuffledrandomoptions = shuffle(randomfactorylist)
        local selectedoptions = {}
        
        for _, option in ipairs(specificfactorylist) do
            table.insert(selectedoptions, option)
        end
        
        table.insert(selectedoptions, shuffledrandomoptions[1])
        
        ud.buildoptions = selectedoptions
    end
end
for name, ud in pairs(UnitDefs) do
    if comvalidunitnames[ud.name] then
        local selectedoptions = {}
        for _, option in ipairs(comspecificbuildoptions) do
            table.insert(selectedoptions, option)
        end
        table.insert(selectedoptions, randomfactorytype)
        ud.buildoptions = selectedoptions
    end
end