AIChatSorian.lua

#****************************************************************************
#**
#**  File     :  /lua/modules/AIChatSorian.lua
#**  Author(s): Mike Robbins aka Sorian
#**
#**  Summary  : AI Chat Functions
#**  Version  : 0.1
#****************************************************************************
local Chat = import('/lua/ui/game/chat.lua')
local ChatTo = import('/lua/lazyvar.lua').Create()

function AIChat(group, text, sender)
    if text then
        if import('/lua/ui/game/taunt.lua').CheckForAndHandleTaunt(text, sender) then
            return
        end
        ChatTo:Set(group)
        msg = { to = ChatTo(), Chat = true }
        msg.text = text
        msg.aisender = sender
        local armynumber = GetArmyData(sender)
        if ChatTo() == 'allies' then
            AISendChatMessage(FindAllies(armynumber), msg)
        elseif ChatTo() == 'enemies' then
            AISendChatMessage(FindEnemies(armynumber), msg)
        elseif type(ChatTo()) == 'number' then
            AISendChatMessage({ChatTo()}, msg)
        else
            AISendChatMessage(nil, msg)
        end
    end
end

function FindAllies(army)
    local t = GetArmiesTable()
    local result = {}
    for k,v in t.armiesTable do
        if IsAlly(k, army) and v.human then
            table.insert(result, k)
        end
    end
    return result
end

function FindEnemies(army)
    local t = GetArmiesTable()
    local result = {}
    for k,v in t.armiesTable do
        if IsEnemy(k, army) and v.human then
            table.insert(result, k)
        end
    end
    return result
end

function AISendChatMessage(towho, msg)
    local t = GetArmiesTable()
    local focus = t.focusArmy
    if msg.Chat then
        if towho then
            for k,v in towho do
                if v == focus then
                    import('/lua/ui/game/chat.lua').ReceiveChat(msg.aisender, msg)
                end
            end
        else
            import('/lua/ui/game/chat.lua').ReceiveChat(msg.aisender, msg)
        end
    elseif msg.Taunt then
        import('/lua/ui/game/taunt.lua').RecieveAITaunt(msg.aisender, msg)
    end
end

function GetArmyData(army)
    local armies = GetArmiesTable()
    local result
    if type(army) == 'string' then
        for i, v in armies.armiesTable do
            if v.nickname == army then
                result = i
                break
            end
        end
    end
    return result
end