package net.fabricmc.fabric.mixin.event.lifecycle;

import java.util.function.BooleanSupplier;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerTickEvents;
import net.minecraft.class_3218;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.At.Shift;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

@Mixin({class_3218.class})
public abstract class ServerWorldMixin {
   @Inject(
      method = {"method_18765(Ljava/util/function/BooleanSupplier;)V"},
      at = {@At(
   value = "FIELD",
   target = "Lnet/minecraft/class_3218;field_13953:Z",
   opcode = 181,
   ordinal = 0,
   shift = Shift.AFTER
)}
   )
   private void startWorldTick(BooleanSupplier shouldKeepTicking, CallbackInfo ci) {
      ((ServerTickEvents.StartWorldTick)ServerTickEvents.START_WORLD_TICK.invoker()).onStartTick((class_3218)this);
   }

   @Inject(
      method = {"method_18765(Ljava/util/function/BooleanSupplier;)V"},
      at = {@At("TAIL")}
   )
   private void endWorldTick(BooleanSupplier shouldKeepTicking, CallbackInfo ci) {
      ((ServerTickEvents.EndWorldTick)ServerTickEvents.END_WORLD_TICK.invoker()).onEndTick((class_3218)this);
   }
}