#include "World.hpp"
#include "ContentManager.hpp"
#include "TOSLib.hpp"
#include <memory>
#include <unordered_set>


namespace LV::Server {


World::World(DefWorldId defId)
    : DefId(defId)
{

}

World::~World() {

}

std::vector<Pos::GlobalRegion> World::onRemoteClient_RegionsEnter(WorldId_t worldId, std::shared_ptr<RemoteClient> cec, const std::vector<Pos::GlobalRegion>& enter) {
    std::vector<Pos::GlobalRegion> out;

    for(const Pos::GlobalRegion &pos : enter) {
        auto iterRegion = Regions.find(pos);
        if(iterRegion == Regions.end()) {
            out.push_back(pos);
            continue;
        }

        auto &region = *iterRegion->second;
        region.RMs.push_back(cec);
        region.NewRMs.push_back(cec);
        // Отправить клиенту информацию о чанках и сущностях
        std::unordered_map<Pos::bvec4u, const std::vector<VoxelCube>*> voxels;
        std::unordered_map<Pos::bvec4u, const Node*> nodes;

        for(auto& [key, value] : region.Voxels) {
            voxels[key] = &value;
        }

        for(int z = 0; z < 4; z++)
            for(int y = 0; y < 4; y++)
                for(int x = 0; x < 4; x++) {
                    nodes[Pos::bvec4u(x, y, z)] = region.Nodes[Pos::bvec4u(x, y, z).pack()].data();
                }

        if(!region.Entityes.empty()) {
            std::vector<std::tuple<ServerEntityId_t, const Entity*>> updates;
            updates.reserve(region.Entityes.size());

            for(size_t iter = 0; iter < region.Entityes.size(); iter++) {
                const Entity& entity = region.Entityes[iter];
                if(entity.IsRemoved)
                    continue;

                ServerEntityId_t entityId = {worldId, pos, static_cast<RegionEntityId_t>(iter)};
                updates.emplace_back(entityId, &entity);
            }

            if(!updates.empty())
                cec->prepareEntitiesUpdate(updates);
        }
    }

    return out;
}

void World::onRemoteClient_RegionsLost(WorldId_t worldId, std::shared_ptr<RemoteClient> cec, const std::vector<Pos::GlobalRegion> &lost) {
    for(const Pos::GlobalRegion &pos : lost) {
        auto region = Regions.find(pos);
        if(region == Regions.end())
            continue;

        if(!region->second->Entityes.empty()) {
            std::vector<ServerEntityId_t> removed;
            removed.reserve(region->second->Entityes.size());

            for(size_t iter = 0; iter < region->second->Entityes.size(); iter++) {
                const Entity& entity = region->second->Entityes[iter];
                if(entity.IsRemoved)
                    continue;

                removed.emplace_back(worldId, pos, static_cast<RegionEntityId_t>(iter));
            }

            if(!removed.empty())
                cec->prepareEntitiesRemove(removed);
        }

        std::vector<std::shared_ptr<RemoteClient>> &CECs = region->second->RMs;
        for(size_t iter = 0; iter < CECs.size(); iter++) {
            if(CECs[iter] == cec) {
                CECs.erase(CECs.begin()+iter);
                break;
            }
        }
    }
}

World::SaveUnloadInfo World::onStepDatabaseSync(ContentManager& cm, float dtime) {
    SaveUnloadInfo out;

    constexpr float kSaveDelay = 15.0f;
    constexpr float kUnloadDelay = 15.0f;

    std::vector<Pos::GlobalRegion> toErase;
    toErase.reserve(16);

    for(auto& [pos, regionPtr] : Regions) {
        Region& region = *regionPtr;

        region.LastSaveTime += dtime;

        const bool hasChanges = region.IsChanged || region.IsChunkChanged_Voxels || region.IsChunkChanged_Nodes;
        const bool needToSave = hasChanges && region.LastSaveTime > kSaveDelay;
        const bool needToUnload = region.RMs.empty() && region.LastSaveTime > kUnloadDelay;

        if(needToSave || needToUnload) {
            SB_Region_In data;
            data.Voxels = region.Voxels;
            data.Nodes = region.Nodes;

            data.Entityes.reserve(region.Entityes.size());
            for(const Entity& entity : region.Entityes) {
                if(entity.IsRemoved || entity.NeedRemove)
                    continue;
                data.Entityes.push_back(entity);
            }

            std::unordered_set<DefVoxelId> voxelIds;
            for(const auto& [chunkPos, voxels] : region.Voxels) {
                (void) chunkPos;
                for(const VoxelCube& cube : voxels)
                    voxelIds.insert(cube.VoxelId);
            }

            std::unordered_set<DefNodeId> nodeIds;
            for(const auto& chunk : region.Nodes) {
                for(const Node& node : chunk)
                    nodeIds.insert(node.NodeId);
            }

            std::unordered_set<DefEntityId> entityIds;
            for(const Entity& entity : data.Entityes)
                entityIds.insert(entity.getDefId());

            data.VoxelsMap.reserve(voxelIds.size());
            for(DefVoxelId id : voxelIds) {
                auto dk = cm.getDK(EnumDefContent::Voxel, id);
                if(!dk)
                    continue;
                data.VoxelsMap.emplace_back(id, dk->Domain + ":" + dk->Key);
            }

            data.NodeMap.reserve(nodeIds.size());
            for(DefNodeId id : nodeIds) {
                auto dk = cm.getDK(EnumDefContent::Node, id);
                if(!dk)
                    continue;
                data.NodeMap.emplace_back(id, dk->Domain + ":" + dk->Key);
            }

            data.EntityMap.reserve(entityIds.size());
            for(DefEntityId id : entityIds) {
                auto dk = cm.getDK(EnumDefContent::Entity, id);
                if(!dk)
                    continue;
                data.EntityMap.emplace_back(id, dk->Domain + ":" + dk->Key);
            }

            out.ToSave.push_back({pos, std::move(data)});

            region.LastSaveTime = 0.0f;
            region.IsChanged = false;
        }

        if(needToUnload) {
            out.ToUnload.push_back(pos);
            toErase.push_back(pos);
        }
    }

    for(const Pos::GlobalRegion& pos : toErase) {
        Regions.erase(pos);
    }

    return out;
}

void World::pushRegions(std::vector<std::pair<Pos::GlobalRegion, RegionIn>> regions) {
    for(auto& [key, value] : regions) {
        Region &region = *(Regions[key] = std::make_unique<Region>());
        region.Voxels = std::move(value.Voxels);
        region.Nodes = value.Nodes;
        region.Entityes = std::move(value.Entityes);
    }
}

void World::onUpdate(GameServer *server, float dtime) {
    
}

}