Added C2S and S2C markings to class names.

SharpChat/C2SPacketHandlers/AuthC2SPacketHandler.cs

@@ -0,0 +1,136 @@
+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();
+        }
+    }
+}