Community resources patches

Author: cheese3660

Runtime/PatchManager.Resources/NonStageableResourcesUIController.cs

@@ -0,0 +1,113 @@
+using System;
+using KSP.Game;
+using KSP.Messages;
+using KSP.Sim.impl;
+using KSP.Sim.ResourceSystem;
+using PatchManager.Shared;
+using UnityEngine;
+
+namespace PatchManager.Resources
+{
+    /// <summary>
+    /// Handles dynamic resizing of non-stageable resources flight HUD window based on number of resources
+    /// </summary>
+    public class NonStageableResourcesUIController : KerbalMonoBehaviour
+    {
+        private const string NON_STAGEABLE_RESOURCES_UI_PANEL_PATH = "GameManager/Default Game Instance(Clone)/UI Manager(Clone)/Scaled Main Canvas/FlightHudRoot(Clone)/NonStageableResources(Clone)/KSP2UIWindow/Root/UIPanel";
+        private VesselComponent? _activeVessel;
+        private GameObject? _nsResourcesUI;
+        private bool _needNsUiUpdate = false;
+
+        private void Start()
+        {
+            Game.Messages.PersistentSubscribe<VesselChangedMessage>(OnVesselChanged);
+        }
+
+        private void OnDestroy()
+        {
+            if (IsGameShuttingDown)
+                return;
+            Game.Messages.Unsubscribe<VesselChangedMessage>(OnVesselChanged);
+        }
+
+        private void Update()
+        {
+            // Get active vessel
+            _activeVessel = Game?.ViewController?.GetActiveSimVessel();
+
+            if (_activeVessel is null)
+                return;
+
+            if (!_needNsUiUpdate) return;
+            
+            _needNsUiUpdate = false;
+
+            // Update Non Stageable Resources UI size
+            UpdateNonStageableResourcesUI();
+        }
+
+        private void OnVesselChanged(MessageCenterMessage msg)
+        {
+            if (msg is not VesselChangedMessage vesselChangedMessage || vesselChangedMessage == null)
+                return;
+
+            _needNsUiUpdate = true;
+        }
+
+        /// <summary>
+        /// Updates the height & position of the non-stageable resources UI
+        /// </summary>
+        /// <param name="nsResourcesCount"></param>
+        private void UpdateNonStageableResourcesUI()
+        {
+            // Get number of non-stageable resources in the active vessel
+            int nsResourcesCount = GetNonStageableResourcesCount(_activeVessel);
+
+            Logging.LogInfo(
+                $"Updating the Non-Stageable Resource UI. Found {nsResourcesCount} Resources to display.");
+
+            // Find the non-stageable resources window game object
+            _nsResourcesUI =
+                GameObject.Find(
+                    NON_STAGEABLE_RESOURCES_UI_PANEL_PATH);
+
+            if (_nsResourcesUI == null) return;
+            
+            // Number of resources above the default maximum of 2 (EC & Monoprop)
+            var extraResources = Math.Max(0, nsResourcesCount - 2);
+
+            // Get the UI object's RectTransform
+            var rect = _nsResourcesUI.GetComponent<RectTransform>();
+
+            // Increase the UI object's vertical size
+            rect.sizeDelta = new Vector2(0, 30 * extraResources);
+
+            // Shift the UI object down
+            rect.localPosition = extraResources * 15 * Vector3.down;
+        }
+
+        /// <summary>
+        /// Returns the number of non-stageable resources present on the vessel
+        /// </summary>
+        private int GetNonStageableResourcesCount(VesselComponent vessel)
+        {
+            var count = 0;
+
+            var controlOwner = vessel.GetControlOwner();
+            
+            if (controlOwner == null || controlOwner.PartOwner == null) return count;
+            
+            var containerGroup = controlOwner.PartOwner.ContainerGroup;
+            
+            foreach (var containedResourceData in containerGroup.GetAllResourcesContainedData())
+            {
+                var definitionData =
+                    Game.ResourceDefinitionDatabase.GetDefinitionData(containedResourceData.ResourceID);
+                if (definitionData.resourceProperties.NonStageable)
+                    ++count;
+            }
+
+            return count;
+        }
+    }
+}

Runtime/PatchManager.Resources/NonStageableResourcesUIController.cs.meta

@@ -1,5 +1,12 @@
-using JetBrains.Annotations;
+using System.Collections.Generic;
+using JetBrains.Annotations;
+using KSP.Game;
+using Newtonsoft.Json.Linq;
+using PatchManager.Shared;
 using PatchManager.Shared.Modules;
+using Unity.VisualScripting;
+using UnityEngine.AddressableAssets;
+using UnityEngine;
 
 namespace PatchManager.Resources
 {
@@ -9,5 +16,35 @@ namespace PatchManager.Resources
     [UsedImplicitly]
     public class ResourcesModule : BaseModule
     {
+        /// <summary>
+        /// Implement the resource units from https://github.com/KSP2Community/CommunityResources
+        /// </summary>
+        internal static readonly Dictionary<string, string> ResourceUnits = new();
+
+
+        public override void Load()
+        {
+            GameManager.Instance.Assets.LoadByLabel("resource_units", RegisterUnits,
+                delegate(IList<TextAsset> assetLocations)
+                {
+                    if (assetLocations != null)
+                    {
+                        Addressables.Release(assetLocations);
+                    }
+                });
+            
+            PatchManager.Instance.AddComponent<NonStageableResourcesUIController>();
+        }
+
+
+        private static void RegisterUnits(TextAsset textAsset)
+        {
+            var dict = JObject.Parse(textAsset.text);
+            foreach (var value in dict)
+            {
+                if (value.Value is not { Type: JTokenType.String }) continue;
+                ResourceUnits[value.Key] = value.Value.Value<string>()!;
+            }
+        }
     }
 }

Runtime/PatchManager.Resources/ResourcesModule.cs

@@ -0,0 +1,32 @@
+using System.Collections.Generic;
+using Newtonsoft.Json.Linq;
+using PatchManager.SassyPatching;
+using PatchManager.SassyPatching.Attributes;
+using PatchManager.SassyPatching.Interfaces;
+using PatchManager.SassyPatching.NewAssets;
+using PatchManager.SassyPatching.Selectables;
+
+namespace PatchManager.Resources.Rulesets
+{
+    /// <summary>
+    /// Implements the resource units ruleset from https://github.com/KSP2Community/CommunityResources
+    /// </summary>
+    [PatcherRuleset("resource_units","resource_units")]
+    public class ResourceUnitRuleset : IPatcherRuleSet
+    {
+        public bool Matches(string label) => label == "resource_units";
+
+        public ISelectable ConvertToSelectable(string type, string name, string jsonData) => new JTokenSelectable(() =>
+        {
+        }, JObject.Parse(jsonData), name, type);
+
+        private static int _globallyIncrementingId = 0;
+        
+        public INewAsset CreateNew(List<DataValue> dataValues)
+        {
+            var id = _globallyIncrementingId++;
+            return new NewGenericAsset("resource_units", $"resource_unit_definition_{id}", new JTokenSelectable(
+                () => { }, new JObject(), $"resource_unit_definition_{id}", "resource_units"));
+        }
+    }
+}

Runtime/PatchManager.Resources/Rulesets/ResourceUnitRuleset.cs

@@ -19,9 +19,11 @@ namespace PatchManager
 {
     public class PatchManager : KerbalMod
     {
+        internal static PatchManager Instance;
         public void Awake()
         {
             // Let's register all our modules!
+            Instance = this;
             ModuleManager.Register(typeof(CoreModule));
             ModuleManager.Register(typeof(GenericModule));
             ModuleManager.Register(typeof(MissionsModule));