package com.customcontent.core.block; import com.customcontent.core.CustomContentCore; import com.cryptomorin.xseries.particles.XParticle; import lombok.Getter; import org.bukkit.Bukkit; import org.bukkit.Chunk; import org.bukkit.Location; import org.bukkit.World; import org.bukkit.block.Block; import java.util.*; import java.util.concurrent.ConcurrentHashMap; public class CustomBlockManager { private final CustomContentCore plugin; @Getter private final Map blocks = new HashMap<>(); private final Map placedBlocks = new ConcurrentHashMap<>(); private final Map> chunkBlocks = new ConcurrentHashMap<>(); @Getter private BlockRegistry blockRegistry; @Getter private CustomBlockPlacer blockPlacer; private int particleTaskId = -1; public CustomBlockManager(CustomContentCore plugin) { this.plugin = plugin; this.blockRegistry = new BlockRegistry(); } public void initialize() { this.blockPlacer = new CustomBlockPlacer(plugin, blockRegistry); } public void registerBlock(CustomBlock block) { this.blocks.put(block.getId(), block); Optional mapping = blockRegistry.allocate(block.getId()); if (mapping.isEmpty()) { plugin.getLogger().severe("Failed to allocate NoteBlock state for block: " + block.getId()); plugin.getLogger().severe("Registry capacity exceeded! Max: " + blockRegistry.getCapacity()); return; } plugin.debug("Registered custom block: " + block.getId() + " with state: " + mapping.get()); if (block.getParticleConfig() != null && block.getParticleConfig().isEnabled()) { ensureParticleTask(); } } public CustomBlock getBlock(String id) { return this.blocks.get(id); } public void registerPlacedBlock(Location location, String blockId) { String key = locationToKey(location); String chunkKey = chunkToKey(location.getChunk()); placedBlocks.put(key, blockId); chunkBlocks.computeIfAbsent(chunkKey, k -> ConcurrentHashMap.newKeySet()).add(key); if (plugin.getConfig().getBoolean("debug.log-block-placements", false)) { plugin.debug("Registered placed block " + blockId + " at " + locationToString(location)); } } public void removeBlock(Location location) { String key = locationToKey(location); String chunkKey = chunkToKey(location.getChunk()); placedBlocks.remove(key); Set chunkSet = chunkBlocks.get(chunkKey); if (chunkSet != null) { chunkSet.remove(key); if (chunkSet.isEmpty()) { chunkBlocks.remove(chunkKey); } } plugin.getBlockStorage().removeBlock(location); } public String getBlockAt(Location location) { String key = locationToKey(location); return placedBlocks.get(key); } public boolean isCustomBlock(Location location) { return getBlockAt(location) != null; } public void loadAllBlocks(CustomBlockStorage storage) { Bukkit.getScheduler().runTaskAsynchronously(plugin, () -> { Map loaded = storage.loadAllBlocks(); Bukkit.getScheduler().runTask(plugin, () -> { for (Map.Entry entry : loaded.entrySet()) { Location loc = entry.getKey(); String blockId = entry.getValue(); if (loc.getWorld() == null || !loc.isChunkLoaded()) { continue; } Block block = loc.getBlock(); Optional mappingOpt = blockRegistry.getMapping(blockId); if (mappingOpt.isPresent()) { NoteBlockStateEncoder.applyState(block, mappingOpt.get()); registerPlacedBlock(loc, blockId); } } plugin.getLogger().info("Loaded " + loaded.size() + " custom blocks from storage"); }); }); } public void loadChunkBlocks(Chunk chunk) { String chunkKey = chunkToKey(chunk); Bukkit.getScheduler().runTaskAsynchronously(plugin, () -> { Map chunkData = new HashMap<>(); for (int x = 0; x < 16; x++) { for (int y = chunk.getWorld().getMinHeight(); y < chunk.getWorld().getMaxHeight(); y++) { for (int z = 0; z < 16; z++) { Location loc = new Location(chunk.getWorld(), chunk.getX() * 16 + x, y, chunk.getZ() * 16 + z); String blockId = plugin.getBlockStorage().getBlock(loc); if (blockId != null) { chunkData.put(loc, blockId); } } } } if (!chunkData.isEmpty()) { Bukkit.getScheduler().runTask(plugin, () -> { for (Map.Entry entry : chunkData.entrySet()) { Location loc = entry.getKey(); String blockId = entry.getValue(); Block block = loc.getBlock(); Optional mappingOpt = blockRegistry.getMapping(blockId); if (mappingOpt.isPresent()) { NoteBlockStateEncoder.applyState(block, mappingOpt.get()); registerPlacedBlock(loc, blockId); } } plugin.debug("Loaded " + chunkData.size() + " blocks in chunk " + chunkKey); }); } }); } public void unloadChunkBlocks(Chunk chunk) { String chunkKey = chunkToKey(chunk); Set blockKeys = chunkBlocks.remove(chunkKey); if (blockKeys != null) { for (String key : blockKeys) { placedBlocks.remove(key); } plugin.debug("Unloaded " + blockKeys.size() + " blocks from chunk " + chunkKey); } } public void clearCache() { this.placedBlocks.clear(); this.chunkBlocks.clear(); } private void ensureParticleTask() { if (particleTaskId != -1) return; particleTaskId = Bukkit.getScheduler().scheduleSyncRepeatingTask(plugin, () -> { for (Map.Entry entry : placedBlocks.entrySet()) { Location location = keyToLocation(entry.getKey()); if (location == null || location.getWorld() == null || !location.isChunkLoaded()) { continue; } CustomBlock block = getBlock(entry.getValue()); if (block == null || block.getParticleConfig() == null || !block.getParticleConfig().isEnabled()) { continue; } XParticle.of(block.getParticleConfig().getType()).ifPresent(xp -> { location.getWorld().spawnParticle( xp.get(), location.clone().add(0.5, 0.5, 0.5), 3, 0.3, 0.3, 0.3, 0.01 ); }); } }, 20L, 20L); } private String locationToKey(Location location) { return location.getWorld().getUID() + "," + location.getBlockX() + "," + location.getBlockY() + "," + location.getBlockZ(); } private String chunkToKey(Chunk chunk) { return chunk.getWorld().getUID() + "," + chunk.getX() + "," + chunk.getZ(); } private Location keyToLocation(String key) { try { String[] parts = key.split(","); World world = Bukkit.getWorld(UUID.fromString(parts[0])); if (world == null) return null; int x = Integer.parseInt(parts[1]); int y = Integer.parseInt(parts[2]); int z = Integer.parseInt(parts[3]); return new Location(world, x, y, z); } catch (Exception e) { return null; } } private String locationToString(Location location) { return String.format("%s: %d, %d, %d", location.getWorld().getName(), location.getBlockX(), location.getBlockY(), location.getBlockZ()); } }