Auto Updater done!

Author: ArjixWasTaken

Server/AutoUpdater/Update.cs

@@ -1,6 +1,42 @@
-namespace Server.AutoUpdater;
+using System.Diagnostics;
+using System.IO;
+using System.Linq;
+using UnityEngine;
+
+namespace Server.AutoUpdater;
 
 public static class Update
 {
+    public static string GetUpdateScript(string url, string path)
+    {
+        var commands = new string[] {
+            "Wait-Process -Name 'INVAXION' -ErrorAction SilentlyContinue",
+            $"Invoke-WebRequest -Uri '{url}' -OutFile '{path}'",
+            "Remove-Item -Path $MyInvocation.MyCommand.Source"
+        };
+        return string.Join("\n", commands);
+    }
+
+    public static void Procceed(GithubReleases.Release release)
+    {
+        var assetLink = release.assets.Where(asset => asset.name == "ServerEmulator.dll").FirstOrDefault().browser_download_url;
+        var outFile = Path.Combine(Path.Combine(Path.Combine(Directory.GetCurrentDirectory(), "BepInEx"), "plugins"), "ServerEmulator.dll");
+        var script = GetUpdateScript(assetLink, outFile);
+
+        var scriptName = $"ServerEmulator_{new Tag(release.tag_name)}_update.ps1";
+        var scriptPath = Path.Combine(Path.GetTempPath(), scriptName);
+        File.WriteAllText(scriptPath, script);
 
+        var updateProcess = new Process()
+        {
+            StartInfo =
+                {
+                    FileName = "powershell.exe",
+                    Arguments = $"-ExecutionPolicy ByPass -File {scriptName}",
+                    WorkingDirectory = Path.GetTempPath(),
+                }
+        };
+        updateProcess.Start();
+        Application.Quit();
+    }
 }

Server/GeneralPatches/TipHelperInputPatch.cs

@@ -0,0 +1,37 @@
+using Aquatrax;
+using HarmonyLib;
+using UnityEngine;
+
+namespace Server.GeneralPatches;
+
+
+public static class TipHelperInputPatches
+{
+    private static readonly Harmony _harmony = new Harmony("tiphelper-patches");
+
+
+    public static bool PatchedUpdate(ref TipHelper __instance)
+    {
+        // The original function used Input.GetKeyDown() for some reason, so I changed it to Input.GetKeyUp() instead.
+
+        if (InputManager.currentRef.getCurentGroup() == "TipHelper")
+        {
+            if (Input.GetKeyUp(KeyCode.Escape) || InputDevice.isButtonChecked_Cancel)
+            {
+                __instance.ButtonCancel();
+            }
+            else if (Input.GetKeyUp(KeyCode.Return) || InputDevice.isButtonChecked_Submit)
+            {
+                __instance.ButtonCommit();
+            }
+        }
+        return false;
+    }
+
+    public static void Hook()
+    {
+        var TipHelper__Update = AccessTools.Method(typeof(TipHelper), "Update");
+        var TipHelperInputPatches__PatchedUpdate = AccessTools.Method(typeof(TipHelperInputPatches), "PatchedUpdate");
+        _harmony.Patch(TipHelper__Update, prefix: new HarmonyMethod(TipHelperInputPatches__PatchedUpdate));
+    }
+}

Server/Server.cs

@@ -20,6 +20,7 @@ public class Server : BaseUnityPlugin
     public static bool Debug = true;
     private static Process _process;
     private static bool ShuttingDown = false;
+    private static bool ManiaLoaderDisabled = true;
 
     private void Awake()
     {
@@ -29,6 +30,7 @@ private void Awake()
         PlaceholderServerData = new();
         Logger.LogInfo($"Plugin {PluginInfo.PLUGIN_GUID} is loaded!");
 
+        GeneralPatches.TipHelperInputPatches.Hook();
         HookManager.Instance.Create();
         var currentVersion = new AutoUpdater.Tag(PluginInfo.PLUGIN_VERSION);
 
@@ -63,12 +65,7 @@ void ShowUpdateDialog()
                                 Aquatrax.TipHelper.Instance.InitText($"An update for ServerEmulator has been found ({currentVersion} -> {releaseVersion}), update now?");
                                 Aquatrax.TipHelper.Instance.Commit = delegate ()
                                 {
-                                    // TODO: Download the update and install it.
-                                    Logger.LogInfo("Clicked yes!");
-                                };
-                                Aquatrax.TipHelper.Instance.Cancel = delegate ()
-                                {
-                                    Logger.LogInfo("Clicked no!");
+                                    AutoUpdater.Update.Procceed(NewVersion);
                                 };
                             });
                             DiscordRichPresence.GameEvents.switchScene -= ShowUpdateDialog;
@@ -79,8 +76,11 @@ void ShowUpdateDialog()
             }
         })));
 
-        ManiaBeatmapsLoader = new OsuManiaLoader.Loader();
-        if (ManiaBeatmapsLoader.BeatmapPacks.Count > 0) Logger.LogInfo($"Loaded {ManiaBeatmapsLoader.BeatmapPacks.Count} osu!mania packs!");
+        if (!ManiaLoaderDisabled)
+        {
+            ManiaBeatmapsLoader = new OsuManiaLoader.Loader();
+            if (ManiaBeatmapsLoader.BeatmapPacks.Count > 0) Logger.LogInfo($"Loaded {ManiaBeatmapsLoader.BeatmapPacks.Count} osu!mania packs!");
+        }
 
         if (File.Exists(Path.Combine(Path.Combine("INVAXION_Data", "Plugins"), "discord_game_sdk.dll")))
         {
@@ -149,6 +149,8 @@ private void OnApplicationQuit()
             _process = null;
         }
 
+        if (!Debug) return;
+
         var commands = new List<string>();
 
         foreach (var command in MustImplement)

Server/Server.csproj

@@ -3,7 +3,7 @@
     <TargetFramework>net35</TargetFramework>
     <AssemblyName>ServerEmulator</AssemblyName>
     <Description>INVAXION Server Emulator</Description>
-    <Version>1.0.2</Version>
+    <Version>1.0.3</Version>
     <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
     <LangVersion>latest</LangVersion>
   </PropertyGroup>