Files
CypherCore/Source/Game/DataStorage/ConversationDataStorage.cs

359 lines
17 KiB
C#

// 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();
}
}