Skip to content

Instantly share code, notes, and snippets.

@lucko
Created Sep 10, 2018
Embed
What would you like to do?
package net.minecraft.command.arguments;
import com.google.common.collect.Maps;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.ArgumentType;
import com.mojang.brigadier.tree.ArgumentCommandNode;
import com.mojang.brigadier.tree.CommandNode;
import com.mojang.brigadier.tree.LiteralCommandNode;
import com.mojang.brigadier.tree.RootCommandNode;
import java.util.Collection;
import java.util.Map;
import javax.annotation.Nullable;
import net.minecraft.command.arguments.serializers.BrigadierSerializers;
import net.minecraft.network.PacketBuffer;
import net.minecraft.util.ResourceLocation;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
public class ArgumentTypes {
private static final Logger LOGGER = LogManager.getLogger();
private static final Map < Class<?>, ArgumentTypes.Entry<? >> CLASS_TYPE_MAP = Maps. < Class<?>, ArgumentTypes.Entry<? >> newHashMap();
private static final Map < ResourceLocation, ArgumentTypes.Entry<? >> ID_TYPE_MAP = Maps. < ResourceLocation, ArgumentTypes.Entry<? >> newHashMap();
public static < T extends ArgumentType<? >> void register(ResourceLocation id, Class<T> argumentTypeClass, IArgumentSerializer<T> serializer) {
if (CLASS_TYPE_MAP.containsKey(argumentTypeClass)) {
throw new IllegalArgumentException("Class " + argumentTypeClass.getName() + " already has a serializer!");
} else if (ID_TYPE_MAP.containsKey(id)) {
throw new IllegalArgumentException("'" + id + "' is already a registered serializer!");
} else {
ArgumentTypes.Entry<T> entry = new ArgumentTypes.Entry<T>(argumentTypeClass, serializer, id);
CLASS_TYPE_MAP.put(argumentTypeClass, entry);
ID_TYPE_MAP.put(id, entry);
}
}
public static void registerArgumentTypes() {
BrigadierSerializers.registerArgumentTypes();
register(new ResourceLocation("minecraft:entity"), EntityArgument.class, new EntityArgument.Serializer());
register(new ResourceLocation("minecraft:game_profile"), GameProfileArgument.class, new ArgumentSerializer(GameProfileArgument::gameProfile));
register(new ResourceLocation("minecraft:block_pos"), BlockPosArgument.class, new ArgumentSerializer(BlockPosArgument::blockPos));
register(new ResourceLocation("minecraft:vec3"), Vec3Argument.class, new ArgumentSerializer(Vec3Argument::vec3));
register(new ResourceLocation("minecraft:vec2"), Vec2Argument.class, new ArgumentSerializer(Vec2Argument::vec2));
register(new ResourceLocation("minecraft:block_state"), BlockStateArgument.class, new ArgumentSerializer(BlockStateArgument::blockState));
register(new ResourceLocation("minecraft:block_predicate"), BlockPredicateArgument.class, new ArgumentSerializer(BlockPredicateArgument::blockPredicateArgument));
register(new ResourceLocation("minecraft:item_stack"), ItemArgument.class, new ArgumentSerializer(ItemArgument::itemStack));
register(new ResourceLocation("minecraft:item_predicate"), ItemPredicateArgument.class, new ArgumentSerializer(ItemPredicateArgument::itemPredicate));
register(new ResourceLocation("minecraft:color"), ColorArgument.class, new ArgumentSerializer(ColorArgument::color));
register(new ResourceLocation("minecraft:component"), ComponentArgument.class, new ArgumentSerializer(ComponentArgument::component));
register(new ResourceLocation("minecraft:message"), MessageArgument.class, new ArgumentSerializer(MessageArgument::message));
register(new ResourceLocation("minecraft:nbt"), NBTArgument.class, new ArgumentSerializer(NBTArgument::nbt));
register(new ResourceLocation("minecraft:nbt_path"), NBTPathArgument.class, new ArgumentSerializer(NBTPathArgument::nbtPath));
register(new ResourceLocation("minecraft:objective"), ObjectiveArgument.class, new ArgumentSerializer(ObjectiveArgument::objective));
register(new ResourceLocation("minecraft:objective_criteria"), ObjectiveCriteriaArgument.class, new ArgumentSerializer(ObjectiveCriteriaArgument::objectiveCriteria));
register(new ResourceLocation("minecraft:operation"), OperationArgument.class, new ArgumentSerializer(OperationArgument::operation));
register(new ResourceLocation("minecraft:particle"), ParticleArgument.class, new ArgumentSerializer(ParticleArgument::particle));
register(new ResourceLocation("minecraft:rotation"), RotationArgument.class, new ArgumentSerializer(RotationArgument::rotation));
register(new ResourceLocation("minecraft:scoreboard_slot"), ScoreboardSlotArgument.class, new ArgumentSerializer(ScoreboardSlotArgument::scoreboardSlot));
register(new ResourceLocation("minecraft:score_holder"), ScoreHolderArgument.class, new ScoreHolderArgument.Serializer());
register(new ResourceLocation("minecraft:swizzle"), SwizzleArgument.class, new ArgumentSerializer(SwizzleArgument::swizzle));
register(new ResourceLocation("minecraft:team"), TeamArgument.class, new ArgumentSerializer(TeamArgument::team));
register(new ResourceLocation("minecraft:item_slot"), SlotArgument.class, new ArgumentSerializer(SlotArgument::itemSlot));
register(new ResourceLocation("minecraft:resource_location"), ResourceLocationArgument.class, new ArgumentSerializer(ResourceLocationArgument::resourceLocation));
register(new ResourceLocation("minecraft:mob_effect"), PotionArgument.class, new ArgumentSerializer(PotionArgument::mobEffect));
register(new ResourceLocation("minecraft:function"), FunctionArgument.class, new ArgumentSerializer(FunctionArgument::function));
register(new ResourceLocation("minecraft:entity_anchor"), EntityAnchorArgument.class, new ArgumentSerializer(EntityAnchorArgument::entityAnchor));
register(new ResourceLocation("minecraft:int_range"), RangeArgument.IntRange.class, new RangeArgument.IntRange.Serializer());
register(new ResourceLocation("minecraft:float_range"), RangeArgument.FloatRange.class, new RangeArgument.FloatRange.Serializer());
register(new ResourceLocation("minecraft:item_enchantment"), EnchantmentArgument.class, new ArgumentSerializer(EnchantmentArgument::itemEnchantment));
register(new ResourceLocation("minecraft:entity_summon"), EntitySummonArgument.class, new ArgumentSerializer(EntitySummonArgument::entitySummon));
}
@Nullable
private static ArgumentTypes.Entry<?> get(ResourceLocation id) {
return ID_TYPE_MAP.get(id);
}
@Nullable
private static ArgumentTypes.Entry<?> get(ArgumentType<?> type) {
return CLASS_TYPE_MAP.get(type.getClass());
}
public static < T extends ArgumentType<? >> void serialize(PacketBuffer buffer, T type) {
ArgumentTypes.Entry<T> entry = get(type);
if (entry == null) {
LOGGER.error("Could not serialize {} ({}) - will not be sent to client!", type, type.getClass());
buffer.writeResourceLocation(new ResourceLocation(""));
} else {
buffer.writeResourceLocation(entry.id);
entry.serializer.write(type, buffer);
}
}
@Nullable
public static ArgumentType<?> deserialize(PacketBuffer buffer) {
ResourceLocation resourcelocation = buffer.readResourceLocation();
ArgumentTypes.Entry<?> entry = get(resourcelocation);
if (entry == null) {
LOGGER.error("Could not deserialize {}", (Object)resourcelocation);
return null;
} else {
return entry.serializer.read(buffer);
}
}
private static < T extends ArgumentType<? >> void serialize(JsonObject json, T type) {
ArgumentTypes.Entry<T> entry = get(type);
if (entry == null) {
LOGGER.error("Could not serialize argument {} ({})!", type, type.getClass());
json.addProperty("type", "unknown");
} else {
json.addProperty("type", "argument");
json.addProperty("parser", entry.id.toString());
JsonObject jsonobject = new JsonObject();
entry.serializer.write(type, jsonobject);
if (jsonobject.size() > 0) {
json.add("properties", jsonobject);
}
}
}
public static <S> JsonObject serialize(CommandDispatcher<S> dispatcher, CommandNode<S> node) {
JsonObject jsonobject = new JsonObject();
if (node instanceof RootCommandNode) {
jsonobject.addProperty("type", "root");
} else if (node instanceof LiteralCommandNode) {
jsonobject.addProperty("type", "literal");
} else if (node instanceof ArgumentCommandNode) {
serialize(jsonobject, ((ArgumentCommandNode)node).getType());
} else {
LOGGER.error("Could not serialize node {} ({})!", node, node.getClass());
jsonobject.addProperty("type", "unknown");
}
JsonObject jsonobject1 = new JsonObject();
for (CommandNode<S> commandnode : node.getChildren()) {
jsonobject1.add(commandnode.getName(), serialize(dispatcher, commandnode));
}
if (jsonobject1.size() > 0) {
jsonobject.add("children", jsonobject1);
}
if (node.getCommand() != null) {
jsonobject.addProperty("executable", true);
}
if (node.getRedirect() != null) {
Collection<String> collection = dispatcher.getPath(node.getRedirect());
if (!collection.isEmpty()) {
JsonArray jsonarray = new JsonArray();
for (String s : collection) {
jsonarray.add(s);
}
jsonobject.add("redirect", jsonarray);
}
}
return jsonobject;
}
static class Entry < T extends ArgumentType<? >> {
public final Class<T> argumentClass;
public final IArgumentSerializer<T> serializer;
public final ResourceLocation id;
private Entry(Class<T> argumentClassIn, IArgumentSerializer<T> serializerIn, ResourceLocation idIn) {
this.argumentClass = argumentClassIn;
this.serializer = serializerIn;
this.id = idIn;
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment