mirror of
				https://github.com/chylex/Minecraft-Phantom-Panel.git
				synced 2025-10-31 02:17:16 +01:00 
			
		
		
		
	Compare commits
	
		
			4 Commits
		
	
	
		
			wip-forge
			...
			15d45fe1a3
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 15d45fe1a3 | |||
| 0b8a870d10 | |||
| 16888c9b10 | |||
| bfefb9063b | 
| @@ -3,7 +3,7 @@ | |||||||
|   "isRoot": true, |   "isRoot": true, | ||||||
|   "tools": { |   "tools": { | ||||||
|     "dotnet-ef": { |     "dotnet-ef": { | ||||||
|       "version": "8.0.3", |       "version": "7.0.0-rc.1.22426.7", | ||||||
|       "commands": [ |       "commands": [ | ||||||
|         "dotnet-ef" |         "dotnet-ef" | ||||||
|       ] |       ] | ||||||
|   | |||||||
| @@ -1,6 +1,5 @@ | |||||||
| using Phantom.Utils.Collections; | using Phantom.Utils.Collections; | ||||||
| using Phantom.Utils.Processes; | using Phantom.Utils.Processes; | ||||||
| using Phantom.Utils.Tasks; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Instance; | namespace Phantom.Agent.Minecraft.Instance; | ||||||
|  |  | ||||||
| @@ -14,7 +13,6 @@ public sealed class InstanceProcess : IDisposable { | |||||||
| 	public bool HasEnded { get; private set; } | 	public bool HasEnded { get; private set; } | ||||||
|  |  | ||||||
| 	private readonly Process process; | 	private readonly Process process; | ||||||
| 	private readonly TaskCompletionSource processExited = AsyncTasks.CreateCompletionSource(); |  | ||||||
|  |  | ||||||
| 	internal InstanceProcess(InstanceProperties instanceProperties, Process process) { | 	internal InstanceProcess(InstanceProperties instanceProperties, Process process) { | ||||||
| 		this.InstanceProperties = instanceProperties; | 		this.InstanceProperties = instanceProperties; | ||||||
| @@ -48,15 +46,16 @@ public sealed class InstanceProcess : IDisposable { | |||||||
| 		OutputEvent = null; | 		OutputEvent = null; | ||||||
| 		HasEnded = true; | 		HasEnded = true; | ||||||
| 		Ended?.Invoke(this, EventArgs.Empty); | 		Ended?.Invoke(this, EventArgs.Empty); | ||||||
| 		processExited.SetResult(); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void Kill() { | 	public void Kill() { | ||||||
| 		process.Kill(true); | 		process.Kill(true); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task WaitForExit(TimeSpan timeout) { | 	public async Task WaitForExit(CancellationToken cancellationToken) { | ||||||
| 		await processExited.Task.WaitAsync(timeout); | 		if (!HasEnded) { | ||||||
|  | 			await process.WaitForExitAsync(cancellationToken); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void Dispose() { | 	public void Dispose() { | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| using System.Diagnostics; | using System.Diagnostics; | ||||||
| using Phantom.Common.Data.Java; | using Phantom.Common.Data.Java; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Java; | namespace Phantom.Agent.Minecraft.Java; | ||||||
|   | |||||||
| @@ -1,5 +1,4 @@ | |||||||
| using System.Collections.ObjectModel; | using System.Text; | ||||||
| using System.Text; |  | ||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Agent.Minecraft.Java; | using Phantom.Agent.Minecraft.Java; | ||||||
| using Phantom.Agent.Minecraft.Server; | using Phantom.Agent.Minecraft.Server; | ||||||
| @@ -12,7 +11,6 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 	private readonly InstanceProperties instanceProperties; | 	private readonly InstanceProperties instanceProperties; | ||||||
|  |  | ||||||
| 	protected string MinecraftVersion => instanceProperties.ServerVersion; | 	protected string MinecraftVersion => instanceProperties.ServerVersion; | ||||||
| 	protected string InstanceFolder => instanceProperties.InstanceFolder; |  | ||||||
|  |  | ||||||
| 	private protected BaseLauncher(InstanceProperties instanceProperties) { | 	private protected BaseLauncher(InstanceProperties instanceProperties) { | ||||||
| 		this.instanceProperties = instanceProperties; | 		this.instanceProperties = instanceProperties; | ||||||
| @@ -53,14 +51,16 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 		 | 		 | ||||||
| 		var processConfigurator = new ProcessConfigurator { | 		var processConfigurator = new ProcessConfigurator { | ||||||
| 			FileName = javaRuntimeExecutable.ExecutablePath, | 			FileName = javaRuntimeExecutable.ExecutablePath, | ||||||
| 			WorkingDirectory = InstanceFolder, | 			WorkingDirectory = instanceProperties.InstanceFolder, | ||||||
| 			RedirectInput = true, | 			RedirectInput = true, | ||||||
| 			UseShellExecute = false | 			UseShellExecute = false | ||||||
| 		}; | 		}; | ||||||
| 		 | 		 | ||||||
| 		var processArguments = processConfigurator.ArgumentList; | 		var processArguments = processConfigurator.ArgumentList; | ||||||
| 		PrepareJvmArguments(serverJar).Build(processArguments); | 		PrepareJvmArguments(serverJar).Build(processArguments); | ||||||
| 		PrepareJavaProcessArguments(processArguments, serverJar.FilePath); | 		processArguments.Add("-jar"); | ||||||
|  | 		processArguments.Add(serverJar.FilePath); | ||||||
|  | 		processArguments.Add("nogui"); | ||||||
|  |  | ||||||
| 		var process = processConfigurator.CreateProcess(); | 		var process = processConfigurator.CreateProcess(); | ||||||
| 		var instanceProcess = new InstanceProcess(instanceProperties, process); | 		var instanceProcess = new InstanceProcess(instanceProperties, process); | ||||||
| @@ -99,12 +99,6 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
|  |  | ||||||
| 	private protected virtual void CustomizeJvmArguments(JvmArgumentBuilder arguments) {} | 	private protected virtual void CustomizeJvmArguments(JvmArgumentBuilder arguments) {} | ||||||
|  |  | ||||||
| 	protected virtual void PrepareJavaProcessArguments(Collection<string> processArguments, string serverJarFilePath) { |  | ||||||
| 		processArguments.Add("-jar"); |  | ||||||
| 		processArguments.Add(serverJarFilePath); |  | ||||||
| 		processArguments.Add("nogui"); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private protected virtual Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { | 	private protected virtual Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { | ||||||
| 		return Task.FromResult(new ServerJarInfo(serverJarPath)); | 		return Task.FromResult(new ServerJarInfo(serverJarPath)); | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,29 +0,0 @@ | |||||||
| using System.Collections.ObjectModel; |  | ||||||
| using Phantom.Agent.Minecraft.Instance; |  | ||||||
| using Phantom.Agent.Minecraft.Java; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Launcher.Types;  |  | ||||||
|  |  | ||||||
| public sealed class ForgeLauncher : BaseLauncher { |  | ||||||
| 	public ForgeLauncher(InstanceProperties instanceProperties) : base(instanceProperties) {} |  | ||||||
| 	 |  | ||||||
| 	private protected override void CustomizeJvmArguments(JvmArgumentBuilder arguments) { |  | ||||||
| 		arguments.AddProperty("terminal.ansi", "true"); // TODO |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	protected override void PrepareJavaProcessArguments(Collection<string> processArguments, string serverJarFilePath) { |  | ||||||
| 		if (OperatingSystem.IsWindows()) { |  | ||||||
| 			processArguments.Add("@libraries/net/minecraftforge/forge/1.20.1-47.2.0/win_args.txt"); |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			processArguments.Add("@libraries/net/minecraftforge/forge/1.20.1-47.2.0/unix_args.txt"); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		processArguments.Add("nogui"); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private protected override Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { |  | ||||||
| 		return Task.FromResult(new ServerJarInfo(Path.Combine(InstanceFolder, "run.sh"))); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -11,8 +11,8 @@ | |||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Data\Phantom.Common.Data.csproj" /> |     <ProjectReference Include="..\..\Common\Phantom.Common.Data\Phantom.Common.Data.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Common\Phantom.Common.Logging\Phantom.Common.Logging.csproj" /> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> |     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Logging\Phantom.Utils.Logging.csproj" /> |  | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|  |  | ||||||
| </Project> | </Project> | ||||||
|   | |||||||
| @@ -18,5 +18,4 @@ static class MinecraftServerProperties { | |||||||
| 	public static readonly MinecraftServerProperty<ushort> ServerPort = new UnsignedShort("server-port"); | 	public static readonly MinecraftServerProperty<ushort> ServerPort = new UnsignedShort("server-port"); | ||||||
| 	public static readonly MinecraftServerProperty<ushort> RconPort = new UnsignedShort("rcon.port"); | 	public static readonly MinecraftServerProperty<ushort> RconPort = new UnsignedShort("rcon.port"); | ||||||
| 	public static readonly MinecraftServerProperty<bool> EnableRcon = new Boolean("enable-rcon"); | 	public static readonly MinecraftServerProperty<bool> EnableRcon = new Boolean("enable-rcon"); | ||||||
| 	public static readonly MinecraftServerProperty<bool> SyncChunkWrites = new Boolean("sync-chunk-writes"); |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,13 +5,11 @@ namespace Phantom.Agent.Minecraft.Properties; | |||||||
| public sealed record ServerProperties( | public sealed record ServerProperties( | ||||||
| 	ushort ServerPort, | 	ushort ServerPort, | ||||||
| 	ushort RconPort, | 	ushort RconPort, | ||||||
| 	bool EnableRcon = true, | 	bool EnableRcon = true | ||||||
| 	bool SyncChunkWrites = false |  | ||||||
| ) { | ) { | ||||||
| 	internal void SetTo(JavaPropertiesFileEditor properties) { | 	internal void SetTo(JavaPropertiesFileEditor properties) { | ||||||
| 		MinecraftServerProperties.ServerPort.Set(properties, ServerPort); | 		MinecraftServerProperties.ServerPort.Set(properties, ServerPort); | ||||||
| 		MinecraftServerProperties.RconPort.Set(properties, RconPort); | 		MinecraftServerProperties.RconPort.Set(properties, RconPort); | ||||||
| 		MinecraftServerProperties.EnableRcon.Set(properties, EnableRcon); | 		MinecraftServerProperties.EnableRcon.Set(properties, EnableRcon); | ||||||
| 		MinecraftServerProperties.SyncChunkWrites.Set(properties, SyncChunkWrites); |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,8 +1,8 @@ | |||||||
| using System.Security.Cryptography; | using System.Security.Cryptography; | ||||||
| using Phantom.Common.Data.Minecraft; | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Cryptography; | using Phantom.Utils.Cryptography; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Runtime; | using Phantom.Utils.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| @@ -16,47 +16,32 @@ sealed class MinecraftServerExecutableDownloader { | |||||||
| 	public event EventHandler? Completed; | 	public event EventHandler? Completed; | ||||||
| 	 | 	 | ||||||
| 	private readonly CancellationTokenSource cancellationTokenSource = new (); | 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||||
| 	 | 	private int listeners = 0; | ||||||
| 	private readonly List<CancellationTokenRegistration> listenerCancellationRegistrations = new (); |  | ||||||
| 	private int listenerCount = 0; |  | ||||||
|  |  | ||||||
| 	public MinecraftServerExecutableDownloader(FileDownloadInfo fileDownloadInfo, string minecraftVersion, string filePath, MinecraftServerExecutableDownloadListener listener) { | 	public MinecraftServerExecutableDownloader(FileDownloadInfo fileDownloadInfo, string minecraftVersion, string filePath, MinecraftServerExecutableDownloadListener listener) { | ||||||
| 		Register(listener); | 		Register(listener); | ||||||
| 		Task = DownloadAndGetPath(fileDownloadInfo, minecraftVersion, filePath, new DownloadProgressCallback(this), cancellationTokenSource.Token); | 		Task = DownloadAndGetPath(fileDownloadInfo, minecraftVersion, filePath); | ||||||
| 		Task.ContinueWith(OnCompleted, TaskScheduler.Default); | 		Task.ContinueWith(OnCompleted, TaskScheduler.Default); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void Register(MinecraftServerExecutableDownloadListener listener) { | 	public void Register(MinecraftServerExecutableDownloadListener listener) { | ||||||
| 		int newListenerCount; | 		++listeners; | ||||||
|  | 		Logger.Debug("Registered download listener, current listener count: {Listeners}", listeners); | ||||||
| 		 | 		 | ||||||
| 		lock (this) { | 		DownloadProgress += listener.DownloadProgressEventHandler; | ||||||
| 			newListenerCount = ++listenerCount; | 		listener.CancellationToken.Register(Unregister, listener); | ||||||
| 			 |  | ||||||
| 			DownloadProgress += listener.DownloadProgressEventHandler; |  | ||||||
| 			listenerCancellationRegistrations.Add(listener.CancellationToken.Register(Unregister, listener)); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		Logger.Debug("Registered download listener, current listener count: {Listeners}", newListenerCount); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void Unregister(object? listenerObject) { | 	private void Unregister(object? listenerObject) { | ||||||
| 		int newListenerCount; | 		MinecraftServerExecutableDownloadListener listener = (MinecraftServerExecutableDownloadListener) listenerObject!; | ||||||
| 		 | 		DownloadProgress -= listener.DownloadProgressEventHandler; | ||||||
| 		lock (this) { |  | ||||||
| 			MinecraftServerExecutableDownloadListener listener = (MinecraftServerExecutableDownloadListener) listenerObject!; |  | ||||||
| 			DownloadProgress -= listener.DownloadProgressEventHandler; |  | ||||||
|  |  | ||||||
| 			newListenerCount = --listenerCount; | 		if (--listeners <= 0) { | ||||||
| 			if (newListenerCount <= 0) { |  | ||||||
| 				cancellationTokenSource.Cancel(); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		if (newListenerCount <= 0) { |  | ||||||
| 			Logger.Debug("Unregistered last download listener, cancelling download."); | 			Logger.Debug("Unregistered last download listener, cancelling download."); | ||||||
|  | 			cancellationTokenSource.Cancel(); | ||||||
| 		} | 		} | ||||||
| 		else { | 		else { | ||||||
| 			Logger.Debug("Unregistered download listener, current listener count: {Listeners}", newListenerCount); | 			Logger.Debug("Unregistered download listener, current listener count: {Listeners}", listeners); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -66,19 +51,9 @@ sealed class MinecraftServerExecutableDownloader { | |||||||
|  |  | ||||||
| 	private void OnCompleted(Task task) { | 	private void OnCompleted(Task task) { | ||||||
| 		Logger.Debug("Download task completed."); | 		Logger.Debug("Download task completed."); | ||||||
|  | 		Completed?.Invoke(this, EventArgs.Empty); | ||||||
| 		lock (this) { | 		Completed = null; | ||||||
| 			Completed?.Invoke(this, EventArgs.Empty); | 		DownloadProgress = null; | ||||||
| 			Completed = null; |  | ||||||
| 			DownloadProgress = null; |  | ||||||
|  |  | ||||||
| 			foreach (var registration in listenerCancellationRegistrations) { |  | ||||||
| 				registration.Dispose(); |  | ||||||
| 			} |  | ||||||
| 			 |  | ||||||
| 			listenerCancellationRegistrations.Clear(); |  | ||||||
| 			cancellationTokenSource.Dispose(); |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private sealed class DownloadProgressCallback { | 	private sealed class DownloadProgressCallback { | ||||||
| @@ -93,14 +68,15 @@ sealed class MinecraftServerExecutableDownloader { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static async Task<string?> DownloadAndGetPath(FileDownloadInfo fileDownloadInfo, string minecraftVersion, string filePath, DownloadProgressCallback progressCallback, CancellationToken cancellationToken) { | 	private async Task<string?> DownloadAndGetPath(FileDownloadInfo fileDownloadInfo, string minecraftVersion, string filePath) { | ||||||
| 		string tmpFilePath = filePath + ".tmp"; | 		string tmpFilePath = filePath + ".tmp"; | ||||||
|  |  | ||||||
|  | 		var cancellationToken = cancellationTokenSource.Token; | ||||||
| 		try { | 		try { | ||||||
| 			Logger.Information("Downloading server version {Version} from: {Url} ({Size})", minecraftVersion, fileDownloadInfo.DownloadUrl, fileDownloadInfo.Size.ToHumanReadable(decimalPlaces: 1)); | 			Logger.Information("Downloading server version {Version} from: {Url} ({Size})", minecraftVersion, fileDownloadInfo.DownloadUrl, fileDownloadInfo.Size.ToHumanReadable(decimalPlaces: 1)); | ||||||
| 			try { | 			try { | ||||||
| 				using var http = new HttpClient(); | 				using var http = new HttpClient(); | ||||||
| 				await FetchServerExecutableFile(http, progressCallback, fileDownloadInfo, tmpFilePath, cancellationToken); | 				await FetchServerExecutableFile(http, new DownloadProgressCallback(this), fileDownloadInfo, tmpFilePath, cancellationToken); | ||||||
| 			} catch (Exception) { | 			} catch (Exception) { | ||||||
| 				TryDeleteExecutableAfterFailure(tmpFilePath); | 				TryDeleteExecutableAfterFailure(tmpFilePath); | ||||||
| 				throw; | 				throw; | ||||||
| @@ -118,6 +94,8 @@ sealed class MinecraftServerExecutableDownloader { | |||||||
| 		} catch (Exception e) { | 		} catch (Exception e) { | ||||||
| 			Logger.Error(e, "An unexpected error occurred."); | 			Logger.Error(e, "An unexpected error occurred."); | ||||||
| 			return null; | 			return null; | ||||||
|  | 		} finally { | ||||||
|  | 			cancellationTokenSource.Dispose(); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| using System.Text.RegularExpressions; | using System.Text.RegularExpressions; | ||||||
| using Phantom.Common.Data.Minecraft; | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Server; | namespace Phantom.Agent.Minecraft.Server; | ||||||
|   | |||||||
| @@ -3,12 +3,28 @@ using System.Buffers.Binary; | |||||||
| using System.Net; | using System.Net; | ||||||
| using System.Net.Sockets; | using System.Net.Sockets; | ||||||
| using System.Text; | using System.Text; | ||||||
| using Phantom.Common.Data.Instance; | using Phantom.Common.Logging; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Server; | namespace Phantom.Agent.Minecraft.Server; | ||||||
|  |  | ||||||
| public static class ServerStatusProtocol { | public sealed class ServerStatusProtocol { | ||||||
| 	public static async Task<InstancePlayerCounts> GetPlayerCounts(ushort serverPort, CancellationToken cancellationToken) { | 	private readonly ILogger logger; | ||||||
|  |  | ||||||
|  | 	public ServerStatusProtocol(string loggerName) { | ||||||
|  | 		this.logger = PhantomLogger.Create<ServerStatusProtocol>(loggerName); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<int?> GetOnlinePlayerCount(int serverPort, CancellationToken cancellationToken) { | ||||||
|  | 		try { | ||||||
|  | 			return await GetOnlinePlayerCountOrThrow(serverPort, cancellationToken); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			logger.Error(e, "Caught exception while checking if players are online."); | ||||||
|  | 			return null; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<int?> GetOnlinePlayerCountOrThrow(int serverPort, CancellationToken cancellationToken) { | ||||||
| 		using var tcpClient = new TcpClient(); | 		using var tcpClient = new TcpClient(); | ||||||
| 		await tcpClient.ConnectAsync(IPAddress.Loopback, serverPort, cancellationToken); | 		await tcpClient.ConnectAsync(IPAddress.Loopback, serverPort, cancellationToken); | ||||||
| 		var tcpStream = tcpClient.GetStream(); | 		var tcpStream = tcpClient.GetStream(); | ||||||
| @@ -17,22 +33,24 @@ public static class ServerStatusProtocol { | |||||||
| 		tcpStream.WriteByte(0xFE); | 		tcpStream.WriteByte(0xFE); | ||||||
| 		await tcpStream.FlushAsync(cancellationToken); | 		await tcpStream.FlushAsync(cancellationToken); | ||||||
|  |  | ||||||
| 		short messageLength = await ReadStreamHeader(tcpStream, cancellationToken); | 		short? messageLength = await ReadStreamHeader(tcpStream, cancellationToken); | ||||||
| 		return await ReadPlayerCounts(tcpStream, messageLength * 2, cancellationToken); | 		return messageLength == null ? null : await ReadOnlinePlayerCount(tcpStream, messageLength.Value * 2, cancellationToken); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static async Task<short> ReadStreamHeader(NetworkStream tcpStream, CancellationToken cancellationToken) { | 	private async Task<short?> ReadStreamHeader(NetworkStream tcpStream, CancellationToken cancellationToken) { | ||||||
| 		var headerBuffer = ArrayPool<byte>.Shared.Rent(3); | 		var headerBuffer = ArrayPool<byte>.Shared.Rent(3); | ||||||
| 		try { | 		try { | ||||||
| 			await tcpStream.ReadExactlyAsync(headerBuffer, 0, 3, cancellationToken); | 			await tcpStream.ReadExactlyAsync(headerBuffer, 0, 3, cancellationToken); | ||||||
|  |  | ||||||
| 			if (headerBuffer[0] != 0xFF) { | 			if (headerBuffer[0] != 0xFF) { | ||||||
| 				throw new ProtocolException("Unexpected first byte in response from server: " + headerBuffer[0]); | 				logger.Error("Unexpected first byte in response from server: {FirstByte}.", headerBuffer[0]); | ||||||
|  | 				return null; | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			short messageLength = BinaryPrimitives.ReadInt16BigEndian(headerBuffer.AsSpan(1)); | 			short messageLength = BinaryPrimitives.ReadInt16BigEndian(headerBuffer.AsSpan(1)); | ||||||
| 			if (messageLength <= 0) { | 			if (messageLength <= 0) { | ||||||
| 				throw new ProtocolException("Unexpected message length in response from server: " + messageLength); | 				logger.Error("Unexpected message length in response from server: {MessageLength}.", messageLength); | ||||||
|  | 				return null; | ||||||
| 			} | 			} | ||||||
| 			 | 			 | ||||||
| 			return messageLength; | 			return messageLength; | ||||||
| @@ -41,54 +59,35 @@ public static class ServerStatusProtocol { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static async Task<InstancePlayerCounts> ReadPlayerCounts(NetworkStream tcpStream, int messageLength, CancellationToken cancellationToken) { | 	private async Task<int?> ReadOnlinePlayerCount(NetworkStream tcpStream, int messageLength, CancellationToken cancellationToken) { | ||||||
| 		var messageBuffer = ArrayPool<byte>.Shared.Rent(messageLength); | 		var messageBuffer = ArrayPool<byte>.Shared.Rent(messageLength); | ||||||
| 		try { | 		try { | ||||||
| 			await tcpStream.ReadExactlyAsync(messageBuffer, 0, messageLength, cancellationToken); | 			await tcpStream.ReadExactlyAsync(messageBuffer, 0, messageLength, cancellationToken); | ||||||
| 			return ReadPlayerCountsFromResponse(messageBuffer.AsSpan(0, messageLength)); |  | ||||||
|  | 			// Valid response separator encoded in UTF-16BE is 0x00 0xA7 (§). | ||||||
|  | 			const byte SeparatorSecondByte = 0xA7; | ||||||
|  | 			 | ||||||
|  | 			static bool IsValidSeparator(ReadOnlySpan<byte> buffer, int index) { | ||||||
|  | 				return index > 0 && buffer[index - 1] == 0x00; | ||||||
|  | 			} | ||||||
|  | 			 | ||||||
|  | 			int separator2 = Array.LastIndexOf(messageBuffer, SeparatorSecondByte); | ||||||
|  | 			int separator1 = separator2 == -1 ? -1 : Array.LastIndexOf(messageBuffer, SeparatorSecondByte, separator2 - 1); | ||||||
|  | 			if (!IsValidSeparator(messageBuffer, separator1) || !IsValidSeparator(messageBuffer, separator2)) { | ||||||
|  | 				logger.Error("Could not find message separators in response from server."); | ||||||
|  | 				return null; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			string onlinePlayerCountStr = Encoding.BigEndianUnicode.GetString(messageBuffer.AsSpan((separator1 + 1)..(separator2 - 1))); | ||||||
|  | 			if (!int.TryParse(onlinePlayerCountStr, out int onlinePlayerCount)) { | ||||||
|  | 				logger.Error("Could not parse online player count in response from server: {OnlinePlayerCount}.", onlinePlayerCountStr); | ||||||
|  | 				return null; | ||||||
|  | 			} | ||||||
|  | 			 | ||||||
|  | 			logger.Debug("Detected {OnlinePlayerCount} online player(s).", onlinePlayerCount); | ||||||
|  | 			return onlinePlayerCount; | ||||||
| 		} finally { | 		} finally { | ||||||
| 			ArrayPool<byte>.Shared.Return(messageBuffer); | 			ArrayPool<byte>.Shared.Return(messageBuffer); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	 |  | ||||||
| 	/// <summary> |  | ||||||
| 	/// Legacy query protocol uses the paragraph symbol (§) as separator encoded in UTF-16BE. |  | ||||||
| 	/// </summary> |  | ||||||
| 	private static readonly byte[] Separator = { 0x00, 0xA7 }; |  | ||||||
| 	 |  | ||||||
| 	private static InstancePlayerCounts ReadPlayerCountsFromResponse(ReadOnlySpan<byte> messageBuffer) { |  | ||||||
| 		int lastSeparator = messageBuffer.LastIndexOf(Separator); |  | ||||||
| 		int middleSeparator = messageBuffer[..lastSeparator].LastIndexOf(Separator); |  | ||||||
| 		 |  | ||||||
| 		if (lastSeparator == -1 || middleSeparator == -1) { |  | ||||||
| 			throw new ProtocolException("Could not find message separators in response from server."); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		var onlinePlayerCountBuffer = messageBuffer[(middleSeparator + Separator.Length)..lastSeparator]; |  | ||||||
| 		var maximumPlayerCountBuffer = messageBuffer[(lastSeparator + Separator.Length)..]; |  | ||||||
| 		 |  | ||||||
| 		// Player counts are integers, whose maximum string length is 10 characters. |  | ||||||
| 		Span<char> integerStringBuffer = stackalloc char[10]; |  | ||||||
| 		 |  | ||||||
| 		return new InstancePlayerCounts( |  | ||||||
| 			DecodeAndParsePlayerCount(onlinePlayerCountBuffer, integerStringBuffer, "online"), |  | ||||||
| 			DecodeAndParsePlayerCount(maximumPlayerCountBuffer, integerStringBuffer, "maximum") |  | ||||||
| 		); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private static int DecodeAndParsePlayerCount(ReadOnlySpan<byte> inputBuffer, Span<char> tempCharBuffer, string countType) { |  | ||||||
| 		if (!Encoding.BigEndianUnicode.TryGetChars(inputBuffer, tempCharBuffer, out int charCount)) { |  | ||||||
| 			throw new ProtocolException("Could not decode " + countType + " player count in response from server."); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		if (!int.TryParse(tempCharBuffer, out int playerCount)) { |  | ||||||
| 			throw new ProtocolException("Could not parse " + countType + " player count in response from server: " + tempCharBuffer[..charCount].ToString()); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		return playerCount; |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public sealed class ProtocolException : Exception { |  | ||||||
| 		internal ProtocolException(string message) : base(message) {} |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Common.Messages.Agent; | ||||||
| using Phantom.Utils.Rpc.Runtime; | using Phantom.Utils.Rpc; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; | namespace Phantom.Agent.Rpc; | ||||||
| @@ -8,9 +8,9 @@ namespace Phantom.Agent.Rpc; | |||||||
| public sealed class ControllerConnection { | public sealed class ControllerConnection { | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ControllerConnection)); | 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ControllerConnection)); | ||||||
|  |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToController> connection; | 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||||
| 	 | 	 | ||||||
| 	public ControllerConnection(RpcConnectionToServer<IMessageToController> connection) { | 	public ControllerConnection(RpcConnectionToServer<IMessageToControllerListener> connection) { | ||||||
| 		this.connection = connection; | 		this.connection = connection; | ||||||
| 		Logger.Information("Connection ready."); | 		Logger.Information("Connection ready."); | ||||||
| 	} | 	} | ||||||
| @@ -18,4 +18,8 @@ public sealed class ControllerConnection { | |||||||
| 	public Task Send<TMessage>(TMessage message) where TMessage : IMessageToController { | 	public Task Send<TMessage>(TMessage message) where TMessage : IMessageToController { | ||||||
| 		return connection.Send(message); | 		return connection.Send(message); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	public Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToController<TReply> where TReply : class { | ||||||
|  | 		return connection.Send<TMessage, TReply>(message, waitForReplyTime, waitForReplyCancellationToken); | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Logging; | ||||||
|  | using Phantom.Common.Messages.Agent; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.Agent.ToController; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Utils.Rpc; | ||||||
| using Phantom.Utils.Rpc.Runtime; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; | namespace Phantom.Agent.Rpc; | ||||||
| @@ -11,24 +11,22 @@ sealed class KeepAliveLoop { | |||||||
|  |  | ||||||
| 	private static readonly TimeSpan KeepAliveInterval = TimeSpan.FromSeconds(10); | 	private static readonly TimeSpan KeepAliveInterval = TimeSpan.FromSeconds(10); | ||||||
|  |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToController> connection; | 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||||
| 	private readonly CancellationTokenSource cancellationTokenSource = new (); | 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||||
|  |  | ||||||
| 	public KeepAliveLoop(RpcConnectionToServer<IMessageToController> connection) { | 	public KeepAliveLoop(RpcConnectionToServer<IMessageToControllerListener> connection) { | ||||||
| 		this.connection = connection; | 		this.connection = connection; | ||||||
| 		Task.Run(Run); | 		Task.Run(Run); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task Run() { | 	private async Task Run() { | ||||||
| 		var cancellationToken = cancellationTokenSource.Token; | 		var cancellationToken = cancellationTokenSource.Token; | ||||||
| 		 |  | ||||||
|  | 		Logger.Information("Started keep-alive loop."); | ||||||
| 		try { | 		try { | ||||||
| 			await connection.IsReady.WaitAsync(cancellationToken); |  | ||||||
| 			Logger.Information("Started keep-alive loop."); |  | ||||||
| 			 |  | ||||||
| 			while (true) { | 			while (true) { | ||||||
| 				await Task.Delay(KeepAliveInterval, cancellationToken); | 				await Task.Delay(KeepAliveInterval, cancellationToken); | ||||||
| 				await connection.Send(new AgentIsAliveMessage()).WaitAsync(cancellationToken); | 				await connection.Send(new AgentIsAliveMessage()); | ||||||
| 			} | 			} | ||||||
| 		} catch (OperationCanceledException) { | 		} catch (OperationCanceledException) { | ||||||
| 			// Ignore. | 			// Ignore. | ||||||
|   | |||||||
| @@ -1,32 +1,33 @@ | |||||||
| using NetMQ; | using NetMQ; | ||||||
| using NetMQ.Sockets; | using NetMQ.Sockets; | ||||||
|  | using Phantom.Common.Data.Agent; | ||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Messages.Agent; | ||||||
| using Phantom.Common.Messages.Agent.BiDirectional; | using Phantom.Common.Messages.Agent.BiDirectional; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.Agent.ToController; | ||||||
| using Phantom.Utils.Actor; | using Phantom.Utils.Rpc; | ||||||
| using Phantom.Utils.Rpc.Runtime; |  | ||||||
| using Phantom.Utils.Rpc.Sockets; | using Phantom.Utils.Rpc.Sockets; | ||||||
|  | using Phantom.Utils.Tasks; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; | namespace Phantom.Agent.Rpc; | ||||||
|  |  | ||||||
| public sealed class RpcClientRuntime : RpcClientRuntime<IMessageToAgent, IMessageToController, ReplyMessage> { | public sealed class RpcClientRuntime : RpcClientRuntime<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> { | ||||||
| 	public static Task Launch(RpcClientSocket<IMessageToAgent, IMessageToController, ReplyMessage> socket, ActorRef<IMessageToAgent> handlerActorRef, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { | 	public static Task Launch(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, AgentInfo agentInfo, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { | ||||||
| 		return new RpcClientRuntime(socket, handlerActorRef, disconnectSemaphore, receiveCancellationToken).Launch(); | 		return new RpcClientRuntime(socket, messageListener, disconnectSemaphore, receiveCancellationToken).Launch(); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private RpcClientRuntime(RpcClientSocket<IMessageToAgent, IMessageToController, ReplyMessage> socket, ActorRef<IMessageToAgent> handlerActor, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(socket, handlerActor, disconnectSemaphore, receiveCancellationToken) {} | 	private RpcClientRuntime(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(socket, messageListener, disconnectSemaphore, receiveCancellationToken) {} | ||||||
|  |  | ||||||
| 	protected override async Task RunWithConnection(ClientSocket socket, RpcConnectionToServer<IMessageToController> connection) { | 	protected override void RunWithConnection(ClientSocket socket, RpcConnectionToServer<IMessageToControllerListener> connection, ILogger logger, TaskManager taskManager) { | ||||||
| 		var keepAliveLoop = new KeepAliveLoop(connection); | 		var keepAliveLoop = new KeepAliveLoop(connection); | ||||||
| 		try { | 		try { | ||||||
| 			await base.RunWithConnection(socket, connection); | 			base.RunWithConnection(socket, connection, logger, taskManager); | ||||||
| 		} finally { | 		} finally { | ||||||
| 			keepAliveLoop.Cancel(); | 			keepAliveLoop.Cancel(); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	protected override async Task SendDisconnectMessage(ClientSocket socket, ILogger logger) { | 	protected override async Task Disconnect(ClientSocket socket, ILogger logger) { | ||||||
| 		var unregisterMessageBytes = AgentMessageRegistries.ToController.Write(new UnregisterAgentMessage()).ToArray(); | 		var unregisterMessageBytes = AgentMessageRegistries.ToController.Write(new UnregisterAgentMessage()).ToArray(); | ||||||
| 		try { | 		try { | ||||||
| 			await socket.SendAsync(unregisterMessageBytes).AsTask().WaitAsync(TimeSpan.FromSeconds(5), CancellationToken.None); | 			await socket.SendAsync(unregisterMessageBytes).AsTask().WaitAsync(TimeSpan.FromSeconds(5), CancellationToken.None); | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| using Phantom.Utils.IO; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Utils.IO; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services; | namespace Phantom.Agent.Services; | ||||||
|   | |||||||
| @@ -1,11 +1,10 @@ | |||||||
| using Akka.Actor; | using Phantom.Agent.Minecraft.Java; | ||||||
| using Phantom.Agent.Minecraft.Java; |  | ||||||
| using Phantom.Agent.Rpc; | using Phantom.Agent.Rpc; | ||||||
| using Phantom.Agent.Services.Backups; | using Phantom.Agent.Services.Backups; | ||||||
| using Phantom.Agent.Services.Instances; | using Phantom.Agent.Services.Instances; | ||||||
| using Phantom.Common.Data.Agent; | using Phantom.Common.Data.Agent; | ||||||
| using Phantom.Utils.Actor; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Utils.Tasks; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services; | namespace Phantom.Agent.Services; | ||||||
| @@ -13,28 +12,19 @@ namespace Phantom.Agent.Services; | |||||||
| public sealed class AgentServices { | public sealed class AgentServices { | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<AgentServices>(); | 	private static readonly ILogger Logger = PhantomLogger.Create<AgentServices>(); | ||||||
| 	 | 	 | ||||||
| 	public ActorSystem ActorSystem { get; } |  | ||||||
|  |  | ||||||
| 	private AgentFolders AgentFolders { get; } | 	private AgentFolders AgentFolders { get; } | ||||||
| 	private AgentState AgentState { get; } | 	private TaskManager TaskManager { get; } | ||||||
| 	private BackupManager BackupManager { get; } | 	private BackupManager BackupManager { get; } | ||||||
|  |  | ||||||
| 	internal JavaRuntimeRepository JavaRuntimeRepository { get; } | 	internal JavaRuntimeRepository JavaRuntimeRepository { get; } | ||||||
| 	internal InstanceTicketManager InstanceTicketManager { get; } | 	internal InstanceSessionManager InstanceSessionManager { get; } | ||||||
| 	internal ActorRef<InstanceManagerActor.ICommand> InstanceManager { get; } |  | ||||||
|  |  | ||||||
| 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders, AgentServiceConfiguration serviceConfiguration, ControllerConnection controllerConnection) { | 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders, AgentServiceConfiguration serviceConfiguration, ControllerConnection controllerConnection) { | ||||||
| 		this.ActorSystem = ActorSystemFactory.Create("Agent"); |  | ||||||
| 		 |  | ||||||
| 		this.AgentFolders = agentFolders; | 		this.AgentFolders = agentFolders; | ||||||
| 		this.AgentState = new AgentState(); | 		this.TaskManager = new TaskManager(PhantomLogger.Create<TaskManager, AgentServices>()); | ||||||
| 		this.BackupManager = new BackupManager(agentFolders, serviceConfiguration.MaxConcurrentCompressionTasks); | 		this.BackupManager = new BackupManager(agentFolders, serviceConfiguration.MaxConcurrentCompressionTasks); | ||||||
| 		 |  | ||||||
| 		this.JavaRuntimeRepository = new JavaRuntimeRepository(); | 		this.JavaRuntimeRepository = new JavaRuntimeRepository(); | ||||||
| 		this.InstanceTicketManager = new InstanceTicketManager(agentInfo, controllerConnection); | 		this.InstanceSessionManager = new InstanceSessionManager(controllerConnection, agentInfo, agentFolders, JavaRuntimeRepository, TaskManager, BackupManager); | ||||||
| 		 |  | ||||||
| 		var instanceManagerInit = new InstanceManagerActor.Init(controllerConnection, agentFolders, AgentState, JavaRuntimeRepository, InstanceTicketManager, BackupManager); |  | ||||||
| 		this.InstanceManager = ActorSystem.ActorOf(InstanceManagerActor.Factory(instanceManagerInit), "InstanceManager"); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task Initialize() { | 	public async Task Initialize() { | ||||||
| @@ -46,13 +36,11 @@ public sealed class AgentServices { | |||||||
| 	public async Task Shutdown() { | 	public async Task Shutdown() { | ||||||
| 		Logger.Information("Stopping services..."); | 		Logger.Information("Stopping services..."); | ||||||
| 		 | 		 | ||||||
| 		await InstanceManager.Stop(new InstanceManagerActor.ShutdownCommand()); | 		await InstanceSessionManager.DisposeAsync(); | ||||||
|  | 		await TaskManager.Stop(); | ||||||
| 		 | 		 | ||||||
| 		BackupManager.Dispose(); | 		BackupManager.Dispose(); | ||||||
| 		 | 		 | ||||||
| 		await ActorSystem.Terminate(); |  | ||||||
| 		ActorSystem.Dispose(); |  | ||||||
| 		 |  | ||||||
| 		Logger.Information("Services stopped."); | 		Logger.Information("Services stopped."); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,15 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using Phantom.Agent.Services.Instances; |  | ||||||
| using Phantom.Utils.Actor.Event; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services; |  | ||||||
|  |  | ||||||
| sealed class AgentState { |  | ||||||
| 	private readonly ObservableState<ImmutableDictionary<Guid, Instance>> instancesByGuid = new (ImmutableDictionary<Guid, Instance>.Empty); |  | ||||||
| 	 |  | ||||||
| 	public ImmutableDictionary<Guid, Instance> InstancesByGuid => instancesByGuid.State; |  | ||||||
| 	 |  | ||||||
| 	public void UpdateInstance(Instance instance) { |  | ||||||
| 		instancesByGuid.PublisherSide.Publish(static (instancesByGuid, instance) => instancesByGuid.SetItem(instance.InstanceGuid, instance), instance); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -3,8 +3,8 @@ using System.Diagnostics.CodeAnalysis; | |||||||
| using System.Formats.Tar; | using System.Formats.Tar; | ||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Common.Data.Backups; | using Phantom.Common.Data.Backups; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
| @@ -25,7 +25,7 @@ sealed class BackupArchiver { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private bool IsFolderSkipped(ImmutableList<string> relativePath) { | 	private bool IsFolderSkipped(ImmutableList<string> relativePath) { | ||||||
| 		return relativePath is ["cache" or "crash-reports" or "debug" or "libraries" or "logs" or "mods" or "servermods" or "versions"]; | 		return relativePath is ["cache" or "crash-reports" or "debug" or "libraries" or "logs" or "mods" or "versions"]; | ||||||
| 	} | 	} | ||||||
| 	 | 	 | ||||||
| 	[SuppressMessage("ReSharper", "ConvertIfStatementToReturnStatement")] | 	[SuppressMessage("ReSharper", "ConvertIfStatementToReturnStatement")] | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| using Phantom.Utils.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Processes; | using Phantom.Utils.Processes; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Common.Data.Backups; | using Phantom.Common.Data.Backups; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Common.Logging; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
| @@ -67,10 +67,6 @@ sealed class BackupManager : IDisposable { | |||||||
| 				resultBuilder.Kind = BackupCreationResultKind.BackupCancelled; | 				resultBuilder.Kind = BackupCreationResultKind.BackupCancelled; | ||||||
| 				logger.Warning("Backup creation was cancelled."); | 				logger.Warning("Backup creation was cancelled."); | ||||||
| 				return null; | 				return null; | ||||||
| 			} catch (TimeoutException) { |  | ||||||
| 				resultBuilder.Kind = BackupCreationResultKind.BackupTimedOut; |  | ||||||
| 				logger.Warning("Backup creation timed out."); |  | ||||||
| 				return null; |  | ||||||
| 			} catch (Exception e) { | 			} catch (Exception e) { | ||||||
| 				resultBuilder.Kind = BackupCreationResultKind.UnknownError; | 				resultBuilder.Kind = BackupCreationResultKind.UnknownError; | ||||||
| 				logger.Error(e, "Caught exception while creating an instance backup."); | 				logger.Error(e, "Caught exception while creating an instance backup."); | ||||||
| @@ -80,9 +76,6 @@ sealed class BackupManager : IDisposable { | |||||||
| 					await dispatcher.EnableAutomaticSaving(); | 					await dispatcher.EnableAutomaticSaving(); | ||||||
| 				} catch (OperationCanceledException) { | 				} catch (OperationCanceledException) { | ||||||
| 					// Ignore. | 					// Ignore. | ||||||
| 				} catch (TimeoutException) { |  | ||||||
| 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotRestoreAutomaticSaving; |  | ||||||
| 					logger.Warning("Timed out waiting for automatic saving to be re-enabled."); |  | ||||||
| 				} catch (Exception e) { | 				} catch (Exception e) { | ||||||
| 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotRestoreAutomaticSaving; | 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotRestoreAutomaticSaving; | ||||||
| 					logger.Error(e, "Caught exception while enabling automatic saving after creating an instance backup."); | 					logger.Error(e, "Caught exception while enabling automatic saving after creating an instance backup."); | ||||||
| @@ -127,7 +120,6 @@ sealed class BackupManager : IDisposable { | |||||||
| 				BackupCreationResultKind.Success                            => "Backup created successfully.", | 				BackupCreationResultKind.Success                            => "Backup created successfully.", | ||||||
| 				BackupCreationResultKind.InstanceNotRunning                 => "Instance is not running.", | 				BackupCreationResultKind.InstanceNotRunning                 => "Instance is not running.", | ||||||
| 				BackupCreationResultKind.BackupCancelled                    => "Backup cancelled.", | 				BackupCreationResultKind.BackupCancelled                    => "Backup cancelled.", | ||||||
| 				BackupCreationResultKind.BackupTimedOut                     => "Backup timed out.", |  | ||||||
| 				BackupCreationResultKind.BackupAlreadyRunning               => "A backup is already being created.", | 				BackupCreationResultKind.BackupAlreadyRunning               => "A backup is already being created.", | ||||||
| 				BackupCreationResultKind.BackupFileAlreadyExists            => "Backup with the same name already exists.", | 				BackupCreationResultKind.BackupFileAlreadyExists            => "Backup with the same name already exists.", | ||||||
| 				BackupCreationResultKind.CouldNotCreateBackupFolder         => "Could not create backup folder.", | 				BackupCreationResultKind.CouldNotCreateBackupFolder         => "Could not create backup folder.", | ||||||
|   | |||||||
| @@ -1,8 +1,11 @@ | |||||||
| using Phantom.Agent.Services.Instances; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Agent.Services.Instances.State; | using Phantom.Agent.Minecraft.Server; | ||||||
|  | using Phantom.Agent.Services.Instances; | ||||||
|  | using Phantom.Agent.Services.Instances.Procedures; | ||||||
| using Phantom.Common.Data.Backups; | using Phantom.Common.Data.Backups; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Tasks; | ||||||
|  | using Phantom.Utils.Threading; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
|  |  | ||||||
| @@ -13,24 +16,28 @@ sealed class BackupScheduler : CancellableBackgroundTask { | |||||||
| 	private static readonly TimeSpan BackupFailureRetryDelay = TimeSpan.FromMinutes(5); | 	private static readonly TimeSpan BackupFailureRetryDelay = TimeSpan.FromMinutes(5); | ||||||
|  |  | ||||||
| 	private readonly BackupManager backupManager; | 	private readonly BackupManager backupManager; | ||||||
| 	private readonly InstanceContext context; | 	private readonly InstanceProcess process; | ||||||
|  | 	private readonly IInstanceContext context; | ||||||
| 	private readonly SemaphoreSlim backupSemaphore = new (1, 1); | 	private readonly SemaphoreSlim backupSemaphore = new (1, 1); | ||||||
|  | 	private readonly int serverPort; | ||||||
|  | 	private readonly ServerStatusProtocol serverStatusProtocol; | ||||||
| 	private readonly ManualResetEventSlim serverOutputWhileWaitingForOnlinePlayers = new (); | 	private readonly ManualResetEventSlim serverOutputWhileWaitingForOnlinePlayers = new (); | ||||||
| 	private readonly InstancePlayerCountTracker playerCountTracker; |  | ||||||
| 	 | 	 | ||||||
| 	public event EventHandler<BackupCreationResult>? BackupCompleted; | 	public event EventHandler<BackupCreationResult>? BackupCompleted; | ||||||
|  |  | ||||||
| 	public BackupScheduler(InstanceContext context, InstancePlayerCountTracker playerCountTracker) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName)) { | 	public BackupScheduler(TaskManager taskManager, BackupManager backupManager, InstanceProcess process, IInstanceContext context, int serverPort) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName), taskManager, "Backup scheduler for " + context.ShortName) { | ||||||
| 		this.backupManager = context.Services.BackupManager; | 		this.backupManager = backupManager; | ||||||
|  | 		this.process = process; | ||||||
| 		this.context = context; | 		this.context = context; | ||||||
| 		this.playerCountTracker = playerCountTracker; | 		this.serverPort = serverPort; | ||||||
|  | 		this.serverStatusProtocol = new ServerStatusProtocol(context.ShortName); | ||||||
| 		Start(); | 		Start(); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	protected override async Task RunTask() { | 	protected override async Task RunTask() { | ||||||
| 		await Task.Delay(InitialDelay, CancellationToken); | 		await Task.Delay(InitialDelay, CancellationToken); | ||||||
| 		Logger.Information("Starting a new backup after server launched."); | 		Logger.Information("Starting a new backup after server launched."); | ||||||
| 		 | 			 | ||||||
| 		while (!CancellationToken.IsCancellationRequested) { | 		while (!CancellationToken.IsCancellationRequested) { | ||||||
| 			var result = await CreateBackup(); | 			var result = await CreateBackup(); | ||||||
| 			BackupCompleted?.Invoke(this, result); | 			BackupCompleted?.Invoke(this, result); | ||||||
| @@ -53,28 +60,52 @@ sealed class BackupScheduler : CancellableBackgroundTask { | |||||||
| 		} | 		} | ||||||
| 		 | 		 | ||||||
| 		try { | 		try { | ||||||
| 			context.ActorCancellationToken.ThrowIfCancellationRequested(); | 			var procedure = new BackupInstanceProcedure(backupManager); | ||||||
| 			return await context.Actor.Request(new InstanceActor.BackupInstanceCommand(backupManager), context.ActorCancellationToken); | 			context.EnqueueProcedure(procedure); | ||||||
| 		} catch (OperationCanceledException) { | 			return await procedure.Result; | ||||||
| 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); |  | ||||||
| 		} finally { | 		} finally { | ||||||
| 			backupSemaphore.Release(); | 			backupSemaphore.Release(); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task WaitForOnlinePlayers() { | 	private async Task WaitForOnlinePlayers() { | ||||||
| 		var task = playerCountTracker.WaitForOnlinePlayers(CancellationToken); | 		bool needsToLogOfflinePlayersMessage = true; | ||||||
| 		if (!task.IsCompleted) { |  | ||||||
| 			Logger.Information("Waiting for someone to join before starting a new backup."); |  | ||||||
| 		} |  | ||||||
| 		 | 		 | ||||||
|  | 		process.AddOutputListener(ServerOutputListener, maxLinesToReadFromHistory: 0); | ||||||
| 		try { | 		try { | ||||||
| 			await task; | 			while (!CancellationToken.IsCancellationRequested) { | ||||||
| 			Logger.Information("Players are online, starting a new backup."); | 				serverOutputWhileWaitingForOnlinePlayers.Reset(); | ||||||
| 		} catch (OperationCanceledException) { | 				 | ||||||
| 			throw; | 				var onlinePlayerCount = await serverStatusProtocol.GetOnlinePlayerCount(serverPort, CancellationToken); | ||||||
| 		} catch (Exception) { | 				if (onlinePlayerCount == null) { | ||||||
| 			Logger.Warning("Could not detect whether any players are online, starting a new backup."); | 					Logger.Warning("Could not detect whether any players are online, starting a new backup."); | ||||||
|  | 					break; | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				if (onlinePlayerCount > 0) { | ||||||
|  | 					Logger.Information("Players are online, starting a new backup."); | ||||||
|  | 					break; | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				if (needsToLogOfflinePlayersMessage) { | ||||||
|  | 					needsToLogOfflinePlayersMessage = false; | ||||||
|  | 					Logger.Information("No players are online, waiting for someone to join before starting a new backup."); | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				await Task.Delay(TimeSpan.FromSeconds(10), CancellationToken); | ||||||
|  | 				 | ||||||
|  | 				Logger.Debug("Waiting for server output before checking for online players again..."); | ||||||
|  | 				await serverOutputWhileWaitingForOnlinePlayers.WaitHandle.WaitOneAsync(CancellationToken); | ||||||
|  | 			} | ||||||
|  | 		} finally { | ||||||
|  | 			process.RemoveOutputListener(ServerOutputListener); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void ServerOutputListener(object? sender, string line) { | ||||||
|  | 		if (!serverOutputWhileWaitingForOnlinePlayers.IsSet) { | ||||||
|  | 			serverOutputWhileWaitingForOnlinePlayers.Set(); | ||||||
|  | 			Logger.Debug("Detected server output, signalling to check for online players again."); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,5 +1,4 @@ | |||||||
| using System.Collections.Immutable; | using System.Text.RegularExpressions; | ||||||
| using System.Text.RegularExpressions; |  | ||||||
| using Phantom.Agent.Minecraft.Command; | using Phantom.Agent.Minecraft.Command; | ||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Tasks; | ||||||
| @@ -8,26 +7,8 @@ using Serilog; | |||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
|  |  | ||||||
| sealed partial class BackupServerCommandDispatcher : IDisposable { | sealed partial class BackupServerCommandDispatcher : IDisposable { | ||||||
| 	[GeneratedRegex(@"^(?:(?:\[.*?\] \[Server thread/INFO\].*?:)|(?:[\d-]+? [\d:]+? \[INFO\])) (.*?)$", RegexOptions.NonBacktracking)] | 	[GeneratedRegex(@"^\[(?:.*?)\] \[Server thread/INFO\]: (.*?)$", RegexOptions.NonBacktracking)] | ||||||
| 	private static partial Regex ServerThreadInfoRegex(); | 	private static partial Regex ServerThreadInfoRegex(); | ||||||
| 	 |  | ||||||
| 	private static readonly ImmutableHashSet<string> AutomaticSavingDisabledMessages = ImmutableHashSet.Create( |  | ||||||
| 		"Automatic saving is now disabled", |  | ||||||
| 		"Turned off world auto-saving", |  | ||||||
| 		"CONSOLE: Disabling level saving.." |  | ||||||
| 	); |  | ||||||
| 	 |  | ||||||
| 	private static readonly ImmutableHashSet<string> SavedTheGameMessages = ImmutableHashSet.Create( |  | ||||||
| 		"Saved the game", |  | ||||||
| 		"Saved the world", |  | ||||||
| 		"CONSOLE: Save complete." |  | ||||||
| 	); |  | ||||||
| 	 |  | ||||||
| 	private static readonly ImmutableHashSet<string> AutomaticSavingEnabledMessages = ImmutableHashSet.Create( |  | ||||||
| 		"Automatic saving is now enabled", |  | ||||||
| 		"Turned on world auto-saving", |  | ||||||
| 		"CONSOLE: Enabling level saving.." |  | ||||||
| 	); |  | ||||||
|  |  | ||||||
| 	private readonly ILogger logger; | 	private readonly ILogger logger; | ||||||
| 	private readonly InstanceProcess process; | 	private readonly InstanceProcess process; | ||||||
| @@ -51,17 +32,18 @@ sealed partial class BackupServerCommandDispatcher : IDisposable { | |||||||
|  |  | ||||||
| 	public async Task DisableAutomaticSaving() { | 	public async Task DisableAutomaticSaving() { | ||||||
| 		await process.SendCommand(MinecraftCommand.SaveOff, cancellationToken); | 		await process.SendCommand(MinecraftCommand.SaveOff, cancellationToken); | ||||||
| 		await automaticSavingDisabled.Task.WaitAsync(TimeSpan.FromSeconds(30), cancellationToken); | 		await automaticSavingDisabled.Task.WaitAsync(cancellationToken); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task SaveAllChunks() { | 	public async Task SaveAllChunks() { | ||||||
|  | 		// TODO Try if not flushing and waiting a few seconds before flushing reduces lag. | ||||||
| 		await process.SendCommand(MinecraftCommand.SaveAll(flush: true), cancellationToken); | 		await process.SendCommand(MinecraftCommand.SaveAll(flush: true), cancellationToken); | ||||||
| 		await savedTheGame.Task.WaitAsync(TimeSpan.FromMinutes(1), cancellationToken); | 		await savedTheGame.Task.WaitAsync(cancellationToken); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task EnableAutomaticSaving() { | 	public async Task EnableAutomaticSaving() { | ||||||
| 		await process.SendCommand(MinecraftCommand.SaveOn, cancellationToken); | 		await process.SendCommand(MinecraftCommand.SaveOn, cancellationToken); | ||||||
| 		await automaticSavingEnabled.Task.WaitAsync(TimeSpan.FromMinutes(1), cancellationToken); | 		await automaticSavingEnabled.Task.WaitAsync(cancellationToken); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void OnOutput(object? sender, string? line) { | 	private void OnOutput(object? sender, string? line) { | ||||||
| @@ -77,19 +59,19 @@ sealed partial class BackupServerCommandDispatcher : IDisposable { | |||||||
| 		string info = match.Groups[1].Value; | 		string info = match.Groups[1].Value; | ||||||
|  |  | ||||||
| 		if (!automaticSavingDisabled.Task.IsCompleted) { | 		if (!automaticSavingDisabled.Task.IsCompleted) { | ||||||
| 			if (AutomaticSavingDisabledMessages.Contains(info)) { | 			if (info == "Automatic saving is now disabled") { | ||||||
| 				logger.Debug("Detected that automatic saving is disabled."); | 				logger.Debug("Detected that automatic saving is disabled."); | ||||||
| 				automaticSavingDisabled.SetResult(); | 				automaticSavingDisabled.SetResult(); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		else if (!savedTheGame.Task.IsCompleted) { | 		else if (!savedTheGame.Task.IsCompleted) { | ||||||
| 			if (SavedTheGameMessages.Contains(info)) { | 			if (info == "Saved the game") { | ||||||
| 				logger.Debug("Detected that the game is saved."); | 				logger.Debug("Detected that the game is saved."); | ||||||
| 				savedTheGame.SetResult(); | 				savedTheGame.SetResult(); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		else if (!automaticSavingEnabled.Task.IsCompleted) { | 		else if (!automaticSavingEnabled.Task.IsCompleted) { | ||||||
| 			if (AutomaticSavingEnabledMessages.Contains(info)) { | 			if (info == "Automatic saving is now enabled") { | ||||||
| 				logger.Debug("Detected that automatic saving is enabled."); | 				logger.Debug("Detected that automatic saving is enabled."); | ||||||
| 				automaticSavingEnabled.SetResult(); | 				automaticSavingEnabled.SetResult(); | ||||||
| 			} | 			} | ||||||
|   | |||||||
							
								
								
									
										25
									
								
								Agent/Phantom.Agent.Services/Instances/IInstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								Agent/Phantom.Agent.Services/Instances/IInstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,25 @@ | |||||||
|  | using Phantom.Agent.Services.Instances.Procedures; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
|  | interface IInstanceContext { | ||||||
|  | 	string ShortName { get; } | ||||||
|  | 	ILogger Logger { get; } | ||||||
|  | 	 | ||||||
|  | 	InstanceServices Services { get; } | ||||||
|  | 	IInstanceState CurrentState { get; } | ||||||
|  |  | ||||||
|  | 	void SetStatus(IInstanceStatus newStatus); | ||||||
|  | 	void ReportEvent(IInstanceEvent instanceEvent); | ||||||
|  | 	void EnqueueProcedure(IInstanceProcedure procedure, bool immediate = false); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static class InstanceContextExtensions { | ||||||
|  | 	public static void SetLaunchFailedStatusAndReportEvent(this IInstanceContext context, InstanceLaunchFailReason reason) { | ||||||
|  | 		context.SetStatus(InstanceStatus.Failed(reason)); | ||||||
|  | 		context.ReportEvent(new InstanceLaunchFailedEvent(reason)); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,5 +1,187 @@ | |||||||
| using Phantom.Common.Data.Instance; | using Phantom.Agent.Minecraft.Launcher; | ||||||
|  | using Phantom.Agent.Services.Instances.Procedures; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  | using Phantom.Common.Logging; | ||||||
|  | using Phantom.Common.Messages.Agent.ToController; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
| sealed record Instance(Guid InstanceGuid, IInstanceStatus Status); | sealed class Instance : IAsyncDisposable { | ||||||
|  | 	private InstanceServices Services { get; } | ||||||
|  | 	 | ||||||
|  | 	public InstanceConfiguration Configuration { get; private set; } | ||||||
|  | 	private IServerLauncher Launcher { get; set; } | ||||||
|  | 	private readonly SemaphoreSlim configurationSemaphore = new (1, 1); | ||||||
|  |  | ||||||
|  | 	private readonly string shortName; | ||||||
|  | 	private readonly ILogger logger; | ||||||
|  |  | ||||||
|  | 	private IInstanceStatus currentStatus; | ||||||
|  | 	private int statusUpdateCounter; | ||||||
|  | 	 | ||||||
|  | 	private IInstanceState currentState; | ||||||
|  | 	public bool IsRunning => currentState is not InstanceNotRunningState; | ||||||
|  | 	 | ||||||
|  | 	public event EventHandler? IsRunningChanged; | ||||||
|  | 	 | ||||||
|  | 	private readonly InstanceProcedureManager procedureManager; | ||||||
|  |  | ||||||
|  | 	public Instance(string shortName, InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||||
|  | 		this.shortName = shortName; | ||||||
|  | 		this.logger = PhantomLogger.Create<Instance>(shortName); | ||||||
|  |  | ||||||
|  | 		this.Services = services; | ||||||
|  | 		this.Configuration = configuration; | ||||||
|  | 		this.Launcher = launcher; | ||||||
|  | 		 | ||||||
|  | 		this.currentState = new InstanceNotRunningState(); | ||||||
|  | 		this.currentStatus = InstanceStatus.NotRunning; | ||||||
|  | 		 | ||||||
|  | 		this.procedureManager = new InstanceProcedureManager(this, new Context(this), services.TaskManager); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void TryUpdateStatus(string taskName, Func<Task> getUpdateTask) { | ||||||
|  | 		int myStatusUpdateCounter = Interlocked.Increment(ref statusUpdateCounter); | ||||||
|  | 		 | ||||||
|  | 		Services.TaskManager.Run(taskName, async () => { | ||||||
|  | 			if (myStatusUpdateCounter == statusUpdateCounter) { | ||||||
|  | 				await getUpdateTask(); | ||||||
|  | 			} | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void ReportLastStatus() { | ||||||
|  | 		TryUpdateStatus("Report last status of instance " + shortName, async () => { | ||||||
|  | 			await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, currentStatus)); | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void ReportAndSetStatus(IInstanceStatus status) { | ||||||
|  | 		TryUpdateStatus("Report status of instance " + shortName + " as " + status.GetType().Name, async () => { | ||||||
|  | 			if (status != currentStatus) { | ||||||
|  | 				currentStatus = status; | ||||||
|  | 				await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, status)); | ||||||
|  | 			} | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void ReportEvent(IInstanceEvent instanceEvent) { | ||||||
|  | 		var message = new ReportInstanceEventMessage(Guid.NewGuid(), DateTime.UtcNow, Configuration.InstanceGuid, instanceEvent); | ||||||
|  | 		Services.TaskManager.Run("Report event for instance " + shortName, async () => await Services.ControllerConnection.Send(message)); | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	internal void TransitionState(IInstanceState newState) { | ||||||
|  | 		if (currentState == newState) { | ||||||
|  | 			return; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (currentState is IDisposable disposable) { | ||||||
|  | 			disposable.Dispose(); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		logger.Debug("Transitioning instance state to: {NewState}", newState.GetType().Name); | ||||||
|  | 		 | ||||||
|  | 		var wasRunning = IsRunning; | ||||||
|  | 		currentState = newState; | ||||||
|  | 		currentState.Initialize(); | ||||||
|  |  | ||||||
|  | 		if (IsRunning != wasRunning) { | ||||||
|  | 			IsRunningChanged?.Invoke(this, EventArgs.Empty); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task Reconfigure(InstanceConfiguration configuration, IServerLauncher launcher, CancellationToken cancellationToken) { | ||||||
|  | 		await configurationSemaphore.WaitAsync(cancellationToken); | ||||||
|  | 		try { | ||||||
|  | 			Configuration = configuration; | ||||||
|  | 			Launcher = launcher; | ||||||
|  | 		} finally { | ||||||
|  | 			configurationSemaphore.Release(); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<LaunchInstanceResult> Launch(CancellationToken cancellationToken) { | ||||||
|  | 		if (IsRunning) { | ||||||
|  | 			return LaunchInstanceResult.InstanceAlreadyRunning; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is LaunchInstanceProcedure) { | ||||||
|  | 			return LaunchInstanceResult.InstanceAlreadyLaunching; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		LaunchInstanceProcedure procedure; | ||||||
|  | 		 | ||||||
|  | 		await configurationSemaphore.WaitAsync(cancellationToken); | ||||||
|  | 		try { | ||||||
|  | 			procedure = new LaunchInstanceProcedure(Configuration, Launcher); | ||||||
|  | 		} finally { | ||||||
|  | 			configurationSemaphore.Release(); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		ReportAndSetStatus(InstanceStatus.Launching); | ||||||
|  | 		await procedureManager.Enqueue(procedure); | ||||||
|  | 		return LaunchInstanceResult.LaunchInitiated; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<StopInstanceResult> Stop(MinecraftStopStrategy stopStrategy, CancellationToken cancellationToken) { | ||||||
|  | 		if (!IsRunning) { | ||||||
|  | 			return StopInstanceResult.InstanceAlreadyStopped; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is StopInstanceProcedure) { | ||||||
|  | 			return StopInstanceResult.InstanceAlreadyStopping; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		ReportAndSetStatus(InstanceStatus.Stopping); | ||||||
|  | 		await procedureManager.Enqueue(new StopInstanceProcedure(stopStrategy)); | ||||||
|  | 		return StopInstanceResult.StopInitiated; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
|  | 		return await currentState.SendCommand(command, cancellationToken); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async ValueTask DisposeAsync() { | ||||||
|  | 		await procedureManager.DisposeAsync(); | ||||||
|  |  | ||||||
|  | 		while (currentState is not InstanceNotRunningState) { | ||||||
|  | 			await Task.Delay(TimeSpan.FromMilliseconds(250), CancellationToken.None); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (currentState is IDisposable disposable) { | ||||||
|  | 			disposable.Dispose(); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		configurationSemaphore.Dispose(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private sealed class Context : IInstanceContext { | ||||||
|  | 		public string ShortName => instance.shortName; | ||||||
|  | 		public ILogger Logger => instance.logger; | ||||||
|  | 		 | ||||||
|  | 		public InstanceServices Services => instance.Services; | ||||||
|  | 		public IInstanceState CurrentState => instance.currentState; | ||||||
|  |  | ||||||
|  | 		private readonly Instance instance; | ||||||
|  | 		 | ||||||
|  | 		public Context(Instance instance) { | ||||||
|  | 			this.instance = instance; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		public void SetStatus(IInstanceStatus newStatus) { | ||||||
|  | 			instance.ReportAndSetStatus(newStatus); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		public void ReportEvent(IInstanceEvent instanceEvent) { | ||||||
|  | 			instance.ReportEvent(instanceEvent); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		public void EnqueueProcedure(IInstanceProcedure procedure, bool immediate) { | ||||||
|  | 			Services.TaskManager.Run("Enqueue procedure for instance " + instance.shortName, () => instance.procedureManager.Enqueue(procedure, immediate)); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,161 +0,0 @@ | |||||||
| using Phantom.Agent.Minecraft.Launcher; |  | ||||||
| using Phantom.Agent.Services.Backups; |  | ||||||
| using Phantom.Agent.Services.Instances.State; |  | ||||||
| using Phantom.Common.Data.Backups; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Data.Minecraft; |  | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Actor; |  | ||||||
| using Phantom.Utils.Actor.Mailbox; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; |  | ||||||
|  |  | ||||||
| sealed class InstanceActor : ReceiveActor<InstanceActor.ICommand> { |  | ||||||
| 	public readonly record struct Init(AgentState AgentState, Guid InstanceGuid, string ShortName, InstanceServices InstanceServices, InstanceTicketManager InstanceTicketManager, CancellationToken ShutdownCancellationToken); |  | ||||||
|  |  | ||||||
| 	public static Props<ICommand> Factory(Init init) { |  | ||||||
| 		return Props<ICommand>.Create(() => new InstanceActor(init), new ActorConfiguration { SupervisorStrategy = SupervisorStrategies.Resume, MailboxType = UnboundedJumpAheadMailbox.Name }); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private readonly AgentState agentState; |  | ||||||
| 	private readonly CancellationToken shutdownCancellationToken; |  | ||||||
| 	 |  | ||||||
| 	private readonly Guid instanceGuid; |  | ||||||
| 	private readonly InstanceServices instanceServices; |  | ||||||
| 	private readonly InstanceTicketManager instanceTicketManager; |  | ||||||
| 	private readonly InstanceContext context; |  | ||||||
| 	 |  | ||||||
| 	private readonly CancellationTokenSource actorCancellationTokenSource = new (); |  | ||||||
|  |  | ||||||
| 	private IInstanceStatus currentStatus = InstanceStatus.NotRunning; |  | ||||||
| 	private InstanceRunningState? runningState = null; |  | ||||||
|  |  | ||||||
| 	private InstanceActor(Init init) { |  | ||||||
| 		this.agentState = init.AgentState; |  | ||||||
| 		this.instanceGuid = init.InstanceGuid; |  | ||||||
| 		this.instanceServices = init.InstanceServices; |  | ||||||
| 		this.instanceTicketManager = init.InstanceTicketManager; |  | ||||||
| 		this.shutdownCancellationToken = init.ShutdownCancellationToken; |  | ||||||
| 		 |  | ||||||
| 		var logger = PhantomLogger.Create<InstanceActor>(init.ShortName); |  | ||||||
| 		this.context = new InstanceContext(instanceGuid, init.ShortName, logger, instanceServices, SelfTyped, actorCancellationTokenSource.Token); |  | ||||||
|  |  | ||||||
| 		Receive<ReportInstanceStatusCommand>(ReportInstanceStatus); |  | ||||||
| 		ReceiveAsync<LaunchInstanceCommand>(LaunchInstance); |  | ||||||
| 		ReceiveAsync<StopInstanceCommand>(StopInstance); |  | ||||||
| 		ReceiveAsyncAndReply<SendCommandToInstanceCommand, SendCommandToInstanceResult>(SendCommandToInstance); |  | ||||||
| 		ReceiveAsyncAndReply<BackupInstanceCommand, BackupCreationResult>(BackupInstance); |  | ||||||
| 		Receive<HandleProcessEndedCommand>(HandleProcessEnded); |  | ||||||
| 		ReceiveAsync<ShutdownCommand>(Shutdown); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void SetAndReportStatus(IInstanceStatus status) { |  | ||||||
| 		currentStatus = status; |  | ||||||
| 		ReportCurrentStatus(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void ReportCurrentStatus() { |  | ||||||
| 		agentState.UpdateInstance(new Instance(instanceGuid, currentStatus)); |  | ||||||
| 		instanceServices.ControllerConnection.Send(new ReportInstanceStatusMessage(instanceGuid, currentStatus)); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void TransitionState(InstanceRunningState? newState) { |  | ||||||
| 		if (runningState == newState) { |  | ||||||
| 			return; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		runningState?.Dispose(); |  | ||||||
| 		runningState = newState; |  | ||||||
| 		runningState?.Initialize(); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public interface ICommand {} |  | ||||||
|  |  | ||||||
| 	public sealed record ReportInstanceStatusCommand : ICommand; |  | ||||||
| 	 |  | ||||||
| 	public sealed record LaunchInstanceCommand(InstanceConfiguration Configuration, IServerLauncher Launcher, InstanceTicketManager.Ticket Ticket, bool IsRestarting) : ICommand; |  | ||||||
| 	 |  | ||||||
| 	public sealed record StopInstanceCommand(MinecraftStopStrategy StopStrategy) : ICommand; |  | ||||||
| 	 |  | ||||||
| 	public sealed record SendCommandToInstanceCommand(string Command) : ICommand, ICanReply<SendCommandToInstanceResult>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record BackupInstanceCommand(BackupManager BackupManager) : ICommand, ICanReply<BackupCreationResult>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record HandleProcessEndedCommand(IInstanceStatus Status) : ICommand, IJumpAhead; |  | ||||||
| 	 |  | ||||||
| 	public sealed record ShutdownCommand : ICommand; |  | ||||||
|  |  | ||||||
| 	private void ReportInstanceStatus(ReportInstanceStatusCommand command) { |  | ||||||
| 		ReportCurrentStatus(); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private async Task LaunchInstance(LaunchInstanceCommand command) { |  | ||||||
| 		if (command.IsRestarting || runningState is null) { |  | ||||||
| 			SetAndReportStatus(command.IsRestarting ? InstanceStatus.Restarting : InstanceStatus.Launching); |  | ||||||
| 			 |  | ||||||
| 			var newState = await InstanceLaunchProcedure.Run(context, command.Configuration, command.Launcher, instanceTicketManager, command.Ticket, SetAndReportStatus, shutdownCancellationToken); |  | ||||||
| 			if (newState is null) { |  | ||||||
| 				instanceTicketManager.Release(command.Ticket); |  | ||||||
| 			} |  | ||||||
| 			 |  | ||||||
| 			TransitionState(newState); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task StopInstance(StopInstanceCommand command) { |  | ||||||
| 		if (runningState is null) { |  | ||||||
| 			return; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		IInstanceStatus oldStatus = currentStatus; |  | ||||||
| 		SetAndReportStatus(InstanceStatus.Stopping); |  | ||||||
| 		 |  | ||||||
| 		if (await InstanceStopProcedure.Run(context, command.StopStrategy, runningState, SetAndReportStatus, shutdownCancellationToken)) { |  | ||||||
| 			instanceTicketManager.Release(runningState.Ticket); |  | ||||||
| 			TransitionState(null); |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			SetAndReportStatus(oldStatus); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<SendCommandToInstanceResult> SendCommandToInstance(SendCommandToInstanceCommand command) { |  | ||||||
| 		if (runningState is null) { |  | ||||||
| 			return SendCommandToInstanceResult.InstanceNotRunning; |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			return await runningState.SendCommand(command.Command, shutdownCancellationToken); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private async Task<BackupCreationResult> BackupInstance(BackupInstanceCommand command) { |  | ||||||
| 		if (runningState is null || runningState.Process.HasEnded) { |  | ||||||
| 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			SetAndReportStatus(InstanceStatus.BackingUp); |  | ||||||
| 			try { |  | ||||||
| 				return await command.BackupManager.CreateBackup(context.ShortName, runningState.Process, shutdownCancellationToken); |  | ||||||
| 			} finally { |  | ||||||
| 				SetAndReportStatus(InstanceStatus.Running); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void HandleProcessEnded(HandleProcessEndedCommand command) { |  | ||||||
| 		if (runningState is { Process.HasEnded: true }) { |  | ||||||
| 			SetAndReportStatus(command.Status); |  | ||||||
| 			context.ReportEvent(InstanceEvent.Stopped); |  | ||||||
| 			instanceTicketManager.Release(runningState.Ticket); |  | ||||||
| 			TransitionState(null); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task Shutdown(ShutdownCommand command) { |  | ||||||
| 		await StopInstance(new StopInstanceCommand(MinecraftStopStrategy.Instant)); |  | ||||||
| 		await actorCancellationTokenSource.CancelAsync(); |  | ||||||
| 		Context.Stop(Self); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,12 +0,0 @@ | |||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Actor; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; |  | ||||||
|  |  | ||||||
| sealed record InstanceContext(Guid InstanceGuid, string ShortName, ILogger Logger, InstanceServices Services, ActorRef<InstanceActor.ICommand> Actor, CancellationToken ActorCancellationToken) { |  | ||||||
| 	public void ReportEvent(IInstanceEvent instanceEvent) { |  | ||||||
| 		Services.ControllerConnection.Send(new ReportInstanceEventMessage(Guid.NewGuid(), DateTime.UtcNow, InstanceGuid, instanceEvent)); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,14 +1,14 @@ | |||||||
| using System.Collections.Immutable; | using System.Collections.Immutable; | ||||||
| using System.Threading.Channels; | using System.Threading.Channels; | ||||||
| using Phantom.Agent.Rpc; | using Phantom.Agent.Rpc; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.Agent.ToController; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Tasks; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Agent.Services.Instances.State; | namespace Phantom.Agent.Services.Instances; | ||||||
| 
 | 
 | ||||||
| sealed class InstanceLogSender : CancellableBackgroundTask { | sealed class InstanceLogSender : CancellableBackgroundTask { | ||||||
| 	private static readonly BoundedChannelOptions BufferOptions = new (capacity: 100) { | 	private static readonly BoundedChannelOptions BufferOptions = new (capacity: 64) { | ||||||
| 		SingleReader = true, | 		SingleReader = true, | ||||||
| 		SingleWriter = true, | 		SingleWriter = true, | ||||||
| 		FullMode = BoundedChannelFullMode.DropNewest | 		FullMode = BoundedChannelFullMode.DropNewest | ||||||
| @@ -22,7 +22,7 @@ sealed class InstanceLogSender : CancellableBackgroundTask { | |||||||
| 	 | 	 | ||||||
| 	private int droppedLinesSinceLastSend; | 	private int droppedLinesSinceLastSend; | ||||||
| 
 | 
 | ||||||
| 	public InstanceLogSender(ControllerConnection controllerConnection, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName)) { | 	public InstanceLogSender(ControllerConnection controllerConnection, TaskManager taskManager, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName), taskManager, "Instance log sender for " + loggerName) { | ||||||
| 		this.controllerConnection = controllerConnection; | 		this.controllerConnection = controllerConnection; | ||||||
| 		this.instanceGuid = instanceGuid; | 		this.instanceGuid = instanceGuid; | ||||||
| 		this.outputChannel = Channel.CreateBounded<string>(BufferOptions, OnLineDropped); | 		this.outputChannel = Channel.CreateBounded<string>(BufferOptions, OnLineDropped); | ||||||
| @@ -36,14 +36,14 @@ sealed class InstanceLogSender : CancellableBackgroundTask { | |||||||
| 		try { | 		try { | ||||||
| 			while (await lineReader.WaitToReadAsync(CancellationToken)) { | 			while (await lineReader.WaitToReadAsync(CancellationToken)) { | ||||||
| 				await Task.Delay(SendDelay, CancellationToken); | 				await Task.Delay(SendDelay, CancellationToken); | ||||||
| 				SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | 				await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | ||||||
| 			} | 			} | ||||||
| 		} catch (OperationCanceledException) { | 		} catch (OperationCanceledException) { | ||||||
| 			// Ignore. | 			// Ignore. | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		// Flush remaining lines. | 		// Flush remaining lines. | ||||||
| 		SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | 		await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private ImmutableArray<string> ReadLinesFromChannel(ChannelReader<string> reader, ImmutableArray<string>.Builder builder) { | 	private ImmutableArray<string> ReadLinesFromChannel(ChannelReader<string> reader, ImmutableArray<string>.Builder builder) { | ||||||
| @@ -61,9 +61,9 @@ sealed class InstanceLogSender : CancellableBackgroundTask { | |||||||
| 		return builder.ToImmutable(); | 		return builder.ToImmutable(); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private void SendOutputToServer(ImmutableArray<string> lines) { | 	private async Task SendOutputToServer(ImmutableArray<string> lines) { | ||||||
| 		if (!lines.IsEmpty) { | 		if (!lines.IsEmpty) { | ||||||
| 			controllerConnection.Send(new InstanceOutputMessage(instanceGuid, lines)); | 			await controllerConnection.Send(new InstanceOutputMessage(instanceGuid, lines)); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| @@ -1,209 +0,0 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; |  | ||||||
| using Phantom.Agent.Minecraft.Java; |  | ||||||
| using Phantom.Agent.Minecraft.Launcher; |  | ||||||
| using Phantom.Agent.Minecraft.Launcher.Types; |  | ||||||
| using Phantom.Agent.Minecraft.Properties; |  | ||||||
| using Phantom.Agent.Minecraft.Server; |  | ||||||
| using Phantom.Agent.Rpc; |  | ||||||
| using Phantom.Agent.Services.Backups; |  | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Data.Minecraft; |  | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
| using Phantom.Utils.Actor; |  | ||||||
| using Phantom.Utils.IO; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; |  | ||||||
|  |  | ||||||
| sealed class InstanceManagerActor : ReceiveActor<InstanceManagerActor.ICommand> { |  | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceManagerActor>(); |  | ||||||
|  |  | ||||||
| 	public readonly record struct Init(ControllerConnection ControllerConnection, AgentFolders AgentFolders, AgentState AgentState, JavaRuntimeRepository JavaRuntimeRepository, InstanceTicketManager InstanceTicketManager, BackupManager BackupManager); |  | ||||||
|  |  | ||||||
| 	public static Props<ICommand> Factory(Init init) { |  | ||||||
| 		return Props<ICommand>.Create(() => new InstanceManagerActor(init), new ActorConfiguration { SupervisorStrategy = SupervisorStrategies.Resume }); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private readonly AgentState agentState; |  | ||||||
| 	private readonly string basePath; |  | ||||||
|  |  | ||||||
| 	private readonly InstanceServices instanceServices; |  | ||||||
| 	private readonly InstanceTicketManager instanceTicketManager; |  | ||||||
| 	private readonly Dictionary<Guid, InstanceInfo> instances = new (); |  | ||||||
|  |  | ||||||
| 	private readonly CancellationTokenSource shutdownCancellationTokenSource = new (); |  | ||||||
| 	private readonly CancellationToken shutdownCancellationToken; |  | ||||||
|  |  | ||||||
| 	private uint instanceLoggerSequenceId = 0; |  | ||||||
|  |  | ||||||
| 	private InstanceManagerActor(Init init) { |  | ||||||
| 		this.agentState = init.AgentState; |  | ||||||
| 		this.basePath = init.AgentFolders.InstancesFolderPath; |  | ||||||
| 		this.instanceTicketManager = init.InstanceTicketManager; |  | ||||||
| 		this.shutdownCancellationToken = shutdownCancellationTokenSource.Token; |  | ||||||
|  |  | ||||||
| 		var minecraftServerExecutables = new MinecraftServerExecutables(init.AgentFolders.ServerExecutableFolderPath); |  | ||||||
| 		var launchServices = new LaunchServices(minecraftServerExecutables, init.JavaRuntimeRepository); |  | ||||||
|  |  | ||||||
| 		this.instanceServices = new InstanceServices(init.ControllerConnection, init.BackupManager, launchServices); |  | ||||||
| 		 |  | ||||||
| 		ReceiveAndReply<ConfigureInstanceCommand, Result<ConfigureInstanceResult, InstanceActionFailure>>(ConfigureInstance); |  | ||||||
| 		ReceiveAndReply<LaunchInstanceCommand, Result<LaunchInstanceResult, InstanceActionFailure>>(LaunchInstance); |  | ||||||
| 		ReceiveAndReply<StopInstanceCommand, Result<StopInstanceResult, InstanceActionFailure>>(StopInstance); |  | ||||||
| 		ReceiveAsyncAndReply<SendCommandToInstanceCommand, Result<SendCommandToInstanceResult, InstanceActionFailure>>(SendCommandToInstance); |  | ||||||
| 		ReceiveAsync<ShutdownCommand>(Shutdown); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private string GetInstanceLoggerName(Guid guid) { |  | ||||||
| 		var prefix = guid.ToString(); |  | ||||||
| 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref instanceLoggerSequenceId); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private sealed record InstanceInfo(ActorRef<InstanceActor.ICommand> Actor, InstanceConfiguration Configuration, IServerLauncher Launcher); |  | ||||||
| 	 |  | ||||||
| 	public interface ICommand {} |  | ||||||
| 	 |  | ||||||
| 	public sealed record ConfigureInstanceCommand(Guid InstanceGuid, InstanceConfiguration Configuration, InstanceLaunchProperties LaunchProperties, bool LaunchNow, bool AlwaysReportStatus) : ICommand, ICanReply<Result<ConfigureInstanceResult, InstanceActionFailure>>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record LaunchInstanceCommand(Guid InstanceGuid) : ICommand, ICanReply<Result<LaunchInstanceResult, InstanceActionFailure>>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record StopInstanceCommand(Guid InstanceGuid, MinecraftStopStrategy StopStrategy) : ICommand, ICanReply<Result<StopInstanceResult, InstanceActionFailure>>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record SendCommandToInstanceCommand(Guid InstanceGuid, string Command) : ICommand, ICanReply<Result<SendCommandToInstanceResult, InstanceActionFailure>>; |  | ||||||
| 	 |  | ||||||
| 	public sealed record ShutdownCommand : ICommand; |  | ||||||
|  |  | ||||||
| 	private Result<ConfigureInstanceResult, InstanceActionFailure> ConfigureInstance(ConfigureInstanceCommand command) { |  | ||||||
| 		var instanceGuid = command.InstanceGuid; |  | ||||||
| 		var configuration = command.Configuration; |  | ||||||
|  |  | ||||||
| 		var instanceFolder = Path.Combine(basePath, instanceGuid.ToString()); |  | ||||||
| 		Directories.Create(instanceFolder, Chmod.URWX_GRX); |  | ||||||
|  |  | ||||||
| 		var heapMegabytes = configuration.MemoryAllocation.InMegabytes; |  | ||||||
| 		var jvmProperties = new JvmProperties( |  | ||||||
| 			InitialHeapMegabytes: heapMegabytes / 2, |  | ||||||
| 			MaximumHeapMegabytes: heapMegabytes |  | ||||||
| 		); |  | ||||||
|  |  | ||||||
| 		var properties = new InstanceProperties( |  | ||||||
| 			instanceGuid, |  | ||||||
| 			configuration.JavaRuntimeGuid, |  | ||||||
| 			jvmProperties, |  | ||||||
| 			configuration.JvmArguments, |  | ||||||
| 			instanceFolder, |  | ||||||
| 			configuration.MinecraftVersion, |  | ||||||
| 			new ServerProperties(configuration.ServerPort, configuration.RconPort), |  | ||||||
| 			command.LaunchProperties |  | ||||||
| 		); |  | ||||||
|  |  | ||||||
| 		IServerLauncher launcher = configuration.MinecraftServerKind switch { |  | ||||||
| 			MinecraftServerKind.Vanilla => new VanillaLauncher(properties), |  | ||||||
| 			MinecraftServerKind.Fabric  => new FabricLauncher(properties), |  | ||||||
| 			MinecraftServerKind.Forge   => new ForgeLauncher(properties), |  | ||||||
| 			_                           => InvalidLauncher.Instance |  | ||||||
| 		}; |  | ||||||
|  |  | ||||||
| 		if (instances.TryGetValue(instanceGuid, out var instance)) { |  | ||||||
| 			instances[instanceGuid] = instance with { |  | ||||||
| 				Configuration = configuration, |  | ||||||
| 				Launcher = launcher |  | ||||||
| 			}; |  | ||||||
| 				 |  | ||||||
| 			Logger.Information("Reconfigured instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, instanceGuid); |  | ||||||
|  |  | ||||||
| 			if (command.AlwaysReportStatus) { |  | ||||||
| 				instance.Actor.Tell(new InstanceActor.ReportInstanceStatusCommand()); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			var instanceInit = new InstanceActor.Init(agentState, instanceGuid, GetInstanceLoggerName(instanceGuid), instanceServices, instanceTicketManager, shutdownCancellationToken); |  | ||||||
| 			instances[instanceGuid] = instance = new InstanceInfo(Context.ActorOf(InstanceActor.Factory(instanceInit), "Instance-" + instanceGuid), configuration, launcher); |  | ||||||
| 				 |  | ||||||
| 			Logger.Information("Created instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, instanceGuid); |  | ||||||
|  |  | ||||||
| 			instance.Actor.Tell(new InstanceActor.ReportInstanceStatusCommand()); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if (command.LaunchNow) { |  | ||||||
| 			LaunchInstance(new LaunchInstanceCommand(instanceGuid)); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		return ConfigureInstanceResult.Success; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private Result<LaunchInstanceResult, InstanceActionFailure> LaunchInstance(LaunchInstanceCommand command) { |  | ||||||
| 		var instanceGuid = command.InstanceGuid; |  | ||||||
| 		if (!instances.TryGetValue(instanceGuid, out var instanceInfo)) { |  | ||||||
| 			return InstanceActionFailure.InstanceDoesNotExist; |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		var ticket = instanceTicketManager.Reserve(instanceInfo.Configuration); |  | ||||||
| 		if (!ticket) { |  | ||||||
| 			return ticket.Error; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if (agentState.InstancesByGuid.TryGetValue(instanceGuid, out var instance)) { |  | ||||||
| 			var status = instance.Status; |  | ||||||
| 			if (status.IsRunning()) { |  | ||||||
| 				return LaunchInstanceResult.InstanceAlreadyRunning; |  | ||||||
| 			} |  | ||||||
| 			else if (status.IsLaunching()) { |  | ||||||
| 				return LaunchInstanceResult.InstanceAlreadyLaunching; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		instanceInfo.Actor.Tell(new InstanceActor.LaunchInstanceCommand(instanceInfo.Configuration, instanceInfo.Launcher, ticket.Value, IsRestarting: false)); |  | ||||||
| 		return LaunchInstanceResult.LaunchInitiated; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private Result<StopInstanceResult, InstanceActionFailure> StopInstance(StopInstanceCommand command) { |  | ||||||
| 		var instanceGuid = command.InstanceGuid; |  | ||||||
| 		if (!instances.TryGetValue(instanceGuid, out var instanceInfo)) { |  | ||||||
| 			return InstanceActionFailure.InstanceDoesNotExist; |  | ||||||
| 		} |  | ||||||
|          |  | ||||||
| 		if (agentState.InstancesByGuid.TryGetValue(instanceGuid, out var instance)) { |  | ||||||
| 			var status = instance.Status; |  | ||||||
| 			if (status.IsStopping()) { |  | ||||||
| 				return StopInstanceResult.InstanceAlreadyStopping; |  | ||||||
| 			} |  | ||||||
| 			else if (!status.CanStop()) { |  | ||||||
| 				return StopInstanceResult.InstanceAlreadyStopped; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 			 |  | ||||||
| 		instanceInfo.Actor.Tell(new InstanceActor.StopInstanceCommand(command.StopStrategy)); |  | ||||||
| 		return StopInstanceResult.StopInitiated; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<Result<SendCommandToInstanceResult, InstanceActionFailure>> SendCommandToInstance(SendCommandToInstanceCommand command) { |  | ||||||
| 		var instanceGuid = command.InstanceGuid; |  | ||||||
| 		if (!instances.TryGetValue(instanceGuid, out var instanceInfo)) { |  | ||||||
| 			return InstanceActionFailure.InstanceDoesNotExist; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		try { |  | ||||||
| 			return await instanceInfo.Actor.Request(new InstanceActor.SendCommandToInstanceCommand(command.Command), shutdownCancellationToken); |  | ||||||
| 		} catch (OperationCanceledException) { |  | ||||||
| 			return InstanceActionFailure.AgentShuttingDown; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task Shutdown(ShutdownCommand command) { |  | ||||||
| 		Logger.Information("Stopping all instances..."); |  | ||||||
| 		 |  | ||||||
| 		await shutdownCancellationTokenSource.CancelAsync(); |  | ||||||
| 		 |  | ||||||
| 		await Task.WhenAll(instances.Values.Select(static instance => instance.Actor.Stop(new InstanceActor.ShutdownCommand()))); |  | ||||||
| 		instances.Clear(); |  | ||||||
| 		 |  | ||||||
| 		shutdownCancellationTokenSource.Dispose(); |  | ||||||
| 		 |  | ||||||
| 		Logger.Information("All instances stopped."); |  | ||||||
| 		 |  | ||||||
| 		Context.Stop(Self); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -0,0 +1,85 @@ | |||||||
|  | using Phantom.Agent.Services.Instances.Procedures; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Utils.Tasks; | ||||||
|  | using Phantom.Utils.Threading; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
|  | sealed class InstanceProcedureManager : IAsyncDisposable { | ||||||
|  | 	private readonly record struct CurrentProcedure(IInstanceProcedure Procedure, CancellationTokenSource CancellationTokenSource); | ||||||
|  |  | ||||||
|  | 	private readonly ThreadSafeStructRef<CurrentProcedure> currentProcedure = new (); | ||||||
|  | 	private readonly ThreadSafeLinkedList<IInstanceProcedure> procedureQueue = new (); | ||||||
|  | 	private readonly AutoResetEvent procedureQueueReady = new (false); | ||||||
|  | 	private readonly ManualResetEventSlim procedureQueueFinished = new (false); | ||||||
|  |  | ||||||
|  | 	private readonly Instance instance; | ||||||
|  | 	private readonly IInstanceContext context; | ||||||
|  | 	private readonly CancellationTokenSource shutdownCancellationTokenSource = new (); | ||||||
|  |  | ||||||
|  | 	public InstanceProcedureManager(Instance instance, IInstanceContext context, TaskManager taskManager) { | ||||||
|  | 		this.instance = instance; | ||||||
|  | 		this.context = context; | ||||||
|  | 		taskManager.Run("Procedure manager for instance " + context.ShortName, Run); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task Enqueue(IInstanceProcedure procedure, bool immediate = false) { | ||||||
|  | 		await procedureQueue.Add(procedure, toFront: immediate, shutdownCancellationTokenSource.Token); | ||||||
|  | 		procedureQueueReady.Set(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<IInstanceProcedure?> GetCurrentProcedure(CancellationToken cancellationToken) { | ||||||
|  | 		return (await currentProcedure.Get(cancellationToken))?.Procedure; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task Run() { | ||||||
|  | 		try { | ||||||
|  | 			var shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||||
|  | 			while (true) { | ||||||
|  | 				await procedureQueueReady.WaitOneAsync(shutdownCancellationToken); | ||||||
|  | 				while (await procedureQueue.TryTakeFromFront(shutdownCancellationToken) is {} nextProcedure) { | ||||||
|  | 					using var procedureCancellationTokenSource = new CancellationTokenSource(); | ||||||
|  | 					await currentProcedure.Set(new CurrentProcedure(nextProcedure, procedureCancellationTokenSource), shutdownCancellationToken); | ||||||
|  | 					await RunProcedure(nextProcedure, procedureCancellationTokenSource.Token); | ||||||
|  | 					await currentProcedure.Set(null, shutdownCancellationToken); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			// Ignore. | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		await RunProcedure(new StopInstanceProcedure(MinecraftStopStrategy.Instant), CancellationToken.None); | ||||||
|  | 		procedureQueueFinished.Set(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task RunProcedure(IInstanceProcedure procedure, CancellationToken cancellationToken) { | ||||||
|  | 		var procedureName = procedure.GetType().Name; | ||||||
|  |  | ||||||
|  | 		context.Logger.Debug("Started procedure: {Procedure}", procedureName); | ||||||
|  | 		try { | ||||||
|  | 			var newState = await procedure.Run(context, cancellationToken); | ||||||
|  | 			context.Logger.Debug("Finished procedure: {Procedure}", procedureName); | ||||||
|  |  | ||||||
|  | 			if (newState != null) { | ||||||
|  | 				instance.TransitionState(newState); | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			context.Logger.Debug("Cancelled procedure: {Procedure}", procedureName); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			context.Logger.Error(e, "Caught exception while running procedure: {Procedure}", procedureName); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async ValueTask DisposeAsync() { | ||||||
|  | 		shutdownCancellationTokenSource.Cancel(); | ||||||
|  |  | ||||||
|  | 		(await currentProcedure.Get(CancellationToken.None))?.CancellationTokenSource.Cancel(); | ||||||
|  | 		await procedureQueueFinished.WaitHandle.WaitOneAsync(); | ||||||
|  |  | ||||||
|  | 		currentProcedure.Dispose(); | ||||||
|  | 		procedureQueue.Dispose(); | ||||||
|  | 		procedureQueueReady.Dispose(); | ||||||
|  | 		procedureQueueFinished.Dispose(); | ||||||
|  | 		shutdownCancellationTokenSource.Dispose(); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,7 +1,8 @@ | |||||||
| using Phantom.Agent.Minecraft.Launcher; | using Phantom.Agent.Minecraft.Launcher; | ||||||
| using Phantom.Agent.Rpc; | using Phantom.Agent.Rpc; | ||||||
| using Phantom.Agent.Services.Backups; | using Phantom.Agent.Services.Backups; | ||||||
|  | using Phantom.Utils.Tasks; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
| sealed record InstanceServices(ControllerConnection ControllerConnection, BackupManager BackupManager, LaunchServices LaunchServices); | sealed record InstanceServices(ControllerConnection ControllerConnection, TaskManager TaskManager, PortManager PortManager, BackupManager BackupManager, LaunchServices LaunchServices); | ||||||
|   | |||||||
							
								
								
									
										198
									
								
								Agent/Phantom.Agent.Services/Instances/InstanceSessionManager.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										198
									
								
								Agent/Phantom.Agent.Services/Instances/InstanceSessionManager.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,198 @@ | |||||||
|  | using System.Collections.Immutable; | ||||||
|  | using System.Diagnostics.CodeAnalysis; | ||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Agent.Minecraft.Java; | ||||||
|  | using Phantom.Agent.Minecraft.Launcher; | ||||||
|  | using Phantom.Agent.Minecraft.Launcher.Types; | ||||||
|  | using Phantom.Agent.Minecraft.Properties; | ||||||
|  | using Phantom.Agent.Minecraft.Server; | ||||||
|  | using Phantom.Agent.Rpc; | ||||||
|  | using Phantom.Agent.Services.Backups; | ||||||
|  | using Phantom.Common.Data; | ||||||
|  | using Phantom.Common.Data.Agent; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  | using Phantom.Common.Logging; | ||||||
|  | using Phantom.Common.Messages.Agent.ToController; | ||||||
|  | using Phantom.Utils.IO; | ||||||
|  | using Phantom.Utils.Tasks; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
|  | sealed class InstanceSessionManager : IAsyncDisposable { | ||||||
|  | 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceSessionManager>(); | ||||||
|  |  | ||||||
|  | 	private readonly ControllerConnection controllerConnection; | ||||||
|  | 	private readonly AgentInfo agentInfo; | ||||||
|  | 	private readonly string basePath; | ||||||
|  |  | ||||||
|  | 	private readonly InstanceServices instanceServices; | ||||||
|  | 	private readonly Dictionary<Guid, Instance> instances = new (); | ||||||
|  |  | ||||||
|  | 	private readonly CancellationTokenSource shutdownCancellationTokenSource = new (); | ||||||
|  | 	private readonly CancellationToken shutdownCancellationToken; | ||||||
|  | 	private readonly SemaphoreSlim semaphore = new (1, 1); | ||||||
|  |  | ||||||
|  | 	private uint instanceLoggerSequenceId = 0; | ||||||
|  |  | ||||||
|  | 	public InstanceSessionManager(ControllerConnection controllerConnection, AgentInfo agentInfo, AgentFolders agentFolders, JavaRuntimeRepository javaRuntimeRepository, TaskManager taskManager, BackupManager backupManager) { | ||||||
|  | 		this.controllerConnection = controllerConnection; | ||||||
|  | 		this.agentInfo = agentInfo; | ||||||
|  | 		this.basePath = agentFolders.InstancesFolderPath; | ||||||
|  | 		this.shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||||
|  |  | ||||||
|  | 		var minecraftServerExecutables = new MinecraftServerExecutables(agentFolders.ServerExecutableFolderPath); | ||||||
|  | 		var launchServices = new LaunchServices(minecraftServerExecutables, javaRuntimeRepository); | ||||||
|  | 		var portManager = new PortManager(agentInfo.AllowedServerPorts, agentInfo.AllowedRconPorts); | ||||||
|  |  | ||||||
|  | 		this.instanceServices = new InstanceServices(controllerConnection, taskManager, portManager, backupManager, launchServices); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<InstanceActionResult<T>> AcquireSemaphoreAndRun<T>(Func<Task<InstanceActionResult<T>>> func) { | ||||||
|  | 		try { | ||||||
|  | 			await semaphore.WaitAsync(shutdownCancellationToken); | ||||||
|  | 			try { | ||||||
|  | 				return await func(); | ||||||
|  | 			} finally { | ||||||
|  | 				semaphore.Release(); | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			return InstanceActionResult.General<T>(InstanceActionGeneralResult.AgentShuttingDown); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	[SuppressMessage("ReSharper", "ConvertIfStatementToReturnStatement")] | ||||||
|  | 	private Task<InstanceActionResult<T>> AcquireSemaphoreAndRunWithInstance<T>(Guid instanceGuid, Func<Instance, Task<T>> func) { | ||||||
|  | 		return AcquireSemaphoreAndRun(async () => { | ||||||
|  | 			if (instances.TryGetValue(instanceGuid, out var instance)) { | ||||||
|  | 				return InstanceActionResult.Concrete(await func(instance)); | ||||||
|  | 			} | ||||||
|  | 			else { | ||||||
|  | 				return InstanceActionResult.General<T>(InstanceActionGeneralResult.InstanceDoesNotExist); | ||||||
|  | 			} | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<InstanceActionResult<ConfigureInstanceResult>> Configure(InstanceConfiguration configuration, InstanceLaunchProperties launchProperties, bool launchNow, bool alwaysReportStatus) { | ||||||
|  | 		return await AcquireSemaphoreAndRun(async () => { | ||||||
|  | 			var instanceGuid = configuration.InstanceGuid; | ||||||
|  | 			var instanceFolder = Path.Combine(basePath, instanceGuid.ToString()); | ||||||
|  | 			Directories.Create(instanceFolder, Chmod.URWX_GRX); | ||||||
|  |  | ||||||
|  | 			var heapMegabytes = configuration.MemoryAllocation.InMegabytes; | ||||||
|  | 			var jvmProperties = new JvmProperties( | ||||||
|  | 				InitialHeapMegabytes: heapMegabytes / 2, | ||||||
|  | 				MaximumHeapMegabytes: heapMegabytes | ||||||
|  | 			); | ||||||
|  |  | ||||||
|  | 			var properties = new InstanceProperties( | ||||||
|  | 				instanceGuid, | ||||||
|  | 				configuration.JavaRuntimeGuid, | ||||||
|  | 				jvmProperties, | ||||||
|  | 				configuration.JvmArguments, | ||||||
|  | 				instanceFolder, | ||||||
|  | 				configuration.MinecraftVersion, | ||||||
|  | 				new ServerProperties(configuration.ServerPort, configuration.RconPort), | ||||||
|  | 				launchProperties | ||||||
|  | 			); | ||||||
|  |  | ||||||
|  | 			IServerLauncher launcher = configuration.MinecraftServerKind switch { | ||||||
|  | 				MinecraftServerKind.Vanilla => new VanillaLauncher(properties), | ||||||
|  | 				MinecraftServerKind.Fabric  => new FabricLauncher(properties), | ||||||
|  | 				_                           => InvalidLauncher.Instance | ||||||
|  | 			}; | ||||||
|  |  | ||||||
|  | 			if (instances.TryGetValue(instanceGuid, out var instance)) { | ||||||
|  | 				await instance.Reconfigure(configuration, launcher, shutdownCancellationToken); | ||||||
|  | 				Logger.Information("Reconfigured instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | ||||||
|  |  | ||||||
|  | 				if (alwaysReportStatus) { | ||||||
|  | 					instance.ReportLastStatus(); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			else { | ||||||
|  | 				instances[instanceGuid] = instance = new Instance(GetInstanceLoggerName(instanceGuid), instanceServices, configuration, launcher); | ||||||
|  | 				Logger.Information("Created instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | ||||||
|  |  | ||||||
|  | 				instance.ReportLastStatus(); | ||||||
|  | 				instance.IsRunningChanged += OnInstanceIsRunningChanged; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if (launchNow) { | ||||||
|  | 				await LaunchInternal(instance); | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			return InstanceActionResult.Concrete(ConfigureInstanceResult.Success); | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private string GetInstanceLoggerName(Guid guid) { | ||||||
|  | 		var prefix = guid.ToString(); | ||||||
|  | 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref instanceLoggerSequenceId); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private ImmutableArray<Instance> GetRunningInstancesInternal() { | ||||||
|  | 		return instances.Values.Where(static instance => instance.IsRunning).ToImmutableArray(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void OnInstanceIsRunningChanged(object? sender, EventArgs e) { | ||||||
|  | 		instanceServices.TaskManager.Run("Handle instance running state changed event", RefreshAgentStatus); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task RefreshAgentStatus() { | ||||||
|  | 		try { | ||||||
|  | 			await semaphore.WaitAsync(shutdownCancellationToken); | ||||||
|  | 			try { | ||||||
|  | 				var runningInstances = GetRunningInstancesInternal(); | ||||||
|  | 				var runningInstanceCount = runningInstances.Length; | ||||||
|  | 				var runningInstanceMemory = runningInstances.Aggregate(RamAllocationUnits.Zero, static (total, instance) => total + instance.Configuration.MemoryAllocation); | ||||||
|  | 				await controllerConnection.Send(new ReportAgentStatusMessage(runningInstanceCount, runningInstanceMemory)); | ||||||
|  | 			} finally { | ||||||
|  | 				semaphore.Release(); | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			// ignore | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<InstanceActionResult<LaunchInstanceResult>> Launch(Guid instanceGuid) { | ||||||
|  | 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, LaunchInternal); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<LaunchInstanceResult> LaunchInternal(Instance instance) { | ||||||
|  | 		var runningInstances = GetRunningInstancesInternal(); | ||||||
|  | 		if (runningInstances.Length + 1 > agentInfo.MaxInstances) { | ||||||
|  | 			return LaunchInstanceResult.InstanceLimitExceeded; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		var availableMemory = agentInfo.MaxMemory - runningInstances.Aggregate(RamAllocationUnits.Zero, static (total, instance) => total + instance.Configuration.MemoryAllocation); | ||||||
|  | 		if (availableMemory < instance.Configuration.MemoryAllocation) { | ||||||
|  | 			return LaunchInstanceResult.MemoryLimitExceeded; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return await instance.Launch(shutdownCancellationToken); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<InstanceActionResult<StopInstanceResult>> Stop(Guid instanceGuid, MinecraftStopStrategy stopStrategy) { | ||||||
|  | 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, instance => instance.Stop(stopStrategy, shutdownCancellationToken)); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<InstanceActionResult<SendCommandToInstanceResult>> SendCommand(Guid instanceGuid, string command) { | ||||||
|  | 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, async instance => await instance.SendCommand(command, shutdownCancellationToken) ? SendCommandToInstanceResult.Success : SendCommandToInstanceResult.UnknownError); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async ValueTask DisposeAsync() { | ||||||
|  | 		Logger.Information("Stopping all instances..."); | ||||||
|  | 		 | ||||||
|  | 		shutdownCancellationTokenSource.Cancel(); | ||||||
|  |  | ||||||
|  | 		await semaphore.WaitAsync(CancellationToken.None); | ||||||
|  | 		await Task.WhenAll(instances.Values.Select(static instance => instance.DisposeAsync().AsTask())); | ||||||
|  | 		instances.Clear(); | ||||||
|  | 		 | ||||||
|  | 		shutdownCancellationTokenSource.Dispose(); | ||||||
|  | 		semaphore.Dispose(); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,100 +0,0 @@ | |||||||
| using Phantom.Agent.Rpc; |  | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Agent; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Tasks; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; |  | ||||||
|  |  | ||||||
| sealed class InstanceTicketManager { |  | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceTicketManager>(); |  | ||||||
| 	 |  | ||||||
| 	private readonly AgentInfo agentInfo; |  | ||||||
| 	private readonly ControllerConnection controllerConnection; |  | ||||||
| 	 |  | ||||||
| 	private readonly HashSet<Guid> activeTicketGuids = new (); |  | ||||||
| 	private readonly HashSet<ushort> usedPorts = new (); |  | ||||||
| 	private RamAllocationUnits usedMemory = new (); |  | ||||||
|  |  | ||||||
| 	public InstanceTicketManager(AgentInfo agentInfo, ControllerConnection controllerConnection) { |  | ||||||
| 		this.agentInfo = agentInfo; |  | ||||||
| 		this.controllerConnection = controllerConnection; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public Result<Ticket, LaunchInstanceResult> Reserve(InstanceConfiguration configuration) { |  | ||||||
| 		var memoryAllocation = configuration.MemoryAllocation; |  | ||||||
| 		var serverPort = configuration.ServerPort; |  | ||||||
| 		var rconPort = configuration.RconPort; |  | ||||||
| 		 |  | ||||||
| 		if (!agentInfo.AllowedServerPorts.Contains(serverPort)) { |  | ||||||
| 			return LaunchInstanceResult.ServerPortNotAllowed; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if (!agentInfo.AllowedRconPorts.Contains(rconPort)) { |  | ||||||
| 			return LaunchInstanceResult.RconPortNotAllowed; |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		lock (this) { |  | ||||||
| 			if (activeTicketGuids.Count + 1 > agentInfo.MaxInstances) { |  | ||||||
| 				return LaunchInstanceResult.InstanceLimitExceeded; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if (usedMemory + memoryAllocation > agentInfo.MaxMemory) { |  | ||||||
| 				return LaunchInstanceResult.MemoryLimitExceeded; |  | ||||||
| 			} |  | ||||||
| 			 |  | ||||||
| 			if (usedPorts.Contains(serverPort)) { |  | ||||||
| 				return LaunchInstanceResult.ServerPortAlreadyInUse; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if (usedPorts.Contains(rconPort)) { |  | ||||||
| 				return LaunchInstanceResult.RconPortAlreadyInUse; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			var ticket = new Ticket(Guid.NewGuid(), memoryAllocation, serverPort, rconPort); |  | ||||||
| 			 |  | ||||||
| 			activeTicketGuids.Add(ticket.TicketGuid); |  | ||||||
| 			usedMemory += memoryAllocation; |  | ||||||
| 			usedPorts.Add(serverPort); |  | ||||||
| 			usedPorts.Add(rconPort); |  | ||||||
| 			 |  | ||||||
| 			RefreshAgentStatus(); |  | ||||||
| 			Logger.Debug("Reserved ticket {TicketGuid} (server port {ServerPort}, rcon port {RconPort}, memory allocation {MemoryAllocation} MB).", ticket.TicketGuid, ticket.ServerPort, ticket.RconPort, ticket.MemoryAllocation.InMegabytes); |  | ||||||
|  |  | ||||||
| 			return ticket; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public bool IsValid(Ticket ticket) { |  | ||||||
| 		lock (this) { |  | ||||||
| 			return activeTicketGuids.Contains(ticket.TicketGuid); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public void Release(Ticket ticket) { |  | ||||||
| 		lock (this) { |  | ||||||
| 			if (!activeTicketGuids.Remove(ticket.TicketGuid)) { |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			usedMemory -= ticket.MemoryAllocation; |  | ||||||
| 			usedPorts.Remove(ticket.ServerPort); |  | ||||||
| 			usedPorts.Remove(ticket.RconPort); |  | ||||||
| 			 |  | ||||||
| 			RefreshAgentStatus(); |  | ||||||
| 			Logger.Debug("Released ticket {TicketGuid} (server port {ServerPort}, rcon port {RconPort}, memory allocation {MemoryAllocation} MB).", ticket.TicketGuid, ticket.ServerPort, ticket.RconPort, ticket.MemoryAllocation.InMegabytes); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public void RefreshAgentStatus() { |  | ||||||
| 		lock (this) { |  | ||||||
| 			controllerConnection.Send(new ReportAgentStatusMessage(activeTicketGuids.Count, usedMemory)); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public sealed record Ticket(Guid TicketGuid, RamAllocationUnits MemoryAllocation, ushort ServerPort, ushort RconPort); |  | ||||||
| } |  | ||||||
							
								
								
									
										58
									
								
								Agent/Phantom.Agent.Services/Instances/PortManager.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										58
									
								
								Agent/Phantom.Agent.Services/Instances/PortManager.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,58 @@ | |||||||
|  | using Phantom.Common.Data; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
|  | sealed class PortManager { | ||||||
|  | 	private readonly AllowedPorts allowedServerPorts; | ||||||
|  | 	private readonly AllowedPorts allowedRconPorts; | ||||||
|  | 	private readonly HashSet<ushort> usedPorts = new (); | ||||||
|  |  | ||||||
|  | 	public PortManager(AllowedPorts allowedServerPorts, AllowedPorts allowedRconPorts) { | ||||||
|  | 		this.allowedServerPorts = allowedServerPorts; | ||||||
|  | 		this.allowedRconPorts = allowedRconPorts; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Result Reserve(InstanceConfiguration configuration) { | ||||||
|  | 		var serverPort = configuration.ServerPort; | ||||||
|  | 		var rconPort = configuration.RconPort; | ||||||
|  | 		 | ||||||
|  | 		if (!allowedServerPorts.Contains(serverPort)) { | ||||||
|  | 			return Result.ServerPortNotAllowed; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (!allowedRconPorts.Contains(rconPort)) { | ||||||
|  | 			return Result.RconPortNotAllowed; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		lock (usedPorts) { | ||||||
|  | 			if (usedPorts.Contains(serverPort)) { | ||||||
|  | 				return Result.ServerPortAlreadyInUse; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if (usedPorts.Contains(rconPort)) { | ||||||
|  | 				return Result.RconPortAlreadyInUse; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			usedPorts.Add(serverPort); | ||||||
|  | 			usedPorts.Add(rconPort); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return Result.Success; | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	public void Release(InstanceConfiguration configuration) { | ||||||
|  | 		lock (usedPorts) { | ||||||
|  | 			usedPorts.Remove(configuration.ServerPort); | ||||||
|  | 			usedPorts.Remove(configuration.RconPort); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public enum Result { | ||||||
|  | 		Success, | ||||||
|  | 		ServerPortNotAllowed, | ||||||
|  | 		ServerPortAlreadyInUse, | ||||||
|  | 		RconPortNotAllowed, | ||||||
|  | 		RconPortAlreadyInUse | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -0,0 +1,29 @@ | |||||||
|  | using Phantom.Agent.Services.Backups; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Backups; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.Procedures; | ||||||
|  |  | ||||||
|  | sealed record BackupInstanceProcedure(BackupManager BackupManager) : IInstanceProcedure { | ||||||
|  | 	private readonly TaskCompletionSource<BackupCreationResult> resultCompletionSource = new (); | ||||||
|  |  | ||||||
|  | 	public Task<BackupCreationResult> Result => resultCompletionSource.Task; | ||||||
|  |  | ||||||
|  | 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||||
|  | 		if (context.CurrentState is not InstanceRunningState runningState || runningState.Process.HasEnded) { | ||||||
|  | 			resultCompletionSource.SetResult(new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning)); | ||||||
|  | 			return null; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		try { | ||||||
|  | 			var result = await BackupManager.CreateBackup(context.ShortName, runningState.Process, cancellationToken); | ||||||
|  | 			resultCompletionSource.SetResult(result); | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			resultCompletionSource.SetCanceled(cancellationToken); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			resultCompletionSource.SetException(e); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return null; | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -0,0 +1,7 @@ | |||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.Procedures; | ||||||
|  |  | ||||||
|  | interface IInstanceProcedure { | ||||||
|  | 	Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken); | ||||||
|  | } | ||||||
| @@ -0,0 +1,97 @@ | |||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Agent.Minecraft.Launcher; | ||||||
|  | using Phantom.Agent.Minecraft.Server; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.Procedures; | ||||||
|  |  | ||||||
|  | sealed record LaunchInstanceProcedure(InstanceConfiguration Configuration, IServerLauncher Launcher, bool IsRestarting = false) : IInstanceProcedure { | ||||||
|  | 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||||
|  | 		if (!IsRestarting && context.CurrentState is InstanceRunningState) { | ||||||
|  | 			return null; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		context.SetStatus(IsRestarting ? InstanceStatus.Restarting : InstanceStatus.Launching); | ||||||
|  |  | ||||||
|  | 		InstanceLaunchFailReason? failReason = context.Services.PortManager.Reserve(Configuration) switch { | ||||||
|  | 			PortManager.Result.ServerPortNotAllowed   => InstanceLaunchFailReason.ServerPortNotAllowed, | ||||||
|  | 			PortManager.Result.ServerPortAlreadyInUse => InstanceLaunchFailReason.ServerPortAlreadyInUse, | ||||||
|  | 			PortManager.Result.RconPortNotAllowed     => InstanceLaunchFailReason.RconPortNotAllowed, | ||||||
|  | 			PortManager.Result.RconPortAlreadyInUse   => InstanceLaunchFailReason.RconPortAlreadyInUse, | ||||||
|  | 			_                                         => null | ||||||
|  | 		}; | ||||||
|  |  | ||||||
|  | 		if (failReason is {} reason) { | ||||||
|  | 			context.SetLaunchFailedStatusAndReportEvent(reason); | ||||||
|  | 			return new InstanceNotRunningState(); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		context.Logger.Information("Session starting..."); | ||||||
|  | 		try { | ||||||
|  | 			InstanceProcess process = await DoLaunch(context, cancellationToken); | ||||||
|  | 			return new InstanceRunningState(Configuration, Launcher, process, context); | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			context.SetStatus(InstanceStatus.NotRunning); | ||||||
|  | 		} catch (LaunchFailureException e) { | ||||||
|  | 			context.Logger.Error(e.LogMessage); | ||||||
|  | 			context.SetLaunchFailedStatusAndReportEvent(e.Reason); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			context.Logger.Error(e, "Caught exception while launching instance."); | ||||||
|  | 			context.SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason.UnknownError); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		context.Services.PortManager.Release(Configuration); | ||||||
|  | 		return new InstanceNotRunningState(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<InstanceProcess> DoLaunch(IInstanceContext context, CancellationToken cancellationToken) { | ||||||
|  | 		cancellationToken.ThrowIfCancellationRequested(); | ||||||
|  |  | ||||||
|  | 		byte lastDownloadProgress = byte.MaxValue; | ||||||
|  |  | ||||||
|  | 		void OnDownloadProgress(object? sender, DownloadProgressEventArgs args) { | ||||||
|  | 			byte progress = (byte) Math.Min(args.DownloadedBytes * 100 / args.TotalBytes, 100); | ||||||
|  |  | ||||||
|  | 			if (lastDownloadProgress != progress) { | ||||||
|  | 				lastDownloadProgress = progress; | ||||||
|  | 				context.SetStatus(InstanceStatus.Downloading(progress)); | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		var launchResult = await Launcher.Launch(context.Logger, context.Services.LaunchServices, OnDownloadProgress, cancellationToken); | ||||||
|  | 		if (launchResult is LaunchResult.InvalidJavaRuntime) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.JavaRuntimeNotFound, "Session failed to launch, invalid Java runtime."); | ||||||
|  | 		} | ||||||
|  | 		else if (launchResult is LaunchResult.CouldNotDownloadMinecraftServer) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotDownloadMinecraftServer, "Session failed to launch, could not download Minecraft server."); | ||||||
|  | 		} | ||||||
|  | 		else if (launchResult is LaunchResult.CouldNotPrepareMinecraftServerLauncher) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotPrepareMinecraftServerLauncher, "Session failed to launch, could not prepare Minecraft server launcher."); | ||||||
|  | 		} | ||||||
|  | 		else if (launchResult is LaunchResult.CouldNotConfigureMinecraftServer) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotConfigureMinecraftServer, "Session failed to launch, could not configure Minecraft server."); | ||||||
|  | 		} | ||||||
|  | 		else if (launchResult is LaunchResult.CouldNotStartMinecraftServer) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotStartMinecraftServer, "Session failed to launch, could not start Minecraft server."); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (launchResult is not LaunchResult.Success launchSuccess) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.UnknownError, "Session failed to launch."); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		context.SetStatus(InstanceStatus.Running); | ||||||
|  | 		context.ReportEvent(InstanceEvent.LaunchSucceded); | ||||||
|  | 		return launchSuccess.Process; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private sealed class LaunchFailureException : Exception { | ||||||
|  | 		public InstanceLaunchFailReason Reason { get; } | ||||||
|  | 		public string LogMessage { get; } | ||||||
|  |  | ||||||
|  | 		public LaunchFailureException(InstanceLaunchFailReason reason, string logMessage) { | ||||||
|  | 			this.Reason = reason; | ||||||
|  | 			this.LogMessage = logMessage; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -0,0 +1,17 @@ | |||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.Procedures; | ||||||
|  |  | ||||||
|  | sealed record SetInstanceToNotRunningStateProcedure(IInstanceStatus Status) : IInstanceProcedure { | ||||||
|  | 	public Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||||
|  | 		if (context.CurrentState is InstanceRunningState { Process.HasEnded: true }) { | ||||||
|  | 			context.SetStatus(Status); | ||||||
|  | 			context.ReportEvent(InstanceEvent.Stopped); | ||||||
|  | 			return Task.FromResult<IInstanceState?>(new InstanceNotRunningState()); | ||||||
|  | 		} | ||||||
|  | 		else { | ||||||
|  | 			return Task.FromResult<IInstanceState?>(null); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,46 +1,51 @@ | |||||||
| using System.Diagnostics; | using System.Diagnostics; | ||||||
| using Phantom.Agent.Minecraft.Command; | using Phantom.Agent.Minecraft.Command; | ||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
| using Phantom.Common.Data.Instance; | using Phantom.Common.Data.Instance; | ||||||
| using Phantom.Common.Data.Minecraft; | using Phantom.Common.Data.Minecraft; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Agent.Services.Instances.State; | namespace Phantom.Agent.Services.Instances.Procedures; | ||||||
| 
 | 
 | ||||||
| static class InstanceStopProcedure { | sealed record StopInstanceProcedure(MinecraftStopStrategy StopStrategy) : IInstanceProcedure { | ||||||
| 	private static readonly ushort[] Stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; | 	private static readonly ushort[] Stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; | ||||||
| 
 | 
 | ||||||
| 	public static async Task<bool> Run(InstanceContext context, MinecraftStopStrategy stopStrategy, InstanceRunningState runningState, Action<IInstanceStatus> reportStatus, CancellationToken cancellationToken) { | 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||||
| 		var process = runningState.Process; | 		if (context.CurrentState is not InstanceRunningState runningState) { | ||||||
| 		runningState.IsStopping = true; | 			return null; | ||||||
|  | 		} | ||||||
| 
 | 
 | ||||||
| 		var seconds = stopStrategy.Seconds; | 		var process = runningState.Process; | ||||||
|  | 
 | ||||||
|  | 		runningState.IsStopping = true; | ||||||
|  | 		context.SetStatus(InstanceStatus.Stopping); | ||||||
|  | 
 | ||||||
|  | 		var seconds = StopStrategy.Seconds; | ||||||
| 		if (seconds > 0) { | 		if (seconds > 0) { | ||||||
| 			try { | 			try { | ||||||
| 				await CountDownWithAnnouncements(context, process, seconds, cancellationToken); | 				await CountDownWithAnnouncements(context, process, seconds, cancellationToken); | ||||||
| 			} catch (OperationCanceledException) { | 			} catch (OperationCanceledException) { | ||||||
| 				runningState.IsStopping = false; | 				runningState.IsStopping = false; | ||||||
| 				return false; | 				return null; | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		try { | 		try { | ||||||
| 			// Too late to cancel the stop procedure now. | 			// Too late to cancel the stop procedure now. | ||||||
| 			runningState.OnStopInitiated(); |  | ||||||
| 			 |  | ||||||
| 			if (!process.HasEnded) { | 			if (!process.HasEnded) { | ||||||
| 				context.Logger.Information("Session stopping now."); | 				context.Logger.Information("Session stopping now."); | ||||||
| 				await DoStop(context, process); | 				await DoStop(context, process); | ||||||
| 			} | 			} | ||||||
| 		} finally { | 		} finally { | ||||||
| 			context.Logger.Information("Session stopped."); | 			context.Logger.Information("Session stopped."); | ||||||
| 			reportStatus(InstanceStatus.NotRunning); | 			context.SetStatus(InstanceStatus.NotRunning); | ||||||
| 			context.ReportEvent(InstanceEvent.Stopped); | 			context.ReportEvent(InstanceEvent.Stopped); | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		return true; | 		return new InstanceNotRunningState(); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private static async Task CountDownWithAnnouncements(InstanceContext context, InstanceProcess process, ushort seconds, CancellationToken cancellationToken) { | 	private async Task CountDownWithAnnouncements(IInstanceContext context, InstanceProcess process, ushort seconds, CancellationToken cancellationToken) { | ||||||
| 		context.Logger.Information("Session stopping in {Seconds} seconds.", seconds); | 		context.Logger.Information("Session stopping in {Seconds} seconds.", seconds); | ||||||
| 
 | 
 | ||||||
| 		foreach (var stop in Stops) { | 		foreach (var stop in Stops) { | ||||||
| @@ -61,7 +66,7 @@ static class InstanceStopProcedure { | |||||||
| 		return MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")); | 		return MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private static async Task DoStop(InstanceContext context, InstanceProcess process) { | 	private async Task DoStop(IInstanceContext context, InstanceProcess process) { | ||||||
| 		context.Logger.Information("Sending stop command..."); | 		context.Logger.Information("Sending stop command..."); | ||||||
| 		await TrySendStopCommand(context, process); | 		await TrySendStopCommand(context, process); | ||||||
| 
 | 
 | ||||||
| @@ -69,7 +74,7 @@ static class InstanceStopProcedure { | |||||||
| 		await WaitForSessionToEnd(context, process); | 		await WaitForSessionToEnd(context, process); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private static async Task TrySendStopCommand(InstanceContext context, InstanceProcess process) { | 	private async Task TrySendStopCommand(IInstanceContext context, InstanceProcess process) { | ||||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); | 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); | ||||||
| 		try { | 		try { | ||||||
| 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); | 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); | ||||||
| @@ -77,17 +82,16 @@ static class InstanceStopProcedure { | |||||||
| 			// Ignore. | 			// Ignore. | ||||||
| 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { | 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { | ||||||
| 			// Ignore. | 			// Ignore. | ||||||
| 		} catch (IOException e) when (e.HResult == -2147024664 /* The pipe is being closed */) { |  | ||||||
| 			// Ignore. |  | ||||||
| 		} catch (Exception e) { | 		} catch (Exception e) { | ||||||
| 			context.Logger.Warning(e, "Caught exception while sending stop command."); | 			context.Logger.Warning(e, "Caught exception while sending stop command."); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	private static async Task WaitForSessionToEnd(InstanceContext context, InstanceProcess process) { | 	private async Task WaitForSessionToEnd(IInstanceContext context, InstanceProcess process) { | ||||||
|  | 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(55)); | ||||||
| 		try { | 		try { | ||||||
| 			await process.WaitForExit(TimeSpan.FromSeconds(55)); | 			await process.WaitForExit(timeout.Token); | ||||||
| 		} catch (TimeoutException) { | 		} catch (OperationCanceledException) { | ||||||
| 			try { | 			try { | ||||||
| 				context.Logger.Warning("Waiting timed out, killing session..."); | 				context.Logger.Warning("Waiting timed out, killing session..."); | ||||||
| 				process.Kill(); | 				process.Kill(); | ||||||
| @@ -1,86 +0,0 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; |  | ||||||
| using Phantom.Agent.Minecraft.Launcher; |  | ||||||
| using Phantom.Agent.Minecraft.Server; |  | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances.State; |  | ||||||
|  |  | ||||||
| static class InstanceLaunchProcedure { |  | ||||||
| 	public static async Task<InstanceRunningState?> Run(InstanceContext context, InstanceConfiguration configuration, IServerLauncher launcher, InstanceTicketManager ticketManager, InstanceTicketManager.Ticket ticket, Action<IInstanceStatus> reportStatus, CancellationToken cancellationToken) { |  | ||||||
| 		context.Logger.Information("Session starting..."); |  | ||||||
|  |  | ||||||
| 		Result<InstanceProcess, InstanceLaunchFailReason> result; |  | ||||||
|  |  | ||||||
| 		if (ticketManager.IsValid(ticket)) { |  | ||||||
| 			try { |  | ||||||
| 				result = await LaunchInstance(context, launcher, reportStatus, cancellationToken); |  | ||||||
| 			} catch (OperationCanceledException) { |  | ||||||
| 				reportStatus(InstanceStatus.NotRunning); |  | ||||||
| 				return null; |  | ||||||
| 			} catch (Exception e) { |  | ||||||
| 				context.Logger.Error(e, "Caught exception while launching instance."); |  | ||||||
| 				result = InstanceLaunchFailReason.UnknownError; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			context.Logger.Error("Attempted to launch instance with an invalid ticket!"); |  | ||||||
| 			result = InstanceLaunchFailReason.UnknownError; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if (result) { |  | ||||||
| 			reportStatus(InstanceStatus.Running); |  | ||||||
| 			context.ReportEvent(InstanceEvent.LaunchSucceeded); |  | ||||||
| 			return new InstanceRunningState(context, configuration, launcher, ticket, result.Value, cancellationToken); |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			reportStatus(InstanceStatus.Failed(result.Error)); |  | ||||||
| 			context.ReportEvent(new InstanceLaunchFailedEvent(result.Error)); |  | ||||||
| 			return null; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private static async Task<Result<InstanceProcess, InstanceLaunchFailReason>> LaunchInstance(InstanceContext context, IServerLauncher launcher, Action<IInstanceStatus> reportStatus, CancellationToken cancellationToken) { |  | ||||||
| 		cancellationToken.ThrowIfCancellationRequested(); |  | ||||||
|  |  | ||||||
| 		byte lastDownloadProgress = byte.MaxValue; |  | ||||||
|  |  | ||||||
| 		void OnDownloadProgress(object? sender, DownloadProgressEventArgs args) { |  | ||||||
| 			byte progress = (byte) Math.Min(args.DownloadedBytes * 100 / args.TotalBytes, 100); |  | ||||||
|  |  | ||||||
| 			if (lastDownloadProgress != progress) { |  | ||||||
| 				lastDownloadProgress = progress; |  | ||||||
| 				reportStatus(InstanceStatus.Downloading(progress)); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		switch (await launcher.Launch(context.Logger, context.Services.LaunchServices, OnDownloadProgress, cancellationToken)) { |  | ||||||
| 			case LaunchResult.Success launchSuccess: |  | ||||||
| 				return launchSuccess.Process; |  | ||||||
|  |  | ||||||
| 			case LaunchResult.InvalidJavaRuntime: |  | ||||||
| 				context.Logger.Error("Session failed to launch, invalid Java runtime."); |  | ||||||
| 				return InstanceLaunchFailReason.JavaRuntimeNotFound; |  | ||||||
|  |  | ||||||
| 			case LaunchResult.CouldNotDownloadMinecraftServer: |  | ||||||
| 				context.Logger.Error("Session failed to launch, could not download Minecraft server."); |  | ||||||
| 				return InstanceLaunchFailReason.CouldNotDownloadMinecraftServer; |  | ||||||
|  |  | ||||||
| 			case LaunchResult.CouldNotPrepareMinecraftServerLauncher: |  | ||||||
| 				context.Logger.Error("Session failed to launch, could not prepare Minecraft server launcher."); |  | ||||||
| 				return InstanceLaunchFailReason.CouldNotPrepareMinecraftServerLauncher; |  | ||||||
|  |  | ||||||
| 			case LaunchResult.CouldNotConfigureMinecraftServer: |  | ||||||
| 				context.Logger.Error("Session failed to launch, could not configure Minecraft server."); |  | ||||||
| 				return InstanceLaunchFailReason.CouldNotConfigureMinecraftServer; |  | ||||||
|  |  | ||||||
| 			case LaunchResult.CouldNotStartMinecraftServer: |  | ||||||
| 				context.Logger.Error("Session failed to launch, could not start Minecraft server."); |  | ||||||
| 				return InstanceLaunchFailReason.CouldNotStartMinecraftServer; |  | ||||||
|  |  | ||||||
| 			default: |  | ||||||
| 				context.Logger.Error("Session failed to launch."); |  | ||||||
| 				return InstanceLaunchFailReason.UnknownError; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,140 +0,0 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; |  | ||||||
| using Phantom.Agent.Minecraft.Server; |  | ||||||
| using Phantom.Agent.Rpc; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Tasks; |  | ||||||
| using Phantom.Utils.Threading; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances.State; |  | ||||||
|  |  | ||||||
| sealed class InstancePlayerCountTracker : CancellableBackgroundTask { |  | ||||||
| 	private readonly ControllerConnection controllerConnection; |  | ||||||
| 	private readonly Guid instanceGuid; |  | ||||||
| 	private readonly ushort serverPort; |  | ||||||
| 	private readonly InstanceProcess process; |  | ||||||
|  |  | ||||||
| 	private readonly TaskCompletionSource firstDetection = AsyncTasks.CreateCompletionSource(); |  | ||||||
| 	private readonly ManualResetEventSlim serverOutputEvent = new (); |  | ||||||
|  |  | ||||||
| 	private InstancePlayerCounts? playerCounts; |  | ||||||
|  |  | ||||||
| 	public InstancePlayerCounts? PlayerCounts { |  | ||||||
| 		get { |  | ||||||
| 			lock (this) { |  | ||||||
| 				return playerCounts; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		private set { |  | ||||||
| 			EventHandler<int?>? onlinePlayerCountChanged; |  | ||||||
| 			lock (this) { |  | ||||||
| 				if (playerCounts == value) { |  | ||||||
| 					return; |  | ||||||
| 				} |  | ||||||
| 				 |  | ||||||
| 				playerCounts = value; |  | ||||||
| 				onlinePlayerCountChanged = OnlinePlayerCountChanged; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			onlinePlayerCountChanged?.Invoke(this, value?.Online); |  | ||||||
| 			controllerConnection.Send(new ReportInstancePlayerCountsMessage(instanceGuid, value)); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private event EventHandler<int?>? OnlinePlayerCountChanged; |  | ||||||
|  |  | ||||||
| 	private bool isDisposed = false; |  | ||||||
|  |  | ||||||
| 	public InstancePlayerCountTracker(InstanceContext context, InstanceProcess process, ushort serverPort) : base(PhantomLogger.Create<InstancePlayerCountTracker>(context.ShortName)) { |  | ||||||
| 		this.controllerConnection = context.Services.ControllerConnection; |  | ||||||
| 		this.instanceGuid = context.InstanceGuid; |  | ||||||
| 		this.process = process; |  | ||||||
| 		this.serverPort = serverPort; |  | ||||||
| 		Start(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	protected override async Task RunTask() { |  | ||||||
| 		// Give the server time to start accepting connections. |  | ||||||
| 		await Task.Delay(TimeSpan.FromSeconds(10), CancellationToken); |  | ||||||
|  |  | ||||||
| 		serverOutputEvent.Set(); |  | ||||||
| 		process.AddOutputListener(OnOutput, maxLinesToReadFromHistory: 0); |  | ||||||
| 		 |  | ||||||
| 		while (!CancellationToken.IsCancellationRequested) { |  | ||||||
| 			serverOutputEvent.Reset(); |  | ||||||
|  |  | ||||||
| 			PlayerCounts = await TryGetPlayerCounts(); |  | ||||||
| 			 |  | ||||||
| 			if (!firstDetection.Task.IsCompleted) { |  | ||||||
| 				firstDetection.SetResult(); |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			await Task.Delay(TimeSpan.FromSeconds(10), CancellationToken); |  | ||||||
| 			await serverOutputEvent.WaitHandle.WaitOneAsync(CancellationToken); |  | ||||||
| 			await Task.Delay(TimeSpan.FromSeconds(1), CancellationToken); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<InstancePlayerCounts?> TryGetPlayerCounts() { |  | ||||||
| 		try { |  | ||||||
| 			var result = await ServerStatusProtocol.GetPlayerCounts(serverPort, CancellationToken); |  | ||||||
| 			Logger.Debug("Detected {OnlinePlayerCount} / {MaximumPlayerCount} online player(s).", result.Online, result.Maximum); |  | ||||||
| 			return result; |  | ||||||
| 		} catch (ServerStatusProtocol.ProtocolException e) { |  | ||||||
| 			Logger.Error(e.Message); |  | ||||||
| 			return null; |  | ||||||
| 		} catch (Exception e) { |  | ||||||
| 			Logger.Error(e, "Caught exception while checking online player count."); |  | ||||||
| 			return null; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public async Task WaitForOnlinePlayers(CancellationToken cancellationToken) { |  | ||||||
| 		await firstDetection.Task.WaitAsync(cancellationToken); |  | ||||||
|  |  | ||||||
| 		var onlinePlayersDetected = AsyncTasks.CreateCompletionSource(); |  | ||||||
|  |  | ||||||
| 		lock (this) { |  | ||||||
| 			if (playerCounts is { Online: > 0 }) { |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
| 			else if (playerCounts == null) { |  | ||||||
| 				throw new InvalidOperationException(); |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			OnlinePlayerCountChanged += OnOnlinePlayerCountChanged; |  | ||||||
|  |  | ||||||
| 			void OnOnlinePlayerCountChanged(object? sender, int? newPlayerCount) { |  | ||||||
| 				if (newPlayerCount == null) { |  | ||||||
| 					onlinePlayersDetected.TrySetException(new InvalidOperationException()); |  | ||||||
| 					OnlinePlayerCountChanged -= OnOnlinePlayerCountChanged; |  | ||||||
| 				} |  | ||||||
| 				else if (newPlayerCount > 0) { |  | ||||||
| 					onlinePlayersDetected.TrySetResult(); |  | ||||||
| 					OnlinePlayerCountChanged -= OnOnlinePlayerCountChanged; |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		await onlinePlayersDetected.Task; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void OnOutput(object? sender, string? line) { |  | ||||||
| 		lock (this) { |  | ||||||
| 			if (!isDisposed) { |  | ||||||
| 				serverOutputEvent.Set(); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	protected override void Dispose() { |  | ||||||
| 		lock (this) { |  | ||||||
| 			isDisposed = true; |  | ||||||
| 			playerCounts = null; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		process.RemoveOutputListener(OnOutput); |  | ||||||
| 		serverOutputEvent.Dispose(); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -0,0 +1,6 @@ | |||||||
|  | namespace Phantom.Agent.Services.Instances.States; | ||||||
|  |  | ||||||
|  | interface IInstanceState { | ||||||
|  | 	void Initialize(); | ||||||
|  | 	Task<bool> SendCommand(string command, CancellationToken cancellationToken); | ||||||
|  | } | ||||||
| @@ -0,0 +1,9 @@ | |||||||
|  | namespace Phantom.Agent.Services.Instances.States; | ||||||
|  |  | ||||||
|  | sealed class InstanceNotRunningState : IInstanceState { | ||||||
|  | 	public void Initialize() {} | ||||||
|  |  | ||||||
|  | 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
|  | 		return Task.FromResult(false); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,41 +1,35 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Agent.Minecraft.Launcher; | using Phantom.Agent.Minecraft.Launcher; | ||||||
| using Phantom.Agent.Services.Backups; | using Phantom.Agent.Services.Backups; | ||||||
|  | using Phantom.Agent.Services.Instances.Procedures; | ||||||
| using Phantom.Common.Data.Backups; | using Phantom.Common.Data.Backups; | ||||||
| using Phantom.Common.Data.Instance; | using Phantom.Common.Data.Instance; | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Agent.Services.Instances.State; | namespace Phantom.Agent.Services.Instances.States; | ||||||
| 
 | 
 | ||||||
| sealed class InstanceRunningState : IDisposable { | sealed class InstanceRunningState : IInstanceState, IDisposable { | ||||||
| 	public InstanceTicketManager.Ticket Ticket { get; } |  | ||||||
| 	public InstanceProcess Process { get; } | 	public InstanceProcess Process { get; } | ||||||
| 
 | 
 | ||||||
| 	internal bool IsStopping { get; set; } | 	internal bool IsStopping { get; set; } | ||||||
| 
 | 
 | ||||||
| 	private readonly InstanceContext context; |  | ||||||
| 	private readonly InstanceConfiguration configuration; | 	private readonly InstanceConfiguration configuration; | ||||||
| 	private readonly IServerLauncher launcher; | 	private readonly IServerLauncher launcher; | ||||||
| 	private readonly CancellationToken cancellationToken; | 	private readonly IInstanceContext context; | ||||||
| 
 | 
 | ||||||
| 	private readonly InstanceLogSender logSender; | 	private readonly InstanceLogSender logSender; | ||||||
| 	private readonly InstancePlayerCountTracker playerCountTracker; |  | ||||||
| 	private readonly BackupScheduler backupScheduler; | 	private readonly BackupScheduler backupScheduler; | ||||||
| 
 | 
 | ||||||
| 	private bool isDisposed; | 	private bool isDisposed; | ||||||
| 
 | 
 | ||||||
| 	public InstanceRunningState(InstanceContext context, InstanceConfiguration configuration, IServerLauncher launcher, InstanceTicketManager.Ticket ticket, InstanceProcess process, CancellationToken cancellationToken) { | 	public InstanceRunningState(InstanceConfiguration configuration, IServerLauncher launcher, InstanceProcess process, IInstanceContext context) { | ||||||
| 		this.context = context; |  | ||||||
| 		this.configuration = configuration; | 		this.configuration = configuration; | ||||||
| 		this.launcher = launcher; | 		this.launcher = launcher; | ||||||
| 		this.Ticket = ticket; | 		this.context = context; | ||||||
| 		this.Process = process; | 		this.Process = process; | ||||||
| 		this.cancellationToken = cancellationToken; |  | ||||||
| 
 | 
 | ||||||
| 		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.InstanceGuid, context.ShortName); | 		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.Services.TaskManager, configuration.InstanceGuid, context.ShortName); | ||||||
| 		this.playerCountTracker = new InstancePlayerCountTracker(context, process, configuration.ServerPort); |  | ||||||
| 
 | 
 | ||||||
| 		this.backupScheduler = new BackupScheduler(context, playerCountTracker); | 		this.backupScheduler = new BackupScheduler(context.Services.TaskManager, context.Services.BackupManager, process, context, configuration.ServerPort); | ||||||
| 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted; | 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| @@ -45,7 +39,7 @@ sealed class InstanceRunningState : IDisposable { | |||||||
| 		if (Process.HasEnded) { | 		if (Process.HasEnded) { | ||||||
| 			if (TryDispose()) { | 			if (TryDispose()) { | ||||||
| 				context.Logger.Warning("Session ended immediately after it was started."); | 				context.Logger.Warning("Session ended immediately after it was started."); | ||||||
| 				context.Actor.Tell(new InstanceActor.HandleProcessEndedCommand(InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError))); | 				context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError)), immediate: true); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		else { | 		else { | ||||||
| @@ -64,17 +58,13 @@ sealed class InstanceRunningState : IDisposable { | |||||||
| 			return; | 			return; | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		if (cancellationToken.IsCancellationRequested) { |  | ||||||
| 			return; |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		if (IsStopping) { | 		if (IsStopping) { | ||||||
| 			context.Actor.Tell(new InstanceActor.HandleProcessEndedCommand(InstanceStatus.NotRunning)); | 			context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.NotRunning), immediate: true); | ||||||
| 		} | 		} | ||||||
| 		else { | 		else { | ||||||
| 			context.Logger.Information("Session ended unexpectedly, restarting..."); | 			context.Logger.Information("Session ended unexpectedly, restarting..."); | ||||||
| 			context.ReportEvent(InstanceEvent.Crashed); | 			context.ReportEvent(InstanceEvent.Crashed); | ||||||
| 			context.Actor.Tell(new InstanceActor.LaunchInstanceCommand(configuration, launcher, Ticket, IsRestarting: true)); | 			context.EnqueueProcedure(new LaunchInstanceProcedure(configuration, launcher, IsRestarting: true)); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| @@ -82,24 +72,19 @@ sealed class InstanceRunningState : IDisposable { | |||||||
| 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	public async Task<SendCommandToInstanceResult> SendCommand(string command, CancellationToken cancellationToken) { | 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
| 		try { | 		try { | ||||||
| 			context.Logger.Information("Sending command: {Command}", command); | 			context.Logger.Information("Sending command: {Command}", command); | ||||||
| 			await Process.SendCommand(command, cancellationToken); | 			await Process.SendCommand(command, cancellationToken); | ||||||
| 			return SendCommandToInstanceResult.Success; | 			return true; | ||||||
| 		} catch (OperationCanceledException) { | 		} catch (OperationCanceledException) { | ||||||
| 			return SendCommandToInstanceResult.UnknownError; | 			return false; | ||||||
| 		} catch (Exception e) { | 		} catch (Exception e) { | ||||||
| 			context.Logger.Warning(e, "Caught exception while sending command."); | 			context.Logger.Warning(e, "Caught exception while sending command."); | ||||||
| 			return SendCommandToInstanceResult.UnknownError; | 			return false; | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	public void OnStopInitiated() { |  | ||||||
| 		backupScheduler.Stop(); |  | ||||||
| 		playerCountTracker.Stop(); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private bool TryDispose() { | 	private bool TryDispose() { | ||||||
| 		lock (this) { | 		lock (this) { | ||||||
| 			if (isDisposed) { | 			if (isDisposed) { | ||||||
| @@ -109,10 +94,11 @@ sealed class InstanceRunningState : IDisposable { | |||||||
| 			isDisposed = true; | 			isDisposed = true; | ||||||
| 		} | 		} | ||||||
| 
 | 
 | ||||||
| 		OnStopInitiated(); |  | ||||||
| 		logSender.Stop(); | 		logSender.Stop(); | ||||||
|  | 		backupScheduler.Stop(); | ||||||
| 		 | 		 | ||||||
| 		Process.Dispose(); | 		Process.Dispose(); | ||||||
|  | 		context.Services.PortManager.Release(configuration); | ||||||
| 		 | 		 | ||||||
| 		return true; | 		return true; | ||||||
| 	} | 	} | ||||||
| @@ -1,101 +0,0 @@ | |||||||
| using Phantom.Agent.Services.Instances; |  | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
| using Phantom.Common.Messages.Agent; |  | ||||||
| using Phantom.Common.Messages.Agent.BiDirectional; |  | ||||||
| using Phantom.Common.Messages.Agent.ToAgent; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Actor; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Rpc.Runtime; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Rpc; |  | ||||||
|  |  | ||||||
| public sealed class ControllerMessageHandlerActor : ReceiveActor<IMessageToAgent> { |  | ||||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create<ControllerMessageHandlerActor>(); |  | ||||||
|  |  | ||||||
| 	public readonly record struct Init(RpcConnectionToServer<IMessageToController> Connection, AgentServices Agent, CancellationTokenSource ShutdownTokenSource); |  | ||||||
| 	 |  | ||||||
| 	public static Props<IMessageToAgent> Factory(Init init) { |  | ||||||
| 		return Props<IMessageToAgent>.Create(() => new ControllerMessageHandlerActor(init), new ActorConfiguration { SupervisorStrategy = SupervisorStrategies.Resume }); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToController> connection; |  | ||||||
| 	private readonly AgentServices agent; |  | ||||||
| 	private readonly CancellationTokenSource shutdownTokenSource; |  | ||||||
|  |  | ||||||
| 	private ControllerMessageHandlerActor(Init init) { |  | ||||||
| 		this.connection = init.Connection; |  | ||||||
| 		this.agent = init.Agent; |  | ||||||
| 		this.shutdownTokenSource = init.ShutdownTokenSource; |  | ||||||
| 		 |  | ||||||
| 		ReceiveAsync<RegisterAgentSuccessMessage>(HandleRegisterAgentSuccess); |  | ||||||
| 		Receive<RegisterAgentFailureMessage>(HandleRegisterAgentFailure); |  | ||||||
| 		ReceiveAndReplyLater<ConfigureInstanceMessage, Result<ConfigureInstanceResult, InstanceActionFailure>>(HandleConfigureInstance); |  | ||||||
| 		ReceiveAndReplyLater<LaunchInstanceMessage, Result<LaunchInstanceResult, InstanceActionFailure>>(HandleLaunchInstance); |  | ||||||
| 		ReceiveAndReplyLater<StopInstanceMessage, Result<StopInstanceResult, InstanceActionFailure>>(HandleStopInstance); |  | ||||||
| 		ReceiveAndReplyLater<SendCommandToInstanceMessage, Result<SendCommandToInstanceResult, InstanceActionFailure>>(HandleSendCommandToInstance); |  | ||||||
| 		Receive<ReplyMessage>(HandleReply); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task HandleRegisterAgentSuccess(RegisterAgentSuccessMessage message) { |  | ||||||
| 		Logger.Information("Agent authentication successful."); |  | ||||||
|  |  | ||||||
| 		void ShutdownAfterConfigurationFailed(Guid instanceGuid, InstanceConfiguration configuration) { |  | ||||||
| 			Logger.Fatal("Unable to configure instance \"{Name}\" (GUID {Guid}), shutting down.", configuration.InstanceName, instanceGuid); |  | ||||||
| 			shutdownTokenSource.Cancel(); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		foreach (var configureInstanceMessage in message.InitialInstanceConfigurations) { |  | ||||||
| 			var result = await HandleConfigureInstance(configureInstanceMessage, alwaysReportStatus: true); |  | ||||||
| 			if (!result.Is(ConfigureInstanceResult.Success)) { |  | ||||||
| 				ShutdownAfterConfigurationFailed(configureInstanceMessage.InstanceGuid, configureInstanceMessage.Configuration); |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		connection.SetIsReady(); |  | ||||||
| 		 |  | ||||||
| 		await connection.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); |  | ||||||
| 		agent.InstanceTicketManager.RefreshAgentStatus(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void HandleRegisterAgentFailure(RegisterAgentFailureMessage message) { |  | ||||||
| 		string errorMessage = message.FailureKind switch { |  | ||||||
| 			RegisterAgentFailure.ConnectionAlreadyHasAnAgent => "This connection already has an associated agent.", |  | ||||||
| 			RegisterAgentFailure.InvalidToken                => "Invalid token.", |  | ||||||
| 			_                                                => "Unknown error " + (byte) message.FailureKind + "." |  | ||||||
| 		}; |  | ||||||
|  |  | ||||||
| 		Logger.Fatal("Agent authentication failed: {Error}", errorMessage); |  | ||||||
| 		 |  | ||||||
| 		PhantomLogger.Dispose(); |  | ||||||
| 		Environment.Exit(1); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private Task<Result<ConfigureInstanceResult, InstanceActionFailure>> HandleConfigureInstance(ConfigureInstanceMessage message, bool alwaysReportStatus) { |  | ||||||
| 		return agent.InstanceManager.Request(new InstanceManagerActor.ConfigureInstanceCommand(message.InstanceGuid, message.Configuration, message.LaunchProperties, message.LaunchNow, alwaysReportStatus)); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	private async Task<Result<ConfigureInstanceResult, InstanceActionFailure>> HandleConfigureInstance(ConfigureInstanceMessage message) { |  | ||||||
| 		return await HandleConfigureInstance(message, alwaysReportStatus: false); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<Result<LaunchInstanceResult, InstanceActionFailure>> HandleLaunchInstance(LaunchInstanceMessage message) { |  | ||||||
| 		return await agent.InstanceManager.Request(new InstanceManagerActor.LaunchInstanceCommand(message.InstanceGuid)); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<Result<StopInstanceResult, InstanceActionFailure>> HandleStopInstance(StopInstanceMessage message) { |  | ||||||
| 		return await agent.InstanceManager.Request(new InstanceManagerActor.StopInstanceCommand(message.InstanceGuid, message.StopStrategy)); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task<Result<SendCommandToInstanceResult, InstanceActionFailure>> HandleSendCommandToInstance(SendCommandToInstanceMessage message) { |  | ||||||
| 		return await agent.InstanceManager.Request(new InstanceManagerActor.SendCommandToInstanceCommand(message.InstanceGuid, message.Command)); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void HandleReply(ReplyMessage message) { |  | ||||||
| 		connection.Receive(message); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										88
									
								
								Agent/Phantom.Agent.Services/Rpc/MessageListener.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										88
									
								
								Agent/Phantom.Agent.Services/Rpc/MessageListener.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,88 @@ | |||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  | using Phantom.Common.Logging; | ||||||
|  | using Phantom.Common.Messages.Agent; | ||||||
|  | using Phantom.Common.Messages.Agent.BiDirectional; | ||||||
|  | using Phantom.Common.Messages.Agent.ToAgent; | ||||||
|  | using Phantom.Common.Messages.Agent.ToController; | ||||||
|  | using Phantom.Utils.Rpc; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Rpc; | ||||||
|  |  | ||||||
|  | public sealed class MessageListener : IMessageToAgentListener { | ||||||
|  | 	private static ILogger Logger { get; } = PhantomLogger.Create<MessageListener>(); | ||||||
|  |  | ||||||
|  | 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||||
|  | 	private readonly AgentServices agent; | ||||||
|  | 	private readonly CancellationTokenSource shutdownTokenSource; | ||||||
|  |  | ||||||
|  | 	public MessageListener(RpcConnectionToServer<IMessageToControllerListener> connection, AgentServices agent, CancellationTokenSource shutdownTokenSource) { | ||||||
|  | 		this.connection = connection; | ||||||
|  | 		this.agent = agent; | ||||||
|  | 		this.shutdownTokenSource = shutdownTokenSource; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<NoReply> HandleRegisterAgentSuccess(RegisterAgentSuccessMessage message) { | ||||||
|  | 		Logger.Information("Agent authentication successful."); | ||||||
|  |  | ||||||
|  | 		void ShutdownAfterConfigurationFailed(InstanceConfiguration configuration) { | ||||||
|  | 			Logger.Fatal("Unable to configure instance \"{Name}\" (GUID {Guid}), shutting down.", configuration.InstanceName, configuration.InstanceGuid); | ||||||
|  | 			shutdownTokenSource.Cancel(); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		foreach (var configureInstanceMessage in message.InitialInstanceConfigurations) { | ||||||
|  | 			var result = await HandleConfigureInstance(configureInstanceMessage, alwaysReportStatus: true); | ||||||
|  | 			if (!result.Is(ConfigureInstanceResult.Success)) { | ||||||
|  | 				ShutdownAfterConfigurationFailed(configureInstanceMessage.Configuration); | ||||||
|  | 				return NoReply.Instance; | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		await connection.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); | ||||||
|  | 		await agent.InstanceSessionManager.RefreshAgentStatus(); | ||||||
|  | 		 | ||||||
|  | 		return NoReply.Instance; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<NoReply> HandleRegisterAgentFailure(RegisterAgentFailureMessage message) { | ||||||
|  | 		string errorMessage = message.FailureKind switch { | ||||||
|  | 			RegisterAgentFailure.ConnectionAlreadyHasAnAgent => "This connection already has an associated agent.", | ||||||
|  | 			RegisterAgentFailure.InvalidToken                => "Invalid token.", | ||||||
|  | 			_                                                => "Unknown error " + (byte) message.FailureKind + "." | ||||||
|  | 		}; | ||||||
|  |  | ||||||
|  | 		Logger.Fatal("Agent authentication failed: {Error}", errorMessage); | ||||||
|  | 		 | ||||||
|  | 		PhantomLogger.Dispose(); | ||||||
|  | 		Environment.Exit(1); | ||||||
|  |  | ||||||
|  | 		return Task.FromResult(NoReply.Instance); | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	private Task<InstanceActionResult<ConfigureInstanceResult>> HandleConfigureInstance(ConfigureInstanceMessage message, bool alwaysReportStatus) { | ||||||
|  | 		return agent.InstanceSessionManager.Configure(message.Configuration, message.LaunchProperties, message.LaunchNow, alwaysReportStatus); | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	public async Task<InstanceActionResult<ConfigureInstanceResult>> HandleConfigureInstance(ConfigureInstanceMessage message) { | ||||||
|  | 		return await HandleConfigureInstance(message, alwaysReportStatus: false); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<InstanceActionResult<LaunchInstanceResult>> HandleLaunchInstance(LaunchInstanceMessage message) { | ||||||
|  | 		return await agent.InstanceSessionManager.Launch(message.InstanceGuid); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<InstanceActionResult<StopInstanceResult>> HandleStopInstance(StopInstanceMessage message) { | ||||||
|  | 		return await agent.InstanceSessionManager.Stop(message.InstanceGuid, message.StopStrategy); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public async Task<InstanceActionResult<SendCommandToInstanceResult>> HandleSendCommandToInstance(SendCommandToInstanceMessage message) { | ||||||
|  | 		return await agent.InstanceSessionManager.SendCommand(message.InstanceGuid, message.Command); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<NoReply> HandleReply(ReplyMessage message) { | ||||||
|  | 		connection.Receive(message); | ||||||
|  | 		return Task.FromResult(NoReply.Instance); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,8 +1,8 @@ | |||||||
| using NetMQ; | using NetMQ; | ||||||
| using Phantom.Common.Data; | using Phantom.Common.Data; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Cryptography; | using Phantom.Utils.Cryptography; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent; | namespace Phantom.Agent; | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| using System.Text; | using System.Text; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent; | namespace Phantom.Agent; | ||||||
|   | |||||||
| @@ -5,26 +5,25 @@ using Phantom.Agent.Rpc; | |||||||
| using Phantom.Agent.Services; | using Phantom.Agent.Services; | ||||||
| using Phantom.Agent.Services.Rpc; | using Phantom.Agent.Services.Rpc; | ||||||
| using Phantom.Common.Data.Agent; | using Phantom.Common.Data.Agent; | ||||||
|  | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Messages.Agent; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.Agent.ToController; | ||||||
| using Phantom.Utils.Actor; |  | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Rpc; | using Phantom.Utils.Rpc; | ||||||
| using Phantom.Utils.Rpc.Sockets; | using Phantom.Utils.Rpc.Sockets; | ||||||
| using Phantom.Utils.Runtime; | using Phantom.Utils.Runtime; | ||||||
|  | using Phantom.Utils.Tasks; | ||||||
|  |  | ||||||
| const int ProtocolVersion = 1; | const int ProtocolVersion = 1; | ||||||
|  |  | ||||||
| var shutdownCancellationTokenSource = new CancellationTokenSource(); | var shutdownCancellationTokenSource = new CancellationTokenSource(); | ||||||
| var shutdownCancellationToken = shutdownCancellationTokenSource.Token; | var shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||||
|  |  | ||||||
| ProgramCulture.UseInvariantCulture(); |  | ||||||
| ThreadPool.SetMinThreads(workerThreads: 2, completionPortThreads: 1); |  | ||||||
|  |  | ||||||
| PosixSignals.RegisterCancellation(shutdownCancellationTokenSource, static () => { | PosixSignals.RegisterCancellation(shutdownCancellationTokenSource, static () => { | ||||||
| 	PhantomLogger.Root.InformationHeading("Stopping Phantom Panel agent..."); | 	PhantomLogger.Root.InformationHeading("Stopping Phantom Panel agent..."); | ||||||
| }); | }); | ||||||
|  |  | ||||||
|  | ThreadPool.SetMinThreads(workerThreads: 2, completionPortThreads: 1); | ||||||
|  |  | ||||||
| try { | try { | ||||||
| 	var fullVersion = AssemblyAttributes.GetFullVersion(Assembly.GetExecutingAssembly()); | 	var fullVersion = AssemblyAttributes.GetFullVersion(Assembly.GetExecutingAssembly()); | ||||||
|  |  | ||||||
| @@ -53,17 +52,15 @@ try { | |||||||
| 	 | 	 | ||||||
| 	PhantomLogger.Root.InformationHeading("Launching Phantom Panel agent..."); | 	PhantomLogger.Root.InformationHeading("Launching Phantom Panel agent..."); | ||||||
| 	 | 	 | ||||||
| 	var rpcConfiguration = new RpcConfiguration("Agent", controllerHost, controllerPort, controllerCertificate); | 	var rpcConfiguration = new RpcConfiguration(PhantomLogger.Create("Rpc"), PhantomLogger.Create<TaskManager>("Rpc"), controllerHost, controllerPort, controllerCertificate); | ||||||
| 	var rpcSocket = RpcClientSocket.Connect(rpcConfiguration, AgentMessageRegistries.Definitions, new RegisterAgentMessage(agentToken, agentInfo)); | 	var rpcSocket = RpcClientSocket.Connect(rpcConfiguration, AgentMessageRegistries.Definitions, new RegisterAgentMessage(agentToken, agentInfo)); | ||||||
|  |  | ||||||
| 	var agentServices = new AgentServices(agentInfo, folders, new AgentServiceConfiguration(maxConcurrentBackupCompressionTasks), new ControllerConnection(rpcSocket.Connection)); | 	var agentServices = new AgentServices(agentInfo, folders, new AgentServiceConfiguration(maxConcurrentBackupCompressionTasks), new ControllerConnection(rpcSocket.Connection)); | ||||||
| 	await agentServices.Initialize(); | 	await agentServices.Initialize(); | ||||||
|  |  | ||||||
| 	var rpcMessageHandlerInit = new ControllerMessageHandlerActor.Init(rpcSocket.Connection, agentServices, shutdownCancellationTokenSource); |  | ||||||
| 	var rpcMessageHandlerActor = agentServices.ActorSystem.ActorOf(ControllerMessageHandlerActor.Factory(rpcMessageHandlerInit), "ControllerMessageHandler"); |  | ||||||
| 	 |  | ||||||
| 	var rpcDisconnectSemaphore = new SemaphoreSlim(0, 1); | 	var rpcDisconnectSemaphore = new SemaphoreSlim(0, 1); | ||||||
| 	var rpcTask = RpcClientRuntime.Launch(rpcSocket, rpcMessageHandlerActor, rpcDisconnectSemaphore, shutdownCancellationToken); | 	var rpcMessageListener = new MessageListener(rpcSocket.Connection, agentServices, shutdownCancellationTokenSource); | ||||||
|  | 	var rpcTask = RpcClientRuntime.Launch(rpcSocket, agentInfo, rpcMessageListener, rpcDisconnectSemaphore, shutdownCancellationToken); | ||||||
| 	try { | 	try { | ||||||
| 		await rpcTask.WaitAsync(shutdownCancellationToken); | 		await rpcTask.WaitAsync(shutdownCancellationToken); | ||||||
| 	} finally { | 	} finally { | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| using Phantom.Agent.Minecraft.Java; | using Phantom.Agent.Minecraft.Java; | ||||||
| using Phantom.Common.Data; | using Phantom.Common.Data; | ||||||
| using Phantom.Utils.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Runtime; | using Phantom.Utils.Runtime; | ||||||
|  |  | ||||||
| namespace Phantom.Agent; | namespace Phantom.Agent; | ||||||
|   | |||||||
| @@ -1,15 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Agent; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Agent; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record Agent( |  | ||||||
| 	[property: MemoryPackOrder(0)] Guid AgentGuid, |  | ||||||
| 	[property: MemoryPackOrder(1)] AgentConfiguration Configuration, |  | ||||||
| 	[property: MemoryPackOrder(2)] AgentStats? Stats, |  | ||||||
| 	[property: MemoryPackOrder(3)] IAgentConnectionStatus ConnectionStatus |  | ||||||
| ) { |  | ||||||
| 	[MemoryPackIgnore] |  | ||||||
| 	public RamAllocationUnits? AvailableMemory => Configuration.MaxMemory - Stats?.RunningInstanceMemory; |  | ||||||
| } |  | ||||||
| @@ -1,19 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Agent; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Agent; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AgentConfiguration( |  | ||||||
| 	[property: MemoryPackOrder(0)] string AgentName, |  | ||||||
| 	[property: MemoryPackOrder(1)] ushort ProtocolVersion, |  | ||||||
| 	[property: MemoryPackOrder(2)] string BuildVersion, |  | ||||||
| 	[property: MemoryPackOrder(3)] ushort MaxInstances, |  | ||||||
| 	[property: MemoryPackOrder(4)] RamAllocationUnits MaxMemory, |  | ||||||
| 	[property: MemoryPackOrder(5)] AllowedPorts? AllowedServerPorts = null, |  | ||||||
| 	[property: MemoryPackOrder(6)] AllowedPorts? AllowedRconPorts = null |  | ||||||
| ) { |  | ||||||
| 	public static AgentConfiguration From(AgentInfo agentInfo) { |  | ||||||
| 		return new AgentConfiguration(agentInfo.AgentName, agentInfo.ProtocolVersion, agentInfo.BuildVersion, agentInfo.MaxInstances, agentInfo.MaxMemory, agentInfo.AllowedServerPorts, agentInfo.AllowedRconPorts); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,27 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Agent; |  | ||||||
|  |  | ||||||
| [MemoryPackable] |  | ||||||
| [MemoryPackUnion(0, typeof(AgentIsOffline))] |  | ||||||
| [MemoryPackUnion(1, typeof(AgentIsDisconnected))] |  | ||||||
| [MemoryPackUnion(2, typeof(AgentIsOnline))] |  | ||||||
| public partial interface IAgentConnectionStatus {} |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AgentIsOffline : IAgentConnectionStatus; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AgentIsDisconnected([property: MemoryPackOrder(0)] DateTimeOffset LastPingTime) : IAgentConnectionStatus; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AgentIsOnline : IAgentConnectionStatus; |  | ||||||
|  |  | ||||||
| public static class AgentConnectionStatus { |  | ||||||
| 	public static readonly IAgentConnectionStatus Offline = new AgentIsOffline(); |  | ||||||
| 	public static readonly IAgentConnectionStatus Online = new AgentIsOnline(); |  | ||||||
|  |  | ||||||
| 	public static IAgentConnectionStatus Disconnected(DateTimeOffset lastPingTime) { |  | ||||||
| 		return new AgentIsDisconnected(lastPingTime); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,14 +1,5 @@ | |||||||
| using MemoryPack; | using System.Text.Json; | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.AuditLog; | namespace Phantom.Common.Data.Web.AuditLog; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | public sealed record AuditLogItem(DateTime UtcTime, Guid? UserGuid, string? UserName, AuditLogEventType EventType, AuditLogSubjectType SubjectType, string? SubjectId, JsonDocument? Data); | ||||||
| public sealed partial record AuditLogItem( |  | ||||||
| 	[property: MemoryPackOrder(0)] DateTime UtcTime, |  | ||||||
| 	[property: MemoryPackOrder(1)] Guid? UserGuid, |  | ||||||
| 	[property: MemoryPackOrder(2)] string? UserName, |  | ||||||
| 	[property: MemoryPackOrder(3)] AuditLogEventType EventType, |  | ||||||
| 	[property: MemoryPackOrder(4)] AuditLogSubjectType SubjectType, |  | ||||||
| 	[property: MemoryPackOrder(5)] string? SubjectId, |  | ||||||
| 	[property: MemoryPackOrder(6)] string? JsonData |  | ||||||
| ); |  | ||||||
|   | |||||||
| @@ -1,24 +1,24 @@ | |||||||
| namespace Phantom.Common.Data.Web.EventLog; | namespace Phantom.Common.Data.Web.EventLog; | ||||||
|  |  | ||||||
| public enum EventLogEventType { | public enum EventLogEventType { | ||||||
| 	InstanceLaunchSucceeded, | 	InstanceLaunchSucceded, | ||||||
| 	InstanceLaunchFailed, | 	InstanceLaunchFailed, | ||||||
| 	InstanceCrashed, | 	InstanceCrashed, | ||||||
| 	InstanceStopped, | 	InstanceStopped, | ||||||
| 	InstanceBackupSucceeded, | 	InstanceBackupSucceeded, | ||||||
| 	InstanceBackupSucceededWithWarnings, | 	InstanceBackupSucceededWithWarnings, | ||||||
| 	InstanceBackupFailed | 	InstanceBackupFailed, | ||||||
| } | } | ||||||
|  |  | ||||||
| public static class EventLogEventTypeExtensions { | public static class EventLogEventTypeExtensions { | ||||||
| 	private static readonly Dictionary<EventLogEventType, EventLogSubjectType> SubjectTypes = new () { | 	private static readonly Dictionary<EventLogEventType, EventLogSubjectType> SubjectTypes = new () { | ||||||
| 		{ EventLogEventType.InstanceLaunchSucceeded, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceLaunchSucceded, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceLaunchFailed, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceLaunchFailed, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceCrashed, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceCrashed, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceStopped, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceStopped, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceBackupSucceeded, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceBackupSucceeded, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceBackupSucceededWithWarnings, EventLogSubjectType.Instance }, | 		{ EventLogEventType.InstanceBackupSucceededWithWarnings, EventLogSubjectType.Instance }, | ||||||
| 		{ EventLogEventType.InstanceBackupFailed, EventLogSubjectType.Instance } | 		{ EventLogEventType.InstanceBackupFailed, EventLogSubjectType.Instance }, | ||||||
| 	}; | 	}; | ||||||
|  |  | ||||||
| 	static EventLogEventTypeExtensions() { | 	static EventLogEventTypeExtensions() { | ||||||
|   | |||||||
| @@ -1,13 +1,5 @@ | |||||||
| using MemoryPack; | using System.Text.Json; | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.EventLog; | namespace Phantom.Common.Data.Web.EventLog; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | public sealed record EventLogItem(DateTime UtcTime, Guid? AgentGuid, EventLogEventType EventType, EventLogSubjectType SubjectType, string SubjectId, JsonDocument? Data); | ||||||
| public sealed partial record EventLogItem( |  | ||||||
| 	[property: MemoryPackOrder(0)] DateTime UtcTime, |  | ||||||
| 	[property: MemoryPackOrder(1)] Guid? AgentGuid, |  | ||||||
| 	[property: MemoryPackOrder(2)] EventLogEventType EventType, |  | ||||||
| 	[property: MemoryPackOrder(3)] EventLogSubjectType SubjectType, |  | ||||||
| 	[property: MemoryPackOrder(4)] string SubjectId, |  | ||||||
| 	[property: MemoryPackOrder(5)] string? JsonData |  | ||||||
| ); |  | ||||||
|   | |||||||
| @@ -1,23 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Web.Instance; |  | ||||||
|  |  | ||||||
| public enum CreateOrUpdateInstanceResult : byte { |  | ||||||
| 	UnknownError, |  | ||||||
| 	Success, |  | ||||||
| 	InstanceNameMustNotBeEmpty, |  | ||||||
| 	InstanceMemoryMustNotBeZero, |  | ||||||
| 	MinecraftVersionDownloadInfoNotFound, |  | ||||||
| 	AgentNotFound |  | ||||||
| } |  | ||||||
|  |  | ||||||
| public static class CreateOrUpdateInstanceResultExtensions { |  | ||||||
| 	public static string ToSentence(this CreateOrUpdateInstanceResult reason) { |  | ||||||
| 		return reason switch { |  | ||||||
| 			CreateOrUpdateInstanceResult.Success                              => "Success.", |  | ||||||
| 			CreateOrUpdateInstanceResult.InstanceNameMustNotBeEmpty           => "Instance name must not be empty.", |  | ||||||
| 			CreateOrUpdateInstanceResult.InstanceMemoryMustNotBeZero          => "Memory must not be 0 MB.", |  | ||||||
| 			CreateOrUpdateInstanceResult.MinecraftVersionDownloadInfoNotFound => "Could not find download information for the selected Minecraft version.", |  | ||||||
| 			CreateOrUpdateInstanceResult.AgentNotFound                        => "Agent not found.", |  | ||||||
| 			_                                                                 => "Unknown error." |  | ||||||
| 		}; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,17 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Instance; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record Instance( |  | ||||||
| 	[property: MemoryPackOrder(0)] Guid InstanceGuid, |  | ||||||
| 	[property: MemoryPackOrder(1)] InstanceConfiguration Configuration, |  | ||||||
| 	[property: MemoryPackOrder(2)] IInstanceStatus Status, |  | ||||||
| 	[property: MemoryPackOrder(3)] InstancePlayerCounts? PlayerCounts, |  | ||||||
| 	[property: MemoryPackOrder(4)] bool LaunchAutomatically |  | ||||||
| ) { |  | ||||||
| 	public static Instance Offline(Guid instanceGuid, InstanceConfiguration configuration, bool launchAutomatically = false) { |  | ||||||
| 		return new Instance(instanceGuid, configuration, InstanceStatus.Offline, PlayerCounts: null, launchAutomatically); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -10,8 +10,4 @@ | |||||||
|     <PackageReference Include="MemoryPack" /> |     <PackageReference Include="MemoryPack" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |  | ||||||
|     <ProjectReference Include="..\Phantom.Common.Data\Phantom.Common.Data.csproj" /> |  | ||||||
|   </ItemGroup> |  | ||||||
|    |  | ||||||
| </Project> | </Project> | ||||||
|   | |||||||
| @@ -1,24 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AuthenticatedUserInfo( |  | ||||||
| 	[property: MemoryPackOrder(0)] Guid Guid, |  | ||||||
| 	[property: MemoryPackOrder(1)] string Name, |  | ||||||
| 	[property: MemoryPackOrder(2)] PermissionSet Permissions, |  | ||||||
| 	[property: MemoryPackOrder(3)] ImmutableHashSet<Guid> ManagedAgentGuids |  | ||||||
| ) { |  | ||||||
| 	public bool CheckPermission(Permission permission) { |  | ||||||
| 		return Permissions.Check(permission); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public bool HasAccessToAgent(Guid agentGuid) { |  | ||||||
| 		return ManagedAgentGuids.Contains(agentGuid) || Permissions.Check(Permission.ManageAllAgents); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public ImmutableHashSet<Guid> FilterAccessibleAgentGuids(ImmutableHashSet<Guid> agentGuids) { |  | ||||||
| 		return Permissions.Check(Permission.ManageAllAgents) ? agentGuids : agentGuids.Intersect(ManagedAgentGuids); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,10 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users;  |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record ChangeUserRolesResult( |  | ||||||
| 	[property: MemoryPackOrder(0)] ImmutableHashSet<Guid> AddedToRoleGuids, |  | ||||||
| 	[property: MemoryPackOrder(1)] ImmutableHashSet<Guid> RemovedFromRoleGuids |  | ||||||
| ); |  | ||||||
| @@ -1,23 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.CreateUserResults; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(Success))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(CreationFailed))] |  | ||||||
| 	[MemoryPackUnion(2, typeof(UnknownError))] |  | ||||||
| 	public abstract partial record CreateUserResult { |  | ||||||
| 		internal CreateUserResult() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.CreateUserResults { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record Success([property: MemoryPackOrder(0)] UserInfo User) : CreateUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record CreationFailed([property: MemoryPackOrder(0)] AddUserError Error) : CreateUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UnknownError : CreateUserResult; |  | ||||||
| } |  | ||||||
| @@ -1,18 +1,14 @@ | |||||||
| using System.Collections.Immutable; | using System.Collections.Immutable; | ||||||
| using MemoryPack; |  | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Common.Data.Web.Users; | namespace Phantom.Common.Data.Web.Users; | ||||||
| 
 | 
 | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | public sealed class IdentityPermissions { | ||||||
| public sealed partial class PermissionSet { | 	public static IdentityPermissions None { get; } = new (ImmutableHashSet<string>.Empty); | ||||||
| 	public static PermissionSet None { get; } = new (ImmutableHashSet<string>.Empty); |  | ||||||
| 	 | 	 | ||||||
| 	[MemoryPackOrder(0)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly ImmutableHashSet<string> permissionIds; | 	private readonly ImmutableHashSet<string> permissionIds; | ||||||
| 
 | 
 | ||||||
| 	public PermissionSet(ImmutableHashSet<string> permissionIds) { | 	public IdentityPermissions(ImmutableHashSet<string> permissionIdsQuery) { | ||||||
| 		this.permissionIds = permissionIds; | 		this.permissionIds = permissionIdsQuery; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	public bool Check(Permission? permission) { | 	public bool Check(Permission? permission) { | ||||||
| @@ -1,10 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record LogInSuccess( |  | ||||||
| 	[property: MemoryPackOrder(0)] AuthenticatedUserInfo UserInfo, |  | ||||||
| 	[property: MemoryPackOrder(1)] ImmutableArray<byte> AuthToken |  | ||||||
| ); |  | ||||||
| @@ -14,9 +14,6 @@ public sealed record Permission(string Id, Permission? Parent) { | |||||||
| 		return Register(id, this); | 		return Register(id, this); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public const string ManageAllAgentsPolicy = "Agents.ManageAll"; |  | ||||||
| 	public static readonly Permission ManageAllAgents = Register(ManageAllAgentsPolicy); |  | ||||||
| 	 |  | ||||||
| 	public const string ViewInstancesPolicy = "Instances.View"; | 	public const string ViewInstancesPolicy = "Instances.View"; | ||||||
| 	public static readonly Permission ViewInstances = Register(ViewInstancesPolicy); | 	public static readonly Permission ViewInstances = Register(ViewInstancesPolicy); | ||||||
| 	 | 	 | ||||||
|   | |||||||
| @@ -1,5 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| public enum UserActionFailure { |  | ||||||
| 	NotAuthorized |  | ||||||
| } |  | ||||||
| @@ -1,25 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Replies; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| [MemoryPackable] |  | ||||||
| [MemoryPackUnion(0, typeof(OfUserActionFailure))] |  | ||||||
| [MemoryPackUnion(1, typeof(OfInstanceActionFailure))] |  | ||||||
| public abstract partial record UserInstanceActionFailure { |  | ||||||
| 	internal UserInstanceActionFailure() {} |  | ||||||
| 	 |  | ||||||
| 	public static implicit operator UserInstanceActionFailure(UserActionFailure failure) { |  | ||||||
| 		return new OfUserActionFailure(failure); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public static implicit operator UserInstanceActionFailure(InstanceActionFailure failure) { |  | ||||||
| 		return new OfInstanceActionFailure(failure); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record OfUserActionFailure([property: MemoryPackOrder(0)] UserActionFailure Failure) : UserInstanceActionFailure; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record OfInstanceActionFailure([property: MemoryPackOrder(0)] InstanceActionFailure Failure) : UserInstanceActionFailure; |  | ||||||
| @@ -4,8 +4,8 @@ namespace Phantom.Common.Data.Agent; | |||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record AgentInfo( | public sealed partial record AgentInfo( | ||||||
| 	[property: MemoryPackOrder(0)] Guid AgentGuid, | 	[property: MemoryPackOrder(0)] Guid Guid, | ||||||
| 	[property: MemoryPackOrder(1)] string AgentName, | 	[property: MemoryPackOrder(1)] string Name, | ||||||
| 	[property: MemoryPackOrder(2)] ushort ProtocolVersion, | 	[property: MemoryPackOrder(2)] ushort ProtocolVersion, | ||||||
| 	[property: MemoryPackOrder(3)] string BuildVersion, | 	[property: MemoryPackOrder(3)] string BuildVersion, | ||||||
| 	[property: MemoryPackOrder(4)] ushort MaxInstances, | 	[property: MemoryPackOrder(4)] ushort MaxInstances, | ||||||
|   | |||||||
| @@ -1,9 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Agent; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record AgentStats( |  | ||||||
| 	[property: MemoryPackOrder(0)] int RunningInstanceCount, |  | ||||||
| 	[property: MemoryPackOrder(1)] RamAllocationUnits RunningInstanceMemory |  | ||||||
| ); |  | ||||||
| @@ -14,7 +14,9 @@ public sealed partial class AuthToken { | |||||||
| 	private readonly byte[] bytes; | 	private readonly byte[] bytes; | ||||||
|  |  | ||||||
| 	internal AuthToken(byte[]? bytes) { | 	internal AuthToken(byte[]? bytes) { | ||||||
| 		ArgumentNullException.ThrowIfNull(bytes); | 		if (bytes == null) { | ||||||
|  | 			throw new ArgumentNullException(nameof(bytes)); | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		if (bytes.Length != Length) { | 		if (bytes.Length != Length) { | ||||||
| 			throw new ArgumentOutOfRangeException(nameof(bytes), "Invalid token length: " + bytes.Length + ". Token length must be exactly " + Length + " bytes."); | 			throw new ArgumentOutOfRangeException(nameof(bytes), "Invalid token length: " + bytes.Length + ". Token length must be exactly " + Length + " bytes."); | ||||||
|   | |||||||
| @@ -1,16 +1,15 @@ | |||||||
| namespace Phantom.Common.Data.Backups; | namespace Phantom.Common.Data.Backups; | ||||||
|  |  | ||||||
| public enum BackupCreationResultKind : byte { | public enum BackupCreationResultKind : byte { | ||||||
| 	UnknownError = 0, | 	UnknownError, | ||||||
| 	Success = 1, | 	Success, | ||||||
| 	InstanceNotRunning = 2, | 	InstanceNotRunning, | ||||||
| 	BackupTimedOut = 3, | 	BackupCancelled, | ||||||
| 	BackupCancelled = 4, | 	BackupAlreadyRunning, | ||||||
| 	BackupAlreadyRunning = 5, | 	BackupFileAlreadyExists, | ||||||
| 	BackupFileAlreadyExists = 6, | 	CouldNotCreateBackupFolder, | ||||||
| 	CouldNotCreateBackupFolder = 7, | 	CouldNotCopyWorldToTemporaryFolder, | ||||||
| 	CouldNotCopyWorldToTemporaryFolder = 8, | 	CouldNotCreateWorldArchive | ||||||
| 	CouldNotCreateWorldArchive = 9 |  | ||||||
| } | } | ||||||
|  |  | ||||||
| public static class BackupCreationResultSummaryExtensions { | public static class BackupCreationResultSummaryExtensions { | ||||||
|   | |||||||
| @@ -4,7 +4,7 @@ using Phantom.Common.Data.Backups; | |||||||
| namespace Phantom.Common.Data.Instance; | namespace Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
| [MemoryPackable] | [MemoryPackable] | ||||||
| [MemoryPackUnion(0, typeof(InstanceLaunchSucceededEvent))] | [MemoryPackUnion(0, typeof(InstanceLaunchSuccededEvent))] | ||||||
| [MemoryPackUnion(1, typeof(InstanceLaunchFailedEvent))] | [MemoryPackUnion(1, typeof(InstanceLaunchFailedEvent))] | ||||||
| [MemoryPackUnion(2, typeof(InstanceCrashedEvent))] | [MemoryPackUnion(2, typeof(InstanceCrashedEvent))] | ||||||
| [MemoryPackUnion(3, typeof(InstanceStoppedEvent))] | [MemoryPackUnion(3, typeof(InstanceStoppedEvent))] | ||||||
| @@ -14,7 +14,7 @@ public partial interface IInstanceEvent { | |||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record InstanceLaunchSucceededEvent : IInstanceEvent { | public sealed partial record InstanceLaunchSuccededEvent : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnLaunchSucceeded(this); | 		visitor.OnLaunchSucceeded(this); | ||||||
| 	} | 	} | ||||||
| @@ -49,7 +49,7 @@ public sealed partial record InstanceBackupCompletedEvent([property: MemoryPackO | |||||||
| } | } | ||||||
|  |  | ||||||
| public static class InstanceEvent { | public static class InstanceEvent { | ||||||
| 	public static readonly IInstanceEvent LaunchSucceeded = new InstanceLaunchSucceededEvent(); | 	public static readonly IInstanceEvent LaunchSucceded = new InstanceLaunchSuccededEvent(); | ||||||
| 	public static readonly IInstanceEvent Crashed = new InstanceCrashedEvent(); | 	public static readonly IInstanceEvent Crashed = new InstanceCrashedEvent(); | ||||||
| 	public static readonly IInstanceEvent Stopped = new InstanceStoppedEvent(); | 	public static readonly IInstanceEvent Stopped = new InstanceStoppedEvent(); | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| namespace Phantom.Common.Data.Instance; | namespace Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
| public interface IInstanceEventVisitor { | public interface IInstanceEventVisitor { | ||||||
| 	void OnLaunchSucceeded(InstanceLaunchSucceededEvent e); | 	void OnLaunchSucceeded(InstanceLaunchSuccededEvent e); | ||||||
| 	void OnLaunchFailed(InstanceLaunchFailedEvent e); | 	void OnLaunchFailed(InstanceLaunchFailedEvent e); | ||||||
| 	void OnCrashed(InstanceCrashedEvent e); | 	void OnCrashed(InstanceCrashedEvent e); | ||||||
| 	void OnStopped(InstanceStoppedEvent e); | 	void OnStopped(InstanceStoppedEvent e); | ||||||
|   | |||||||
| @@ -9,10 +9,9 @@ namespace Phantom.Common.Data.Instance; | |||||||
| [MemoryPackUnion(3, typeof(InstanceIsDownloading))] | [MemoryPackUnion(3, typeof(InstanceIsDownloading))] | ||||||
| [MemoryPackUnion(4, typeof(InstanceIsLaunching))] | [MemoryPackUnion(4, typeof(InstanceIsLaunching))] | ||||||
| [MemoryPackUnion(5, typeof(InstanceIsRunning))] | [MemoryPackUnion(5, typeof(InstanceIsRunning))] | ||||||
| [MemoryPackUnion(6, typeof(InstanceIsBackingUp))] | [MemoryPackUnion(6, typeof(InstanceIsRestarting))] | ||||||
| [MemoryPackUnion(7, typeof(InstanceIsRestarting))] | [MemoryPackUnion(7, typeof(InstanceIsStopping))] | ||||||
| [MemoryPackUnion(8, typeof(InstanceIsStopping))] | [MemoryPackUnion(8, typeof(InstanceIsFailed))] | ||||||
| [MemoryPackUnion(9, typeof(InstanceIsFailed))] |  | ||||||
| public partial interface IInstanceStatus {} | public partial interface IInstanceStatus {} | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| @@ -33,9 +32,6 @@ public sealed partial record InstanceIsLaunching : IInstanceStatus; | |||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record InstanceIsRunning : IInstanceStatus; | public sealed partial record InstanceIsRunning : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record InstanceIsBackingUp : IInstanceStatus; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record InstanceIsRestarting : IInstanceStatus; | public sealed partial record InstanceIsRestarting : IInstanceStatus; | ||||||
|  |  | ||||||
| @@ -50,35 +46,22 @@ public static class InstanceStatus { | |||||||
| 	public static readonly IInstanceStatus NotRunning = new InstanceIsNotRunning(); | 	public static readonly IInstanceStatus NotRunning = new InstanceIsNotRunning(); | ||||||
| 	public static readonly IInstanceStatus Launching = new InstanceIsLaunching(); | 	public static readonly IInstanceStatus Launching = new InstanceIsLaunching(); | ||||||
| 	public static readonly IInstanceStatus Running = new InstanceIsRunning(); | 	public static readonly IInstanceStatus Running = new InstanceIsRunning(); | ||||||
| 	public static readonly IInstanceStatus BackingUp = new InstanceIsBackingUp(); |  | ||||||
| 	public static readonly IInstanceStatus Restarting = new InstanceIsRestarting(); | 	public static readonly IInstanceStatus Restarting = new InstanceIsRestarting(); | ||||||
| 	public static readonly IInstanceStatus Stopping = new InstanceIsStopping(); | 	public static readonly IInstanceStatus Stopping = new InstanceIsStopping(); | ||||||
| 	 | 	 | ||||||
| 	public static IInstanceStatus Invalid(string reason) => new InstanceIsInvalid(reason); | 	public static IInstanceStatus Invalid(string reason) => new InstanceIsInvalid(reason); | ||||||
| 	public static IInstanceStatus Downloading(byte progress) => new InstanceIsDownloading(progress); | 	public static IInstanceStatus Downloading(byte progress) => new InstanceIsDownloading(progress); | ||||||
| 	public static IInstanceStatus Failed(InstanceLaunchFailReason reason) => new InstanceIsFailed(reason); | 	public static IInstanceStatus Failed(InstanceLaunchFailReason reason) => new InstanceIsFailed(reason); | ||||||
|  |  | ||||||
| 	public static bool IsLaunching(this IInstanceStatus status) { |  | ||||||
| 		return status is InstanceIsDownloading or InstanceIsLaunching or InstanceIsRestarting; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static bool IsRunning(this IInstanceStatus status) { |  | ||||||
| 		return status is InstanceIsRunning or InstanceIsBackingUp; |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public static bool IsStopping(this IInstanceStatus status) { |  | ||||||
| 		return status is InstanceIsStopping; |  | ||||||
| 	} |  | ||||||
| 	 | 	 | ||||||
| 	public static bool CanLaunch(this IInstanceStatus status) { | 	public static bool CanLaunch(this IInstanceStatus status) { | ||||||
| 		return status is InstanceIsNotRunning or InstanceIsFailed; | 		return status is InstanceIsNotRunning or InstanceIsFailed; | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public static bool CanStop(this IInstanceStatus status) { | 	public static bool CanStop(this IInstanceStatus status) { | ||||||
| 		return status.IsRunning() || status.IsLaunching(); | 		return status is InstanceIsDownloading or InstanceIsLaunching or InstanceIsRunning; | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public static bool CanSendCommand(this IInstanceStatus status) { | 	public static bool CanSendCommand(this IInstanceStatus status) { | ||||||
| 		return status.IsRunning(); | 		return status is InstanceIsRunning; | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -7,12 +7,13 @@ namespace Phantom.Common.Data.Instance; | |||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record InstanceConfiguration( | public sealed partial record InstanceConfiguration( | ||||||
| 	[property: MemoryPackOrder(0)] Guid AgentGuid, | 	[property: MemoryPackOrder(0)] Guid AgentGuid, | ||||||
| 	[property: MemoryPackOrder(1)] string InstanceName, | 	[property: MemoryPackOrder(1)] Guid InstanceGuid, | ||||||
| 	[property: MemoryPackOrder(2)] ushort ServerPort, | 	[property: MemoryPackOrder(2)] string InstanceName, | ||||||
| 	[property: MemoryPackOrder(3)] ushort RconPort, | 	[property: MemoryPackOrder(3)] ushort ServerPort, | ||||||
| 	[property: MemoryPackOrder(4)] string MinecraftVersion, | 	[property: MemoryPackOrder(4)] ushort RconPort, | ||||||
| 	[property: MemoryPackOrder(5)] MinecraftServerKind MinecraftServerKind, | 	[property: MemoryPackOrder(5)] string MinecraftVersion, | ||||||
| 	[property: MemoryPackOrder(6)] RamAllocationUnits MemoryAllocation, | 	[property: MemoryPackOrder(6)] MinecraftServerKind MinecraftServerKind, | ||||||
| 	[property: MemoryPackOrder(7)] Guid JavaRuntimeGuid, | 	[property: MemoryPackOrder(7)] RamAllocationUnits MemoryAllocation, | ||||||
| 	[property: MemoryPackOrder(8)] ImmutableArray<string> JvmArguments | 	[property: MemoryPackOrder(8)] Guid JavaRuntimeGuid, | ||||||
|  | 	[property: MemoryPackOrder(9)] ImmutableArray<string> JvmArguments | ||||||
| ); | ); | ||||||
|   | |||||||
| @@ -2,6 +2,10 @@ | |||||||
|  |  | ||||||
| public enum InstanceLaunchFailReason : byte { | public enum InstanceLaunchFailReason : byte { | ||||||
| 	UnknownError                           = 0, | 	UnknownError                           = 0, | ||||||
|  | 	ServerPortNotAllowed                   = 1, | ||||||
|  | 	ServerPortAlreadyInUse                 = 2, | ||||||
|  | 	RconPortNotAllowed                     = 3, | ||||||
|  | 	RconPortAlreadyInUse                   = 4, | ||||||
| 	JavaRuntimeNotFound                    = 5, | 	JavaRuntimeNotFound                    = 5, | ||||||
| 	CouldNotDownloadMinecraftServer        = 6, | 	CouldNotDownloadMinecraftServer        = 6, | ||||||
| 	CouldNotConfigureMinecraftServer       = 7, | 	CouldNotConfigureMinecraftServer       = 7, | ||||||
|   | |||||||
| @@ -1,9 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Instance; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public readonly partial record struct InstancePlayerCounts( |  | ||||||
| 	[property: MemoryPackOrder(0)] int Online, |  | ||||||
| 	[property: MemoryPackOrder(1)] int Maximum |  | ||||||
| ); |  | ||||||
| @@ -2,6 +2,5 @@ | |||||||
|  |  | ||||||
| public enum MinecraftServerKind : ushort { | public enum MinecraftServerKind : ushort { | ||||||
| 	Vanilla = 1, | 	Vanilla = 1, | ||||||
| 	Fabric = 2, | 	Fabric = 2 | ||||||
| 	Forge = 3 |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,10 +1,7 @@ | |||||||
| using MemoryPack; | namespace Phantom.Common.Data.Minecraft; | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Minecraft; | public sealed record MinecraftVersion( | ||||||
|  | 	string Id, | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | 	MinecraftVersionType Type, | ||||||
| public sealed partial record MinecraftVersion( | 	string MetadataUrl | ||||||
| 	[property: MemoryPackOrder(0)] string Id, |  | ||||||
| 	[property: MemoryPackOrder(1)] MinecraftVersionType Type, |  | ||||||
| 	[property: MemoryPackOrder(2)] string MetadataUrl |  | ||||||
| ); | ); | ||||||
|   | |||||||
| @@ -1,10 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data; |  | ||||||
|  |  | ||||||
| [MemoryPackable] |  | ||||||
| public readonly partial record struct Optional<T>(T? Value) { |  | ||||||
| 	public static implicit operator Optional<T>(T? value) { |  | ||||||
| 		return new Optional<T>(value); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -3,12 +3,3 @@ | |||||||
| public enum ConfigureInstanceResult : byte { | public enum ConfigureInstanceResult : byte { | ||||||
| 	Success | 	Success | ||||||
| } | } | ||||||
|  |  | ||||||
| public static class ConfigureInstanceResultExtensions { |  | ||||||
| 	public static string ToSentence(this ConfigureInstanceResult reason) { |  | ||||||
| 		return reason switch { |  | ||||||
| 			ConfigureInstanceResult.Success => "Success.", |  | ||||||
| 			_                               => "Unknown error." |  | ||||||
| 		}; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,20 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Replies; |  | ||||||
|  |  | ||||||
| public enum InstanceActionFailure : byte { |  | ||||||
| 	AgentDoesNotExist, |  | ||||||
| 	AgentShuttingDown, |  | ||||||
| 	AgentIsNotResponding, |  | ||||||
| 	InstanceDoesNotExist |  | ||||||
| } |  | ||||||
|  |  | ||||||
| public static class InstanceActionFailureExtensions { |  | ||||||
| 	public static string ToSentence(this InstanceActionFailure failure) { |  | ||||||
| 		return failure switch { |  | ||||||
| 			InstanceActionFailure.AgentDoesNotExist    => "Agent does not exist.", |  | ||||||
| 			InstanceActionFailure.AgentShuttingDown    => "Agent is shutting down.", |  | ||||||
| 			InstanceActionFailure.AgentIsNotResponding => "Agent is not responding.", |  | ||||||
| 			InstanceActionFailure.InstanceDoesNotExist => "Instance does not exist.", |  | ||||||
| 			_                                          => "Unknown error." |  | ||||||
| 		}; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -0,0 +1,8 @@ | |||||||
|  | namespace Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | public enum InstanceActionGeneralResult : byte { | ||||||
|  | 	None, | ||||||
|  | 	AgentShuttingDown, | ||||||
|  | 	AgentIsNotResponding, | ||||||
|  | 	InstanceDoesNotExist | ||||||
|  | } | ||||||
							
								
								
									
										41
									
								
								Common/Phantom.Common.Data/Replies/InstanceActionResult.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								Common/Phantom.Common.Data/Replies/InstanceActionResult.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,41 @@ | |||||||
|  | using MemoryPack; | ||||||
|  |  | ||||||
|  | namespace Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
|  | public sealed partial record InstanceActionResult<T>( | ||||||
|  | 	[property: MemoryPackOrder(0)] InstanceActionGeneralResult GeneralResult, | ||||||
|  | 	[property: MemoryPackOrder(1)] T? ConcreteResult | ||||||
|  | ) { | ||||||
|  | 	public bool Is(T? concreteResult) { | ||||||
|  | 		return GeneralResult == InstanceActionGeneralResult.None && EqualityComparer<T>.Default.Equals(ConcreteResult, concreteResult); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public InstanceActionResult<T2> Map<T2>(Func<T, T2> mapper) { | ||||||
|  | 		return new InstanceActionResult<T2>(GeneralResult, ConcreteResult is not null ? mapper(ConcreteResult) : default); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public string ToSentence(Func<T, string> concreteResultToSentence) { | ||||||
|  | 		return GeneralResult switch { | ||||||
|  | 			InstanceActionGeneralResult.None                 => concreteResultToSentence(ConcreteResult!), | ||||||
|  | 			InstanceActionGeneralResult.AgentShuttingDown    => "Agent is shutting down.", | ||||||
|  | 			InstanceActionGeneralResult.AgentIsNotResponding => "Agent is not responding.", | ||||||
|  | 			InstanceActionGeneralResult.InstanceDoesNotExist => "Instance does not exist.", | ||||||
|  | 			_                                                => "Unknown result." | ||||||
|  | 		}; | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | public static class InstanceActionResult { | ||||||
|  | 	public static InstanceActionResult<T> General<T>(InstanceActionGeneralResult generalResult) { | ||||||
|  | 		return new InstanceActionResult<T>(generalResult, default); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public static InstanceActionResult<T> Concrete<T>(T? concreteResult) { | ||||||
|  | 		return new InstanceActionResult<T>(InstanceActionGeneralResult.None, concreteResult); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public static InstanceActionResult<T> DidNotReplyIfNull<T>(this InstanceActionResult<T>? result) { | ||||||
|  | 		return result ?? General<T>(InstanceActionGeneralResult.AgentIsNotResponding); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -5,9 +5,5 @@ public enum LaunchInstanceResult : byte { | |||||||
| 	InstanceAlreadyLaunching = 2, | 	InstanceAlreadyLaunching = 2, | ||||||
| 	InstanceAlreadyRunning   = 3, | 	InstanceAlreadyRunning   = 3, | ||||||
| 	InstanceLimitExceeded    = 4, | 	InstanceLimitExceeded    = 4, | ||||||
| 	MemoryLimitExceeded      = 5, | 	MemoryLimitExceeded      = 5 | ||||||
| 	ServerPortNotAllowed     = 6, |  | ||||||
| 	ServerPortAlreadyInUse   = 7, |  | ||||||
| 	RconPortNotAllowed       = 8, |  | ||||||
| 	RconPortAlreadyInUse     = 9 |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,7 +1,6 @@ | |||||||
| namespace Phantom.Common.Data.Replies; | namespace Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
| public enum SendCommandToInstanceResult : byte { | public enum SendCommandToInstanceResult : byte { | ||||||
| 	UnknownError = 0, | 	UnknownError, | ||||||
| 	Success = 1, | 	Success | ||||||
| 	InstanceNotRunning = 2 |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,108 +0,0 @@ | |||||||
| using System.Diagnostics.CodeAnalysis; |  | ||||||
| using MemoryPack; |  | ||||||
| using Phantom.Utils.Result; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial class Result<TValue, TError> { |  | ||||||
| 	[MemoryPackOrder(0)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly bool hasValue; |  | ||||||
| 	 |  | ||||||
| 	[MemoryPackOrder(1)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly TValue? value; |  | ||||||
| 	 |  | ||||||
| 	[MemoryPackOrder(2)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly TError? error; |  | ||||||
|  |  | ||||||
| 	[MemoryPackIgnore] |  | ||||||
| 	public TValue Value => hasValue ? value! : throw new InvalidOperationException("Attempted to get value from an error result."); |  | ||||||
| 	 |  | ||||||
| 	[MemoryPackIgnore] |  | ||||||
| 	public TError Error => !hasValue ? error! : throw new InvalidOperationException("Attempted to get error from a success result."); |  | ||||||
| 	 |  | ||||||
| 	private Result(bool hasValue, TValue? value, TError? error) { |  | ||||||
| 		this.hasValue = hasValue; |  | ||||||
| 		this.value = value; |  | ||||||
| 		this.error = error; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public bool Is(TValue expectedValue) { |  | ||||||
| 		return hasValue && EqualityComparer<TValue>.Default.Equals(value, expectedValue); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public TOutput Into<TOutput>(Func<TValue, TOutput> valueConverter, Func<TError, TOutput> errorConverter) { |  | ||||||
| 		return hasValue ? valueConverter(value!) : errorConverter(error!); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public Result<TValue, TNewError> MapError<TNewError>(Func<TError, TNewError> errorConverter) { |  | ||||||
| 		return hasValue ? value! : errorConverter(error!); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public Utils.Result.Result Variant() { |  | ||||||
| 		return hasValue ? new Ok<TValue>(Value) : new Err<TError>(Error); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static implicit operator Result<TValue, TError>(TValue value) { |  | ||||||
| 		return new Result<TValue, TError>(hasValue: true, value, default); |  | ||||||
| 	} |  | ||||||
| 	 |  | ||||||
| 	public static implicit operator Result<TValue, TError>(TError error) { |  | ||||||
| 		return new Result<TValue, TError>(hasValue: false, default, error); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static implicit operator bool(Result<TValue, TError> result) { |  | ||||||
| 		return result.hasValue; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial class Result<TError> { |  | ||||||
| 	[MemoryPackOrder(0)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly bool hasValue; |  | ||||||
| 	 |  | ||||||
| 	[MemoryPackOrder(1)] |  | ||||||
| 	[MemoryPackInclude] |  | ||||||
| 	private readonly TError? error; |  | ||||||
|  |  | ||||||
| 	[MemoryPackIgnore] |  | ||||||
| 	public TError Error => !hasValue ? error! : throw new InvalidOperationException("Attempted to get error from a success result."); |  | ||||||
|  |  | ||||||
| 	private Result(bool hasValue, TError? error) { |  | ||||||
| 		this.hasValue = hasValue; |  | ||||||
| 		this.error = error; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public bool TryGetError([MaybeNullWhen(false)] out TError error) { |  | ||||||
| 		if (hasValue) { |  | ||||||
| 			error = default; |  | ||||||
| 			return false; |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			error = this.error!; |  | ||||||
| 			return true; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static implicit operator Result<TError>([SuppressMessage("ReSharper", "UnusedParameter.Global")] Result.OkType _) { |  | ||||||
| 		return new Result<TError>(hasValue: true, default); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static implicit operator Result<TError>(TError error) { |  | ||||||
| 		return new Result<TError>(hasValue: false, error); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static implicit operator bool(Result<TError> result) { |  | ||||||
| 		return result.hasValue; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| public static class Result { |  | ||||||
| 	public static OkType Ok { get; }  = new (); |  | ||||||
|  |  | ||||||
| 	public readonly record struct OkType; |  | ||||||
| } |  | ||||||
| @@ -1,7 +1,7 @@ | |||||||
| using System.Diagnostics.CodeAnalysis; | using System.Diagnostics.CodeAnalysis; | ||||||
| using Serilog.Events; | using Serilog.Events; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Utils.Logging; | namespace Phantom.Common.Logging; | ||||||
| 
 | 
 | ||||||
| static class DefaultLogLevel { | static class DefaultLogLevel { | ||||||
| 	private const string ENVIRONMENT_VARIABLE = "LOG_LEVEL"; | 	private const string ENVIRONMENT_VARIABLE = "LOG_LEVEL"; | ||||||
| @@ -1,6 +1,6 @@ | |||||||
| using Serilog; | using Serilog; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Utils.Logging; | namespace Phantom.Common.Logging; | ||||||
| 
 | 
 | ||||||
| public static class LoggerExtensions { | public static class LoggerExtensions { | ||||||
| 	private static readonly string HeadingPadding = new (' ', 23); | 	private static readonly string HeadingPadding = new (' ', 23); | ||||||
| @@ -1,4 +1,4 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
| @@ -4,7 +4,7 @@ using Serilog.Core; | |||||||
| using Serilog.Events; | using Serilog.Events; | ||||||
| using Serilog.Sinks.SystemConsole.Themes; | using Serilog.Sinks.SystemConsole.Themes; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Utils.Logging; | namespace Phantom.Common.Logging; | ||||||
| 
 | 
 | ||||||
| public static class PhantomLogger { | public static class PhantomLogger { | ||||||
| 	public static Logger Root { get; } = CreateLogger("[{Timestamp:HH:mm:ss} {Level:u}] {Message:lj}{NewLine}{Exception}"); | 	public static Logger Root { get; } = CreateLogger("[{Timestamp:HH:mm:ss} {Level:u}] {Message:lj}{NewLine}{Exception}"); | ||||||
| @@ -1,26 +1,25 @@ | |||||||
| using Phantom.Common.Data; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent.BiDirectional; | using Phantom.Common.Messages.Agent.BiDirectional; | ||||||
| using Phantom.Common.Messages.Agent.ToAgent; | using Phantom.Common.Messages.Agent.ToAgent; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.Agent.ToController; | ||||||
| using Phantom.Utils.Logging; |  | ||||||
| using Phantom.Utils.Rpc.Message; | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
| namespace Phantom.Common.Messages.Agent; | namespace Phantom.Common.Messages.Agent; | ||||||
|  |  | ||||||
| public static class AgentMessageRegistries { | public static class AgentMessageRegistries { | ||||||
| 	public static MessageRegistry<IMessageToAgent> ToAgent { get; } = new (PhantomLogger.Create("MessageRegistry", nameof(ToAgent))); | 	public static MessageRegistry<IMessageToAgentListener> ToAgent { get; } = new (PhantomLogger.Create("MessageRegistry", nameof(ToAgent))); | ||||||
| 	public static MessageRegistry<IMessageToController> ToController { get; } = new (PhantomLogger.Create("MessageRegistry", nameof(ToController))); | 	public static MessageRegistry<IMessageToControllerListener> ToController { get; } = new (PhantomLogger.Create("MessageRegistry", nameof(ToController))); | ||||||
| 	 | 	 | ||||||
| 	public static IMessageDefinitions<IMessageToAgent, IMessageToController, ReplyMessage> Definitions { get; } = new MessageDefinitions(); | 	public static IMessageDefinitions<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> Definitions { get; } = new MessageDefinitions(); | ||||||
|  |  | ||||||
| 	static AgentMessageRegistries() { | 	static AgentMessageRegistries() { | ||||||
| 		ToAgent.Add<RegisterAgentSuccessMessage>(0); | 		ToAgent.Add<RegisterAgentSuccessMessage>(0); | ||||||
| 		ToAgent.Add<RegisterAgentFailureMessage>(1); | 		ToAgent.Add<RegisterAgentFailureMessage>(1); | ||||||
| 		ToAgent.Add<ConfigureInstanceMessage, Result<ConfigureInstanceResult, InstanceActionFailure>>(2); | 		ToAgent.Add<ConfigureInstanceMessage, InstanceActionResult<ConfigureInstanceResult>>(2); | ||||||
| 		ToAgent.Add<LaunchInstanceMessage, Result<LaunchInstanceResult, InstanceActionFailure>>(3); | 		ToAgent.Add<LaunchInstanceMessage, InstanceActionResult<LaunchInstanceResult>>(3); | ||||||
| 		ToAgent.Add<StopInstanceMessage, Result<StopInstanceResult, InstanceActionFailure>>(4); | 		ToAgent.Add<StopInstanceMessage, InstanceActionResult<StopInstanceResult>>(4); | ||||||
| 		ToAgent.Add<SendCommandToInstanceMessage, Result<SendCommandToInstanceResult, InstanceActionFailure>>(5); | 		ToAgent.Add<SendCommandToInstanceMessage, InstanceActionResult<SendCommandToInstanceResult>>(5); | ||||||
| 		ToAgent.Add<ReplyMessage>(127); | 		ToAgent.Add<ReplyMessage>(127); | ||||||
| 		 | 		 | ||||||
| 		ToController.Add<RegisterAgentMessage>(0); | 		ToController.Add<RegisterAgentMessage>(0); | ||||||
| @@ -31,13 +30,16 @@ public static class AgentMessageRegistries { | |||||||
| 		ToController.Add<InstanceOutputMessage>(5); | 		ToController.Add<InstanceOutputMessage>(5); | ||||||
| 		ToController.Add<ReportAgentStatusMessage>(6); | 		ToController.Add<ReportAgentStatusMessage>(6); | ||||||
| 		ToController.Add<ReportInstanceEventMessage>(7); | 		ToController.Add<ReportInstanceEventMessage>(7); | ||||||
| 		ToController.Add<ReportInstancePlayerCountsMessage>(8); |  | ||||||
| 		ToController.Add<ReplyMessage>(127); | 		ToController.Add<ReplyMessage>(127); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private sealed class MessageDefinitions : IMessageDefinitions<IMessageToAgent, IMessageToController, ReplyMessage> { | 	private sealed class MessageDefinitions : IMessageDefinitions<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> { | ||||||
| 		public MessageRegistry<IMessageToAgent> ToClient => ToAgent; | 		public MessageRegistry<IMessageToAgentListener> ToClient => ToAgent; | ||||||
| 		public MessageRegistry<IMessageToController> ToServer => ToController; | 		public MessageRegistry<IMessageToControllerListener> ToServer => ToController; | ||||||
|  |  | ||||||
|  | 		public bool IsRegistrationMessage(Type messageType) { | ||||||
|  | 			return messageType == typeof(RegisterAgentMessage); | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		public ReplyMessage CreateReplyMessage(uint sequenceId, byte[] serializedReply) { | 		public ReplyMessage CreateReplyMessage(uint sequenceId, byte[] serializedReply) { | ||||||
| 			return new ReplyMessage(sequenceId, serializedReply); | 			return new ReplyMessage(sequenceId, serializedReply); | ||||||
|   | |||||||
| @@ -7,4 +7,12 @@ namespace Phantom.Common.Messages.Agent.BiDirectional; | |||||||
| public sealed partial record ReplyMessage( | public sealed partial record ReplyMessage( | ||||||
| 	[property: MemoryPackOrder(0)] uint SequenceId, | 	[property: MemoryPackOrder(0)] uint SequenceId, | ||||||
| 	[property: MemoryPackOrder(1)] byte[] SerializedReply | 	[property: MemoryPackOrder(1)] byte[] SerializedReply | ||||||
| ) : IMessageToController, IMessageToAgent, IReply; | ) : IMessageToController, IMessageToAgent, IReply { | ||||||
|  | 	public Task<NoReply> Accept(IMessageToControllerListener listener) { | ||||||
|  | 		return listener.HandleReply(this); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<NoReply> Accept(IMessageToAgentListener listener) { | ||||||
|  | 		return listener.HandleReply(this); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,3 +1,7 @@ | |||||||
| namespace Phantom.Common.Messages.Agent; | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
| public interface IMessageToAgent {} | namespace Phantom.Common.Messages.Agent; | ||||||
|  |  | ||||||
|  | public interface IMessageToAgent<TReply> : IMessage<IMessageToAgentListener, TReply> {} | ||||||
|  |  | ||||||
|  | public interface IMessageToAgent : IMessageToAgent<NoReply> {} | ||||||
|   | |||||||
| @@ -0,0 +1,16 @@ | |||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  | using Phantom.Common.Messages.Agent.BiDirectional; | ||||||
|  | using Phantom.Common.Messages.Agent.ToAgent; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
|  | namespace Phantom.Common.Messages.Agent; | ||||||
|  |  | ||||||
|  | public interface IMessageToAgentListener { | ||||||
|  | 	Task<NoReply> HandleRegisterAgentSuccess(RegisterAgentSuccessMessage message); | ||||||
|  | 	Task<NoReply> HandleRegisterAgentFailure(RegisterAgentFailureMessage message); | ||||||
|  | 	Task<InstanceActionResult<ConfigureInstanceResult>> HandleConfigureInstance(ConfigureInstanceMessage message); | ||||||
|  | 	Task<InstanceActionResult<LaunchInstanceResult>> HandleLaunchInstance(LaunchInstanceMessage message); | ||||||
|  | 	Task<InstanceActionResult<StopInstanceResult>> HandleStopInstance(StopInstanceMessage message); | ||||||
|  | 	Task<InstanceActionResult<SendCommandToInstanceResult>> HandleSendCommandToInstance(SendCommandToInstanceMessage message); | ||||||
|  | 	Task<NoReply> HandleReply(ReplyMessage message); | ||||||
|  | } | ||||||
| @@ -1,3 +1,7 @@ | |||||||
| namespace Phantom.Common.Messages.Agent; | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
| public interface IMessageToController {} | namespace Phantom.Common.Messages.Agent; | ||||||
|  |  | ||||||
|  | public interface IMessageToController<TReply> : IMessage<IMessageToControllerListener, TReply> {} | ||||||
|  |  | ||||||
|  | public interface IMessageToController : IMessageToController<NoReply> {} | ||||||
|   | |||||||
| @@ -0,0 +1,17 @@ | |||||||
|  | using Phantom.Common.Messages.Agent.BiDirectional; | ||||||
|  | using Phantom.Common.Messages.Agent.ToController; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
|  | namespace Phantom.Common.Messages.Agent; | ||||||
|  |  | ||||||
|  | public interface IMessageToControllerListener { | ||||||
|  | 	Task<NoReply> HandleRegisterAgent(RegisterAgentMessage message); | ||||||
|  | 	Task<NoReply> HandleUnregisterAgent(UnregisterAgentMessage message); | ||||||
|  | 	Task<NoReply> HandleAgentIsAlive(AgentIsAliveMessage message); | ||||||
|  | 	Task<NoReply> HandleAdvertiseJavaRuntimes(AdvertiseJavaRuntimesMessage message); | ||||||
|  | 	Task<NoReply> HandleReportAgentStatus(ReportAgentStatusMessage message); | ||||||
|  | 	Task<NoReply> HandleReportInstanceStatus(ReportInstanceStatusMessage message); | ||||||
|  | 	Task<NoReply> HandleReportInstanceEvent(ReportInstanceEventMessage message); | ||||||
|  | 	Task<NoReply> HandleInstanceOutput(InstanceOutputMessage message); | ||||||
|  | 	Task<NoReply> HandleReply(ReplyMessage message); | ||||||
|  | } | ||||||
| @@ -6,8 +6,8 @@ | |||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|  |     <ProjectReference Include="..\Phantom.Common.Logging\Phantom.Common.Logging.csproj" /> | ||||||
|     <ProjectReference Include="..\Phantom.Common.Data\Phantom.Common.Data.csproj" /> |     <ProjectReference Include="..\Phantom.Common.Data\Phantom.Common.Data.csproj" /> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Logging\Phantom.Utils.Logging.csproj" /> |  | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Rpc\Phantom.Utils.Rpc.csproj" /> |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Rpc\Phantom.Utils.Rpc.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,15 +1,16 @@ | |||||||
| using MemoryPack; | using MemoryPack; | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Instance; | using Phantom.Common.Data.Instance; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Utils.Actor; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Messages.Agent.ToAgent; | namespace Phantom.Common.Messages.Agent.ToAgent; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record ConfigureInstanceMessage( | public sealed partial record ConfigureInstanceMessage( | ||||||
| 	[property: MemoryPackOrder(0)] Guid InstanceGuid, | 	[property: MemoryPackOrder(0)] InstanceConfiguration Configuration, | ||||||
| 	[property: MemoryPackOrder(1)] InstanceConfiguration Configuration, | 	[property: MemoryPackOrder(1)] InstanceLaunchProperties LaunchProperties, | ||||||
| 	[property: MemoryPackOrder(2)] InstanceLaunchProperties LaunchProperties, | 	[property: MemoryPackOrder(2)] bool LaunchNow = false | ||||||
| 	[property: MemoryPackOrder(3)] bool LaunchNow = false | ) : IMessageToAgent<InstanceActionResult<ConfigureInstanceResult>> { | ||||||
| ) : IMessageToAgent, ICanReply<Result<ConfigureInstanceResult, InstanceActionFailure>>; | 	public Task<InstanceActionResult<ConfigureInstanceResult>> Accept(IMessageToAgentListener listener) { | ||||||
|  | 		return listener.HandleConfigureInstance(this); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,11 +1,13 @@ | |||||||
| using MemoryPack; | using MemoryPack; | ||||||
| using Phantom.Common.Data; |  | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Utils.Actor; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Messages.Agent.ToAgent; | namespace Phantom.Common.Messages.Agent.ToAgent; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record LaunchInstanceMessage( | public sealed partial record LaunchInstanceMessage( | ||||||
| 	[property: MemoryPackOrder(0)] Guid InstanceGuid | 	[property: MemoryPackOrder(0)] Guid InstanceGuid | ||||||
| ) : IMessageToAgent, ICanReply<Result<LaunchInstanceResult, InstanceActionFailure>>; | ) : IMessageToAgent<InstanceActionResult<LaunchInstanceResult>> { | ||||||
|  | 	public Task<InstanceActionResult<LaunchInstanceResult>> Accept(IMessageToAgentListener listener) { | ||||||
|  | 		return listener.HandleLaunchInstance(this); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,9 +1,14 @@ | |||||||
| using MemoryPack; | using MemoryPack; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
| namespace Phantom.Common.Messages.Agent.ToAgent; | namespace Phantom.Common.Messages.Agent.ToAgent; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable(GenerateType.VersionTolerant)] | ||||||
| public sealed partial record RegisterAgentFailureMessage( | public sealed partial record RegisterAgentFailureMessage( | ||||||
| 	[property: MemoryPackOrder(0)] RegisterAgentFailure FailureKind | 	[property: MemoryPackOrder(0)] RegisterAgentFailure FailureKind | ||||||
| ) : IMessageToAgent; | ) : IMessageToAgent { | ||||||
|  | 	public Task<NoReply> Accept(IMessageToAgentListener listener) { | ||||||
|  | 		return listener.HandleRegisterAgentFailure(this); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user