-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
83a1380
commit 7d75a72
Showing
7 changed files
with
163 additions
and
23 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
134 changes: 134 additions & 0 deletions
134
...ersiontools/reflection/provider/v1_16_R2/data/hook/BukkitContainerAdapterHook1_16_R2.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| package net.sourcewriters.minecraft.versiontools.reflection.provider.v1_16_R2.data.hook; | ||
|
|
||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import java.util.Objects; | ||
| import java.util.function.Function; | ||
|
|
||
| import org.bukkit.craftbukkit.v1_16_R2.entity.CraftEntity; | ||
| import org.bukkit.craftbukkit.v1_16_R2.persistence.CraftPersistentDataContainer; | ||
| import org.bukkit.craftbukkit.v1_16_R2.persistence.CraftPersistentDataTypeRegistry; | ||
| import org.bukkit.persistence.PersistentDataContainer; | ||
|
|
||
| import com.syntaxphoenix.syntaxapi.data.IDataContainer; | ||
| import com.syntaxphoenix.syntaxapi.data.container.nbt.NbtContainer; | ||
| import com.syntaxphoenix.syntaxapi.nbt.NbtCompound; | ||
| import com.syntaxphoenix.syntaxapi.reflection.AbstractReflect; | ||
| import com.syntaxphoenix.syntaxapi.reflection.Reflect; | ||
|
|
||
| import net.minecraft.server.v1_16_R2.NBTTagCompound; | ||
| import net.sourcewriters.minecraft.versiontools.reflection.VersionControl; | ||
| import net.sourcewriters.minecraft.versiontools.reflection.data.WrappedContainer; | ||
| import net.sourcewriters.minecraft.versiontools.reflection.provider.v1_16_R2.data.BukkitContainer1_16_R2; | ||
| import net.sourcewriters.minecraft.versiontools.reflection.provider.v1_16_R2.data.SyntaxContainer1_16_R2; | ||
|
|
||
| @SuppressWarnings({ | ||
| "rawtypes", | ||
| "unchecked" | ||
| }) | ||
| public final class BukkitContainerAdapterHook1_16_R2 { | ||
|
|
||
| private static final BukkitContainerAdapterHook1_16_R2 HOOK = new BukkitContainerAdapterHook1_16_R2(); | ||
|
|
||
| private final AbstractReflect registryRef = new Reflect(CraftPersistentDataTypeRegistry.class) | ||
| .searchMethod("create", "createAdapter", Class.class, Class.class, Function.class, Function.class) | ||
| .searchField("adapters", "adapters") | ||
| .searchField("function", "CREATE_ADAPTER"); | ||
| private final AbstractReflect entityRef = new Reflect(CraftEntity.class).searchField("registry", "DATA_TYPE_REGISTRY"); | ||
|
|
||
| private BukkitContainerAdapterHook1_16_R2() {} | ||
|
|
||
| private final HashMap<CraftPersistentDataTypeRegistry, Function> map = new HashMap<>(); | ||
|
|
||
| private CraftPersistentDataTypeRegistry getEntityRegistry() { | ||
| return (CraftPersistentDataTypeRegistry) entityRef.getFieldValue("registry"); | ||
| } | ||
|
|
||
| private void uninjectAll() { | ||
| for (CraftPersistentDataTypeRegistry registry : map.keySet()) { | ||
| Map adapters = (Map) registryRef.getFieldValue("adapters", registry); | ||
| adapters.remove(BukkitContainer1_16_R2.class); | ||
| adapters.remove(SyntaxContainer1_16_R2.class); | ||
| registryRef.setFieldValue(registry, "function", map.get(registry)); | ||
| } | ||
| map.clear(); | ||
| } | ||
|
|
||
| private void inject(CraftPersistentDataTypeRegistry registry) { | ||
| if (map.containsKey(registry)) { | ||
| return; | ||
| } | ||
| map.put(registry, (Function) registryRef.getFieldValue("function", registry)); | ||
| Function function = clazz -> createAdapter(registry, registryRef.getMethod("create").getReturnType(), (Class) clazz); | ||
| registryRef.setFieldValue(registry, "function", function); | ||
| } | ||
|
|
||
| private <E> E createAdapter(CraftPersistentDataTypeRegistry registry, Class<E> adapterType, Class type) { | ||
| if (Objects.equals(BukkitContainer1_16_R2.class, type)) { | ||
| return (E) buildAdapter(registry, BukkitContainer1_16_R2.class, tag -> fromPrimitiveSyntax(tag)); | ||
| } | ||
| if (Objects.equals(SyntaxContainer1_16_R2.class, type)) { | ||
| return (E) buildAdapter(registry, SyntaxContainer1_16_R2.class, tag -> fromPrimitiveBukkit(registry, tag)); | ||
| } | ||
| return (E) map.get(registry).apply(type); | ||
| } | ||
|
|
||
| private <C extends WrappedContainer> Object buildAdapter(Object handle, Class<C> type, Function<NBTTagCompound, C> function) { | ||
| return registryRef.run(handle, "create", type, NBTTagCompound.class, (Function<C, NBTTagCompound>) input -> toPrimitive(input), | ||
| function); | ||
| } | ||
|
|
||
| private NBTTagCompound toPrimitive(WrappedContainer input) { | ||
| Object handle = findFinalContainer(input).getHandle(); | ||
| if (handle instanceof PersistentDataContainer) { | ||
| if (handle instanceof CraftPersistentDataContainer) { | ||
| return ((CraftPersistentDataContainer) handle).toTagCompound(); | ||
| } | ||
| throw new IllegalArgumentException( | ||
| "Expected 'CraftPersistentDataContainer' got '" + handle.getClass().getSimpleName() + " instead'!"); | ||
| } | ||
| if (handle instanceof IDataContainer) { | ||
| if (handle instanceof NbtContainer) { | ||
| return (NBTTagCompound) VersionControl.get().getBukkitConversion().toMinecraftCompound(((NbtContainer) handle).asNbt()); | ||
| } | ||
| throw new IllegalArgumentException( | ||
| "Expected 'CraftPersistentDataContainer' got '" + handle.getClass().getSimpleName() + " instead'!"); | ||
| } | ||
| throw new IllegalArgumentException("Unknown WrappedContainer implementation!"); | ||
| } | ||
|
|
||
| private BukkitContainer1_16_R2 fromPrimitiveSyntax(NBTTagCompound data) { | ||
| VersionControl control = VersionControl.get(); | ||
| NbtContainer container = new NbtContainer(control.getDataProvider().getRegistry()); | ||
| NbtCompound compound = control.getBukkitConversion().fromMinecraftCompound(data); | ||
| container.fromNbt(compound); | ||
| return new BukkitContainer1_16_R2(container); | ||
| } | ||
|
|
||
| private SyntaxContainer1_16_R2 fromPrimitiveBukkit(CraftPersistentDataTypeRegistry registry, NBTTagCompound data) { | ||
| CraftPersistentDataContainer container = new CraftPersistentDataContainer(registry); | ||
| container.putAll(data); | ||
| return new SyntaxContainer1_16_R2(container); | ||
| } | ||
|
|
||
| private WrappedContainer findFinalContainer(WrappedContainer container) { | ||
| WrappedContainer output = container; | ||
| while (output.getHandle() instanceof WrappedContainer) { | ||
| output = (WrappedContainer) output.getHandle(); | ||
| } | ||
| return output; | ||
| } | ||
|
|
||
| public static void unhookAll() { | ||
| HOOK.uninjectAll(); | ||
| } | ||
|
|
||
| public static void hookEntity() { | ||
| HOOK.inject(HOOK.getEntityRegistry()); | ||
| } | ||
|
|
||
| public static void hook(CraftPersistentDataTypeRegistry registry) { | ||
| HOOK.inject(registry); | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters