IW4M-Admin/Integrations/Source/SourceRConConnection.cs

using System.Linq;
using System.Net.Sockets;
using System.Threading.Tasks;
using Integrations.Source.Interfaces;
using Microsoft.Extensions.Logging;
using RconSharp;
using Serilog.Context;
using SharedLibraryCore;
using SharedLibraryCore.Exceptions;
using SharedLibraryCore.Interfaces;
using SharedLibraryCore.RCon;
using ILogger = Microsoft.Extensions.Logging.ILogger;

namespace Integrations.Source
{
    public class SourceRConConnection : IRConConnection
    {
        private readonly ILogger _logger;
        private readonly string _password;
        private readonly string _hostname;
        private readonly int _port;
        private readonly IRConClientFactory _rconClientFactory;

        private RconClient _rconClient;

        public SourceRConConnection(ILogger<SourceRConConnection> logger, IRConClientFactory rconClientFactory,
            string hostname, int port, string password)
        {
            _rconClient = rconClientFactory.CreateClient(hostname, port);
            _rconClientFactory = rconClientFactory;
            _password = password;
            _hostname = hostname;
            _port = port;
            _logger = logger;
        }

        public async Task<string[]> SendQueryAsync(StaticHelpers.QueryType type, string parameters = "")
        {
            await _rconClient.ConnectAsync();

            bool authenticated;

            try
            {
                authenticated = await _rconClient.AuthenticateAsync(_password);
            }
            catch (SocketException ex)
            {
                // occurs when the server comes back from hibernation
                // this is probably a bug in the library
                if (ex.ErrorCode == 10053 || ex.ErrorCode == 10054)
                {
                    using (LogContext.PushProperty("Server", $"{_hostname}:{_port}"))
                    {
                        _logger.LogWarning(ex,
                            "Server appears to resumed from hibernation, so we are using a new socket");
                    }

                    _rconClient = _rconClientFactory.CreateClient(_hostname, _port);
                }

                using (LogContext.PushProperty("Server", $"{_hostname}:{_port}"))
                {
                    _logger.LogError("Could not login to server");
                }

                throw new NetworkException("Could not authenticate with server");
            }

            if (!authenticated)
            {
                using (LogContext.PushProperty("Server", $"{_hostname}:{_port}"))
                {
                    _logger.LogError("Could not login to server");
                }

                throw new ServerException("Could not authenticate to server with provided password");
            }

            if (type == StaticHelpers.QueryType.COMMAND_STATUS)
            {
                parameters = "status";
            }

            using (LogContext.PushProperty("Server", $"{_hostname}:{_port}"))
            {
                _logger.LogDebug("Sending query {Type} with parameters {Parameters}", type, parameters);
            }

            var response = await _rconClient.ExecuteCommandAsync(parameters.StripColors(), true);

            using (LogContext.PushProperty("Server", $"{_rconClient.Host}:{_rconClient.Port}"))
            {
                _logger.LogDebug("Received RCon response {Response}", response);
            }

            var split = response.TrimEnd('\n').Split('\n');
            return split.Take(split.Length - 1).ToArray();
        }

        public void SetConfiguration(IRConParser config)
        {
        }
    }
}