rjw-plasticsurgeries/Source/RJW_PlasticSurgeries/Recipe_Surgery_Beautify.cs

using System.Collections.Generic;
using RimWorld;
using Verse;
using static RimWorld.TraitDefOf;

namespace RJW_PlasticSurgeries
{
    /// <inheritdoc />
    public abstract class Recipe_Surgery_Beautify : Recipe_Surgery
    {
        /// <inheritdoc />
        public override IEnumerable<BodyPartRecord> GetPartsToApplyOn(Pawn pawn, RecipeDef recipe)
        {
            if (!pawn.story.traits.HasTrait(Beauty) ||
                pawn.story.traits.HasTrait(Beauty) && pawn.story.traits.GetTrait(Beauty).Degree < 2)
                yield return pawn.RaceProps.body.corePart;
        }

        /// <inheritdoc />
        public override void ApplyOnPawn(Pawn pawn, BodyPartRecord part, Pawn billDoer, List<Thing> ingredients,
            Bill bill)
        {
            if (billDoer != null)
            {
                TaleRecorder.RecordTale(TaleDefOf.DidSurgery, billDoer, pawn);
                SurgeryResult(pawn);
            }
        }

        /// <summary>
        /// 
        /// </summary>
        /// <param name="pawn"></param>
        public abstract void SurgeryResult(Pawn pawn);

        /// <summary>
        /// Setts the severity of the Beautiful trait for the selected pawn.
        /// If the trait doesn't exist in the pawns traits, it will be added.
        /// </summary>
        /// <param name="pawn">the pawn to modify</param>
        /// <param name="severity">the new severity of the pawn's Beautiful trait</param>
        protected void SurgeryX(Pawn pawn, int severity)
        {
            if (pawn.story.traits.HasTrait(Beauty))
            {
                pawn.story.traits.allTraits.FindAll(t => Beauty.ConflictsWith(t))
                    .ForEach(t => pawn.story.traits.RemoveTrait(t));
                pawn.story.traits.RemoveTrait(pawn.story.traits.allTraits.Find(t => t.def == Beauty));
            }

            pawn.story.traits.GainTrait(new Trait(Beauty, severity));
        }
    }

    /// <inheritdoc />
    public class Recipe_Surgery_Beautify_Beautiful : Recipe_Surgery_Beautify
    {
        /// <inheritdoc />
        public override void SurgeryResult(Pawn pawn) => SurgeryX(pawn, 2);
    }
}