暂存充电床

2025-08-14 14:28:28 +08:00
parent e3b25dbda2
commit 1fafac51bb
13 changed files with 260 additions and 11 deletions
--- a/Source/WulaFallenEmpire/JobDriver_WulaLayDownToCharge.cs
+++ b/Source/WulaFallenEmpire/JobDriver_WulaLayDownToCharge.cs
@@ -0,0 +1,58 @@
+using RimWorld;
+using System.Collections.Generic;
+using Verse;
+using Verse.AI;
+
+namespace WulaFallenEmpire
+{
+    public class JobDriver_WulaLayDownToCharge : JobDriver_LayDown
+    {
+        protected override IEnumerable<Toil> MakeNewToils()
+        {
+            this.AddFinishAction(jobCondition =>
+            {
+                Log.Message($"[JobDriver_WulaLayDownToCharge] Job finishing for {pawn.Name.ToStringShort} with condition {jobCondition}. Removing hediff.");
+                var hediff = pawn.health.hediffSet.GetFirstHediffOfDef(HediffDef.Named("WULA_ChargingHediff"));
+                if (hediff != null)
+                {
+                    pawn.health.RemoveHediff(hediff);
+                    Log.Message($"[JobDriver_WulaLayDownToCharge] Hediff removed from {pawn.Name.ToStringShort}.");
+                }
+                else
+                {
+                    Log.Message($"[JobDriver_WulaLayDownToCharge] No hediff found on {pawn.Name.ToStringShort} to remove.");
+                }
+            });
+
+            foreach (Toil toil in base.MakeNewToils())
+            {
+                yield return toil;
+            }
+
+            var bed = (Building_Bed)job.targetA.Thing;
+            var powerComp = bed.GetComp<CompPowerTrader>();
+
+            var checkToil = new Toil
+            {
+                tickAction = delegate
+                {
+                    if (powerComp != null && !powerComp.PowerOn)
+                    {
+                        Log.Message($"[JobDriver_WulaLayDownToCharge] Power lost for {pawn.Name.ToStringShort}. Ending job.");
+                        EndJobWith(JobCondition.Incompletable);
+                        return;
+                    }
+
+                    Need_WulaEnergy energyNeed = pawn.needs.TryGetNeed<Need_WulaEnergy>();
+                    if (energyNeed != null && energyNeed.CurLevelPercentage >= 0.99f)
+                    {
+                        Log.Message($"[JobDriver_WulaLayDownToCharge] {pawn.Name.ToStringShort} is fully charged. Ending job.");
+                        EndJobWith(JobCondition.Succeeded);
+                    }
+                },
+                defaultCompleteMode = ToilCompleteMode.Never
+            };
+            yield return checkToil;
+        }
+    }
+}