119 lines
4.3 KiB
Java
119 lines
4.3 KiB
Java
/*
|
|
* This file is part of ViaBackwards - https://github.com/ViaVersion/ViaBackwards
|
|
* Copyright (C) 2016-2021 ViaVersion and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package nl.matsv.viabackwards.api.rewriters;
|
|
|
|
import nl.matsv.viabackwards.api.BackwardsProtocol;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import us.myles.ViaVersion.api.minecraft.item.Item;
|
|
import us.myles.viaversion.libs.opennbt.tag.builtin.CompoundTag;
|
|
import us.myles.viaversion.libs.opennbt.tag.builtin.ListTag;
|
|
import us.myles.viaversion.libs.opennbt.tag.builtin.StringTag;
|
|
import us.myles.viaversion.libs.opennbt.tag.builtin.Tag;
|
|
|
|
public abstract class ItemRewriterBase<T extends BackwardsProtocol> extends Rewriter<T> {
|
|
|
|
protected final String nbtTagName;
|
|
protected final boolean jsonNameFormat;
|
|
|
|
protected ItemRewriterBase(T protocol, boolean jsonNameFormat) {
|
|
super(protocol);
|
|
this.jsonNameFormat = jsonNameFormat;
|
|
nbtTagName = "VB|" + protocol.getClass().getSimpleName();
|
|
}
|
|
|
|
@Nullable
|
|
public Item handleItemToClient(Item item) {
|
|
if (item == null) return null;
|
|
if (protocol.getMappingData() != null && protocol.getMappingData().getItemMappings() != null) {
|
|
item.setIdentifier(protocol.getMappingData().getNewItemId(item.getIdentifier()));
|
|
}
|
|
return item;
|
|
}
|
|
|
|
@Nullable
|
|
public Item handleItemToServer(Item item) {
|
|
if (item == null) return null;
|
|
if (protocol.getMappingData() != null && protocol.getMappingData().getItemMappings() != null) {
|
|
item.setIdentifier(protocol.getMappingData().getOldItemId(item.getIdentifier()));
|
|
}
|
|
restoreDisplayTag(item);
|
|
return item;
|
|
}
|
|
|
|
protected boolean hasBackupTag(CompoundTag displayTag, String tagName) {
|
|
return displayTag.contains(nbtTagName + "|o" + tagName);
|
|
}
|
|
|
|
protected void saveStringTag(CompoundTag displayTag, StringTag original, String name) {
|
|
// Multiple places might try to backup data
|
|
String backupName = nbtTagName + "|o" + name;
|
|
if (!displayTag.contains(backupName)) {
|
|
displayTag.put(backupName, new StringTag(original.getValue()));
|
|
}
|
|
}
|
|
|
|
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);
|
|
}
|
|
}
|
|
|
|
protected void restoreDisplayTag(Item item) {
|
|
if (item.getTag() == null) return;
|
|
|
|
CompoundTag display = item.getTag().get("display");
|
|
if (display != null) {
|
|
// Remove custom name / restore original name
|
|
if (display.remove(nbtTagName + "|customName") != null) {
|
|
display.remove("Name");
|
|
} else {
|
|
restoreStringTag(display, "Name");
|
|
}
|
|
|
|
// Restore lore
|
|
restoreListTag(display, "Lore");
|
|
}
|
|
}
|
|
|
|
protected void restoreStringTag(CompoundTag tag, String tagName) {
|
|
StringTag original = tag.remove(nbtTagName + "|o" + tagName);
|
|
if (original != null) {
|
|
tag.put(tagName, new StringTag(original.getValue()));
|
|
}
|
|
}
|
|
|
|
protected void restoreListTag(CompoundTag tag, String tagName) {
|
|
ListTag original = tag.remove(nbtTagName + "|o" + tagName);
|
|
if (original != null) {
|
|
tag.put(tagName, new ListTag(original.getValue()));
|
|
}
|
|
}
|
|
|
|
public String getNbtTagName() {
|
|
return nbtTagName;
|
|
}
|
|
}
|