mirror of
				https://github.com/chylex/Minecraft-Phantom-Panel.git
				synced 2025-10-31 20:17:16 +01:00 
			
		
		
		
	Compare commits
	
		
			1 Commits
		
	
	
		
			15d45fe1a3
			...
			wip-launch
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 101ca865fe | 
| @@ -1,6 +1,6 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Agent 1" type="DotNetProject" factoryName=".NET Project"> |   <configuration default="false" name="Agent 1" type="DotNetProject" factoryName=".NET Project"> | ||||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> |     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||||
|     <option name="PROGRAM_PARAMETERS" value="" /> |     <option name="PROGRAM_PARAMETERS" value="" /> | ||||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent1" /> |     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent1" /> | ||||||
|     <option name="PASS_PARENT_ENVS" value="1" /> |     <option name="PASS_PARENT_ENVS" value="1" /> | ||||||
| @@ -9,10 +9,10 @@ | |||||||
|       <env name="AGENT_NAME" value="Agent 1" /> |       <env name="AGENT_NAME" value="Agent 1" /> | ||||||
|       <env name="ALLOWED_RCON_PORTS" value="25575,27000,27001" /> |       <env name="ALLOWED_RCON_PORTS" value="25575,27000,27001" /> | ||||||
|       <env name="ALLOWED_SERVER_PORTS" value="25565,26000,26001" /> |       <env name="ALLOWED_SERVER_PORTS" value="25565,26000,26001" /> | ||||||
|       <env name="CONTROLLER_HOST" value="localhost" /> |  | ||||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> |       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||||
|       <env name="MAX_INSTANCES" value="3" /> |       <env name="MAX_INSTANCES" value="3" /> | ||||||
|       <env name="MAX_MEMORY" value="12G" /> |       <env name="MAX_MEMORY" value="12G" /> | ||||||
|  |       <env name="SERVER_HOST" value="localhost" /> | ||||||
|     </envs> |     </envs> | ||||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> |     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||||
|     <option name="USE_MONO" value="0" /> |     <option name="USE_MONO" value="0" /> | ||||||
| @@ -22,7 +22,7 @@ | |||||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> |     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> |     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> |     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||||
|     <option name="PROJECT_TFM" value="net8.0" /> |     <option name="PROJECT_TFM" value="net7.0" /> | ||||||
|     <method v="2"> |     <method v="2"> | ||||||
|       <option name="Build" /> |       <option name="Build" /> | ||||||
|     </method> |     </method> | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Agent 2" type="DotNetProject" factoryName=".NET Project"> |   <configuration default="false" name="Agent 2" type="DotNetProject" factoryName=".NET Project"> | ||||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> |     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||||
|     <option name="PROGRAM_PARAMETERS" value="" /> |     <option name="PROGRAM_PARAMETERS" value="" /> | ||||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent2" /> |     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent2" /> | ||||||
|     <option name="PASS_PARENT_ENVS" value="1" /> |     <option name="PASS_PARENT_ENVS" value="1" /> | ||||||
| @@ -9,10 +9,10 @@ | |||||||
|       <env name="AGENT_NAME" value="Agent 2" /> |       <env name="AGENT_NAME" value="Agent 2" /> | ||||||
|       <env name="ALLOWED_RCON_PORTS" value="27002-27006" /> |       <env name="ALLOWED_RCON_PORTS" value="27002-27006" /> | ||||||
|       <env name="ALLOWED_SERVER_PORTS" value="26002-26006" /> |       <env name="ALLOWED_SERVER_PORTS" value="26002-26006" /> | ||||||
|       <env name="CONTROLLER_HOST" value="localhost" /> |  | ||||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> |       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||||
|       <env name="MAX_INSTANCES" value="5" /> |       <env name="MAX_INSTANCES" value="5" /> | ||||||
|       <env name="MAX_MEMORY" value="10G" /> |       <env name="MAX_MEMORY" value="10G" /> | ||||||
|  |       <env name="SERVER_HOST" value="localhost" /> | ||||||
|     </envs> |     </envs> | ||||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> |     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||||
|     <option name="USE_MONO" value="0" /> |     <option name="USE_MONO" value="0" /> | ||||||
| @@ -22,7 +22,7 @@ | |||||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> |     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> |     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> |     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||||
|     <option name="PROJECT_TFM" value="net8.0" /> |     <option name="PROJECT_TFM" value="net7.0" /> | ||||||
|     <method v="2"> |     <method v="2"> | ||||||
|       <option name="Build" /> |       <option name="Build" /> | ||||||
|     </method> |     </method> | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Agent 3" type="DotNetProject" factoryName=".NET Project"> |   <configuration default="false" name="Agent 3" type="DotNetProject" factoryName=".NET Project"> | ||||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> |     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||||
|     <option name="PROGRAM_PARAMETERS" value="" /> |     <option name="PROGRAM_PARAMETERS" value="" /> | ||||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent3" /> |     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent3" /> | ||||||
|     <option name="PASS_PARENT_ENVS" value="1" /> |     <option name="PASS_PARENT_ENVS" value="1" /> | ||||||
| @@ -9,10 +9,10 @@ | |||||||
|       <env name="AGENT_NAME" value="Agent 3" /> |       <env name="AGENT_NAME" value="Agent 3" /> | ||||||
|       <env name="ALLOWED_RCON_PORTS" value="27007" /> |       <env name="ALLOWED_RCON_PORTS" value="27007" /> | ||||||
|       <env name="ALLOWED_SERVER_PORTS" value="26007" /> |       <env name="ALLOWED_SERVER_PORTS" value="26007" /> | ||||||
|       <env name="CONTROLLER_HOST" value="localhost" /> |  | ||||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> |       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||||
|       <env name="MAX_INSTANCES" value="1" /> |       <env name="MAX_INSTANCES" value="1" /> | ||||||
|       <env name="MAX_MEMORY" value="2560M" /> |       <env name="MAX_MEMORY" value="2560M" /> | ||||||
|  |       <env name="SERVER_HOST" value="localhost" /> | ||||||
|     </envs> |     </envs> | ||||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> |     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||||
|     <option name="USE_MONO" value="0" /> |     <option name="USE_MONO" value="0" /> | ||||||
| @@ -22,7 +22,7 @@ | |||||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> |     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> |     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> |     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||||
|     <option name="PROJECT_TFM" value="net8.0" /> |     <option name="PROJECT_TFM" value="net7.0" /> | ||||||
|     <method v="2"> |     <method v="2"> | ||||||
|       <option name="Build" /> |       <option name="Build" /> | ||||||
|     </method> |     </method> | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Compile Bootstrap" type="js.build_tools.npm"> |   <configuration default="false" name="Compile Bootstrap" type="js.build_tools.npm"> | ||||||
|     <package-json value="$PROJECT_DIR$/Web/Phantom.Web.Bootstrap/package.json" /> |     <package-json value="$PROJECT_DIR$/Server/Phantom.Server.Web.Bootstrap/package.json" /> | ||||||
|     <command value="run" /> |     <command value="run" /> | ||||||
|     <scripts> |     <scripts> | ||||||
|       <script value="compile" /> |       <script value="compile" /> | ||||||
|   | |||||||
| @@ -1,8 +0,0 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> |  | ||||||
|   <configuration default="false" name="Controller + Web + Agent" type="CompoundRunConfigurationType"> |  | ||||||
|     <toRun name="Agent 1" type="DotNetProject" /> |  | ||||||
|     <toRun name="Controller" type="DotNetProject" /> |  | ||||||
|     <toRun name="Web" type="DotNetProject" /> |  | ||||||
|     <method v="2" /> |  | ||||||
|   </configuration> |  | ||||||
| </component> |  | ||||||
| @@ -1,10 +1,9 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Controller + Web + Agent x3" type="CompoundRunConfigurationType"> |   <configuration default="false" name="Server + Agent x3" type="CompoundRunConfigurationType"> | ||||||
|     <toRun name="Agent 1" type="DotNetProject" /> |     <toRun name="Agent 1" type="DotNetProject" /> | ||||||
|     <toRun name="Agent 2" type="DotNetProject" /> |     <toRun name="Agent 2" type="DotNetProject" /> | ||||||
|     <toRun name="Agent 3" type="DotNetProject" /> |     <toRun name="Agent 3" type="DotNetProject" /> | ||||||
|     <toRun name="Controller" type="DotNetProject" /> |     <toRun name="Server" type="DotNetProject" /> | ||||||
|     <toRun name="Web" type="DotNetProject" /> |  | ||||||
|     <method v="2" /> |     <method v="2" /> | ||||||
|   </configuration> |   </configuration> | ||||||
| </component> | </component> | ||||||
							
								
								
									
										7
									
								
								.run/Server + Agent.run.xml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								.run/Server + Agent.run.xml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,7 @@ | |||||||
|  | <component name="ProjectRunConfigurationManager"> | ||||||
|  |   <configuration default="false" name="Server + Agent" type="CompoundRunConfigurationType"> | ||||||
|  |     <toRun name="Agent 1" type="DotNetProject" /> | ||||||
|  |     <toRun name="Server" type="DotNetProject" /> | ||||||
|  |     <method v="2" /> | ||||||
|  |   </configuration> | ||||||
|  | </component> | ||||||
| @@ -1,27 +1,28 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> | <component name="ProjectRunConfigurationManager"> | ||||||
|   <configuration default="false" name="Controller" type="DotNetProject" factoryName=".NET Project"> |   <configuration default="false" name="Server" type="DotNetProject" factoryName=".NET Project"> | ||||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Controller/debug/Phantom.Controller.exe" /> |     <option name="EXE_PATH" value="$PROJECT_DIR$/Server/Phantom.Server/bin/Debug/net7.0/Phantom.Server.exe" /> | ||||||
|     <option name="PROGRAM_PARAMETERS" value="" /> |     <option name="PROGRAM_PARAMETERS" value="" /> | ||||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Controller" /> |     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Server" /> | ||||||
|     <option name="PASS_PARENT_ENVS" value="1" /> |     <option name="PASS_PARENT_ENVS" value="1" /> | ||||||
|     <envs> |     <envs> | ||||||
|       <env name="AGENT_RPC_SERVER_HOST" value="localhost" /> |       <env name="ASPNETCORE_ENVIRONMENT" value="Development" /> | ||||||
|       <env name="PG_DATABASE" value="postgres" /> |       <env name="PG_DATABASE" value="postgres" /> | ||||||
|       <env name="PG_HOST" value="localhost" /> |       <env name="PG_HOST" value="localhost" /> | ||||||
|       <env name="PG_PASS" value="development" /> |       <env name="PG_PASS" value="development" /> | ||||||
|       <env name="PG_PORT" value="9403" /> |       <env name="PG_PORT" value="9402" /> | ||||||
|       <env name="PG_USER" value="postgres" /> |       <env name="PG_USER" value="postgres" /> | ||||||
|       <env name="WEB_RPC_SERVER_HOST" value="localhost" /> |       <env name="RPC_SERVER_HOST" value="localhost" /> | ||||||
|  |       <env name="WEB_SERVER_HOST" value="localhost" /> | ||||||
|     </envs> |     </envs> | ||||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> |     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||||
|     <option name="USE_MONO" value="0" /> |     <option name="USE_MONO" value="0" /> | ||||||
|     <option name="RUNTIME_ARGUMENTS" value="" /> |     <option name="RUNTIME_ARGUMENTS" value="" /> | ||||||
|     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Controller/Phantom.Controller/Phantom.Controller.csproj" /> |     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Server/Phantom.Server/Phantom.Server.csproj" /> | ||||||
|     <option name="PROJECT_EXE_PATH_TRACKING" value="1" /> |     <option name="PROJECT_EXE_PATH_TRACKING" value="1" /> | ||||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> |     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> |     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> |     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||||
|     <option name="PROJECT_TFM" value="net8.0" /> |     <option name="PROJECT_TFM" value="net7.0" /> | ||||||
|     <method v="2"> |     <method v="2"> | ||||||
|       <option name="Build" /> |       <option name="Build" /> | ||||||
|     </method> |     </method> | ||||||
| @@ -1,26 +0,0 @@ | |||||||
| <component name="ProjectRunConfigurationManager"> |  | ||||||
|   <configuration default="false" name="Web" type="DotNetProject" factoryName=".NET Project"> |  | ||||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Web/debug/Phantom.Web.exe" /> |  | ||||||
|     <option name="PROGRAM_PARAMETERS" value="" /> |  | ||||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Web" /> |  | ||||||
|     <option name="PASS_PARENT_ENVS" value="1" /> |  | ||||||
|     <envs> |  | ||||||
|       <env name="ASPNETCORE_ENVIRONMENT" value="Development" /> |  | ||||||
|       <env name="CONTROLLER_HOST" value="localhost" /> |  | ||||||
|       <env name="WEB_KEY" value="BMNHM9RRPMCBBY29D9XHS6KBKZSRY7F5XFN27YZX96XXWJC2NM2D6YRHM9PZN9JGQGCSJ6FMB2GGZ" /> |  | ||||||
|       <env name="WEB_SERVER_HOST" value="localhost" /> |  | ||||||
|     </envs> |  | ||||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> |  | ||||||
|     <option name="USE_MONO" value="0" /> |  | ||||||
|     <option name="RUNTIME_ARGUMENTS" value="" /> |  | ||||||
|     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Web/Phantom.Web/Phantom.Web.csproj" /> |  | ||||||
|     <option name="PROJECT_EXE_PATH_TRACKING" value="1" /> |  | ||||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> |  | ||||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> |  | ||||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> |  | ||||||
|     <option name="PROJECT_TFM" value="net8.0" /> |  | ||||||
|     <method v="2"> |  | ||||||
|       <option name="Build" /> |  | ||||||
|     </method> |  | ||||||
|   </configuration> |  | ||||||
| </component> |  | ||||||
							
								
								
									
										1
									
								
								.workdir/Controller/.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.workdir/Controller/.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1 +0,0 @@ | |||||||
|  |  | ||||||
| @@ -1,2 +0,0 @@ | |||||||
| <EFBFBD><EFBFBD>h?Ο<05>Bx |  | ||||||
| <02> |  | ||||||
| @@ -1 +0,0 @@ | |||||||
| T<EFBFBD>./g<11><>N<EFBFBD><4E>t<EFBFBD>$<24>!<21>(<28><>#<23>~<7E><>}<14><: |  | ||||||
| @@ -5,4 +5,4 @@ if [%1]==[] ( | |||||||
|     exit |     exit | ||||||
| ) | ) | ||||||
|  |  | ||||||
| dotnet ef migrations add %~1 --project Controller/Phantom.Controller.Database.Postgres --msbuildprojectextensionspath .artifacts/obj/Phantom.Controller.Database.Postgres | dotnet ef migrations add %~1 --project Server/Phantom.Server.Database.Postgres | ||||||
|   | |||||||
| @@ -3,4 +3,4 @@ if [ -z "$1" ]; then | |||||||
|     exit 1 |     exit 1 | ||||||
| fi | fi | ||||||
|  |  | ||||||
| dotnet ef migrations add "$1" --project Controller/Phantom.Controller.Database.Postgres --msbuildprojectextensionspath .artifacts/obj/Phantom.Controller.Database.Postgres | dotnet ef migrations add "$1" --project Server/Phantom.Server.Database.Postgres | ||||||
|   | |||||||
| @@ -1,12 +1,13 @@ | |||||||
| using Phantom.Utils.Collections; | using Phantom.Utils.Collections; | ||||||
| using Phantom.Utils.Processes; | using Phantom.Utils.Runtime; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Instance;  | namespace Phantom.Agent.Minecraft.Instance;  | ||||||
|  |  | ||||||
| public sealed class InstanceProcess : IDisposable { | public sealed class InstanceProcess : IDisposable { | ||||||
| 	public InstanceProperties InstanceProperties { get; } | 	public InstanceProperties InstanceProperties { get; } | ||||||
|  | 	public CancellableSemaphore BackupSemaphore { get; } = new (1, 1); | ||||||
| 	 | 	 | ||||||
| 	private readonly RingBuffer<string> outputBuffer = new (100); | 	private readonly RingBuffer<string> outputBuffer = new (10000); | ||||||
| 	private event EventHandler<string>? OutputEvent; | 	private event EventHandler<string>? OutputEvent; | ||||||
|  |  | ||||||
| 	public event EventHandler? Ended; | 	public event EventHandler? Ended; | ||||||
| @@ -60,6 +61,7 @@ public sealed class InstanceProcess : IDisposable { | |||||||
|  |  | ||||||
| 	public void Dispose() { | 	public void Dispose() { | ||||||
| 		process.Dispose(); | 		process.Dispose(); | ||||||
|  | 		BackupSemaphore.Dispose(); | ||||||
| 		OutputEvent = null; | 		OutputEvent = null; | ||||||
| 		Ended = null; | 		Ended = null; | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,92 +0,0 @@ | |||||||
| using System.Text; |  | ||||||
| using Kajabity.Tools.Java; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Java; |  | ||||||
|  |  | ||||||
| sealed class JavaPropertiesFileEditor { |  | ||||||
| 	private static readonly Encoding Encoding = Encoding.GetEncoding("ISO-8859-1"); |  | ||||||
|  |  | ||||||
| 	private readonly Dictionary<string, string> overriddenProperties = new (); |  | ||||||
|  |  | ||||||
| 	public void Set(string key, string value) { |  | ||||||
| 		overriddenProperties[key] = value; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public async Task EditOrCreate(string filePath) { |  | ||||||
| 		if (File.Exists(filePath)) { |  | ||||||
| 			string tmpFilePath = filePath + ".tmp"; |  | ||||||
| 			File.Copy(filePath, tmpFilePath, overwrite: true); |  | ||||||
| 			await EditFromCopyOrCreate(filePath, tmpFilePath); |  | ||||||
| 			File.Move(tmpFilePath, filePath, overwrite: true); |  | ||||||
| 		} |  | ||||||
| 		else { |  | ||||||
| 			await EditFromCopyOrCreate(null, filePath); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task EditFromCopyOrCreate(string? sourceFilePath, string targetFilePath) { |  | ||||||
| 		var properties = new JavaProperties(); |  | ||||||
|  |  | ||||||
| 		if (sourceFilePath != null) { |  | ||||||
| 			// TODO replace with custom async parser |  | ||||||
| 			await using var sourceStream = new FileStream(sourceFilePath, FileMode.Open, FileAccess.Read, FileShare.Read); |  | ||||||
| 			properties.Load(sourceStream, Encoding); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		foreach (var (key, value) in overriddenProperties) { |  | ||||||
| 			properties[key] = value; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		await using var targetStream = new FileStream(targetFilePath, FileMode.Create, FileAccess.Write, FileShare.Read); |  | ||||||
| 		await using var targetWriter = new StreamWriter(targetStream, Encoding); |  | ||||||
|  |  | ||||||
| 		await targetWriter.WriteLineAsync("# Properties"); |  | ||||||
|  |  | ||||||
| 		foreach (var (key, value) in properties) { |  | ||||||
| 			await WriteProperty(targetWriter, key, value); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private static async Task WriteProperty(StreamWriter writer, string key, string value) { |  | ||||||
| 		await WritePropertyComponent(writer, key, escapeSpaces: true); |  | ||||||
| 		await writer.WriteAsync('='); |  | ||||||
| 		await WritePropertyComponent(writer, value, escapeSpaces: false); |  | ||||||
| 		await writer.WriteLineAsync(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private static async Task WritePropertyComponent(TextWriter writer, string component, bool escapeSpaces) { |  | ||||||
| 		for (int index = 0; index < component.Length; index++) { |  | ||||||
| 			var c = component[index]; |  | ||||||
| 			switch (c) { |  | ||||||
| 				case '\\': |  | ||||||
| 				case '#': |  | ||||||
| 				case '!': |  | ||||||
| 				case '=': |  | ||||||
| 				case ':': |  | ||||||
| 				case ' ' when escapeSpaces || index == 0: |  | ||||||
| 					await writer.WriteAsync('\\'); |  | ||||||
| 					await writer.WriteAsync(c); |  | ||||||
| 					break; |  | ||||||
| 				case var _ when c > 31 && c < 127: |  | ||||||
| 					await writer.WriteAsync(c); |  | ||||||
| 					break; |  | ||||||
| 				case '\t': |  | ||||||
| 					await writer.WriteAsync("\\t"); |  | ||||||
| 					break; |  | ||||||
| 				case '\n': |  | ||||||
| 					await writer.WriteAsync("\\n"); |  | ||||||
| 					break; |  | ||||||
| 				case '\r': |  | ||||||
| 					await writer.WriteAsync("\\r"); |  | ||||||
| 					break; |  | ||||||
| 				case '\f': |  | ||||||
| 					await writer.WriteAsync("\\f"); |  | ||||||
| 					break; |  | ||||||
| 				default: |  | ||||||
| 					await writer.WriteAsync("\\u"); |  | ||||||
| 					await writer.WriteAsync(((int) c).ToString("X4")); |  | ||||||
| 					break; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -79,9 +79,10 @@ public sealed class JavaRuntimeDiscovery { | |||||||
| 			WorkingDirectory = Path.GetDirectoryName(javaExecutablePath), | 			WorkingDirectory = Path.GetDirectoryName(javaExecutablePath), | ||||||
| 			Arguments = "-XshowSettings:properties -version", | 			Arguments = "-XshowSettings:properties -version", | ||||||
| 			RedirectStandardInput = false, | 			RedirectStandardInput = false, | ||||||
| 			RedirectStandardOutput = false, | 			RedirectStandardOutput = true, | ||||||
| 			RedirectStandardError = true, | 			RedirectStandardError = true, | ||||||
| 			UseShellExecute = false | 			UseShellExecute = false, | ||||||
|  | 			CreateNoWindow = false | ||||||
| 		}; | 		}; | ||||||
|  |  | ||||||
| 		var process = new Process { StartInfo = startInfo }; | 		var process = new Process { StartInfo = startInfo }; | ||||||
|   | |||||||
| @@ -1,4 +1,5 @@ | |||||||
| using System.Collections.ObjectModel; | using System.Collections.Immutable; | ||||||
|  | using System.Collections.ObjectModel; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Java; | namespace Phantom.Agent.Minecraft.Java; | ||||||
|  |  | ||||||
| @@ -6,12 +7,12 @@ sealed class JvmArgumentBuilder { | |||||||
| 	private readonly JvmProperties basicProperties; | 	private readonly JvmProperties basicProperties; | ||||||
| 	private readonly List<string> customArguments = new (); | 	private readonly List<string> customArguments = new (); | ||||||
|  |  | ||||||
| 	public JvmArgumentBuilder(JvmProperties basicProperties) { | 	public JvmArgumentBuilder(JvmProperties basicProperties, ImmutableArray<string> customArguments) { | ||||||
| 		this.basicProperties = basicProperties; | 		this.basicProperties = basicProperties; | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public void Add(string argument) { | 		foreach (var jvmArgument in customArguments) { | ||||||
| 		customArguments.Add(argument); | 			this.customArguments.Add(jvmArgument); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void AddProperty(string key, string value) { | 	public void AddProperty(string key, string value) { | ||||||
| @@ -23,8 +24,8 @@ sealed class JvmArgumentBuilder { | |||||||
| 			target.Add(property); | 			target.Add(property); | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// In case of duplicate JVM arguments, typically the last one wins. |  | ||||||
| 		target.Add("-Xms" + basicProperties.InitialHeapMegabytes + "M"); | 		target.Add("-Xms" + basicProperties.InitialHeapMegabytes + "M"); | ||||||
| 		target.Add("-Xmx" + basicProperties.MaximumHeapMegabytes + "M"); | 		target.Add("-Xmx" + basicProperties.MaximumHeapMegabytes + "M"); | ||||||
|  | 		target.Add("-Xrs"); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,8 +1,10 @@ | |||||||
| using System.Text; | using System.Text; | ||||||
|  | using Kajabity.Tools.Java; | ||||||
| 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; | ||||||
| using Phantom.Utils.Processes; | using Phantom.Common.Minecraft; | ||||||
|  | using Phantom.Utils.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Launcher; | namespace Phantom.Agent.Minecraft.Launcher; | ||||||
| @@ -21,6 +23,10 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 			return new LaunchResult.InvalidJavaRuntime(); | 			return new LaunchResult.InvalidJavaRuntime(); | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		if (JvmArgumentsHelper.Validate(instanceProperties.JvmArguments) != null) { | ||||||
|  | 			return new LaunchResult.InvalidJvmArguments(); | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		var vanillaServerJarPath = await services.ServerExecutables.DownloadAndGetPath(instanceProperties.LaunchProperties.ServerDownloadInfo, MinecraftVersion, downloadProgressEventHandler, cancellationToken); | 		var vanillaServerJarPath = await services.ServerExecutables.DownloadAndGetPath(instanceProperties.LaunchProperties.ServerDownloadInfo, MinecraftVersion, downloadProgressEventHandler, cancellationToken); | ||||||
| 		if (vanillaServerJarPath == null) { | 		if (vanillaServerJarPath == null) { | ||||||
| 			return new LaunchResult.CouldNotDownloadMinecraftServer(); | 			return new LaunchResult.CouldNotDownloadMinecraftServer(); | ||||||
| @@ -56,8 +62,16 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 			UseShellExecute = false | 			UseShellExecute = false | ||||||
| 		}; | 		}; | ||||||
| 		 | 		 | ||||||
|  | 		var jvmArguments = new JvmArgumentBuilder(instanceProperties.JvmProperties, instanceProperties.JvmArguments); | ||||||
|  | 		CustomizeJvmArguments(jvmArguments); | ||||||
|  |  | ||||||
| 		var processArguments = processConfigurator.ArgumentList; | 		var processArguments = processConfigurator.ArgumentList; | ||||||
| 		PrepareJvmArguments(serverJar).Build(processArguments); | 		jvmArguments.Build(processArguments); | ||||||
|  | 		 | ||||||
|  | 		foreach (var extraArgument in serverJar.ExtraArgs) { | ||||||
|  | 			processArguments.Add(extraArgument); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
| 		processArguments.Add("-jar"); | 		processArguments.Add("-jar"); | ||||||
| 		processArguments.Add(serverJar.FilePath); | 		processArguments.Add(serverJar.FilePath); | ||||||
| 		processArguments.Add("nogui"); | 		processArguments.Add("nogui"); | ||||||
| @@ -82,21 +96,6 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 		return new LaunchResult.Success(instanceProcess); | 		return new LaunchResult.Success(instanceProcess); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private JvmArgumentBuilder PrepareJvmArguments(ServerJarInfo serverJar) { |  | ||||||
| 		var builder = new JvmArgumentBuilder(instanceProperties.JvmProperties); |  | ||||||
|  |  | ||||||
| 		foreach (string argument in instanceProperties.JvmArguments) { |  | ||||||
| 			builder.Add(argument); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		foreach (var argument in serverJar.ExtraArgs) { |  | ||||||
| 			builder.Add(argument); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		CustomizeJvmArguments(builder); |  | ||||||
| 		return builder; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private protected virtual void CustomizeJvmArguments(JvmArgumentBuilder arguments) {} | 	private protected virtual void CustomizeJvmArguments(JvmArgumentBuilder arguments) {} | ||||||
|  |  | ||||||
| 	private protected virtual Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { | 	private protected virtual Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { | ||||||
| @@ -109,8 +108,21 @@ public abstract class BaseLauncher : IServerLauncher { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static async Task UpdateServerProperties(InstanceProperties instanceProperties) { | 	private static async Task UpdateServerProperties(InstanceProperties instanceProperties) { | ||||||
| 		var serverPropertiesEditor = new JavaPropertiesFileEditor(); | 		var serverPropertiesFilePath = Path.Combine(instanceProperties.InstanceFolder, "server.properties"); | ||||||
| 		instanceProperties.ServerProperties.SetTo(serverPropertiesEditor); | 		var serverPropertiesData = new JavaProperties(); | ||||||
| 		await serverPropertiesEditor.EditOrCreate(Path.Combine(instanceProperties.InstanceFolder, "server.properties")); |  | ||||||
|  | 		await using var fileStream = new FileStream(serverPropertiesFilePath, FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.None); | ||||||
|  | 		try { | ||||||
|  | 			serverPropertiesData.Load(fileStream); | ||||||
|  | 		} catch (ParseException e) { | ||||||
|  | 			throw new Exception("Could not parse server.properties file: " + serverPropertiesFilePath, e); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		instanceProperties.ServerProperties.SetTo(serverPropertiesData); | ||||||
|  |  | ||||||
|  | 		fileStream.Seek(0L, SeekOrigin.Begin); | ||||||
|  | 		fileStream.SetLength(0L); | ||||||
|  | 		 | ||||||
|  | 		serverPropertiesData.Store(fileStream, true); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -9,6 +9,8 @@ public abstract record LaunchResult { | |||||||
|  |  | ||||||
| 	public sealed record InvalidJavaRuntime : LaunchResult; | 	public sealed record InvalidJavaRuntime : LaunchResult; | ||||||
| 	 | 	 | ||||||
|  | 	public sealed record InvalidJvmArguments : LaunchResult; | ||||||
|  |  | ||||||
| 	public sealed record CouldNotDownloadMinecraftServer : LaunchResult; | 	public sealed record CouldNotDownloadMinecraftServer : LaunchResult; | ||||||
| 	 | 	 | ||||||
| 	public sealed record CouldNotPrepareMinecraftServerLauncher : LaunchResult; | 	public sealed record CouldNotPrepareMinecraftServerLauncher : LaunchResult; | ||||||
|   | |||||||
| @@ -0,0 +1,12 @@ | |||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Agent.Minecraft.Java; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Minecraft.Launcher.Types;  | ||||||
|  |  | ||||||
|  | public class ForgeLauncher : BaseLauncher { | ||||||
|  | 	public ForgeLauncher(InstanceProperties instanceProperties) : base(instanceProperties) {} | ||||||
|  | 	 | ||||||
|  | 	private protected override void CustomizeJvmArguments(JvmArgumentBuilder arguments) { | ||||||
|  | 		arguments.AddProperty("terminal.ansi", "true"); // TODO | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,6 +1,7 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
| @@ -12,7 +13,11 @@ | |||||||
|   <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="..\..\Common\Phantom.Common.Logging\Phantom.Common.Logging.csproj" /> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> |     <ProjectReference Include="..\..\Common\Phantom.Common.Minecraft\Phantom.Common.Minecraft.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Collections\Phantom.Utils.Collections.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Cryptography\Phantom.Utils.Cryptography.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.IO\Phantom.Utils.IO.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Runtime\Phantom.Utils.Runtime.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|  |  | ||||||
| </Project> | </Project> | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| using Phantom.Agent.Minecraft.Java; | using Kajabity.Tools.Java; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Properties;  | namespace Phantom.Agent.Minecraft.Properties;  | ||||||
|  |  | ||||||
| @@ -12,7 +12,7 @@ abstract class MinecraftServerProperty<T> { | |||||||
| 	protected abstract T Read(string value); | 	protected abstract T Read(string value); | ||||||
| 	protected abstract string Write(T value); | 	protected abstract string Write(T value); | ||||||
|  |  | ||||||
| 	public void Set(JavaPropertiesFileEditor properties, T value) { | 	public void Set(JavaProperties properties, T value) { | ||||||
| 		properties.Set(key, Write(value)); | 		properties.SetProperty(key, Write(value)); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| using Phantom.Agent.Minecraft.Java; | using Kajabity.Tools.Java; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Minecraft.Properties; | namespace Phantom.Agent.Minecraft.Properties; | ||||||
|  |  | ||||||
| @@ -7,7 +7,7 @@ public sealed record ServerProperties( | |||||||
| 	ushort RconPort, | 	ushort RconPort, | ||||||
| 	bool EnableRcon = true | 	bool EnableRcon = true | ||||||
| ) { | ) { | ||||||
| 	internal void SetTo(JavaPropertiesFileEditor properties) { | 	internal void SetTo(JavaProperties 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); | ||||||
|   | |||||||
| @@ -10,7 +10,7 @@ public sealed partial class MinecraftServerExecutables { | |||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<MinecraftServerExecutables>(); | 	private static readonly ILogger Logger = PhantomLogger.Create<MinecraftServerExecutables>(); | ||||||
|  |  | ||||||
| 	[GeneratedRegex(@"[^a-zA-Z0-9_\-\.]", RegexOptions.Compiled)] | 	[GeneratedRegex(@"[^a-zA-Z0-9_\-\.]", RegexOptions.Compiled)] | ||||||
| 	private static partial Regex SanitizePathRegex(); | 	private static partial Regex VersionFolderSanitizeRegex(); | ||||||
|  |  | ||||||
| 	private readonly string basePath; | 	private readonly string basePath; | ||||||
| 	private readonly Dictionary<string, MinecraftServerExecutableDownloader> runningDownloadersByVersion = new (); | 	private readonly Dictionary<string, MinecraftServerExecutableDownloader> runningDownloadersByVersion = new (); | ||||||
| @@ -20,7 +20,7 @@ public sealed partial class MinecraftServerExecutables { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	internal async Task<string?> DownloadAndGetPath(FileDownloadInfo? fileDownloadInfo, string minecraftVersion, EventHandler<DownloadProgressEventArgs> progressEventHandler, CancellationToken cancellationToken) { | 	internal async Task<string?> DownloadAndGetPath(FileDownloadInfo? fileDownloadInfo, string minecraftVersion, EventHandler<DownloadProgressEventArgs> progressEventHandler, CancellationToken cancellationToken) { | ||||||
| 		string serverExecutableFolderPath = Path.Combine(basePath, SanitizePathRegex().IsMatch(minecraftVersion) ? SanitizePathRegex().Replace(minecraftVersion, "_") : minecraftVersion); | 		string serverExecutableFolderPath = Path.Combine(basePath, VersionFolderSanitizeRegex().Replace(minecraftVersion, "_")); | ||||||
| 		string serverExecutableFilePath = Path.Combine(serverExecutableFolderPath, "server.jar"); | 		string serverExecutableFilePath = Path.Combine(serverExecutableFolderPath, "server.jar"); | ||||||
|  |  | ||||||
| 		if (File.Exists(serverExecutableFilePath)) { | 		if (File.Exists(serverExecutableFilePath)) { | ||||||
|   | |||||||
| @@ -78,7 +78,7 @@ public sealed class ServerStatusProtocol { | |||||||
| 				return null; | 				return null; | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			string onlinePlayerCountStr = Encoding.BigEndianUnicode.GetString(messageBuffer.AsSpan((separator1 + 1)..(separator2 - 1))); | 			string onlinePlayerCountStr = Encoding.BigEndianUnicode.GetString(messageBuffer[(separator1 + 1)..(separator2 - 1)]); | ||||||
| 			if (!int.TryParse(onlinePlayerCountStr, out int onlinePlayerCount)) { | 			if (!int.TryParse(onlinePlayerCountStr, out int onlinePlayerCount)) { | ||||||
| 				logger.Error("Could not parse online player count in response from server: {OnlinePlayerCount}.", onlinePlayerCountStr); | 				logger.Error("Could not parse online player count in response from server: {OnlinePlayerCount}.", onlinePlayerCountStr); | ||||||
| 				return null; | 				return null; | ||||||
|   | |||||||
| @@ -1,25 +0,0 @@ | |||||||
| using Phantom.Common.Logging; |  | ||||||
| using Phantom.Common.Messages.Agent; |  | ||||||
| using Phantom.Utils.Rpc; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; |  | ||||||
|  |  | ||||||
| public sealed class ControllerConnection { |  | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ControllerConnection)); |  | ||||||
|  |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; |  | ||||||
| 	 |  | ||||||
| 	public ControllerConnection(RpcConnectionToServer<IMessageToControllerListener> connection) { |  | ||||||
| 		this.connection = connection; |  | ||||||
| 		Logger.Information("Connection ready."); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public Task Send<TMessage>(TMessage message) where TMessage : IMessageToController { |  | ||||||
| 		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,5 @@ | |||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Messages.ToServer; | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Rpc; |  | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; | namespace Phantom.Agent.Rpc; | ||||||
| @@ -11,10 +9,10 @@ sealed class KeepAliveLoop { | |||||||
|  |  | ||||||
| 	private static readonly TimeSpan KeepAliveInterval = TimeSpan.FromSeconds(10); | 	private static readonly TimeSpan KeepAliveInterval = TimeSpan.FromSeconds(10); | ||||||
|  |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | 	private readonly RpcServerConnection connection; | ||||||
| 	private readonly CancellationTokenSource cancellationTokenSource = new (); | 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||||
|  |  | ||||||
| 	public KeepAliveLoop(RpcConnectionToServer<IMessageToControllerListener> connection) { | 	public KeepAliveLoop(RpcServerConnection connection) { | ||||||
| 		this.connection = connection; | 		this.connection = connection; | ||||||
| 		Task.Run(Run); | 		Task.Run(Run); | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,12 +1,13 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages.Agent\Phantom.Common.Messages.Agent.csproj" /> |     <ProjectReference Include="..\..\Common\Phantom.Common.Messages\Phantom.Common.Messages.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|  |  | ||||||
| </Project> | </Project> | ||||||
|   | |||||||
| @@ -1,38 +0,0 @@ | |||||||
| using NetMQ; |  | ||||||
| using NetMQ.Sockets; |  | ||||||
| using Phantom.Common.Data.Agent; |  | ||||||
| using Phantom.Common.Messages.Agent; |  | ||||||
| using Phantom.Common.Messages.Agent.BiDirectional; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Rpc; |  | ||||||
| using Phantom.Utils.Rpc.Sockets; |  | ||||||
| using Phantom.Utils.Tasks; |  | ||||||
| using Serilog; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Rpc; |  | ||||||
|  |  | ||||||
| public sealed class RpcClientRuntime : RpcClientRuntime<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> { |  | ||||||
| 	public static Task Launch(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, AgentInfo agentInfo, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { |  | ||||||
| 		return new RpcClientRuntime(socket, messageListener, disconnectSemaphore, receiveCancellationToken).Launch(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private RpcClientRuntime(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(socket, messageListener, disconnectSemaphore, receiveCancellationToken) {} |  | ||||||
|  |  | ||||||
| 	protected override void RunWithConnection(ClientSocket socket, RpcConnectionToServer<IMessageToControllerListener> connection, ILogger logger, TaskManager taskManager) { |  | ||||||
| 		var keepAliveLoop = new KeepAliveLoop(connection); |  | ||||||
| 		try { |  | ||||||
| 			base.RunWithConnection(socket, connection, logger, taskManager); |  | ||||||
| 		} finally { |  | ||||||
| 			keepAliveLoop.Cancel(); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	protected override async Task Disconnect(ClientSocket socket, ILogger logger) { |  | ||||||
| 		var unregisterMessageBytes = AgentMessageRegistries.ToController.Write(new UnregisterAgentMessage()).ToArray(); |  | ||||||
| 		try { |  | ||||||
| 			await socket.SendAsync(unregisterMessageBytes).AsTask().WaitAsync(TimeSpan.FromSeconds(5), CancellationToken.None); |  | ||||||
| 		} catch (TimeoutException) { |  | ||||||
| 			logger.Error("Timed out communicating agent shutdown with the controller."); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										107
									
								
								Agent/Phantom.Agent.Rpc/RpcLauncher.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										107
									
								
								Agent/Phantom.Agent.Rpc/RpcLauncher.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,107 @@ | |||||||
|  | using NetMQ; | ||||||
|  | using NetMQ.Sockets; | ||||||
|  | using Phantom.Common.Data.Agent; | ||||||
|  | using Phantom.Common.Messages; | ||||||
|  | using Phantom.Common.Messages.BiDirectional; | ||||||
|  | using Phantom.Common.Messages.ToServer; | ||||||
|  | using Phantom.Utils.Rpc; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  | using Phantom.Utils.Runtime; | ||||||
|  | using Serilog; | ||||||
|  | using Serilog.Events; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Rpc; | ||||||
|  |  | ||||||
|  | public sealed class RpcLauncher : RpcRuntime<ClientSocket> { | ||||||
|  | 	public static async Task Launch(RpcConfiguration config, AgentAuthToken authToken, AgentInfo agentInfo, Func<RpcServerConnection, IMessageToAgentListener> listenerFactory, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { | ||||||
|  | 		var socket = new ClientSocket(); | ||||||
|  | 		var options = socket.Options; | ||||||
|  |  | ||||||
|  | 		options.CurveServerCertificate = config.ServerCertificate; | ||||||
|  | 		options.CurveCertificate = new NetMQCertificate(); | ||||||
|  | 		options.HelloMessage = MessageRegistries.ToServer.Write(new RegisterAgentMessage(authToken, agentInfo)).ToArray(); | ||||||
|  |  | ||||||
|  | 		await new RpcLauncher(config, socket, agentInfo.Guid, listenerFactory, disconnectSemaphore, receiveCancellationToken).Launch(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private readonly RpcConfiguration config; | ||||||
|  | 	private readonly Guid agentGuid; | ||||||
|  | 	private readonly Func<RpcServerConnection, IMessageToAgentListener> messageListenerFactory; | ||||||
|  |  | ||||||
|  | 	private readonly SemaphoreSlim disconnectSemaphore; | ||||||
|  | 	private readonly CancellationToken receiveCancellationToken; | ||||||
|  |  | ||||||
|  | 	private RpcLauncher(RpcConfiguration config, ClientSocket socket, Guid agentGuid, Func<RpcServerConnection, IMessageToAgentListener> messageListenerFactory, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(config, socket) { | ||||||
|  | 		this.config = config; | ||||||
|  | 		this.agentGuid = agentGuid; | ||||||
|  | 		this.messageListenerFactory = messageListenerFactory; | ||||||
|  | 		this.disconnectSemaphore = disconnectSemaphore; | ||||||
|  | 		this.receiveCancellationToken = receiveCancellationToken; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	protected override void Connect(ClientSocket socket) { | ||||||
|  | 		var logger = config.RuntimeLogger; | ||||||
|  | 		var url = config.TcpUrl; | ||||||
|  |  | ||||||
|  | 		logger.Information("Starting ZeroMQ client and connecting to {Url}...", url); | ||||||
|  | 		socket.Connect(url); | ||||||
|  | 		logger.Information("ZeroMQ client ready."); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	protected override void Run(ClientSocket socket, MessageReplyTracker replyTracker, TaskManager taskManager) { | ||||||
|  | 		var connection = new RpcServerConnection(socket, replyTracker); | ||||||
|  | 		ServerMessaging.SetCurrentConnection(connection); | ||||||
|  | 		 | ||||||
|  | 		var logger = config.RuntimeLogger; | ||||||
|  | 		var handler = new MessageToAgentHandler(messageListenerFactory(connection), logger, taskManager, receiveCancellationToken); | ||||||
|  | 		var keepAliveLoop = new KeepAliveLoop(connection); | ||||||
|  |  | ||||||
|  | 		try { | ||||||
|  | 			while (!receiveCancellationToken.IsCancellationRequested) { | ||||||
|  | 				var data = socket.Receive(receiveCancellationToken); | ||||||
|  | 				 | ||||||
|  | 				LogMessageType(logger, data); | ||||||
|  | 				 | ||||||
|  | 				if (data.Length > 0) { | ||||||
|  | 					MessageRegistries.ToAgent.Handle(data, handler); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			// Ignore. | ||||||
|  | 		} finally { | ||||||
|  | 			logger.Debug("ZeroMQ client stopped receiving messages."); | ||||||
|  |  | ||||||
|  | 			disconnectSemaphore.Wait(CancellationToken.None); | ||||||
|  | 			keepAliveLoop.Cancel(); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private static void LogMessageType(ILogger logger, ReadOnlyMemory<byte> data) { | ||||||
|  | 		if (!logger.IsEnabled(LogEventLevel.Verbose)) { | ||||||
|  | 			return; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (data.Length > 0 && MessageRegistries.ToAgent.TryGetType(data, out var type)) { | ||||||
|  | 			logger.Verbose("Received {MessageType} ({Bytes} B) from server.", type.Name, data.Length); | ||||||
|  | 		} | ||||||
|  | 		else { | ||||||
|  | 			logger.Verbose("Received {Bytes} B message from server.", data.Length); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	protected override async Task Disconnect() { | ||||||
|  | 		var unregisterTimeoutTask = Task.Delay(TimeSpan.FromSeconds(5), CancellationToken.None); | ||||||
|  | 		var finishedTask = await Task.WhenAny(ServerMessaging.Send(new UnregisterAgentMessage(agentGuid)), unregisterTimeoutTask); | ||||||
|  | 		if (finishedTask == unregisterTimeoutTask) { | ||||||
|  | 			config.RuntimeLogger.Error("Timed out communicating agent shutdown with the server."); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private sealed class MessageToAgentHandler : MessageHandler<IMessageToAgentListener> { | ||||||
|  | 		public MessageToAgentHandler(IMessageToAgentListener listener, ILogger logger, TaskManager taskManager, CancellationToken cancellationToken) : base(listener, logger, taskManager, cancellationToken) {} | ||||||
|  | 		 | ||||||
|  | 		protected override Task SendReply(uint sequenceId, byte[] serializedReply) { | ||||||
|  | 			return ServerMessaging.Send(new ReplyMessage(sequenceId, serializedReply)); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										41
									
								
								Agent/Phantom.Agent.Rpc/RpcServerConnection.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								Agent/Phantom.Agent.Rpc/RpcServerConnection.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,41 @@ | |||||||
|  | using NetMQ; | ||||||
|  | using NetMQ.Sockets; | ||||||
|  | using Phantom.Common.Messages; | ||||||
|  | using Phantom.Common.Messages.BiDirectional; | ||||||
|  | using Phantom.Utils.Rpc.Message; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Rpc;  | ||||||
|  |  | ||||||
|  | public sealed class RpcServerConnection { | ||||||
|  | 	private readonly ClientSocket socket; | ||||||
|  | 	private readonly MessageReplyTracker replyTracker; | ||||||
|  |  | ||||||
|  | 	internal RpcServerConnection(ClientSocket socket, MessageReplyTracker replyTracker) { | ||||||
|  | 		this.socket = socket; | ||||||
|  | 		this.replyTracker = replyTracker; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	internal async Task Send<TMessage>(TMessage message) where TMessage : IMessageToServer { | ||||||
|  | 		var bytes = MessageRegistries.ToServer.Write(message).ToArray(); | ||||||
|  | 		if (bytes.Length > 0) { | ||||||
|  | 			await socket.SendAsync(bytes); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	internal async Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToServer<TReply> where TReply : class { | ||||||
|  | 		var sequenceId = replyTracker.RegisterReply(); | ||||||
|  | 		 | ||||||
|  | 		var bytes = MessageRegistries.ToServer.Write<TMessage, TReply>(sequenceId, message).ToArray(); | ||||||
|  | 		if (bytes.Length == 0) { | ||||||
|  | 			replyTracker.ForgetReply(sequenceId); | ||||||
|  | 			return null; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		await socket.SendAsync(bytes); | ||||||
|  | 		return await replyTracker.WaitForReply<TReply>(sequenceId, waitForReplyTime, waitForReplyCancellationToken); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Receive(ReplyMessage message) { | ||||||
|  | 		replyTracker.ReceiveReply(message.SequenceId, message.SerializedReply); | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										34
									
								
								Agent/Phantom.Agent.Rpc/ServerMessaging.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								Agent/Phantom.Agent.Rpc/ServerMessaging.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | |||||||
|  | using Phantom.Common.Logging; | ||||||
|  | using Phantom.Common.Messages; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Rpc; | ||||||
|  |  | ||||||
|  | public static class ServerMessaging { | ||||||
|  | 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ServerMessaging)); | ||||||
|  | 	 | ||||||
|  | 	private static RpcServerConnection? CurrentConnection { get; set; } | ||||||
|  | 	private static RpcServerConnection CurrentConnectionOrThrow => CurrentConnection ?? throw new InvalidOperationException("Server connection not ready."); | ||||||
|  | 	 | ||||||
|  | 	private static readonly object SetCurrentConnectionLock = new (); | ||||||
|  |  | ||||||
|  | 	internal static void SetCurrentConnection(RpcServerConnection connection) { | ||||||
|  | 		lock (SetCurrentConnectionLock) { | ||||||
|  | 			if (CurrentConnection != null) { | ||||||
|  | 				throw new InvalidOperationException("Server connection can only be set once."); | ||||||
|  | 			} | ||||||
|  | 			 | ||||||
|  | 			CurrentConnection = connection; | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		Logger.Information("Server connection ready."); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public static Task Send<TMessage>(TMessage message) where TMessage : IMessageToServer { | ||||||
|  | 		return CurrentConnectionOrThrow.Send(message); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public static Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToServer<TReply> where TReply : class { | ||||||
|  | 		return CurrentConnectionOrThrow.Send<TMessage, TReply>(message, waitForReplyTime, waitForReplyCancellationToken); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,3 +0,0 @@ | |||||||
| namespace Phantom.Agent.Services; |  | ||||||
|  |  | ||||||
| public readonly record struct AgentServiceConfiguration(int MaxConcurrentCompressionTasks); |  | ||||||
| @@ -1,10 +1,9 @@ | |||||||
| using Phantom.Agent.Minecraft.Java; | using Phantom.Agent.Minecraft.Java; | ||||||
| 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.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services; | namespace Phantom.Agent.Services; | ||||||
| @@ -19,12 +18,12 @@ public sealed class AgentServices { | |||||||
| 	internal JavaRuntimeRepository JavaRuntimeRepository { get; } | 	internal JavaRuntimeRepository JavaRuntimeRepository { get; } | ||||||
| 	internal InstanceSessionManager InstanceSessionManager { get; } | 	internal InstanceSessionManager InstanceSessionManager { get; } | ||||||
|  |  | ||||||
| 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders, AgentServiceConfiguration serviceConfiguration, ControllerConnection controllerConnection) { | 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders) { | ||||||
| 		this.AgentFolders = agentFolders; | 		this.AgentFolders = agentFolders; | ||||||
| 		this.TaskManager = new TaskManager(PhantomLogger.Create<TaskManager, AgentServices>()); | 		this.TaskManager = new TaskManager(PhantomLogger.Create<TaskManager, AgentServices>()); | ||||||
| 		this.BackupManager = new BackupManager(agentFolders, serviceConfiguration.MaxConcurrentCompressionTasks); | 		this.BackupManager = new BackupManager(agentFolders); | ||||||
| 		this.JavaRuntimeRepository = new JavaRuntimeRepository(); | 		this.JavaRuntimeRepository = new JavaRuntimeRepository(); | ||||||
| 		this.InstanceSessionManager = new InstanceSessionManager(controllerConnection, agentInfo, agentFolders, JavaRuntimeRepository, TaskManager, BackupManager); | 		this.InstanceSessionManager = new InstanceSessionManager(agentInfo, agentFolders, JavaRuntimeRepository, TaskManager, BackupManager); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task Initialize() { | 	public async Task Initialize() { | ||||||
| @@ -36,10 +35,10 @@ public sealed class AgentServices { | |||||||
| 	public async Task Shutdown() { | 	public async Task Shutdown() { | ||||||
| 		Logger.Information("Stopping services..."); | 		Logger.Information("Stopping services..."); | ||||||
| 		 | 		 | ||||||
| 		await InstanceSessionManager.DisposeAsync(); | 		await InstanceSessionManager.StopAll(); | ||||||
| 		await TaskManager.Stop(); | 		InstanceSessionManager.Dispose(); | ||||||
| 		 | 		 | ||||||
| 		BackupManager.Dispose(); | 		await TaskManager.Stop(); | ||||||
| 		 | 		 | ||||||
| 		Logger.Information("Services stopped."); | 		Logger.Information("Services stopped."); | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Processes; | using Phantom.Utils.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups;  | namespace Phantom.Agent.Services.Backups;  | ||||||
| @@ -47,8 +47,10 @@ static class BackupCompressor { | |||||||
| 				Quality, | 				Quality, | ||||||
| 				Memory, | 				Memory, | ||||||
| 				Threads, | 				Threads, | ||||||
|  | 				"-c", | ||||||
| 				"--rm", | 				"--rm", | ||||||
| 				"--no-progress", | 				"--no-progress", | ||||||
|  | 				"-c", | ||||||
| 				"-o", destinationFilePath, | 				"-o", destinationFilePath, | ||||||
| 				"--", sourceFilePath | 				"--", sourceFilePath | ||||||
| 			} | 			} | ||||||
|   | |||||||
| @@ -5,34 +5,44 @@ using Serilog; | |||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
|  |  | ||||||
| sealed class BackupManager : IDisposable { | sealed class BackupManager { | ||||||
| 	private readonly string destinationBasePath; | 	private readonly string destinationBasePath; | ||||||
| 	private readonly string temporaryBasePath; | 	private readonly string temporaryBasePath; | ||||||
| 	private readonly SemaphoreSlim compressionSemaphore; |  | ||||||
|  |  | ||||||
| 	public BackupManager(AgentFolders agentFolders, int maxConcurrentCompressionTasks) { | 	public BackupManager(AgentFolders agentFolders) { | ||||||
| 		this.destinationBasePath = agentFolders.BackupsFolderPath; | 		this.destinationBasePath = agentFolders.BackupsFolderPath; | ||||||
| 		this.temporaryBasePath = Path.Combine(agentFolders.TemporaryFolderPath, "backups"); | 		this.temporaryBasePath = Path.Combine(agentFolders.TemporaryFolderPath, "backups"); | ||||||
| 		this.compressionSemaphore = new SemaphoreSlim(maxConcurrentCompressionTasks, maxConcurrentCompressionTasks); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public Task<BackupCreationResult> CreateBackup(string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | 	public async Task<BackupCreationResult> CreateBackup(string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||||
| 		return new BackupCreator(this, loggerName, process, cancellationToken).CreateBackup(); | 		try { | ||||||
| 	} | 			if (!await process.BackupSemaphore.Wait(TimeSpan.FromSeconds(1), cancellationToken)) { | ||||||
|  | 				return new BackupCreationResult(BackupCreationResultKind.BackupAlreadyRunning); | ||||||
|  | 			} | ||||||
|  | 		} catch (ObjectDisposedException) { | ||||||
|  | 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 	public void Dispose() { | 		try { | ||||||
| 		compressionSemaphore.Dispose(); | 			return await new BackupCreator(destinationBasePath, temporaryBasePath, loggerName, process, cancellationToken).CreateBackup(); | ||||||
|  | 		} finally { | ||||||
|  | 			process.BackupSemaphore.Release(); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private sealed class BackupCreator { | 	private sealed class BackupCreator { | ||||||
| 		private readonly BackupManager manager; | 		private readonly string destinationBasePath; | ||||||
|  | 		private readonly string temporaryBasePath; | ||||||
| 		private readonly string loggerName; | 		private readonly string loggerName; | ||||||
| 		private readonly ILogger logger; | 		private readonly ILogger logger; | ||||||
| 		private readonly InstanceProcess process; | 		private readonly InstanceProcess process; | ||||||
| 		private readonly CancellationToken cancellationToken; | 		private readonly CancellationToken cancellationToken; | ||||||
|  |  | ||||||
| 		public BackupCreator(BackupManager manager, string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | 		public BackupCreator(string destinationBasePath, string temporaryBasePath, string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||||
| 			this.manager = manager; | 			this.destinationBasePath = destinationBasePath; | ||||||
|  | 			this.temporaryBasePath = temporaryBasePath; | ||||||
| 			this.loggerName = loggerName; | 			this.loggerName = loggerName; | ||||||
| 			this.logger = PhantomLogger.Create<BackupManager>(loggerName); | 			this.logger = PhantomLogger.Create<BackupManager>(loggerName); | ||||||
| 			this.process = process; | 			this.process = process; | ||||||
| @@ -62,7 +72,7 @@ sealed class BackupManager : IDisposable { | |||||||
| 			try { | 			try { | ||||||
| 				await dispatcher.DisableAutomaticSaving(); | 				await dispatcher.DisableAutomaticSaving(); | ||||||
| 				await dispatcher.SaveAllChunks(); | 				await dispatcher.SaveAllChunks(); | ||||||
| 				return await new BackupArchiver(manager.destinationBasePath, manager.temporaryBasePath, loggerName, process.InstanceProperties, cancellationToken).ArchiveWorld(resultBuilder); | 				return await new BackupArchiver(destinationBasePath, temporaryBasePath, loggerName, process.InstanceProperties, cancellationToken).ArchiveWorld(resultBuilder); | ||||||
| 			} catch (OperationCanceledException) { | 			} catch (OperationCanceledException) { | ||||||
| 				resultBuilder.Kind = BackupCreationResultKind.BackupCancelled; | 				resultBuilder.Kind = BackupCreationResultKind.BackupCancelled; | ||||||
| 				logger.Warning("Backup creation was cancelled."); | 				logger.Warning("Backup creation was cancelled."); | ||||||
| @@ -75,7 +85,7 @@ sealed class BackupManager : IDisposable { | |||||||
| 				try { | 				try { | ||||||
| 					await dispatcher.EnableAutomaticSaving(); | 					await dispatcher.EnableAutomaticSaving(); | ||||||
| 				} catch (OperationCanceledException) { | 				} catch (OperationCanceledException) { | ||||||
| 					// Ignore. | 					// ignore | ||||||
| 				} 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."); | ||||||
| @@ -84,25 +94,15 @@ sealed class BackupManager : IDisposable { | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		private async Task CompressWorldArchive(string filePath, BackupCreationResult.Builder resultBuilder) { | 		private async Task CompressWorldArchive(string filePath, BackupCreationResult.Builder resultBuilder) { | ||||||
| 			if (!await manager.compressionSemaphore.WaitAsync(TimeSpan.FromSeconds(1), cancellationToken)) { | 			var compressedFilePath = await BackupCompressor.Compress(filePath, cancellationToken); | ||||||
| 				logger.Information("Too many compression tasks running, waiting for one of them to complete..."); | 			if (compressedFilePath == null) { | ||||||
| 				await manager.compressionSemaphore.WaitAsync(cancellationToken); | 				resultBuilder.Warnings |= BackupCreationWarnings.CouldNotCompressWorldArchive; | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			logger.Information("Compressing backup..."); |  | ||||||
| 			try { |  | ||||||
| 				var compressedFilePath = await BackupCompressor.Compress(filePath, cancellationToken); |  | ||||||
| 				if (compressedFilePath == null) { |  | ||||||
| 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotCompressWorldArchive; |  | ||||||
| 				} |  | ||||||
| 			} finally { |  | ||||||
| 				manager.compressionSemaphore.Release(); |  | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		private void LogBackupResult(BackupCreationResult result) { | 		private void LogBackupResult(BackupCreationResult result) { | ||||||
| 			if (result.Kind != BackupCreationResultKind.Success) { | 			if (result.Kind != BackupCreationResultKind.Success) { | ||||||
| 				logger.Warning("Backup failed: {Reason}", DescribeResult(result.Kind)); | 				logger.Warning("Backup failed: {Reason}", result.Kind.ToSentence()); | ||||||
| 				return; | 				return; | ||||||
| 			} | 			} | ||||||
| 			 | 			 | ||||||
| @@ -114,19 +114,5 @@ sealed class BackupManager : IDisposable { | |||||||
| 				logger.Information("Backup finished successfully."); | 				logger.Information("Backup finished successfully."); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		private static string DescribeResult(BackupCreationResultKind kind) { |  | ||||||
| 			return kind switch { |  | ||||||
| 				BackupCreationResultKind.Success                            => "Backup created successfully.", |  | ||||||
| 				BackupCreationResultKind.InstanceNotRunning                 => "Instance is not running.", |  | ||||||
| 				BackupCreationResultKind.BackupCancelled                    => "Backup cancelled.", |  | ||||||
| 				BackupCreationResultKind.BackupAlreadyRunning               => "A backup is already being created.", |  | ||||||
| 				BackupCreationResultKind.BackupFileAlreadyExists            => "Backup with the same name already exists.", |  | ||||||
| 				BackupCreationResultKind.CouldNotCreateBackupFolder         => "Could not create backup folder.", |  | ||||||
| 				BackupCreationResultKind.CouldNotCopyWorldToTemporaryFolder => "Could not copy world to temporary folder.", |  | ||||||
| 				BackupCreationResultKind.CouldNotCreateWorldArchive         => "Could not create world archive.", |  | ||||||
| 				_                                                           => "Unknown error." |  | ||||||
| 			}; |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,11 +1,8 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Agent.Minecraft.Server; | 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.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Runtime; | ||||||
| using Phantom.Utils.Threading; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
|  |  | ||||||
| @@ -15,23 +12,21 @@ sealed class BackupScheduler : CancellableBackgroundTask { | |||||||
| 	private static readonly TimeSpan BackupInterval = TimeSpan.FromMinutes(30); | 	private static readonly TimeSpan BackupInterval = TimeSpan.FromMinutes(30); | ||||||
| 	private static readonly TimeSpan BackupFailureRetryDelay = TimeSpan.FromMinutes(5); | 	private static readonly TimeSpan BackupFailureRetryDelay = TimeSpan.FromMinutes(5); | ||||||
|  |  | ||||||
|  | 	private readonly string loggerName; | ||||||
| 	private readonly BackupManager backupManager; | 	private readonly BackupManager backupManager; | ||||||
| 	private readonly InstanceProcess process; | 	private readonly InstanceProcess process; | ||||||
| 	private readonly IInstanceContext context; |  | ||||||
| 	private readonly SemaphoreSlim backupSemaphore = new (1, 1); |  | ||||||
| 	private readonly int serverPort; | 	private readonly int serverPort; | ||||||
| 	private readonly ServerStatusProtocol serverStatusProtocol; | 	private readonly ServerStatusProtocol serverStatusProtocol; | ||||||
| 	private readonly ManualResetEventSlim serverOutputWhileWaitingForOnlinePlayers = new (); | 	private readonly ManualResetEventSlim serverOutputWhileWaitingForOnlinePlayers = new (); | ||||||
| 	 | 	 | ||||||
| 	public event EventHandler<BackupCreationResult>? BackupCompleted;  | 	public event EventHandler<BackupCreationResult>? BackupCompleted;  | ||||||
|  |  | ||||||
| 	public BackupScheduler(TaskManager taskManager, BackupManager backupManager, InstanceProcess process, IInstanceContext context, int serverPort) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName), taskManager, "Backup scheduler for " + context.ShortName) { | 	public BackupScheduler(TaskManager taskManager, BackupManager backupManager, InstanceProcess process, int serverPort, string loggerName) : base(PhantomLogger.Create<BackupScheduler>(loggerName), taskManager, "Backup scheduler for " + loggerName) { | ||||||
|  | 		this.loggerName = loggerName; | ||||||
| 		this.backupManager = backupManager; | 		this.backupManager = backupManager; | ||||||
| 		this.process = process; | 		this.process = process; | ||||||
| 		this.context = context; |  | ||||||
| 		this.serverPort = serverPort; | 		this.serverPort = serverPort; | ||||||
| 		this.serverStatusProtocol = new ServerStatusProtocol(context.ShortName); | 		this.serverStatusProtocol = new ServerStatusProtocol(loggerName); | ||||||
| 		Start(); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	protected override async Task RunTask() { | 	protected override async Task RunTask() { | ||||||
| @@ -55,17 +50,7 @@ sealed class BackupScheduler : CancellableBackgroundTask { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task<BackupCreationResult> CreateBackup() { | 	private async Task<BackupCreationResult> CreateBackup() { | ||||||
| 		if (!await backupSemaphore.WaitAsync(TimeSpan.FromSeconds(1))) { | 		return await backupManager.CreateBackup(loggerName, process, CancellationToken.None); | ||||||
| 			return new BackupCreationResult(BackupCreationResultKind.BackupAlreadyRunning); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		try { |  | ||||||
| 			var procedure = new BackupInstanceProcedure(backupManager); |  | ||||||
| 			context.EnqueueProcedure(procedure); |  | ||||||
| 			return await procedure.Result; |  | ||||||
| 		} finally { |  | ||||||
| 			backupSemaphore.Release(); |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task WaitForOnlinePlayers() { | 	private async Task WaitForOnlinePlayers() { | ||||||
| @@ -108,9 +93,4 @@ sealed class BackupScheduler : CancellableBackgroundTask { | |||||||
| 			Logger.Debug("Detected server output, signalling to check for online players again."); | 			Logger.Debug("Detected server output, signalling to check for online players again."); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	protected override void Dispose() { |  | ||||||
| 		backupSemaphore.Dispose(); |  | ||||||
| 		serverOutputWhileWaitingForOnlinePlayers.Dispose(); |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| 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.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Backups; | namespace Phantom.Agent.Services.Backups; | ||||||
| @@ -14,9 +14,9 @@ sealed partial class BackupServerCommandDispatcher : IDisposable { | |||||||
| 	private readonly InstanceProcess process; | 	private readonly InstanceProcess process; | ||||||
| 	private readonly CancellationToken cancellationToken; | 	private readonly CancellationToken cancellationToken; | ||||||
|  |  | ||||||
| 	private readonly TaskCompletionSource automaticSavingDisabled = AsyncTasks.CreateCompletionSource(); | 	private readonly TaskCompletionSource automaticSavingDisabled = Tasks.CreateCompletionSource(); | ||||||
| 	private readonly TaskCompletionSource savedTheGame = AsyncTasks.CreateCompletionSource(); | 	private readonly TaskCompletionSource savedTheGame = Tasks.CreateCompletionSource(); | ||||||
| 	private readonly TaskCompletionSource automaticSavingEnabled = AsyncTasks.CreateCompletionSource(); | 	private readonly TaskCompletionSource automaticSavingEnabled = Tasks.CreateCompletionSource(); | ||||||
|  |  | ||||||
| 	public BackupServerCommandDispatcher(ILogger logger, InstanceProcess process, CancellationToken cancellationToken) { | 	public BackupServerCommandDispatcher(ILogger logger, InstanceProcess process, CancellationToken cancellationToken) { | ||||||
| 		this.logger = logger; | 		this.logger = logger; | ||||||
|   | |||||||
| @@ -1,25 +0,0 @@ | |||||||
| 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,21 +1,27 @@ | |||||||
| using Phantom.Agent.Minecraft.Launcher; | using Phantom.Agent.Minecraft.Launcher; | ||||||
| using Phantom.Agent.Services.Instances.Procedures; | using Phantom.Agent.Rpc; | ||||||
| using Phantom.Agent.Services.Instances.States; | 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; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.ToServer; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
| sealed class Instance : IAsyncDisposable { | sealed class Instance : IDisposable { | ||||||
|  | 	private static uint loggerSequenceId = 0; | ||||||
|  |  | ||||||
|  | 	private static string GetLoggerName(Guid guid) { | ||||||
|  | 		var prefix = guid.ToString(); | ||||||
|  | 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref loggerSequenceId); | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	private InstanceServices Services { get; } | 	private InstanceServices Services { get; } | ||||||
| 	 | 	 | ||||||
| 	public InstanceConfiguration Configuration { get; private set; } | 	public InstanceConfiguration Configuration { get; private set; } | ||||||
| 	private IServerLauncher Launcher { get; set; } | 	private IServerLauncher Launcher { get; set; } | ||||||
| 	private readonly SemaphoreSlim configurationSemaphore = new (1, 1); |  | ||||||
|  |  | ||||||
| 	private readonly string shortName; | 	private readonly string shortName; | ||||||
| 	private readonly ILogger logger; | 	private readonly ILogger logger; | ||||||
| @@ -24,14 +30,14 @@ sealed class Instance : IAsyncDisposable { | |||||||
| 	private int statusUpdateCounter; | 	private int statusUpdateCounter; | ||||||
| 	 | 	 | ||||||
| 	private IInstanceState currentState; | 	private IInstanceState currentState; | ||||||
|  | 	private readonly SemaphoreSlim stateTransitioningActionSemaphore = new (1, 1); | ||||||
|  |  | ||||||
| 	public bool IsRunning => currentState is not InstanceNotRunningState; | 	public bool IsRunning => currentState is not InstanceNotRunningState; | ||||||
| 	 | 	 | ||||||
| 	public event EventHandler? IsRunningChanged;  | 	public event EventHandler? IsRunningChanged;  | ||||||
|  |  | ||||||
| 	private readonly InstanceProcedureManager procedureManager; | 	public Instance(InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||||
|  | 		this.shortName = GetLoggerName(configuration.InstanceGuid); | ||||||
| 	public Instance(string shortName, InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { |  | ||||||
| 		this.shortName = shortName; |  | ||||||
| 		this.logger = PhantomLogger.Create<Instance>(shortName); | 		this.logger = PhantomLogger.Create<Instance>(shortName); | ||||||
|  |  | ||||||
| 		this.Services = services; | 		this.Services = services; | ||||||
| @@ -40,8 +46,6 @@ sealed class Instance : IAsyncDisposable { | |||||||
| 		 | 		 | ||||||
| 		this.currentState = new InstanceNotRunningState(); | 		this.currentState = new InstanceNotRunningState(); | ||||||
| 		this.currentStatus = InstanceStatus.NotRunning; | 		this.currentStatus = InstanceStatus.NotRunning; | ||||||
| 		 |  | ||||||
| 		this.procedureManager = new InstanceProcedureManager(this, new Context(this), services.TaskManager); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void TryUpdateStatus(string taskName, Func<Task> getUpdateTask) { | 	private void TryUpdateStatus(string taskName, Func<Task> getUpdateTask) { | ||||||
| @@ -56,25 +60,23 @@ sealed class Instance : IAsyncDisposable { | |||||||
|  |  | ||||||
| 	public void ReportLastStatus() { | 	public void ReportLastStatus() { | ||||||
| 		TryUpdateStatus("Report last status of instance " + shortName, async () => { | 		TryUpdateStatus("Report last status of instance " + shortName, async () => { | ||||||
| 			await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, currentStatus)); | 			await ServerMessaging.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, currentStatus)); | ||||||
| 		}); | 		}); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void ReportAndSetStatus(IInstanceStatus status) { | 	private void ReportAndSetStatus(IInstanceStatus status) { | ||||||
| 		TryUpdateStatus("Report status of instance " + shortName + " as " + status.GetType().Name, async () => { | 		TryUpdateStatus("Report status of instance " + shortName + " as " + status.GetType().Name, async () => { | ||||||
| 			if (status != currentStatus) { | 			currentStatus = status; | ||||||
| 				currentStatus = status; | 			await ServerMessaging.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, status)); | ||||||
| 				await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, status)); |  | ||||||
| 			} |  | ||||||
| 		}); | 		}); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void ReportEvent(IInstanceEvent instanceEvent) { | 	private void ReportEvent(IInstanceEvent instanceEvent) { | ||||||
| 		var message = new ReportInstanceEventMessage(Guid.NewGuid(), DateTime.UtcNow, Configuration.InstanceGuid, 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)); | 		Services.TaskManager.Run("Report event for instance " + shortName, async () => await ServerMessaging.Send(message)); | ||||||
| 	} | 	} | ||||||
| 	 | 	 | ||||||
| 	internal void TransitionState(IInstanceState newState) { | 	private void TransitionState(IInstanceState newState) { | ||||||
| 		if (currentState == newState) { | 		if (currentState == newState) { | ||||||
| 			return; | 			return; | ||||||
| 		} | 		} | ||||||
| @@ -94,94 +96,114 @@ sealed class Instance : IAsyncDisposable { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	private T TransitionStateAndReturn<T>((IInstanceState State, T Result) newStateAndResult) { | ||||||
|  | 		TransitionState(newStateAndResult.State); | ||||||
|  | 		return newStateAndResult.Result; | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	public async Task Reconfigure(InstanceConfiguration configuration, IServerLauncher launcher, CancellationToken cancellationToken) { | 	public async Task Reconfigure(InstanceConfiguration configuration, IServerLauncher launcher, CancellationToken cancellationToken) { | ||||||
| 		await configurationSemaphore.WaitAsync(cancellationToken); | 		await stateTransitioningActionSemaphore.WaitAsync(cancellationToken); | ||||||
| 		try { | 		try { | ||||||
| 			Configuration = configuration; | 			Configuration = configuration; | ||||||
| 			Launcher = launcher; | 			Launcher = launcher; | ||||||
| 		} finally { | 		} finally { | ||||||
| 			configurationSemaphore.Release(); | 			stateTransitioningActionSemaphore.Release(); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task<LaunchInstanceResult> Launch(CancellationToken cancellationToken) { | 	public async Task<LaunchInstanceResult> Launch(CancellationToken shutdownCancellationToken) { | ||||||
| 		if (IsRunning) { | 		await stateTransitioningActionSemaphore.WaitAsync(shutdownCancellationToken); | ||||||
| 			return LaunchInstanceResult.InstanceAlreadyRunning; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is LaunchInstanceProcedure) { |  | ||||||
| 			return LaunchInstanceResult.InstanceAlreadyLaunching; |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		LaunchInstanceProcedure procedure; |  | ||||||
| 		 |  | ||||||
| 		await configurationSemaphore.WaitAsync(cancellationToken); |  | ||||||
| 		try { | 		try { | ||||||
| 			procedure = new LaunchInstanceProcedure(Configuration, Launcher); | 			return TransitionStateAndReturn(currentState.Launch(new InstanceContextImpl(this, shutdownCancellationToken))); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			logger.Error(e, "Caught exception while launching instance."); | ||||||
|  | 			return LaunchInstanceResult.UnknownError; | ||||||
| 		} finally { | 		} finally { | ||||||
| 			configurationSemaphore.Release(); | 			stateTransitioningActionSemaphore.Release(); | ||||||
| 		} | 		} | ||||||
| 		 |  | ||||||
| 		ReportAndSetStatus(InstanceStatus.Launching); |  | ||||||
| 		await procedureManager.Enqueue(procedure); |  | ||||||
| 		return LaunchInstanceResult.LaunchInitiated; |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task<StopInstanceResult> Stop(MinecraftStopStrategy stopStrategy, CancellationToken cancellationToken) { | 	public async Task<StopInstanceResult> Stop(MinecraftStopStrategy stopStrategy) { | ||||||
| 		if (!IsRunning) { | 		await stateTransitioningActionSemaphore.WaitAsync(); | ||||||
| 			return StopInstanceResult.InstanceAlreadyStopped; | 		try { | ||||||
|  | 			return TransitionStateAndReturn(currentState.Stop(stopStrategy)); | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			logger.Error(e, "Caught exception while stopping instance."); | ||||||
|  | 			return StopInstanceResult.UnknownError; | ||||||
|  | 		} finally { | ||||||
|  | 			stateTransitioningActionSemaphore.Release(); | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is StopInstanceProcedure) { | 	public async Task StopAndWait(TimeSpan waitTime) { | ||||||
| 			return StopInstanceResult.InstanceAlreadyStopping; | 		await Stop(MinecraftStopStrategy.Instant); | ||||||
|  |  | ||||||
|  | 		using var waitTokenSource = new CancellationTokenSource(waitTime); | ||||||
|  | 		var waitToken = waitTokenSource.Token; | ||||||
|  |  | ||||||
|  | 		while (currentState is not InstanceNotRunningState) { | ||||||
|  | 			await Task.Delay(TimeSpan.FromMilliseconds(250), waitToken); | ||||||
| 		} | 		} | ||||||
| 		 |  | ||||||
| 		ReportAndSetStatus(InstanceStatus.Stopping); |  | ||||||
| 		await procedureManager.Enqueue(new StopInstanceProcedure(stopStrategy)); |  | ||||||
| 		return StopInstanceResult.StopInitiated; |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
| 		return await currentState.SendCommand(command, cancellationToken); | 		return await currentState.SendCommand(command, cancellationToken); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async ValueTask DisposeAsync() { | 	private sealed class InstanceContextImpl : InstanceContext { | ||||||
| 		await procedureManager.DisposeAsync(); | 		private readonly Instance instance; | ||||||
|  | 		private readonly CancellationToken shutdownCancellationToken; | ||||||
| 		 | 		 | ||||||
| 		while (currentState is not InstanceNotRunningState) { | 		public InstanceContextImpl(Instance instance, CancellationToken shutdownCancellationToken) : base(instance.Services, instance.Configuration, instance.Launcher) { | ||||||
| 			await Task.Delay(TimeSpan.FromMilliseconds(250), CancellationToken.None); | 			this.instance = instance; | ||||||
|  | 			this.shutdownCancellationToken = shutdownCancellationToken; | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
|  | 		public override ILogger Logger => instance.logger; | ||||||
|  | 		public override string ShortName => instance.shortName; | ||||||
|  |  | ||||||
|  | 		public override void SetStatus(IInstanceStatus newStatus) { | ||||||
|  | 			instance.ReportAndSetStatus(newStatus); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		public override void ReportEvent(IInstanceEvent instanceEvent) { | ||||||
|  | 			instance.ReportEvent(instanceEvent); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		public override void TransitionState(Func<(IInstanceState, IInstanceStatus?)> newStateAndStatus) { | ||||||
|  | 			instance.stateTransitioningActionSemaphore.Wait(CancellationToken.None); | ||||||
|  | 			try { | ||||||
|  | 				var (state, status) = newStateAndStatus(); | ||||||
|  | 				 | ||||||
|  | 				if (!instance.IsRunning) { | ||||||
|  | 					// Only InstanceSessionManager is allowed to transition an instance out of a non-running state. | ||||||
|  | 					instance.logger.Debug("Cancelled state transition to {State} because instance is not running.", state.GetType().Name); | ||||||
|  | 					return; | ||||||
|  | 				} | ||||||
|  | 				 | ||||||
|  | 				if (state is not InstanceNotRunningState && shutdownCancellationToken.IsCancellationRequested) { | ||||||
|  | 					instance.logger.Debug("Cancelled state transition to {State} due to Agent shutdown.", state.GetType().Name); | ||||||
|  | 					return; | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				if (status != null) { | ||||||
|  | 					SetStatus(status); | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				instance.TransitionState(state); | ||||||
|  | 			} catch (Exception e) { | ||||||
|  | 				instance.logger.Error(e, "Caught exception during state transition."); | ||||||
|  | 			} finally { | ||||||
|  | 				instance.stateTransitioningActionSemaphore.Release(); | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Dispose() { | ||||||
|  | 		stateTransitioningActionSemaphore.Dispose(); | ||||||
|  | 		 | ||||||
| 		if (currentState is IDisposable disposable) { | 		if (currentState is IDisposable disposable) { | ||||||
| 			disposable.Dispose(); | 			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)); |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										35
									
								
								Agent/Phantom.Agent.Services/Instances/InstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								Agent/Phantom.Agent.Services/Instances/InstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | |||||||
|  | using Phantom.Agent.Minecraft.Launcher; | ||||||
|  | using Phantom.Agent.Services.Instances.States; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Serilog; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
|  | abstract class InstanceContext { | ||||||
|  | 	public InstanceServices Services { get; } | ||||||
|  | 	public InstanceConfiguration Configuration { get; } | ||||||
|  | 	public IServerLauncher Launcher { get; } | ||||||
|  | 	 | ||||||
|  | 	public abstract ILogger Logger { get; } | ||||||
|  | 	public abstract string ShortName { get; } | ||||||
|  |  | ||||||
|  | 	protected InstanceContext(InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||||
|  | 		Services = services; | ||||||
|  | 		Configuration = configuration; | ||||||
|  | 		Launcher = launcher; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public abstract void SetStatus(IInstanceStatus newStatus); | ||||||
|  |  | ||||||
|  | 	public void SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason reason) { | ||||||
|  | 		SetStatus(InstanceStatus.Failed(reason)); | ||||||
|  | 		ReportEvent(new InstanceLaunchFailedEvent(reason)); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public abstract void ReportEvent(IInstanceEvent instanceEvent); | ||||||
|  | 	public abstract void TransitionState(Func<(IInstanceState, IInstanceStatus?)> newStateAndStatus); | ||||||
|  |  | ||||||
|  | 	public void TransitionState(IInstanceState newState, IInstanceStatus? newStatus = null) { | ||||||
|  | 		TransitionState(() => (newState, newStatus)); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,84 +1,71 @@ | |||||||
| using System.Collections.Immutable; | using System.Collections.Immutable; | ||||||
| using System.Threading.Channels; |  | ||||||
| using Phantom.Agent.Rpc; | using Phantom.Agent.Rpc; | ||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.ToServer; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Collections; | ||||||
|  | using Phantom.Utils.Runtime; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
| sealed class InstanceLogSender : CancellableBackgroundTask { | sealed class InstanceLogSender : CancellableBackgroundTask { | ||||||
| 	private static readonly BoundedChannelOptions BufferOptions = new (capacity: 64) { |  | ||||||
| 		SingleReader = true, |  | ||||||
| 		SingleWriter = true, |  | ||||||
| 		FullMode = BoundedChannelFullMode.DropNewest |  | ||||||
| 	}; |  | ||||||
| 	 |  | ||||||
| 	private static readonly TimeSpan SendDelay = TimeSpan.FromMilliseconds(200); | 	private static readonly TimeSpan SendDelay = TimeSpan.FromMilliseconds(200); | ||||||
|  |  | ||||||
| 	private readonly ControllerConnection controllerConnection; |  | ||||||
| 	private readonly Guid instanceGuid; | 	private readonly Guid instanceGuid; | ||||||
| 	private readonly Channel<string> outputChannel; |  | ||||||
|  |  | ||||||
| 	private int droppedLinesSinceLastSend; | 	private readonly SemaphoreSlim semaphore = new (1, 1); | ||||||
|  | 	private readonly RingBuffer<string> buffer = new (1000); | ||||||
|  |  | ||||||
| 	public InstanceLogSender(ControllerConnection controllerConnection, TaskManager taskManager, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName), taskManager, "Instance log sender for " + loggerName) { | 	public InstanceLogSender(TaskManager taskManager, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName), taskManager, "Instance log sender for " + loggerName) { | ||||||
| 		this.controllerConnection = controllerConnection; |  | ||||||
| 		this.instanceGuid = instanceGuid; | 		this.instanceGuid = instanceGuid; | ||||||
| 		this.outputChannel = Channel.CreateBounded<string>(BufferOptions, OnLineDropped); |  | ||||||
| 		Start(); |  | ||||||
| 	} | 	} | ||||||
| 	 | 	 | ||||||
| 	protected override async Task RunTask() { | 	protected override async Task RunTask() { | ||||||
| 		var lineReader = outputChannel.Reader; |  | ||||||
| 		var lineBuilder = ImmutableArray.CreateBuilder<string>(); |  | ||||||
| 		 |  | ||||||
| 		try { | 		try { | ||||||
| 			while (await lineReader.WaitToReadAsync(CancellationToken)) { | 			while (!CancellationToken.IsCancellationRequested) { | ||||||
|  | 				await SendOutputToServer(await DequeueOrThrow()); | ||||||
| 				await Task.Delay(SendDelay, CancellationToken); | 				await Task.Delay(SendDelay, CancellationToken); | ||||||
| 				await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); |  | ||||||
| 			} | 			} | ||||||
| 		} catch (OperationCanceledException) { | 		} catch (OperationCanceledException) { | ||||||
| 			// Ignore. | 			// Ignore. | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// Flush remaining lines. | 		// Flush remaining lines. | ||||||
| 		await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | 		await SendOutputToServer(DequeueWithoutSemaphore()); | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private ImmutableArray<string> ReadLinesFromChannel(ChannelReader<string> reader, ImmutableArray<string>.Builder builder) { |  | ||||||
| 		builder.Clear(); |  | ||||||
| 		 |  | ||||||
| 		while (reader.TryRead(out string? line)) { |  | ||||||
| 			builder.Add(line); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		int droppedLines = Interlocked.Exchange(ref droppedLinesSinceLastSend, 0); |  | ||||||
| 		if (droppedLines > 0) { |  | ||||||
| 			builder.Add($"Dropped {droppedLines} {(droppedLines == 1 ? "line" : "lines")} due to buffer overflow."); |  | ||||||
| 		} |  | ||||||
| 		 |  | ||||||
| 		return builder.ToImmutable(); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task SendOutputToServer(ImmutableArray<string> lines) { | 	private async Task SendOutputToServer(ImmutableArray<string> lines) { | ||||||
| 		if (!lines.IsEmpty) { | 		if (!lines.IsEmpty) { | ||||||
| 			await controllerConnection.Send(new InstanceOutputMessage(instanceGuid, lines)); | 			await ServerMessaging.Send(new InstanceOutputMessage(instanceGuid, lines)); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void OnLineDropped(string line) { | 	private ImmutableArray<string> DequeueWithoutSemaphore() { | ||||||
| 		Logger.Warning("Buffer is full, dropped line: {Line}", line); | 		ImmutableArray<string> lines = buffer.Count > 0 ? buffer.EnumerateLast(uint.MaxValue).ToImmutableArray() : ImmutableArray<string>.Empty; | ||||||
| 		Interlocked.Increment(ref droppedLinesSinceLastSend); | 		buffer.Clear(); | ||||||
|  | 		return lines; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<ImmutableArray<string>> DequeueOrThrow() { | ||||||
|  | 		await semaphore.WaitAsync(CancellationToken); | ||||||
|  |  | ||||||
|  | 		try { | ||||||
|  | 			return DequeueWithoutSemaphore(); | ||||||
|  | 		} finally { | ||||||
|  | 			semaphore.Release(); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void Enqueue(string line) { | 	public void Enqueue(string line) { | ||||||
| 		outputChannel.Writer.TryWrite(line); | 		try { | ||||||
| 	} | 			semaphore.Wait(CancellationToken); | ||||||
|  | 		} catch (Exception) { | ||||||
|  | 			return; | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 	protected override void Dispose() { | 		try { | ||||||
| 		if (!outputChannel.Writer.TryComplete()) { | 			buffer.Add(line); | ||||||
| 			Logger.Error("Could not mark channel as completed."); | 		} finally { | ||||||
|  | 			semaphore.Release(); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,85 +0,0 @@ | |||||||
| 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,8 +1,7 @@ | |||||||
| using Phantom.Agent.Minecraft.Launcher; | using Phantom.Agent.Minecraft.Launcher; | ||||||
| using Phantom.Agent.Rpc; |  | ||||||
| using Phantom.Agent.Services.Backups; | using Phantom.Agent.Services.Backups; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Runtime; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances;  | namespace Phantom.Agent.Services.Instances;  | ||||||
|  |  | ||||||
| sealed record InstanceServices(ControllerConnection ControllerConnection, TaskManager TaskManager, PortManager PortManager, BackupManager BackupManager, LaunchServices LaunchServices); | sealed record InstanceServices(TaskManager TaskManager, PortManager PortManager, BackupManager BackupManager, LaunchServices LaunchServices); | ||||||
|   | |||||||
| @@ -14,17 +14,16 @@ using Phantom.Common.Data.Instance; | |||||||
| using Phantom.Common.Data.Minecraft; | using Phantom.Common.Data.Minecraft; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.ToServer; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| using Phantom.Utils.Tasks; | using Phantom.Utils.Runtime; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances; | namespace Phantom.Agent.Services.Instances; | ||||||
|  |  | ||||||
| sealed class InstanceSessionManager : IAsyncDisposable { | sealed class InstanceSessionManager : IDisposable { | ||||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceSessionManager>(); | 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceSessionManager>(); | ||||||
| 	 | 	 | ||||||
| 	private readonly ControllerConnection controllerConnection; |  | ||||||
| 	private readonly AgentInfo agentInfo; | 	private readonly AgentInfo agentInfo; | ||||||
| 	private readonly string basePath; | 	private readonly string basePath; | ||||||
|  |  | ||||||
| @@ -35,10 +34,7 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 	private readonly CancellationToken shutdownCancellationToken; | 	private readonly CancellationToken shutdownCancellationToken; | ||||||
| 	private readonly SemaphoreSlim semaphore = new (1, 1); | 	private readonly SemaphoreSlim semaphore = new (1, 1); | ||||||
|  |  | ||||||
| 	private uint instanceLoggerSequenceId = 0; | 	public InstanceSessionManager(AgentInfo agentInfo, AgentFolders agentFolders, JavaRuntimeRepository javaRuntimeRepository, TaskManager taskManager, BackupManager backupManager) { | ||||||
|  |  | ||||||
| 	public InstanceSessionManager(ControllerConnection controllerConnection, AgentInfo agentInfo, AgentFolders agentFolders, JavaRuntimeRepository javaRuntimeRepository, TaskManager taskManager, BackupManager backupManager) { |  | ||||||
| 		this.controllerConnection = controllerConnection; |  | ||||||
| 		this.agentInfo = agentInfo; | 		this.agentInfo = agentInfo; | ||||||
| 		this.basePath = agentFolders.InstancesFolderPath; | 		this.basePath = agentFolders.InstancesFolderPath; | ||||||
| 		this.shutdownCancellationToken = shutdownCancellationTokenSource.Token; | 		this.shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||||
| @@ -47,7 +43,7 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 		var launchServices = new LaunchServices(minecraftServerExecutables, javaRuntimeRepository); | 		var launchServices = new LaunchServices(minecraftServerExecutables, javaRuntimeRepository); | ||||||
| 		var portManager = new PortManager(agentInfo.AllowedServerPorts, agentInfo.AllowedRconPorts); | 		var portManager = new PortManager(agentInfo.AllowedServerPorts, agentInfo.AllowedRconPorts); | ||||||
| 		 | 		 | ||||||
| 		this.instanceServices = new InstanceServices(controllerConnection, taskManager, portManager, backupManager, launchServices); | 		this.instanceServices = new InstanceServices(taskManager, portManager, backupManager, launchServices); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private async Task<InstanceActionResult<T>> AcquireSemaphoreAndRun<T>(Func<Task<InstanceActionResult<T>>> func) { | 	private async Task<InstanceActionResult<T>> AcquireSemaphoreAndRun<T>(Func<Task<InstanceActionResult<T>>> func) { | ||||||
| @@ -113,7 +109,7 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 			else { | 			else { | ||||||
| 				instances[instanceGuid] = instance = new Instance(GetInstanceLoggerName(instanceGuid), instanceServices, configuration, launcher); | 				instances[instanceGuid] = instance = new Instance(instanceServices, configuration, launcher); | ||||||
| 				Logger.Information("Created instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | 				Logger.Information("Created instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | ||||||
| 				 | 				 | ||||||
| 				instance.ReportLastStatus(); | 				instance.ReportLastStatus(); | ||||||
| @@ -128,11 +124,6 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 		}); | 		}); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private string GetInstanceLoggerName(Guid guid) { |  | ||||||
| 		var prefix = guid.ToString(); |  | ||||||
| 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref instanceLoggerSequenceId); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private ImmutableArray<Instance> GetRunningInstancesInternal() { | 	private ImmutableArray<Instance> GetRunningInstancesInternal() { | ||||||
| 		return instances.Values.Where(static instance => instance.IsRunning).ToImmutableArray(); | 		return instances.Values.Where(static instance => instance.IsRunning).ToImmutableArray(); | ||||||
| 	} | 	} | ||||||
| @@ -148,7 +139,7 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 				var runningInstances = GetRunningInstancesInternal(); | 				var runningInstances = GetRunningInstancesInternal(); | ||||||
| 				var runningInstanceCount = runningInstances.Length; | 				var runningInstanceCount = runningInstances.Length; | ||||||
| 				var runningInstanceMemory = runningInstances.Aggregate(RamAllocationUnits.Zero, static (total, instance) => total + instance.Configuration.MemoryAllocation); | 				var runningInstanceMemory = runningInstances.Aggregate(RamAllocationUnits.Zero, static (total, instance) => total + instance.Configuration.MemoryAllocation); | ||||||
| 				await controllerConnection.Send(new ReportAgentStatusMessage(runningInstanceCount, runningInstanceMemory)); | 				await ServerMessaging.Send(new ReportAgentStatusMessage(runningInstanceCount, runningInstanceMemory)); | ||||||
| 			} finally { | 			} finally { | ||||||
| 				semaphore.Release(); | 				semaphore.Release(); | ||||||
| 			} | 			} | ||||||
| @@ -176,23 +167,38 @@ sealed class InstanceSessionManager : IAsyncDisposable { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public Task<InstanceActionResult<StopInstanceResult>> Stop(Guid instanceGuid, MinecraftStopStrategy stopStrategy) { | 	public Task<InstanceActionResult<StopInstanceResult>> Stop(Guid instanceGuid, MinecraftStopStrategy stopStrategy) { | ||||||
| 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, instance => instance.Stop(stopStrategy, shutdownCancellationToken)); | 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, instance => instance.Stop(stopStrategy)); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public Task<InstanceActionResult<SendCommandToInstanceResult>> SendCommand(Guid instanceGuid, string command) { | 	public Task<InstanceActionResult<SendCommandToInstanceResult>> SendCommand(Guid instanceGuid, string command) { | ||||||
| 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, async instance => await instance.SendCommand(command, shutdownCancellationToken) ? SendCommandToInstanceResult.Success : SendCommandToInstanceResult.UnknownError); | 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, async instance => await instance.SendCommand(command, shutdownCancellationToken) ? SendCommandToInstanceResult.Success : SendCommandToInstanceResult.UnknownError); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async ValueTask DisposeAsync() { | 	public async Task StopAll() { | ||||||
| 		Logger.Information("Stopping all instances..."); |  | ||||||
| 		 |  | ||||||
| 		shutdownCancellationTokenSource.Cancel(); | 		shutdownCancellationTokenSource.Cancel(); | ||||||
| 		 | 		 | ||||||
| 		await semaphore.WaitAsync(CancellationToken.None); | 		Logger.Information("Stopping all instances..."); | ||||||
| 		await Task.WhenAll(instances.Values.Select(static instance => instance.DisposeAsync().AsTask())); |  | ||||||
| 		instances.Clear(); |  | ||||||
|  |  | ||||||
|  | 		await semaphore.WaitAsync(CancellationToken.None); | ||||||
|  | 		try { | ||||||
|  | 			await Task.WhenAll(instances.Values.Select(static instance => instance.StopAndWait(TimeSpan.FromSeconds(30)))); | ||||||
|  | 			DisposeAllInstances(); | ||||||
|  | 		} finally { | ||||||
|  | 			semaphore.Release(); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Dispose() { | ||||||
|  | 		DisposeAllInstances(); | ||||||
| 		shutdownCancellationTokenSource.Dispose(); | 		shutdownCancellationTokenSource.Dispose(); | ||||||
| 		semaphore.Dispose(); | 		semaphore.Dispose(); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	private void DisposeAllInstances() { | ||||||
|  | 		foreach (var (_, instance) in instances) { | ||||||
|  | 			instance.Dispose(); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		instances.Clear(); | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,29 +0,0 @@ | |||||||
| 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; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,7 +0,0 @@ | |||||||
| using Phantom.Agent.Services.Instances.States; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances.Procedures; |  | ||||||
|  |  | ||||||
| interface IInstanceProcedure { |  | ||||||
| 	Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken); |  | ||||||
| } |  | ||||||
| @@ -1,97 +0,0 @@ | |||||||
| 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; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,17 +0,0 @@ | |||||||
| 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,103 +0,0 @@ | |||||||
| using System.Diagnostics; |  | ||||||
| using Phantom.Agent.Minecraft.Command; |  | ||||||
| using Phantom.Agent.Minecraft.Instance; |  | ||||||
| using Phantom.Agent.Services.Instances.States; |  | ||||||
| using Phantom.Common.Data.Instance; |  | ||||||
| using Phantom.Common.Data.Minecraft; |  | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances.Procedures; |  | ||||||
|  |  | ||||||
| sealed record StopInstanceProcedure(MinecraftStopStrategy StopStrategy) : IInstanceProcedure { |  | ||||||
| 	private static readonly ushort[] Stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; |  | ||||||
|  |  | ||||||
| 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { |  | ||||||
| 		if (context.CurrentState is not InstanceRunningState runningState) { |  | ||||||
| 			return null; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		var process = runningState.Process; |  | ||||||
|  |  | ||||||
| 		runningState.IsStopping = true; |  | ||||||
| 		context.SetStatus(InstanceStatus.Stopping); |  | ||||||
|  |  | ||||||
| 		var seconds = StopStrategy.Seconds; |  | ||||||
| 		if (seconds > 0) { |  | ||||||
| 			try { |  | ||||||
| 				await CountDownWithAnnouncements(context, process, seconds, cancellationToken); |  | ||||||
| 			} catch (OperationCanceledException) { |  | ||||||
| 				runningState.IsStopping = false; |  | ||||||
| 				return null; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		try { |  | ||||||
| 			// Too late to cancel the stop procedure now. |  | ||||||
| 			if (!process.HasEnded) { |  | ||||||
| 				context.Logger.Information("Session stopping now."); |  | ||||||
| 				await DoStop(context, process); |  | ||||||
| 			} |  | ||||||
| 		} finally { |  | ||||||
| 			context.Logger.Information("Session stopped."); |  | ||||||
| 			context.SetStatus(InstanceStatus.NotRunning); |  | ||||||
| 			context.ReportEvent(InstanceEvent.Stopped); |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		return new InstanceNotRunningState(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task CountDownWithAnnouncements(IInstanceContext context, InstanceProcess process, ushort seconds, CancellationToken cancellationToken) { |  | ||||||
| 		context.Logger.Information("Session stopping in {Seconds} seconds.", seconds); |  | ||||||
|  |  | ||||||
| 		foreach (var stop in Stops) { |  | ||||||
| 			// TODO change to event-based cancellation |  | ||||||
| 			if (process.HasEnded) { |  | ||||||
| 				return; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if (seconds > stop) { |  | ||||||
| 				await process.SendCommand(GetCountDownAnnouncementCommand(seconds), cancellationToken); |  | ||||||
| 				await Task.Delay(TimeSpan.FromSeconds(seconds - stop), cancellationToken); |  | ||||||
| 				seconds = stop; |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private static string GetCountDownAnnouncementCommand(ushort seconds) { |  | ||||||
| 		return MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task DoStop(IInstanceContext context, InstanceProcess process) { |  | ||||||
| 		context.Logger.Information("Sending stop command..."); |  | ||||||
| 		await TrySendStopCommand(context, process); |  | ||||||
|  |  | ||||||
| 		context.Logger.Information("Waiting for session to end..."); |  | ||||||
| 		await WaitForSessionToEnd(context, process); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task TrySendStopCommand(IInstanceContext context, InstanceProcess process) { |  | ||||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); |  | ||||||
| 		try { |  | ||||||
| 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); |  | ||||||
| 		} catch (OperationCanceledException) { |  | ||||||
| 			// Ignore. |  | ||||||
| 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { |  | ||||||
| 			// Ignore. |  | ||||||
| 		} catch (Exception e) { |  | ||||||
| 			context.Logger.Warning(e, "Caught exception while sending stop command."); |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private async Task WaitForSessionToEnd(IInstanceContext context, InstanceProcess process) { |  | ||||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(55)); |  | ||||||
| 		try { |  | ||||||
| 			await process.WaitForExit(timeout.Token); |  | ||||||
| 		} catch (OperationCanceledException) { |  | ||||||
| 			try { |  | ||||||
| 				context.Logger.Warning("Waiting timed out, killing session..."); |  | ||||||
| 				process.Kill(); |  | ||||||
| 			} catch (Exception e) { |  | ||||||
| 				context.Logger.Error(e, "Caught exception while killing session."); |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -0,0 +1,28 @@ | |||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.Sessions;  | ||||||
|  |  | ||||||
|  | sealed class InstanceSession : IDisposable { | ||||||
|  | 	private readonly InstanceProcess process; | ||||||
|  | 	private readonly InstanceContext context; | ||||||
|  | 	private readonly InstanceLogSender logSender; | ||||||
|  |  | ||||||
|  | 	public InstanceSession(InstanceProcess process, InstanceContext context) { | ||||||
|  | 		this.process = process; | ||||||
|  | 		this.context = context; | ||||||
|  | 		this.logSender = new InstanceLogSender(context.Services.TaskManager, context.Configuration.InstanceGuid, context.ShortName); | ||||||
|  | 		 | ||||||
|  | 		this.process.AddOutputListener(SessionOutput); | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	private void SessionOutput(object? sender, string line) { | ||||||
|  | 		context.Logger.Debug("[Server] {Line}", line); | ||||||
|  | 		logSender.Enqueue(line); | ||||||
|  | 	} | ||||||
|  | 	 | ||||||
|  | 	public void Dispose() { | ||||||
|  | 		logSender.Stop(); | ||||||
|  | 		process.Dispose(); | ||||||
|  | 		context.Services.PortManager.Release(context.Configuration); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,6 +1,11 @@ | |||||||
| namespace Phantom.Agent.Services.Instances.States; | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.States;  | ||||||
|  |  | ||||||
| interface IInstanceState { | interface IInstanceState { | ||||||
| 	void Initialize(); | 	void Initialize(); | ||||||
|  | 	(IInstanceState, LaunchInstanceResult) Launch(InstanceContext context); | ||||||
|  | 	(IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy); | ||||||
| 	Task<bool> SendCommand(string command, CancellationToken cancellationToken); | 	Task<bool> SendCommand(string command, CancellationToken cancellationToken); | ||||||
| } | } | ||||||
|   | |||||||
| @@ -0,0 +1,127 @@ | |||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Agent.Minecraft.Launcher; | ||||||
|  | using Phantom.Agent.Minecraft.Server; | ||||||
|  | using Phantom.Agent.Services.Instances.Sessions; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.States; | ||||||
|  |  | ||||||
|  | sealed class InstanceLaunchingState : IInstanceState, IDisposable { | ||||||
|  | 	private readonly InstanceContext context; | ||||||
|  | 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||||
|  | 	private byte lastDownloadProgress = byte.MaxValue; | ||||||
|  |  | ||||||
|  | 	public InstanceLaunchingState(InstanceContext context) { | ||||||
|  | 		this.context = context; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Initialize() { | ||||||
|  | 		context.Logger.Information("Session starting..."); | ||||||
|  |  | ||||||
|  | 		var launchTask = context.Services.TaskManager.Run("Launch procedure for instance " + context.ShortName, DoLaunch); | ||||||
|  | 		launchTask.ContinueWith(OnLaunchSuccess, CancellationToken.None, TaskContinuationOptions.OnlyOnRanToCompletion, TaskScheduler.Default); | ||||||
|  | 		launchTask.ContinueWith(OnLaunchFailure, CancellationToken.None, TaskContinuationOptions.NotOnRanToCompletion, TaskScheduler.Default); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task<InstanceProcess> DoLaunch() { | ||||||
|  | 		var cancellationToken = cancellationTokenSource.Token; | ||||||
|  | 		cancellationToken.ThrowIfCancellationRequested(); | ||||||
|  |  | ||||||
|  | 		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 context.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.InvalidJvmArguments) { | ||||||
|  | 			throw new LaunchFailureException(InstanceLaunchFailReason.InvalidJvmArguments, "Session failed to launch, invalid JVM arguments."); | ||||||
|  | 		} | ||||||
|  | 		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.Launching); | ||||||
|  | 		return launchSuccess.Process; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void OnLaunchSuccess(Task<InstanceProcess> task) { | ||||||
|  | 		context.TransitionState(() => { | ||||||
|  | 			context.ReportEvent(InstanceEvent.LaunchSucceded); | ||||||
|  | 			 | ||||||
|  | 			var process = task.Result; | ||||||
|  | 			var session = new InstanceSession(process, context); | ||||||
|  | 			 | ||||||
|  | 			if (cancellationTokenSource.IsCancellationRequested) { | ||||||
|  | 				return (new InstanceStoppingState(context, process, session), InstanceStatus.Stopping); | ||||||
|  | 			} | ||||||
|  | 			else { | ||||||
|  | 				return (new InstanceRunningState(context, process, session), null); | ||||||
|  | 			} | ||||||
|  | 		}); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void OnLaunchFailure(Task task) { | ||||||
|  | 		if (task.IsFaulted) { | ||||||
|  | 			if (task.Exception is { InnerException: LaunchFailureException e }) { | ||||||
|  | 				context.Logger.Error(e.LogMessage); | ||||||
|  | 				context.SetLaunchFailedStatusAndReportEvent(e.Reason); | ||||||
|  | 			} | ||||||
|  | 			else { | ||||||
|  | 				context.Logger.Error(task.Exception, "Caught exception while launching instance."); | ||||||
|  | 				context.SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason.UnknownError); | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		context.Services.PortManager.Release(context.Configuration); | ||||||
|  | 		context.TransitionState(new InstanceNotRunningState()); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	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; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||||
|  | 		return (this, LaunchInstanceResult.InstanceAlreadyLaunching); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||||
|  | 		cancellationTokenSource.Cancel(); | ||||||
|  | 		return (this, StopInstanceResult.StopInitiated); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
|  | 		return Task.FromResult(false); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Dispose() { | ||||||
|  | 		cancellationTokenSource.Dispose(); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,8 +1,34 @@ | |||||||
| namespace Phantom.Agent.Services.Instances.States; | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.States;  | ||||||
|  |  | ||||||
| sealed class InstanceNotRunningState : IInstanceState { | sealed class InstanceNotRunningState : IInstanceState { | ||||||
| 	public void Initialize() {} | 	public void Initialize() {} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||||
|  | 		InstanceLaunchFailReason? failReason = context.Services.PortManager.Reserve(context.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 (this, LaunchInstanceResult.LaunchInitiated); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		context.SetStatus(InstanceStatus.Launching); | ||||||
|  | 		return (new InstanceLaunchingState(context), LaunchInstanceResult.LaunchInitiated); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||||
|  | 		return (this, StopInstanceResult.InstanceAlreadyStopped); | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
| 		return Task.FromResult(false); | 		return Task.FromResult(false); | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,81 +1,133 @@ | |||||||
| using Phantom.Agent.Minecraft.Instance; | using Phantom.Agent.Minecraft.Command; | ||||||
| using Phantom.Agent.Minecraft.Launcher; | using Phantom.Agent.Minecraft.Instance; | ||||||
| using Phantom.Agent.Services.Backups; | using Phantom.Agent.Services.Backups; | ||||||
| using Phantom.Agent.Services.Instances.Procedures; | using Phantom.Agent.Services.Instances.Sessions; | ||||||
| using Phantom.Common.Data.Backups; | using Phantom.Common.Data.Backups; | ||||||
| using Phantom.Common.Data.Instance; | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
| namespace Phantom.Agent.Services.Instances.States; | namespace Phantom.Agent.Services.Instances.States; | ||||||
|  |  | ||||||
| sealed class InstanceRunningState : IInstanceState, IDisposable { | sealed class InstanceRunningState : IInstanceState { | ||||||
| 	public InstanceProcess Process { get; } | 	private readonly InstanceContext context; | ||||||
|  | 	private readonly InstanceProcess process; | ||||||
| 	internal bool IsStopping { get; set; } |  | ||||||
|  |  | ||||||
| 	private readonly InstanceConfiguration configuration; |  | ||||||
| 	private readonly IServerLauncher launcher; |  | ||||||
| 	private readonly IInstanceContext context; |  | ||||||
|  |  | ||||||
| 	private readonly InstanceLogSender logSender; |  | ||||||
| 	private readonly BackupScheduler backupScheduler; | 	private readonly BackupScheduler backupScheduler; | ||||||
|  | 	private readonly RunningSessionDisposer runningSessionDisposer; | ||||||
| 	 | 	 | ||||||
| 	private bool isDisposed; | 	private readonly CancellationTokenSource delayedStopCancellationTokenSource = new (); | ||||||
|  | 	private bool stateOwnsDelayedStopCancellationTokenSource = true; | ||||||
|  | 	private bool isStopping; | ||||||
|  |  | ||||||
| 	public InstanceRunningState(InstanceConfiguration configuration, IServerLauncher launcher, InstanceProcess process, IInstanceContext context) { | 	public InstanceRunningState(InstanceContext context, InstanceProcess process, InstanceSession session) { | ||||||
| 		this.configuration = configuration; |  | ||||||
| 		this.launcher = launcher; |  | ||||||
| 		this.context = context; | 		this.context = context; | ||||||
| 		this.Process = process; | 		this.process = process; | ||||||
|  | 		this.backupScheduler = new BackupScheduler(context.Services.TaskManager, context.Services.BackupManager, process, context.Configuration.ServerPort, context.ShortName); | ||||||
| 		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.Services.TaskManager, configuration.InstanceGuid, context.ShortName); |  | ||||||
|  |  | ||||||
| 		this.backupScheduler = new BackupScheduler(context.Services.TaskManager, context.Services.BackupManager, process, context, configuration.ServerPort); |  | ||||||
| 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted; | 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted; | ||||||
|  | 		this.runningSessionDisposer = new RunningSessionDisposer(this, session); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public void Initialize() { | 	public void Initialize() { | ||||||
| 		Process.Ended += ProcessEnded; | 		process.Ended += ProcessEnded; | ||||||
| 		 | 		 | ||||||
| 		if (Process.HasEnded) { | 		if (process.HasEnded) { | ||||||
| 			if (TryDispose()) { | 			if (runningSessionDisposer.Dispose()) { | ||||||
| 				context.Logger.Warning("Session ended immediately after it was started."); | 				context.Logger.Warning("Session ended immediately after it was started."); | ||||||
| 				context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError)), immediate: true); | 				context.ReportEvent(InstanceEvent.Stopped); | ||||||
|  | 				context.Services.TaskManager.Run("Transition state of instance " + context.ShortName + " to not running", () => context.TransitionState(new InstanceNotRunningState(), InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError))); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 		else { | 		else { | ||||||
|  | 			context.SetStatus(InstanceStatus.Running); | ||||||
| 			context.Logger.Information("Session started."); | 			context.Logger.Information("Session started."); | ||||||
| 			Process.AddOutputListener(SessionOutput); |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void SessionOutput(object? sender, string line) { |  | ||||||
| 		context.Logger.Debug("[Server] {Line}", line); |  | ||||||
| 		logSender.Enqueue(line); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	private void ProcessEnded(object? sender, EventArgs e) { | 	private void ProcessEnded(object? sender, EventArgs e) { | ||||||
| 		if (!TryDispose()) { | 		if (!runningSessionDisposer.Dispose()) { | ||||||
| 			return; | 			return; | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if (IsStopping) { | 		if (isStopping) { | ||||||
| 			context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.NotRunning), immediate: true); | 			context.Logger.Information("Session ended."); | ||||||
|  | 			context.ReportEvent(InstanceEvent.Stopped); | ||||||
|  | 			context.TransitionState(new InstanceNotRunningState(), InstanceStatus.NotRunning); | ||||||
| 		} | 		} | ||||||
| 		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.EnqueueProcedure(new LaunchInstanceProcedure(configuration, launcher, IsRestarting: true)); | 			context.TransitionState(new InstanceLaunchingState(context), InstanceStatus.Restarting); | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private void OnScheduledBackupCompleted(object? sender, BackupCreationResult e) { | 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||||
| 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | 		return (this, LaunchInstanceResult.InstanceAlreadyRunning); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||||
|  | 		if (stopStrategy == MinecraftStopStrategy.Instant) { | ||||||
|  | 			CancelDelayedStop(); | ||||||
|  | 			return (PrepareStoppedState(), StopInstanceResult.StopInitiated); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if (isStopping) { | ||||||
|  | 			// TODO change delay or something | ||||||
|  | 			return (this, StopInstanceResult.InstanceAlreadyStopping); | ||||||
|  | 		} | ||||||
|  | 		 | ||||||
|  | 		isStopping = true; | ||||||
|  | 		context.Services.TaskManager.Run("Delayed stop timer for instance " + context.ShortName, () => StopLater(stopStrategy.Seconds)); | ||||||
|  | 		return (this, StopInstanceResult.StopInitiated); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private IInstanceState PrepareStoppedState() { | ||||||
|  | 		process.Ended -= ProcessEnded; | ||||||
|  | 		backupScheduler.Stop(); | ||||||
|  | 		return new InstanceStoppingState(context, process, runningSessionDisposer); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private void CancelDelayedStop() { | ||||||
|  | 		try { | ||||||
|  | 			delayedStopCancellationTokenSource.Cancel(); | ||||||
|  | 		} catch (ObjectDisposedException) { | ||||||
|  | 			// ignore | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task StopLater(int seconds) { | ||||||
|  | 		var cancellationToken = delayedStopCancellationTokenSource.Token; | ||||||
|  |  | ||||||
|  | 		try { | ||||||
|  | 			stateOwnsDelayedStopCancellationTokenSource = false; | ||||||
|  |  | ||||||
|  | 			int[] stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; | ||||||
|  | 			 | ||||||
|  | 			foreach (var stop in stops) { | ||||||
|  | 				if (seconds > stop) { | ||||||
|  | 					await SendCommand(MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")), cancellationToken); | ||||||
|  | 					await Task.Delay(TimeSpan.FromSeconds(seconds - stop), cancellationToken); | ||||||
|  | 					seconds = stop; | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			context.Logger.Debug("Cancelled delayed stop."); | ||||||
|  | 			return; | ||||||
|  | 		} catch (ObjectDisposedException) { | ||||||
|  | 			return; | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			context.Logger.Warning(e, "Caught exception during delayed stop."); | ||||||
|  | 			return; | ||||||
|  | 		} finally { | ||||||
|  | 			delayedStopCancellationTokenSource.Dispose(); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		context.TransitionState(PrepareStoppedState()); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public async Task<bool> 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 true; | 			return true; | ||||||
| 		} catch (OperationCanceledException) { | 		} catch (OperationCanceledException) { | ||||||
| 			return false; | 			return false; | ||||||
| @@ -85,25 +137,42 @@ sealed class InstanceRunningState : IInstanceState, IDisposable { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private bool TryDispose() { | 	private void OnScheduledBackupCompleted(object? sender, BackupCreationResult e) { | ||||||
| 		lock (this) { | 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | ||||||
| 			if (isDisposed) { | 	} | ||||||
| 				return false; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			isDisposed = true; | 	private sealed class RunningSessionDisposer : IDisposable { | ||||||
|  | 		private readonly InstanceRunningState state; | ||||||
|  | 		private readonly InstanceSession session; | ||||||
|  | 		private bool isDisposed; | ||||||
|  |  | ||||||
|  | 		public RunningSessionDisposer(InstanceRunningState state, InstanceSession session) { | ||||||
|  | 			this.state = state; | ||||||
|  | 			this.session = session; | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		logSender.Stop(); | 		public bool Dispose() { | ||||||
| 		backupScheduler.Stop(); | 			lock (this) { | ||||||
|  | 				if (isDisposed) { | ||||||
|  | 					return false; | ||||||
|  | 				} | ||||||
|  |  | ||||||
| 		Process.Dispose(); | 				isDisposed = true; | ||||||
| 		context.Services.PortManager.Release(configuration); | 			} | ||||||
|  |  | ||||||
| 		return true; | 			if (state.stateOwnsDelayedStopCancellationTokenSource) { | ||||||
| 	} | 				state.delayedStopCancellationTokenSource.Dispose(); | ||||||
|  | 			} | ||||||
|  | 			else { | ||||||
|  | 				state.CancelDelayedStop(); | ||||||
|  | 			} | ||||||
| 			 | 			 | ||||||
| 	public void Dispose() { | 			session.Dispose(); | ||||||
| 		TryDispose(); | 			return true; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		void IDisposable.Dispose() { | ||||||
|  | 			Dispose(); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -0,0 +1,89 @@ | |||||||
|  | using System.Diagnostics; | ||||||
|  | using Phantom.Agent.Minecraft.Command; | ||||||
|  | using Phantom.Agent.Minecraft.Instance; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
|  | using Phantom.Common.Data.Minecraft; | ||||||
|  | using Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
|  | namespace Phantom.Agent.Services.Instances.States;  | ||||||
|  |  | ||||||
|  | sealed class InstanceStoppingState : IInstanceState, IDisposable { | ||||||
|  | 	private readonly InstanceContext context; | ||||||
|  | 	private readonly InstanceProcess process; | ||||||
|  | 	private readonly IDisposable sessionDisposer; | ||||||
|  |  | ||||||
|  | 	public InstanceStoppingState(InstanceContext context, InstanceProcess process, IDisposable sessionDisposer) { | ||||||
|  | 		this.context = context; | ||||||
|  | 		this.process = process; | ||||||
|  | 		this.sessionDisposer = sessionDisposer; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Initialize() { | ||||||
|  | 		context.Logger.Information("Session stopping."); | ||||||
|  | 		context.SetStatus(InstanceStatus.Stopping); | ||||||
|  | 		context.Services.TaskManager.Run("Stop procedure for instance " + context.ShortName, DoStop); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task DoStop() { | ||||||
|  | 		try { | ||||||
|  | 			// Do not release the semaphore after this point. | ||||||
|  | 			if (!await process.BackupSemaphore.CancelAndWait(TimeSpan.FromSeconds(1))) { | ||||||
|  | 				context.Logger.Information("Waiting for backup to finish..."); | ||||||
|  | 				await process.BackupSemaphore.CancelAndWait(Timeout.InfiniteTimeSpan); | ||||||
|  | 			} | ||||||
|  | 			 | ||||||
|  | 			context.Logger.Information("Sending stop command..."); | ||||||
|  | 			await DoSendStopCommand(); | ||||||
|  |  | ||||||
|  | 			context.Logger.Information("Waiting for session to end..."); | ||||||
|  | 			await DoWaitForSessionToEnd(); | ||||||
|  | 		} finally { | ||||||
|  | 			context.Logger.Information("Session stopped."); | ||||||
|  | 			context.ReportEvent(InstanceEvent.Stopped); | ||||||
|  | 			context.TransitionState(new InstanceNotRunningState(), InstanceStatus.NotRunning); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task DoSendStopCommand() { | ||||||
|  | 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); | ||||||
|  | 		try { | ||||||
|  | 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			// ignore | ||||||
|  | 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { | ||||||
|  | 			// ignore | ||||||
|  | 		} catch (Exception e) { | ||||||
|  | 			context.Logger.Warning(e, "Caught exception while sending stop command."); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	private async Task DoWaitForSessionToEnd() { | ||||||
|  | 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(55)); | ||||||
|  | 		try { | ||||||
|  | 			await process.WaitForExit(timeout.Token); | ||||||
|  | 		} catch (OperationCanceledException) { | ||||||
|  | 			try { | ||||||
|  | 				context.Logger.Warning("Waiting timed out, killing session..."); | ||||||
|  | 				process.Kill(); | ||||||
|  | 			} catch (Exception e) { | ||||||
|  | 				context.Logger.Error(e, "Caught exception while killing session."); | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||||
|  | 		return (this, LaunchInstanceResult.InstanceIsStopping); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||||
|  | 		return (this, StopInstanceResult.InstanceAlreadyStopping); // TODO maybe provide a way to kill? | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||||
|  | 		return Task.FromResult(false); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public void Dispose() { | ||||||
|  | 		sessionDisposer.Dispose(); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,12 +1,13 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages.Agent\Phantom.Common.Messages.Agent.csproj" /> |     <ProjectReference Include="..\..\Common\Phantom.Common.Messages\Phantom.Common.Messages.csproj" /> | ||||||
|     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> |     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> | ||||||
|     <ProjectReference Include="..\Phantom.Agent.Rpc\Phantom.Agent.Rpc.csproj" /> |     <ProjectReference Include="..\Phantom.Agent.Rpc\Phantom.Agent.Rpc.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|   | |||||||
| @@ -1,11 +1,11 @@ | |||||||
| using Phantom.Common.Data.Instance; | using Phantom.Agent.Rpc; | ||||||
|  | using Phantom.Common.Data.Instance; | ||||||
| using Phantom.Common.Data.Replies; | using Phantom.Common.Data.Replies; | ||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Common.Messages.Agent; | using Phantom.Common.Messages; | ||||||
| using Phantom.Common.Messages.Agent.BiDirectional; | using Phantom.Common.Messages.BiDirectional; | ||||||
| using Phantom.Common.Messages.Agent.ToAgent; | using Phantom.Common.Messages.ToAgent; | ||||||
| using Phantom.Common.Messages.Agent.ToController; | using Phantom.Common.Messages.ToServer; | ||||||
| using Phantom.Utils.Rpc; |  | ||||||
| using Phantom.Utils.Rpc.Message; | using Phantom.Utils.Rpc.Message; | ||||||
| using Serilog; | using Serilog; | ||||||
|  |  | ||||||
| @@ -14,11 +14,11 @@ namespace Phantom.Agent.Services.Rpc; | |||||||
| public sealed class MessageListener : IMessageToAgentListener { | public sealed class MessageListener : IMessageToAgentListener { | ||||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create<MessageListener>(); | 	private static ILogger Logger { get; } = PhantomLogger.Create<MessageListener>(); | ||||||
|  |  | ||||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | 	private readonly RpcServerConnection connection; | ||||||
| 	private readonly AgentServices agent; | 	private readonly AgentServices agent; | ||||||
| 	private readonly CancellationTokenSource shutdownTokenSource; | 	private readonly CancellationTokenSource shutdownTokenSource; | ||||||
|  |  | ||||||
| 	public MessageListener(RpcConnectionToServer<IMessageToControllerListener> connection, AgentServices agent, CancellationTokenSource shutdownTokenSource) { | 	public MessageListener(RpcServerConnection connection, AgentServices agent, CancellationTokenSource shutdownTokenSource) { | ||||||
| 		this.connection = connection; | 		this.connection = connection; | ||||||
| 		this.agent = agent; | 		this.agent = agent; | ||||||
| 		this.shutdownTokenSource = shutdownTokenSource; | 		this.shutdownTokenSource = shutdownTokenSource; | ||||||
| @@ -40,7 +40,7 @@ public sealed class MessageListener : IMessageToAgentListener { | |||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		await connection.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); | 		await ServerMessaging.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); | ||||||
| 		await agent.InstanceSessionManager.RefreshAgentStatus(); | 		await agent.InstanceSessionManager.RefreshAgentStatus(); | ||||||
| 		 | 		 | ||||||
| 		return NoReply.Instance; | 		return NoReply.Instance; | ||||||
| @@ -54,8 +54,6 @@ public sealed class MessageListener : IMessageToAgentListener { | |||||||
| 		}; | 		}; | ||||||
|  |  | ||||||
| 		Logger.Fatal("Agent authentication failed: {Error}", errorMessage); | 		Logger.Fatal("Agent authentication failed: {Error}", errorMessage); | ||||||
| 		 |  | ||||||
| 		PhantomLogger.Dispose(); |  | ||||||
| 		Environment.Exit(1); | 		Environment.Exit(1); | ||||||
|  |  | ||||||
| 		return Task.FromResult(NoReply.Instance); | 		return Task.FromResult(NoReply.Instance); | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| using NetMQ; | using NetMQ; | ||||||
| using Phantom.Common.Data; | using Phantom.Common.Data.Agent; | ||||||
| using Phantom.Common.Logging; | using Phantom.Common.Logging; | ||||||
| using Phantom.Utils.Cryptography; | using Phantom.Utils.Cryptography; | ||||||
| using Phantom.Utils.IO; | using Phantom.Utils.IO; | ||||||
| @@ -10,7 +10,7 @@ namespace Phantom.Agent; | |||||||
| static class AgentKey { | static class AgentKey { | ||||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create(nameof(AgentKey)); | 	private static ILogger Logger { get; } = PhantomLogger.Create(nameof(AgentKey)); | ||||||
|  |  | ||||||
| 	public static Task<(NetMQCertificate, AuthToken)?> Load(string? agentKeyToken, string? agentKeyFilePath) { | 	public static Task<(NetMQCertificate, AgentAuthToken)?> Load(string? agentKeyToken, string? agentKeyFilePath) { | ||||||
| 		if (agentKeyFilePath != null) { | 		if (agentKeyFilePath != null) { | ||||||
| 			return LoadFromFile(agentKeyFilePath); | 			return LoadFromFile(agentKeyFilePath); | ||||||
| 		} | 		} | ||||||
| @@ -22,7 +22,7 @@ static class AgentKey { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static async Task<(NetMQCertificate, AuthToken)?> LoadFromFile(string agentKeyFilePath) { | 	private static async Task<(NetMQCertificate, AgentAuthToken)?> LoadFromFile(string agentKeyFilePath) { | ||||||
| 		if (!File.Exists(agentKeyFilePath)) { | 		if (!File.Exists(agentKeyFilePath)) { | ||||||
| 			Logger.Fatal("Missing agent key file: {AgentKeyFilePath}", agentKeyFilePath); | 			Logger.Fatal("Missing agent key file: {AgentKeyFilePath}", agentKeyFilePath); | ||||||
| 			return null; | 			return null; | ||||||
| @@ -41,7 +41,7 @@ static class AgentKey { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static (NetMQCertificate, AuthToken)? LoadFromToken(string agentKey) { | 	private static (NetMQCertificate, AgentAuthToken)? LoadFromToken(string agentKey) { | ||||||
| 		try { | 		try { | ||||||
| 			return LoadFromBytes(TokenGenerator.DecodeBytes(agentKey)); | 			return LoadFromBytes(TokenGenerator.DecodeBytes(agentKey)); | ||||||
| 		} catch (Exception) { | 		} catch (Exception) { | ||||||
| @@ -50,11 +50,11 @@ static class AgentKey { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	private static (NetMQCertificate, AuthToken)? LoadFromBytes(byte[] agentKey) { | 	private static (NetMQCertificate, AgentAuthToken)? LoadFromBytes(byte[] agentKey) { | ||||||
| 		var (publicKey, agentToken) = ConnectionCommonKey.FromBytes(agentKey); | 		var (publicKey, agentToken) = AgentKeyData.FromBytes(agentKey); | ||||||
| 		var controllerCertificate = NetMQCertificate.FromPublicKey(publicKey); | 		var serverCertificate = NetMQCertificate.FromPublicKey(publicKey); | ||||||
| 		 | 		 | ||||||
| 		Logger.Information("Loaded agent key."); | 		Logger.Information("Loaded agent key."); | ||||||
| 		return (controllerCertificate, agentToken); | 		return (serverCertificate, agentToken); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
| @@ -11,7 +12,8 @@ | |||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> |     <ProjectReference Include="..\..\Utils\Phantom.Utils.IO\Phantom.Utils.IO.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Runtime\Phantom.Utils.Runtime.csproj" /> | ||||||
|     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> |     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> | ||||||
|     <ProjectReference Include="..\Phantom.Agent.Services\Phantom.Agent.Services.csproj" /> |     <ProjectReference Include="..\Phantom.Agent.Services\Phantom.Agent.Services.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|   | |||||||
| @@ -1,17 +1,12 @@ | |||||||
| using System.Reflection; | using System.Reflection; | ||||||
| using NetMQ; |  | ||||||
| using Phantom.Agent; | using Phantom.Agent; | ||||||
| using Phantom.Agent.Rpc; | 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.Logging; | ||||||
| using Phantom.Common.Messages.Agent; |  | ||||||
| using Phantom.Common.Messages.Agent.ToController; |  | ||||||
| using Phantom.Utils.Rpc; | using Phantom.Utils.Rpc; | ||||||
| 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; | ||||||
|  |  | ||||||
| @@ -22,45 +17,44 @@ 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()); | ||||||
|  |  | ||||||
| 	PhantomLogger.Root.InformationHeading("Initializing Phantom Panel agent..."); | 	PhantomLogger.Root.InformationHeading("Initializing Phantom Panel agent..."); | ||||||
| 	PhantomLogger.Root.Information("Agent version: {Version}", fullVersion); | 	PhantomLogger.Root.Information("Agent version: {Version}", fullVersion); | ||||||
|  |  | ||||||
| 	var (controllerHost, controllerPort, javaSearchPath, agentKeyToken, agentKeyFilePath, agentName, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts, maxConcurrentBackupCompressionTasks) = Variables.LoadOrStop(); | 	var (serverHost, serverPort, javaSearchPath, agentKeyToken, agentKeyFilePath, agentName, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts) = Variables.LoadOrExit(); | ||||||
|  |  | ||||||
| 	var agentKey = await AgentKey.Load(agentKeyToken, agentKeyFilePath); | 	var agentKey = await AgentKey.Load(agentKeyToken, agentKeyFilePath); | ||||||
| 	if (agentKey == null) { | 	if (agentKey == null) { | ||||||
| 		return 1; | 		Environment.Exit(1); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var folders = new AgentFolders("./data", "./temp", javaSearchPath); | 	var folders = new AgentFolders("./data", "./temp", javaSearchPath); | ||||||
| 	if (!folders.TryCreate()) { | 	if (!folders.TryCreate()) { | ||||||
| 		return 1; | 		Environment.Exit(1); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var agentGuid = await GuidFile.CreateOrLoad(folders.DataFolderPath); | 	var agentGuid = await GuidFile.CreateOrLoad(folders.DataFolderPath); | ||||||
| 	if (agentGuid == null) { | 	if (agentGuid == null) { | ||||||
| 		return 1; | 		Environment.Exit(1); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var (controllerCertificate, agentToken) = agentKey.Value; | 	var (serverCertificate, agentToken) = agentKey.Value; | ||||||
| 	var agentInfo = new AgentInfo(agentGuid.Value, agentName, ProtocolVersion, fullVersion, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts); | 	var agentInfo = new AgentInfo(agentGuid.Value, agentName, ProtocolVersion, fullVersion, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts); | ||||||
|  | 	var agentServices = new AgentServices(agentInfo, folders); | ||||||
|  |  | ||||||
|  | 	MessageListener MessageListenerFactory(RpcServerConnection connection) { | ||||||
|  | 		return new MessageListener(connection, agentServices, shutdownCancellationTokenSource); | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	PhantomLogger.Root.InformationHeading("Launching Phantom Panel agent..."); | 	PhantomLogger.Root.InformationHeading("Launching Phantom Panel agent..."); | ||||||
|  |  | ||||||
| 	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 agentServices = new AgentServices(agentInfo, folders, new AgentServiceConfiguration(maxConcurrentBackupCompressionTasks), new ControllerConnection(rpcSocket.Connection)); |  | ||||||
| 	await agentServices.Initialize(); | 	await agentServices.Initialize(); | ||||||
|  |  | ||||||
| 	var rpcDisconnectSemaphore = new SemaphoreSlim(0, 1); | 	var rpcDisconnectSemaphore = new SemaphoreSlim(0, 1); | ||||||
| 	var rpcMessageListener = new MessageListener(rpcSocket.Connection, agentServices, shutdownCancellationTokenSource); | 	var rpcConfiguration = new RpcConfiguration(PhantomLogger.Create("Rpc"), PhantomLogger.Create<TaskManager>("Rpc"), serverHost, serverPort, serverCertificate); | ||||||
| 	var rpcTask = RpcClientRuntime.Launch(rpcSocket, agentInfo, rpcMessageListener, rpcDisconnectSemaphore, shutdownCancellationToken); | 	var rpcTask = RpcLauncher.Launch(rpcConfiguration, agentToken, agentInfo, MessageListenerFactory, rpcDisconnectSemaphore, shutdownCancellationToken); | ||||||
| 	try { | 	try { | ||||||
| 		await rpcTask.WaitAsync(shutdownCancellationToken); | 		await rpcTask.WaitAsync(shutdownCancellationToken); | ||||||
| 	} finally { | 	} finally { | ||||||
| @@ -70,18 +64,11 @@ try { | |||||||
| 		rpcDisconnectSemaphore.Release(); | 		rpcDisconnectSemaphore.Release(); | ||||||
| 		await rpcTask; | 		await rpcTask; | ||||||
| 		rpcDisconnectSemaphore.Dispose(); | 		rpcDisconnectSemaphore.Dispose(); | ||||||
| 		 |  | ||||||
| 		NetMQConfig.Cleanup(); |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return 0; |  | ||||||
| } catch (OperationCanceledException) { | } catch (OperationCanceledException) { | ||||||
| 	return 0; | 	// Ignore. | ||||||
| } catch (StopProcedureException) { |  | ||||||
| 	return 1; |  | ||||||
| } catch (Exception e) { | } catch (Exception e) { | ||||||
| 	PhantomLogger.Root.Fatal(e, "Caught exception in entry point."); | 	PhantomLogger.Root.Fatal(e, "Caught exception in entry point."); | ||||||
| 	return 1; |  | ||||||
| } finally { | } finally { | ||||||
| 	shutdownCancellationTokenSource.Dispose(); | 	shutdownCancellationTokenSource.Dispose(); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -6,8 +6,8 @@ using Phantom.Utils.Runtime; | |||||||
| namespace Phantom.Agent; | namespace Phantom.Agent; | ||||||
|  |  | ||||||
| sealed record Variables( | sealed record Variables( | ||||||
| 	string ControllerHost, | 	string ServerHost, | ||||||
| 	ushort ControllerPort, | 	ushort ServerPort, | ||||||
| 	string JavaSearchPath, | 	string JavaSearchPath, | ||||||
| 	string? AgentKeyToken, | 	string? AgentKeyToken, | ||||||
| 	string? AgentKeyFilePath, | 	string? AgentKeyFilePath, | ||||||
| @@ -15,16 +15,15 @@ sealed record Variables( | |||||||
| 	ushort MaxInstances, | 	ushort MaxInstances, | ||||||
| 	RamAllocationUnits MaxMemory, | 	RamAllocationUnits MaxMemory, | ||||||
| 	AllowedPorts AllowedServerPorts, | 	AllowedPorts AllowedServerPorts, | ||||||
| 	AllowedPorts AllowedRconPorts, | 	AllowedPorts AllowedRconPorts | ||||||
| 	ushort MaxConcurrentBackupCompressionTasks |  | ||||||
| ) { | ) { | ||||||
| 	private static Variables LoadOrThrow() { | 	private static Variables LoadOrThrow() { | ||||||
| 		var (agentKeyToken, agentKeyFilePath) = EnvironmentVariables.GetEitherString("AGENT_KEY", "AGENT_KEY_FILE").Require; | 		var (agentKeyToken, agentKeyFilePath) = EnvironmentVariables.GetEitherString("AGENT_KEY", "AGENT_KEY_FILE").Require; | ||||||
| 		var javaSearchPath = EnvironmentVariables.GetString("JAVA_SEARCH_PATH").WithDefaultGetter(GetDefaultJavaSearchPath); | 		var javaSearchPath = EnvironmentVariables.GetString("JAVA_SEARCH_PATH").WithDefaultGetter(GetDefaultJavaSearchPath); | ||||||
|  |  | ||||||
| 		return new Variables( | 		return new Variables( | ||||||
| 			EnvironmentVariables.GetString("CONTROLLER_HOST").Require, | 			EnvironmentVariables.GetString("SERVER_HOST").Require, | ||||||
| 			EnvironmentVariables.GetPortNumber("CONTROLLER_PORT").WithDefault(9401), | 			EnvironmentVariables.GetPortNumber("SERVER_PORT").WithDefault(9401), | ||||||
| 			javaSearchPath, | 			javaSearchPath, | ||||||
| 			agentKeyToken, | 			agentKeyToken, | ||||||
| 			agentKeyFilePath, | 			agentKeyFilePath, | ||||||
| @@ -32,8 +31,7 @@ sealed record Variables( | |||||||
| 			(ushort) EnvironmentVariables.GetInteger("MAX_INSTANCES", min: 1, max: 10000).Require, | 			(ushort) EnvironmentVariables.GetInteger("MAX_INSTANCES", min: 1, max: 10000).Require, | ||||||
| 			EnvironmentVariables.GetString("MAX_MEMORY").MapParse(RamAllocationUnits.FromString).Require, | 			EnvironmentVariables.GetString("MAX_MEMORY").MapParse(RamAllocationUnits.FromString).Require, | ||||||
| 			EnvironmentVariables.GetString("ALLOWED_SERVER_PORTS").MapParse(AllowedPorts.FromString).Require, | 			EnvironmentVariables.GetString("ALLOWED_SERVER_PORTS").MapParse(AllowedPorts.FromString).Require, | ||||||
| 			EnvironmentVariables.GetString("ALLOWED_RCON_PORTS").MapParse(AllowedPorts.FromString).Require, | 			EnvironmentVariables.GetString("ALLOWED_RCON_PORTS").MapParse(AllowedPorts.FromString).Require | ||||||
| 			(ushort) EnvironmentVariables.GetInteger("MAX_CONCURRENT_BACKUP_COMPRESSION_TASKS", min: 1, max: 10000).WithDefault(1) |  | ||||||
| 		); | 		); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -41,12 +39,13 @@ sealed record Variables( | |||||||
| 		return JavaRuntimeDiscovery.GetSystemSearchPath() ?? throw new Exception("Could not automatically determine the path to Java installations on this system. Please set the JAVA_SEARCH_PATH environment variable to the folder containing Java installations."); | 		return JavaRuntimeDiscovery.GetSystemSearchPath() ?? throw new Exception("Could not automatically determine the path to Java installations on this system. Please set the JAVA_SEARCH_PATH environment variable to the folder containing Java installations."); | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public static Variables LoadOrStop() { | 	public static Variables LoadOrExit() { | ||||||
| 		try { | 		try { | ||||||
| 			return LoadOrThrow(); | 			return LoadOrThrow(); | ||||||
| 		} catch (Exception e) { | 		} catch (Exception e) { | ||||||
| 			PhantomLogger.Root.Fatal(e.Message); | 			PhantomLogger.Root.Fatal(e.Message); | ||||||
| 			throw StopProcedureException.Instance; | 			Environment.Exit(1); | ||||||
|  | 			throw; | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
| @@ -14,6 +15,7 @@ | |||||||
|     <PackageReference Include="NUnit" /> |     <PackageReference Include="NUnit" /> | ||||||
|     <PackageReference Include="NUnit3TestAdapter" /> |     <PackageReference Include="NUnit3TestAdapter" /> | ||||||
|     <PackageReference Include="NUnit.Analyzers" /> |     <PackageReference Include="NUnit.Analyzers" /> | ||||||
|  |     <PackageReference Include="coverlet.collector" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|   | |||||||
| @@ -1,5 +0,0 @@ | |||||||
| using System.Text.Json; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.AuditLog; |  | ||||||
|  |  | ||||||
| public sealed record AuditLogItem(DateTime UtcTime, Guid? UserGuid, string? UserName, AuditLogEventType EventType, AuditLogSubjectType SubjectType, string? SubjectId, JsonDocument? Data); |  | ||||||
| @@ -1,8 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| public enum AddRoleError : byte { |  | ||||||
| 	NameIsEmpty, |  | ||||||
| 	NameIsTooLong, |  | ||||||
| 	NameAlreadyExists, |  | ||||||
| 	UnknownError |  | ||||||
| } |  | ||||||
| @@ -1,28 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.AddUserErrors; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(NameIsInvalid))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(PasswordIsInvalid))] |  | ||||||
| 	[MemoryPackUnion(2, typeof(NameAlreadyExists))] |  | ||||||
| 	[MemoryPackUnion(3, typeof(UnknownError))] |  | ||||||
| 	public abstract partial record AddUserError { |  | ||||||
| 		internal AddUserError() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.AddUserErrors { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record NameIsInvalid([property: MemoryPackOrder(0)] UsernameRequirementViolation Violation) : AddUserError; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record PasswordIsInvalid([property: MemoryPackOrder(0)] ImmutableArray<PasswordRequirementViolation> Violations) : AddUserError; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record NameAlreadyExists : AddUserError; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UnknownError : AddUserError; |  | ||||||
| } |  | ||||||
| @@ -1,31 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.CreateOrUpdateAdministratorUserResults; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(Success))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(CreationFailed))] |  | ||||||
| 	[MemoryPackUnion(2, typeof(UpdatingFailed))] |  | ||||||
| 	[MemoryPackUnion(3, typeof(AddingToRoleFailed))] |  | ||||||
| 	[MemoryPackUnion(4, typeof(UnknownError))] |  | ||||||
| 	public abstract partial record CreateOrUpdateAdministratorUserResult { |  | ||||||
| 		internal CreateOrUpdateAdministratorUserResult() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.CreateOrUpdateAdministratorUserResults { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record Success([property: MemoryPackOrder(0)] UserInfo User) : CreateOrUpdateAdministratorUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record CreationFailed([property: MemoryPackOrder(0)] AddUserError Error) : CreateOrUpdateAdministratorUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UpdatingFailed([property: MemoryPackOrder(0)] SetUserPasswordError Error) : CreateOrUpdateAdministratorUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record AddingToRoleFailed : CreateOrUpdateAdministratorUserResult; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UnknownError : CreateOrUpdateAdministratorUserResult; |  | ||||||
| } |  | ||||||
| @@ -1,7 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| public enum DeleteUserResult : byte { |  | ||||||
| 	Deleted, |  | ||||||
| 	NotFound, |  | ||||||
| 	Failed |  | ||||||
| } |  | ||||||
| @@ -1,25 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| public sealed class IdentityPermissions { |  | ||||||
| 	public static IdentityPermissions None { get; } = new (ImmutableHashSet<string>.Empty); |  | ||||||
| 	 |  | ||||||
| 	private readonly ImmutableHashSet<string> permissionIds; |  | ||||||
|  |  | ||||||
| 	public IdentityPermissions(ImmutableHashSet<string> permissionIdsQuery) { |  | ||||||
| 		this.permissionIds = permissionIdsQuery; |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public bool Check(Permission? permission) { |  | ||||||
| 		while (permission != null) { |  | ||||||
| 			if (!permissionIds.Contains(permission.Id)) { |  | ||||||
| 				return false; |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			permission = permission.Parent; |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		return true; |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,27 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.PasswordRequirementViolations; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(TooShort))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(MustContainLowercaseLetter))] |  | ||||||
| 	[MemoryPackUnion(2, typeof(MustContainUppercaseLetter))] |  | ||||||
| 	[MemoryPackUnion(3, typeof(MustContainDigit))] |  | ||||||
| 	public abstract partial record PasswordRequirementViolation { |  | ||||||
| 		internal PasswordRequirementViolation() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.PasswordRequirementViolations { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record TooShort([property: MemoryPackOrder(0)] int MinimumLength) : PasswordRequirementViolation; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record MustContainLowercaseLetter : PasswordRequirementViolation; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record MustContainUppercaseLetter : PasswordRequirementViolation; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record MustContainDigit : PasswordRequirementViolation; |  | ||||||
| } |  | ||||||
| @@ -1,9 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record RoleInfo( |  | ||||||
| 	[property: MemoryPackOrder(0)] Guid Guid, |  | ||||||
| 	[property: MemoryPackOrder(1)] string Name |  | ||||||
| ); |  | ||||||
| @@ -1,24 +0,0 @@ | |||||||
| using System.Collections.Immutable; |  | ||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.SetUserPasswordErrors; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(UserNotFound))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(PasswordIsInvalid))] |  | ||||||
| 	[MemoryPackUnion(2, typeof(UnknownError))] |  | ||||||
| 	public abstract partial record SetUserPasswordError { |  | ||||||
| 		internal SetUserPasswordError() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.SetUserPasswordErrors { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UserNotFound : SetUserPasswordError; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record PasswordIsInvalid([property: MemoryPackOrder(0)] ImmutableArray<PasswordRequirementViolation> Violations) : SetUserPasswordError; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record UnknownError : SetUserPasswordError; |  | ||||||
| } |  | ||||||
| @@ -1,9 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| public sealed partial record UserInfo( |  | ||||||
| 	[property: MemoryPackOrder(0)] Guid Guid, |  | ||||||
| 	[property: MemoryPackOrder(1)] string Name |  | ||||||
| ); |  | ||||||
| @@ -1,12 +0,0 @@ | |||||||
| namespace Phantom.Common.Data.Web.Users; |  | ||||||
|  |  | ||||||
| public static class UserPasswords { |  | ||||||
| 	public static string Hash(string password) { |  | ||||||
| 		return BCrypt.Net.BCrypt.HashPassword(password, workFactor: 12); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static bool Verify(string password, string hash) { |  | ||||||
| 		// TODO rehash |  | ||||||
| 		return BCrypt.Net.BCrypt.Verify(password, hash); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,19 +0,0 @@ | |||||||
| using MemoryPack; |  | ||||||
| using Phantom.Common.Data.Web.Users.UsernameRequirementViolations; |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users { |  | ||||||
| 	[MemoryPackable] |  | ||||||
| 	[MemoryPackUnion(0, typeof(IsEmpty))] |  | ||||||
| 	[MemoryPackUnion(1, typeof(TooLong))] |  | ||||||
| 	public abstract partial record UsernameRequirementViolation { |  | ||||||
| 		internal UsernameRequirementViolation() {} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Web.Users.UsernameRequirementViolations { |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record IsEmpty : UsernameRequirementViolation; |  | ||||||
|  |  | ||||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] |  | ||||||
| 	public sealed partial record TooLong([property: MemoryPackOrder(0)] int MaxLength) : UsernameRequirementViolation; |  | ||||||
| } |  | ||||||
| @@ -2,18 +2,18 @@ | |||||||
| using System.Security.Cryptography; | using System.Security.Cryptography; | ||||||
| using MemoryPack; | using MemoryPack; | ||||||
| 
 | 
 | ||||||
| namespace Phantom.Common.Data; | namespace Phantom.Common.Data.Agent; | ||||||
| 
 | 
 | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")] | [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")] | ||||||
| public sealed partial class AuthToken { | public sealed partial class AgentAuthToken { | ||||||
| 	internal const int Length = 12; | 	internal const int Length = 12; | ||||||
| 
 | 
 | ||||||
| 	[MemoryPackOrder(0)] | 	[MemoryPackOrder(0)] | ||||||
| 	[MemoryPackInclude] | 	[MemoryPackInclude] | ||||||
| 	private readonly byte[] bytes; | 	private readonly byte[] bytes; | ||||||
| 
 | 
 | ||||||
| 	internal AuthToken(byte[]? bytes) { | 	internal AgentAuthToken(byte[]? bytes) { | ||||||
| 		if (bytes == null) { | 		if (bytes == null) { | ||||||
| 			throw new ArgumentNullException(nameof(bytes)); | 			throw new ArgumentNullException(nameof(bytes)); | ||||||
| 		} | 		} | ||||||
| @@ -25,7 +25,7 @@ public sealed partial class AuthToken { | |||||||
| 		this.bytes = bytes; | 		this.bytes = bytes; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	public bool FixedTimeEquals(AuthToken providedAuthToken) { | 	public bool FixedTimeEquals(AgentAuthToken providedAuthToken) { | ||||||
| 		return CryptographicOperations.FixedTimeEquals(bytes, providedAuthToken.bytes); | 		return CryptographicOperations.FixedTimeEquals(bytes, providedAuthToken.bytes); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| @@ -33,7 +33,7 @@ public sealed partial class AuthToken { | |||||||
| 		bytes.CopyTo(span); | 		bytes.CopyTo(span); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	public static AuthToken Generate() { | 	public static AgentAuthToken Generate() { | ||||||
| 		return new AuthToken(RandomNumberGenerator.GetBytes(Length)); | 		return new AgentAuthToken(RandomNumberGenerator.GetBytes(Length)); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Agent; | namespace Phantom.Common.Data.Agent; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record AgentInfo( | public sealed partial record AgentInfo( | ||||||
| 	[property: MemoryPackOrder(0)] Guid Guid, | 	[property: MemoryPackOrder(0)] Guid Guid, | ||||||
| 	[property: MemoryPackOrder(1)] string Name, | 	[property: MemoryPackOrder(1)] string Name, | ||||||
|   | |||||||
							
								
								
									
										18
									
								
								Common/Phantom.Common.Data/Agent/AgentKeyData.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								Common/Phantom.Common.Data/Agent/AgentKeyData.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,18 @@ | |||||||
|  | namespace Phantom.Common.Data.Agent; | ||||||
|  |  | ||||||
|  | public static class AgentKeyData { | ||||||
|  | 	private const byte TokenLength = AgentAuthToken.Length; | ||||||
|  |  | ||||||
|  | 	public static byte[] ToBytes(byte[] publicKey, AgentAuthToken agentToken) { | ||||||
|  | 		Span<byte> agentKey = stackalloc byte[TokenLength + publicKey.Length]; | ||||||
|  | 		agentToken.WriteTo(agentKey[..TokenLength]); | ||||||
|  | 		publicKey.CopyTo(agentKey[TokenLength..]); | ||||||
|  | 		return agentKey.ToArray(); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	public static (byte[] PublicKey, AgentAuthToken AgentToken) FromBytes(byte[] agentKey) { | ||||||
|  | 		var token = new AgentAuthToken(agentKey[..TokenLength]); | ||||||
|  | 		var publicKey = agentKey[TokenLength..]; | ||||||
|  | 		return (publicKey, token); | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -4,7 +4,7 @@ using MemoryPack; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data; | namespace Phantom.Common.Data; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial class AllowedPorts { | public sealed partial class AllowedPorts { | ||||||
| 	[MemoryPackOrder(0)] | 	[MemoryPackOrder(0)] | ||||||
| 	[MemoryPackInclude] | 	[MemoryPackInclude] | ||||||
| @@ -53,6 +53,6 @@ public sealed partial class AllowedPorts { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	public static AllowedPorts FromString(string definitions) { | 	public static AllowedPorts FromString(string definitions) { | ||||||
| 		return FromString(definitions.AsSpan()); | 		return FromString((ReadOnlySpan<char>) definitions); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Backups; | namespace Phantom.Common.Data.Backups; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record BackupCreationResult( | public sealed partial record BackupCreationResult( | ||||||
| 	[property: MemoryPackOrder(0)] BackupCreationResultKind Kind, | 	[property: MemoryPackOrder(0)] BackupCreationResultKind Kind, | ||||||
| 	[property: MemoryPackOrder(1)] BackupCreationWarnings Warnings = BackupCreationWarnings.None | 	[property: MemoryPackOrder(1)] BackupCreationWarnings Warnings = BackupCreationWarnings.None | ||||||
|   | |||||||
| @@ -20,4 +20,18 @@ public static class BackupCreationResultSummaryExtensions { | |||||||
| 		       kind != BackupCreationResultKind.BackupAlreadyRunning && | 		       kind != BackupCreationResultKind.BackupAlreadyRunning && | ||||||
| 		       kind != BackupCreationResultKind.BackupFileAlreadyExists; | 		       kind != BackupCreationResultKind.BackupFileAlreadyExists; | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	public static string ToSentence(this BackupCreationResultKind kind) { | ||||||
|  | 		return kind switch { | ||||||
|  | 			BackupCreationResultKind.Success                            => "Backup created successfully.", | ||||||
|  | 			BackupCreationResultKind.InstanceNotRunning                 => "Instance is not running.", | ||||||
|  | 			BackupCreationResultKind.BackupCancelled                    => "Backup cancelled.", | ||||||
|  | 			BackupCreationResultKind.BackupAlreadyRunning               => "A backup is already being created.", | ||||||
|  | 			BackupCreationResultKind.BackupFileAlreadyExists            => "Backup with the same name already exists.", | ||||||
|  | 			BackupCreationResultKind.CouldNotCreateBackupFolder         => "Could not create backup folder.", | ||||||
|  | 			BackupCreationResultKind.CouldNotCopyWorldToTemporaryFolder => "Could not copy world to temporary folder.", | ||||||
|  | 			BackupCreationResultKind.CouldNotCreateWorldArchive         => "Could not create world archive.", | ||||||
|  | 			_                                                           => "Unknown error." | ||||||
|  | 		}; | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,18 +0,0 @@ | |||||||
| namespace Phantom.Common.Data; |  | ||||||
|  |  | ||||||
| public readonly record struct ConnectionCommonKey(byte[] CertificatePublicKey, AuthToken AuthToken) { |  | ||||||
| 	private const byte TokenLength = AuthToken.Length; |  | ||||||
|  |  | ||||||
| 	public byte[] ToBytes() { |  | ||||||
| 		Span<byte> result = stackalloc byte[TokenLength + CertificatePublicKey.Length]; |  | ||||||
| 		AuthToken.WriteTo(result[..TokenLength]); |  | ||||||
| 		CertificatePublicKey.CopyTo(result[TokenLength..]); |  | ||||||
| 		return result.ToArray(); |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	public static ConnectionCommonKey FromBytes(byte[] agentKey) { |  | ||||||
| 		var authToken = new AuthToken(agentKey[..TokenLength]); |  | ||||||
| 		var certificatePublicKey = agentKey[TokenLength..]; |  | ||||||
| 		return new ConnectionCommonKey(certificatePublicKey, authToken); |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -13,35 +13,35 @@ public partial interface IInstanceEvent { | |||||||
| 	void Accept(IInstanceEventVisitor visitor); | 	void Accept(IInstanceEventVisitor visitor); | ||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceLaunchSuccededEvent : IInstanceEvent { | public sealed partial record InstanceLaunchSuccededEvent : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnLaunchSucceeded(this); | 		visitor.OnLaunchSucceeded(this); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceLaunchFailedEvent([property: MemoryPackOrder(0)] InstanceLaunchFailReason Reason) : IInstanceEvent { | public sealed partial record InstanceLaunchFailedEvent([property: MemoryPackOrder(0)] InstanceLaunchFailReason Reason) : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnLaunchFailed(this); | 		visitor.OnLaunchFailed(this); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceCrashedEvent : IInstanceEvent { | public sealed partial record InstanceCrashedEvent : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnCrashed(this); | 		visitor.OnCrashed(this); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceStoppedEvent : IInstanceEvent { | public sealed partial record InstanceStoppedEvent : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnStopped(this); | 		visitor.OnStopped(this); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceBackupCompletedEvent([property: MemoryPackOrder(0)] BackupCreationResultKind Kind, [property: MemoryPackOrder(1)] BackupCreationWarnings Warnings) : IInstanceEvent { | public sealed partial record InstanceBackupCompletedEvent([property: MemoryPackOrder(0)] BackupCreationResultKind Kind, [property: MemoryPackOrder(1)] BackupCreationWarnings Warnings) : IInstanceEvent { | ||||||
| 	public void Accept(IInstanceEventVisitor visitor) { | 	public void Accept(IInstanceEventVisitor visitor) { | ||||||
| 		visitor.OnBackupCompleted(this); | 		visitor.OnBackupCompleted(this); | ||||||
|   | |||||||
| @@ -14,31 +14,31 @@ namespace Phantom.Common.Data.Instance; | |||||||
| [MemoryPackUnion(8, typeof(InstanceIsFailed))] | [MemoryPackUnion(8, typeof(InstanceIsFailed))] | ||||||
| public partial interface IInstanceStatus {} | public partial interface IInstanceStatus {} | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsOffline : IInstanceStatus; | public sealed partial record InstanceIsOffline : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsInvalid([property: MemoryPackOrder(0)] string Reason) : IInstanceStatus; | public sealed partial record InstanceIsInvalid([property: MemoryPackOrder(0)] string Reason) : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsNotRunning : IInstanceStatus; | public sealed partial record InstanceIsNotRunning : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsDownloading([property: MemoryPackOrder(0)] byte Progress) : IInstanceStatus; | public sealed partial record InstanceIsDownloading([property: MemoryPackOrder(0)] byte Progress) : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsLaunching : IInstanceStatus; | public sealed partial record InstanceIsLaunching : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsRunning : IInstanceStatus; | public sealed partial record InstanceIsRunning : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsRestarting : IInstanceStatus; | public sealed partial record InstanceIsRestarting : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsStopping : IInstanceStatus; | public sealed partial record InstanceIsStopping : IInstanceStatus; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceIsFailed([property: MemoryPackOrder(0)] InstanceLaunchFailReason Reason) : IInstanceStatus; | public sealed partial record InstanceIsFailed([property: MemoryPackOrder(0)] InstanceLaunchFailReason Reason) : IInstanceStatus; | ||||||
|  |  | ||||||
| public static class InstanceStatus { | public static class InstanceStatus { | ||||||
|   | |||||||
| @@ -4,7 +4,7 @@ using Phantom.Common.Data.Minecraft; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Instance; | namespace Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceConfiguration( | public sealed partial record InstanceConfiguration( | ||||||
| 	[property: MemoryPackOrder(0)] Guid AgentGuid, | 	[property: MemoryPackOrder(0)] Guid AgentGuid, | ||||||
| 	[property: MemoryPackOrder(1)] Guid InstanceGuid, | 	[property: MemoryPackOrder(1)] Guid InstanceGuid, | ||||||
|   | |||||||
| @@ -1,14 +1,33 @@ | |||||||
| namespace Phantom.Common.Data.Instance; | namespace Phantom.Common.Data.Instance; | ||||||
|  |  | ||||||
| public enum InstanceLaunchFailReason : byte { | public enum InstanceLaunchFailReason : byte { | ||||||
| 	UnknownError                           = 0, | 	UnknownError, | ||||||
| 	ServerPortNotAllowed                   = 1, | 	ServerPortNotAllowed, | ||||||
| 	ServerPortAlreadyInUse                 = 2, | 	ServerPortAlreadyInUse, | ||||||
| 	RconPortNotAllowed                     = 3, | 	RconPortNotAllowed, | ||||||
| 	RconPortAlreadyInUse                   = 4, | 	RconPortAlreadyInUse, | ||||||
| 	JavaRuntimeNotFound                    = 5, | 	JavaRuntimeNotFound, | ||||||
| 	CouldNotDownloadMinecraftServer        = 6, | 	InvalidJvmArguments, | ||||||
| 	CouldNotConfigureMinecraftServer       = 7, | 	CouldNotDownloadMinecraftServer, | ||||||
| 	CouldNotPrepareMinecraftServerLauncher = 8, | 	CouldNotConfigureMinecraftServer, | ||||||
| 	CouldNotStartMinecraftServer           = 9 | 	CouldNotPrepareMinecraftServerLauncher, | ||||||
|  | 	CouldNotStartMinecraftServer | ||||||
|  | } | ||||||
|  |  | ||||||
|  | public static class InstanceLaunchFailReasonExtensions { | ||||||
|  | 	public static string ToSentence(this InstanceLaunchFailReason reason) { | ||||||
|  | 		return reason switch { | ||||||
|  | 			InstanceLaunchFailReason.ServerPortNotAllowed                   => "Server port not allowed.", | ||||||
|  | 			InstanceLaunchFailReason.ServerPortAlreadyInUse                 => "Server port already in use.", | ||||||
|  | 			InstanceLaunchFailReason.RconPortNotAllowed                     => "Rcon port not allowed.", | ||||||
|  | 			InstanceLaunchFailReason.RconPortAlreadyInUse                   => "Rcon port already in use.", | ||||||
|  | 			InstanceLaunchFailReason.JavaRuntimeNotFound                    => "Java runtime not found.", | ||||||
|  | 			InstanceLaunchFailReason.InvalidJvmArguments                    => "Invalid JVM arguments.", | ||||||
|  | 			InstanceLaunchFailReason.CouldNotDownloadMinecraftServer        => "Could not download Minecraft server.", | ||||||
|  | 			InstanceLaunchFailReason.CouldNotConfigureMinecraftServer       => "Could not configure Minecraft server.", | ||||||
|  | 			InstanceLaunchFailReason.CouldNotPrepareMinecraftServerLauncher => "Could not prepare Minecraft server launcher.", | ||||||
|  | 			InstanceLaunchFailReason.CouldNotStartMinecraftServer           => "Could not start Minecraft server.", | ||||||
|  | 			_                                                               => "Unknown error." | ||||||
|  | 		}; | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ using Phantom.Common.Data.Minecraft; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Instance;  | namespace Phantom.Common.Data.Instance;  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceLaunchProperties( | public sealed partial record InstanceLaunchProperties( | ||||||
| 	[property: MemoryPackOrder(0)] FileDownloadInfo? ServerDownloadInfo | 	[property: MemoryPackOrder(0)] FileDownloadInfo? ServerDownloadInfo | ||||||
| ); | ); | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ using MemoryPack; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Java; | namespace Phantom.Common.Data.Java; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record JavaRuntime( | public sealed partial record JavaRuntime( | ||||||
| 	[property: MemoryPackOrder(0)] string MainVersion, | 	[property: MemoryPackOrder(0)] string MainVersion, | ||||||
| 	[property: MemoryPackOrder(1)] string FullVersion, | 	[property: MemoryPackOrder(1)] string FullVersion, | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Java;  | namespace Phantom.Common.Data.Java;  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record TaggedJavaRuntime( | public sealed partial record TaggedJavaRuntime( | ||||||
| 	[property: MemoryPackOrder(0)] Guid Guid, | 	[property: MemoryPackOrder(0)] Guid Guid, | ||||||
| 	[property: MemoryPackOrder(1)] JavaRuntime Runtime | 	[property: MemoryPackOrder(1)] JavaRuntime Runtime | ||||||
|   | |||||||
| @@ -4,7 +4,7 @@ using Phantom.Utils.IO; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Minecraft;  | namespace Phantom.Common.Data.Minecraft;  | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial class FileDownloadInfo { | public sealed partial class FileDownloadInfo { | ||||||
| 	[MemoryPackOrder(0)] | 	[MemoryPackOrder(0)] | ||||||
| 	public string DownloadUrl { get; } | 	public string DownloadUrl { get; } | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Minecraft; | namespace Phantom.Common.Data.Minecraft; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public readonly partial record struct MinecraftStopStrategy( | public readonly partial record struct MinecraftStopStrategy( | ||||||
| 	[property: MemoryPackOrder(0)] ushort Seconds | 	[property: MemoryPackOrder(0)] ushort Seconds | ||||||
| ) { | ) { | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| <Project Sdk="Microsoft.NET.Sdk"> | <Project Sdk="Microsoft.NET.Sdk"> | ||||||
|    |    | ||||||
|   <PropertyGroup> |   <PropertyGroup> | ||||||
|  |     <TargetFramework>net7.0</TargetFramework> | ||||||
|     <ImplicitUsings>enable</ImplicitUsings> |     <ImplicitUsings>enable</ImplicitUsings> | ||||||
|     <Nullable>enable</Nullable> |     <Nullable>enable</Nullable> | ||||||
|   </PropertyGroup> |   </PropertyGroup> | ||||||
| @@ -10,7 +11,8 @@ | |||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|    |    | ||||||
|   <ItemGroup> |   <ItemGroup> | ||||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> |     <ProjectReference Include="..\..\Utils\Phantom.Utils.Cryptography\Phantom.Utils.Cryptography.csproj" /> | ||||||
|  |     <ProjectReference Include="..\..\Utils\Phantom.Utils.IO\Phantom.Utils.IO.csproj" /> | ||||||
|   </ItemGroup> |   </ItemGroup> | ||||||
|  |  | ||||||
| </Project> | </Project> | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ using MemoryPack; | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data; | namespace Phantom.Common.Data; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| readonly partial record struct PortRange( | readonly partial record struct PortRange( | ||||||
| 	[property: MemoryPackOrder(0)] ushort FirstPort, | 	[property: MemoryPackOrder(0)] ushort FirstPort, | ||||||
| 	[property: MemoryPackOrder(1)] ushort LastPort | 	[property: MemoryPackOrder(1)] ushort LastPort | ||||||
|   | |||||||
| @@ -6,7 +6,7 @@ namespace Phantom.Common.Data; | |||||||
| /// <summary> | /// <summary> | ||||||
| /// Represents a number of RAM allocation units, using the conversion factor of 256 MB per unit. Supports allocations up to 16 TB minus 256 MB (65535 units). | /// Represents a number of RAM allocation units, using the conversion factor of 256 MB per unit. Supports allocations up to 16 TB minus 256 MB (65535 units). | ||||||
| /// </summary> | /// </summary> | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")] | [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")] | ||||||
| public readonly partial record struct RamAllocationUnits( | public readonly partial record struct RamAllocationUnits( | ||||||
| 	[property: MemoryPackOrder(0)] ushort RawValue | 	[property: MemoryPackOrder(0)] ushort RawValue | ||||||
| @@ -98,6 +98,6 @@ public readonly partial record struct RamAllocationUnits( | |||||||
| 	/// </summary> | 	/// </summary> | ||||||
| 	/// <exception cref="ArgumentOutOfRangeException">If the <paramref name="definition"/> is in the incorrect format, or the value cannot be converted via <see cref="FromMegabytes"/>.</exception> | 	/// <exception cref="ArgumentOutOfRangeException">If the <paramref name="definition"/> is in the incorrect format, or the value cannot be converted via <see cref="FromMegabytes"/>.</exception> | ||||||
| 	public static RamAllocationUnits FromString(string definition) { | 	public static RamAllocationUnits FromString(string definition) { | ||||||
| 		return FromString(definition.AsSpan()); | 		return FromString((ReadOnlySpan<char>) definition); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| namespace Phantom.Common.Data.Replies; | namespace Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
| [MemoryPackable(GenerateType.VersionTolerant)] | [MemoryPackable] | ||||||
| public sealed partial record InstanceActionResult<T>( | public sealed partial record InstanceActionResult<T>( | ||||||
| 	[property: MemoryPackOrder(0)] InstanceActionGeneralResult GeneralResult, | 	[property: MemoryPackOrder(0)] InstanceActionGeneralResult GeneralResult, | ||||||
| 	[property: MemoryPackOrder(1)] T? ConcreteResult | 	[property: MemoryPackOrder(1)] T? ConcreteResult | ||||||
|   | |||||||
| @@ -1,9 +1,25 @@ | |||||||
| namespace Phantom.Common.Data.Replies; | namespace Phantom.Common.Data.Replies; | ||||||
|  |  | ||||||
| public enum LaunchInstanceResult : byte { | public enum LaunchInstanceResult : byte { | ||||||
| 	LaunchInitiated          = 1, | 	UnknownError, | ||||||
| 	InstanceAlreadyLaunching = 2, | 	LaunchInitiated, | ||||||
| 	InstanceAlreadyRunning   = 3, | 	InstanceAlreadyLaunching, | ||||||
| 	InstanceLimitExceeded    = 4, | 	InstanceAlreadyRunning, | ||||||
| 	MemoryLimitExceeded      = 5 | 	InstanceIsStopping, | ||||||
|  | 	InstanceLimitExceeded, | ||||||
|  | 	MemoryLimitExceeded | ||||||
|  | } | ||||||
|  |  | ||||||
|  | public static class LaunchInstanceResultExtensions { | ||||||
|  | 	public static string ToSentence(this LaunchInstanceResult reason) { | ||||||
|  | 		return reason switch { | ||||||
|  | 			LaunchInstanceResult.LaunchInitiated          => "Launch initiated.", | ||||||
|  | 			LaunchInstanceResult.InstanceAlreadyLaunching => "Instance is already launching.", | ||||||
|  | 			LaunchInstanceResult.InstanceAlreadyRunning   => "Instance is already running.", | ||||||
|  | 			LaunchInstanceResult.InstanceIsStopping       => "Instance is stopping.", | ||||||
|  | 			LaunchInstanceResult.InstanceLimitExceeded    => "Agent does not have any more available instances.", | ||||||
|  | 			LaunchInstanceResult.MemoryLimitExceeded      => "Agent does not have enough available memory.", | ||||||
|  | 			_                                             => "Unknown error." | ||||||
|  | 		}; | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user