Use safer injection point

ramidzkh · Dec 2, 2024 · a029164 · a029164
1 parent be22aea
commit a029164
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/main/java/me/ramidzkh/fabrishot/mixins/RenderSystemMixin.java b/src/main/java/me/ramidzkh/fabrishot/mixins/RenderSystemMixin.java
@@ -1,32 +1,31 @@
 package me.ramidzkh.fabrishot.mixins;
 
+import com.llamalad7.mixinextras.injector.v2.WrapWithCondition;
 import com.mojang.blaze3d.systems.RenderSystem;
 import me.ramidzkh.fabrishot.Fabrishot;
-import org.lwjgl.glfw.GLFW;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Unique;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Redirect;
 
 @Mixin(RenderSystem.class)
 public class RenderSystemMixin {
 
     @Unique
     private static boolean wasLastFrameInCapture;
 
-    @Redirect(method = "flipFrame", at = @At(value = "INVOKE", target = "Lorg/lwjgl/glfw/GLFW;glfwSwapBuffers(J)V"), remap = false)
-    private static void glfwSwapBuffers(long window) {
+    @WrapWithCondition(method = "flipFrame", at = @At(value = "INVOKE", target = "Lorg/lwjgl/glfw/GLFW;glfwSwapBuffers(J)V"), remap = false)
+    private static boolean glfwSwapBuffers(long window) {
         if (Fabrishot.isInCapture()) {
             wasLastFrameInCapture = true;
-            return;
+            return false;
         }
 
         // skip a single black frame
         if (wasLastFrameInCapture) {
             wasLastFrameInCapture = false;
-            return;
+            return false;
         }
 
-        GLFW.glfwSwapBuffers(window);
+        return true;
     }
 }