fix TPS issues with the TES as well as ATs turning off while the TES is on

src/main/java/net/neganote/monilabs/MoniEvents.java

@@ -1,9 +1,11 @@
 package net.neganote.monilabs;
 
 import net.minecraftforge.event.RegisterCommandsEvent;
+import net.minecraftforge.event.server.ServerStoppingEvent;
 import net.minecraftforge.eventbus.api.SubscribeEvent;
 import net.minecraftforge.fml.common.Mod;
 import net.neganote.monilabs.commands.PackSwitcherCommands;
+import net.neganote.monilabs.common.machine.multiblock.CreativeEnergyMultiMachine;
 
 @Mod.EventBusSubscriber(modid = MoniLabs.MOD_ID)
 public class MoniEvents {
@@ -13,4 +15,9 @@ public static void RegisterCommandsEvent(RegisterCommandsEvent event) {
         MoniLabs.LOGGER.debug("Registering commands...");
         PackSwitcherCommands.register(event.getDispatcher(), event.getBuildContext());
     }
+
+    @SubscribeEvent
+    public static void onServerStopping(ServerStoppingEvent event) {
+        CreativeEnergyMultiMachine.clearActiveOwners();
+    }
 }

src/main/java/net/neganote/monilabs/common/machine/multiblock/CreativeEnergyMultiMachine.java

@@ -2,15 +2,22 @@
 
 import com.gregtechceu.gtceu.api.machine.ConditionalSubscriptionHandler;
 import com.gregtechceu.gtceu.api.machine.IMachineBlockEntity;
+import com.gregtechceu.gtceu.common.machine.owner.MachineOwner;
 
 import net.minecraft.server.level.ServerLevel;
-import net.neganote.monilabs.saveddata.CreativeEnergySavedData;
 
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
 import java.util.UUID;
 
 @SuppressWarnings("unused")
 public class CreativeEnergyMultiMachine extends UniqueWorkableElectricMultiblockMachine {
 
+    private static final Set<UUID> ACTIVE_OWNERS = new HashSet<>();
+    private static final Map<UUID, Boolean> PLAYER_CACHE = new HashMap<>();
+
     private final ConditionalSubscriptionHandler creativeEnergySubscription;
 
     public CreativeEnergyMultiMachine(IMachineBlockEntity holder, Object... args) {
@@ -20,21 +27,36 @@ public CreativeEnergyMultiMachine(IMachineBlockEntity holder, Object... args) {
                 this::isSubscriptionActive);
     }
 
+    public static boolean isCreativeEnergyEnabledFor(UUID playerUUID) {
+        if (ACTIVE_OWNERS.isEmpty()) return false;
+        return PLAYER_CACHE.computeIfAbsent(playerUUID, uuid -> {
+            MachineOwner owner = MachineOwner.getOwner(uuid);
+            if (owner == null) return false;
+            return ACTIVE_OWNERS.stream().anyMatch(owner::isPlayerInTeam);
+        });
+    }
+
+    public static void clearActiveOwners() {
+        ACTIVE_OWNERS.clear();
+        PLAYER_CACHE.clear();
+    }
+
     @Override
     public void onStructureFormed() {
         super.onStructureFormed();
+        enableCreativeEnergy(recipeLogic.isWorking());
         creativeEnergySubscription.updateSubscription();
     }
 
     public void enableCreativeEnergy(boolean enabled) {
+        if (!(getLevel() instanceof ServerLevel)) return;
         UUID ownerUUID = getOwnerUUID();
         if (ownerUUID == null) {
             ownerUUID = new UUID(0L, 0L);
         }
-        if (getLevel() instanceof ServerLevel serverLevel) {
-            CreativeEnergySavedData savedData = CreativeEnergySavedData
-                    .getOrCreate(serverLevel.getServer().overworld());
-            savedData.setEnabled(ownerUUID, enabled);
+        boolean changed = enabled ? ACTIVE_OWNERS.add(ownerUUID) : ACTIVE_OWNERS.remove(ownerUUID);
+        if (changed) {
+            PLAYER_CACHE.clear();
         }
     }
 

src/main/java/net/neganote/monilabs/mixin/NotifiableEnergyContainerMixin.java

@@ -8,7 +8,7 @@
 import com.lowdragmc.lowdraglib.syncdata.field.ManagedFieldHolder;
 
 import net.minecraft.server.level.ServerLevel;
-import net.neganote.monilabs.saveddata.CreativeEnergySavedData;
+import net.neganote.monilabs.common.machine.multiblock.CreativeEnergyMultiMachine;
 
 import org.jetbrains.annotations.Nullable;
 import org.spongepowered.asm.mixin.Mixin;
@@ -45,13 +45,12 @@ public MetaMachine getMachine() {
     @Inject(method = "getEnergyStored()J", at = @At(value = "HEAD"), cancellable = true)
     private void monilabs$injectBeforeGetEnergyStored(CallbackInfoReturnable<Long> cir) {
         MetaMachine machine = getMachine();
-        if (machine.getLevel() instanceof ServerLevel serverLevel) {
+        if (machine.getLevel() instanceof ServerLevel) {
             outputSubs = machine.subscribeServerTick(this.outputSubs, this::serverTick);
-            CreativeEnergySavedData savedData = CreativeEnergySavedData
-                    .getOrCreate(serverLevel.getServer().overworld());
             UUID uuid = machine.getOwnerUUID();
-            if (uuid != null && savedData.isEnabledFor(uuid)) {
-                cir.setReturnValue(getEnergyCapacity());
+            if (uuid != null && CreativeEnergyMultiMachine.isCreativeEnergyEnabledFor(uuid)) {
+                // return 1 less so active transformers won't turn off
+                cir.setReturnValue(getEnergyCapacity() - 1);
             }
         }
     }
@@ -61,12 +60,10 @@ public MetaMachine getMachine() {
     @Inject(method = "changeEnergy", at = @At(value = "HEAD"), cancellable = true)
     private void monilabs$injectBeforeChangeEnergy(long energyToAdd, CallbackInfoReturnable<Long> cir) {
         MetaMachine machine = getMachine();
-        if (machine.getLevel() instanceof ServerLevel serverLevel) {
+        if (machine.getLevel() instanceof ServerLevel) {
             outputSubs = machine.subscribeServerTick(this.outputSubs, this::serverTick);
-            CreativeEnergySavedData savedData = CreativeEnergySavedData
-                    .getOrCreate(serverLevel.getServer().overworld());
             UUID uuid = machine.getOwnerUUID();
-            if (uuid != null && savedData.isEnabledFor(uuid)) {
+            if (uuid != null && CreativeEnergyMultiMachine.isCreativeEnergyEnabledFor(uuid)) {
                 cir.setReturnValue(energyToAdd);
             }
         }

src/main/java/net/neganote/monilabs/mixin/PowerSubstationMachineMixin.java

@@ -5,7 +5,7 @@
 import com.gregtechceu.gtceu.common.machine.multiblock.electric.PowerSubstationMachine;
 
 import net.minecraft.server.level.ServerLevel;
-import net.neganote.monilabs.saveddata.CreativeEnergySavedData;
+import net.neganote.monilabs.common.machine.multiblock.CreativeEnergyMultiMachine;
 
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
@@ -22,12 +22,9 @@ public PowerSubstationMachineMixin(IMachineBlockEntity holder) {
     // Prevents substations from performing any power transfers while TES is running
     @Inject(method = "transferEnergyTick()V", at = @At(value = "HEAD"), cancellable = true)
     public void monilabs$injectBeforeTransferEnergyTick(CallbackInfo ci) {
-        if (getLevel() instanceof ServerLevel serverLevel) {
-            CreativeEnergySavedData savedData = CreativeEnergySavedData
-                    .getOrCreate(serverLevel.getServer().overworld());
-            if (savedData.isEnabledFor(getOwnerUUID())) {
-                ci.cancel();
-            }
+        if (getLevel() instanceof ServerLevel &&
+                CreativeEnergyMultiMachine.isCreativeEnergyEnabledFor(getOwnerUUID())) {
+            ci.cancel();
         }
     }
 }