sharp-chat/SharpChat/C2SPacketHandlers/AuthC2SPacketHandler.cs

using SharpChat.Config;
using SharpChat.Misuzu;
using SharpChat.S2CPackets;

namespace SharpChat.C2SPacketHandlers {
    public class AuthC2SPacketHandler(
        MisuzuClient msz,
        ChatChannel defaultChannel,
        CachedValue<int> maxMsgLength,
        CachedValue<int> maxConns
    ) : C2SPacketHandler {
        private readonly MisuzuClient Misuzu = msz ?? throw new ArgumentNullException(nameof(msz));
        private readonly ChatChannel DefaultChannel = defaultChannel ?? throw new ArgumentNullException(nameof(defaultChannel));
        private readonly CachedValue<int> MaxMessageLength = maxMsgLength ?? throw new ArgumentNullException(nameof(maxMsgLength));
        private readonly CachedValue<int> MaxConnections = maxConns ?? throw new ArgumentNullException(nameof(maxConns));

        public bool IsMatch(ChatPacketHandlerContext ctx) {
            return ctx.CheckPacketId("1");
        }

        public void Handle(ChatPacketHandlerContext ctx) {
            string[] args = ctx.SplitText(3);

            string? authMethod = args.ElementAtOrDefault(1);
            if(string.IsNullOrWhiteSpace(authMethod)) {
                ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.AuthInvalid));
                ctx.Connection.Dispose();
                return;
            }

            string? authToken = args.ElementAtOrDefault(2);
            if(string.IsNullOrWhiteSpace(authToken)) {
                ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.AuthInvalid));
                ctx.Connection.Dispose();
                return;
            }

            if(authMethod.All(c => c is >= '0' and <= '9') && authToken.Contains(':')) {
                string[] tokenParts = authToken.Split(':', 2);
                authMethod = tokenParts[0];
                authToken = tokenParts[1];
            }

            Task.Run(async () => {
                MisuzuAuthInfo? fai;
                string ipAddr = ctx.Connection.RemoteAddress.ToString();

                try {
                    fai = await Misuzu.AuthVerifyAsync(authMethod, authToken, ipAddr);
                } catch(Exception ex) {
                    Logger.Write($"<{ctx.Connection.Id}> Failed to authenticate: {ex}");
                    ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.AuthInvalid));
                    ctx.Connection.Dispose();
#if DEBUG
                    throw;
#else
                    return;
#endif
                }

                if(fai?.Success != true) {
                    Logger.Debug($"<{ctx.Connection.Id}> Auth fail: {fai?.Reason ?? "unknown"}");
                    ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.AuthInvalid));
                    ctx.Connection.Dispose();
                    return;
                }

                MisuzuBanInfo? fbi;
                try {
                    fbi = await Misuzu.CheckBanAsync(fai.UserId.ToString(), ipAddr);
                } catch(Exception ex) {
                    Logger.Write($"<{ctx.Connection.Id}> Failed auth ban check: {ex}");
                    ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.AuthInvalid));
                    ctx.Connection.Dispose();
#if DEBUG
                    throw;
#else
                    return;
#endif
                }

                if(fbi?.IsBanned == true && !fbi.HasExpired) {
                    Logger.Write($"<{ctx.Connection.Id}> User is banned.");
                    ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.Banned, fbi));
                    ctx.Connection.Dispose();
                    return;
                }

                await ctx.Chat.ContextAccess.WaitAsync();
                try {
                    ChatUser? user = ctx.Chat.Users.FirstOrDefault(u => u.UserId == fai.UserId);

                    if(user == null)
                        user = new ChatUser(
                            fai.UserId,
                            fai.UserName ?? $"({fai.UserId})",
                            fai.Colour,
                            fai.Rank,
                            fai.Permissions
                        );
                    else
                        ctx.Chat.UpdateUser(
                            user,
                            userName: fai.UserName ?? $"({fai.UserId})",
                            colour: fai.Colour,
                            rank: fai.Rank,
                            perms: fai.Permissions
                        );

                    // Enforce a maximum amount of connections per user
                    if(ctx.Chat.Connections.Count(conn => conn.User == user) >= MaxConnections) {
                        ctx.Connection.Send(new AuthFailS2CPacket(AuthFailReason.MaxSessions));
                        ctx.Connection.Dispose();
                        return;
                    }

                    ctx.Connection.BumpPing();
                    ctx.Connection.User = user;
                    ctx.Connection.Send(new CommandResponseS2CPacket(0, LCR.WELCOME, false, $"Welcome to Flashii Chat, {user.UserName}!"));

                    if(File.Exists("welcome.txt")) {
                        IEnumerable<string> lines = File.ReadAllLines("welcome.txt").Where(x => !string.IsNullOrWhiteSpace(x));
                        string? line = lines.ElementAtOrDefault(RNG.Next(lines.Count()));

                        if(!string.IsNullOrWhiteSpace(line))
                            ctx.Connection.Send(new CommandResponseS2CPacket(0, LCR.WELCOME, false, line));
                    }

                    ctx.Chat.HandleJoin(user, DefaultChannel, ctx.Connection, MaxMessageLength);
                } finally {
                    ctx.Chat.ContextAccess.Release();
                }
            }).Wait();
        }
    }
}