rjw-genes/Source/Interactions/CustomSexInteraction_Helper.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Verse;
using RimWorld;
using rjw;
using rjw.Modules.Interactions.Objects;
using rjw.Modules.Interactions.Helpers;
using rjw.Modules.Interactions.Enums;
using rjw.Modules.Interactions.Implementation;
using rjw.Modules.Interactions.Defs.DefFragment;
namespace RJW_Genes
{
    public class CustomSexInteraction_Helper
    {
		public static List<InteractionDef> GenerateInteractionDefList(Pawn pawn, Pawn pawn2, CompProperties_SexInteractionRequirements sexpropsreq)
		{
			List<InteractionTag> tags = new List<InteractionTag>();
			if (pawn2.IsAnimal())
			{
				tags.Add(InteractionTag.Animal);

			}
			else
			{
				tags = sexpropsreq.tags;
			}

			InteractionRequirement dominantRequirement = sexpropsreq.dominantRequirement;
			InteractionRequirement submissiveRequirement = sexpropsreq.submissiveRequirement;
			List<InteractionDef> list = new List<InteractionDef>();
			//List<InteractionDef> a = from interaction in sexinteractions
			//where InteractionHelper.GetWithExtension(interaction).DominantHasFamily(dominantRequirement.families.)
			//						 select interaction;

			//should use where select but dont fully understand that, so I am using this.
			foreach (InteractionDef interactionDef in SexUtility.SexInterractions)
			{
				//Use rjw function to check if the interaction would be valid
				if (!LewdInteractionValidatorService.Instance.IsValid(interactionDef, pawn, pawn2))
				{
					continue;
				}
				InteractionWithExtension withExtension = InteractionHelper.GetWithExtension(interactionDef);
				bool add_interaction = false;
				//only add interactions which have a correct tag
				foreach (InteractionTag tag in tags)
				{
					if (withExtension.HasInteractionTag(tag))
					{
						add_interaction = true;
						break;
					}
				}
				//In case of failure go to next interaction
				if (!add_interaction)
				{
					continue;
				}
				//goes to next interaction if it doesn't have the required genitals
				if (dominantRequirement != null)
				{
					foreach (GenitalFamily genitalFamily in dominantRequirement.families)
					{
						if (!withExtension.DominantHasFamily(genitalFamily))
						{
							add_interaction = false;
							break;

						}
					}
					if (!add_interaction)
					{
						continue;
					}
					foreach (GenitalTag tag in dominantRequirement.tags)
					{
						if (!withExtension.DominantHasTag(tag))
						{
							add_interaction = false;
							break;

						}
					}
				}
				//goes to next interaction if it doesn't have the required genitals
				if (submissiveRequirement != null)
				{
					foreach (GenitalFamily genitalFamily in submissiveRequirement.families)
					{
						if (!withExtension.SubmissiveHasFamily(genitalFamily))
						{
							add_interaction = false;
							break;

						}
					}
					if (!add_interaction)
					{
						continue;
					}
					foreach (GenitalTag tag in submissiveRequirement.tags)
					{
						if (!withExtension.SubmissiveHasTag(tag))
						{
							add_interaction = false;
							break;

						}

					}
				}
				if (add_interaction)
				{
					list.Add(interactionDef);
				}

			}
			return list;
		}

		//Generates a valid interaction for the requirements and assigns sexprops based on that
		public static SexProps GenerateSexProps(Pawn pawn, Pawn pawn2, CompProperties_SexInteractionRequirements sexpropsreq)
		{
			List<InteractionDef> interactionlist = GenerateInteractionDefList(pawn, pawn2, sexpropsreq);
			if (!interactionlist.Any())
			{
				return null;
			}
			InteractionDef dictionaryKey = interactionlist.RandomElement();
			bool rape = InteractionHelper.GetWithExtension(dictionaryKey).HasInteractionTag(InteractionTag.Rape);
			SexProps sexProps = new SexProps();
			sexProps.pawn = pawn;
			sexProps.partner = pawn2;
			sexProps.sexType = SexUtility.rjwSextypeGet(dictionaryKey);
			sexProps.isRape = rape;
			sexProps.isRapist = rape;
			sexProps.canBeGuilty = false;
			sexProps.dictionaryKey = dictionaryKey;
			sexProps.rulePack = SexUtility.SexRulePackGet(dictionaryKey);
			return sexProps;
		}
	}
}