- Implement better start/stop for the webinterface

- Save session tokens between startups
2024-08-28 14:09:09 +02:00
parent 579d405621
commit 2ababa82b0
8 changed files with 400 additions and 122 deletions
@@ -0,0 +1,135 @@
+using WatsonWebserver.Core;
+using WatsonWebserver.Lite;
+
+namespace ChatTwo.Http;
+
+public class HostContext
+{
+    private readonly Plugin Plugin;
+
+    public bool IsActive;
+    public bool IsStopping;
+
+    internal WebserverLite Host;
+    internal Processing Processing;
+    internal RouteController RouteController;
+
+    internal readonly List<SSEConnection> EventConnections = [];
+
+    internal readonly CancellationTokenSource TokenSource = new();
+    internal readonly string StaticDir = Path.Combine(Plugin.Interface.AssemblyLocation.DirectoryName!, "Http");
+
+    public HostContext(Plugin plugin)
+    {
+        Plugin = plugin;
+    }
+
+    public bool Start()
+    {
+        try
+        {
+            Host = new WebserverLite(new WebserverSettings("*", Plugin.Config.WebinterfacePort), DefaultRoute);
+
+            Processing = new Processing(Plugin);
+            RouteController = new RouteController(Plugin, this);
+
+            Host.Routes.PreAuthentication.Content.BaseDirectory = StaticDir;
+            Host.Routes.AuthenticateRequest = CheckAuthenticationCookie;
+            Host.Events.ExceptionEncountered += ExceptionEncountered;
+
+            // Settings
+            Host.Settings.Debug.Requests = true;
+            Host.Settings.Debug.Routing = true;
+            Host.Settings.Debug.Responses = true;
+            Host.Settings.Debug.AccessControl = true;
+            Host.Events.Logger = logMessage => Plugin.Log.Information(logMessage);
+
+            IsActive = true;
+            return true;
+        }
+        catch (Exception ex)
+        {
+            IsActive = false;
+            Plugin.Log.Error(ex, "Initialization of the webserver failed.");
+            return false;
+        }
+    }
+
+    public void Run()
+    {
+        try
+        {
+            Host.Start(TokenSource.Token);
+        }
+        catch (Exception ex)
+        {
+            Plugin.Log.Error(ex, "Webserver failed to boot up.");
+        }
+    }
+
+    public async ValueTask<bool> Stop()
+    {
+        // Is already stopped
+        if (!IsActive)
+            return true;
+
+        try
+        {
+            IsActive = false;
+            IsStopping = true;
+            await Task.Delay(10000);
+            Host.Stop();
+
+            // Save our session tokens
+            Plugin.SaveConfig();
+
+            // We get a copy, so that the original can be cleaned up successfully
+            foreach (var eventServer in EventConnections.ToArray())
+                await eventServer.DisposeAsync();
+
+            EventConnections.Clear();
+            Host.Dispose();
+            RouteController.Dispose();
+            IsStopping = false;
+
+            return true;
+        }
+        catch (Exception ex)
+        {
+            Plugin.Log.Error(ex, "Webserver failed to stop and dispose all resources.");
+            return false;
+        }
+    }
+
+    public async ValueTask DisposeAsync()
+    {
+        await Stop();
+    }
+
+    #region GeneralHandlers
+    private static void ExceptionEncountered(object? _, ExceptionEventArgs args)
+    {
+        Plugin.Log.Error(args.Exception, "Webserver threw an exception.");
+    }
+
+    private async Task DefaultRoute(HttpContextBase ctx)
+    {
+        await ctx.Response.Send("Nothing to see here.");
+    }
+    #endregion
+
+    private async Task CheckAuthenticationCookie(HttpContextBase ctx)
+    {
+        if (Plugin.Config.SessionTokens.IsEmpty)
+        {
+            await RouteController.Redirect(ctx, "/", ("message", "Invalid session token."));
+            return;
+        }
+
+        var cookies = WebserverUtil.GetCookieData(ctx.Request.Headers.Get("Cookie") ?? "");
+        if (!cookies.TryGetValue("ChatTwo-token", out var token) || !Plugin.Config.SessionTokens.ContainsKey(token))
+            await RouteController.Redirect(ctx, "/", ("message", "Invalid session token."));
+
+        // Do nothing to let auth pass
+    }
+}