ArchiSteamFarm/ArchiSteamFarm/WCF.cs

using System;
using System.ServiceModel;
using System.ServiceModel.Channels;
using System.Threading.Tasks;

namespace ArchiSteamFarm {
	[ServiceContract]
	internal interface IWCF {
		[OperationContract]
		string HandleCommand(string input);
	}

	internal class WCF : IWCF {

		private const string URL = "http://localhost:1242/ASF"; // 1242 = 1024 + A(65) + S(83) + F(70)

		private ServiceHost ServiceHost;
		private Client Client;

		internal void StartServer() {
			if (ServiceHost != null) {
				return;
			}

			Logging.LogGenericNotice("WCF", "Starting WCF server...");
			ServiceHost = new ServiceHost(typeof(WCF));
			ServiceHost.AddServiceEndpoint(typeof(IWCF), new BasicHttpBinding(), URL);

			try {
				ServiceHost.Open();
			} catch (AddressAccessDeniedException) {
				Logging.LogGenericWarning("WCF", "WCF service could not be started because of AddressAccessDeniedException");
				Logging.LogGenericWarning("WCF", "If you want to use WCF service provided by ASF, consider starting ASF as administrator, or giving proper permissions");
				return;
			} catch (Exception e) {
				Logging.LogGenericException("WCF", e);
				return;
			}

			Logging.LogGenericNotice("WCF", "WCF server ready!");
		}

		internal void StopServer() {
			if (ServiceHost == null) {
				return;
			}

			ServiceHost.Close();
			ServiceHost = null;
		}

		internal string SendCommand(string input) {
			if (Client == null) {
				Client = new Client(new BasicHttpBinding(), new EndpointAddress(URL));
			}

			return Client.HandleCommand(input);
		}

		public string HandleCommand(string input) {
			if (string.IsNullOrEmpty(input)) {
				return null;
			}

			string[] args = input.Split(' ');
			if (args.Length < 2) {
				return "ERROR: Too few arguments, expected: <Command> <BotName> (ExtraArgs)";
			}

			// Bot name is args[1]
			string botName = args[1];

			Bot bot;
			if (!Bot.Bots.TryGetValue(botName, out bot)) {
				return "ERROR: Couldn't find any bot named " + botName;
			}

			string command = '!' + input;
			Logging.LogGenericInfo("WCF", "Received command: \"" + command + "\"");

			string output = bot.HandleMessage(command).Result; // TODO: This should be asynchronous

			Logging.LogGenericInfo("WCF", "Answered to command: \"" + input + "\" with: \"" + output + "\"");
			return output;
		}
	}

	internal class Client : ClientBase<IWCF>, IWCF {
		internal Client(Binding binding, EndpointAddress address) : base(binding, address) { }

		public string HandleCommand(string input) {
			try {
				return Channel.HandleCommand(input);
			} catch (Exception e) {
				Logging.LogGenericException("WCF", e);
				return null;
			}
		}
	}
}