From bd731bf51ae84f2a4340606dc31eaa41f1ff87da Mon Sep 17 00:00:00 2001
From: Jordan Dominion <Cyberboss@users.noreply.github.com>
Date: Fri, 12 Jul 2024 14:48:31 -0400
Subject: [PATCH] DMAPI enabled deployments now set to
 `WatchdogStatus.Restoring` on Reboot

---
 .../Components/Watchdog/BasicWatchdog.cs               | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Tgstation.Server.Host/Components/Watchdog/BasicWatchdog.cs b/src/Tgstation.Server.Host/Components/Watchdog/BasicWatchdog.cs
index 8dfb10e412..edc50518d4 100644
--- a/src/Tgstation.Server.Host/Components/Watchdog/BasicWatchdog.cs
+++ b/src/Tgstation.Server.Host/Components/Watchdog/BasicWatchdog.cs
@@ -181,7 +181,8 @@ namespace Tgstation.Server.Host.Components.Watchdog
 					await HandleEventImpl(EventType.WorldPrime, Enumerable.Empty<string>(), false, cancellationToken);
 					break;
 				case MonitorActivationReason.ActiveServerStartup:
-					break; // unused in BasicWatchdog
+					Status = Api.Models.WatchdogStatus.Online;
+					break;
 				case MonitorActivationReason.HealthCheck:
 				default:
 					throw new InvalidOperationException($"Invalid activation reason: {reason}");
@@ -296,7 +297,12 @@ namespace Tgstation.Server.Host.Components.Watchdog
 		/// <param name="cancellationToken">The <see cref="CancellationToken"/> for the operation.</param>
 		/// <returns>A <see cref="ValueTask{TResult}"/> resulting in the <see cref="MonitorAction"/> to take.</returns>
 		protected virtual ValueTask<MonitorAction> HandleNormalReboot(CancellationToken cancellationToken)
-			=> ValueTask.FromResult(MonitorAction.Continue);
+		{
+			if (Server!.CompileJob.DMApiVersion != null)
+				Status = Api.Models.WatchdogStatus.Restoring;
+
+			return ValueTask.FromResult(MonitorAction.Continue);
+		}
 
 		/// <summary>
 		/// Handler for <see cref="MonitorActivationReason.NewDmbAvailable"/>.