1 files changed, 26 insertions, 6 deletions
diff --git a/src/main/java/com/thatgravyboat/skyblockhud/mixins/MixinRenderItem.java b/src/main/java/com/thatgravyboat/skyblockhud/mixins/MixinRenderItem.java
index 269920f..951b549 100644
--- a/src/main/java/com/thatgravyboat/skyblockhud/mixins/MixinRenderItem.java
+++ b/src/main/java/com/thatgravyboat/skyblockhud/mixins/MixinRenderItem.java
@@ -1,6 +1,8 @@
 package com.thatgravyboat.skyblockhud.mixins;
 
 import com.thatgravyboat.skyblockhud.handlers.CooldownHandler;
+import com.thatgravyboat.skyblockhud.location.MinesHandler;
+import java.lang.ref.WeakReference;
 import net.minecraft.client.gui.FontRenderer;
 import net.minecraft.client.renderer.GlStateManager;
 import net.minecraft.client.renderer.Tessellator;
@@ -16,15 +18,33 @@ import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 @Mixin(RenderItem.class)
 public abstract class MixinRenderItem {
 
-    @Shadow
-    protected abstract void draw(WorldRenderer renderer, int x, int y, int width, int height, int red, int green, int blue, int alpha);
+    @Shadow protected abstract void draw(WorldRenderer renderer, int x, int y, int width, int height, int red, int green, int blue, int alpha);
 
-    @Inject(method = "renderItemOverlayIntoGUI", at = @At("RETURN"))
+    @Inject(method="renderItemOverlayIntoGUI", at=@At("RETURN"))
     public void renderItemOverlayIntoGUI(FontRenderer fr, ItemStack stack, int xPosition, int yPosition, String text, CallbackInfo ci) {
-        if (stack == null) return;
+        if(stack == null) return;
         float cooldown = CooldownHandler.getAbilityTime(stack);
 
-        if (cooldown > -1) {
+        WeakReference<MinesHandler.PrehistoricEggProgress> weakProgress = MinesHandler.getEggColorAndProgress(stack);
+
+        if (weakProgress != null) {
+            MinesHandler.PrehistoricEggProgress progress = weakProgress.get();
+            if (progress == null) return;
+            GlStateManager.disableLighting();
+            GlStateManager.disableDepth();
+            GlStateManager.disableTexture2D();
+            GlStateManager.disableAlpha();
+            GlStateManager.disableBlend();
+            WorldRenderer worldrenderer = Tessellator.getInstance().getWorldRenderer();
+            this.draw(worldrenderer, xPosition + 2, yPosition + 13, 13, 2, 0, 0, 0, 255);
+            this.draw(worldrenderer, xPosition + 2, yPosition + 13,  Math.round(progress.progress * 13f), 1, (progress.currentColor >> 16) & 0xFF, (progress.currentColor >> 8) & 0xFF, progress.currentColor & 0xFF, 255);
+            GlStateManager.enableAlpha();
+            GlStateManager.enableTexture2D();
+            GlStateManager.enableLighting();
+            GlStateManager.enableDepth();
+        }
+
+        if (cooldown > -1){
             GlStateManager.disableLighting();
             GlStateManager.disableDepth();
             GlStateManager.disableTexture2D();
@@ -32,7 +52,7 @@ public abstract class MixinRenderItem {
             GlStateManager.disableBlend();
             WorldRenderer worldrenderer = Tessellator.getInstance().getWorldRenderer();
             this.draw(worldrenderer, xPosition + 2, yPosition + 13, 13, 2, 0, 0, 0, 255);
-            this.draw(worldrenderer, xPosition + 2, yPosition + 13, Math.round(cooldown * 13f), 1, 102, 102, 255, 255);
+            this.draw(worldrenderer, xPosition + 2, yPosition + 13,  Math.round(cooldown * 13f), 1, 102, 102, 255, 255);
             GlStateManager.enableAlpha();
             GlStateManager.enableTexture2D();
             GlStateManager.enableLighting();