Allow deployers to give schedules to trains

Author: Julia

common/src/main/java/com/railwayteam/railways/mixin/MixinStationBlock.java

@@ -23,17 +23,23 @@
 import com.railwayteam.railways.mixin_interfaces.ICarriageConductors;
 import com.railwayteam.railways.registry.CRBlocks;
 import com.railwayteam.railways.registry.CREntities;
+import com.simibubi.create.AllItems;
 import com.simibubi.create.Create;
+import com.simibubi.create.content.contraptions.Contraption;
 import com.simibubi.create.content.kinetics.deployer.DeployerFakePlayer;
 import com.simibubi.create.content.trains.entity.Carriage;
+import com.simibubi.create.content.trains.entity.CarriageContraption;
+import com.simibubi.create.content.trains.entity.CarriageContraptionEntity;
 import com.simibubi.create.content.trains.entity.Train;
 import com.simibubi.create.content.trains.schedule.Schedule;
 import com.simibubi.create.content.trains.schedule.ScheduleEntry;
+import com.simibubi.create.content.trains.schedule.ScheduleItem;
 import com.simibubi.create.content.trains.schedule.condition.ScheduledDelay;
 import com.simibubi.create.content.trains.schedule.destination.DestinationInstruction;
 import com.simibubi.create.content.trains.station.GlobalStation;
 import com.simibubi.create.content.trains.station.StationBlock;
 import com.simibubi.create.content.trains.station.StationBlockEntity;
+import net.createmod.catnip.data.Couple;
 import net.createmod.catnip.math.VecHelper;
 import net.minecraft.core.BlockPos;
 import net.minecraft.nbt.CompoundTag;
@@ -45,6 +51,7 @@
 import net.minecraft.world.InteractionHand;
 import net.minecraft.world.InteractionResult;
 import net.minecraft.world.entity.Entity;
+import net.minecraft.world.entity.LivingEntity;
 import net.minecraft.world.entity.item.ItemEntity;
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.item.ItemStack;
@@ -58,6 +65,7 @@
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 import java.util.ArrayList;
+import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.atomic.AtomicBoolean;
 
@@ -231,6 +239,29 @@ private void deployersAssemble(BlockState pState, Level pLevel, BlockPos pPos, P
         }
     }
 
+    @Inject(method = "use", at = @At("HEAD"), cancellable = true, remap = true)
+    private void deployerGiveSchedule(BlockState pState, Level pLevel, BlockPos pPos, Player pPlayer, InteractionHand pHand, BlockHitResult pHit, CallbackInfoReturnable<InteractionResult> cir) {
+        ItemStack item = pPlayer.getItemInHand(pHand);
+        if (!pLevel.isClientSide && pPlayer instanceof DeployerFakePlayer && pLevel.getBlockEntity(pPos) instanceof StationBlockEntity stationBe) {
+            if (stationBe.getStation() != null && stationBe.getStation().getPresentTrain() != null) {
+                Train train = stationBe.getStation().getPresentTrain();
+                if (train.runtime.getSchedule() == null) {
+                    Schedule schedule = ScheduleItem.getSchedule(item);
+                    if(schedule == null || schedule.entries.isEmpty())
+                        return;
+
+                    train.runtime.setSchedule(schedule, false);
+                    item.shrink(1);
+                    pPlayer.setItemInHand(pHand, item.isEmpty() ? ItemStack.EMPTY : item);
+                    cir.setReturnValue(InteractionResult.SUCCESS);
+                } else if(item.isEmpty()) {
+                    pPlayer.setItemInHand(pHand, train.runtime.returnSchedule());
+                    cir.setReturnValue(InteractionResult.SUCCESS);
+                }
+            }
+        }
+    }
+
     private boolean disassembleAndEnterMode(ServerPlayer sender, StationBlockEntity te) {
         GlobalStation station = te.getStation();
         if (station != null) {