------------------- ABOUT ----------------------

--

-- In this adventure hero gets the lost part with

-- the help of the green bananas hogs.

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Animate.lua")

HedgewarsScriptLoad("/Missions/Campaign/A_Space_Adventure/global_functions.lua")

----------------- VARIABLES --------------------

-- globals

local missionName = loc("Getting to the device")

local inBattle = false

local tookPartInBattle = false

local previousHog = -1

local permitCaptainLimeDeath = false

-- dialogs

local dialog01 = {}

local dialog02 = {}

local dialog03 = {}

local dialog04 = {}

local dialog05 = {}

-- mission objectives

local minesTimeText = loc("Mines time: 0 seconds")

local goals

-- crates

local girderCrate = {name = amGirder, x = 1680, y = 1160}

local eagleCrate = {name = amDEagle, x = 1680, y = 1650}

local weaponCrate = { x = 1320, y = 1870}

local deviceCrate = { gear = nil, x = 1360, y = 1870}

local ropeCrate = {name = amRope, x = 1400, y = 1870}

-- hogs

local hero = {}

local green1 = {}

local green2 = {}

local green3 = {}

-- teams

local teamA = {}

local teamB = {}

local teamC = {}

-- hedgehogs values

hero.name = loc("Hog Solo")

hero.x = 1200

hero.y = 820

hero.dead = false

green1.name = loc("Captain Lime")

green1.hat = "war_desertofficer"

green1.x = 1050

green1.y = 820

green1.dead = false

green2.name = loc("Mister Pear")

green2.hat = "war_britmedic"

green2.x = 1350

green2.y = 820

green3.name = loc("Lady Mango")

green3.hat = "hair_red"

green3.x = 1450

green3.y = 820

local redHedgehogs = {

	{ name = loc("Poisonous Apple") },

	{ name = loc("Dark Strawberry") },

	{ name = loc("Watermelon Heart") },

	{ name = loc("Deadly Grape") }

}

-- Hog Solo and Green Bananas

teamA.name = loc("Hog Solo and GB")

teamA.color = -6

-- Captain Lime will use a color which is almost the same as the color of teamA.

-- It works, but it's a hack.

-- Technically, this makes Captain Lime an enemy team in the mission but for the player

-- it looks like an ally. This is because Camptain Lime starts friendly (story-wise),

-- but might become your enemy during the course of this mission.

-- TODO: For teamB, use same color of teamA when friendly, change color when evil

teamB.name = loc("Captain Lime")

teamC.name = loc("Fruit Assassins")

teamC.color = -1

function onGameInit()

	GameFlags = gfDisableWind

	Seed = 1

	TurnTime = 20000

	CaseFreq = 0

	MinesNum = 0

	MinesTime = 1

	Explosives = 0

	-- Disable Sudden Death

	HealthDecrease = 0

	WaterRise = 0

	Map = "fruit02_map"

	Theme = "Fruit"

	local health = 100

	-- Fruit Assassins

	local assasinsHats = { "NinjaFull", "NinjaStraight", "NinjaTriangle" }

	AddTeam(teamC.name, teamC.color, "bp2", "Island", "Default", "cm_scout")

	for i=1,table.getn(redHedgehogs) do

		redHedgehogs[i].gear =  AddHog(redHedgehogs[i].name, 1, 100, assasinsHats[GetRandom(3)+1])

		SetGearPosition(redHedgehogs[i].gear, 2010 + 50*i, 630)

	end

	local assassinsColor = div(GetClanColor(GetHogClan(redHedgehogs[1].gear)), 0x100)

	-- Hero and Green Bananas

	teamA.name = AddMissionTeam(teamA.color)

	hero.gear = AddMissionHog(health)

	hero.name = GetHogName(hero.gear)

	SetHogTeamName(hero.gear, string.format(loc("%s and GB"), teamA.name))

	teamA.name = GetHogTeamName(hero.gear)

	SetGearPosition(hero.gear, hero.x, hero.y)

	HogTurnLeft(hero.gear, true)

	local heroColor = div(GetClanColor(GetHogClan(hero.gear)), 0x100)

	-- companions

	-- Change companion identity if they have same name as hero

	-- to avoid confusion.

	if green2.name == hero.name then

		green2.name = loc("Green Hog Grape")

		green2.hat = "war_desertsapper1"

	elseif green3.name == hero.name then

		green3.name = loc("Green Hog Grape")

		green3.hat = "war_desertsapper1"

	end

	green2.gear = AddHog(green2.name, 0, 100, green2.hat)

	SetGearPosition(green2.gear, green2.x, green2.y)

	HogTurnLeft(green2.gear, true)

	green3.gear = AddHog(green3.name, 0, 100, green3.hat)

	SetGearPosition(green3.gear, green3.x, green3.y)

	HogTurnLeft(green3.gear, true)

	-- Captain Lime

	-- Returns a color that is as close as possible to the color argument

	-- but does not equal color and otherCollision.

	local getSimilarColor = function(color, otherCollision)

		local goingUp = false

		local collision1 = color

		while(color == collision1 or color == otherCollision) do

			-- Try out colors by varying the blue color component until

			-- we don't collide with any of the 2 colors.

			if ((color % 0x100) > 0) and (not goingUp) then

				color = color - 0x1

			else

				goingUp = true

				color = color + 0x1

			end

		end

		return color

	end

	-- Captain Lime gets a fake color clore to hero's clan color.

	-- This is a hack, but it works. See explanation at top of file

	local cptnColor = getSimilarColor(heroColor, assassinsColor)

	AddTeam(teamB.name, cptnColor, "Cherry", "Island", "Default", "congo-brazzaville")

	green1.gear= AddHog(green1.name, 0, 100, green1.hat)

	SetGearPosition(green1.gear, green1.x, green1.y)

	-- Populate goals table

	goals = {

		[dialog01] = {missionName, loc("Exploring the tunnel"), loc("Search for the device with the help of the other hedgehogs.").."|"..string.format(loc("%s must collect the final crates."), hero.name) .. "|" .. minesTimeText, 1, 4000},

		[dialog02] = {missionName, loc("Exploring the tunnel"), loc("Explore the tunnel with the other hedgehogs and search for the device.").."|"..string.format(loc("%s must collect the final crates."), hero.name) .. "|" .. minesTimeText, 1, 4000},

		[dialog03] = {missionName, loc("Return to the Surface"), loc("Go to the surface!").."|"..loc("Attack Captain Lime before he attacks back.").."|"..minesTimeText, 1, 4000},

		[dialog04] = {missionName, loc("Return to the Surface"), loc("Go to the surface!").."|"..loc("Attack the assassins before they attack back.").."|"..minesTimeText, 1, 4000},

	}

	AnimInit(true)

	AnimationSetup()

end

function onGameStart()

	AnimWait(hero.gear, 3000)

	FollowGear(hero.gear)

	if GetCampaignVar("Fruit01JoinedBattle") and GetCampaignVar("Fruit01JoinedBattle") == "true" then

		tookPartInBattle = true

	end

	AddEvent(onHeroDeath, {hero.gear}, heroDeath, {hero.gear}, 0)

	AddEvent(onDeviceCrates, {hero.gear}, deviceCrateEvent, {hero.gear}, 0)

	-- Hero and Green Bananas weapons

	AddAmmo(hero.gear, amSwitch, 100)

	-- Assassins weapons

	AddAmmo(redHedgehogs[1].gear, amBazooka, 6)

	AddAmmo(redHedgehogs[1].gear, amGrenade, 6)

	AddAmmo(redHedgehogs[1].bot, amDEagle, 6)

	for i=1,table.getn(redHedgehogs) do

		HideHog(redHedgehogs[i].gear)

	end

	-- explosives

	-- I wanted to use FindPlace but doesn't accept height values...

	local x1 = 950

	local x2 = 1306

	local y1 = 1210

	local y2 = 1620

	while true do

		if y2<y1 then

			break

		end

		if x2<x1 then

			x2 = 1305

			y2 = y2 - 50

		end

		if not TestRectForObstacle(x2+25, y2+25, x2-25, y2-25, true) then

			AddGear(x2, y2, gtExplosives, 0, 0, 0, 0)

		end

		x2 = x2 - 25

	end

	AddGear(3128, 1680, gtExplosives, 0, 0, 0, 0)

	--mines

	AddGear(3135, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3145, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3155, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3165, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3175, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3115, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3105, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3095, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3085, 1680, gtMine, 0, 0, 0, 0)

	AddGear(3075, 1680, gtMine, 0, 0, 0, 0)

	AddAmmo(hero.gear, amFirePunch, 3)

	if tookPartInBattle then

		AddAnim(dialog01)

	else

		AddAnim(dialog02)

	end

	-- place crates

	SpawnSupplyCrate(girderCrate.x, girderCrate.y, girderCrate.name)

	SpawnSupplyCrate(eagleCrate.x, eagleCrate.y, eagleCrate.name)

	deviceCrate.gear = SpawnFakeUtilityCrate(deviceCrate.x, deviceCrate.y, false, false) -- anti-gravity device

	-- Rope crate is placed after device crate has been collected.

	-- This is done so it is impossible the player can rope before getting

	-- the device part.

	if tookPartInBattle then

		SpawnSupplyCrate(weaponCrate.x, weaponCrate.y, amWatermelon)

	else

		SpawnSupplyCrate(weaponCrate.x, weaponCrate.y, amSniperRifle)

	end

	SendHealthStatsOff()

end

function onNewTurn()

	if not inBattle and CurrentHedgehog == green1.gear then

		EndTurn(true)

	elseif (not inBattle) and GetHogTeamName(CurrentHedgehog) == teamA.name then

		if CurrentHedgehog ~= hero.gear then

			SwitchHog(hero.gear)

		end

		SetTurnTimeLeft(MAX_TURN_TIME)

		wind()

	elseif inBattle then

		if CurrentHedgehog == green1.gear and previousHog ~= hero.gear then

			EndTurn(true)

			return

		end

		for i=1,table.getn(redHedgehogs) do

			if CurrentHedgehog == redHedgehogs[i].gear and previousHog ~= hero.gear then

				EndTurn(true)

				return

			end

		end

		SetTurnTimeLeft(20000)

		wind()

	else

		EndTurn(true)

	end

	previousHog = CurrentHedgehog

end

function onGameTick()

	AnimUnWait()

	if ShowAnimation() == false then

		return

	end

	ExecuteAfterAnimations()

	CheckEvents()

end

function onGameTick20()

	if not permitCaptainLimeDeath and not GetHealth(green1.gear) then

		-- game ends with the according stat messages

		heroDeath()

		permitCaptainLimeDeath = true

	end

	if CurrentHedgehog and GetY(CurrentHedgehog) > 1350 then

		SetWind(-40)

	end

end

function onGearDelete(gear)

	if gear == hero.gear then

		hero.dead = true

	elseif gear == green1.gear then

		green1.dead = true

	elseif gear == deviceCrate.gear then

		if band(GetGearMessage(gear), gmDestroy) ~= 0 then

			PlaySound(sndShotgunReload)

			AddCaption(loc("Anti-Gravity Device Part (+1)"), GetClanColor(GetHogClan(CurrentHedgehog)), capgrpAmmostate)

			deviceCrate.collected = true

			deviceCrate.collector = CurrentHedgehog

			-- Spawn rope crate

			SpawnSupplyCrate(ropeCrate.x, ropeCrate.y, ropeCrate.name)

		end

	end

end

function onGearDamage(gear, damage)

	if GetGearType(gear) == gtCase then

		-- in this mode every crate is essential in order to complete the mission

		-- destroying a crate ends the game

		heroDeath()

	end

end

function onAmmoStoreInit()

	SetAmmo(amDEagle, 0, 0, 0, 6)

	SetAmmo(amGirder, 0, 0, 0, 2)

	SetAmmo(amRope, 0, 0, 0, 1)

	SetAmmo(amSkip, 9, 0, 0, 1)

	if tonumber(getBonus(2)) == 1 then

		SetAmmo(amWatermelon, 0, 0, 0, 2)

		SetAmmo(amSniperRifle, 0, 0, 0, 2)

	else

		SetAmmo(amWatermelon, 0, 0, 0, 1)

		SetAmmo(amSniperRifle, 0, 0, 0, 1)

	end

end

function onPrecise()

	if GameTime > 3000 then

		SetAnimSkip(true)

	end

end

-------------- EVENTS ------------------

function onHeroDeath(gear)

	if hero.dead then

		return true

	end

	return false

end

function onDeviceCrates(gear)

	if not hero.dead and deviceCrate.collected and StoppedGear(hero.gear) then

		return true

	end

	return false

end

function onSurface(gear)

	if not hero.dead and GetY(hero.gear)<850 and  StoppedGear(hero.gear) then

		return true

	end

	return false

end

function onGaptainLimeDeath(gear)

	if green1.dead then

		return true

	end

	return false

end

function onRedTeamDeath(gear)

	local redDead = true

	for i=1,table.getn(redHedgehogs) do

		if GetHealth(redHedgehogs[i].gear) then

			redDead = false

			break

		end

	end

	return redDead

end

-------------- ACTIONS ------------------

ended = false

function heroDeath(gear)

	if not ended then

		SendStat(siGameResult, string.format(loc("%s lost, try again!"), hero.name))

		SendStat(siCustomAchievement, string.format(loc("To win the game, %s has to get the bottom crates and come back to the surface."), hero.name))

		SendStat(siCustomAchievement, loc("You can use the other 2 hogs to assist you."))

		SendStat(siCustomAchievement, loc("Do not destroy the crates!"))

		if tookPartInBattle then

			if permitCaptainLimeDeath then

				SendStat(siCustomAchievement, string.format(loc("You'll have to eliminate %s at the end."), teamC.name))

				sendSimpleTeamRankings({teamC.name, teamA.name})

			else

				sendSimpleTeamRankings({teamA.name})

			end

		else

			if permitCaptainLimeDeath then

				SendStat(siCustomAchievement, loc("You'll have to eliminate Captain Lime at the end."))

				sendSimpleTeamRankings({teamB.name, teamA.name})

			else

				SendStat(siCustomAchievement, loc("Don't eliminate Captain Lime before collecting the last crate!"))

				sendSimpleTeamRankings({teamA.name})

			end

		end

		EndGame()

		ended = true

	end

end

-- Device crate got taken

function deviceCrateEvent(gear)

	-- Stop hedgehog

	SetGearMessage(deviceCrate.collector, 0)

	if deviceCrate.collector == hero.gear then

		-- Hero collected the device crate

		if not tookPartInBattle then

			-- Captain Lime turns evil

			AddAnim(dialog03)

		else

			-- Fruit Assassins attack

			for i=1,table.getn(redHedgehogs) do

				RestoreHog(redHedgehogs[i].gear)

			end

			AddAnim(dialog04)

		end

		-- needs to be set to true for both plots

		permitCaptainLimeDeath = true

		AddAmmo(hero.gear, amSwitch, 0)

		AddEvent(onSurface, {hero.gear}, surface, {hero.gear}, 0)

	else

		-- Player let the Green Bananas collect the crate.

		-- How dumb!

		-- Player will lose for this.

		AnimationSetup05(deviceCrate.collector)

		AddAnim(dialog05)

	end

end

function surface(gear)

	previousHog = -1

	if tookPartInBattle then

		escapeHog(green1.gear)

		AddEvent(onRedTeamDeath, {green1.gear}, redTeamDeath, {green1.gear}, 0)

	else

		SetHogLevel(green1.gear, 1)

		-- Equip Captain Lime with weapons

		AddAmmo(green1.gear, amBazooka, 6)

		AddAmmo(green1.gear, amGrenade, 6)

		AddAmmo(green1.gear, amDEagle, 2)

		AddEvent(onGaptainLimeDeath, {green1.gear}, captainLimeDeath, {green1.gear}, 0)

	end

	EndTurn(true)

	escapeHog(green2.gear)

	escapeHog(green3.gear)

	inBattle = true

end

function captainLimeDeath(gear)

	-- hero win in scenario of escape in 1st part

	saveCompletedStatus(3)

	SendStat(siGameResult, loc("Congratulations, you won!"))

	SendStat(siCustomAchievement, loc("You retrieved the lost part."))

	SendStat(siCustomAchievement, loc("You defended yourself against Captain Lime."))

	sendSimpleTeamRankings({teamA.name, teamB.name})

	EndGame()

end

function redTeamDeath(gear)

	-- hero win in battle scenario

	saveCompletedStatus(3)

	SendStat(siGameResult, loc("Congratulations, you won!"))

	SendStat(siCustomAchievement, loc("You retrieved the lost part."))

	SendStat(siCustomAchievement, string.format(loc("You defended yourself against %s."), teamC.name))

	sendSimpleTeamRankings({teamA.name, teamC.name})

	EndGame()

end

-------------- ANIMATIONS ------------------

function Skipanim(anim)

	if goals[anim] ~= nil then

		ShowMission(unpack(goals[anim]))

	end

	if anim == dialog03 then

		makeCptLimeEvil()

	elseif anim == dialog05 then

		heroIsAStupidFool()

	else