HellionChat/ChatTwo/Http/HostContext.cs

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
    }
}