// Copyright (c) CypherCore <http://github.com/CypherCore> All rights reserved.
// Licensed under the GNU GENERAL PUBLIC LICENSE. See LICENSE file in the project root for full license information.

using Framework.Constants;
using Framework.Database;
using System.Collections.Generic;
using System.Linq;

namespace Game.DataStorage
{
    public class ConversationDataStorage : Singleton<ConversationDataStorage>
    {
        ConversationDataStorage() { }

        public void LoadConversationTemplates()
        {
            _conversationLineTemplateStorage.Clear();
            _conversationTemplateStorage.Clear();

            Dictionary<uint, List<ConversationActorTemplate>> actorsByConversation = new();

            SQLResult lineTemplates = DB.World.Query("SELECT Id, UiCameraID, ActorIdx, Flags, ChatType FROM conversation_line_template");
            if (!lineTemplates.IsEmpty())
            {
                uint oldMSTime = Time.GetMSTime();

                do
                {
                    uint id = lineTemplates.Read<uint>(0);

                    if (!CliDB.ConversationLineStorage.ContainsKey(id))
                    {
                        Log.outError(LogFilter.Sql, "Table `conversation_line_template` has template for non existing ConversationLine (ID: {0}), skipped", id);
                        continue;
                    }

                    ConversationLineTemplate conversationLine = new();
                    conversationLine.Id = id;
                    conversationLine.UiCameraID = lineTemplates.Read<uint>(1);
                    conversationLine.ActorIdx = lineTemplates.Read<byte>(2);
                    conversationLine.Flags = lineTemplates.Read<byte>(3);
                    conversationLine.ChatType = lineTemplates.Read<byte>(4);

                    _conversationLineTemplateStorage[id] = conversationLine;
                }
                while (lineTemplates.NextRow());

                Log.outInfo(LogFilter.ServerLoading, "Loaded {0} Conversation line templates in {1} ms", _conversationLineTemplateStorage.Count, Time.GetMSTimeDiffToNow(oldMSTime));
            }
            else
            {
                Log.outInfo(LogFilter.ServerLoading, "Loaded 0 Conversation line templates. DB table `conversation_line_template` is empty.");
            }

            SQLResult actorResult = DB.World.Query("SELECT ConversationId, ConversationActorId, ConversationActorGuid, Idx, CreatureId, CreatureDisplayInfoId, NoActorObject, ActivePlayerObject FROM conversation_actors");
            if (!actorResult.IsEmpty())
            {
                uint oldMSTime = Time.GetMSTime();
                uint count = 0;

                do
                {
                    ConversationActorDbRow data;
                    ConversationActorTemplate actor = new();

                    data.ConversationId = actorResult.Read<uint>(0);
                    data.ConversationId = actorResult.Read<uint>(1);
                    data.SpawnId = actorResult.Read<ulong>(2);
                    data.ActorIndex = actor.Index = actorResult.Read<ushort>(3);
                    data.CreatureId = actorResult.Read<uint>(4);
                    data.CreatureDisplayInfoId = actorResult.Read<uint>(5);
                    bool noActorObject = actorResult.Read<byte>(6) == 1;
                    bool activePlayerObject = actorResult.Read<byte>(7) == 1;

                    if (activePlayerObject)
                        actor.ActivePlayerTemplate = new();
                    else if (noActorObject)
                        actor.NoObjectTemplate = new();
                    else if (data.SpawnId != 0 || data.CreatureId == 0) // @TODO: remove CreatureId check when actor flags are implemented
                        actor.WorldObjectTemplate = new();
                    else
                        actor.TalkingHeadTemplate = new();

                    bool valid = data.Invoke(actor);
                    if (!valid)
                        continue;

                    if (!actorsByConversation.ContainsKey(data.ConversationId))
                        actorsByConversation[data.ConversationId] = new();

                    actorsByConversation[data.ConversationId].Add(actor);
                    ++count;
                } while (actorResult.NextRow());

                Log.outInfo(LogFilter.ServerLoading, "Loaded {0} Conversation actors in {1} ms", count, Time.GetMSTimeDiffToNow(oldMSTime));
            }
            else
            {
                Log.outInfo(LogFilter.ServerLoading, "Loaded 0 Conversation actors. DB table `conversation_actors` is empty.");
            }

            // TODO: Remove this hack when NextConversationLineID is changed to uint32
            uint getNextConversationLineId(ConversationLineRecord conversationLine)
            {
                if (conversationLine != null && conversationLine.NextConversationLineID != 0)
                {
                    uint FirstLineId = 60000; // Arbitrary id to cover the affected rows

                    if (conversationLine.Id > FirstLineId && conversationLine.NextConversationLineID < (CliDB.ConversationLineStorage.GetNumRows() - ushort.MaxValue - 1))
                        return (uint)(ushort.MaxValue + conversationLine.NextConversationLineID + 1);

                    return (uint)conversationLine.NextConversationLineID;
                }

                return 0u;
            };

            // Validate FirstLineId
            Dictionary<uint, uint> prevConversationLineIds = new();
            foreach (var conversationLine in CliDB.ConversationLineStorage.Values)
            {
                uint nextConversationLineId = getNextConversationLineId(conversationLine);
                if (nextConversationLineId != 0)
                    prevConversationLineIds[nextConversationLineId] = conversationLine.Id;
            }

            uint getFirstLineIdFromAnyLineId(uint lineId)
            {
                uint prevLineId;
                while ((prevLineId = prevConversationLineIds.LookupByKey(lineId)) != 0)
                    lineId = prevLineId;

                return lineId;
            }

            SQLResult templateResult = DB.World.Query("SELECT Id, FirstLineId, TextureKitId, Flags, ScriptName FROM conversation_template");
            if (!templateResult.IsEmpty())
            {
                uint oldMSTime = Time.GetMSTime();

                do
                {
                    ConversationTemplate conversationTemplate = new();
                    conversationTemplate.Id = templateResult.Read<uint>(0);
                    conversationTemplate.FirstLineId = templateResult.Read<uint>(1);
                    conversationTemplate.TextureKitId = templateResult.Read<uint>(2);
                    conversationTemplate.Flags = (ConversationFlags)templateResult.Read<byte>(3);
                    conversationTemplate.ScriptId = Global.ObjectMgr.GetScriptId(templateResult.Read<string>(4));

                    conversationTemplate.Actors = actorsByConversation.TryGetValue(conversationTemplate.Id, out var actors) ? actors.ToList() : new();

                    uint correctedFirstLineId = getFirstLineIdFromAnyLineId(conversationTemplate.FirstLineId);
                    if (conversationTemplate.FirstLineId != correctedFirstLineId)
                    {
                        Log.outError(LogFilter.Sql, $"Table `conversation_template` has incorrect FirstLineId {conversationTemplate.FirstLineId}, it should be {correctedFirstLineId} for Conversation {conversationTemplate.Id}, corrected");
                        conversationTemplate.FirstLineId = correctedFirstLineId;
                    }

                    ConversationLineRecord currentConversationLine = CliDB.ConversationLineStorage.LookupByKey(conversationTemplate.FirstLineId);
                    if (currentConversationLine == null)
                        Log.outError(LogFilter.Sql, "Table `conversation_template` references an invalid line (ID: {0}) for Conversation {1}, skipped", conversationTemplate.FirstLineId, conversationTemplate.Id);

                    while (currentConversationLine != null)
                    {
                        ConversationLineTemplate conversationLineTemplate = _conversationLineTemplateStorage.LookupByKey(currentConversationLine.Id);
                        if (conversationLineTemplate != null)
                            conversationTemplate.Lines.Add(conversationLineTemplate);
                        else
                            Log.outError(LogFilter.Sql, "Table `conversation_line_template` has missing template for line (ID: {0}) in Conversation {1}, skipped", currentConversationLine.Id, conversationTemplate.Id);

                        uint nextConversationLineId = getNextConversationLineId(currentConversationLine);
                        if (nextConversationLineId == 0)
                            break;

                        currentConversationLine = CliDB.ConversationLineStorage.LookupByKey(nextConversationLineId);
                    }

                    _conversationTemplateStorage[conversationTemplate.Id] = conversationTemplate;
                }
                while (templateResult.NextRow());

                Log.outInfo(LogFilter.ServerLoading, "Loaded {0} Conversation templates in {1} ms", _conversationTemplateStorage.Count, Time.GetMSTimeDiffToNow(oldMSTime));
            }
            else
            {
                Log.outInfo(LogFilter.ServerLoading, "Loaded 0 Conversation templates. DB table `conversation_template` is empty.");
            }
        }

        public ConversationTemplate GetConversationTemplate(uint conversationId)
        {
            return _conversationTemplateStorage.LookupByKey(conversationId);
        }

        public ConversationLineTemplate GetConversationLineTemplate(uint conversationLineId)
        {
            return _conversationLineTemplateStorage.LookupByKey(conversationLineId);
        }

        Dictionary<uint, ConversationTemplate> _conversationTemplateStorage = new();
        Dictionary<uint, ConversationLineTemplate> _conversationLineTemplateStorage = new();

        struct ConversationActorDbRow
        {
            public uint ConversationId;
            public uint ActorIndex;

            public ulong SpawnId;
            public uint CreatureId;
            public uint CreatureDisplayInfoId;

            public bool Invoke(ConversationActorTemplate template)
            {
                if (template.WorldObjectTemplate != null)
                    return Invoke(template.WorldObjectTemplate);

                if (template.NoObjectTemplate != null)
                    return Invoke(template.NoObjectTemplate);

                if (template.ActivePlayerTemplate != null)
                    return Invoke(template.ActivePlayerTemplate);

                if (template.TalkingHeadTemplate != null)
                    return Invoke(template.TalkingHeadTemplate);

                return false;
            }

            public bool Invoke(ConversationActorWorldObjectTemplate worldObject)
            {
                if (Global.ObjectMgr.GetCreatureData(SpawnId) == null)
                {
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` references an invalid creature guid (GUID: {SpawnId}) for Conversation {ConversationId} and Idx {ActorIndex}, skipped.");
                    return false;
                }

                if (CreatureId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with ConversationActorGuid cannot have CreatureId ({CreatureId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                if (CreatureDisplayInfoId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with ConversationActorGuid cannot have CreatureDisplayInfoId ({CreatureDisplayInfoId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                worldObject.SpawnId = SpawnId;
                return true;
            }

            public bool Invoke(ConversationActorNoObjectTemplate noObject)
            {
                if (Global.ObjectMgr.GetCreatureTemplate(CreatureId) == null)
                {
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` references an invalid creature id ({CreatureId}) for Conversation {ConversationId} and Idx {ActorIndex}, skipped.");
                    return false;
                }
                if (CreatureDisplayInfoId != 0 && !CliDB.CreatureDisplayInfoStorage.ContainsKey(CreatureDisplayInfoId))
                {
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` references an invalid creature display id ({CreatureDisplayInfoId}) for Conversation {ConversationId} and Idx {ActorIndex}, skipped.");
                    return false;
                }

                if (SpawnId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with NoActorObject cannot have ConversationActorGuid ({SpawnId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                noObject.CreatureId = CreatureId;
                noObject.CreatureDisplayInfoId = CreatureDisplayInfoId;
                return true;
            }

            public bool Invoke(ConversationActorActivePlayerTemplate activePlayer)
            {
                if (SpawnId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with ActivePlayerObject cannot have ConversationActorGuid ({SpawnId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                if (CreatureId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with ActivePlayerObject cannot have CreatureId ({CreatureId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                if (CreatureDisplayInfoId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with ActivePlayerObject cannot have CreatureDisplayInfoId ({CreatureDisplayInfoId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                return true;
            }

            public bool Invoke(ConversationActorTalkingHeadTemplate talkingHead)
            {
                if (Global.ObjectMgr.GetCreatureTemplate(CreatureId) == null)
                {
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` references an invalid creature id ({CreatureId}) for Conversation {ConversationId} and Idx {ActorIndex}, skipped.");
                    return false;
                }
                if (CreatureDisplayInfoId != 0 && !CliDB.CreatureDisplayInfoStorage.ContainsKey(CreatureDisplayInfoId))
                {
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` references an invalid creature display id ({CreatureDisplayInfoId}) for Conversation {ConversationId} and Idx {ActorIndex}, skipped.");
                    return false;
                }

                if (SpawnId != 0)
                    Log.outError(LogFilter.Sql, $"Table `conversation_actors` with TalkingHead cannot have ConversationActorGuid ({SpawnId}). Conversation {ConversationId} and Idx {ActorIndex}.");

                talkingHead.CreatureId = CreatureId;
                talkingHead.CreatureDisplayInfoId = CreatureDisplayInfoId;
                return true;
            }
        }
    }

    public class ConversationActorWorldObjectTemplate
    {
        public ulong SpawnId;
    }

    public class ConversationActorNoObjectTemplate
    {
        public uint CreatureId;
        public uint CreatureDisplayInfoId;
    }

    public class ConversationActorActivePlayerTemplate
    {
    }

    public class ConversationActorTalkingHeadTemplate
    {
        public uint CreatureId;
        public uint CreatureDisplayInfoId;
    }

    public struct ConversationActorTemplate
    {
        public int Id;
        public uint Index;
        public ConversationActorWorldObjectTemplate WorldObjectTemplate;
        public ConversationActorNoObjectTemplate NoObjectTemplate;
        public ConversationActorActivePlayerTemplate ActivePlayerTemplate;
        public ConversationActorTalkingHeadTemplate TalkingHeadTemplate;
    }

    public class ConversationLineTemplate
    {
        public uint Id;          // Link to ConversationLine.db2
        public uint UiCameraID;  // Link to UiCamera.db2
        public byte ActorIdx;    // Index from conversation_actors
        public byte Flags;
        public byte ChatType;
    }

    public class ConversationTemplate
    {
        public uint Id;
        public uint FirstLineId;     // Link to ConversationLine.db2
        public uint TextureKitId;    // Background texture
        public ConversationFlags Flags = ConversationFlags.None;
        public uint ScriptId;

        public List<ConversationActorTemplate> Actors = new();
        public List<ConversationLineTemplate> Lines = new();
    }


}