Minecraft-Phantom-Panel/Agent/Phantom.Agent.Services/Instances/PortManager.cs

using Phantom.Common.Data;
using Phantom.Common.Data.Instance;

namespace Phantom.Agent.Services.Instances;

sealed class PortManager {
	private readonly AllowedPorts allowedServerPorts;
	private readonly AllowedPorts allowedRconPorts;
	private readonly HashSet<ushort> usedPorts = new ();

	public PortManager(AllowedPorts allowedServerPorts, AllowedPorts allowedRconPorts) {
		this.allowedServerPorts = allowedServerPorts;
		this.allowedRconPorts = allowedRconPorts;
	}

	public Result Reserve(InstanceConfiguration configuration) {
		lock (usedPorts) {
			if (usedPorts.Contains(configuration.ServerPort)) {
				return Result.ServerPortAlreadyInUse;
			}

			if (usedPorts.Contains(configuration.RconPort)) {
				return Result.RconPortAlreadyInUse;
			}

			usedPorts.Add(configuration.ServerPort);
			usedPorts.Add(configuration.RconPort);
		}

		return Result.Success;
	}
	
	public void Release(InstanceConfiguration configuration) {
		lock (usedPorts) {
			usedPorts.Remove(configuration.ServerPort);
			usedPorts.Remove(configuration.RconPort);
		}
	}

	public enum Result {
		Success,
		ServerPortNotAllowed,
		ServerPortAlreadyInUse,
		RconPortNotAllowed,
		RconPortAlreadyInUse,
	}
}