63 lines
2.7 KiB
C#
63 lines
2.7 KiB
C#
using System;
|
|
using Ooui.AspNetCore;
|
|
|
|
namespace Microsoft.AspNetCore.Builder
|
|
{
|
|
public static class OouiMiddlewareExtensions
|
|
{
|
|
public static void UseOoui (this IApplicationBuilder app, string jsPath = "/ooui.js", string webSocketPath = "/ooui.ws", TimeSpan? sessionTimeout = null)
|
|
{
|
|
if (string.IsNullOrWhiteSpace (webSocketPath))
|
|
throw new ArgumentException ("A path to be used for Ooui web sockets must be specified", nameof (webSocketPath));
|
|
|
|
if (string.IsNullOrWhiteSpace (jsPath))
|
|
throw new ArgumentException ("A path to be used for Ooui JavaScript must be specified", nameof (jsPath));
|
|
|
|
WebSocketHandler.WebSocketPath = webSocketPath;
|
|
|
|
if (sessionTimeout.HasValue) {
|
|
WebSocketHandler.SessionTimeout = sessionTimeout.Value;
|
|
}
|
|
|
|
var webSocketOptions = new WebSocketOptions () {
|
|
KeepAliveInterval = WebSocketHandler.SessionTimeout,
|
|
ReceiveBufferSize = 4 * 1024
|
|
};
|
|
app.UseWebSockets (webSocketOptions);
|
|
|
|
app.Use (async (context, next) =>
|
|
{
|
|
if (context.Request.Path == jsPath) {
|
|
var response = context.Response;
|
|
var clientJsBytes = Ooui.UI.ClientJsBytes;
|
|
var clientJsEtag = Ooui.UI.ClientJsEtag;
|
|
if (context.Request.Headers.TryGetValue ("If-None-Match", out var inms) && inms.Count > 0 && inms[0] == clientJsEtag) {
|
|
response.StatusCode = 304;
|
|
}
|
|
else {
|
|
response.StatusCode = 200;
|
|
response.ContentLength = clientJsBytes.Length;
|
|
response.ContentType = "application/javascript; charset=utf-8";
|
|
response.Headers.Add ("Cache-Control", "public, max-age=60");
|
|
response.Headers.Add ("Etag", clientJsEtag);
|
|
using (var s = response.Body) {
|
|
await s.WriteAsync (clientJsBytes, 0, clientJsBytes.Length).ConfigureAwait (false);
|
|
}
|
|
}
|
|
}
|
|
else if (context.Request.Path == WebSocketHandler.WebSocketPath) {
|
|
if (context.WebSockets.IsWebSocketRequest) {
|
|
await WebSocketHandler.HandleWebSocketRequestAsync (context).ConfigureAwait (false);
|
|
}
|
|
else {
|
|
context.Response.StatusCode = 400;
|
|
}
|
|
}
|
|
else {
|
|
await next ().ConfigureAwait (false);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}
|