using Unity.Burst;
using Unity.Entities;
using Unity.Jobs;
using Unity.Collections;
using Unity.NetCode;
using Unity.Mathematics;

namespace Asteroids.Server
{
    public struct LevelRequestedTag : IComponentData
    {
    }

    [BurstCompile]
    [WorldSystemFilter(WorldSystemFilterFlags.ServerSimulation)]
    [UpdateBefore(typeof(RpcSystem))]
    public partial struct LoadLevelSystem : ISystem
    {
        private EntityQuery m_LevelGroup;
        private PortableFunctionPointer<GhostImportance.ScaleImportanceDelegate> m_ScaleFunctionPointer;

        public void OnCreate(ref SystemState state)
        {
            var builder = new EntityQueryBuilder(Allocator.Temp).WithAllRW<LevelComponent>();
            m_LevelGroup = state.GetEntityQuery(builder);

            state.RequireForUpdate<ServerSettings>();
            m_ScaleFunctionPointer = GhostDistanceImportance.ScaleFunctionPointer;
        }

        [BurstCompile]
        public void OnUpdate(ref SystemState state)
        {
            var settings = SystemAPI.GetSingleton<ServerSettings>();
            if (!SystemAPI.HasSingleton<GhostImportance>())
            {
                // Try to store a bit less than full chunks to avoid fragmenting the data too much
                var maxAsteroidsPerTile = 25;
                var minTileSize = 256;
                float asteroidsPerPx = (float)settings.levelData.numAsteroids / (float)(settings.levelData.levelWidth*settings.levelData.levelHeight);
                // We want to make sure that asteroidsPerPx * tileSize * tileSize = maxAsteroidsPerTile
                int tileSize = math.max(minTileSize, (int)math.ceil(math.sqrt((float)maxAsteroidsPerTile / asteroidsPerPx)));

                var gridSingleton = state.EntityManager.CreateSingleton(new GhostDistanceData
                {
                    TileSize = new int3(tileSize, tileSize, 256),
                    TileCenter = new int3(0, 0, 128),
                    TileBorderWidth = new float3(1f, 1f, 1f),
                });
                state.EntityManager.AddComponentData(gridSingleton, new GhostImportance
                {
                    ScaleImportanceFunction = m_ScaleFunctionPointer,
                    GhostConnectionComponentType = ComponentType.ReadOnly<GhostConnectionPosition>(),
                    GhostImportanceDataType = ComponentType.ReadOnly<GhostDistanceData>(),
                    GhostImportancePerChunkDataType = ComponentType.ReadOnly<GhostDistancePartitionShared>(),
                });
            }

            if (m_LevelGroup.IsEmptyIgnoreFilter)
            {
                var newLevel = state.EntityManager.CreateEntity();
                state.EntityManager.AddComponentData(newLevel, settings.levelData);
                return;
            }

            var level = m_LevelGroup.ToComponentDataListAsync<LevelComponent>(state.WorldUpdateAllocator,
                out var levelHandle);

            var levelJob = new LoadLevelJob
            {
                ecb = SystemAPI.GetSingleton<EndSimulationEntityCommandBufferSystem.Singleton>()
                .CreateCommandBuffer(state.WorldUnmanaged),
                level = level,
            };
            state.Dependency = JobHandle.CombineDependencies(state.Dependency, levelHandle);
            levelJob.Schedule();
        }

        [BurstCompile]
        [WithNone(typeof(LevelRequestedTag))]
        internal partial struct LoadLevelJob : IJobEntity
        {
            public EntityCommandBuffer ecb;
            [ReadOnly] public NativeList<LevelComponent> level;

            public void Execute(Entity entity, in NetworkId networkId)
            {
                ecb.AddComponent(entity, new LevelRequestedTag());
                var req = ecb.CreateEntity();
                ecb.AddComponent(req, new LevelLoadRequest
                {
                    levelData = level[0],
                });
                ecb.AddComponent(req, new SendRpcCommandRequest {TargetConnection = entity});
            }
        }
    }

}