Add engineMode.propellant field (addresses #130)

untoldwind · Mar 5, 2024 · 90b7272 · 90b7272
1 parent f0c6041
commit 90b7272
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -1,5 +1,10 @@
 # Changes
 
+## 0.5.4.1 ->
+
+* Add `engineMode.propellant` field
+
+
 ## 0.5.4.0 -> 0.5.4.1
 
 Hotfix:

diff --git a/KSP2Runtime/KSPVessel/KSPVesselModule.EngineMode.cs b/KSP2Runtime/KSPVessel/KSPVesselModule.EngineMode.cs
@@ -1,4 +1,5 @@
-using KontrolSystem.TO2.Binding;
+using KontrolSystem.KSP.Runtime.KSPResource;
+using KontrolSystem.TO2.Binding;
 using KSP.Modules;
 
 namespace KontrolSystem.KSP.Runtime.KSPVessel;
@@ -25,5 +26,18 @@ public EngineModeAdapter(Data_Engine.EngineMode engineMode) {
         [KSField] public double MaxThrust => engineMode.maxThrust;
 
         [KSField] public EngineType EngineType => engineMode.engineType;
+
+        [KSField]
+        public KSPResourceModule.ResourceDefinitionAdapter Propellant {
+            get {
+                var context = KSPContext.CurrentContext;
+                var resourceId =
+                    context.Game.ResourceDefinitionDatabase.GetResourceIDFromName(engineMode.propellant.mixtureName);
+                var resourceDefinitionData =
+                    context.Game.ResourceDefinitionDatabase.GetDefinitionData(resourceId);
+
+                return new KSPResourceModule.ResourceDefinitionAdapter(resourceDefinitionData);
+            }
+        }
     }
 }
diff --git a/TO2/AST/VariableDeclaration.cs b/TO2/AST/VariableDeclaration.cs
@@ -1,5 +1,4 @@
-using System.Diagnostics.CodeAnalysis;
-using KontrolSystem.Parsing;
+using KontrolSystem.Parsing;
 using KontrolSystem.TO2.Generator;
 using KontrolSystem.TO2.Runtime;