Cache info about users, cleanup for module

This commit is contained in:
2022-01-03 23:25:20 +03:00
parent 64a4a1fabb
commit cf34fcd89a
2 changed files with 26 additions and 5 deletions
+21 -5
View File
@@ -8,9 +8,11 @@ using BotFramework.Attributes;
using BotFramework.Enums;
using BotFramework.Utils;
using Castle.Core.Internal;
using Kruzya.TelegramBot.Core.Cache;
using Kruzya.TelegramBot.Core.Data;
using Kruzya.TelegramBot.Core.Extensions;
using Telegram.Bot;
using Telegram.Bot.Types;
using Telegram.Bot.Types.Enums;
namespace UserGroupTag
@@ -18,16 +20,17 @@ namespace UserGroupTag
public class Handler : BotEventHandler
{
private readonly CoreContext _db;
public Handler(CoreContext db)
private readonly ICacheStorage<long, User> _userCache;
public Handler(CoreContext db, ICacheStorage<long, User> userCache)
{
_db = db;
_userCache = userCache;
}
[ParametrizedCommand(InChat.Public, "add_group", CommandParseMode.Both)]
public async Task HandleAddGroup(string groupName)
{
Console.WriteLine("add_group");
// var groupName = group.Text;
var repliedMessage = RawUpdate.Message!.ReplyToMessage;
@@ -56,6 +59,7 @@ namespace UserGroupTag
userGroupOption.SetValue(tagDictionary);
_db.MarkAsModified(userGroupOption);
_userCache.SetValue(repliedMessage.From.Id, repliedMessage.From, Int32.MaxValue);
await Bot.SendTextMessageAsync(chatId, $"{from.ToHtml()} добавлен в группу {groupName}",
ParseMode.Html);
}
@@ -79,13 +83,25 @@ namespace UserGroupTag
var msg = new HtmlString()
.Text("Пользователь ")
.UserMention(From)
.Text(" охуел и хочет доебаться к: ");
.Text($" призывает группу пользователей {groupName}: ");
foreach (var userId in userList)
{
msg.User(userId, userId.ToString()).Text(", ");
msg.UserMention(await GetUser(userId, Chat)).Text(", ");
}
await Bot.SendTextMessageAsync(Chat.Id, msg.ToString().TrimEnd(',', ' '), ParseMode.Html);
}
protected async Task<User> GetUser(long userId, Chat chat)
{
User user;
if (!_userCache.TryGetValue(userId, out user))
{
user = (await Bot.GetChatMemberAsync(Chat, userId)).User;
_userCache.SetValue(userId, user, Int32.MaxValue);
}
return user;
}
}
}