using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using System.Threading; using System.Threading.Tasks; using System.Net; using System.Runtime.InteropServices; using System.Text.RegularExpressions; namespace Ooui { public static class UI { public const int MaxFps = 30; static readonly ManualResetEvent started = new ManualResetEvent (false); static CancellationTokenSource serverCts; static readonly Dictionary publishedPaths = new Dictionary (); static readonly byte[] clientJsBytes; static readonly string clientJsEtag; public static byte[] ClientJsBytes => clientJsBytes; public static string ClientJsEtag => clientJsEtag; public static string HeadHtml { get; set; } = @""; public static string BodyHeaderHtml { get; set; } = @""; public static string BodyFooterHtml { get; set; } = @""; static string host = "*"; public static string Host { get => host; set { if (!string.IsNullOrWhiteSpace (value) && host != value) { host = value; Restart (); } } } static int port = 8080; public static int Port { get => port; set { if (port != value) { port = value; Restart (); } } } static bool serverEnabled = true; public static bool ServerEnabled { get => serverEnabled; set { if (serverEnabled != value) { serverEnabled = value; if (serverEnabled) Restart (); else Stop (); } } } [Preserve] static void DisableServer () { ServerEnabled = false; } static UI () { var asm = typeof(UI).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")) { if (s == null) throw new Exception ("Missing Client.js"); using (var r = new StreamReader (s)) { clientJsBytes = Encoding.UTF8.GetBytes (r.ReadToEnd ()); } } clientJsEtag = "\"" + Utilities.GetShaHash (clientJsBytes) + "\""; } class PublishedPath { public string Path { get; } public Regex RegexPath { get; } public RequestHandler Handler { get; } public PublishedPath (string path, RequestHandler handler) { Path = path; RegexPath = new Regex ("^" + path + "$"); Handler = handler; } } static void Publish (string path, RequestHandler handler) { //Console.WriteLine ($"PUBLISH {path} {handler}"); lock (publishedPaths) publishedPaths[path] = new PublishedPath (path, handler); Start (); } public static void Publish (string path, Func elementCtor, bool disposeElementWhenDone = true) { Publish (path, new ElementHandler (elementCtor, disposeElementWhenDone)); } public static void Publish (string path, Func elementCtor, bool disposeElementWhenDone = true) { Publish (path, new ElementHandler (_ => elementCtor (), disposeElementWhenDone)); } public static void Publish (string path, Element element, bool disposeElementWhenDone = true) { Publish (path, () => element, disposeElementWhenDone); } public static void PublishFile (string filePath) { var path = "/" + System.IO.Path.GetFileName (filePath); PublishFile (path, filePath); } public static void PublishFile (string path, string filePath, string contentType = null) { var data = System.IO.File.ReadAllBytes (filePath); if (contentType == null) { contentType = GuessContentType (path, filePath); } var etag = "\"" + Utilities.GetShaHash (data) + "\""; Publish (path, new DataHandler (data, etag, contentType)); } public static void PublishFile (string path, byte[] data, string contentType) { var etag = "\"" + Utilities.GetShaHash (data) + "\""; Publish (path, new DataHandler (data, etag, contentType)); } public static void PublishFile (string path, byte[] data, string etag, string contentType) { Publish (path, new DataHandler (data, etag, contentType)); } public static bool TryGetFileContentAtPath (string path, out FileContent file) { PublishedPath pp; lock (publishedPaths) { if (!publishedPaths.TryGetValue (path, out pp)) { file = null; return false; } } if (pp.Handler is DataHandler dh) { file = new FileContent { Etag = dh.Etag, Content = dh.Data, ContentType = dh.ContentType, }; return true; } file = null; return false; } public class FileContent { public string ContentType { get; set; } public string Etag { get; set; } public byte[] Content { get; set; } } public static void PublishJson (string path, Func ctor) { Publish (path, new JsonHandler (_ => ctor ())); } public static void PublishJson (string path, Func ctor) { Publish (path, new JsonHandler (ctor)); } public static void PublishJson (string path, object value) { var data = JsonHandler.GetData (value); var etag = "\"" + Utilities.GetShaHash (data) + "\""; Publish (path, new DataHandler (data, etag, JsonHandler.ContentType)); } public static void PublishCustomResponse (string path, Action responder) { Publish (path, new CustomHandler (responder)); } static string GuessContentType (string path, string filePath) { return null; } public static void Present (string path, object presenter = null) { WaitUntilStarted (); var url = GetUrl (path); Console.WriteLine ($"PRESENT {url}"); Platform.OpenBrowser (url, presenter); } public static string GetUrl (string path) { var localhost = host == "*" ? "localhost" : host; var url = $"http://{localhost}:{port}{path}"; return url; } public static string GetWebSocketUrl (string path) { var localhost = host == "*" ? "localhost" : host; var url = $"ws://{localhost}:{port}{path}"; return url; } public static void WaitUntilStarted () => started.WaitOne (); static void Start () { if (!serverEnabled) return; if (serverCts != null) return; serverCts = new CancellationTokenSource (); var token = serverCts.Token; var listenerPrefix = $"http://{host}:{port}/"; Task.Run (() => RunAsync (listenerPrefix, token), token); } static void Stop () { var scts = serverCts; if (scts == null) return; serverCts = null; started.Reset (); Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine ($"Stopping..."); Console.ResetColor (); scts.Cancel (); } static void Restart () { if (serverCts == null) return; Stop (); Start (); } static async Task RunAsync (string listenerPrefix, CancellationToken token) { HttpListener listener = null; var wait = 5; started.Reset (); while (!started.WaitOne(0) && !token.IsCancellationRequested) { try { listener = new HttpListener (); listener.Prefixes.Add (listenerPrefix); listener.Start (); started.Set (); } catch (System.Net.Sockets.SocketException ex) { Console.WriteLine ($"{listenerPrefix} error: {ex.Message}. Trying again in {wait} seconds..."); await Task.Delay (wait * 1000).ConfigureAwait (false); } catch (System.Net.HttpListenerException ex) { Console.WriteLine ($"{listenerPrefix} error: {ex.Message}. Trying again in {wait} seconds..."); await Task.Delay (wait * 1000).ConfigureAwait (false); } catch (Exception ex) { Error ("Error listening", ex); return; } } Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine ($"Listening at {listenerPrefix}..."); Console.ResetColor (); while (!token.IsCancellationRequested) { var listenerContext = await listener.GetContextAsync ().ConfigureAwait (false); if (listenerContext.Request.IsWebSocketRequest) { ProcessWebSocketRequest (listenerContext, token); } else { ProcessRequest (listenerContext, token); } } } static 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; if (path == "/ooui.js") { var inm = listenerContext.Request.Headers.Get ("If-None-Match"); if (string.IsNullOrEmpty (inm) || inm != clientJsEtag) { response.StatusCode = 200; response.ContentLength64 = clientJsBytes.LongLength; response.ContentType = "application/javascript"; response.ContentEncoding = Encoding.UTF8; response.AddHeader ("Cache-Control", "public, max-age=60"); response.AddHeader ("Etag", clientJsEtag); using (var s = response.OutputStream) { s.Write (clientJsBytes, 0, clientJsBytes.Length); } response.Close (); } else { response.StatusCode = 304; response.Close (); } } else { var found = false; PublishedPath pp; var variables = new Dictionary (); lock (publishedPaths) found = publishedPaths.TryGetValue (path, out pp); if (!found) { // Try regex List pps; lock (publishedPaths) pps = publishedPaths.Values.ToList(); foreach (var p in pps) { var m = p.RegexPath.Match (path); if (m.Success) { pp = p; found = true; for (var ig = 1; ig < m.Groups.Count; ig++) { variables[p.RegexPath.GroupNameFromNumber(ig)] = m.Groups[ig].Value; } break; } } } if (found) { try { var context = UIContext.ForListenerContext (listenerContext, variables); pp.Handler.Respond (listenerContext, context, token); } catch (Exception ex) { Error ("Handler failed to respond", ex); try { response.StatusCode = 500; response.Close (); } catch { // Ignore ending the response errors } } } else { response.StatusCode = 404; response.Close (); } } } abstract class RequestHandler { public abstract void Respond (HttpListenerContext listenerContext, UIContext context, CancellationToken token); } class ElementHandler : RequestHandler { readonly Func ctor; public bool DisposeElementWhenDone { get; } public ElementHandler (Func ctor, bool disposeElementWhenDone) { this.ctor = ctor; DisposeElementWhenDone = disposeElementWhenDone; } public Element GetElement (UIContext context) => ctor (context); public override void Respond (HttpListenerContext listenerContext, UIContext context, CancellationToken token) { var url = listenerContext.Request.Url; var path = url.LocalPath; var response = listenerContext.Response; response.StatusCode = 200; response.ContentType = "text/html"; response.ContentEncoding = Encoding.UTF8; var html = Encoding.UTF8.GetBytes (RenderTemplate (path)); response.ContentLength64 = html.LongLength; using (var s = response.OutputStream) { s.Write (html, 0, html.Length); } response.Close (); } } public static string RenderTemplate (string webSocketPath, string title = "", string initialHtml = "") { using (var w = new System.IO.StringWriter ()) { RenderTemplate (w, webSocketPath, title, initialHtml); return w.ToString (); } } static string EscapeHtml (string text) { return text.Replace ("&", "&").Replace ("<", "<"); } public static void RenderTemplate (TextWriter writer, string webSocketPath, string title, string initialHtml) { writer.Write (@" "); writer.Write (EscapeHtml (title)); writer.Write (@" "); writer.WriteLine (HeadHtml); writer.WriteLine (@" "); writer.WriteLine (BodyHeaderHtml); writer.WriteLine (@"
"); writer.WriteLine (initialHtml); writer.Write (@"
"); writer.WriteLine (BodyFooterHtml); writer.WriteLine (@" "); } class DataHandler : RequestHandler { readonly byte[] data; readonly string etag; readonly string contentType; public byte[] Data => data; public string Etag => etag; public string ContentType => contentType; public DataHandler (byte[] data, string etag, string contentType = null) { this.data = data; this.etag = etag; this.contentType = contentType; } public override void Respond (HttpListenerContext listenerContext, UIContext context, CancellationToken token) { var url = listenerContext.Request.Url; var path = url.LocalPath; var response = listenerContext.Response; var inm = listenerContext.Request.Headers.Get ("If-None-Match"); if (!string.IsNullOrEmpty (inm) && inm == etag) { response.StatusCode = 304; } else { response.StatusCode = 200; response.AddHeader ("Etag", etag); if (!string.IsNullOrEmpty (contentType)) response.ContentType = contentType; response.ContentLength64 = data.LongLength; using (var s = response.OutputStream) { s.Write (data, 0, data.Length); } } response.Close (); } } class JsonHandler : RequestHandler { public const string ContentType = "application/json; charset=utf-8"; readonly Func ctor; public JsonHandler (Func ctor) { this.ctor = ctor; } public static byte[] GetData (object obj) { var r = Ooui.JsonConvert.SerializeObject (obj); var e = new UTF8Encoding (false); return e.GetBytes (r); } public override void Respond (HttpListenerContext listenerContext, UIContext context, CancellationToken token) { var response = listenerContext.Response; var data = GetData (ctor (context)); response.StatusCode = 200; response.ContentType = ContentType; response.ContentLength64 = data.LongLength; using (var s = response.OutputStream) { s.Write (data, 0, data.Length); } response.Close (); } } class CustomHandler : RequestHandler { readonly Action responder; public CustomHandler (Action responder) { this.responder = responder; } public override void Respond (HttpListenerContext listenerContext, UIContext context, CancellationToken token) { responder (listenerContext, token); } } static async void ProcessWebSocketRequest (HttpListenerContext listenerContext, CancellationToken serverToken) { // // Find the element // var url = listenerContext.Request.Url; var path = url.LocalPath; PublishedPath pp; var found = false; var variables = new Dictionary (); lock (publishedPaths) found = publishedPaths.TryGetValue (path, out pp); if (!found) { List pps; lock (publishedPaths) pps = publishedPaths.Values.ToList (); foreach (var p in pps) { var m = p.RegexPath.Match (path); if (m.Success) { pp = p; found = true; for (var ig = 1; ig < m.Groups.Count; ig++) { variables[p.RegexPath.GroupNameFromNumber (ig)] = m.Groups[ig].Value; } break; } } } var elementHandler = pp?.Handler as ElementHandler; if (!found || elementHandler == null) { listenerContext.Response.StatusCode = 404; listenerContext.Response.Close (); return; } Element element = null; bool disposeElementWhenDone = true; try { var context = UIContext.ForListenerContext (listenerContext, variables); element = elementHandler.GetElement (context); disposeElementWhenDone = elementHandler.DisposeElementWhenDone; if (element == null) throw new Exception ("Handler returned a null element"); } catch (Exception ex) { listenerContext.Response.StatusCode = 500; listenerContext.Response.Close(); Error ("Failed to create element", ex); return; } // // Connect the web socket // System.Net.WebSockets.WebSocketContext webSocketContext = null; System.Net.WebSockets.WebSocket webSocket = null; try { webSocketContext = await listenerContext.AcceptWebSocketAsync (subProtocol: "ooui").ConfigureAwait (false); webSocket = webSocketContext.WebSocket; Console.WriteLine ("WEBSOCKET {0}", listenerContext.Request.Url.LocalPath); } catch (Exception ex) { listenerContext.Response.StatusCode = 500; listenerContext.Response.Close(); Error ("Failed to accept WebSocket", ex); return; } // // Set the element's dimensions // var query = (from part in listenerContext.Request.Url.Query.Split (new[] { '?', '&' }) where part.Length > 0 let kvs = part.Split ('=') where kvs.Length == 2 select kvs).ToDictionary (x => Uri.UnescapeDataString (x[0]), x => Uri.UnescapeDataString (x[1])); if (!query.TryGetValue ("w", out var wValue) || string.IsNullOrEmpty (wValue)) { wValue = "640"; } if (!query.TryGetValue ("h", out var hValue) || string.IsNullOrEmpty (hValue)) { hValue = "480"; } var icult = System.Globalization.CultureInfo.InvariantCulture; if (!double.TryParse (wValue, System.Globalization.NumberStyles.Any, icult, out var w)) w = 640; if (!double.TryParse (hValue, System.Globalization.NumberStyles.Any, icult, out var h)) h = 480; // // Create a new session and let it handle everything from here // try { var session = new WebSocketSession (webSocket, element, disposeElementWhenDone, w, h, Error, serverToken); await session.RunAsync ().ConfigureAwait (false); } catch (System.Net.WebSockets.WebSocketException ex) when (ex.WebSocketErrorCode == System.Net.WebSockets.WebSocketError.ConnectionClosedPrematurely) { // The remote party closed the WebSocket connection without completing the close handshake. } catch (Exception ex) { Error ("Web socket failed", ex); } finally { webSocket?.Dispose (); } } static void Error (string message, Exception ex) { Console.ForegroundColor = ConsoleColor.Red; Console.WriteLine ("{0}: {1}", message, ex); Console.ResetColor (); } static readonly Dictionary globalElementSessions = new Dictionary (); [Preserve] public static void StartWebAssemblySession (string sessionId, string elementPath, string initialSize) { Element element; PublishedPath pp; lock (publishedPaths) { publishedPaths.TryGetValue (elementPath, out pp); } var disposeElementWhenDone = true; if (pp.Handler is ElementHandler eh) { var context = UIContext.ForWebAssemblySession (sessionId, elementPath); element = eh.GetElement (context); disposeElementWhenDone = eh.DisposeElementWhenDone; } else { element = new Div (); } var ops = initialSize.Split (' '); var initialWidth = double.Parse (ops[0]); var initialHeight = double.Parse (ops[1]); var g = new WebAssemblySession (sessionId, element, disposeElementWhenDone, initialWidth, initialHeight, Error); lock (globalElementSessions) { globalElementSessions[sessionId] = g; } g.StartSession (); } [Preserve] public static void ReceiveWebAssemblySessionMessageJson (string sessionId, string json) { WebAssemblySession g; lock (globalElementSessions) { if (!globalElementSessions.TryGetValue (sessionId, out g)) return; } g.ReceiveMessageJson (json); } static readonly Dictionary styles = new Dictionary (); static readonly StyleSelectors rules = new StyleSelectors (); public static StyleSelectors Styles => rules; public class StyleSelectors { public Style this[string selector] { get { var key = selector ?? ""; lock (styles) { if (!styles.TryGetValue (key, out Style r)) { r = new Style (); styles.Add (key, r); } return r; } } set { var key = selector ?? ""; lock (styles) { if (value == null) { styles.Remove (key); } else { styles[key] = value; } } } } public void Clear () { lock (styles) { styles.Clear (); } } public override string ToString() { lock (styles) { var q = from s in styles let v = s.Value.ToString () where v.Length > 0 select s.Key + " {" + s.Value.ToString () + "}"; return String.Join ("\n", q); } } } } public class UIContext { readonly Dictionary variables = new Dictionary (); public Uri RequestUrl { get; } public string this[string key] => variables.TryGetValue (key, out var v) ? v : ""; public UIContext (Uri requestUrl) { RequestUrl = requestUrl ?? throw new ArgumentNullException (nameof (requestUrl)); } public static UIContext ForListenerContext (HttpListenerContext listenerContext, Dictionary variables) { var c = new UIContext (listenerContext.Request.Url); foreach (var kv in variables) { c.variables.Add (kv.Key, kv.Value); } return c; } public static UIContext ForWebAssemblySession (string sessionId, string elementPath) { return new UIContext (new Uri ("/" + elementPath, UriKind.Relative)); } } }