rjw-sexperience-ideology/Source/IdeologyAddon/SexTame/JobDriver_LewdTrainSex.cs

using System.Collections.Generic;
using Verse;
using RimWorld;
using rjw;
using Verse.AI;

namespace RJWSexperience.Ideology
{
	public class JobDriver_LewdTrainSex : LewdTrainBase
	{
		private Pawn originalPartner;
		protected override IEnumerable<Toil> MakeNewToils()
		{
			originalPartner = Partner;
			if (SexAppraiser.would_fuck_animal(pawn, Partner) > 0f)
			{
				foreach (var baseToil in base.MakeNewToils())
				{
					yield return baseToil;
				}
			}

			if (originalPartner != Partner)
			{
				// Pawn got interrupted by another partner, then the original partner finished before them.
				yield break;
			}

			yield return Toils_Interpersonal.SetLastInteractTime(iTarget);

			yield return Toils_Interpersonal.TryTrain(iTarget);
		}
	}
}