1.1

Source/ResearchWhatever15/Patches/WorkGiver_ResearcherPatch.cs

@@ -0,0 +1,86 @@
+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.CostApparent);
+
+            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);
+        }
+    }
+}