using Content.Server.AI.Pathfinding.Pathfinders;
using Content.Shared.Access.Systems;
using Content.Shared.AI;
using Content.Shared.GameTicking;
using JetBrains.Annotations;
using Robust.Server.Player;
using Robust.Shared.Map;
using Robust.Shared.Physics;
using Robust.Shared.Timing;
using Robust.Shared.Utility;

namespace Content.Server.AI.Pathfinding.Accessible
{
    /// <summary>
    /// Determines whether an AI has access to a specific pathfinding node.
    /// </summary>
    /// Long-term can be used to do hierarchical pathfinding
    [UsedImplicitly]
    public sealed class AiReachableSystem : EntitySystem
    {
        /*
         * The purpose of this is to provide a higher-level / hierarchical abstraction of the actual pathfinding graph
         * The goal is so that we can more quickly discern if a specific node is reachable or not rather than
         * Pathfinding the entire graph.
         *
         * There's a lot of different implementations of hierarchical or some variation of it: HPA*, PRA, HAA*, etc.
         * (HPA* technically caches the edge nodes of each chunk), e.g. Rimworld, Factorio, etc.
         * so we'll just write one with SS14's requirements in mind.
         *
         * There's probably a better data structure to use though you'd need to benchmark multiple ones to compare,
         * at the very least on the memory side it could definitely be better.
         */
        [Dependency] private readonly IMapManager _mapManager = default!;
        [Dependency] private readonly IGameTiming _gameTiming = default!;
        [Dependency] private readonly PathfindingSystem _pathfindingSystem = default!;
        [Dependency] private readonly AccessReaderSystem _accessReader = default!;

        /// <summary>
        /// Queued region updates
        /// </summary>
        private readonly HashSet<PathfindingChunk> _queuedUpdates = new();

        // Oh god the nesting. Shouldn't need to go beyond this
        /// <summary>
        /// The corresponding regions for each PathfindingChunk.
        /// Regions are groups of nodes with the same profile (for pathfinding purposes)
        /// i.e. same collision, not-space, same access, etc.
        /// </summary>
        private readonly Dictionary<GridId, Dictionary<PathfindingChunk, HashSet<PathfindingRegion>>> _regions =
            new();

        /// <summary>
        /// Minimum time for the cached reachable regions to be stored
        /// </summary>
        private const float MinCacheTime = 1.0f;

        // Cache what regions are accessible from this region. Cached per ReachableArgs
        // so multiple entities in the same region with the same args should all be able to share their reachable lookup
        // Also need to store when we cached it to know if it's stale if the chunks have updated

        // TODO: There's probably a more memory-efficient way to cache this
        // Then again, there's likely also a more memory-efficient way to implement regions.

        // Also, didn't use a dictionary because there didn't seem to be a clean way to do the lookup
        // Plus this way we can check if everything is equal except for vision so an entity with a lower vision radius can use an entity with a higher vision radius' cached result
        private readonly Dictionary<ReachableArgs, Dictionary<PathfindingRegion, (TimeSpan CacheTime, HashSet<PathfindingRegion> Regions)>> _cachedAccessible =
            new();

        private readonly List<PathfindingRegion> _queuedCacheDeletions = new();

#if DEBUG
        private HashSet<IPlayerSession> _subscribedSessions = new();
        private int _runningCacheIdx = 0;
#endif

        public override void Initialize()
        {
            SubscribeLocalEvent<RoundRestartCleanupEvent>(Reset);
            SubscribeLocalEvent<PathfindingChunkUpdateMessage>(RecalculateNodeRegions);
            SubscribeLocalEvent<GridRemovalEvent>(GridRemoved);
#if DEBUG
            SubscribeNetworkEvent<SharedAiDebug.SubscribeReachableMessage>(HandleSubscription);
            SubscribeNetworkEvent<SharedAiDebug.UnsubscribeReachableMessage>(HandleUnsubscription);
#endif
        }

        public override void Shutdown()
        {
            base.Shutdown();

            _queuedUpdates.Clear();
            _regions.Clear();
            _cachedAccessible.Clear();
            _queuedCacheDeletions.Clear();

        }

        private void GridRemoved(GridRemovalEvent ev)
        {
            _regions.Remove(ev.GridId);
        }

        public override void Update(float frameTime)
        {
            base.Update(frameTime);
            foreach (var chunk in _queuedUpdates)
            {
                GenerateRegions(chunk);
            }

            // TODO: Only send diffs instead
#if DEBUG
            if (_subscribedSessions.Count > 0 && _queuedUpdates.Count > 0)
            {
                foreach (var (gridId, regs) in _regions)
                {
                    if (regs.Count > 0)
                    {
                        SendRegionsDebugMessage(gridId);
                    }
                }
            }
#endif
            _queuedUpdates.Clear();

            foreach (var region in _queuedCacheDeletions)
            {
                ClearCache(region);
            }

            _queuedCacheDeletions.Clear();
        }

#if DEBUG
        private void HandleSubscription(SharedAiDebug.SubscribeReachableMessage message, EntitySessionEventArgs eventArgs)
        {
            _subscribedSessions.Add((IPlayerSession) eventArgs.SenderSession);
            foreach (var (gridId, _) in _regions)
            {
                SendRegionsDebugMessage(gridId);
            }
        }

        private void HandleUnsubscription(SharedAiDebug.UnsubscribeReachableMessage message, EntitySessionEventArgs eventArgs)
        {
            _subscribedSessions.Remove((IPlayerSession) eventArgs.SenderSession);
        }
#endif

        private void RecalculateNodeRegions(PathfindingChunkUpdateMessage message)
        {
            // TODO: Only need to do changed nodes ideally
            // For now this is fine but it's a low-hanging fruit optimisation
            _queuedUpdates.Add(message.Chunk);
        }

        /// <summary>
        /// Can the entity reach the target?
        /// </summary>
        /// First it does a quick check to see if there are any traversable nodes in range.
        /// Then it will go through the regions to try and see if there's a region connection between the target and itself
        /// Will used a cached region if available
        /// <param name="entity"></param>
        /// <param name="target"></param>
        /// <param name="range"></param>
        /// <returns></returns>
        public bool CanAccess(EntityUid entity, EntityUid target, float range = 0.0f)
        {
            var xform = EntityManager.GetComponent<TransformComponent>(target);
            // TODO: Handle this gracefully instead of just failing.
            if (!xform.GridID.IsValid())
                return false;

            var targetTile = _mapManager.GetGrid(xform.GridID).GetTileRef(xform.Coordinates);
            var targetNode = _pathfindingSystem.GetNode(targetTile);

            var collisionMask = 0;
            if (EntityManager.TryGetComponent(entity, out IPhysBody? physics))
            {
                collisionMask = physics.CollisionMask;
            }

            var access = _accessReader.FindAccessTags(entity);

            // We'll do a quick traversable check before going through regions
            // If we can't access it we'll try to get a valid node in range (this is essentially an early-out)
            if (!PathfindingHelpers.Traversable(collisionMask, access, targetNode))
            {
                // ReSharper disable once CompareOfFloatsByEqualityOperator
                if (range == 0.0f)
                {
                    return false;
                }

                var pathfindingArgs = new PathfindingArgs(entity, access, collisionMask, default, targetTile, range);
                foreach (var node in BFSPathfinder.GetNodesInRange(pathfindingArgs, false))
                {
                    targetNode = node;
                }
            }

            return CanAccess(entity, targetNode);
        }

        public bool CanAccess(EntityUid entity, PathfindingNode targetNode)
        {
            var xform = EntityManager.GetComponent<TransformComponent>(entity);

            if (xform.GridID != targetNode.TileRef.GridIndex)
                return false;

            var entityTile = _mapManager.GetGrid(xform.GridID).GetTileRef(xform.Coordinates);
            var entityNode = _pathfindingSystem.GetNode(entityTile);
            var entityRegion = GetRegion(entityNode);
            var targetRegion = GetRegion(targetNode);
            // TODO: Regional pathfind from target to entity
            // Early out
            if (entityRegion == targetRegion)
            {
                return true;
            }

            // We'll go from target's position to us because most of the time it's probably in a locked room rather than vice versa
            var reachableArgs = ReachableArgs.GetArgs(entity);
            var reachableRegions = GetReachableRegions(reachableArgs, targetRegion);

            return entityRegion != null && reachableRegions.Contains(entityRegion);
        }

        /// <summary>
        /// Retrieve the reachable regions
        /// </summary>
        /// <param name="reachableArgs"></param>
        /// <param name="region"></param>
        /// <returns></returns>
        public HashSet<PathfindingRegion> GetReachableRegions(ReachableArgs reachableArgs, PathfindingRegion? region)
        {
            // if we're on a node that's not tracked at all atm then region will be null
            if (region == null)
            {
                return new HashSet<PathfindingRegion>();
            }

            var cachedArgs = GetCachedArgs(reachableArgs);
            (TimeSpan CacheTime, HashSet<PathfindingRegion> Regions) cached;

            if (!IsCacheValid(cachedArgs, region))
            {
                cached = GetVisionReachable(cachedArgs, region);
                _cachedAccessible[cachedArgs][region] = cached;
#if DEBUG
                SendRegionCacheMessage(region.ParentChunk.GridId, cached.Regions, false);
#endif
            }
            else
            {
                cached = _cachedAccessible[cachedArgs][region];
#if DEBUG
                SendRegionCacheMessage(region.ParentChunk.GridId, cached.Regions, true);
#endif
            }

            return cached.Regions;
        }

        /// <summary>
        /// Get any adequate cached args if possible, otherwise just use ours
        /// </summary>
        /// Essentially any args that have the same access AND >= our vision radius can be used
        /// <param name="accessibleArgs"></param>
        /// <returns></returns>
        private ReachableArgs GetCachedArgs(ReachableArgs accessibleArgs)
        {
            ReachableArgs? foundArgs = null;

            foreach (var (cachedAccessible, _) in _cachedAccessible)
            {
                if (Equals(cachedAccessible.Access, accessibleArgs.Access) &&
                    cachedAccessible.CollisionMask == accessibleArgs.CollisionMask &&
                    cachedAccessible.VisionRadius <= accessibleArgs.VisionRadius)
                {
                    foundArgs = cachedAccessible;
                    break;
                }
            }

            return foundArgs ?? accessibleArgs;
        }

        /// <summary>
        /// Checks whether there's a valid cache for our accessibility args.
        /// Most regular mobs can share their cached accessibility with each other
        /// </summary>
        /// Will also remove it from the cache if it is invalid
        /// <param name="accessibleArgs"></param>
        /// <param name="region"></param>
        /// <returns></returns>
        private bool IsCacheValid(ReachableArgs accessibleArgs, PathfindingRegion region)
        {
            if (!_cachedAccessible.TryGetValue(accessibleArgs, out var cachedArgs))
            {
                _cachedAccessible.Add(accessibleArgs, new Dictionary<PathfindingRegion, (TimeSpan, HashSet<PathfindingRegion>)>());
                return false;
            }

            if (!cachedArgs.TryGetValue(region, out var regionCache))
            {
                return false;
            }

            // Just so we don't invalidate the cache every tick we'll store it for a minimum amount of time
            var currentTime = _gameTiming.CurTime;
            if ((currentTime - regionCache.CacheTime).TotalSeconds < MinCacheTime)
            {
                return true;
            }

            var checkedAccess = new HashSet<PathfindingRegion>();
            // Check if cache is stale
            foreach (var accessibleRegion in regionCache.Regions)
            {
                if (checkedAccess.Contains(accessibleRegion)) continue;

                // Any applicable chunk has been invalidated OR one of our neighbors has been invalidated (i.e. new connections)
                // TODO: Could look at storing the TimeSpan directly on the region so our neighbor can tell us straight-up
                if (accessibleRegion.ParentChunk.LastUpdate > regionCache.CacheTime)
                {
                    // Remove the stale cache, to be updated later
                    _cachedAccessible[accessibleArgs].Remove(region);
                    return false;
                }

                foreach (var neighbor in accessibleRegion.Neighbors)
                {
                    if (checkedAccess.Contains(neighbor)) continue;
                    if (neighbor.ParentChunk.LastUpdate > regionCache.CacheTime)
                    {
                        _cachedAccessible[accessibleArgs].Remove(region);
                        return false;
                    }
                    checkedAccess.Add(neighbor);
                }
                checkedAccess.Add(accessibleRegion);
            }
            return true;
        }

        /// <summary>
        /// Caches the entity's nearby accessible regions in vision radius
        /// </summary>
        /// Longer-term TODO: Hierarchical pathfinding in which case this function would probably get bulldozed, BRRRTT
        /// <param name="reachableArgs"></param>
        /// <param name="entityRegion"></param>
        private (TimeSpan, HashSet<PathfindingRegion>) GetVisionReachable(ReachableArgs reachableArgs, PathfindingRegion entityRegion)
        {
            var openSet = new Queue<PathfindingRegion>();
            openSet.Enqueue(entityRegion);
            var closedSet = new HashSet<PathfindingRegion>();
            var accessible = new HashSet<PathfindingRegion> {entityRegion};

            while (openSet.Count > 0)
            {
                var region = openSet.Dequeue();
                closedSet.Add(region);

                foreach (var neighbor in region.Neighbors)
                {
                    if (closedSet.Contains(neighbor))
                    {
                        continue;
                    }

                    // Distance is an approximation here so we'll be generous with it
                    // TODO: Could do better; the fewer nodes the better it is.
                    if (!neighbor.RegionTraversable(reachableArgs) ||
                        neighbor.Distance(entityRegion) > reachableArgs.VisionRadius + 1)
                    {
                        closedSet.Add(neighbor);
                        continue;
                    }

                    openSet.Enqueue(neighbor);
                    accessible.Add(neighbor);
                }
            }

            return (_gameTiming.CurTime, accessible);
        }

        /// <summary>
        /// Grab the related cardinal nodes and if they're in different regions then add to our edge and their edge
        /// </summary>
        /// Implicitly they would've already been merged if possible
        /// <param name="region"></param>
        /// <param name="node"></param>
        private void UpdateRegionEdge(PathfindingRegion region, PathfindingNode node)
        {
            DebugTools.Assert(region.Nodes.Contains(node));
            // Originally I tried just doing bottom and left but that doesn't work as the chunk update order is not guaranteed

            var checkDirections = new[] {Direction.East, Direction.South, Direction.West, Direction.North};
            foreach (var direction in checkDirections)
            {
                var directionNode = node.GetNeighbor(direction);
                if (directionNode == null) continue;

                var directionRegion = GetRegion(directionNode);
                if (directionRegion == null || directionRegion == region) continue;

                region.Neighbors.Add(directionRegion);
                directionRegion.Neighbors.Add(region);
            }
        }

        /// <summary>
        /// Get the current region for this entity
        /// </summary>
        /// <param name="entity"></param>
        /// <returns></returns>
        public PathfindingRegion? GetRegion(EntityUid entity)
        {
            var xform = EntityManager.GetComponent<TransformComponent>(entity);

            if (!xform.GridID.IsValid())
            {
                return null;
            }

            var entityTile = _mapManager.GetGrid(xform.GridID).GetTileRef(xform.Coordinates);
            var entityNode = _pathfindingSystem.GetNode(entityTile);
            return GetRegion(entityNode);
        }

        /// <summary>
        /// Get the current region for this node
        /// </summary>
        /// <param name="node"></param>
        /// <returns></returns>
        public PathfindingRegion? GetRegion(PathfindingNode node)
        {
            // Not sure on the best way to optimise this
            // On the one hand, just storing each node's region is faster buuutttt muh memory
            // On the other hand, you might need O(n) lookups on regions for each chunk, though it's probably not too bad with smaller chunk sizes?
            // Someone smarter than me will know better
            var parentChunk = node.ParentChunk;

            // No guarantee the node even has a region yet (if we're doing neighbor lookups)
            if (!_regions[parentChunk.GridId].TryGetValue(parentChunk, out var regions))
            {
                return null;
            }

            foreach (var region in regions)
            {
                if (region.Nodes.Contains(node))
                {
                    return region;
                }
            }

            // Longer term this will probably be guaranteed a region but for now space etc. are no region
            return null;
        }

        /// <summary>
        /// Add this node to the relevant region.
        /// </summary>
        /// <param name="node"></param>
        /// <param name="existingRegions">The cached region for each node</param>
        /// <param name="chunkRegions">The existing regions in the chunk</param>
        /// <param name="x">This is already calculated in advance so may as well re-use it</param>
        /// <param name="y">This is already calculated in advance so may as well re-use it</param>
        /// <returns></returns>
        private PathfindingRegion? CalculateNode(
            PathfindingNode node,
            Dictionary<PathfindingNode, PathfindingRegion> existingRegions,
            HashSet<PathfindingRegion> chunkRegions,
            int x, int y)
        {
            DebugTools.Assert(_regions.ContainsKey(node.ParentChunk.GridId));
            DebugTools.Assert(_regions[node.ParentChunk.GridId].ContainsKey(node.ParentChunk));
            // TODO For now we don't have these regions but longer-term yeah sure
            if (node.BlockedCollisionMask != 0x0 || node.TileRef.Tile.IsEmpty)
            {
                return null;
            }

            var parentChunk = node.ParentChunk;
            // Doors will be their own separate region
            // We won't store them in existingRegions so they don't show up and can't be connected to (at least for now)
            if (node.AccessReaders.Count > 0)
            {
                var region = new PathfindingRegion(node, new HashSet<PathfindingNode>(1) {node}, true);
                _regions[parentChunk.GridId][parentChunk].Add(region);
                UpdateRegionEdge(region, node);
                return region;
            }

            // Relative x and y of the chunk
            // If one of our bottom / left neighbors are in a region try to join them
            // Otherwise, make our own region.
            var leftNeighbor = x > 0 ? parentChunk.Nodes[x - 1, y] : null;
            var bottomNeighbor = y > 0 ? parentChunk.Nodes[x, y - 1] : null;
            PathfindingRegion? leftRegion;
            PathfindingRegion? bottomRegion;

            // We'll check if our left or down neighbors are already in a region and join them

            // Is left node valid to connect to
            if (leftNeighbor != null &&
                existingRegions.TryGetValue(leftNeighbor, out leftRegion) &&
                !leftRegion.IsDoor)
            {
                // We'll try and connect the left node's region to the bottom region if they're separate (yay merge)
                if (bottomNeighbor != null &&
                    existingRegions.TryGetValue(bottomNeighbor, out bottomRegion) &&
                    bottomRegion != leftRegion &&
                    !bottomRegion.IsDoor)
                {
                    bottomRegion.Add(node);
                    existingRegions.Add(node, bottomRegion);
                    MergeInto(leftRegion, bottomRegion, existingRegions);

                    // Cleanup leftRegion
                    // MergeInto will remove it from the overall region chunk cache while we need to remove it from
                    // our short-term ones (chunkRegions and existingRegions)
                    chunkRegions.Remove(leftRegion);

                    foreach (var leftNode in leftRegion.Nodes)
                    {
                        existingRegions[leftNode] = bottomRegion;
                    }

                    return bottomRegion;
                }

                leftRegion.Add(node);
                existingRegions.Add(node, leftRegion);
                UpdateRegionEdge(leftRegion, node);
                return leftRegion;
            }

            //Is bottom node valid to connect to
            if (bottomNeighbor != null &&
                existingRegions.TryGetValue(bottomNeighbor, out bottomRegion) &&
                !bottomRegion.IsDoor)
            {
                bottomRegion.Add(node);
                existingRegions.Add(node, bottomRegion);
                UpdateRegionEdge(bottomRegion, node);
                return bottomRegion;
            }

            // If we can't join an existing region then we'll make our own
            var newRegion = new PathfindingRegion(node, new HashSet<PathfindingNode> {node}, node.AccessReaders.Count > 0);
            _regions[parentChunk.GridId][parentChunk].Add(newRegion);
            existingRegions.Add(node, newRegion);
            UpdateRegionEdge(newRegion, node);
            return newRegion;
        }

        /// <summary>
        /// Combines the two regions into one bigger region
        /// </summary>
        /// <param name="source"></param>
        /// <param name="target"></param>
        private void MergeInto(PathfindingRegion source, PathfindingRegion target, Dictionary<PathfindingNode, PathfindingRegion>? existingRegions = null)
        {
            DebugTools.AssertNotNull(source);
            DebugTools.AssertNotNull(target);
            DebugTools.Assert(source != target);
            foreach (var node in source.Nodes)
            {
                target.Add(node);
            }

            if (existingRegions != null)
            {
                foreach (var node in source.Nodes)
                {
                    existingRegions[node] = target;
                }
            }

            source.Shutdown();
            // This doesn't check the cachedaccessible to see if it's reachable but maybe it should?
            // Although currently merge gets spammed so maybe when some other stuff is improved
            // MergeInto is also only called by GenerateRegions currently so nothing should hold onto the original region
            _regions[source.ParentChunk.GridId][source.ParentChunk].Remove(source);

            foreach (var node in target.Nodes)
            {
                UpdateRegionEdge(target, node);
            }
        }

        /// <summary>
        /// Remove the cached accessibility lookup for this region
        /// </summary>
        /// <param name="region"></param>
        private void ClearCache(PathfindingRegion region)
        {
            DebugTools.Assert(region.Deleted);

            // Need to forcibly clear cache for ourself and anything that includes us
            foreach (var (_, cachedRegions) in _cachedAccessible)
            {
                if (cachedRegions.ContainsKey(region))
                {
                    cachedRegions.Remove(region);
                }

                // Seemed like the safest way to remove this
                // We could just have GetVisionAccessible remove us if it can tell we're deleted but that
                // seems like it could be unreliable
                var regionsToClear = new List<PathfindingRegion>();

                foreach (var (otherRegion, cache) in cachedRegions)
                {
                    if (cache.Regions.Contains(region))
                    {
                        regionsToClear.Add(otherRegion);
                    }
                }

                foreach (var otherRegion in regionsToClear)
                {
                    cachedRegions.Remove(otherRegion);
                }
            }

#if DEBUG
            if (_regions.TryGetValue(region.ParentChunk.GridId, out var chunks) &&
                chunks.TryGetValue(region.ParentChunk, out var regions))
            {
                DebugTools.Assert(!regions.Contains(region));
            }
#endif
        }

        /// <summary>
        /// Generate all of the regions within a chunk
        /// </summary>
        /// These can't across over into another chunk and doors are their own region
        /// <param name="chunk"></param>
        private void GenerateRegions(PathfindingChunk chunk)
        {
            // Grid deleted while update queued, or invalid grid.
            if (!_mapManager.TryGetGrid(chunk.GridId, out _))
            {
                return;
            }

            if (!_regions.ContainsKey(chunk.GridId))
            {
                _regions.Add(chunk.GridId, new Dictionary<PathfindingChunk, HashSet<PathfindingRegion>>());
            }

            if (_regions[chunk.GridId].TryGetValue(chunk, out var regions))
            {
                foreach (var region in regions)
                {
                    _queuedCacheDeletions.Add(region);
                    region.Shutdown();
                }

                _regions[chunk.GridId].Remove(chunk);
            }

            // Temporarily store the corresponding region for each node
            // Makes merging regions or adding nodes to existing regions neater.
            var nodeRegions = new Dictionary<PathfindingNode, PathfindingRegion>();
            var chunkRegions = new HashSet<PathfindingRegion>();
            _regions[chunk.GridId].Add(chunk, chunkRegions);

            for (var y = 0; y < PathfindingChunk.ChunkSize; y++)
            {
                for (var x = 0; x < PathfindingChunk.ChunkSize; x++)
                {
                    var node = chunk.Nodes[x, y];
                    var region = CalculateNode(node, nodeRegions, chunkRegions, x, y);
                    // Currently we won't store a separate region for each mask / space / whatever because muh effort
                    // Long-term you'll want to account for it probably
                    if (region == null)
                    {
                        continue;
                    }

                    chunkRegions.Add(region);
                }
            }
#if DEBUG
            foreach (var region in chunkRegions)
            {
                DebugTools.Assert(!region.Deleted);
            }

            DebugTools.Assert(chunkRegions.Count < Math.Pow(PathfindingChunk.ChunkSize, 2));
            SendRegionsDebugMessage(chunk.GridId);
#endif
        }

        public void Reset(RoundRestartCleanupEvent ev)
        {
            _queuedUpdates.Clear();
            _regions.Clear();
            _cachedAccessible.Clear();
            _queuedCacheDeletions.Clear();
        }

#if DEBUG
        private void SendRegionsDebugMessage(GridId gridId)
        {
            if (_subscribedSessions.Count == 0) return;
            var grid = _mapManager.GetGrid(gridId);
            // Chunk / Regions / Nodes
            var debugResult = new Dictionary<int, Dictionary<int, List<Vector2>>>();
            var chunkIdx = 0;
            var regionIdx = 0;

            if (!_regions.TryGetValue(gridId, out var dict))
            {
                return;
            }

            foreach (var (_, regions) in dict)
            {
                var debugRegions = new Dictionary<int, List<Vector2>>();
                debugResult.Add(chunkIdx, debugRegions);

                foreach (var region in regions)
                {
                    var debugRegionNodes = new List<Vector2>(region.Nodes.Count);
                    debugResult[chunkIdx].Add(regionIdx, debugRegionNodes);

                    foreach (var node in region.Nodes)
                    {
                        var nodeVector = grid.GridTileToLocal(node.TileRef.GridIndices).ToMapPos(EntityManager);
                        debugRegionNodes.Add(nodeVector);
                    }

                    regionIdx++;
                }

                chunkIdx++;
            }

            foreach (var session in _subscribedSessions)
            {
                RaiseNetworkEvent(new SharedAiDebug.ReachableChunkRegionsDebugMessage(gridId, debugResult), session.ConnectedClient);
            }
        }

        /// <summary>
        ///     Sent whenever the reachable cache for a particular mob is built or retrieved
        /// </summary>
        /// <param name="gridId"></param>
        /// <param name="regions"></param>
        /// <param name="cached"></param>
        private void SendRegionCacheMessage(GridId gridId, IEnumerable<PathfindingRegion> regions, bool cached)
        {
            if (_subscribedSessions.Count == 0) return;

            var grid = _mapManager.GetGrid(gridId);
            var debugResult = new Dictionary<int, List<Vector2>>();

            foreach (var region in regions)
            {
                debugResult.Add(_runningCacheIdx, new List<Vector2>());

                foreach (var node in region.Nodes)
                {
                    var nodeVector = grid.GridTileToLocal(node.TileRef.GridIndices).ToMapPos(EntityManager);

                    debugResult[_runningCacheIdx].Add(nodeVector);
                }

                _runningCacheIdx++;
            }

            foreach (var session in _subscribedSessions)
            {
                RaiseNetworkEvent(new SharedAiDebug.ReachableCacheDebugMessage(gridId, debugResult, cached), session.ConnectedClient);
            }
        }
#endif
    }
}