From ec248029babe6d484184b8d5d4501778588f5c76 Mon Sep 17 00:00:00 2001
From: Cyberboss <Cyberboss@users.noreply.github.com>
Date: Fri, 29 Jun 2018 16:50:14 -0400
Subject: [PATCH] Moh werrk?

---
 .../Components/DreamDaemonControl.cs          | 19 +++++++++++++++++
 .../DreamDaemonReattachInformation.cs         |  5 +++++
 .../Components/DreamDaemonRebootState.cs      | 21 +++++++++++++++++++
 .../Components/IDreamDaemonControl.cs         | 17 +++++++++++++--
 .../Components/IDreamDaemonFactory.cs         | 16 ++++++++++++++
 5 files changed, 76 insertions(+), 2 deletions(-)
 create mode 100644 src/Tgstation.Server.Host/Components/DreamDaemonRebootState.cs
 create mode 100644 src/Tgstation.Server.Host/Components/IDreamDaemonFactory.cs

diff --git a/src/Tgstation.Server.Host/Components/DreamDaemonControl.cs b/src/Tgstation.Server.Host/Components/DreamDaemonControl.cs
index 85d8238029..272cf23d69 100644
--- a/src/Tgstation.Server.Host/Components/DreamDaemonControl.cs
+++ b/src/Tgstation.Server.Host/Components/DreamDaemonControl.cs
@@ -57,6 +57,16 @@ namespace Tgstation.Server.Host.Components
 			}
 		}
 
+		/// <inheritdoc />
+		public DreamDaemonRebootState RebootState
+		{
+			get
+			{
+				CheckDisposed();
+				return reattachInformation.RebootState;
+			}
+		}
+
 		/// <summary>
 		/// The up to date <see cref="DreamDaemonReattachInformation"/>
 		/// </summary>
@@ -209,5 +219,14 @@ namespace Tgstation.Server.Host.Components
 				throw new ArgumentOutOfRangeException(nameof(port), port, "port must not be zero!");
 			return await SetPortImpl(port, cancellatonToken).ConfigureAwait(false);
 		}
+
+		/// <inheritdoc />
+		public async Task<bool> SetRebootState(DreamDaemonRebootState newRebootState, CancellationToken cancellationToken)
+		{
+			var oldActive = RebootState != DreamDaemonRebootState.Normal;
+			var newActive = RebootState != DreamDaemonRebootState.Normal;
+			if (oldActive == newActive)
+				return true;
+		}
 	}
 }
diff --git a/src/Tgstation.Server.Host/Components/DreamDaemonReattachInformation.cs b/src/Tgstation.Server.Host/Components/DreamDaemonReattachInformation.cs
index 75b079cec8..5c24f46e21 100644
--- a/src/Tgstation.Server.Host/Components/DreamDaemonReattachInformation.cs
+++ b/src/Tgstation.Server.Host/Components/DreamDaemonReattachInformation.cs
@@ -27,6 +27,11 @@ namespace Tgstation.Server.Host.Components
 		/// </summary>
 		public ushort Port { get; set; }
 
+		/// <summary>
+		/// The current DreamDaemon reboot state
+		/// </summary>
+		public DreamDaemonRebootState RebootState { get; set; }
+
 		/// <summary>
 		/// The <see cref="IDmbProvider"/> used by DreamDaemon
 		/// </summary>
diff --git a/src/Tgstation.Server.Host/Components/DreamDaemonRebootState.cs b/src/Tgstation.Server.Host/Components/DreamDaemonRebootState.cs
new file mode 100644
index 0000000000..8608ab6005
--- /dev/null
+++ b/src/Tgstation.Server.Host/Components/DreamDaemonRebootState.cs
@@ -0,0 +1,21 @@
+namespace Tgstation.Server.Host.Components
+{
+	/// <summary>
+	/// Represents the action to take when /world/Reboot() is called
+	/// </summary>
+    enum DreamDaemonRebootState
+    {
+		/// <summary>
+		/// Run DreamDaemon's normal reboot process
+		/// </summary>
+		Normal,
+		/// <summary>
+		/// Shutdown DreamDaemon
+		/// </summary>
+		Shutdown,
+		/// <summary>
+		/// Restart the DreamDaemon process
+		/// </summary>
+		Restart
+    }
+}
diff --git a/src/Tgstation.Server.Host/Components/IDreamDaemonControl.cs b/src/Tgstation.Server.Host/Components/IDreamDaemonControl.cs
index b5ecc131b9..81454a6047 100644
--- a/src/Tgstation.Server.Host/Components/IDreamDaemonControl.cs
+++ b/src/Tgstation.Server.Host/Components/IDreamDaemonControl.cs
@@ -7,8 +7,8 @@ namespace Tgstation.Server.Host.Components
 	/// <summary>
 	/// Handles communication with a <see cref="IDreamDaemonSession"/>
 	/// </summary>
-    interface IDreamDaemonControl : IDisposable
-    {
+	interface IDreamDaemonControl : IDisposable
+	{
 		/// <summary>
 		/// If the <see cref="IDmbProvider.PrimaryDirectory"/> of <see cref="Dmb"/> is being used
 		/// </summary>
@@ -24,6 +24,11 @@ namespace Tgstation.Server.Host.Components
 		/// </summary>
 		ushort Port { get; }
 
+		/// <summary>
+		/// The current <see cref="DreamDaemonRebootState"/>
+		/// </summary>
+		DreamDaemonRebootState RebootState { get; }
+
 		/// <summary>
 		/// Releases the <see cref="IDreamDaemonSession"/> without terminating it. Also calls <see cref="IDisposable.Dispose"/>
 		/// </summary>
@@ -50,5 +55,13 @@ namespace Tgstation.Server.Host.Components
 		/// <param name="cancellatonToken">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> SetPort(ushort newPort, CancellationToken cancellatonToken);
+
+		/// <summary>
+		/// Attempts to change the current <see cref="RebootState"/> to <paramref name="newRebootState"/>
+		/// </summary>
+		/// <param name="newRebootState">The new <see cref="DreamDaemonRebootState"/></param>
+		/// <param name="cancellatonToken">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(DreamDaemonRebootState newRebootState, CancellationToken cancellationToken);
     }
 }
diff --git a/src/Tgstation.Server.Host/Components/IDreamDaemonFactory.cs b/src/Tgstation.Server.Host/Components/IDreamDaemonFactory.cs
new file mode 100644
index 0000000000..76ae5ce3f4
--- /dev/null
+++ b/src/Tgstation.Server.Host/Components/IDreamDaemonFactory.cs
@@ -0,0 +1,16 @@
+using System.Threading;
+using System.Threading.Tasks;
+using Tgstation.Server.Api.Models.Internal;
+
+namespace Tgstation.Server.Host.Components
+{
+	/// <summary>
+	/// Factory for <see cref="IDreamDaemonControl"/>s
+	/// </summary>
+    interface IDreamDaemonFactory
+    {
+		IDreamDaemonControl LaunchNew(DreamDaemonLaunchParameters launchParameters);
+
+		Task<IDreamDaemonControl> Reattach(DreamDaemonReattachInformation reattachInformation, CancellationToken cancellationToken);
+    }
+}