Merge remote-tracking branch 'origin/master' into pre116

4 files changed, 135 insertions, 57 deletions

diff --git a/src/main/java/re/jag/parquet/CustomStats.java b/src/main/java/re/jag/parquet/CustomStats.java
new file mode 100644
index 0000000..c9d69b8
--- /dev/null
+++ b/src/main/java/re/jag/parquet/CustomStats.java
@@ -0,0 +1,14 @@
+package re.jag.parquet;
+
+import net.minecraft.item.Item;
+import net.minecraft.stat.StatType;
+import net.minecraft.util.Identifier;
+import net.minecraft.util.registry.Registry;
+
+public class CustomStats {
+	public static final StatType<Item> TRADED;
+
+	static {
+		TRADED = (StatType<Item>) Registry.STAT_TYPE.get(new Identifier("traded"));
+	}
+}
diff --git a/src/main/java/re/jag/parquet/commands/Savedata.java b/src/main/java/re/jag/parquet/commands/Savedata.java
index 3a8e6d6..11978ef 100644
--- a/src/main/java/re/jag/parquet/commands/Savedata.java
+++ b/src/main/java/re/jag/parquet/commands/Savedata.java
@@ -1,12 +1,13 @@
 package re.jag.parquet.commands;
 
 import java.io.File;
-import java.util.UUID;
+import java.util.*;
 
 import static net.minecraft.server.command.CommandManager.literal;
 import static net.minecraft.server.command.CommandSource.suggestMatching;
 import static net.minecraft.server.command.CommandManager.argument;
 
+import com.google.common.collect.Lists;
 import com.mojang.brigadier.CommandDispatcher;
 import com.mojang.brigadier.builder.LiteralArgumentBuilder;
 import com.mojang.brigadier.arguments.FloatArgumentType;
@@ -16,10 +17,6 @@ import com.mojang.authlib.GameProfile;
 import net.minecraft.server.command.ServerCommandSource;
 import net.minecraft.server.network.ServerPlayerEntity;
 import net.minecraft.command.arguments.ObjectiveArgumentType;
-import net.minecraft.command.arguments.ObjectiveCriteriaArgumentType;
-import net.minecraft.entity.EntityType;
-import net.minecraft.item.Item;
-import net.minecraft.scoreboard.ScoreboardCriterion;
 import net.minecraft.scoreboard.ScoreboardObjective;
 import net.minecraft.scoreboard.ScoreboardPlayerScore;
 import net.minecraft.scoreboard.ServerScoreboard;
@@ -31,7 +28,6 @@ import re.jag.parquet.Parquet;
 import net.minecraft.stat.ServerStatHandler;
 import net.minecraft.stat.Stat;
 import net.minecraft.stat.StatType;
-import net.minecraft.block.Block;
 
 public class Savedata {
 	public static void register(CommandDispatcher<ServerCommandSource> dispatcher) {
@@ -46,23 +42,24 @@ public class Savedata {
 	                then(literal("stats").
 	                		then(argument("player", StringArgumentType.word()).
 	                				suggests( (c, b) -> suggestMatching(c.getSource().getPlayerNames() , b)).
-			                		then(argument("criteria", ObjectiveCriteriaArgumentType.objectiveCriteria()).
+			                		then(argument("criteria", StringArgumentType.string()).
+							                suggests((c,b) -> suggestMatching(get_stat_list(), b)).
 			                				executes((c) -> print_player_stat (
 			                						c.getSource(),
 			                						StringArgumentType.getString(c, "player"),
-			                						ObjectiveCriteriaArgumentType.getCriteria(c, "criteria") 
+			                						StringArgumentType.getString(c, "criteria")
 			                				)).
 			                				then(literal("import").then(argument("score", ObjectiveArgumentType.objective()).
 			                						executes((c) -> import_stat_to_scoreboard (
 			                								c.getSource(), 
-			                								StringArgumentType.getString(c, "player"), 
-			                								ObjectiveCriteriaArgumentType.getCriteria(c, "criteria"), 
+			                								StringArgumentType.getString(c, "player"),
+											                StringArgumentType.getString(c, "criteria"),
 			                								ObjectiveArgumentType.getWritableObjective(c, "score"), 1)).
 			                						then(argument("multiplier", FloatArgumentType.floatArg()).
 			                								executes((c) ->  import_stat_to_scoreboard(
 			                										c.getSource(), 
-			                										StringArgumentType.getString(c, "player"), 
-			                										ObjectiveCriteriaArgumentType.getCriteria(c, "criteria"), 
+			                										StringArgumentType.getString(c, "player"),
+													                StringArgumentType.getString(c, "criteria"),
 			                										ObjectiveArgumentType.getWritableObjective(c, "score"), 
 			                										FloatArgumentType.getFloat(c, "multiplier"))
 			                								)
@@ -78,7 +75,52 @@ public class Savedata {
 		source.sendFeedback(new LiteralText("Hölp goes here"), false);
 		return 1;
 	}
-	
+
+	private static int test(ServerCommandSource source,  String string) {
+		source.sendFeedback(new LiteralText(string),false);
+		return 1;
+	}
+
+	private static List<String> get_stat_list() {
+		List<String> ret = Lists.newArrayList();
+		Iterator stat_type_iter = Registry.STAT_TYPE.iterator();
+
+		while(stat_type_iter.hasNext()) {
+			StatType<Object> statType = (StatType)stat_type_iter.next();
+			Iterator stat_iter = statType.getRegistry().iterator();
+
+			while(stat_iter.hasNext()) {
+				Object object = stat_iter.next();
+				String stat_name = "\"" + Stat.getName(statType, object) + "\"";
+				ret.add(stat_name);
+			}
+		}
+
+		return ret;
+	}
+
+	private static Stat<?> get_stat_from_string(String argument) {
+
+		int criteria_name_seperator = argument.indexOf(':');
+
+		String registry_name="";
+		String stat_name="";
+
+		if (criteria_name_seperator < 0) {
+			return null;
+		} else {
+			registry_name = argument.substring(0,criteria_name_seperator);
+			stat_name = argument.substring(criteria_name_seperator + 1, argument.length() );
+		}
+
+		StatType<Object> stat_type = (StatType<Object>) Registry.STAT_TYPE.get(Identifier.splitOn(registry_name, '.'));
+		Object stat_obj = stat_type.getRegistry().get(Identifier.splitOn(stat_name, '.'));
+
+		Stat stat = stat_type.getOrCreateStat(stat_obj);
+
+		return stat;
+	}
+
 	private static int list_local_saves(ServerCommandSource source) {
 		//TODO 1.16 fix
 
@@ -100,7 +142,7 @@ public class Savedata {
 		return 1;
 	}
 	
-	private static int import_stat_to_scoreboard(ServerCommandSource source, String player,ScoreboardCriterion stat, ScoreboardObjective objective, float multiplier) {
+	private static int import_stat_to_scoreboard(ServerCommandSource source, String player,String stat, ScoreboardObjective objective, float multiplier) {
 		int score = get_player_stat(source, player, stat);
 		if (score < 0)
 			return 0;
@@ -115,7 +157,7 @@ public class Savedata {
 		return 1;
 	}
 	
-	public static ServerStatHandler get_player_stat_handler(ServerCommandSource source, String player_name) {
+	private static ServerStatHandler get_player_stat_handler(ServerCommandSource source, String player_name) {
 		ServerPlayerEntity player = source.getMinecraftServer().getPlayerManager().getPlayer(player_name);
 		if (player != null) {
 			return player.getStatHandler();
@@ -148,7 +190,7 @@ public class Savedata {
 		*/
 	}
 
-	private static int print_player_stat(ServerCommandSource source, String player, ScoreboardCriterion stat) {
+	private static int print_player_stat(ServerCommandSource source, String player, String stat) {
 		Integer stat_value = get_player_stat(source, player, stat);
 		
 		if (stat_value >= 0)
@@ -159,54 +201,21 @@ public class Savedata {
 		return 0;
 	}
 	
-	private static int get_player_stat(ServerCommandSource source, String player_name, ScoreboardCriterion criteria) {
+	private static int get_player_stat(ServerCommandSource source, String player_name, String stat_name) {
 		ServerStatHandler stat_handler = get_player_stat_handler(source, player_name);
 		
 		if (stat_handler == null) {
 			source.sendError(new LiteralText("Failed to get Statistics for Player " + player_name));
 			return -1;
 		}
-		
-		String criteria_name = criteria.getName();
-		int criteria_name_seperator = criteria_name.indexOf(':');
-		
-		String registry_name="";
-		String stat_name="";
-		
-		if (criteria_name_seperator < 0) {
-			source.sendError(new LiteralText("Expected valid registry and stat ID"));
-			return -1;
-		} else {
-			registry_name = criteria_name.substring(0,criteria_name_seperator);
-			stat_name = criteria_name.substring(criteria_name_seperator + 1, criteria_name.length() );
-		}
-		
-		StatType<?> stat_type = Registry.STAT_TYPE.get(Identifier.splitOn(registry_name, '.'));
-		Object stat_obj = stat_type.getRegistry().get(Identifier.splitOn(stat_name, '.'));
-		
-		if (stat_obj instanceof Block) {
-			@SuppressWarnings("unchecked")
-			StatType<Block> stat_type_block = (StatType<Block>) stat_type;
-			Stat<Block> stat = stat_type_block.getOrCreateStat((Block) stat_obj);
-			return stat_handler.getStat(stat);
-		} else if (stat_obj instanceof Item) {
-			@SuppressWarnings("unchecked")
-			StatType<Item> stat_type_item = (StatType<Item>) stat_type;
-			Stat<Item> stat = stat_type_item.getOrCreateStat((Item) stat_obj);
-			return stat_handler.getStat(stat);	
-		} else if (stat_obj instanceof Identifier) {
-			@SuppressWarnings("unchecked")
-			StatType<Identifier> stat_type_ident = (StatType<Identifier>) stat_type;
-			Stat<Identifier> stat = stat_type_ident.getOrCreateStat((Identifier) stat_obj);
-			return stat_handler.getStat(stat);
-		} else if (stat_obj instanceof EntityType<?>) {
-			@SuppressWarnings("unchecked")
-			StatType<EntityType<?>> stat_type_ident = (StatType<EntityType<?>>) stat_type;
-			Stat<EntityType<?>> stat = stat_type_ident.getOrCreateStat((EntityType<?>) stat_obj);
-			return stat_handler.getStat(stat);
-		} else {
-			source.sendError(new LiteralText("Unknown Object " + stat_obj.getClass().getName()));
+
+		Stat stat = get_stat_from_string(stat_name);
+
+		if (stat == null) {
+			source.sendError(new LiteralText("Invalid stat: " + stat_name));
 			return -1;
 		}
+
+		return stat_handler.getStat(stat);
 	}
 }
diff --git a/src/main/java/re/jag/parquet/mixin/AbstractTraderEntityMixin.java b/src/main/java/re/jag/parquet/mixin/AbstractTraderEntityMixin.java
new file mode 100644
index 0000000..654c7cd
--- /dev/null
+++ b/src/main/java/re/jag/parquet/mixin/AbstractTraderEntityMixin.java
@@ -0,0 +1,26 @@
+package re.jag.parquet.mixin;
+
+import net.minecraft.entity.passive.AbstractTraderEntity;
+import net.minecraft.entity.player.PlayerEntity;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.village.TradeOffer;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import re.jag.parquet.CustomStats;
+import re.jag.parquet.Parquet;
+
+@Mixin(AbstractTraderEntity.class)
+public class AbstractTraderEntityMixin {
+	@Shadow
+	PlayerEntity customer;
+
+	@Inject(method="trade", at = @At("RETURN"))
+	private void onTrade(TradeOffer tradeOffer, CallbackInfo ci) {
+		if (this.customer instanceof ServerPlayerEntity) {
+			((ServerPlayerEntity)customer).incrementStat( CustomStats.TRADED.getOrCreateStat( tradeOffer.getMutableSellItem().getItem() ) );
+		}
+	}
+}
diff --git a/src/main/java/re/jag/parquet/mixin/StatsMixin.java b/src/main/java/re/jag/parquet/mixin/StatsMixin.java
new file mode 100644
index 0000000..89463d8
--- /dev/null
+++ b/src/main/java/re/jag/parquet/mixin/StatsMixin.java
@@ -0,0 +1,29 @@
+package re.jag.parquet.mixin;
+
+import net.minecraft.item.Item;
+import net.minecraft.stat.StatFormatter;
+import net.minecraft.stat.StatType;
+import net.minecraft.stat.Stats;
+import net.minecraft.util.Identifier;
+import net.minecraft.util.registry.Registry;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+/**
+ * Has to be a Mixin for the game to load on startup.
+ * */
+
+@Mixin(Stats.class)
+public class StatsMixin {
+	@Shadow
+	private static StatType<Item> registerType(String string, Registry<Item> registry){ return null; };
+
+	@Shadow
+	private static Identifier register(String string, StatFormatter statFormatter) { return  null; }
+
+	private static final StatType<Item> TRADED;
+
+	static {
+		TRADED = registerType("traded", Registry.ITEM);
+	}
+}