RimWorld_ResearchWhatever/Source/ResearchWhatever15/Patches/WorkGiver_ResearcherPatch.cs

using System.Linq;
using System.Collections.Generic;
using HarmonyLib;
using RimWorld;
using Verse;
using Verse.Sound;

namespace ResearchWhatever.Patches
{
    [HarmonyPatch(typeof(WorkGiver_Researcher), "ShouldSkip")]
    public static class WorkGiver_Researcher_ShouldSkip_ResearchWhateverPatch
    {
        public static bool Prefix(ref bool __result)
        {
            __result = false;
            return false;
        }
    }

    [HarmonyPatch(typeof(WorkGiver_Researcher), "PotentialWorkThingRequest", MethodType.Getter)]
    public static class WorkGiver_PotentialWorkThingRequest_ResearchWhateverPatch
    {
        public static bool Prefix(ref ThingRequest __result)
        {
            __result = ThingRequest.ForGroup(ThingRequestGroup.ResearchBench);
            return false;
        }
    }

    [HarmonyPatch(typeof(WorkGiver_Researcher), "HasJobOnThing")]
    public static class WorkGiver_Researcher_HasJobOnThing_ResearchWhateverPatch
    {
        private static bool HasFacilities(this Building_ResearchBench bench, List<ThingDef> requiredFacilities)
        {
            if (requiredFacilities.NullOrEmpty()) return true;
            CompAffectedByFacilities comp = bench.TryGetComp<CompAffectedByFacilities>();
            if (comp == null) return false;

            foreach (var rf in requiredFacilities)
                if (comp.LinkedFacilitiesListForReading.FirstOrDefault(x => x.def == rf && comp.IsFacilityActive(x)) == null)
                    return false;

            return true;
        }

        private static bool CanBePicked(this ResearchProjectDef research, Building_ResearchBench bench)
        {
            return research.baseCost > 0f
                && research.CanStartNow
                && Find.Storyteller.difficulty.AllowedBy(research.hideWhen)
                && (research.requiredResearchBuilding == null || research.requiredResearchBuilding == bench.def && bench.HasFacilities(research.requiredResearchFacilities))
                && research.GetModExtension<ResearchWhateverExtansion>()?.ignore != true;
        }

        internal static void Prefix(Pawn pawn, Thing t, bool forced)
        {
            ResearchProjectDef currentProj = Find.ResearchManager.GetProject();
            if (currentProj != null) return;
            if (!(t is Building_ResearchBench bench)) return;

            ResearchWhateverComp comp = bench.TryGetComp<ResearchWhateverComp>();

            if (comp == null || !comp.Active) return;

            List<ResearchProjectDef> projects = new List<ResearchProjectDef>(
                from x in DefDatabase<ResearchProjectDef>.AllDefsListForReading
                where x.CanBePicked(bench)
                select x);

            if (projects.NullOrEmpty())
            {
                comp.Active = false;
                Messages.Message("ResearchWhateverNothingLeftToResearch".Translate(bench.Label).CapitalizeFirst(), new TargetInfo(bench.Position, bench.Map, false), MessageTypeDefOf.NeutralEvent);
                return;
            }
            projects.SortBy(x => x.GetModExtension<ResearchWhateverExtansion>()?.lowPriority == true ? 100000000f + x.CostApparent - x.ProgressApparent : x.CostApparent - x.ProgressApparent);

            ResearchProjectDef def = projects.First();
            projects.TryRandomElementByWeight(x => x.CostApparent == def.CostApparent ? 1f : 0f, out def);
            SoundDefOf.ResearchStart.PlayOneShotOnCamera(null);
            Find.ResearchManager.SetCurrentProject(def);
            TutorSystem.Notify_Event("StartResearchProject");
            Messages.Message( "ResearchWhateverNewResearch".Translate(pawn.Name.ToStringFull, def.label).CapitalizeFirst(), new TargetInfo(bench.Position, bench.Map, false), MessageTypeDefOf.SilentInput);
        }
    }
}