Fix #72

1 files changed, 25 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/mixin/MixinClientConnection.java b/src/main/java/me/shedaniel/rei/mixin/MixinClientConnection.java
new file mode 100644
index 000000000..8b137aff7
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/mixin/MixinClientConnection.java
@@ -0,0 +1,25 @@
+package me.shedaniel.rei.mixin;
+
+import io.netty.channel.Channel;
+import me.shedaniel.rei.RoughlyEnoughItemsCore;
+import net.minecraft.network.ClientConnection;
+import net.minecraft.text.TextComponent;
+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;
+
+@Mixin(ClientConnection.class)
+public class MixinClientConnection {
+    
+    @Shadow
+    private Channel channel;
+    
+    @Inject(method = "disconnect", at = @At("HEAD"))
+    public void disconnect(TextComponent reason, CallbackInfo callback) {
+        if (channel.isOpen())
+            RoughlyEnoughItemsCore.reiIsOnServer = false;
+    }
+    
+}