require("PGBase")
require("GameManager/Constants")

---@class GUIManager
---@field TacticalGame TacticalGameClass|nil The local Tactical Game state.
---@field SkirmishGame SkirmishGameClass|nil The local Skirmish Game state.
---@field ShowTeamId integer The Team ID units to display.
GUIManager = {}
GUIManager.__index = GUIManager

---Creates a new GUI Manager.
---@return GUIManager
function GUIManager:New()
    local self = setmetatable({}, GUIManager)

    self.TacticalGame = nil
    self.SkirmishGame = nil
    self.ShowTeamId = 0

    return self
end

---Initializes the GUI Manager for Skirmish mode.
---@param game SkirmishGameClass The Skirmish Game state.
function GUIManager:InitSkirmish(game)
    self.SkirmishGame = game
    self.TacticalGame = game.TacticalGame

    GUI_Component_Visibility(UI_GameTime, true)

    if self.SkirmishGame.IsLocalSpectator then
        GUI_Component_Visibility(UI_CreditsT1, true)
        GUI_Component_Visibility(UI_IconT1, true)

        GUI_Component_Visibility(UI_CreditsT2, true)
        GUI_Component_Visibility(UI_IconT2, true)

        GUI_Component_Visibility(UI_PlanetaryPop, false)
        GUI_Component_Visibility(UI_CreditsTactical, false)
    end
end

---Initializes the GUI Manager for Tactical mode.
---@param game TacticalGameClass The Tactical Game state.
function GUIManager:InitTactical(game)
    self.TacticalGame = game

    GUI_Component_Visibility(UI_GameTime, true)
end

---Resets the GUI Manager.
function GUIManager:Reset()
    self.SkirmishGame = nil
    self.TacticalGame = nil

    GUI_Component_Text(UI_GameTime, "")
    GUI_Text_Color(UI_GameTime, 1, 1, 1, 1)
    GUI_Component_Visibility(UI_GameTime, false)

    GUI_Component_Text(UI_CreditsT1, "")
    GUI_Text_Color(UI_CreditsT1, 1, 1, 1, 1)
    GUI_Component_Visibility(UI_CreditsT1, false)

    GUI_Button_Icon(UI_IconT1, "", 1, 1, 1, 1)
    GUI_Component_Visibility(UI_IconT1, false)

    GUI_Component_Text(UI_CreditsT2, "")
    GUI_Text_Color(UI_CreditsT2, 1, 1, 1, 1)
    GUI_Component_Visibility(UI_CreditsT2, false)

    GUI_Button_Icon(UI_IconT2, "", 1, 1, 1, 1)
    GUI_Component_Visibility(UI_IconT2, false)

    GUI_Component_Visibility(UI_PlanetaryPop, true)
    GUI_Component_Visibility(UI_CreditsTactical, true)
end

---Services the GUI Manager.
function GUIManager:Service()
    if not self.TacticalGame then
        -- Nothing to service
        return
    end

    ScriptMessage("Servicing GUI Updates...")
    ServiceGameTime(self.TacticalGame)
    ServiceTeamCredits(self.SkirmishGame)
    --ServiceUnitDisplay(self.SkirmishGame, self.ShowTeamId)
end

---Services Game Time display.
---@param tacticalGame TacticalGameClass The current Tactical Game state
function ServiceGameTime(tacticalGame)
    if not tacticalGame then
        return
    end

    local gameTime = Dirty_Floor(GetCurrentTime()) - tacticalGame.StartTime
    local minutes = Dirty_Floor(gameTime / 60)
    local minutesText = tostring(minutes)

    if tonumber(minutes) < 10 then
        minutesText = "0" .. minutesText
    end

    local seconds = gameTime - (minutes * 60)
    local secondsText = tostring(seconds)

    if tonumber(seconds) < 10 then
        secondsText = "0" .. secondsText
    end

    local text = string.format("Time: %s:%s", minutesText, secondsText)

    GUI_Component_Text(UI_GameTime, text)
end

---Services Team Credits display.
---@param skirmishGame SkirmishGameClass The current Skirmish Game state.
function ServiceTeamCredits(skirmishGame)
    if not skirmishGame then
        return
    end

    if not skirmishGame.IsLocalSpectator then
        return
    end

    ---Displays credits for the specified team
    ---@param team TeamStruct The team to service.
    ---@param uiText string The name of the UI Component to display Team credits in.
    ---@param uiIcon string The name of the UI Component to display the Faction icon in.
    local function DisplayTeamCredits(team, uiText, uiIcon)
        local teamCredits = skirmishGame:CalculateTeamTotalCredits(team.Id)
        local teamIncome = skirmishGame:CalculateTeamTotalIncome(team.Id)
        local positiveText = "+"

        if teamIncome < 0 then
            positiveText = ""
        end

        local text = string.format("Team %d: $%d (%s%d)", team.Number, teamCredits, positiveText, teamIncome)
        local icon = team.Faction.Icon
        local color = team.Faction.Color

        GUI_Button_Icon(uiIcon, icon, color.R, color.G, color.B, 1)
        GUI_Component_Text(uiText, text)
        GUI_Text_Color(uiText, color.R, color.G, color.B, 1)
    end

    for _, team in pairs(skirmishGame.Teams) do
        if team.Number == 1 then
            DisplayTeamCredits(team, UI_CreditsT1, UI_IconT1)
        elseif team.Number == 2 then
            DisplayTeamCredits(team, UI_CreditsT2, UI_IconT2)
        end
    end
end

---Services Team Units display.
---@param skirmishGame SkirmishGameClass The current Skirmish Game state.
---@param teamId integer The Team ID's units to display.
function ServiceUnitDisplay(skirmishGame, teamId)
    if not skirmishGame then
        return
    end

    if not skirmishGame.IsLocalSpectator then
        return
    end

    local function DisplayTeamUnit(unit, uiComponent)

    end

    local localPlayer = Find_Player("local")
    local players = skirmishGame:GetPlayersOnTeam(teamId)
    ---@type table<string, integer> Units Table by name with alive count.
    local units = {}

    for _, player in pairs(players) do
        local playerUnits = player:GetAliveUnits()

        for _, playerUnit in pairs(playerUnits) do
            localPlayer.Select_Object(playerUnit.GameObjectWrapper)

            if not units[playerUnit.Name] then
                units[playerUnit.Name] = 1
            else
                units[playerUnit.Name] = units[playerUnit.Name] + 1
            end
        end
    end
end