CypherCore/Source/Game/Text/ChatTextBuilder.cs

/*
 * Copyright (C) 2012-2020 CypherCore <http://github.com/CypherCore>
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

using Framework.Constants;
using Framework.Dynamic;
using Game.DataStorage;
using Game.Entities;
using Game.Maps;
using Game.Networking.Packets;
using System.Collections.Generic;

namespace Game.Chat
{
    public class MessageBuilder
    {
        public virtual dynamic Invoke(Locale locale = Locale.enUS) { return default; }
    }

    public class ChatPacketSender : IDoWork<Player>
    {
        ChatMsg Type;
        Language Language;
        WorldObject Sender;
        WorldObject Receiver;
        string Text;
        uint AchievementId;
        Locale Locale;

        // caches
        public ChatPkt UntranslatedPacket;
        public Optional<ChatPkt> TranslatedPacket;

        public ChatPacketSender(ChatMsg chatType, Language language, WorldObject sender, WorldObject receiver, string message, uint achievementId = 0, Locale locale = Locale.enUS)
        {
            Type = chatType;
            Language = language;
            Sender = sender;
            Receiver = receiver;
            Text = message;
            AchievementId = achievementId;
            Locale = locale;

            UntranslatedPacket.Initialize(Type, Language, Sender, Receiver, Text, AchievementId, "", Locale);
            UntranslatedPacket.Write();
        }

        public void Invoke(Player player)
        {
            if (Language == Language.Universal || Language == Language.Addon || Language == Language.AddonLogged || player.CanUnderstandLanguage(Language))
            {
                player.SendPacket(UntranslatedPacket);
                return;
            }

            if (!TranslatedPacket.HasValue)
            {
                TranslatedPacket.HasValue = true;
                TranslatedPacket.Value.Initialize(Type, Language, Sender, Receiver, Global.LanguageMgr.Translate(Text, (uint)Language, player.GetSession().GetSessionDbcLocale()), AchievementId, "", Locale);
                TranslatedPacket.Value.Write();
            }

            player.SendPacket(TranslatedPacket.Value);
        }
    }
    
    public class BroadcastTextBuilder : MessageBuilder
    {
        public BroadcastTextBuilder(WorldObject obj, ChatMsg msgtype, uint textId, Gender gender, WorldObject target = null, uint achievementId = 0)
        {
            _source = obj;
            _msgType = msgtype;
            _textId = textId;
            _gender = gender;
            _target = target;
            _achievementId = achievementId;
        }

        public override ChatPacketSender Invoke(Locale locale = Locale.enUS)
        {
            BroadcastTextRecord bct = CliDB.BroadcastTextStorage.LookupByKey(_textId);
            return new ChatPacketSender(_msgType, bct != null ? (Language)bct.LanguageID : Language.Universal, _source, _target, bct != null ? Global.DB2Mgr.GetBroadcastTextValue(bct, locale, _gender) : "", _achievementId, locale);
        }

        WorldObject _source;
        ChatMsg _msgType;
        uint _textId;
        Gender _gender;
        WorldObject _target;
        uint _achievementId;
    }

    public class CustomChatTextBuilder : MessageBuilder
    {
        public CustomChatTextBuilder(WorldObject obj, ChatMsg msgType, string text, Language language = Language.Universal, WorldObject target = null)
        {
            _source = obj;
            _msgType = msgType;
            _text = text;
            _language = language;
            _target = target;
        }

        public override ChatPacketSender Invoke(Locale locale)
        {
            return new ChatPacketSender(_msgType, _language, _source, _target, _text, 0, locale);
        }

        WorldObject _source;
        ChatMsg _msgType;
        string _text;
        Language _language;
        WorldObject _target;
    }

    class CypherStringChatBuilder : MessageBuilder
    {
        public CypherStringChatBuilder(WorldObject obj, ChatMsg msgType, CypherStrings textId, WorldObject target = null, object[] args = null)
        {
            _source = obj;
            _msgType = msgType;
            _textId = textId;
            _target = target;
            _args = args;
        }

        public override ChatPacketSender Invoke(Locale locale)
        {
            string text = Global.ObjectMgr.GetCypherString(_textId, locale);

            if (_args != null)
                return new ChatPacketSender(_msgType, Language.Universal, _source, _target, string.Format(text, _args), 0, locale);
            else
                return new ChatPacketSender(_msgType, Language.Universal, _source, _target, text, 0, locale);
        }

        WorldObject _source;
        ChatMsg _msgType;
        CypherStrings _textId;
        WorldObject _target;
        object[] _args;
    }
}