--[[                  ___                   ___

                    (   )                 (   )

___ .-. .-. ___  ___ | |_    .---. ___ .-. | |_

(   )   '   (   )(   (   __) / .-, (   )   (   __)

|  .-.  .-. | |  | | | |   (__) ; ||  .-. .| |

| |  | |  | | |  | | | | ___ .'`  || |  | || | ___

| |  | |  | | |  | | | |(   / .'| || |  | || |(   )

| |  | |  | | |  | | | | | | /  | || |  | || | | |

| |  | |  | | |  ; ' | ' | ; |  ; || |  | || ' | |

| |  | |  | ' `-'  / ' `-' ' `-'  || |  | |' `-' ;

(___)(___)(___'.__.'   `.__.`.__.'_(___)(___)`.__.

----  Recommended settings:

----    * one hedgehog per team

----    * 'Small' one-island map

--]]

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

--[[

    MUTANT SCRIPT

    To Do:  -Clean-up this fucking piece of code

            -Debug

            -Find a girlfriend

            -Fix Sheepluva's hat  +[p]

            -Cookies

-----------------------]]

local hhs = {}

local numhhs = 0

local meh = false

local gameOver=false

local mutant = nil

local mutant_base_health = 200

local mutant_base_disease = 25

local disease_timer = 2000

local kill_reward = nil

local mt_hurt=false

local killsCounter = 0

local team_fire_punishment = 3

local mutant_kill_reward = 2

local hh_weapons = { amBazooka, amGrenade, amShotgun, amMine}

local mt_weapons = {amWatermelon, amHellishBomb, amBallgun, amRCPlane, amTeleport}

local disease=0

local timer=0

local winScore = 15

local hogsLimit = 1

local teams = {}

local circles = {}

local circleFrame = -1

-- Variables for custom achievements

-- Most kills in 1 turn

local recordKills = 0

local recordKillsHogName = nil

local recordKillsTeamName = nil

-- Most suicides

local recordSuicides = 0

local recordSuicidesHogName = nil

local recordSuicidesTeamName = nil

-- Most skips

local recordSkips = 0

local recordSkipsHogName = nil

local recordSkipsTeamName = nil

-- Most crates collected 

local recordCrates = 0

local recordCratesHogName = nil

local recordCratesTeamName = nil

-- Most deaths

local recordDeaths = 0

local recordDeathsHogName = nil

local recordDeathsTeamName = nil

-- Total killed hedgehogs

local totalKills = 0

-- Total damage

local totalDamage = 0

function rules()

	local ruleSet = loc("Rules: ") .. "|" ..

	loc("Hedgehogs will be revived after their death.") .. "|" ..

	string.format(loc("Mines explode after %d s."), div(MinesTime, 1000)) .. "|" ..

	" |"..

	loc("The first hedgehog to kill someone becomes the Mutant.") .. "|" ..

	loc("The Mutant has super-weapons and a lot of health.") .. "|" ..

	loc("The Mutant loses health quickly if he doesn't keep scoring kills.") .. "|" ..

	" |" ..

	loc("Score points by killing other hedgehogs (see below).") .. "|" ..

	loc("The hedgehog with least points (or most deaths) becomes the Bottom Feeder.") .. "|" ..

	string.format(loc("The first hedgehog which scores %d or more wins the game."), winScore) .. "|" ..

        " |" ..

	loc("Scoring: ") .. "|" ..

	loc("+2 for becoming the Mutant") .. "|" ..

	loc("+1 to the Mutant for killing anyone") .. "|" ..

	loc("+1 to the Bottom Feeder for killing anyone") .. "|" ..

	loc("-1 to anyone for a suicide") .. "|" ..

	loc("Other kills don't give you points.")

	return ruleSet

end

function showStartingInfo()

	ShowMission(loc("Mutant"), loc("A Hedgewars tag game"), rules(), 1, 5000)

end

function onGameInit()

    TurnTime = 20000

    WaterRise = 0

    HealthDecrease = 0

    EnableGameFlags(gfResetWeps, gfPerHogAmmo)

    HealthCaseProb=0

    HealthCaseAmount=0

    MinesTime=1000

    CaseFreq = 2

end

function limitHogs(gear)

    cnthhs = cnthhs + 1

        if cnthhs > 1 then

            hogLimitHit = true

            SetEffect(gear, heResurrectable, false)

            SetGearPosition(gear, -100,LAND_HEIGHT)

        end

end

function onGameStart()

    SendHealthStatsOff()

    trackTeams()

    teamScan()

    runOnHogs(saveStuff)

    hogLimitHit = false

    for i=0 , TeamsCount - 1 do

        cnthhs = 0

        runOnHogsInTeam(limitHogs, teams[i])

    end

    if hogLimitHit then

        AddCaption(loc("Only one hog per team allowed! Excess hogs will be removed"), 0xFFFFFFFF, capgrpGameState)

    end

    showStartingInfo()

end

function giveWeapons(gear)

    if gear == mutant then

        AddAmmo(gear, amRope)

        for i=1, #mt_weapons do

            AddAmmo(gear, mt_weapons[i])

        end

    else

        for i=1, #hh_weapons do

            AddAmmo(gear,hh_weapons[i])

        end

    end

end

function onAmmoStoreInit()

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

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

    SetAmmo(amSnowball,0,1,0,1)

    for i=1, #hh_weapons do

        SetAmmo(hh_weapons[i], 0, 0, 0, 1)

    end

    for i=1, #mt_weapons do

        SetAmmo(mt_weapons[i], 0, 3, 0, 1)

    end

end

function drawCircles()

    for i = 0, #hhs do

        if circles[hhs[i]] ~= nil then

            DeleteVisualGear(circles[hhs[i]])

            circles[hhs[i]] = nil

        end

        if hhs[i] ~= CurrentHedgehog then

            if mutant == nil then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 5, 0xff000080)

            elseif CurrentHedgehog == mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 3, 0xaa000070)

            elseif getGearValue(CurrentHedgehog, "Feeder") and hhs[i] ~= mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 3, 0xaa000070)

            elseif hhs[i] == mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 5, 0xff000080)

            end

        end

    end

    circleFrame = 0

end

function onNewTurn()

    trackTeams()

    killsCounter = 0

    if mutant == nil then

        AddCaption( loc("First killer will mutate"), 0xFFFFFFFF, capgrpGameState )

    end

    checkScore()

    for i=0, TeamsCount-1 do

        SendStat(siClanHealth, getTeamValue(teams[i], "Score"), teams[i])

    end

    giveWeapons(CurrentHedgehog)

    drawCircles()

    setAIHints()

    kill_reward= numhhs*10

    if CurrentHedgehog == mutant then

        mt_hurt=true

        disease= mutant_base_disease - numhhs

    else

        mt_hurt=false

    end

    setGearValue(CurrentHedgehog, "Alive", true)

end

function countBodies()

        if killsCounter == 2 then

            AddCaption(loc("Double kill!"), 0xFFFFFFFF, capgrpGameState )

        elseif killsCounter == 3 then

            AddCaption(loc("Mega kill!"), 0xFFFFFFFF, capgrpGameState )

            PlaySound(sndRegret)

        elseif killsCounter == 4 then

            AddCaption(loc("Ultra kill!"), 0xFFFFFFFF, capgrpGameState )

        elseif killsCounter == 5 then

            AddCaption(loc("Monster kill!"), 0xFFFFFFFF, capgrpGameState )

            PlaySound(sndIllGetYou)

        elseif killsCounter == 6 then

            AddCaption(loc("Ludicrous kill!"), 0xFFFFFFFF, capgrpGameState )

            PlaySound(sndNutter)

        elseif killsCounter == 7 then

            AddCaption(loc("Holy shit!"), 0xFFFFFFFF, capgrpGameState )

            PlaySound(sndLaugh)

        elseif killsCounter > 8 then

            AddCaption(loc("Insanity!"), 0xFFFFFFFF, capgrpGameState )

        end

        if killsCounter > recordKills then

            recordKills = killsCounter

            recordKillsHogName = getGearValue(CurrentHedgehog, "Name")

            recordKillsTeamName = GetHogTeamName(CurrentHedgehog)

        end

end

function onGameTick()

    if circleFrame > -1 then

        for i = 0, #hhs do

            if circles[hhs[i]] ~= nil and hhs[i]~= nil then

                hhx, hhy = GetGearPosition(hhs[i])

                X, Y, dX, dY, Angle, Frame, FrameTicks, State, Timer, Tint = GetVisualGearValues(circles[hhs[i]])

                SetVisualGearValues(circles[hhs[i]], hhx + 1, hhy - 3, 0, 0, 0, 0, 0, 40 - (circleFrame % 25), Timer, Tint)

            end

        end

        circleFrame = circleFrame + 0.06

        if circleFrame >= 25 then

            for i = 0, #hhs do

                if circles[hhs[i]] ~= nil then

                    DeleteVisualGear(circles[hhs[i]])

                    circles[hhs[i]] = nil

                end

            end

        end

    end

    if TurnTimeLeft==0 and mt_hurt then

        mt_hurt = false

    end

    if mt_hurt and mutant~=nil then

        timer = timer + 1

            if timer > disease_timer then

                timer = 0

                SetHealth(mutant, GetHealth(mutant)-disease )

                AddVisualGear(GetX(mutant), GetY(mutant)-5, vgtHealthTag, disease, true)

                    if GetHealth(mutant)<=0 then

                        SetHealth(mutant,0)

                        mt_hurt= false

                        setGearValue(mutant,"SelfDestruct",true)

                        TurnTimeLeft = 0

                    end

            end

    end

end

function saveStuff(gear)

    setGearValue(gear,"Name",GetHogName(gear))

    setGearValue(gear,"Hat",GetHogHat(gear))

end

function armageddon(gear)

    SetState(gear, gstLoser)

    SetEffect(gear, heResurrectable, false)

    SetHealth(gear, 0)

end

function updateScore()

    local showScore = ""

    for i=0, TeamsCount-1 do

        if teams[i]~= nil then

            local curr_score = getTeamValue(teams[i], "Score")

            showScore = showScore .. string.format(loc("%s: %d (deaths: %d)"), teams[i], curr_score, getTeamValue(teams[i], "DeadHogs")) .. "|"

        end

    end

    ShowMission(loc("Mutant"),

                loc("Scores"),

                showScore, 4, 1)

    HideMission()

end

function checkScore()

local showScore = ""

local lowest_score_team = nil

local min_score=nil

local winTeam = nil

local only_low_score = true

    for i=0, TeamsCount-1 do

        if teams[i]~=nil then

            local curr_score = getTeamValue(teams[i], "Score")

            runOnHogsInTeam(removeFeeder, teams[i])

            showScore = showScore .. string.format(loc("%s: %d (deaths: %d)"), teams[i], curr_score, getTeamValue(teams[i], "DeadHogs")) .. "|"

            if curr_score >= winScore then

                gameOver = true

                winTeam = teams[i]

            end

            if min_score==nil then

                min_score= curr_score

                lowest_score_team = teams[i]

            else

                if curr_score <= min_score then

                    if curr_score == min_score then

                        if getTeamValue(teams[i], "DeadHogs") == getTeamValue(lowest_score_team, "DeadHogs") then

                            only_low_score = false

                        else

                            if getTeamValue(teams[i], "DeadHogs") > getTeamValue(lowest_score_team, "DeadHogs") then

                                lowest_score_team = teams[i]

                            end

                            only_low_score = true

                        end

                    else

                        min_score= curr_score

                        lowest_score_team = teams[i]

                        only_low_score = true

                    end

                end

            end

        end

    end

    if gameOver then

        SendStat(siGraphTitle, loc("Score graph"))

        TurnTimeLeft = 0

        teamsSorted = {}

        for i=0, TeamsCount-1 do

            if teams[i]~=winTeam then

                runOnHogsInTeam(armageddon, teams[i])

            end

            teamsSorted[i+1] = teams[i]

        end

        -- Achievements stuff

        local achievements = 0

        --- Most kills per turn

        if recordKills >= 3 then

            SendStat(siMaxStepKills, string.format("%d %s (%s)", recordKills, recordKillsHogName, recordKillsTeamName))

            achievements = achievements + 1

        end

        --- Most crates collected

        if recordCrates >= 5 then

            SendStat(siCustomAchievement, string.format(loc("%s (%s) was the greediest hedgehog and collected %d crates."), recordCratesHogName, recordCratesTeamName, recordCrates))

            achievements = achievements + 1

        end

        --- Most suicides

        if recordSuicides >= 5 then

            SendStat(siCustomAchievement, string.format(loc("%s (%s) hate life and suicided %d times."), recordSuicidesHogName, recordSuicidesTeamName, recordSuicides))

            achievements = achievements + 1

        end

        --- Most deaths

        if recordDeaths >= 5 then

            SendStat(siCustomAchievement, string.format(loc("Poor %s (%s) died %d times."), recordDeathsHogName, recordDeathsTeamName, recordDeaths))

            achievements = achievements + 1

        end

        --- Most skips

        if recordSkips >= 3 then

            SendStat(siMaxTurnSkips, string.format("%d %s (%s)", recordSkips, recordSkipsHogName, recordSkipsTeamName))

            achievements = achievements + 1

        end

        --- Total damage 

        if totalDamage >= 900 then

            SendStat(siCustomAchievement, string.format(loc("%d damage was dealt in this game."), totalDamage))

            achievements = achievements + 1

        end

        --- Total kills

        if totalKills >= 20 or achievements <= 0 then

            SendStat(siKilledHHs, tostring(totalKills))

            achievements = achievements + 1

        end

        -- Score and stats stuff

        table.sort(teamsSorted, function(team1, team2) return getTeamValue(team1, "Score") < getTeamValue(team2, "Score") end)

        for i=TeamsCount, 1, -1 do

            SendStat(siPointType, loc("point(s)"))

            SendStat(siPlayerKills, getTeamValue(teamsSorted[i], "Score"), teamsSorted[i])

        end

        ShowMission(    loc("Mutant"),

                        loc("Final result"),

                        string.format(loc("Winner: %s"), winTeam) .. "| |" .. loc("Scores:") .. " |" ..

                        showScore, 0, 15000)

    else

    if only_low_score then

        runOnHogsInTeam(setFeeder, lowest_score_team)

    end

    if meh == false then

		meh = true

	else

		ShowMission(    loc("Mutant"),

                    loc("Scores"),

                    showScore, 4, 1)

		HideMission()

	end

    end

end

function backToNormal(gear)