Examples with Tag com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag used on opensource projects

Example 6 with Tag

use of com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag in project ViaBackwards by ViaVersion.

the class BlockItemPackets1_13 method registerPackets.

@Override
protected void registerPackets() {
    protocol.registerClientbound(ClientboundPackets1_13.COOLDOWN, new PacketRemapper() {

        @Override
        public void registerMap() {
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int itemId = wrapper.read(Type.VAR_INT);
                    int oldId = protocol.getMappingData().getItemMappings().get(itemId);
                    if (oldId != -1) {
                        Optional<String> eggEntityId = SpawnEggRewriter.getEntityId(oldId);
                        if (eggEntityId.isPresent()) {
                            itemId = 383 << 16;
                        } else {
                            itemId = (oldId >> 4) << 16 | oldId & 0xF;
                        }
                    }
                    wrapper.write(Type.VAR_INT, itemId);
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.BLOCK_ACTION, new PacketRemapper() {

        @Override
        public void registerMap() {
            // Location
            map(Type.POSITION);
            // Action Id
            map(Type.UNSIGNED_BYTE);
            // Action param
            map(Type.UNSIGNED_BYTE);
            // Block Id - /!\ NOT BLOCK STATE ID
            map(Type.VAR_INT);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int blockId = wrapper.get(Type.VAR_INT, 0);
                    if (blockId == 73)
                        blockId = 25;
                    else if (blockId == 99)
                        blockId = 33;
                    else if (blockId == 92)
                        blockId = 29;
                    else if (blockId == 142)
                        blockId = 54;
                    else if (blockId == 305)
                        blockId = 146;
                    else if (blockId == 249)
                        blockId = 130;
                    else if (blockId == 257)
                        blockId = 138;
                    else if (blockId == 140)
                        blockId = 52;
                    else if (blockId == 472)
                        blockId = 209;
                    else if (blockId >= 483 && blockId <= 498)
                        blockId = blockId - 483 + 219;
                    wrapper.set(Type.VAR_INT, 0, blockId);
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.BLOCK_ENTITY_DATA, new PacketRemapper() {

        @Override
        public void registerMap() {
            // 0 - Position
            map(Type.POSITION);
            // 1 - Action
            map(Type.UNSIGNED_BYTE);
            // 2 - NBT Data
            map(Type.NBT);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    BackwardsBlockEntityProvider provider = Via.getManager().getProviders().get(BackwardsBlockEntityProvider.class);
                    // TODO conduit handling
                    if (wrapper.get(Type.UNSIGNED_BYTE, 0) == 5) {
                        wrapper.cancel();
                    }
                    wrapper.set(Type.NBT, 0, provider.transform(wrapper.user(), wrapper.get(Type.POSITION, 0), wrapper.get(Type.NBT, 0)));
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.UNLOAD_CHUNK, new PacketRemapper() {

        @Override
        public void registerMap() {
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int chunkMinX = wrapper.passthrough(Type.INT) << 4;
                    int chunkMinZ = wrapper.passthrough(Type.INT) << 4;
                    int chunkMaxX = chunkMinX + 15;
                    int chunkMaxZ = chunkMinZ + 15;
                    BackwardsBlockStorage blockStorage = wrapper.user().get(BackwardsBlockStorage.class);
                    blockStorage.getBlocks().entrySet().removeIf(entry -> {
                        Position position = entry.getKey();
                        return position.getX() >= chunkMinX && position.getZ() >= chunkMinZ && position.getX() <= chunkMaxX && position.getZ() <= chunkMaxZ;
                    });
                }
            });
        }
    });
    // Block Change
    protocol.registerClientbound(ClientboundPackets1_13.BLOCK_CHANGE, new PacketRemapper() {

        @Override
        public void registerMap() {
            // 0 - Position
            map(Type.POSITION);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int blockState = wrapper.read(Type.VAR_INT);
                    Position position = wrapper.get(Type.POSITION, 0);
                    // Store blocks
                    BackwardsBlockStorage storage = wrapper.user().get(BackwardsBlockStorage.class);
                    storage.checkAndStore(position, blockState);
                    wrapper.write(Type.VAR_INT, protocol.getMappingData().getNewBlockStateId(blockState));
                    // Flower pot special treatment
                    flowerPotSpecialTreatment(wrapper.user(), blockState, position);
                }
            });
        }
    });
    // Multi Block Change
    protocol.registerClientbound(ClientboundPackets1_13.MULTI_BLOCK_CHANGE, new PacketRemapper() {

        @Override
        public void registerMap() {
            // 0 - Chunk X
            map(Type.INT);
            // 1 - Chunk Z
            map(Type.INT);
            map(Type.BLOCK_CHANGE_RECORD_ARRAY);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    BackwardsBlockStorage storage = wrapper.user().get(BackwardsBlockStorage.class);
                    for (BlockChangeRecord record : wrapper.get(Type.BLOCK_CHANGE_RECORD_ARRAY, 0)) {
                        int chunkX = wrapper.get(Type.INT, 0);
                        int chunkZ = wrapper.get(Type.INT, 1);
                        int block = record.getBlockId();
                        Position position = new Position(record.getSectionX() + (chunkX * 16), record.getY(), record.getSectionZ() + (chunkZ * 16));
                        // Store if needed
                        storage.checkAndStore(position, block);
                        // Flower pot special treatment
                        flowerPotSpecialTreatment(wrapper.user(), block, position);
                        // Change to old id
                        record.setBlockId(protocol.getMappingData().getNewBlockStateId(block));
                    }
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.WINDOW_ITEMS, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.UNSIGNED_BYTE);
            map(Type.FLAT_ITEM_ARRAY, Type.ITEM_ARRAY);
            handler(itemArrayHandler(Type.ITEM_ARRAY));
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.SET_SLOT, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.UNSIGNED_BYTE);
            map(Type.SHORT);
            map(Type.FLAT_ITEM, Type.ITEM);
            handler(itemToClientHandler(Type.ITEM));
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.CHUNK_DATA, new PacketRemapper() {

        @Override
        public void registerMap() {
            handler(wrapper -> {
                ClientWorld clientWorld = wrapper.user().get(ClientWorld.class);
                Chunk1_9_3_4Type type_old = new Chunk1_9_3_4Type(clientWorld);
                Chunk1_13Type type = new Chunk1_13Type(clientWorld);
                Chunk chunk = wrapper.read(type);
                // Handle Block Entities before block rewrite
                BackwardsBlockEntityProvider provider = Via.getManager().getProviders().get(BackwardsBlockEntityProvider.class);
                BackwardsBlockStorage storage = wrapper.user().get(BackwardsBlockStorage.class);
                for (CompoundTag tag : chunk.getBlockEntities()) {
                    Tag idTag = tag.get("id");
                    if (idTag == null)
                        continue;
                    String id = (String) idTag.getValue();
                    // Ignore if we don't handle it
                    if (!provider.isHandled(id))
                        continue;
                    int sectionIndex = ((NumberTag) tag.get("y")).asInt() >> 4;
                    if (sectionIndex < 0 || sectionIndex > 15) {
                        // 1.17 chunks
                        continue;
                    }
                    ChunkSection section = chunk.getSections()[sectionIndex];
                    int x = ((NumberTag) tag.get("x")).asInt();
                    int y = ((NumberTag) tag.get("y")).asInt();
                    int z = ((NumberTag) tag.get("z")).asInt();
                    Position position = new Position(x, (short) y, z);
                    int block = section.getFlatBlock(x & 0xF, y & 0xF, z & 0xF);
                    storage.checkAndStore(position, block);
                    provider.transform(wrapper.user(), position, tag);
                }
                // Rewrite new blocks to old blocks
                for (int i = 0; i < chunk.getSections().length; i++) {
                    ChunkSection section = chunk.getSections()[i];
                    if (section == null) {
                        continue;
                    }
                    // Flower pots require a special treatment, they are no longer block entities :(
                    for (int y = 0; y < 16; y++) {
                        for (int z = 0; z < 16; z++) {
                            for (int x = 0; x < 16; x++) {
                                int block = section.getFlatBlock(x, y, z);
                                // Check if the block is a flower
                                if (FlowerPotHandler.isFlowah(block)) {
                                    Position pos = new Position((x + (chunk.getX() << 4)), (short) (y + (i << 4)), (z + (chunk.getZ() << 4)));
                                    // Store block
                                    storage.checkAndStore(pos, block);
                                    CompoundTag nbt = provider.transform(wrapper.user(), pos, "minecraft:flower_pot");
                                    chunk.getBlockEntities().add(nbt);
                                }
                            }
                        }
                    }
                    for (int p = 0; p < section.getPaletteSize(); p++) {
                        int old = section.getPaletteEntry(p);
                        if (old != 0) {
                            int oldId = protocol.getMappingData().getNewBlockStateId(old);
                            section.setPaletteEntry(p, oldId);
                        }
                    }
                }
                if (chunk.isBiomeData()) {
                    for (int i = 0; i < 256; i++) {
                        int biome = chunk.getBiomeData()[i];
                        int newId = -1;
                        switch(biome) {
                            // end biomes
                            case 40:
                            case 41:
                            case 42:
                            case 43:
                                newId = 9;
                                break;
                            // deep ocean biomes
                            case 47:
                            case 48:
                            case 49:
                                newId = 24;
                                break;
                            case // deep frozen... let's just pick the frozen variant
                            50:
                                newId = 10;
                                break;
                            // the other new ocean biomes
                            case 44:
                            case 45:
                            case 46:
                                newId = 0;
                                break;
                        }
                        if (newId != -1) {
                            chunk.getBiomeData()[i] = newId;
                        }
                    }
                }
                wrapper.write(type_old, chunk);
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.EFFECT, new PacketRemapper() {

        @Override
        public void registerMap() {
            // Effect Id
            map(Type.INT);
            // Location
            map(Type.POSITION);
            // Data
            map(Type.INT);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int id = wrapper.get(Type.INT, 0);
                    int data = wrapper.get(Type.INT, 1);
                    if (id == 1010) {
                        // Play record
                        wrapper.set(Type.INT, 1, protocol.getMappingData().getItemMappings().get(data) >> 4);
                    } else if (id == 2001) {
                        // Block break + block break sound
                        data = protocol.getMappingData().getNewBlockStateId(data);
                        int blockId = data >> 4;
                        int blockData = data & 0xF;
                        wrapper.set(Type.INT, 1, (blockId & 0xFFF) | (blockData << 12));
                    }
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.MAP_DATA, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.VAR_INT);
            map(Type.BYTE);
            map(Type.BOOLEAN);
            handler(new PacketHandler() {

                @Override
                public void handle(PacketWrapper wrapper) throws Exception {
                    int iconCount = wrapper.passthrough(Type.VAR_INT);
                    for (int i = 0; i < iconCount; i++) {
                        int type = wrapper.read(Type.VAR_INT);
                        byte x = wrapper.read(Type.BYTE);
                        byte z = wrapper.read(Type.BYTE);
                        byte direction = wrapper.read(Type.BYTE);
                        if (wrapper.read(Type.BOOLEAN)) {
                            wrapper.read(Type.COMPONENT);
                        }
                        if (type > 9) {
                            wrapper.set(Type.VAR_INT, 1, wrapper.get(Type.VAR_INT, 1) - 1);
                            continue;
                        }
                        wrapper.write(Type.BYTE, (byte) ((type << 4) | (direction & 0x0F)));
                        wrapper.write(Type.BYTE, x);
                        wrapper.write(Type.BYTE, z);
                    }
                }
            });
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.ENTITY_EQUIPMENT, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.VAR_INT);
            map(Type.VAR_INT);
            map(Type.FLAT_ITEM, Type.ITEM);
            handler(itemToClientHandler(Type.ITEM));
        }
    });
    protocol.registerClientbound(ClientboundPackets1_13.WINDOW_PROPERTY, new PacketRemapper() {

        @Override
        public void registerMap() {
            // Window Id
            map(Type.UNSIGNED_BYTE);
            // Property
            map(Type.SHORT);
            // Value
            map(Type.SHORT);
            handler(wrapper -> {
                short property = wrapper.get(Type.SHORT, 0);
                // Enchantment table
                if (property >= 4 && property <= 6) {
                    short oldId = wrapper.get(Type.SHORT, 1);
                    wrapper.set(Type.SHORT, 1, (short) protocol.getMappingData().getEnchantmentMappings().getNewId(oldId));
                }
            });
        }
    });
    protocol.registerServerbound(ServerboundPackets1_12_1.CREATIVE_INVENTORY_ACTION, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.SHORT);
            map(Type.ITEM, Type.FLAT_ITEM);
            handler(itemToServerHandler(Type.FLAT_ITEM));
        }
    });
    protocol.registerServerbound(ServerboundPackets1_12_1.CLICK_WINDOW, new PacketRemapper() {

        @Override
        public void registerMap() {
            map(Type.UNSIGNED_BYTE);
            map(Type.SHORT);
            map(Type.BYTE);
            map(Type.SHORT);
            map(Type.VAR_INT);
            map(Type.ITEM, Type.FLAT_ITEM);
            handler(itemToServerHandler(Type.FLAT_ITEM));
        }
    });
}

Example 7 with Tag

use of com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag in project ViaBackwards by ViaVersion.

the class BlockItemPackets1_12 method handleNbtToServer.

private void handleNbtToServer(CompoundTag compoundTag, CompoundTag backupTag) {
    // Restore the removed long array tags
    for (Map.Entry<String, Tag> entry : backupTag) {
        if (entry.getValue() instanceof CompoundTag) {
            CompoundTag nestedTag = compoundTag.get(entry.getKey());
            handleNbtToServer(nestedTag, (CompoundTag) entry.getValue());
        } else {
            compoundTag.put(entry.getKey(), fromIntArrayTag((IntArrayTag) entry.getValue()));
        }
    }
}

Example 8 with Tag

use of com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag in project ViaBackwards by ViaVersion.

the class ItemRewriterBase method saveListTag.

protected void saveListTag(CompoundTag displayTag, ListTag original, String name) {
    // Multiple places might try to backup data
    String backupName = nbtTagName + "|o" + name;
    if (!displayTag.contains(backupName)) {
        // Clone all tag entries
        ListTag listTag = new ListTag();
        for (Tag tag : original.getValue()) {
            listTag.add(tag.clone());
        }
        displayTag.put(backupName, listTag);
    }
}

Example 9 with Tag

use of com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag in project ViaBackwards by ViaVersion.

the class EnchantmentRewriter method rewriteEnchantmentsToClient.

public void rewriteEnchantmentsToClient(CompoundTag tag, boolean storedEnchant) {
    String key = storedEnchant ? "StoredEnchantments" : "Enchantments";
    ListTag enchantments = tag.get(key);
    List<Tag> loreToAdd = new ArrayList<>();
    boolean changed = false;
    Iterator<Tag> iterator = enchantments.iterator();
    while (iterator.hasNext()) {
        CompoundTag enchantmentEntry = (CompoundTag) iterator.next();
        Tag idTag = enchantmentEntry.get("id");
        if (!(idTag instanceof StringTag))
            continue;
        String enchantmentId = ((StringTag) idTag).getValue();
        String remappedName = enchantmentMappings.get(enchantmentId);
        if (remappedName != null) {
            if (!changed) {
                // Backup original before doing modifications
                itemRewriter.saveListTag(tag, enchantments, key);
                changed = true;
            }
            iterator.remove();
            int level = ((NumberTag) enchantmentEntry.get("lvl")).asInt();
            String loreValue = remappedName + " " + getRomanNumber(level);
            if (jsonFormat) {
                loreValue = ChatRewriter.legacyTextToJsonString(loreValue);
            }
            loreToAdd.add(new StringTag(loreValue));
        }
    }
    if (!loreToAdd.isEmpty()) {
        // Add dummy enchant for the glow effect if there are no actual enchantments left
        if (!storedEnchant && enchantments.size() == 0) {
            CompoundTag dummyEnchantment = new CompoundTag();
            dummyEnchantment.put("id", new StringTag());
            dummyEnchantment.put("lvl", new ShortTag((short) 0));
            enchantments.add(dummyEnchantment);
        }
        CompoundTag display = tag.get("display");
        if (display == null) {
            tag.put("display", display = new CompoundTag());
        }
        ListTag loreTag = display.get("Lore");
        if (loreTag == null) {
            display.put("Lore", loreTag = new ListTag(StringTag.class));
        } else {
            // Save original lore
            itemRewriter.saveListTag(display, loreTag, "Lore");
        }
        loreToAdd.addAll(loreTag.getValue());
        loreTag.setValue(loreToAdd);
    }
}

Example 10 with Tag

use of com.viaversion.viaversion.libs.opennbt.tag.builtin.Tag in project ViaBackwards by ViaVersion.

the class BlockItemPackets1_12 method handleNbtToClient.

private boolean handleNbtToClient(CompoundTag compoundTag, CompoundTag backupTag) {
    // Long array tags were introduced in 1.12 - just remove them
    // Only save the removed tags instead of blindly copying the entire nbt again
    Iterator<Map.Entry<String, Tag>> iterator = compoundTag.iterator();
    boolean hasLongArrayTag = false;
    while (iterator.hasNext()) {
        Map.Entry<String, Tag> entry = iterator.next();
        if (entry.getValue() instanceof CompoundTag) {
            CompoundTag nestedBackupTag = new CompoundTag();
            backupTag.put(entry.getKey(), nestedBackupTag);
            hasLongArrayTag |= handleNbtToClient((CompoundTag) entry.getValue(), nestedBackupTag);
        } else if (entry.getValue() instanceof LongArrayTag) {
            backupTag.put(entry.getKey(), fromLongArrayTag((LongArrayTag) entry.getValue()));
            iterator.remove();
            hasLongArrayTag = true;
        }
    }
    return hasLongArrayTag;
}