AAEmu.Game/GameData/AiGameData.cs - develop - NLObP/AAEmu

AAEmu.Game/GameData/AiGameData.cs
156 строк5 KB
Roger Barreto
Collection expressions check + blank lines update (#1110)
11 дек 2024, 19:38
11 дек 2024, 19:380ce5643
using System;
using System.Collections.Generic;

using AAEmu.Commons.Utils;
using AAEmu.Game.Core.Managers.UnitManagers;
using AAEmu.Game.GameData.Framework;
using AAEmu.Game.Models.Game.AI.Enums;
using AAEmu.Game.Models.Game.AI.v2.Params;
using AAEmu.Game.Utils.DB;

using Microsoft.Data.Sqlite;

using NLog;

#pragma warning disable IDE0079 // Remove unnecessary suppression

namespace AAEmu.Game.GameData;

[GameData]
public class AiGameData : Singleton<AiGameData>, IGameDataLoader
{
    private static Logger Logger { get; } = LogManager.GetCurrentClassLogger();

    private Dictionary<uint, AiParams> _aiParams;
    private Dictionary<uint, List<AiCommands>> _aiCommands;
    private Dictionary<uint, AiCommandSets> _aiCommandSets;

    public AiParams GetAiParamsForId(uint id)
    {
        if (_aiParams.TryGetValue(id, out var value))
            return value;
        return null;
    }

    public List<AiCommands> GetAiCommands(uint id)
    {
        if (_aiCommands.TryGetValue(id, out var value))
            return value;
        return null;
    }

    public void Load(SqliteConnection connection)
    {
        _aiParams = [];
        _aiCommands = [];
        _aiCommandSets = [];

        var fileTypeToId = new Dictionary<uint, AiParamType>();

        using (var command = connection.CreateCommand())
        {
            command.CommandText = "SELECT id, ai_file_id, npc_ai_param_id FROM npcs";
            command.Prepare();
            using (var sqliteReader = command.ExecuteReader())
            using (var reader = new SQLiteWrapperReader(sqliteReader))
            {
                while (reader.Read())
                {
                    var npcId = reader.GetUInt32("id");
                    var type = (AiParamType)reader.GetUInt32("ai_file_id");
                    var id = reader.GetUInt32("npc_ai_param_id");
                    fileTypeToId.TryAdd(id, type);
                }
            }
        }

        using (var command = connection.CreateCommand())
        {
            command.CommandText = "SELECT * FROM npc_ai_params";
            command.Prepare();
            using (var sqliteReader = command.ExecuteReader())
            using (var reader = new SQLiteWrapperReader(sqliteReader))
            {
                while (reader.Read())
                {
                    var id = reader.GetUInt32("id");
                    if (!fileTypeToId.ContainsKey(id))
                        continue;

                    var fileType = fileTypeToId[id];
                    try
                    {
                        var data = reader.IsDBNull("ai_param") ? string.Empty : reader.GetString("ai_param");
                        var aiParams = AiParams.CreateByType(fileType, data);
#pragma warning disable CA1508 // Avoid dead conditional code
                        if (aiParams != null)
                            _aiParams.TryAdd(id, aiParams);
#pragma warning restore CA1508 // Avoid dead conditional code
                    }
                    catch (Exception e)
                    {
                        Logger.Warn("Impossible to parse npc_ai_params {0}\n{1}", id, e.Message);
                    }
                }
            }
        }

        using (var command = connection.CreateCommand())
        {
            command.CommandText = "SELECT * FROM ai_commands";
            command.Prepare();
            using (var sqliteReader = command.ExecuteReader())
            using (var reader = new SQLiteWrapperReader(sqliteReader))
            {
                var tempListId = new List<uint>();
                while (reader.Read())
                {
                    var template = new AiCommands();
                    template.Id = reader.GetUInt32("id");
                    if (tempListId.Contains(template.Id))
                    {
                        continue; // в таблице есть дубли
                    }

                    tempListId.Add(template.Id);
                    template.CmdSetId = reader.GetUInt32("cmd_set_id");
                    template.CmdId = (AiCommandCategory)reader.GetUInt32("cmd_id");
                    template.Param1 = reader.GetUInt32("param1");
                    template.Param2 = reader.GetString("param2");

                    if (!_aiCommands.TryGetValue(template.CmdSetId, out var value))
                    {
                        value = [];
                        _aiCommands.Add(template.CmdSetId, value);
                    }

                    value.Add(template);
                }
            }
        }

        using (var command = connection.CreateCommand())
        {
            command.CommandText = "SELECT * FROM ai_command_sets";
            command.Prepare();
            using (var sqliteReader = command.ExecuteReader())
            using (var reader = new SQLiteWrapperReader(sqliteReader))
            {
                while (reader.Read())
                {
                    var template = new AiCommandSets();
                    template.Id = reader.GetUInt32("id");
                    template.Name = reader.GetString("name");
                    template.CanInteract = reader.GetBoolean("can_interact");

                    _aiCommandSets.TryAdd(template.Id, template);
                }
            }
        }
    }

    public void PostLoad()
    {
        NpcManager.Instance.LoadAiParams();
    }
}