RJW-Sexperience/RJWSexperience/RJWSexperience/Patches/RJW_Patch.cs

using HarmonyLib;
using RimWorld;
using rjw;
using rjw.Modules.Interactions.Enums;
using RJWSexperience.Cum;
using RJWSexperience.ExtensionMethods;
using RJWSexperience.Logs;
using System.Collections.Generic;
using UnityEngine;
using Verse;
using Verse.AI;

namespace RJWSexperience
{
	[HarmonyPatch(typeof(JobDriver_Sex), "Orgasm")]
	public static class RJW_Patch_Orgasm
	{
		public static void Postfix(JobDriver_Sex __instance)
		{
			if (__instance.Sexprops.sexType != xxx.rjwSextype.Masturbation && !(__instance is JobDriver_Masturbate))
			{
				if (__instance.Sexprops.isRape && __instance.Sexprops.isReceiver)
				{
					__instance.pawn?.skills?.Learn(VariousDefOf.SexSkill, 0.05f, true);
				}
				else
				{
					__instance.pawn?.skills?.Learn(VariousDefOf.SexSkill, 0.35f, true);
				}
			}
		}
	}

	[HarmonyPatch(typeof(WhoringHelper), "WhoreAbilityAdjustmentMin")]
	public static class RJW_Patch_WhoreAbilityAdjustmentMin
	{
		public static void Postfix(Pawn whore, ref float __result)
		{
			__result *= whore.GetSexStat();
		}
	}

	[HarmonyPatch(typeof(WhoringHelper), "WhoreAbilityAdjustmentMax")]
	public static class RJW_Patch_WhoreAbilityAdjustmentMax
	{
		public static void Postfix(Pawn whore, ref float __result)
		{
			__result *= whore.GetSexStat();
		}
	}

	[HarmonyPatch(typeof(SexUtility), nameof(SexUtility.SatisfyPersonal))]
	public static class RJW_Patch_SatisfyPersonal
	{
		private const float base_sat_per_fuck = 0.4f;

		public static void Prefix(SexProps props, ref float satisfaction)
		{
			satisfaction = Mathf.Max(base_sat_per_fuck, satisfaction * props.partner.GetSexStat());
		}

		public static void Postfix(SexProps props, ref float satisfaction)
		{
			LustUtility.UpdateLust(props, satisfaction, base_sat_per_fuck);
			FillCumBuckets(props);
			props.pawn.records?.Increment(VariousDefOf.OrgasmCount);
			if (props.partner != null)
				props.pawn.TryGetComp<SexHistoryComp>()?.RecordSatisfaction(props.partner, props, satisfaction);
		}

		private static void FillCumBuckets(SexProps props)
		{
			xxx.rjwSextype sextype = props.sexType;

			bool sexFillsCumbuckets =
				// Base: Fill Cumbuckets on Masturbation. Having no partner means it must be masturbation too
				sextype == xxx.rjwSextype.Masturbation || props.partner == null
				// Depending on configuration, also fill cumbuckets when certain sextypes are matched 
				|| (SexperienceMod.Settings.SexCanFillBuckets && (sextype == xxx.rjwSextype.Boobjob || sextype == xxx.rjwSextype.Footjob || sextype == xxx.rjwSextype.Handjob));

			if (!sexFillsCumbuckets)
				return;

			IEnumerable<Building_CumBucket> buckets = props.pawn.GetAdjacentBuildings<Building_CumBucket>();

			if (buckets?.EnumerableCount() > 0)
			{
				var initialCum = CumUtility.GetCumVolume(props.pawn);
				foreach (Building_CumBucket bucket in buckets)
				{
					bucket.AddCum(initialCum / buckets.EnumerableCount());
				}
			}
		}
	}

	[HarmonyPatch(typeof(SexUtility), "TransferNutrition")]
	public static class RJW_Patch_TransferNutrition
	{
		public static void Postfix(SexProps props)
		{
			TryFeedCum(props);
		}

		private static void TryFeedCum(SexProps props)
		{
			if (!Genital_Helper.has_penis_fertile(props.pawn))
				return;

			if (!PawnsPenisIsInPartnersMouth(props))
				return;

			float cumAmount = CumUtility.GetOnePartCumVolume(props.pawn);

			if (cumAmount <= 0)
				return;

			CumUtility.FeedCum(props.partner, cumAmount);
		}

		private static bool PawnsPenisIsInPartnersMouth(SexProps props)
		{
			var interaction = rjw.Modules.Interactions.Helpers.InteractionHelper.GetWithExtension(props.dictionaryKey);

			if (props.pawn == props.GetInteractionInitiator())
			{
				if (!interaction.DominantHasTag(GenitalTag.CanPenetrate) && !interaction.DominantHasFamily(GenitalFamily.Penis))
					return false;
				var requirement = interaction.SelectorExtension.submissiveRequirement;
				if (!requirement.mouth && !requirement.beak && !requirement.mouthORbeak)
					return false;
			}
			else
			{
				if (!interaction.SubmissiveHasTag(GenitalTag.CanPenetrate) && !interaction.SubmissiveHasFamily(GenitalFamily.Penis))
					return false;
				var requirement = interaction.SelectorExtension.dominantRequirement;
				if (!requirement.mouth && !requirement.beak && !requirement.mouthORbeak)
					return false;
			}

			return true;
		}
	}

	[HarmonyPatch(typeof(Nymph_Generator), "set_skills")]
	public static class RJW_Patch_Nymph_set_skills
	{
		public static void Postfix(Pawn pawn)
		{
			SkillRecord sexskill = pawn.skills.GetSkill(VariousDefOf.SexSkill);
			if (sexskill != null)
			{
				sexskill.passion = Passion.Major;
				sexskill.Level = (int)Utility.RandGaussianLike(7f, 20.99f);
				sexskill.xpSinceLastLevel = sexskill.XpRequiredForLevelUp * Rand.Range(0.10f, 0.90f);
			}
		}
	}

	[HarmonyPatch(typeof(AfterSexUtility), "UpdateRecords")]
	public static class RJW_Patch_UpdateRecords
	{
		public static void Postfix(SexProps props)
		{
			RJWUtility.UpdateSextypeRecords(props);

			if (props.partner == null)
				return;

			props.pawn.TryGetComp<SexHistoryComp>()?.RecordSex(props.partner, props);
			props.partner.TryGetComp<SexHistoryComp>()?.RecordSex(props.pawn, props);
		}
	}

	[HarmonyPatch(typeof(JobDriver_SexBaseInitiator), "Start")]
	public static class RJW_Patch_LogSextype
	{
		public static void Postfix(JobDriver_SexBaseInitiator __instance)
		{
			if (__instance.Partner != null)
			{
				__instance.pawn.PoptheCherry(__instance.Partner, __instance.Sexprops);
				__instance.Partner.PoptheCherry(__instance.pawn, __instance.Sexprops);
			}
		}
	}

	[HarmonyPatch(typeof(WorkGiver_CleanSelf), "JobOnThing")]
	public static class RJW_Patch_CleanSelf_JobOnThing
	{
		public static bool Prefix(Pawn pawn, Thing t, bool forced, ref Job __result)
		{
			Building_CumBucket bucket = pawn.GetAdjacentBuilding<Building_CumBucket>();
			if (bucket == null) bucket = pawn.FindClosestBucket();
			if (bucket != null)
			{
				__result = JobMaker.MakeJob(VariousDefOf.CleanSelfwithBucket, null, bucket, bucket.Position);
				return false;
			}

			return true;
		}
	}

	[HarmonyPatch(typeof(CasualSex_Helper), nameof(CasualSex_Helper.FindSexLocation))]
	public static class RJW_Patch_CasualSex_Helper_FindSexLocation
	{
		/// <summary>
		/// If masturbation and current map has a bucket, return location near the bucket
		/// </summary>
		/// <param name="pawn"></param>
		/// <param name="partner"></param>
		/// <param name="__result"></param>
		/// <returns></returns>
		public static bool Prefix(Pawn pawn, Pawn partner, ref IntVec3 __result)
		{
			if (partner != null)
				return true; // Not masturbation

			var log = LogManager.GetLogger<DebugLogProvider>("RJW_Patch_CasualSex_Helper_FindSexLocation");
			log.Message($"Called for {pawn.NameShortColored}");

			if (pawn.Faction?.IsPlayer != true && !pawn.IsPrisonerOfColony)
			{
				log.Message("Not a player's faction or a prisoner");
				return true;
			}

			Building_CumBucket bucket = pawn.FindClosestBucket();

			if (bucket == null)
			{
				log.Message("Bucket not found");
				return true;
			}

			__result = bucket.RandomAdjacentCell8Way();
			log.Message($"Bucket location: {__result}");
			return false;
		}
	}
}