1 files changed, 25 insertions, 0 deletions
diff --git a/src/main/java/org/chrisoft/trashyaddon/mixin/EntityMixin.java b/src/main/java/org/chrisoft/trashyaddon/mixin/EntityMixin.java
new file mode 100644
index 0000000..39f6bb6
--- /dev/null
+++ b/src/main/java/org/chrisoft/trashyaddon/mixin/EntityMixin.java
@@ -0,0 +1,25 @@
+package org.chrisoft.trashyaddon.mixin;
+
+import com.mojang.logging.LogUtils;
+import meteordevelopment.meteorclient.utils.render.color.Color;
+import meteordevelopment.meteorclient.utils.render.postprocess.PostProcessShaders;
+import net.minecraft.entity.Entity;
+import org.chrisoft.trashyaddon.commands.EntityHighlightCommand;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+@Mixin(value = Entity.class, priority = 1001)
+public abstract class EntityMixin {
+    @Inject(method = "getTeamColorValue", at = @At("HEAD"), cancellable = true)
+    private void onGetTeamColorValue(CallbackInfoReturnable<Integer> info) {
+        if (PostProcessShaders.rendering) {
+            Color color = EntityHighlightCommand.getEntityColor((Entity) (Object) this);
+            if (color != null) {
+                info.setReturnValue(color.getPacked());
+                info.cancel();
+            }
+        }
+    }
+}