CypherCore/Source/Game/World/WorldStateManager.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.Collections;
using Framework.Constants;
using Framework.Database;
using Game.DataStorage;
using Game.Maps;
using Game.Networking.Packets;
using System;
using System.Collections.Generic;
using System.Linq;

namespace Game
{
    public class WorldStateManager : Singleton<WorldStateManager>
    {
        static int AnyMap = -1;

        Dictionary<int, WorldStateTemplate> _worldStateTemplates = new();
        Dictionary<int, int> _realmWorldStateValues = new();
        Dictionary<int, Dictionary<int, int>> _worldStatesByMap = new();

        WorldStateManager() { }

        public void LoadFromDB()
        {
            uint oldMSTime = Time.GetMSTime();

            //                                         0   1             2       3        4
            SQLResult result = DB.World.Query("SELECT ID, DefaultValue, MapIDs, AreaIDs, ScriptName FROM world_state");
            if (result.IsEmpty())
                return;

            do
            {
                int id = result.Read<int>(0);
                WorldStateTemplate worldState = new();
                worldState.Id = id;
                worldState.DefaultValue = result.Read<int>(1);

                string mapIds = result.Read<string>(2);
                if (!mapIds.IsEmpty())
                {
                    foreach (string mapIdToken in new StringArray(mapIds, ','))
                    {
                        if (!int.TryParse(mapIdToken, out int mapId))
                        {
                            Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with non-integer MapID ({mapIdToken}), map ignored");
                            continue;
                        }

                        if (!CliDB.MapStorage.ContainsKey(mapId) && mapId != AnyMap)
                        {
                            Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with invalid MapID ({mapId}), map ignored");
                            continue;
                        }

                        worldState.MapIds.Add(mapId);
                    }
                }

                if (!mapIds.IsEmpty() && worldState.MapIds.Empty())
                {
                    Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with nonempty MapIDs ({mapIds}) but no valid map id was found, ignored");
                    continue;
                }

                string areaIds = result.Read<string>(3);
                if (!areaIds.IsEmpty() && !worldState.MapIds.Empty())
                {
                    foreach (string areaIdToken in new StringArray(areaIds, ','))
                    {
                        if (!uint.TryParse(areaIdToken, out uint areaId))
                        {
                            Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with non-integer AreaID ({areaIdToken}), area ignored");
                            continue;
                        }

                        var areaTableEntry = CliDB.AreaTableStorage.LookupByKey(areaId);
                        if (areaTableEntry == null)
                        {
                            Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with invalid AreaID ({areaId}), area ignored");
                            continue;
                        }

                        if (!worldState.MapIds.Contains(areaTableEntry.ContinentID))
                        {
                            Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with AreaID ({areaId}) not on any of required maps, area ignored");
                            continue;
                        }

                        worldState.AreaIds.Add(areaId);
                    }

                    if (!areaIds.IsEmpty() && worldState.AreaIds.Empty())
                    {
                        Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with nonempty AreaIDs ({areaIds}) but no valid area id was found, ignored");
                        continue;
                    }
                }
                else if (!areaIds.IsEmpty())
                {
                    Log.outError(LogFilter.Sql, $"Table `world_state` contains a world state {id} with nonempty AreaIDs ({areaIds}) but is a realm wide world state, area requirement ignored");
                }

                worldState.ScriptId = Global.ObjectMgr.GetScriptId(result.Read<string>(4));

                if (!worldState.MapIds.Empty())
                {
                    foreach (int mapId in worldState.MapIds)
                    {
                        if (!_worldStatesByMap.ContainsKey(mapId))
                            _worldStatesByMap[mapId] = new();

                        _worldStatesByMap[mapId][id] = worldState.DefaultValue;
                    }
                }
                else
                    _realmWorldStateValues[id] = worldState.DefaultValue;

                _worldStateTemplates[id] = worldState;

            } while (result.NextRow());

            Log.outInfo(LogFilter.ServerLoading, $"Loaded {_worldStateTemplates.Count} world state templates {Time.GetMSTimeDiffToNow(oldMSTime)} ms");

            oldMSTime = Time.GetMSTime();

            result = DB.Characters.Query("SELECT Id, Value FROM world_state_value");
            uint savedValueCount = 0;
            if (!result.IsEmpty())
            {
                do
                {
                    int worldStateId = result.Read<int>(0);
                    WorldStateTemplate worldState = _worldStateTemplates.LookupByKey(worldStateId);
                    if (worldState == null)
                    {
                        Log.outError(LogFilter.Sql, $"Table `world_state_value` contains a value for unknown world state {worldStateId}, ignored");
                        continue;
                    }

                    int value = result.Read<int>(1);

                    if (!worldState.MapIds.Empty())
                    {
                        foreach (int mapId in worldState.MapIds)
                        {
                            if (!_worldStatesByMap.ContainsKey(mapId))
                                _worldStatesByMap[mapId] = new();

                            _worldStatesByMap[mapId][worldStateId] = value;
                        }
                    }
                    else
                        _realmWorldStateValues[worldStateId] = value;

                    ++savedValueCount;
                }
                while (result.NextRow());
            }

            Log.outInfo(LogFilter.ServerLoading, $"Loaded {savedValueCount} saved world state values {Time.GetMSTimeDiffToNow(oldMSTime)} ms");
        }

        public WorldStateTemplate GetWorldStateTemplate(int worldStateId)
        {
            return _worldStateTemplates.LookupByKey(worldStateId);
        }

        public int GetValue(WorldStates worldStateId, Map map)
        {
            return GetValue((int)worldStateId, map);
        }

        public int GetValue(int worldStateId, Map map)
        {
            WorldStateTemplate worldStateTemplate = GetWorldStateTemplate(worldStateId);
            if (worldStateTemplate == null || worldStateTemplate.MapIds.Empty())
                return _realmWorldStateValues.LookupByKey(worldStateId);

            if (map == null || (!worldStateTemplate.MapIds.Contains((int)map.GetId()) && !worldStateTemplate.MapIds.Contains(AnyMap)))
                return 0;

            return map.GetWorldStateValue(worldStateId);
        }

        public void SetValue(WorldStates worldStateId, int value, bool hidden, Map map)
        {
            SetValue((int)worldStateId, value, hidden, map);
        }

        public void SetValue(uint worldStateId, int value, bool hidden, Map map)
        {
            SetValue((int)worldStateId, value, hidden, map);
        }

        public void SetValue(int worldStateId, int value, bool hidden, Map map)
        {
            WorldStateTemplate worldStateTemplate = GetWorldStateTemplate(worldStateId);
            if (worldStateTemplate == null || worldStateTemplate.MapIds.Empty())
            {
                int oldValue = 0;
                if (!_realmWorldStateValues.TryAdd(worldStateId, 0))
                {
                    oldValue = _realmWorldStateValues[worldStateId];
                    if (oldValue == value)
                        return;
                }

                _realmWorldStateValues[worldStateId] = value;

                if (worldStateTemplate != null)
                    Global.ScriptMgr.OnWorldStateValueChange(worldStateTemplate, oldValue, value, null);

                // Broadcast update to all players on the server
                UpdateWorldState updateWorldState = new();
                updateWorldState.VariableID = (uint)worldStateId;
                updateWorldState.Value = value;
                updateWorldState.Hidden = hidden;
                Global.WorldMgr.SendGlobalMessage(updateWorldState);
                return;
            }

            if (map == null || (!worldStateTemplate.MapIds.Contains((int)map.GetId()) && !worldStateTemplate.MapIds.Contains(AnyMap)))
                return;

            map.SetWorldStateValue(worldStateId, value, hidden);
        }

        public void SaveValueInDb(int worldStateId, int value)
        {
            if (GetWorldStateTemplate(worldStateId) == null)
                return;

            PreparedStatement stmt = DB.Characters.GetPreparedStatement(CharStatements.REP_WORLD_STATE);
            stmt.AddValue(0, worldStateId);
            stmt.AddValue(1, value);
            DB.Characters.Execute(stmt);
        }

        public void SetValueAndSaveInDb(WorldStates worldStateId, int value, bool hidden, Map map)
        {
            SetValueAndSaveInDb((int)worldStateId, value, hidden, map);
        }

        public void SetValueAndSaveInDb(int worldStateId, int value, bool hidden, Map map)
        {
            SetValue(worldStateId, value, hidden, map);
            SaveValueInDb(worldStateId, value);
        }

        public Dictionary<int, int> GetInitialWorldStatesForMap(Map map)
        {
            Dictionary<int, int> initialValues = new();

            if (_worldStatesByMap.TryGetValue((int)map.GetId(), out Dictionary<int, int> valuesTemplate))
            {
                foreach (var (key, value) in valuesTemplate)
                    initialValues.Add(key, value);
            }

            if (_worldStatesByMap.TryGetValue(AnyMap, out valuesTemplate))
            {
                foreach (var (key, value) in valuesTemplate)
                    initialValues.Add(key, value);
            }

            return initialValues;
        }

        public void FillInitialWorldStates(InitWorldStates initWorldStates, Map map, uint playerAreaId)
        {
            foreach (var (worldStateId, value) in _realmWorldStateValues)
                initWorldStates.AddState(worldStateId, value);

            foreach (var (worldStateId, value) in map.GetWorldStateValues())
            {
                WorldStateTemplate worldStateTemplate = GetWorldStateTemplate(worldStateId);
                if (worldStateTemplate != null && !worldStateTemplate.AreaIds.Empty())
                {
                    bool isInAllowedArea = worldStateTemplate.AreaIds.Any(requiredAreaId => Global.DB2Mgr.IsInArea(playerAreaId, requiredAreaId));
                    if (!isInAllowedArea)
                        continue;
                }

                initWorldStates.AddState(worldStateId, value);
            }
        }
    }

    public class WorldStateTemplate
    {
        public int Id;
        public int DefaultValue;
        public uint ScriptId;

        public List<int> MapIds = new();
        public List<uint> AreaIds = new();
    }
}