mirror of
https://github.com/Bubuni-Team/telegram-bot.git
synced 2026-07-31 00:29:24 +03:00
[reputation] Add rating.
This commit is contained in:
@@ -0,0 +1,8 @@
|
|||||||
|
namespace Kruzya.TelegramBot.Core.Data;
|
||||||
|
|
||||||
|
public interface IReputationEntity
|
||||||
|
{
|
||||||
|
public long UserId { get; set; }
|
||||||
|
public long ChatId { get; set; }
|
||||||
|
public double Value { get; set; }
|
||||||
|
}
|
||||||
@@ -1,4 +1,6 @@
|
|||||||
using System.Threading.Tasks;
|
using System.Collections.Generic;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Kruzya.TelegramBot.Core.Data;
|
||||||
using Telegram.Bot.Types;
|
using Telegram.Bot.Types;
|
||||||
|
|
||||||
namespace Kruzya.TelegramBot.Core.Service
|
namespace Kruzya.TelegramBot.Core.Service
|
||||||
@@ -9,5 +11,6 @@ namespace Kruzya.TelegramBot.Core.Service
|
|||||||
public Task<double> SetUserReputationAsync(Chat chat, User user, double value);
|
public Task<double> SetUserReputationAsync(Chat chat, User user, double value);
|
||||||
public Task<double> GetReputationDiffAsync(Chat chat, User sender, User receiver);
|
public Task<double> GetReputationDiffAsync(Chat chat, User sender, User receiver);
|
||||||
public Task IncrementReputationAsync(Chat chat, User user, double diff);
|
public Task IncrementReputationAsync(Chat chat, User user, double diff);
|
||||||
|
public Task<IEnumerable<IReputationEntity>> GetChatRatingAsync(Chat chat, int limit = 10, int offset = 0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,12 @@
|
|||||||
using System;
|
using System;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using BotFramework.Abstractions;
|
||||||
|
using Kruzya.TelegramBot.Core.Cache;
|
||||||
using Microsoft.Extensions.Configuration;
|
using Microsoft.Extensions.Configuration;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
using Telegram.Bot;
|
||||||
using Telegram.Bot.Types;
|
using Telegram.Bot.Types;
|
||||||
|
|
||||||
namespace Kruzya.TelegramBot.Core.Service
|
namespace Kruzya.TelegramBot.Core.Service
|
||||||
@@ -11,10 +16,17 @@ namespace Kruzya.TelegramBot.Core.Service
|
|||||||
public class UserService
|
public class UserService
|
||||||
{
|
{
|
||||||
private readonly List<long> _superUsers;
|
private readonly List<long> _superUsers;
|
||||||
|
|
||||||
private readonly Dictionary<long, string> _statusMap;
|
private readonly Dictionary<long, string> _statusMap;
|
||||||
|
private readonly ICacheStorage<long, User> _userCache;
|
||||||
|
private readonly ITelegramBotClient _bot;
|
||||||
|
private readonly ILogger<UserService> _logger;
|
||||||
|
|
||||||
public UserService(IConfiguration configuration)
|
|
||||||
|
public UserService(
|
||||||
|
IConfiguration configuration,
|
||||||
|
IBotInstance bot,
|
||||||
|
ICacheStorage<long, User> userCache,
|
||||||
|
ILogger<UserService> logger)
|
||||||
{
|
{
|
||||||
_superUsers = configuration.GetSection("SuperUsers")
|
_superUsers = configuration.GetSection("SuperUsers")
|
||||||
.GetChildren()
|
.GetChildren()
|
||||||
@@ -24,6 +36,41 @@ namespace Kruzya.TelegramBot.Core.Service
|
|||||||
_statusMap = configuration.GetSection("CustomMemberStatus")
|
_statusMap = configuration.GetSection("CustomMemberStatus")
|
||||||
.GetChildren()
|
.GetChildren()
|
||||||
.ToDictionary(x => Convert.ToInt64(x.Key), x => x.Value);
|
.ToDictionary(x => Convert.ToInt64(x.Key), x => x.Value);
|
||||||
|
|
||||||
|
_userCache = userCache;
|
||||||
|
_bot = bot.BotClient;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task<User> GetUser(long userId, long chatId, bool bypassCache = false)
|
||||||
|
{
|
||||||
|
if (bypassCache || !_userCache.TryGetValue(userId, out var user))
|
||||||
|
{
|
||||||
|
if (bypassCache)
|
||||||
|
{
|
||||||
|
_logger.LogDebug("Bypassing cache.");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_logger.LogDebug("User {UserId} is not found in cache. Fetching from Telegram API.", userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
user = await GetUserByChat(chatId, userId);
|
||||||
|
CacheUser(user);
|
||||||
|
}
|
||||||
|
|
||||||
|
return user;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CacheUser(User user)
|
||||||
|
{
|
||||||
|
// TODO: config entry for TTL
|
||||||
|
_userCache.SetValue(user.Id, user, Convert.ToInt32(TimeSpan.FromHours(1).TotalSeconds));
|
||||||
|
}
|
||||||
|
|
||||||
|
private async Task<User> GetUserByChat(long chatId, long userId)
|
||||||
|
{
|
||||||
|
return (await _bot.GetChatMemberAsync(chatId, userId)).User;
|
||||||
}
|
}
|
||||||
|
|
||||||
public bool IsUserSuper(User user)
|
public bool IsUserSuper(User user)
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
namespace West.TelegramBot.Reputation.Data;
|
using Kruzya.TelegramBot.Core.Data;
|
||||||
|
|
||||||
public class UserReputation
|
namespace West.TelegramBot.Reputation.Data;
|
||||||
|
|
||||||
|
public class UserReputation : IReputationEntity
|
||||||
{
|
{
|
||||||
public long UserId { get; set; }
|
public long UserId { get; set; }
|
||||||
|
|
||||||
public long ChatId { get; set; }
|
public long ChatId { get; set; }
|
||||||
|
|
||||||
public double Value { get; set; }
|
public double Value { get; set; }
|
||||||
}
|
}
|
||||||
@@ -17,7 +17,7 @@ public class Reputation : CommonHandler
|
|||||||
private readonly UserService _userService;
|
private readonly UserService _userService;
|
||||||
private readonly ILogger _logger;
|
private readonly ILogger _logger;
|
||||||
private readonly IReputation _reputationService;
|
private readonly IReputation _reputationService;
|
||||||
|
|
||||||
private static readonly List<string> RepUpChars = new()
|
private static readonly List<string> RepUpChars = new()
|
||||||
{
|
{
|
||||||
"+",
|
"+",
|
||||||
@@ -43,8 +43,7 @@ public class Reputation : CommonHandler
|
|||||||
CoreContext db,
|
CoreContext db,
|
||||||
UserService userService,
|
UserService userService,
|
||||||
IReputation reputationService,
|
IReputation reputationService,
|
||||||
ILogger<Reputation> logger
|
ILogger<Reputation> logger) : base(db)
|
||||||
) : base(db)
|
|
||||||
{
|
{
|
||||||
_reputationService = reputationService;
|
_reputationService = reputationService;
|
||||||
_userService = userService;
|
_userService = userService;
|
||||||
@@ -129,6 +128,22 @@ public class Reputation : CommonHandler
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Command(InChat.Public, "top", CommandParseMode.Both)]
|
||||||
|
public async Task HandleTop()
|
||||||
|
{
|
||||||
|
var msg = new HtmlString();
|
||||||
|
msg.TextBr("Топ репутации чата: ");
|
||||||
|
var i = 1;
|
||||||
|
foreach (var rep in await _reputationService.GetChatRatingAsync(Chat))
|
||||||
|
{
|
||||||
|
var user = await _userService.GetUser(rep.UserId, rep.ChatId);
|
||||||
|
msg.Text($"{i}. ").UserMention(user).Text($" ({rep.Value})").Br();
|
||||||
|
i++;
|
||||||
|
}
|
||||||
|
|
||||||
|
await Bot.SendHtmlStringAsync(Chat, msg);
|
||||||
|
}
|
||||||
|
|
||||||
private async Task<double> GetUserReputation(User user)
|
private async Task<double> GetUserReputation(User user)
|
||||||
{
|
{
|
||||||
return await _reputationService.GetUserReputationAsync(Chat, user);
|
return await _reputationService.GetUserReputationAsync(Chat, user);
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
using Kruzya.TelegramBot.Core.Extensions;
|
using Kruzya.TelegramBot.Core.Data;
|
||||||
|
using Kruzya.TelegramBot.Core.Extensions;
|
||||||
using Kruzya.TelegramBot.Core.Service;
|
using Kruzya.TelegramBot.Core.Service;
|
||||||
using Microsoft.EntityFrameworkCore;
|
using Microsoft.EntityFrameworkCore;
|
||||||
using Telegram.Bot.Types;
|
using Telegram.Bot.Types;
|
||||||
@@ -54,6 +55,16 @@ public class ReputationService : IReputation
|
|||||||
return Math.Round(senderValue == 0 ? 1 : Math.Sqrt(senderValue), 2);
|
return Math.Round(senderValue == 0 ? 1 : Math.Sqrt(senderValue), 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public async Task<IEnumerable<IReputationEntity>> GetChatRatingAsync(Chat chat, int limit = 10, int offset = 0)
|
||||||
|
{
|
||||||
|
return await _reputationContext.UserReputation
|
||||||
|
.Where(r => r.ChatId == chat.Id && r.Value != 0D)
|
||||||
|
.OrderByDescending(r => r.Value)
|
||||||
|
.Skip(offset)
|
||||||
|
.Take(limit)
|
||||||
|
.ToListAsync();
|
||||||
|
}
|
||||||
|
|
||||||
private async Task<UserReputation?> FindUserReputation(Chat chat, User user)
|
private async Task<UserReputation?> FindUserReputation(Chat chat, User user)
|
||||||
{
|
{
|
||||||
return await _reputationContext.UserReputation.SingleOrDefaultAsync(
|
return await _reputationContext.UserReputation.SingleOrDefaultAsync(
|
||||||
|
|||||||
Reference in New Issue
Block a user