--[[                  ___                   ___

                    (   )                 (   )

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

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

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

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

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

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

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

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

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

----  Recommended settings:

----    * one hedgehog per team (forced by game)

----    * one team per clan

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

--]]

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

--[[

    MUTANT SCRIPT

]]

local hhs = {}

local crates = {}

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 teamsDead = {}

local teamsDeleted = {}

local hogLimitHit = false

local cnthhs

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

local mutantHat = "WhySoSerious"

local feederHat = "poke_slowpoke"

function rules()

    local mineStr

    if MinesTime < 0 then

        mineStr = loc("Mines time: 0s-5s")

    else

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

    end

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

    mineStr .. "|" ..

    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, but gains health by killing.") .. "|" ..

    " |" ..

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

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

    loc("The score and deaths are shown next to the team bar.") .. "|" ..

    string.format(loc("Goal: Score %d points or more to win!"), 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")

    return ruleSet

end

function showStartingInfo()

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

end

function onGameInit()

    -- Sudden Death would be weird

    WaterRise = 0

    HealthDecrease = 0

    -- Weapons must be reset for the Mutant mechanic to work

    EnableGameFlags(gfResetWeps)

    -- King Mode messes with game too much

    DisableGameFlags(gfKing)

end

function limitHogsTeam(gear)

    cnthhs = cnthhs + 1

    if cnthhs > 1 then

        hogLimitHit = true

        SetEffect(gear, heResurrectable, 0)

        setGearValue(gear, "excess", true)

        DeleteGear(gear)

    end

end

function limitHogsClan(gear)

    hogLimitHit = true

    SetEffect(gear, heResurrectable, 0)

    setGearValue(gear, "excess", true)

    DeleteGear(gear)

end

function onGameStart()

    if ClansCount >= 2 then

        SendHealthStatsOff()

        SendAchievementsStatsOff()

    end

    SendRankingStatsOff()

    trackTeams()

    teamScan()

    runOnHogs(saveStuff)

    -- Enforce team and hog limits

    hogLimitHit = false

    -- Rule 1: One team per clan

    if TeamsCount > ClansCount then

        local usedClans = {}

        for i=0, TeamsCount - 1 do

            local teamName = GetTeamName(i)

            local clanNumber = GetTeamClan(teamName)

            if not usedClans[clanNumber] then

                usedClans[clanNumber] = true

            else

                runOnHogsInTeam(limitHogsClan, teamName)

                teamsDeleted[teamName] = true

                setTeamValue(teamName, "Score", getTeamValue(teamName, "Score") -99999)

            end

        end

    end

    -- Rule 2: One hog per team

    for i=0 , TeamsCount - 1 do

        cnthhs = 0

        runOnHogsInTeam(limitHogsTeam, GetTeamName(i))

    end

    if hogLimitHit then

        -- TODO: Update warning message to include excess teams as well

        WriteLnToChat(loc("Only one hog per team allowed! Excess hogs will be removed."))

    end

    trackTeams()

    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()

    killsCounter = 0

    if mutant == nil and TotalRounds >= 0 then

        AddCaption( loc("First killer will mutate"), capcolDefault, capgrpGameState )

    end

    checkScore()

    for i=0, TeamsCount-1 do

        local teamName = GetTeamName(i)

        if not teamsDeleted[teamName] then

            SendStat(siClanHealth, getTeamValue(teamName, "Score"), teamName)

        end

    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!"), capcolDefault, capgrpGameState )

        elseif killsCounter == 3 then

            AddCaption(loc("Mega kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndRegret)

        elseif killsCounter == 4 then

            AddCaption(loc("Ultra kill!"), capcolDefault, capgrpGameState )

        elseif killsCounter == 5 then

            AddCaption(loc("Monster kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndIllGetYou)

        elseif killsCounter == 6 then

            AddCaption(loc("Ludicrous kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndNutter)

        elseif killsCounter == 7 then

            AddCaption(loc("Holy shit!"), capcolDefault, capgrpGameState )

            PlaySound(sndLaugh)

        elseif killsCounter > 8 then

            AddCaption(loc("Insanity!"), capcolDefault, 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])

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

            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 or band(GetState(mutant), gstHHDriven) == 0) and mt_hurt then

        mt_hurt = false

    end

    -- Mutant's disease

    -- Hurt Mutant during its turn time

    -- Mutant's health is safe in ready phase

    if mt_hurt and mutant~=nil and ReadyTimeLeft == 0 then

        timer = timer + 1

        if timer > disease_timer then

            timer = 0

            local h = GetHealth(mutant)-disease

            SetHealth(mutant, h)

            -- Low health warning

            if h <= 75 then

                PlaySound(sndPoisonMoan, mutant)

            elseif h <= 150 then

                PlaySound(sndPoisonCough, mutant)

            end

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

            SetVisualGearValues(tag, nil, nil, nil, nil, nil, nil, nil, nil, nil, GetClanColor(GetHogClan(mutant)))

            if GetHealth(mutant)<=0 then

                SetHealth(mutant,0)

                mt_hurt= false

                setGearValue(mutant,"SelfDestruct",true)

                EndTurn()

            end

        end

    end

end

--[[

Forces the special mutant/feeder names and hats only to be

taken by those who deserved it.

Names and hats will be changed (and ridiculed) if neccesary.

]]

function exposeIdentityTheft(gear)

    local lon = string.lower(GetHogName(gear)) -- lowercase origina name

    local name, hat

    -- Change name if hog uses a reserved one

    if lon == "mutant" or lon == string.lower(loc("Mutant")) then

       SetHogName(gear, loc("Identity Thief"))

       SetHogHat(gear, "Disguise")

    elseif lon == "bottom feeder" or lon == string.lower(loc("Bottom Feeder")) then

       -- Word play on "Bottom Feeder". Someone who is low on cotton. :D

       -- Either translate literally or make up your ow word play

       SetHogName(gear, loc("Cotton Needer"))

       SetHogHat(gear, "StrawHat")

    end

    -- Strip hog off its special hat

    if GetHogHat(gear) == mutantHat or GetHogHat(gear) == feederHat then

       SetHogHat(gear, "NoHat")

    end

end

function saveStuff(gear)

    exposeIdentityTheft(gear)

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

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

end

function armageddon(gear)

    SetState(gear, gstLoser)

    SetEffect(gear, heResurrectable, 0)

    SetHealth(gear, 0)

end

function renderScores()

    for i=0, TeamsCount-1 do

        local name = GetTeamName(i)

        SetTeamLabel(name, string.format(loc("%d | %d"), getTeamValue(name, "Score"), getTeamValue(name, "DeadHogs")))

    end

end

function createEndGameStats()

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

    local teamsSorted = {}

    for i=0, TeamsCount-1, 1 do

        teamsSorted[i+1] = GetTeamName(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 and totalKills >= 1) then

        SendStat(siKilledHHs, tostring(totalKills))

        achievements = achievements + 1

    end

    -- Score and stats stuff

    local showScore = ""

    local rank = 0

    local rankPlus = 1

    local prevScore

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

    for i=1, TeamsCount do

        local score = getTeamValue(teamsSorted[i], "Score")

        local deaths = getTeamValue(teamsSorted[i], "DeadHogs")

        if i == 1 or score < prevScore then

            rank = rank + rankPlus

            rankPlus = 1

            prevScore = score

        else

            rankPlus = rankPlus + 1

        end

        SendStat(siPointType, "!POINTS")

        SendStat(siTeamRank, rank)

        SendStat(siPlayerKills, score, teamsSorted[i])