Add more config options

Tomate0613 · Dec 6, 2023 · fbc16a2 · fbc16a2
1 parent bc08a85
commit fbc16a2
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 5 deletions.
diff --git a/src/main/java/tomate/totaldragon/DragonConfig.java b/src/main/java/tomate/totaldragon/DragonConfig.java
@@ -2,10 +2,17 @@
 
 // TODO Replace this with an actual config (placeholder for now)
 public class DragonConfig {
+    public enum PhantomBehaviour {
+        NO_PHANTOMS_IN_FIGHT,
+        VANILLA_PHANTOMS,
+        IMPROVED_PHANTOMS_IN_FIGHT,
+        IMPROVED_PHANTOMS_ALWAYS
+    };
+
     public static final boolean logDragonPhasesToInGameChat = false;
     public static final boolean playersFallIntoOverworld = true;
-
     public static final boolean chainReactionEndCrystals = true;
-
     public static final boolean endCrystalSpawns = true;
+    public static final boolean spawnEndermitesInFight = true;
+    public static final PhantomBehaviour phantomBehaviour = PhantomBehaviour.IMPROVED_PHANTOMS_IN_FIGHT;
 }
diff --git a/src/main/java/tomate/totaldragon/mixin/EndCrystalMixin.java b/src/main/java/tomate/totaldragon/mixin/EndCrystalMixin.java
@@ -35,7 +35,6 @@ public boolean hurt(DamageSource damageSource, float f) {
         if (damageSource.getEntity() instanceof EnderDragon) {
             return false;
         }
-
         if (!this.isRemoved() && !this.level().isClientSide && active) {
             active = false;
 
@@ -76,13 +75,19 @@ public boolean hurt(DamageSource damageSource, float f) {
 
     @Unique
     private void spawnEndermite() {
+        if(!DragonConfig.spawnEndermitesInFight)
+            return;
+
         var endermite = new Endermite(EntityType.ENDERMITE, level());
         endermite.setPos(position());
         level().addFreshEntity(endermite);
     }
 
     @Unique
     private void spawnPhantoms(LivingEntity target) {
+        if(DragonConfig.phantomBehaviour == DragonConfig.PhantomBehaviour.NO_PHANTOMS_IN_FIGHT)
+            return;
+
         var phantom = new Phantom(EntityType.PHANTOM, level());
         phantom.setPos(position());
         phantom.setTarget(target);

diff --git a/src/main/java/tomate/totaldragon/mixin/PhantomMixin.java b/src/main/java/tomate/totaldragon/mixin/PhantomMixin.java
@@ -12,6 +12,7 @@
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import tomate.totaldragon.DragonConfig;
 
 @Mixin(Phantom.class)
 public abstract class PhantomMixin extends FlyingMob {
@@ -23,8 +24,8 @@ protected PhantomMixin(EntityType<? extends FlyingMob> entityType, Level level)
 
     @Inject(at = @At("TAIL"), method = "<init>")
     void init(EntityType<?> entityType, Level level, CallbackInfo ci) {
-        var dimensionsAccessible = ((EntityDimensionsAccessor)this);
-        dimensionsAccessible.setDimensions(dimensionsAccessible.getDimensions().scale(2, 1.2f));
+        if(DragonConfig.phantomBehaviour != DragonConfig.PhantomBehaviour.IMPROVED_PHANTOMS_ALWAYS && !(DragonConfig.phantomBehaviour == DragonConfig.PhantomBehaviour.IMPROVED_PHANTOMS_IN_FIGHT && level().dimension() != Level.END))
+            return;
 
         updatePhantomSize();
     }
@@ -46,6 +47,9 @@ public boolean hurt(DamageSource damageSource, float f) {
     }
 
     private void updatePhantomSize() {
+        var dimensionsAccessible = ((EntityDimensionsAccessor)this);
+        dimensionsAccessible.setDimensions(dimensionsAccessible.getDimensions().scale(2, 1.2f));
+
         if(level().dimension() == Level.END) {
             getAttribute(Attributes.ATTACK_DAMAGE).setBaseValue(25 + this.getPhantomSize());
             return;