Some minor cleanup

Pespiri · Jan 28, 2020 · 5c7f934 · 5c7f934
1 parent 5d94e43
commit 5c7f934
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/CustomWalls/Settings/UI/MaterialListViewController.cs b/CustomWalls/Settings/UI/MaterialListViewController.cs
@@ -10,7 +10,7 @@
 
 namespace CustomWalls.Settings.UI
 {
-    internal class MaterialListView : BSMLResourceViewController
+    internal class MaterialListViewController : BSMLResourceViewController
     {
         public override string ResourceName => "CustomWalls.Settings.UI.Views.materialList.bsml";
 

diff --git a/CustomWalls/Settings/UI/MaterialsFlowCoordinator.cs b/CustomWalls/Settings/UI/MaterialsFlowCoordinator.cs
@@ -1,24 +1,30 @@
 using BeatSaberMarkupLanguage;
 using HMUI;
-using IPA.Utilities;
 using System;
 
 namespace CustomWalls.Settings.UI
 {
     internal class MaterialsFlowCoordinator : FlowCoordinator
     {
-        private MaterialListView materialsListView;
+        private MaterialListViewController materialsListView;
         private MaterialPreviewViewController materialsPreviewView;
         private MaterialDetailsViewController materialsDescriptionView;
 
         public void Awake()
         {
-            if (materialsListView == null)
+            if (!materialsPreviewView)
             {
-                materialsListView = BeatSaberUI.CreateViewController<MaterialListView>();
                 materialsPreviewView = BeatSaberUI.CreateViewController<MaterialPreviewViewController>();
+            }
+
+            if (!materialsDescriptionView)
+            {
                 materialsDescriptionView = BeatSaberUI.CreateViewController<MaterialDetailsViewController>();
+            }
 
+            if (!materialsListView)
+            {
+                materialsListView = BeatSaberUI.CreateViewController<MaterialListViewController>();
                 materialsListView.customMaterialChanged += materialsDescriptionView.OnMaterialWasChanged;
             }
         }
@@ -43,8 +49,7 @@ protected override void DidActivate(bool firstActivation, ActivationType activat
         protected override void BackButtonWasPressed(ViewController topViewController)
         {
             // Dismiss ourselves
-            MainFlowCoordinator mainFlow = BeatSaberUI.MainFlowCoordinator;
-            mainFlow.InvokePrivateMethod("DismissFlowCoordinator", this, null, false);
+            BeatSaberUI.MainFlowCoordinator.DismissFlowCoordinator(this, null, false);
         }
     }
 }