Initial Commit

3 files changed, 371 insertions, 0 deletions

diff --git a/src/main/java/com/thatgravyboat/skyblockhud/tracker/KillTrackerHandler.java b/src/main/java/com/thatgravyboat/skyblockhud/tracker/KillTrackerHandler.java
new file mode 100644
index 0000000..6734283
--- /dev/null
+++ b/src/main/java/com/thatgravyboat/skyblockhud/tracker/KillTrackerHandler.java
@@ -0,0 +1,76 @@
+package com.thatgravyboat.skyblockhud.tracker;
+
+import net.minecraft.client.Minecraft;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.util.DamageSource;
+import net.minecraftforge.event.entity.EntityJoinWorldEvent;
+import net.minecraftforge.event.entity.living.LivingDeathEvent;
+import net.minecraftforge.event.entity.player.AttackEntityEvent;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+
+import java.util.HashSet;
+import java.util.Set;
+import java.util.UUID;
+
+public class KillTrackerHandler {
+
+    public static final Set<UUID> attackedEntities = new HashSet<>();
+
+    @SubscribeEvent
+    public void onAttack(AttackEntityEvent event){
+        if (event.target != null) {
+            attackedEntities.add(event.target.getUniqueID());
+        }
+    }
+
+    @SubscribeEvent
+    public void onDeath(LivingDeathEvent event){
+
+        if (false) {
+            //Used for testing
+            System.out.println("----------------------------------------------------------------------------------------------------------------");
+            System.out.println("Name : " + event.entity.getName());
+            System.out.println("UUID : " + event.entity.getUniqueID());
+            NBTTagCompound tag = new NBTTagCompound();
+            event.entity.writeToNBT(tag);
+            System.out.println("Tag : " + tag);
+            System.out.println("Damage : " + getDamageSourceString(event.source));
+            System.out.println("----------------------------------------------------------------------------------------------------------------");
+        }
+
+        attackedEntities.remove(event.entity.getUniqueID());
+    }
+
+    public static String getDamageSourceString(DamageSource source){
+        return "{ " +
+                source.getDamageType() +
+                ", " +
+                source.isDamageAbsolute() +
+                ", " +
+                source.isDifficultyScaled() +
+                ", " +
+                source.isFireDamage() +
+                ", " +
+                source.isProjectile() +
+                ", " +
+                source.isUnblockable() +
+                ", " +
+                source.isExplosion() +
+                ", " +
+                source.isMagicDamage() +
+                ", " +
+                source.isCreativePlayer() +
+                ", " +
+                source.getSourceOfDamage() +
+                " }";
+    }
+
+    @SubscribeEvent
+    public void onWorldChange(EntityJoinWorldEvent event){
+        if (event.entity != null) {
+            if (event.entity.getUniqueID().equals(Minecraft.getMinecraft().thePlayer.getUniqueID())) {
+                attackedEntities.clear();
+            }
+        }
+    }
+}
diff --git a/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerFileLoader.java b/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerFileLoader.java
new file mode 100644
index 0000000..d7f7512
--- /dev/null
+++ b/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerFileLoader.java
@@ -0,0 +1,171 @@
+package com.thatgravyboat.skyblockhud.tracker;
+
+import com.google.gson.*;
+import com.thatgravyboat.skyblockhud.location.Locations;
+import net.minecraft.client.Minecraft;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTBase;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.nbt.NBTTagList;
+import net.minecraft.util.ResourceLocation;
+
+import java.io.*;
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class TrackerFileLoader {
+
+    private static final Gson gson = new GsonBuilder().create();
+
+    public static ItemStack getDisplayItem(JsonObject jsonObject){
+        int meta = jsonObject.get("meta").getAsInt();
+        String displayItemId = jsonObject.get("item").getAsString();
+        Item item = Item.itemRegistry.getObject(new ResourceLocation(displayItemId));
+        ItemStack stack = new ItemStack(item, 0, meta);
+        if (jsonObject.has("skullData") && displayItemId.equals("minecraft:skull") && meta == 3){
+            stack.setTagInfo("SkullOwner",getSkullTag(jsonObject.getAsJsonObject("skullData")));
+        }
+        if (jsonObject.has("enchanted") && jsonObject.get("enchanted").getAsBoolean()) stack.setTagInfo("ench", new NBTTagList());
+        return stack;
+    }
+
+    public static NBTBase getSkullTag(JsonObject skullObject){
+        NBTTagCompound skullOwner = new NBTTagCompound();
+        NBTTagCompound properties = new NBTTagCompound();
+        NBTTagList textures = new NBTTagList();
+        NBTTagCompound value = new NBTTagCompound();
+
+        skullOwner.setString("Id", skullObject.get("id").getAsString());
+
+        value.setString("Value", skullObject.get("texture").getAsString());
+        textures.appendTag(value);
+
+        properties.setTag("textures",textures);
+
+        skullOwner.setTag("Properties", properties);
+        return skullOwner;
+    }
+
+    private static void loadTrackers(JsonObject object){
+        for (JsonElement element : object.get("trackers").getAsJsonArray()) {
+            JsonObject tracker = element.getAsJsonObject();
+            StringBuilder builder = new StringBuilder();
+            tracker.get("location").getAsJsonArray().forEach(loc -> builder.append(loc.getAsString()));
+            String location = builder.toString();
+
+
+            Map<String, ItemStack> stacks = new HashMap<>();
+            for (JsonElement drop :tracker.get("drops").getAsJsonArray()) {
+                JsonObject dropObject = drop.getAsJsonObject();
+
+                //Display Item Creation
+                ItemStack stack = getDisplayItem(dropObject.getAsJsonObject("displayItem"));
+                String itemId = dropObject.get("id").getAsString();
+
+                stacks.put(itemId, stack);
+            }
+
+            String event = tracker.has("event") ? tracker.get("event").getAsString() : null;
+
+            Map<String, Map<String, ItemStack>> events = new HashMap<>();
+            events.put(event, stacks);
+
+            if (TrackerHandler.trackers.containsKey(location)){
+                TrackerHandler.trackers.get(location).dropTrackers.put(event, stacks);
+            }else {
+                TrackerHandler.trackers.putIfAbsent(location, new TrackerHandler.TrackerData(events));
+            }
+
+            tracker.get("location").getAsJsonArray().forEach(loc -> TrackerHandler.trackerIds.put(Locations.get(loc.getAsString()), location));
+        }
+    }
+
+    private static JsonElement getTrackerFile(){
+        List<JsonObject> trackerStats = new ArrayList<>();
+        TrackerHandler.trackers.forEach((locations, trackerData) ->
+            trackerData.dropTrackers.forEach((event, drops) -> {
+                JsonObject jsonObject = new JsonObject();
+                jsonObject.addProperty("location", locations);
+
+                if (event == null) jsonObject.add("event", new JsonNull());
+                else jsonObject.addProperty("event", event);
+
+                JsonObject dropsData = new JsonObject();
+                drops.forEach((s, stack) -> dropsData.addProperty(s, stack.stackSize));
+                jsonObject.add("drops", dropsData);
+                trackerStats.add(jsonObject);
+            }
+        ));
+        JsonArray stats = new JsonArray();
+        trackerStats.forEach(stats::add);
+        return stats;
+    }
+
+    public static void loadTrackersFile(){
+        try {
+            ResourceLocation trackers = new ResourceLocation("skyblockhud:data/trackers.json");
+            InputStream is = Minecraft.getMinecraft().getResourceManager().getResource(trackers).getInputStream();
+
+            try (BufferedReader reader = new BufferedReader(new InputStreamReader(is, StandardCharsets.UTF_8))) {
+                loadTrackers(gson.fromJson(reader, JsonObject.class));
+            }
+        }catch (Exception ignored){}
+    }
+
+    public static boolean loadTrackerStatsFile(File configDirectory){
+        File configFile = new File(configDirectory, "sbh-trackers-stats.json");
+
+        try {
+            if (configFile.createNewFile()){
+                return true;
+            }
+
+            try (BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(configFile), StandardCharsets.UTF_8))) {
+                JsonObject json = gson.fromJson(reader, JsonObject.class);
+                if (json.has("trackerStats")){
+                    json.getAsJsonArray("trackerStats").forEach(element -> {
+                        if (element.isJsonObject()){
+                            JsonObject object = element.getAsJsonObject();
+                            String location = object.get("location").getAsString();
+                            Map<String, Map<String, ItemStack>> trackers = TrackerHandler.trackers.get(location).dropTrackers;
+
+                            JsonElement event = object.get("event");
+                            String eventString = event == null || event.isJsonNull() ? null : event.getAsString();
+                            Map<String, ItemStack> drops = trackers.get(eventString);
+
+                            if (drops != null) {
+                                for (Map.Entry <String ,JsonElement> drop :object.getAsJsonObject("drops").entrySet()) {
+                                    if (drops.containsKey(drop.getKey())) {
+                                        drops.get(drop.getKey()).stackSize = drop.getValue().getAsInt();
+                                    }
+                                }
+                                drops = TrackerHandler.sortTrackers(drops);
+                                trackers.put(eventString, drops);
+                            }
+                        }
+                    });
+                }
+            }
+        } catch(Exception ignored) {}
+        return false;
+    }
+
+    public static void saveTrackerStatsFile(File configDirectory){
+        File configFile = new File(configDirectory, "sbh-trackers-stats.json");
+
+        try {
+            configFile.createNewFile();
+
+            try(BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(configFile), StandardCharsets.UTF_8))) {
+                JsonObject json = new JsonObject();
+                json.add("trackerStats", getTrackerFile());
+                writer.write(gson.toJson(json));
+            }
+        } catch(IOException ignored) {}
+    }
+
+}
diff --git a/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerHandler.java b/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerHandler.java
new file mode 100644
index 0000000..9d8261b
--- /dev/null
+++ b/src/main/java/com/thatgravyboat/skyblockhud/tracker/TrackerHandler.java
@@ -0,0 +1,124 @@
+package com.thatgravyboat.skyblockhud.tracker;
+
+import com.thatgravyboat.skyblockhud.SkyblockHud;
+import com.thatgravyboat.skyblockhud.Utils;
+import com.thatgravyboat.skyblockhud.core.config.Position;
+import com.thatgravyboat.skyblockhud.location.LocationHandler;
+import com.thatgravyboat.skyblockhud.location.Locations;
+import com.thatgravyboat.skyblockhud.seasons.SeasonDateHandler;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.Gui;
+import net.minecraft.client.renderer.GlStateManager;
+import net.minecraft.client.renderer.RenderHelper;
+import net.minecraft.client.renderer.entity.RenderItem;
+import net.minecraft.item.ItemStack;
+import net.minecraftforge.client.event.RenderGameOverlayEvent;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+
+import java.util.*;
+
+public class TrackerHandler {
+
+    public static class TrackerData {
+        public Map<String, Map<String,ItemStack>> dropTrackers;
+
+        public TrackerData(Map<String, Map<String,ItemStack>> trackers) {
+            this.dropTrackers = trackers;
+        }
+
+        public String getDropId(String event){
+            if (event == null || event.isEmpty() || !eventGoing() || !dropTrackers.containsKey(event.toLowerCase().trim())) return null;
+            return event.toLowerCase().trim();
+        }
+
+        private boolean eventGoing(){
+            return SeasonDateHandler.getCurrentEventTime().trim().toLowerCase().contains("ends in");
+        }
+    }
+
+    public static Map<String, TrackerData> trackers = new HashMap<>();
+    public static Map<Locations,String> trackerIds = new HashMap<>();
+
+    public static Map<String, ItemStack> sortTrackers(Map<String, ItemStack> map) {
+        List<Map.Entry<String, ItemStack>> list = new ArrayList<>(map.entrySet());
+        list.sort((entry1, entry2) -> Integer.compare(entry2.getValue().stackSize, entry1.getValue().stackSize));
+
+        Map<String, ItemStack> result = new LinkedHashMap<>();
+        for (Map.Entry<String, ItemStack> entry : list) {
+            result.put(entry.getKey(), entry.getValue());
+        }
+
+        return result;
+    }
+
+    public static void onItemAdded(String id, int amount, String enchant, int level){
+        if (SkyblockHud.hasSkyblockScoreboard() && trackerIds.containsKey(LocationHandler.getCurrentLocation())){
+            String trackerId = trackerIds.get(LocationHandler.getCurrentLocation());
+            TrackerData tracked = trackers.get(trackerId);
+            String dropTrackerId = tracked.getDropId(SeasonDateHandler.getCurrentEvent());
+            Map<String,ItemStack> tracker = tracked.dropTrackers.get(dropTrackerId);
+            String dropId = id;
+            if (enchant != null){
+                dropId = enchant.toUpperCase() + ";" + level;
+            }
+
+            if (tracker != null && tracker.containsKey(dropId)){
+                ItemStack stack = tracker.get(dropId);
+                stack.stackSize += amount;
+                tracked.dropTrackers.put(dropTrackerId, sortTrackers(tracker));
+            }
+        }
+    }
+
+    public static void drawItemStack(ItemStack stack, int x, int y) {
+        if(stack == null)return;
+        RenderItem itemRender = Minecraft.getMinecraft().getRenderItem();
+        RenderHelper.enableGUIStandardItemLighting();
+        itemRender.zLevel = -145;
+        itemRender.renderItemAndEffectIntoGUI(stack, x, y);
+        itemRender.zLevel = 0;
+        RenderHelper.disableStandardItemLighting();
+    }
+
+    @SubscribeEvent
+    public void renderOverlay(RenderGameOverlayEvent.Post event) {
+        if (Utils.overlayShouldRender(event.type, SkyblockHud.hasSkyblockScoreboard(), trackerIds.containsKey(LocationHandler.getCurrentLocation()), !SkyblockHud.config.trackers.hideTracker)) {
+            String trackerId = trackerIds.get(LocationHandler.getCurrentLocation());
+            Minecraft mc = Minecraft.getMinecraft();
+            TrackerData tracked = trackers.get(trackerId);
+
+            Map<String, ItemStack> tracker = tracked.dropTrackers.get(tracked.getDropId(SeasonDateHandler.getCurrentEvent()));
+            if (tracker != null) {
+                Position pos = SkyblockHud.config.trackers.trackerPosition;
+                int startPos = pos.getAbsX(event.resolution, (tracker.size() >= 6 ? 120 : tracker.size() * 20));
+                int y = pos.getAbsY(event.resolution, (int) (10 + Math.ceil(tracker.size() / 5d) * 20));
+
+                Gui.drawRect(startPos, y, startPos + 120, y + 10, -1072689136);
+                mc.fontRendererObj.drawString("Tracker", startPos + 4, y + 1, 0xffffff, false);
+                y += 10;
+                Gui.drawRect(startPos, y, startPos + (tracker.size() >= 6 ? 120 : tracker.size() * 20), (int) (y + (Math.ceil(tracker.size() / 5d) * 20)), 1610612736);
+                int x = startPos;
+                for (ItemStack stack : tracker.values()) {
+                    String s = String.valueOf(stack.stackSize);
+                    GlStateManager.disableLighting();
+                    GlStateManager.enableDepth();
+                    drawItemStack(stack, x, y);
+                    GlStateManager.disableDepth();
+                    GlStateManager.disableBlend();
+                    mc.fontRendererObj.drawStringWithShadow(s, (float) (x + 19 - 2 - mc.fontRendererObj.getStringWidth(s)), (float) (y + 9), stack.stackSize < 1 ? 16733525 : 16777215);
+                    GlStateManager.enableBlend();
+                    GlStateManager.enableDepth();
+
+                    if ((x - startPos) / 20 == 5) {
+                        x = startPos;
+                        y += 20;
+                    } else {
+                        x += 20;
+                    }
+                }
+            }
+        }
+    }
+
+
+}