RJW-Sexperience/Source/RJWSexperienceCum/JobDriver_CleanSelfWithBucket.cs

using RJWSexperience;
using System.Collections.Generic;
using Verse;
using Verse.AI;

namespace RJWSexperienceCum
{
	public class JobDriver_CleanSelfWithBucket : JobDriver
	{
		protected const int UNITTIME = 240;//ticks - 120 = 2 real seconds, 3 in-game minutes
		protected float progress = 0;
		protected float severitycache = 1;
		protected Hediff hediffcache;
		protected float CleaningTime
		{
			get
			{
				return severitycache * UNITTIME;
			}
		}

		protected Building_CumBucket Bucket => TargetB.Thing as Building_CumBucket;

		public override bool TryMakePreToilReservations(bool errorOnFailed)
		{
			return pawn.Reserve(pawn, job, 1, -1, null, errorOnFailed);
		}

		protected override IEnumerable<Toil> MakeNewToils()
		{
			this.FailOn(delegate
			{
				List<Hediff> hediffs = pawn.health.hediffSet.hediffs;
				return !hediffs.Exists(x => x.def == HediffDefOf.Hediff_CumController);
			});
			yield return Toils_Goto.GotoThing(TargetIndex.B, PathEndMode.ClosestTouch);
			Toil cleaning = new Toil
			{
				initAction = CleaningInit,
				tickAction = CleaningTick,
				defaultCompleteMode = ToilCompleteMode.Never
			};
			cleaning.AddFinishAction(Finish);
			cleaning.WithProgressBar(TargetIndex.A, () => progress / CleaningTime);

			yield return cleaning;
		}

		protected void CleaningInit()
		{
			hediffcache = pawn.health.hediffSet.hediffs.Find(x => x.def == HediffDefOf.Hediff_Cum || x.def == HediffDefOf.Hediff_InsectSpunk);
			if (hediffcache == null)
			{
				pawn.jobs.EndCurrentJob(JobCondition.Succeeded);
			}
			else
			{
				progress = 0f;
				severitycache = hediffcache.Severity;
				if (float.IsNaN(severitycache)) //TODO: Figure out WHY NaN is here
					severitycache = 0.1f;
			}
		}

		protected void CleaningTick()
		{
			progress++;
			if (progress > CleaningTime)
			{
				Cleaned();
			}
		}

		protected void Cleaned()
		{
			if (hediffcache != null)
			{
				float cumamount = hediffcache.Severity * 10f;
				hediffcache.Severity = 0f;
				Bucket.AddCum(cumamount);
			}
			CleaningInit();
		}

		protected void Finish()
		{
			if (pawn.CurJobDef == RimWorld.JobDefOf.Wait_MaintainPosture)
			{
				pawn.jobs.EndCurrentJob(JobCondition.InterruptForced);
			}
		}
	}
}