The Random Specialists v2

HedgewarsScriptLoad("/Scripts/Locale.lua")

local hhs = {}
local weapons = {}
local amount = {}
local hogammo = {}
local hoghealth = {}
local classes = {
	[amGrenade] = {loc("Veteran"), "Rambo"},
	[amClusterBomb] = {loc("Bomber"), "ushanka"},
	[amMortar] = {loc("Terminator"), "Terminator_Glasses"},
	[amBee] = {loc("Researcher"), "scif_2001Y"},
	[amShotgun] = {loc("Hunter"), "RobinHood"},
	[amPickHammer] = {loc("Worker"), "constructor"},
	[amRope] = {loc("Ninja"), "NinjaTriangle"},
	[amMine] = {loc("Terrorist"), "Skull"},
	[amDEagle] = {"Taxi driver", "punkman"},
	[amDynamite] = {loc("Pirate"), "pirate_jack"},
	[amFirePunch] = {loc("Fighter"), "StrawHatFacial"},
	[amWhip] = {loc("Adventurer"), "anzac"},
	[amBaseballBat] = {loc("Gangster"), "cap_blue"},
	[amParachute] = {loc("Infiltrator"), "Disguise"},
	[amAirAttack] = {loc("Colonel"), "war_trenchfrench01"},
	[amMineStrike] = {loc("Air General"), "war_desertofficer"},
	[amBlowTorch] = {loc("Miner"), "dwarf"},
	[amGirder] = {loc("Engineer"), "Glasses"},
	[amTeleport] = {loc("Wizard"), "WizardHat"},
	[amSwitch] = {"The trickster", "sf_balrog"},
	[amBazooka] = {loc("Soldier"), "war_americanww2helmet"},
	[amKamikaze] = {loc("Fanatic"), "Mummy"},
	[amCake] = {loc("Santa"), "Santa"},
	[amSeduction] = {loc("Siren"), "hair_red"},
	[amWatermelon] = {loc("Greengrocer"), "fr_banana"},
	[amHellishBomb] = {loc("Devil"), "Evil"},
	[amNapalm] = {loc("Commander"), "leprechaun"},
	[amDrill] = {loc("Oilman"), "flag_usa"},
	[amBallgun] = {loc("Joker"), "WhySoSerious"},
	[amRCPlane] = {loc("Pilot"), "tf_scout"},
	[amLowGravity] = {loc("Alien"), "mv_Venom"},
	[amExtraDamage] = {loc("Spartan"), "spartan"},
	[amInvulnerable] = {loc("Shaman"), "AkuAku"},
	[amExtraTime] = {loc("Turtle"), "zoo_turtle"},
	[amLaserSight] = {loc("Cyborg"), "cyborg2"},
	[amVampiric] = {loc("Vampire"), "vampirichog"},
	[amSniperRifle] = {loc("Sniper"), "hair_green"},
	[amJetpack] = {loc("Astronaut"), "Meteorhelmet"},
	[amMolotov] = {loc("Thug"), "thugclean"},
	[amBirdy] = {loc("Birdman"), "OldMan"},
	[amPortalGun] = {loc("Scientist"), "Einstein"},
	[amPiano] = {loc("Musician"), "fr_lemon"},
	[amGasBomb] = {loc("Infector"), "Gasmask"},
	[amSineGun] = {loc("Tesla"), "Moustache"},
	[amFlamethrower] = {loc("Pyromaniac"), "poke_charmander"},
	[amSMine] = {loc("Trapper"), "rasta"},
	[amHammer] = {"Chief Broom", "IndianChief"},
	[amResurrector] = {loc("Angel"), "angel"},
	[amDrillStrike] = {loc("King"), "crown"},
	[amSnowball] = {loc("Kid"), "cap_junior"},
	[amTardis] = {"Time traveler", "ShaggyYeti"},
	[amLandGun] = {loc("Gardener"), "bushhider"},
	[amIceGun] = {"Ice Lord", "snowhog"},
	[amKnife] = {loc("Chef"), "chef"},
	[amRubber] = {loc("Bouncer"), "bubble"},
	[amAirMine] = {loc("Coward"), "barrelhider"},
	[amMinigun] = {loc("Clone"), "scif_swStormtrooper"}
}

function onNewAmmoStore(groupIndex, hogIndex)
	if #weapons == 0 then
		for ammotype = 1, AmmoTypeMax do
			local count = GetAmmo(ammotype)
			if 0 < count and ammotype ~= amSkip then
				table.insert(weapons, ammotype)
				amount[ammotype] = count
			end
		end
	end

	if not hogammo[hogIndex] then
		hogammo[hogIndex] = {}
	end

	while #hogammo[hogIndex] < 3 do
		if #weapons == 0 then EndGame() end
		local random = 1 + GetRandom(#weapons)
		table.insert(hogammo[hogIndex], weapons[random])
		table.remove(weapons, random)
	end

	for i, ammo in ipairs(hogammo[hogIndex]) do
		SetAmmo(ammo, amount[ammo], 0, 0, 0)
	end

	SetAmmo(amSkip, 9, 0, 0, 0)
end

function onGameInit()
	DisableGameFlags(gfSharedAmmo, gfKing)
	EnableGameFlags(gfPerHogAmmo, gfResetWeps)

	HealthCaseProb = 100
end

function onGameStart()
	currTeam = ""
	lastTeam = ""
	z = 0
	mission = {}

	for i in ipairs(hhs) do

			currTeam = GetHogTeamName(hhs[i])

			if currTeam == lastTeam then
					z = z + 1
			else
					z = 0
			end

			if not hoghealth[z] then hoghealth[z] = 80 + GetRandom(13) * 10 end
			SetHealth(hhs[i], hoghealth[z])
			SetHogName(hhs[i], classes[hogammo[z][1]][1])
			SetHogHat(hhs[i], classes[hogammo[z][1]][2])

			mission[1 + z] = string.format("%s: %s, %s, %s", classes[hogammo[z][1]][1], GetAmmoName(hogammo[z][1]), GetAmmoName(hogammo[z][2]), GetAmmoName(hogammo[z][3]))

			lastTeam = GetHogTeamName(hhs[i])

	end

	ShowMission(loc("Random Specialists"), loc("minigame"), table.concat(mission, "|"), 0, 0)
end

function onNewTurn()
	AddCaption(string.format(loc("Prepare yourself, %s!"), GetHogName(CurrentHedgehog)))
end

function onGearAdd(gear)
	if GetGearType(gear) == gtHedgehog then
		table.insert(hhs, gear)
	elseif GetGearType(gear) == gtRCPlane then
		SetHealth(gear, 1)
	elseif GetGearType(gear) == gtAirAttack then
		SetHealth(gear, 1)
	elseif GetGearType(gear) == gtDrill and band(GetState(gear), gsttmpFlag) == 0 then
		SetGearValues(gear, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, 35)
	elseif GetGearType(gear) == gtCake then
		SetGearValues(gear, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, 35)
	end
end