Ooui-tws-port/Ooui/UI.cs

789 lines
28 KiB
C#

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<string, PublishedPath> publishedPaths =
new Dictionary<string, PublishedPath> ();
static readonly byte[] clientJsBytes;
static readonly string clientJsEtag;
public static byte[] ClientJsBytes => clientJsBytes;
public static string ClientJsEtag => clientJsEtag;
public static string HeadHtml { get; set; } = @"<link rel=""stylesheet"" href=""https://ajax.aspnetcdn.com/ajax/bootstrap/3.3.7/css/bootstrap.min.css"" />";
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<UIContext, Element> elementCtor, bool disposeElementWhenDone = true)
{
Publish (path, new ElementHandler (elementCtor, disposeElementWhenDone));
}
public static void Publish (string path, Func<Element> 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<object> ctor)
{
Publish (path, new JsonHandler (_ => ctor ()));
}
public static void PublishJson (string path, Func<UIContext, object> 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<HttpListenerContext, CancellationToken> 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<string, string> ();
lock (publishedPaths) found = publishedPaths.TryGetValue (path, out pp);
if (!found) {
// Try regex
List<PublishedPath> 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<UIContext, Element> ctor;
public bool DisposeElementWhenDone { get; }
public ElementHandler (Func<UIContext, Element> 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 ("&", "&amp;").Replace ("<", "&lt;");
}
public static void RenderTemplate (TextWriter writer, string webSocketPath, string title, string initialHtml)
{
writer.Write (@"<!DOCTYPE html>
<html>
<head>
<title>");
writer.Write (EscapeHtml (title));
writer.Write (@"</title>
<meta name=""viewport"" content=""width=device-width, initial-scale=1"" />
");
writer.WriteLine (HeadHtml);
writer.WriteLine (@" <style>");
writer.WriteLine (rules.ToString ());
writer.WriteLine (@" </style>
</head>
<body>");
writer.WriteLine (BodyHeaderHtml);
writer.WriteLine (@"<div id=""ooui-body"" class=""container-fluid"" style=""padding:0;margin:0"">");
writer.WriteLine (initialHtml);
writer.Write (@"</div>
<script src=""/ooui.js""></script>
<script>ooui(""");
writer.Write (webSocketPath);
writer.WriteLine (@""");</script>");
writer.WriteLine (BodyFooterHtml);
writer.WriteLine (@"</body>
</html>");
}
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<UIContext, object> ctor;
public JsonHandler (Func<UIContext, object> 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<HttpListenerContext, CancellationToken> responder;
public CustomHandler (Action<HttpListenerContext, CancellationToken> 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<string, string> ();
lock (publishedPaths) found = publishedPaths.TryGetValue (path, out pp);
if (!found) {
List<PublishedPath> 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<string, WebAssemblySession> globalElementSessions = new Dictionary<string, WebAssemblySession> ();
[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<string, Style> styles =
new Dictionary<string, Style> ();
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<string, string> variables = new Dictionary<string, string> ();
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<string, string> 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));
}
}
}