JobHandler.cs

using DV;
using DV.Booklets;
using DV.Booklets.Rendered;
using DV.Common;
using DV.InventorySystem;
using DV.Logic.Job;
using DV.ServicePenalty;
using DV.ThingTypes;
using DV.Utils;
using HarmonyLib;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using UnityEngine;
using static HarmonyLib.AccessTools;

namespace dvRadioJobControl
{
    public static class JobHandler
    {
        private static FieldRef<StationController, List<JobOverview>> SpawnedJobOverviews = AccessTools.FieldRefAccess<StationController, List<JobOverview>>("spawnedJobOverviews");
        private static Action<JobOverview>? FoundJobOverviewAction;

        //code taken partially from vanilla
        public static void HandInJob(Job job)
        {
            bool completed = false;
            JobState newState = SingletonBehaviour<JobsManager>.Instance.TryToCompleteAJob(job);
            if (newState == JobState.Completed) completed = true;

            DisplayableDebt displayableDebt = (completed ? SingletonBehaviour<JobDebtController>.Instance.LastStagedJobDebt : SingletonBehaviour<JobDebtController>.Instance.GetExistingJobDebtForJob(job));
            if (displayableDebt != null && !displayableDebt.IsStaged) displayableDebt.UpdateDebtState();
            var jobReport = BookletCreator.CreateJobReport(job, displayableDebt, Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
            CommsRadioController.PlayAudioFromRadio(JobValidator_Awake_Patch.ValidateSound, PlayerManager.PlayerCamera.transform);
            Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobReport.gameObject));

            if (completed)
            {
                JobBooklet b = JobBooklet.allExistingJobBooklets.FirstOrDefault(jb => jb.job.ID == job.ID);
                SingletonBehaviour<Inventory>.Instance.DropItemFromHandsOrInventory(b.gameObject);
                b.DestroyJobBooklet();
                SingletonBehaviour<Inventory>.Instance.AddMoney(job.GetWageForTheJob());
                CommsRadioController.PlayAudioFromRadio(MoneyPrinterJobValidator_Awake_Patch.MoneySound, PlayerManager.PlayerCamera.transform);
            }
        }

        private static IEnumerator CreateAndInitializePrintedObject(GameObject printedObject)
        {
            yield return null;
            CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.PrintingSound, PlayerManager.PlayerCamera.transform);
            while ((!(printedObject.GetComponent<RenderedTexturesBase>()?.IsGenerated())) is true) yield return null;

            if (SingletonBehaviour<Inventory>.Instance && SingletonBehaviour<Inventory>.Instance.HasFreeSlots()) SingletonBehaviour<Inventory>.Instance.AddItemToInventory(printedObject);
            else if (PlayerManager.PlayerCamera != null)
            {
                printedObject.transform.position = PlayerManager.PlayerCamera.transform.position + PlayerManager.PlayerCamera.transform.forward * 0.75f;
                Rigidbody component = printedObject.GetComponent<Rigidbody>();
                if (component != null)
                {
                    component.velocity = Vector3.zero;
                    component.angularVelocity = Vector3.zero;
                }

                if ((bool)SingletonBehaviour<StorageController>.Instance) SingletonBehaviour<StorageController>.Instance.AddItemToWorldStorage(printedObject);
                else UnityEngine.Debug.LogError("StorageController doesn't exist! Can't add item to WorldStorage!");
            }
            else UnityEngine.Debug.LogError("Unexpected: No free space in inventory and PlayerCamera not found!");
        }

        public static void TryTakeOrDiscardJob(Job job, bool discard = false)
        {
            var sc = StationController.GetStationByYardID(job.ID.Split('-')[0]);
            if (sc != null && sc.logicStation.availableJobs.Contains(job))
            {
                FoundJobOverviewAction += TryAcceptJobOverview;
                Main.RadioEntry.StartCoroutine(GetJobOverview(job, sc, discard));
            }
        }

        private static IEnumerator GetJobOverview(Job job, StationController sc, bool discard = false)
        {
            SpawnedJobOverviews ??= AccessTools.FieldRefAccess<StationController, List<JobOverview>>("spawnedJobOverviews");
            List<JobOverview>? joList = SpawnedJobOverviews(sc);
            if ((joList?.Any(jo => jo.job.ID == job.ID)) is not true)
            {
                joList = null;
                StationJobGenerationRange_IsPlayerInRangeForBookletGeneration_Patch.scToOverride = sc;
                yield return new WaitUntil(() =>
                {
                    Stopwatch st = Stopwatch.StartNew();
                    joList = SpawnedJobOverviews(sc);
                    return ((joList is not null && joList is { Count: > 0 }) || st.ElapsedMilliseconds > 5000);
                });
            }
            var jobOverview = joList.FirstOrDefault(jo => jo.job.ID == job.ID);
            if (discard)
            {
                if (jobOverview != null)
                {
                    jobOverview.job.ExpireJob();
                    jobOverview.DestroyJobOverview();
                }
            }
            else
            {
                FoundJobOverviewAction?.Invoke(jobOverview);
                FoundJobOverviewAction -= TryAcceptJobOverview;
            }
            yield return new WaitForSeconds(2f);
            StationJobGenerationRange_IsPlayerInRangeForBookletGeneration_Patch.scToOverride = null;
        }

        private static void TryAcceptJobOverview(JobOverview jobOverview)
        {
            if (jobOverview == null)
            {
                UnityEngine.Debug.LogError("jobOverview is null, this shouldn´t have happened");
                CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                return;
            }

            JobState state = jobOverview.job.State;
            if (state != JobState.Available)
            {
                if (state != JobState.Expired)
                {
                    UnityEngine.Debug.LogError(string.Concat(["Job[", jobOverview.job.ID, "] was already taken, but jobOverview: ", jobOverview.name, " was not destroyed! Destroying jobOverview"]));
                    CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                    jobOverview.DestroyJobOverview();
                    return;
                }
                var jobEReport = BookletCreator.CreateJobExpiredReport(jobOverview.job, Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
                CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobEReport.gameObject));
                jobOverview.DestroyJobOverview();
                return;
            }
            else
            {
                StationController stationController = StationController.allStations.FirstOrDefault<StationController>((StationController st) => st.logicStation.availableJobs.Contains(jobOverview.job));
                if (!(stationController != null))
                {
                    UnityEngine.Debug.LogError("Job[" + jobOverview.job.ID + "] is in available state, but is not part of any station! Destroying jobOverview");
                    jobOverview.DestroyJobOverview();
                    return;
                }
                Job job = jobOverview.job;
                if (!GameFeatureFlags.IsAllowed(GameFeatureFlags.Flag.UseJobValidator))
                {
                    var jobTWReport = BookletCreator.CreateTutorialWarningReport(Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
                    CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                    Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobTWReport.gameObject));
                    return;
                }
                if (SingletonBehaviour<JobsManager>.Instance.currentJobs.Count >= SingletonBehaviour<LicenseManager>.Instance.GetNumberOfAllowedConcurrentJobs())
                {
                    var jobMLReport = BookletCreator.CreateMissingLicenseReport(job, false, Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
                    CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                    Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobMLReport.gameObject));
                    return;
                }
                if (!SingletonBehaviour<LicenseManager>.Instance.IsLicensedForJob(JobLicenseType_v2.ToV2List(job.requiredLicenses)))
                {
                    var jobML2Report = BookletCreator.CreateMissingLicenseReport(job, true, Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
                    CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                    Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobML2Report.gameObject));
                    return;
                }
                if (!SingletonBehaviour<CareerManagerDebtController>.Instance.IsPlayerAllowedToTakeJob())
                {
                    var jobDWReport = BookletCreator.CreateDebtWarningReport(Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent);
                    CommsRadioController.PlayAudioFromRadio(PrinterController_Awake_Patch.ErrorSound, PlayerManager.PlayerCamera.transform);
                    Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobDWReport.gameObject));
                    return;
                }
                stationController.TakeJobFromStation(jobOverview);
                var jobReport = BookletCreator.CreateJobBooklet(job, Vector3.zero, Quaternion.identity, WorldMover.OriginShiftParent, true);
                CommsRadioController.PlayAudioFromRadio(JobValidator_Awake_Patch.ValidateSound, PlayerManager.PlayerCamera.transform);
                Main.RadioEntry.StartCoroutine(CreateAndInitializePrintedObject(jobReport.gameObject));
                return;
            }
        }
    }
}