Redo ad_astra compat

src/main/java/com/minelittlepony/unicopia/compat/ad_astra/OxygenApi.java

@@ -0,0 +1,11 @@
+package com.minelittlepony.unicopia.compat.ad_astra;
+
+import java.util.concurrent.atomic.AtomicReference;
+
+import net.minecraft.entity.Entity;
+
+public interface OxygenApi {
+    AtomicReference<OxygenApi> API = new AtomicReference<>(entity -> false);
+
+    boolean hasOxygen(Entity entity);
+}

src/main/java/com/minelittlepony/unicopia/compat/ad_astra/OxygenUtils.java

@@ -1,11 +0,0 @@
-package com.minelittlepony.unicopia.compat.ad_astra;
-
-import net.minecraft.entity.Entity;
-
-public final class OxygenUtils {
-    public static OxygenApi API = entity -> false;
-
-    public interface OxygenApi {
-        boolean hasOxygen(Entity entity);
-    }
-}

src/main/java/com/minelittlepony/unicopia/entity/player/PlayerPhysics.java

@@ -12,7 +12,7 @@ import com.minelittlepony.unicopia.ability.magic.spell.effect.SpellType;
 import com.minelittlepony.unicopia.advancement.UCriteria;
 import com.minelittlepony.unicopia.client.minelittlepony.MineLPDelegate;
 import com.minelittlepony.unicopia.client.render.PlayerPoser.Animation;
-import com.minelittlepony.unicopia.compat.ad_astra.OxygenUtils;
+import com.minelittlepony.unicopia.compat.ad_astra.OxygenApi;
 import com.minelittlepony.unicopia.entity.*;
 import com.minelittlepony.unicopia.entity.damage.UDamageTypes;
 import com.minelittlepony.unicopia.entity.duck.LivingEntityDuck;
@@ -200,7 +200,7 @@ public class PlayerPhysics extends EntityPhysics<PlayerEntity> implements Tickab
 
         if ((RegistryUtils.isIn(entity.getWorld(), dimension, RegistryKeys.DIMENSION_TYPE, UTags.DimensionTypes.HAS_NO_ATMOSPHERE)
                 || Unicopia.getConfig().dimensionsWithoutAtmosphere.get().contains(RegistryUtils.getId(entity.getWorld(), dimension, RegistryKeys.DIMENSION_TYPE).toString()))
-                && !OxygenUtils.API.hasOxygen(entity)) {
+                && !OxygenApi.API.get().hasOxygen(entity)) {
             return FlightType.NONE;
         }
 

src/main/java/com/minelittlepony/unicopia/mixin/ad_astra/MixinOxygenUtils.java

@@ -1,30 +1,31 @@
 package com.minelittlepony.unicopia.mixin.ad_astra;
 
-import org.spongepowered.asm.mixin.Dynamic;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Pseudo;
-import org.spongepowered.asm.mixin.gen.Accessor;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Coerce;
 import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
-import com.minelittlepony.unicopia.compat.ad_astra.OxygenUtils;
+import com.minelittlepony.unicopia.compat.ad_astra.OxygenApi;
 
 @Pseudo
 @Mixin(
         targets = { "earth.terrarium.adastra.api.systems.OxygenApi" },
         remap = false
 )
-public interface MixinOxygenUtils extends OxygenUtils.OxygenApi {
+interface MixinOxygenApi extends OxygenApi {
-    @Accessor("API")
-    @Coerce
-    static Object getAPI() {
-        throw new AbstractMethodError("stub");
 }
 
-    @Dynamic("Compiler-generated class-init() method")
+@Pseudo
-    @Inject(method = "<clinit>()V", at = @At("RETURN"), remap = false)
+@Mixin(
-    private static void classInit() {
+        targets = { "earth.terrarium.adastra.api.ApiHelper" },
-        OxygenUtils.API = (OxygenUtils.OxygenApi)getAPI();
+        remap = false
+)
+abstract class MixinApiHelper {
+    @Inject(method = "load", at = @At("RETURN"), require = 0)
+    private static <T> void onLoad(Class<T> clazz, CallbackInfoReturnable<T> info) {
+        if (info.getReturnValue() instanceof OxygenApi api) {
+            OxygenApi.API.set(api);
+        }
     }
 }

src/main/resources/unicopia.mixin.json

@@ -71,7 +71,8 @@
     "trinkets.MixinTrinketInventory",
     "trinkets.MixinScreenHandler",
     "seasons.MixinFertilizableUtil",
-    "ad_astra.MixinOxygenUtils",
+    "ad_astra.MixinApiHelper",
+    "ad_astra.MixinOxygenApi",
     "forgified.MixinIForgeBoat"
   ],
   "client": [