Nibriboard/Nibriboard/NibriboardServer.cs

using System;
using System.Threading.Tasks;
using System.Threading;

using IotWeb.Server;
using IotWeb.Common.Http;

using Nibriboard.RippleSpace;
using Nibriboard.Client;
using System.Net.Sockets;
using System.Net;
using System.IO;

namespace Nibriboard
{
	/// <summary>
	/// The main Nibriboard server.
	/// This class manages not only the connected clients, but also holds the master reference to the <see cref="Nibriboard.RippleSpace.RippleSpaceManager"/>.
	/// </summary>
	public class NibriboardServer
	{
		private TcpListener commandServer;
		private HttpServer httpServer;

		private ClientSettings clientSettings;
		private RippleSpaceManager planeManager = new RippleSpaceManager() { SourceFilename = "./test.ripplespace.zip" };

		private readonly CancellationTokenSource clientManagerCanceller = new CancellationTokenSource();
		private NibriClientManager clientManager;

		public readonly int CommandPort = 31587;
		public readonly int Port = 31586;

		public NibriboardServer(int inPort = 31586)
		{
			Port = inPort;

			clientSettings = new ClientSettings() {
				SecureWebSocket = false,
				WebSocketHost = "192.168.0.56",
				WebSocketPort = Port,
				WebSocketPath = "/RipplespaceLink"
			};

			// HTTP Server setup
			httpServer = new HttpServer(Port);
			httpServer.AddHttpRequestHandler(
				"/",
				new HttpEmbeddedFileHandler("Nibriboard.ClientFiles")
				/*new HttpResourceHandler(
					Assembly.GetExecutingAssembly(),
					"ClientFiles",
					"index.html"
				)*/
			);
			httpServer.AddHttpRequestHandler(
				"/Settings.json",
				new HttpClientSettingsHandler(clientSettings)
			);

			// Websocket setup
			clientManager = new NibriClientManager(
				clientSettings,
				planeManager,
				clientManagerCanceller.Token
			);
			httpServer.AddWebSocketRequestHandler(
				clientSettings.WebSocketPath,

				clientManager
			);
		}

		public async Task Start()
		{
			httpServer.Start();
			Log.WriteLine("[NibriboardServer] Started on port {0}", Port);

			await planeManager.StartMaintenanceMonkey();
		}

		/// <summary>
		/// Starts the command listener.
		/// The command listener is a light tcp-based command console that allows control
		/// of the Nibriboard server, since C# doesn't currently have support for signal handling.
		/// It listeners on [::1] _only_, to avoid security issues.
		/// In the future, a simple secret might be required to use it to aid security further.
		/// </summary>
		public async Task StartCommandListener()
		{
			commandServer = new TcpListener(IPAddress.IPv6Loopback, CommandPort);
			commandServer.Start();
			Log.WriteLine("[CommandConsole] Listening on {0}.", new IPEndPoint(IPAddress.IPv6Loopback, CommandPort));
			while(true)
			{
				TcpClient nextClient = await commandServer.AcceptTcpClientAsync();

				StreamReader source = new StreamReader(nextClient.GetStream());
				StreamWriter destination = new StreamWriter(nextClient.GetStream()) { AutoFlush = true };

				string rawCommand = await source.ReadLineAsync();
				string[] commandParts = rawCommand.Split(" \t".ToCharArray());
				Log.WriteLine("[CommandConsole] Client executing {0}", rawCommand);

				try
				{
					switch(commandParts[0].Trim())
					{
						case "help":
							await destination.WriteLineAsync("NibriboardServer Command Console");
							await destination.WriteLineAsync("================================");
							await destination.WriteLineAsync("Available commands:");
							await destination.WriteLineAsync("    help     Show this message");
							await destination.WriteLineAsync("    save     Save the ripplespace to disk");
							break;
						case "save":
							await planeManager.Save();
							break;
					}
				}
				catch(Exception error)
				{
					await destination.WriteLineAsync(error.ToString());
				}
				nextClient.Close();
			}
		}
	}
}