1 files changed, 35 insertions, 14 deletions

diff --git a/src/main/java/net/justlearning/arslaan3102/arslaansmagichax/modules/impl/Fullbright.java b/src/main/java/net/justlearning/arslaan3102/arslaansmagichax/modules/impl/Fullbright.java
index bf72c6a..eefd5fb 100644
--- a/src/main/java/net/justlearning/arslaan3102/arslaansmagichax/modules/impl/Fullbright.java
+++ b/src/main/java/net/justlearning/arslaan3102/arslaansmagichax/modules/impl/Fullbright.java
@@ -3,7 +3,6 @@ package net.justlearning.arslaan3102.arslaansmagichax.modules.impl;
 import net.justlearning.arslaan3102.arslaansmagichax.modules.Category;
 import net.justlearning.arslaan3102.arslaansmagichax.modules.Module;
 import net.justlearning.arslaan3102.arslaansmagichax.modules.settings.ModeSetting;
-import net.minecraft.client.option.SimpleOption;
 import net.minecraft.entity.effect.StatusEffectInstance;
 import net.minecraft.entity.effect.StatusEffects;
 import net.minecraft.registry.Registries;
@@ -11,23 +10,24 @@ import net.minecraft.registry.Registries;
 import java.util.Objects;
 
 public class Fullbright extends Module {
-    private ModeSetting mode = new ModeSetting("Mode", "Gamma", "Effect", "Gamma", "Effect + Gamma");
-    private SimpleOption<Double> option;
+    public ModeSetting mode = new ModeSetting("Mode", "Gamma", "Effect", "Gamma", "Effect + Gamma", "Break AO Renderer");
     private double previousGamma = 1.0;
     private boolean setPreviousGamma = false;
+    private String lastMode = ""; // track previous mode
 
     public Fullbright() {
         super("Fullbright", "Brightens up your world!", -1, Category.RENDER);
         addSetting(mode);
+        lastMode = mode.getMode(); // initialize
     }
 
     @Override
     protected void onEnable() {
         if (mc.options == null) return;
-        if (Objects.equals(mode.getMode(), "Gamma")) {
-            if (!setPreviousGamma) previousGamma = mc.options.getGamma().getValue();
-            setPreviousGamma = true;
-            mc.options.getGamma().setValue(1000.0);
+        applyMode(mode.getMode());
+
+        if (mc.worldRenderer != null) {
+            mc.worldRenderer.reload();
         }
     }
 
@@ -35,31 +35,52 @@ public class Fullbright extends Module {
     public void tick() {
         if (!this.isEnabled() || mc.player == null) return;
 
-        if (Objects.equals(mode.getMode(), "Effect")) {
-            mc.options.getGamma().setValue(previousGamma);
-            setPreviousGamma = false;
+        String currentMode = mode.getMode();
+        if (!Objects.equals(currentMode, lastMode)) {
+            applyMode(currentMode);
+
+            if (mc.worldRenderer != null) {
+                mc.worldRenderer.reload();
+            }
+
+            lastMode = currentMode;
+        }
+
+        if (Objects.equals(currentMode, "Effect")) {
+            mc.player.addStatusEffect(new StatusEffectInstance(Registries.STATUS_EFFECT.getEntry(StatusEffects.NIGHT_VISION.value()), 1, 255));
+        } else if (Objects.equals(currentMode, "Effect + Gamma")) {
             mc.player.addStatusEffect(new StatusEffectInstance(Registries.STATUS_EFFECT.getEntry(StatusEffects.NIGHT_VISION.value()), 1, 255));
         }
-        if (Objects.equals(mode.getMode(), "Gamma")) {
+    }
+
+    private void applyMode(String modeName) {
+        if (Objects.equals(modeName, "Gamma")) {
             if (!setPreviousGamma) previousGamma = mc.options.getGamma().getValue();
             setPreviousGamma = true;
             mc.options.getGamma().setValue(1000.0);
-        }
-        if (Objects.equals(mode.getMode(), "Effect + Gamma")) {
+        } else if (Objects.equals(modeName, "Effect + Gamma")) {
             if (!setPreviousGamma) previousGamma = mc.options.getGamma().getValue();
             setPreviousGamma = true;
             mc.options.getGamma().setValue(1000.0);
-            mc.player.addStatusEffect(new StatusEffectInstance(Registries.STATUS_EFFECT.getEntry(StatusEffects.NIGHT_VISION.value()), 1, 255));
+        } else if (Objects.equals(modeName, "Effect")) {
+            mc.options.getGamma().setValue(previousGamma);
+            setPreviousGamma = false;
         }
     }
 
     @Override
     protected void onDisable() {
         if (mc.player == null) return;
+
         if (mc.player.hasStatusEffect(Registries.STATUS_EFFECT.getEntry(StatusEffects.NIGHT_VISION.value()))) {
             mc.player.removeStatusEffect(Registries.STATUS_EFFECT.getEntry(StatusEffects.NIGHT_VISION.value()));
         }
 
         mc.options.getGamma().setValue(previousGamma);
+
+        if (mc.worldRenderer != null) {
+            mc.worldRenderer.reload();
+        }
     }
 }
+