1

Source/WulaFallenEmpire/Work/EnterMech/JobDriver_EnterMech.cs

@@ -0,0 +1,76 @@
+// File: JobDriver_EnterMech.cs (不再保留机甲)
+using RimWorld;
+using System.Collections.Generic;
+using Verse;
+using Verse.AI;
+namespace WulaFallenEmpire
+{
+    public class JobDriver_EnterMech : JobDriver
+    {
+        private const TargetIndex MechIndex = TargetIndex.A;
+
+        public override bool TryMakePreToilReservations(bool errorOnFailed)
+        {
+            Pawn pawn = this.pawn;
+            LocalTargetInfo target = this.job.GetTarget(MechIndex);
+
+            // 不再保留机甲，这样多个殖民者可以同时被命令进入同一个机甲
+            // 只需要检查殖民者是否可以到达机甲
+            if (!pawn.CanReach(target, PathEndMode.Touch, Danger.Deadly))
+            {
+                return false;
+            }
+            
+            return true;
+        }
+
+        protected override IEnumerable<Toil> MakeNewToils()
+        {
+            // 0. 初始检查
+            AddFailCondition(() =>
+            {
+                var mech = TargetThingA as Wulamechunit;
+                if (mech == null || mech.Destroyed)
+                {
+                    return true;
+                }
+
+                var comp = mech.GetComp<CompMechPilotHolder>();
+                if (comp == null || comp.IsFull || !comp.CanAddPilot(pawn))
+                {
+                    return true;
+                }
+
+                if (pawn.Downed || pawn.Dead)
+                    return true;
+
+                return false;
+            });
+
+            // 1. 走到机甲旁边
+            yield return Toils_Goto.GotoThing(MechIndex, PathEndMode.Touch);
+
+            // 2. 检查是否仍然可以进入
+            yield return Toils_General.Wait(10).WithProgressBarToilDelay(MechIndex);
+
+            // 3. 进入机甲
+            Toil enterToil = new Toil();
+            enterToil.initAction = () =>
+            {
+                var mech = TargetThingA as Wulamechunit;
+                if (mech == null)
+                    return;
+
+                var comp = mech.GetComp<CompMechPilotHolder>();
+                if (comp != null && comp.CanAddPilot(pawn))
+                {
+                    comp.AddPilot(pawn);
+                    Messages.Message("WULA_PilotEnteredMech".Translate(pawn.LabelShort, mech.LabelShort),
+                        MessageTypeDefOf.PositiveEvent, false);
+                }
+            };
+            enterToil.defaultCompleteMode = ToilCompleteMode.Instant;
+            yield return enterToil;
+        }
+    }
+}