using System;
using Cysharp.Threading.Tasks;

namespace DanieleMarotta.RiversongCodeShowcase
{
    public class BlockMapUpdateSystem : GameSystem, IInitializable, IDisposable
    {
        [InjectService]
        private ISignalBus _signalBus;

        [InjectService]
        private World _world;

        [InjectService]
        private GameConfig _config;

        public BlockMapUpdateSystem(IServiceLocator serviceLocator) : base(serviceLocator)
        {
        }

        public UniTask InitializeAsync()
        {
            _signalBus.Subscribe<BuildingPlacementAnimationStartedSignal>(OnBuildingPlacementAnimationStarted);
            _signalBus.Subscribe<BuildingCreatedSignal>(OnBuildingCreated);
            _signalBus.Subscribe<BuildingDeleteAnimationCompletedSignal>(OnBuildingDeleteAnimationCompleted);
            _signalBus.Subscribe<ConstructionSiteDeletedSignal>(OnConstructionSiteDeleted);
            _signalBus.Subscribe<RoadTileUpdatedSignal>(OnRoadTileUpdated);
            _signalBus.Subscribe<RawResourcesRemovedSignal>(OnRawResourcesRemoved);

            return UniTask.CompletedTask;
        }

        public void Dispose()
        {
            _signalBus.Unsubscribe<BuildingPlacementAnimationStartedSignal>(OnBuildingPlacementAnimationStarted);
            _signalBus.Unsubscribe<BuildingCreatedSignal>(OnBuildingCreated);
            _signalBus.Unsubscribe<BuildingDeleteAnimationCompletedSignal>(OnBuildingDeleteAnimationCompleted);
            _signalBus.Unsubscribe<ConstructionSiteDeletedSignal>(OnConstructionSiteDeleted);
            _signalBus.Unsubscribe<RoadTileUpdatedSignal>(OnRoadTileUpdated);
            _signalBus.Unsubscribe<RawResourcesRemovedSignal>(OnRawResourcesRemoved);
        }

        private void OnBuildingPlacementAnimationStarted(BuildingPlacementAnimationStartedSignal signal)
        {
            _world.BlockMap.AddReason(signal.Rect.Min, signal.Rect.Max, BlockReason.Building);
        }

        private void OnBuildingCreated(BuildingCreatedSignal signal)
        {
            var rect = signal.Building.Rect;
            _world.BlockMap.AddReason(rect.Min, rect.Max, BlockReason.Building);
        }

        private void OnBuildingDeleteAnimationCompleted(BuildingDeleteAnimationCompletedSignal signal)
        {
            var rect = signal.Building.Rect;
            _world.BlockMap.RemoveReason(rect.Min, rect.Max, BlockReason.Building);
        }

        private void OnConstructionSiteDeleted(ConstructionSiteDeletedSignal signal)
        {
            var rect = signal.ConstructionSite.Rect;
            _world.BlockMap.RemoveReason(rect.Min, rect.Max, BlockReason.Building);
        }

        private void OnRoadTileUpdated(RoadTileUpdatedSignal signal)
        {
            if (_world.RoadNetwork.IsRoadTile(signal.Tile))
                _world.BlockMap.AddReason(signal.Tile, BlockReason.Road);
            else
                _world.BlockMap.RemoveReason(signal.Tile, BlockReason.Road);
        }

        private void OnRawResourcesRemoved(RawResourcesRemovedSignal signal)
        {
            foreach (var resourceNode in signal.ResourceNodes) _world.BlockMap.RemoveReason(resourceNode.Tile, BlockReason.RawResource);
        }
    }
}