{element}

using System; using System.Collections.Generic; using System.IO; using System.Text; using System.Threading; using System.Threading.Tasks; using System.Net; using System.Net.WebSockets; namespace Ooui { public class Server { readonly Dictionary> publishedPaths = new Dictionary> (); readonly static byte[] clientJsBytes; static Server () { var asm = typeof(Server).Assembly; System.Console.WriteLine("ASM = {0}", asm); foreach (var n in asm.GetManifestResourceNames()) { System.Console.WriteLine(" {0}", n); } using (var s = asm.GetManifestResourceStream ("Ooui.Client.js")) { using (var r = new StreamReader (s)) { clientJsBytes = Encoding.UTF8.GetBytes (r.ReadToEnd ()); } } } public Task RunAsync (string listenerPrefix) { return RunAsync (listenerPrefix, CancellationToken.None); } public async Task RunAsync (string listenerPrefix, CancellationToken token) { var listener = new HttpListener (); listener.Prefixes.Add (listenerPrefix); listener.Start (); Console.WriteLine ($"Listening at {listenerPrefix}..."); while (!token.IsCancellationRequested) { var listenerContext = await listener.GetContextAsync ().ConfigureAwait (false); if (listenerContext.Request.IsWebSocketRequest) { ProcessWebSocketRequest (listenerContext, token); } else { ProcessRequest (listenerContext, token); } } } public void Publish (string path, Func elementCtor) { System.Console.WriteLine($"PUBLISH {path}"); publishedPaths[path] = elementCtor; } public void Publish (string path, Element element) { Publish (path, () => element); } void ProcessRequest (HttpListenerContext listenerContext, CancellationToken token) { var url = listenerContext.Request.Url; var path = url.LocalPath; Console.WriteLine ($"{listenerContext.Request.HttpMethod} {url.LocalPath}"); var response = listenerContext.Response; Func ctor; if (path == "/client.js") { response.ContentLength64 = clientJsBytes.LongLength; response.ContentType = "application/javascript"; response.ContentEncoding = Encoding.UTF8; response.AddHeader ("Cache-Control", "public, max-age=3600"); using (var s = response.OutputStream) { s.Write (clientJsBytes, 0, clientJsBytes.Length); } } else if (publishedPaths.TryGetValue (path, out ctor)) { var element = ctor (); RegisterElement (element); WriteElementHtml (element, response); } else { response.StatusCode = 404; response.Close (); } } void RegisterElement (Element element) { } void WriteElementHtml (Element element, HttpListenerResponse response) { response.StatusCode = 200; response.ContentType = "text/html"; response.ContentEncoding = Encoding.UTF8; var html = Encoding.UTF8.GetBytes ($@" {element} "); response.ContentLength64 = html.LongLength; using (var s = response.OutputStream) { s.Write (html, 0, html.Length); } response.Close (); } async void ProcessWebSocketRequest (HttpListenerContext listenerContext, CancellationToken token) { WebSocketContext webSocketContext = null; try { webSocketContext = await listenerContext.AcceptWebSocketAsync(subProtocol: "ooui-1.0").ConfigureAwait (false); Console.WriteLine ("Accepted WebSocket: {0}", webSocketContext); } catch (Exception e) { listenerContext.Response.StatusCode = 500; listenerContext.Response.Close(); Console.WriteLine ("Failed to accept WebSocket: {0}", e); return; } WebSocket webSocket = null; try { webSocket = webSocketContext.WebSocket; var receiveBuffer = new byte[1024]; while (webSocket.State == WebSocketState.Open && !token.IsCancellationRequested) { var receiveResult = await webSocket.ReceiveAsync(new ArraySegment(receiveBuffer), token); if (receiveResult.MessageType == WebSocketMessageType.Close) { await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token).ConfigureAwait (false); } else if (receiveResult.MessageType == WebSocketMessageType.Binary) { await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token).ConfigureAwait (false); } else { var size = receiveResult.Count; while (!receiveResult.EndOfMessage) { if (size >= receiveBuffer.Length) { await webSocket.CloseAsync (WebSocketCloseStatus.MessageTooBig, "Message too big", token).ConfigureAwait (false); return; } receiveResult = await webSocket.ReceiveAsync (new ArraySegment(receiveBuffer, size, receiveBuffer.Length - size), token).ConfigureAwait (false); size += receiveResult.Count; } var receivedString = Encoding.UTF8.GetString (receiveBuffer, 0, size); Console.WriteLine ("RECEIVED: {0}", receivedString); var outputBuffer = new ArraySegment (Encoding.UTF8.GetBytes ($"You said: {receivedString}")); await webSocket.SendAsync (outputBuffer, WebSocketMessageType.Text, true, token).ConfigureAwait (false); } } } catch (Exception e) { Console.WriteLine ("Exception: {0}", e); } finally { webSocket?.Dispose(); } } } }