Game update detection (#557)

* Detect any game updates and restart if found.

* Change log level and change wording.

Author: Bishbash777

Torch.API/ITorchConfig.cs

@@ -33,6 +33,8 @@ public interface ITorchConfig
         bool SendLogsToKeen { get; set; }
         bool DeleteMiniDumps { get; set; }
         string LoginToken { get; set; }
+        bool RestartOnGameUpdate { get; set; }
+        int GameUpdateRestartDelayMins { get; set; }
 
         void Save(string path = null);
     }

Torch.Server/Managers/GameUpdateManager.cs

@@ -0,0 +1,101 @@
+using System;
+using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using System.Xml.Linq;
+using NLog;
+using NLog.Fluent;
+using Sandbox.Game;
+using Torch.API;
+using Torch.API.Managers;
+using Torch.API.ModAPI;
+using Torch.Managers;
+using Torch.Managers.ChatManager;
+using Torch.Server;
+
+namespace Torch.Server.Managers
+{
+    public class GameUpdateManager : Manager
+    {
+        private CancellationTokenSource _cancellationTokenSource;
+        private readonly HttpClient _httpClient;
+        private const string UpdateCheckUrl = "https://mirror.keenswh.com/news/SpaceEngineersChangelog.xml";
+        
+        private static readonly Logger _log = LogManager.GetCurrentClassLogger();
+
+
+        public GameUpdateManager(ITorchBase torchInstance) : base(torchInstance)
+        {
+            _httpClient = new HttpClient();
+        }
+
+        /// <summary>
+        ///    Starts the game update manager.
+        /// </summary>
+        public override void Attach()
+        {
+            _log.Info("Starting game update manager");
+            base.Attach();
+            int updateIntervalSeconds = TorchBase.Instance.Config.GameUpdateRestartDelayMins * 60;
+            
+            _cancellationTokenSource = new CancellationTokenSource();
+
+            Task.Run(async () =>
+            {
+                while (!_cancellationTokenSource.Token.IsCancellationRequested)
+                {
+                    await CheckForUpdates();
+
+                    await Task.Delay(TimeSpan.FromSeconds(updateIntervalSeconds), _cancellationTokenSource.Token);
+                }
+            }, _cancellationTokenSource.Token);
+        }
+
+        /// <summary>
+        /// 
+        /// </summary>
+        public override void Detach()
+        {
+            _cancellationTokenSource?.Cancel();
+            base.Detach();
+        }
+
+        private async Task CheckForUpdates()
+        {
+            if(TorchBase.Instance.GameState != TorchGameState.Loaded || !TorchBase.Instance.Config.RestartOnGameUpdate)
+                return;
+            
+            var response = await _httpClient.GetStringAsync(UpdateCheckUrl);
+            var xml = XDocument.Parse(response);
+
+            var latestEntry = xml.Root.Element("Entry");
+            var latestVersion = latestEntry?.Attribute("version")?.Value;
+            
+            var currentVersion = MyPerGameSettings.BasicGameInfo.GameVersion.Value;
+            
+            if (latestVersion != null && int.Parse(latestVersion) > currentVersion)
+            {
+                _log.Warn($"Game update detectected!");
+                try
+                {
+                    Torch.CurrentSession?.Managers.GetManager<ChatManagerServer>()?.SendMessageAsOther("Server",
+                        $"A new version of Space Engineers is available! The server will restart in {TorchBase.Instance.Config.GameUpdateRestartDelayMins} minutes to update to version {latestVersion}.");
+
+                    // Wait 2 minutes or until the task is cancelled
+                    await Task.Delay(TimeSpan.FromMinutes(2), _cancellationTokenSource.Token);
+
+                    // Restart the server
+                    Torch.Restart();
+                }
+                catch (TaskCanceledException)
+                {
+                    // Task was cancelled, do nothing
+                }
+                catch
+                {
+                    // ignored
+                }
+            }
+        }
+    }
+}

Torch.Server/Torch.Server.csproj

@@ -245,6 +245,7 @@
     <Compile Include="FlowDocumentTarget.cs" />
     <Compile Include="ListBoxExtensions.cs" />
     <Compile Include="Managers\EntityControlManager.cs" />
+    <Compile Include="Managers\GameUpdateManager.cs" />
     <Compile Include="Managers\MultiplayerManagerDedicated.cs" />
     <Compile Include="Managers\InstanceManager.cs" />
     <Compile Include="Managers\MultiplayerManagerDedicatedEventShim.cs" />

Torch.Server/TorchConfig.cs

@@ -25,6 +25,8 @@ public class TorchConfig : CommandLine, ITorchConfig, INotifyPropertyChanged
         private string _instanceName = "Instance";
         private bool _autostart;
         private bool _restartOnCrash;
+        private bool _restartOnGameUpdate;
+        private int _gameUpdateRestartDelayMins = 5;
         private bool _noGui;
         private bool _getPluginUpdates = true;
         private bool _getTorchUpdates = true;
@@ -86,7 +88,21 @@ public string InstancePath
         [Arg("restartoncrash", "Automatically restart the server if it crashes.")]
         [Display(Name = "Restart On Crash", Description = "Automatically restart the server if it crashes.", GroupName = "Server")]
         public bool RestartOnCrash { get => _restartOnCrash; set => Set(value, ref _restartOnCrash); }
-
+        
+        /// <summary>
+        /// Enable Game update detection. If enabled, server will restart when game updates are found.
+        /// </summary>
+        [Arg("gameupdatedetection", "Automatically restart the server if the game updates.")]
+        [Display(Name = "Restart On Game Update", Description = "Automatically restart the server if the game updates.", GroupName = "Update Detection")]
+        public bool RestartOnGameUpdate { get => _restartOnGameUpdate; set => Set(value, ref _restartOnGameUpdate); }
+        
+        /// <summary>
+        /// How long (in minutes) to wait before restarting after a game update is detected.
+        /// </summary>
+        [Arg("gameupdaterestartdelay", "How long (in minutes) to wait before restarting after a game update is detected.")]
+        [Display(Name = "Game Update Restart Delay", Description = "How long (in minutes) to wait before restarting after a game update is detected.", GroupName = "Update Detection")]
+        public int GameUpdateRestartDelayMins { get => _gameUpdateRestartDelayMins; set => Set(value, ref _gameUpdateRestartDelayMins); }
+        
         /// <inheritdoc />
         [Arg("nogui", "Do not show the Torch UI.")]
         [Display(Name = "No GUI", Description = "Do not show the Torch UI.", GroupName = "Window")]

Torch.Server/TorchServer.cs

@@ -67,6 +67,7 @@ public TorchServer(TorchConfig config) : base(config)
             AddManager(new EntityControlManager(this));
             AddManager(new RemoteAPIManager(this));
             AddManager(new UpdateManager(this));
+            AddManager(new GameUpdateManager(this));
 
             Managers.GetManager<UpdateManager>().CheckAndUpdateTorch();