Files

90 lines
2.9 KiB
C#

using BotFramework.Attributes;
using BotFramework.Enums;
using System.Threading.Tasks;
using System;
using System.Linq;
using Kruzya.TelegramBot.Core;
using Kruzya.TelegramBot.Core.Data;
using Kruzya.TelegramBot.Core.Extensions;
using Kruzya.TelegramBot.Core.Service;
using Microsoft.Extensions.Logging;
namespace West.Entertainment.Handler;
public class Roll : CommonHandler
{
private readonly ILogger<Roll> _logger;
private readonly IReputation _reputation;
private readonly UserService _userService;
[Command(InChat.Public, "roll", CommandParseMode.Both)]
public async Task HandleRoll()
{
var cdOption = await Db.UserValues.FindOrCreateOption(Chat.Id, From.Id, "rollNextUse");
var cd = cdOption.GetValue<DateTime>();
_logger.LogInformation("[{Datetime}] {User} used /roll in {Chat}. Cooldown is: {Cooldown}",
DateTime.Now,
From.ToLog(),
Chat.ToLog(),
cd
);
if (cd > DateTime.Now)
{
await Reply($"Следующее использование через {cd - DateTime.Now:hh\\:mm\\:ss}.");
return;
}
var newRep = new Random().Next(-50, 50);
await Reply($"Вы получили {newRep} кармы.");
await _reputation.IncrementReputationAsync(Chat, From, newRep);
_logger.LogInformation("[{Datetime}] Added {Count} reputation to {User} in {Chat} from /roll",
DateTime.Now,
newRep,
From.ToLog(),
Chat.ToLog()
);
cdOption.SetValue(DateTime.Now.AddDays(1));
Db.AddOrUpdate(cdOption);
_logger.LogInformation("[{Datetime}] Updated /roll cooldown for {User} in {Chat} to {Cooldown}",
DateTime.Now,
From.ToLog(),
Chat.ToLog(),
cdOption.GetValue<DateTime>()
);
}
[Command(InChat.Public, "reset_roll", CommandParseMode.Both)]
public async Task HandleResetRoll()
{
if (!_userService.IsUserSuper(From))
{
_logger.LogWarning("{User} attempted to use super-user command /reset_roll in {Chat}",
From.ToLog(),
Chat.ToLog()
);
return;
}
var cooldownList = Db.UserValues
.Where(q => q.BotUser.ChatId == Chat.Id && q.Name == "rollNextUse");
Db.UserValues.RemoveRange(cooldownList);
await Db.SaveChangesAsync();
_logger.LogInformation("/roll cooldowns have been reset for {Chat} by {User}", Chat.ToLog(), From.ToLog());
await Reply($"Кулдаун /roll сброшен для чата \"{Chat.Title}\".");
}
public Roll(CoreContext db, IReputation reputation, UserService userService, ILogger<Roll> logger) : base(db)
{
_reputation = reputation;
_userService = userService;
_logger = logger;
}
}