[core] move core handlers into subfolder

2026-07-31 00:29:24 +03:00 · 2023-07-20 21:22:35 +03:00
parent 8de8eac959
commit 467ccd4fde
10 changed files with 17 additions and 17 deletions
@@ -0,0 +1,76 @@
+using System;
+using System.Threading.Tasks;
+using BotFramework;
+using BotFramework.Attributes;
+using Kruzya.TelegramBot.Core.Data;
+using Microsoft.EntityFrameworkCore;
+using Telegram.Bot.Types;
+
+namespace Kruzya.TelegramBot.Core.Handler
+{
+    public class GeneralHandler : BotEventHandler
+    {
+        [AttributeUsage(AttributeTargets.Method)]
+        public sealed class HandleEverythingAttribute : HandlerAttribute
+        {
+            protected override bool CanHandle(HandlerParams param)
+            {
+                return true;
+            }
+        }
+
+        protected readonly CoreContext databaseContext;
+
+        public GeneralHandler(CoreContext dbCtx)
+        {
+            databaseContext = dbCtx;
+        }
+
+        [HandleEverything]
+        [Priority(short.MaxValue - 10)]
+        public async Task<bool> Listener()
+        {
+            foreach (var message in new Message[] { RawUpdate.Message, RawUpdate.EditedMessage, RawUpdate.ChannelPost, RawUpdate.EditedChannelPost })
+            {
+                if (message == null)
+                {
+                    continue;
+                }
+
+                await VerifyChatPair(message);
+            }
+
+            return true;
+        }
+
+        public async Task VerifyChatPair(Message message)
+        {
+            await VerifyChatPair(message.Chat, message.From);
+        }
+
+        public async Task VerifyChatPair(Chat chat, User user)
+        {
+            await VerifyChatPair(chat.Id, user.Id);
+        }
+
+        public async Task VerifyChatPair(long chat, long user)
+        {
+            var hasEntry = await databaseContext.Users.AnyAsync(e => e.ChatId == chat && e.UserId == user);
+            if (hasEntry)
+            {
+                return;
+            }
+
+            await MakeChatPair(chat, user);
+        }
+
+        public async Task MakeChatPair(long chat, long user)
+        {
+            await databaseContext.Users.AddAsync(new BotUser()
+            {
+                ChatId = chat,
+                UserId = user
+            });
+        }
+    }
+}