From 4af3bd811ee87b60302a70e44e6cf84e78e511cf Mon Sep 17 00:00:00 2001
From: Cyberboss <Cyberboss@users.noreply.github.com>
Date: Wed, 18 Jul 2018 10:58:02 -0400
Subject: [PATCH] Implement ISessionController.ResetRebootState()

---
 .../Components/Watchdog/ISessionController.cs              | 7 ++++++-
 .../Components/Watchdog/SessionController.cs               | 7 +++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/Tgstation.Server.Host/Components/Watchdog/ISessionController.cs b/src/Tgstation.Server.Host/Components/Watchdog/ISessionController.cs
index faa997cbf1..6624d03145 100644
--- a/src/Tgstation.Server.Host/Components/Watchdog/ISessionController.cs
+++ b/src/Tgstation.Server.Host/Components/Watchdog/ISessionController.cs
@@ -79,5 +79,10 @@ namespace Tgstation.Server.Host.Components.Watchdog
 		/// <param name="cancellationToken">The <see cref="CancellationToken"/> for the operation</param>
 		/// <returns>A <see cref="Task{TResult}"/> resulting in <see langword="true"/> if the operation succeeded, <see langword="false"/> otherwise</returns>
 		Task<bool> SetRebootState(RebootState newRebootState, CancellationToken cancellationToken);
-    }
+
+		/// <summary>
+		/// Changes <see cref="RebootState"/> to <see cref="Components.Watchdog.RebootState.Normal"/> without telling the DMAPI
+		/// </summary>
+		void ResetRebootState();
+	}
 }
diff --git a/src/Tgstation.Server.Host/Components/Watchdog/SessionController.cs b/src/Tgstation.Server.Host/Components/Watchdog/SessionController.cs
index 4937f7e170..757734d4f1 100644
--- a/src/Tgstation.Server.Host/Components/Watchdog/SessionController.cs
+++ b/src/Tgstation.Server.Host/Components/Watchdog/SessionController.cs
@@ -333,5 +333,12 @@ namespace Tgstation.Server.Host.Components.Watchdog
 
 			return await SendCommand(String.Format(CultureInfo.InvariantCulture, "{0}&{1}={2}", InteropConstants.DMTopicChangeReboot, InteropConstants.DMParameterNewRebootMode, (int)newRebootState), cancellationToken).ConfigureAwait(false) == InteropConstants.DMResponseSuccess;
 		}
+
+		/// <inheritdoc />
+		public void ResetRebootState()
+		{
+			CheckDisposed();
+			reattachInformation.RebootState = RebootState.Normal;
+		}
 	}
 }