From 86b9db821a2b32eaed20ac49cd099920e27646f0 Mon Sep 17 00:00:00 2001
From: John Michael Gross <mindstorms11@cox.net>
Date: Mon, 17 Oct 2016 01:14:47 -0700
Subject: [PATCH] idk i wrote this 3 weeks ago

---
 .../MTObservableCollection.cs                 |   0
 Piston/{ => Collections}/PlayerInfoCache.cs   |   0
 .../MyPlayerCollectionExtensions.cs           |   0
 Piston/{ => Extensions}/StringExtensions.cs   |   0
 Piston/Piston.csproj                          |  10 +-
 Piston/PistonPlugin.cs                        |  13 ++
 PistonClient/GameInitializer.cs               | 158 +++++++++++++++++
 PistonClient/Piston.Client.csproj             |   6 +
 PistonClient/PistonConsoleScreen.cs           |  63 +++++++
 PistonClient/PistonSettingsScreen.cs          |   3 +-
 PistonClient/Program.cs                       | 159 +-----------------
 11 files changed, 256 insertions(+), 156 deletions(-)
 rename Piston/{ => Collections}/MTObservableCollection.cs (100%)
 rename Piston/{ => Collections}/PlayerInfoCache.cs (100%)
 rename Piston/{ => Extensions}/MyPlayerCollectionExtensions.cs (100%)
 rename Piston/{ => Extensions}/StringExtensions.cs (100%)
 create mode 100644 Piston/PistonPlugin.cs
 create mode 100644 PistonClient/GameInitializer.cs
 create mode 100644 PistonClient/PistonConsoleScreen.cs

diff --git a/Piston/MTObservableCollection.cs b/Piston/Collections/MTObservableCollection.cs
similarity index 100%
rename from Piston/MTObservableCollection.cs
rename to Piston/Collections/MTObservableCollection.cs
diff --git a/Piston/PlayerInfoCache.cs b/Piston/Collections/PlayerInfoCache.cs
similarity index 100%
rename from Piston/PlayerInfoCache.cs
rename to Piston/Collections/PlayerInfoCache.cs
diff --git a/Piston/MyPlayerCollectionExtensions.cs b/Piston/Extensions/MyPlayerCollectionExtensions.cs
similarity index 100%
rename from Piston/MyPlayerCollectionExtensions.cs
rename to Piston/Extensions/MyPlayerCollectionExtensions.cs
diff --git a/Piston/StringExtensions.cs b/Piston/Extensions/StringExtensions.cs
similarity index 100%
rename from Piston/StringExtensions.cs
rename to Piston/Extensions/StringExtensions.cs
diff --git a/Piston/Piston.csproj b/Piston/Piston.csproj
index 231e886..4befa94 100644
--- a/Piston/Piston.csproj
+++ b/Piston/Piston.csproj
@@ -123,14 +123,15 @@
   <ItemGroup>
     <Compile Include="ConnectionState.cs" />
     <Compile Include="Logger.cs" />
+    <Compile Include="PistonPlugin.cs" />
     <Compile Include="PlayerInfo.cs" />
-    <Compile Include="PlayerInfoCache.cs" />
+    <Compile Include="Collections\PlayerInfoCache.cs" />
     <Compile Include="SteamService.cs" />
     <Compile Include="ViewModels\ModViewModel.cs" />
-    <Compile Include="MTObservableCollection.cs" />
-    <Compile Include="MyPlayerCollectionExtensions.cs" />
+    <Compile Include="Collections\MTObservableCollection.cs" />
+    <Compile Include="Extensions\MyPlayerCollectionExtensions.cs" />
     <Compile Include="SteamHelper.cs" />
-    <Compile Include="StringExtensions.cs" />
+    <Compile Include="Extensions\StringExtensions.cs" />
     <Compile Include="ViewModels\ViewModel.cs" />
     <Compile Include="PluginManager.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
@@ -142,6 +143,7 @@
       <Name>Piston.API</Name>
     </ProjectReference>
   </ItemGroup>
+  <ItemGroup />
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
diff --git a/Piston/PistonPlugin.cs b/Piston/PistonPlugin.cs
new file mode 100644
index 0000000..814b781
--- /dev/null
+++ b/Piston/PistonPlugin.cs
@@ -0,0 +1,13 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Piston
+{
+    public class PistonPlugin
+    {
+
+    }
+}
diff --git a/PistonClient/GameInitializer.cs b/PistonClient/GameInitializer.cs
new file mode 100644
index 0000000..5a96652
--- /dev/null
+++ b/PistonClient/GameInitializer.cs
@@ -0,0 +1,158 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using System.Windows;
+using Sandbox;
+using Sandbox.Game;
+using Sandbox.Game.Gui;
+using Sandbox.Graphics;
+using Sandbox.Graphics.GUI;
+using Sandbox.Gui;
+using SpaceEngineers.Game;
+using SpaceEngineers.Game.GUI;
+using VRage.FileSystem;
+using VRage.Game;
+using VRage.Input;
+using VRage.Utils;
+using VRageRender;
+using Game = Sandbox.Engine.Platform.Game;
+
+namespace Piston.Client
+{
+    public class GameInitializer
+    {
+        private MyCommonProgramStartup _startup;
+        private IMyRender _renderer;
+        private const uint APP_ID = 244850;
+        private PluginManager _pluginManager;
+        private readonly string[] _args;
+        private VRageGameServices _services;
+
+        public GameInitializer(string[] args)
+        {
+            _args = args;
+        }
+
+        public void TryInit()
+        {
+            if (!File.Exists("steam_appid.txt"))
+            {
+                Directory.SetCurrentDirectory(Path.GetDirectoryName(typeof(VRage.FastResourceLock).Assembly.Location) + "\\..");
+            }
+
+            //Add myself to the credits because I'm awesome.
+            var credits = new MyCreditsDepartment("Piston Developed By") { Persons = new List<MyCreditsPerson> { new MyCreditsPerson("JIMMACLE") } };
+            MyPerGameSettings.Credits.Departments.Add(credits);
+
+            SpaceEngineersGame.SetupBasicGameInfo();
+            _startup = new MyCommonProgramStartup(_args);
+            if (_startup.PerformReporting())
+                return;
+
+            _startup.PerformAutoconnect();
+            if (!_startup.CheckSingleInstance())
+                return;
+
+            var appDataPath = _startup.GetAppDataPath();
+            MyInitializer.InvokeBeforeRun(APP_ID, MyPerGameSettings.BasicGameInfo.ApplicationName, appDataPath, false);
+            MyInitializer.InitCheckSum();
+            if (!_startup.Check64Bit())
+                return;
+
+            _startup.DetectSharpDxLeaksBeforeRun();
+            using (var mySteamService = new SteamService(Game.IsDedicated, APP_ID))
+            {
+                _renderer = null;
+                SpaceEngineersGame.SetupPerGameSettings();
+
+                InitializeRender();
+
+                _services = new VRageGameServices(mySteamService);
+                if (!Game.IsDedicated)
+                    MyFileSystem.InitUserSpecific(mySteamService.UserId.ToString());
+            }
+
+            _startup.DetectSharpDxLeaksAfterRun();
+            MyInitializer.InvokeAfterRun();
+        }
+
+        public void RunGame()
+        {
+            using (var spaceEngineersGame = new SpaceEngineersGame(_services, _args))
+            {
+                Logger.Write("Starting SE...");
+                spaceEngineersGame.OnGameLoaded += SpaceEngineersGame_OnGameLoaded;
+                MyGuiSandbox.GuiControlCreated += GuiControlCreated;
+                spaceEngineersGame.Run();
+            }
+        }
+
+        private void GuiControlCreated(object o)
+        {
+            var menu = o as MyGuiScreenMainMenu;
+            if (menu != null)
+            {
+                var pistonBtn = new MyGuiControlImageButton
+                {
+                    Name = "PistonButton",
+                    Text = "Piston",
+                    HighlightType = MyGuiControlHighlightType.WHEN_CURSOR_OVER,
+                    Visible = true,
+                };
+
+                menu.Controls.Add(pistonBtn);
+            }
+        }
+
+        private void SpaceEngineersGame_OnGameLoaded(object sender, EventArgs e)
+        {
+            _pluginManager = new PluginManager();
+            _pluginManager.LoadAllPlugins();
+
+            //Fix Marek's name.
+            MyPerGameSettings.Credits.Departments[1].Persons[0].Name.Clear().Append("MEARK ROAS");
+            MyScreenManager.AddScreen(new PistonConsoleScreen());
+        }
+
+        private void InitializeRender()
+        {
+            try
+            {
+                if (Game.IsDedicated)
+                {
+                    _renderer = new MyNullRender();
+                }
+                else
+                {
+                    var graphicsRenderer = MySandboxGame.Config.GraphicsRenderer;
+                    if (graphicsRenderer == MySandboxGame.DirectX11RendererKey)
+                    {
+                        _renderer = new MyDX11Render();
+                        if (!_renderer.IsSupported)
+                        {
+                            MySandboxGame.Log.WriteLine("DirectX 11 renderer not supported. No renderer to revert back to.");
+                            _renderer = null;
+                        }
+                    }
+                    if (_renderer == null)
+                        throw new MyRenderException("The current version of the game requires a Dx11 card. \\n For more information please see : http://blog.marekrosa.org/2016/02/space-engineers-news-full-source-code_26.html", MyRenderExceptionEnum.GpuNotSupported);
+
+                    MySandboxGame.Config.GraphicsRenderer = graphicsRenderer;
+                }
+
+                MyRenderProxy.Initialize(_renderer);
+                MyRenderProxy.IS_OFFICIAL = true;
+                MyRenderProxy.GetRenderProfiler().SetAutocommit(false);
+                MyRenderProxy.GetRenderProfiler().InitMemoryHack("MainEntryPoint");
+            }
+            catch (Exception ex)
+            {
+                MessageBox.Show(ex.Message, "Render Initialization Failed");
+                Environment.Exit(-1);
+            }
+        }
+    }
+}
diff --git a/PistonClient/Piston.Client.csproj b/PistonClient/Piston.Client.csproj
index a8594e0..a3b2ed8 100644
--- a/PistonClient/Piston.Client.csproj
+++ b/PistonClient/Piston.Client.csproj
@@ -64,6 +64,10 @@
     <Reference Include="VRage.Game">
       <HintPath>C:\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.Game.dll</HintPath>
     </Reference>
+    <Reference Include="VRage.Input, Version=1.0.0.0, Culture=neutral, processorArchitecture=AMD64">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>C:\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.Input.dll</HintPath>
+    </Reference>
     <Reference Include="VRage.Library">
       <HintPath>C:\Program Files (x86)\Steam\steamapps\common\SpaceEngineers\Bin64\VRage.Library.dll</HintPath>
     </Reference>
@@ -82,6 +86,8 @@
     <Reference Include="PresentationFramework" />
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="GameInitializer.cs" />
+    <Compile Include="PistonConsoleScreen.cs" />
     <Compile Include="PistonSettingsScreen.cs" />
     <Compile Include="Program.cs" />
     <Compile Include="Properties\AssemblyInfo.cs">
diff --git a/PistonClient/PistonConsoleScreen.cs b/PistonClient/PistonConsoleScreen.cs
new file mode 100644
index 0000000..d8e1e0d
--- /dev/null
+++ b/PistonClient/PistonConsoleScreen.cs
@@ -0,0 +1,63 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using System.Windows;
+using Sandbox.Graphics;
+using Sandbox.Graphics.GUI;
+using Sandbox.Gui;
+using VRage.Utils;
+using VRageMath;
+
+namespace Piston.Client
+{
+    public class PistonConsoleScreen : MyGuiScreenBase
+    {
+        private MyGuiControlTextbox _textBox;
+
+        public override string GetFriendlyName()
+        {
+            return "Piston Console";
+        }
+
+        public PistonConsoleScreen() : base(isTopMostScreen: true)
+        {
+            BackgroundColor = new Vector4(0, 0, 0, 0.5f);
+            Size = new Vector2(0.5f);
+            RecreateControls(true);
+        }
+
+        public sealed override void RecreateControls(bool constructor)
+        {
+            Elements.Clear();
+            Elements.Add(new MyGuiControlLabel
+            {
+                Text = "Piston Console",
+                OriginAlign = MyGuiDrawAlignEnum.HORISONTAL_RIGHT_AND_VERTICAL_TOP,
+                Position = MyGuiManager.ComputeFullscreenGuiCoordinate(MyGuiDrawAlignEnum.HORISONTAL_RIGHT_AND_VERTICAL_TOP)
+            });
+
+            Controls.Clear();
+            _textBox = new MyGuiControlTextbox
+            {
+                BorderEnabled = false,
+                Enabled = true,
+                OriginAlign = MyGuiDrawAlignEnum.HORISONTAL_LEFT_AND_VERTICAL_TOP,
+                Position = new Vector2(-0.5f)
+            };
+            Controls.Add(_textBox);
+
+            var pistonBtn = new MyGuiControlImageButton
+            {
+                Name = "PistonButton",
+                Text = "Piston",
+                HighlightType = MyGuiControlHighlightType.WHEN_CURSOR_OVER,
+                Visible = true,
+                OriginAlign = MyGuiDrawAlignEnum.HORISONTAL_LEFT_AND_VERTICAL_TOP
+            };
+
+            Controls.Add(pistonBtn);
+        }
+    }
+}
diff --git a/PistonClient/PistonSettingsScreen.cs b/PistonClient/PistonSettingsScreen.cs
index 6df3ec7..c083f71 100644
--- a/PistonClient/PistonSettingsScreen.cs
+++ b/PistonClient/PistonSettingsScreen.cs
@@ -19,8 +19,7 @@ namespace Piston.Client
             RecreateControls(true);
         }
 
-        /// <inheritdoc />
-        public override void RecreateControls(bool constructor)
+        public sealed override void RecreateControls(bool constructor)
         {
             base.RecreateControls(constructor);
             AddCaption(MyStringId.GetOrCompute("Piston Settings"));
diff --git a/PistonClient/Program.cs b/PistonClient/Program.cs
index d2335a5..4bb1dc1 100644
--- a/PistonClient/Program.cs
+++ b/PistonClient/Program.cs
@@ -1,166 +1,25 @@
 using System;
-using System.Collections.Generic;
-using System.IO;
-using System.Security.Cryptography.X509Certificates;
-using System.Text;
 using System.Windows;
-using Sandbox;
-using Sandbox.Game;
-using SpaceEngineers.Game;
-using VRage.FileSystem;
-using VRageRender;
-using Piston;
-using Sandbox.Game.Gui;
-using Sandbox.Graphics;
-using Sandbox.Graphics.GUI;
-using Sandbox.Gui;
-using VRage.Game;
-using VRage.Utils;
-using VRageMath;
-using Game = Sandbox.Engine.Platform.Game;
-using MessageBoxResult = System.Windows.MessageBoxResult;
 
 namespace Piston.Client
 {
-    /// <summary>
-    ///     This is nearly all Keen's code.
-    /// </summary>
-    public static class MyProgram
+    public static class Program
     {
-        private static MyCommonProgramStartup _startup;
-        private static IMyRender _renderer;
-        private static readonly uint AppId = 244850u;
-        private static PluginManager _pluginManager;
-
-        private static void Main(string[] args)
+        public static void Main(string[] args)
         {
-            AppDomain.CurrentDomain.UnhandledException += CurrentDomain_UnhandledException;
-            if (!File.Exists("steam_appid.txt"))
-            {
-                Directory.SetCurrentDirectory(Path.GetDirectoryName(typeof(VRage.FastResourceLock).Assembly.Location) + "\\..");
-            }
+            var game = new GameInitializer(args);
 
-            //Add myself to the credits because I'm awesome.
-            var credits = new MyCreditsDepartment("Piston Developed By") {Persons = new List<MyCreditsPerson> {new MyCreditsPerson("JIMMACLE")}};
-            MyPerGameSettings.Credits.Departments.Add(credits);
-
-            SpaceEngineersGame.SetupBasicGameInfo();
-            _startup = new MyCommonProgramStartup(args);
-            if (_startup.PerformReporting())
-                return;
-
-            _startup.PerformAutoconnect();
-            if (!_startup.CheckSingleInstance())
-                return;
-
-            var appDataPath = _startup.GetAppDataPath();
-            MyInitializer.InvokeBeforeRun(AppId, MyPerGameSettings.BasicGameInfo.ApplicationName, appDataPath, false);
-            MyInitializer.InitCheckSum();
-            if (!_startup.Check64Bit())
-                return;
-
-            _startup.DetectSharpDxLeaksBeforeRun();
-            using (var mySteamService = new SteamService(Game.IsDedicated, AppId))
-            {
-                _renderer = null;
-                SpaceEngineersGame.SetupPerGameSettings();
-
-                SpaceEngineersGame.SetupRender();
-                try
-                {
-                    InitializeRender();
-                }
-                catch (MyRenderException ex)
-                {
-                    MessageBox.Show(ex.Message);
-                    return;
-                }
-
-                var services = new VRageGameServices(mySteamService);
-                if (!Game.IsDedicated)
-                    MyFileSystem.InitUserSpecific(mySteamService.UserId.ToString());
-                using (var spaceEngineersGame = new SpaceEngineersGame(services, args))
-                {
-                    Logger.Write("Starting SE...");
-                    spaceEngineersGame.OnGameLoaded += SpaceEngineersGame_OnGameLoaded;
-                    MyGuiSandbox.GuiControlCreated += GuiControlCreated;
-                    spaceEngineersGame.Run();
-                }
-            }
-
-            _startup.DetectSharpDxLeaksAfterRun();
-            MyInitializer.InvokeAfterRun();
-        }
-
-        private static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e)
-        {
-            var ex = (Exception)e.ExceptionObject;
-            MessageBox.Show($"PistonClient crashed. Go bug Jimmacle and send him a screenshot of this box!\n\n{ex.Message}\n{ex.StackTrace}", "Unhandled Exception");
-        }
-
-        private static void GuiControlCreated(object o)
-        {
-            var menu = o as MyGuiScreenMainMenu;
-            if (menu != null)
-            {
-                var pistonBtn = new MyGuiControlButton(MyGuiManager.ComputeFullscreenGuiCoordinate(MyGuiDrawAlignEnum.HORISONTAL_CENTER_AND_VERTICAL_CENTER), MyGuiControlButtonStyleEnum.Default, null, null, MyGuiDrawAlignEnum.HORISONTAL_CENTER_AND_VERTICAL_BOTTOM, null, new StringBuilder("Piston"), 1f, MyGuiDrawAlignEnum.HORISONTAL_CENTER_AND_VERTICAL_CENTER, MyGuiControlHighlightType.WHEN_ACTIVE, null, GuiSounds.MouseClick, 1f, null, false);
-                pistonBtn.Visible = true;
-                //menu.Controls.Add(pistonBtn);
-
-                var scr = new PistonSettingsScreen();
-                scr.Controls.Add(pistonBtn);
-
-                MyScreenManager.AddScreen(scr);
-                //menu.Controls.Add(new MyGuiControlLabel(new Vector2(0.5f), new Vector2(0.5f), "Piston Enabled"));
-                //menu.RecreateControls(false);
-            }
-        }
-
-        private static void SpaceEngineersGame_OnGameLoaded(object sender, System.EventArgs e)
-        {
-            _pluginManager = new PluginManager();
-            _pluginManager.LoadAllPlugins();
-
-            //Fix Marek's name.
-            MyPerGameSettings.Credits.Departments[1].Persons[0].Name.Clear().Append("MEARK ROAS");
-        }
-
-        private static void InitializeRender()
-        {
             try
             {
-                if (Game.IsDedicated)
-                {
-                    _renderer = new MyNullRender();
-                }
-                else
-                {
-                    var graphicsRenderer = MySandboxGame.Config.GraphicsRenderer;
-                    if (graphicsRenderer == MySandboxGame.DirectX11RendererKey)
-                    {
-                        _renderer = new MyDX11Render();
-                        if (!_renderer.IsSupported)
-                        {
-                            MySandboxGame.Log.WriteLine("DirectX 11 renderer not supported. No renderer to revert back to.");
-                            _renderer = null;
-                        }
-                    }
-                    if (_renderer == null)
-                        throw new MyRenderException("The current version of the game requires a Dx11 card. \\n For more information please see : http://blog.marekrosa.org/2016/02/space-engineers-news-full-source-code_26.html", MyRenderExceptionEnum.GpuNotSupported);
-
-                    MySandboxGame.Config.GraphicsRenderer = graphicsRenderer;
-                }
-
-                MyRenderProxy.Initialize(_renderer);
-                MyRenderProxy.IS_OFFICIAL = true;
-                MyRenderProxy.GetRenderProfiler().SetAutocommit(false);
-                MyRenderProxy.GetRenderProfiler().InitMemoryHack("MainEntryPoint");
+                game.TryInit();
             }
-            catch (TypeLoadException)
+            catch (Exception e)
             {
-                MessageBox.Show("This version of Piston does not run on the Stable branch.", "Initialization Error");
-                Environment.Exit(-1);
+                MessageBox.Show($"Piston encountered an error trying to initialize the game.\n{e.Message}");
+                return;
             }
+
+            game.RunGame();
         }
     }
 }
\ No newline at end of file