mirror of
https://github.com/barelyprofessional/KfChatDotNet.git
synced 2026-05-02 04:22:04 -04:00
Experimental convoluted rain refactor to use Redis instead of semaphores
This commit is contained in:
barelyprofessional
@@ -1,166 +1,15 @@
|
||||
using System.Text.Json;
|
||||
using System.Text.RegularExpressions;
|
||||
using KfChatDotNetBot.Extensions;
|
||||
using KfChatDotNetBot.Models;
|
||||
using KfChatDotNetBot.Models.DbModels;
|
||||
using KfChatDotNetBot.Services;
|
||||
using KfChatDotNetBot.Settings;
|
||||
using KfChatDotNetWsClient.Models.Events;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
namespace KfChatDotNetBot.Commands.Kasino;
|
||||
|
||||
[KasinoCommand]
|
||||
[WagerCommand]
|
||||
public class RainCommand : ICommand
|
||||
{
|
||||
public static class RainManager
|
||||
{
|
||||
public static Dictionary<int,Rain> rainLobbies { get; } = new(); //dictionary of lobbies with the creators id and the lobby
|
||||
public static readonly SemaphoreSlim _lock = new(1, 1);
|
||||
public static ChatBot botInstance;
|
||||
|
||||
|
||||
public static async Task<bool> AddParticipant(GamblerDbModel gambler, UserDbModel user)
|
||||
{
|
||||
await _lock.WaitAsync();
|
||||
bool openLobby = false;
|
||||
try
|
||||
{
|
||||
foreach (var lobby in rainLobbies.Values)
|
||||
{
|
||||
if (lobby.open)
|
||||
{
|
||||
lobby.AddParticipant(gambler);
|
||||
openLobby = true;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
return openLobby;
|
||||
}
|
||||
|
||||
public static async Task<bool> AddRain(int gamblerId, Rain rain)
|
||||
{
|
||||
await _lock.WaitAsync();
|
||||
try
|
||||
{
|
||||
if (!rainLobbies.TryGetValue(gamblerId, out var lobby))
|
||||
{
|
||||
rainLobbies.Add(gamblerId, lobby);
|
||||
}
|
||||
else //can't start multiple rains at the same time
|
||||
{
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{lobby.creator.FormatUsername()}, you can't start multiple rains at the same time.");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public static async Task PayoutRain(Rain rain)
|
||||
{
|
||||
if (rain.participants.Count == 0)
|
||||
{
|
||||
await botInstance.SendChatMessageAsync($"{rain.creator.FormatUsername()} made it rain on nobody.");
|
||||
await Money.ModifyBalanceAsync(rain.creatorGambler.Id, -rain.rainAmount, TransactionSourceEventType.Rain);
|
||||
return;
|
||||
}
|
||||
else
|
||||
{
|
||||
decimal payout = rain.rainAmount / rain.participants.Count;
|
||||
string rainParticipants = "";
|
||||
int counter = 0;
|
||||
foreach (var participant in rain.participants)
|
||||
{
|
||||
rainParticipants += (rain.participants.Count > 1 && counter == rain.participants.Count - 1) ? $"and {participant.User.FormatUsername()}" : $"{participant.User.FormatUsername()}, ";
|
||||
await Money.ModifyBalanceAsync(participant.Id, payout, TransactionSourceEventType.Rain);
|
||||
counter++;
|
||||
}
|
||||
if (rain.participants.Count == 1) rainParticipants = rain.participants[0].User.FormatUsername();
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{rain.creator.FormatUsername()} made it rain {payout.FormatKasinoCurrencyAsync()} on {rainParticipants}",
|
||||
true, autoDeleteAfter: TimeSpan.FromSeconds(30));
|
||||
await Money.ModifyBalanceAsync(rain.creatorGambler.Id, -rain.rainAmount, TransactionSourceEventType.Rain);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
public static async Task RemoveRain(Rain rain)
|
||||
{
|
||||
await _lock.WaitAsync();
|
||||
try
|
||||
{
|
||||
rainLobbies.Remove(rain.creatorGambler.Id);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_lock.Release();
|
||||
}
|
||||
}
|
||||
public class Rain
|
||||
{
|
||||
public List<GamblerDbModel> participants = new List<GamblerDbModel>();
|
||||
public UserDbModel creator;
|
||||
public GamblerDbModel creatorGambler;
|
||||
public DateTime startTime { get; } = DateTime.UtcNow;
|
||||
public bool open = true;
|
||||
public decimal rainAmount;
|
||||
|
||||
public Rain(UserDbModel user, GamblerDbModel gambler, decimal wager)
|
||||
{
|
||||
creator = user;
|
||||
creatorGambler = gambler;
|
||||
rainAmount = wager;
|
||||
_ = RunRain();
|
||||
}
|
||||
|
||||
public void AddParticipant(GamblerDbModel gambler)
|
||||
{
|
||||
participants.Add(gambler);
|
||||
}
|
||||
|
||||
public async Task RunRain()
|
||||
{
|
||||
var chatTimer = TimeSpan.FromSeconds(5);
|
||||
int rainTimer = 60;
|
||||
var msgId = await botInstance.SendChatMessageAsync(
|
||||
$"{creator.FormatUsername()} is making it rain with {rainAmount.FormatKasinoCurrencyAsync()}! You have {rainTimer} seconds left to join!", true,
|
||||
autoDeleteAfter: TimeSpan.FromSeconds(rainTimer));
|
||||
int num = 0;
|
||||
while (msgId.ChatMessageId == null)
|
||||
{
|
||||
num++;
|
||||
if (msgId.Status is SentMessageTrackerStatus.NotSending or SentMessageTrackerStatus.Lost) return;
|
||||
if (num > 100) return;
|
||||
await Task.Delay(100);
|
||||
}
|
||||
for (int i = 0; i < rainTimer / Convert.ToInt32(chatTimer); i++)
|
||||
{
|
||||
await Task.Delay(chatTimer);
|
||||
await botInstance.KfClient.EditMessageAsync(msgId.ChatMessageId!.Value,
|
||||
$"{creator.FormatUsername()} is making it rain with {rainAmount.FormatKasinoCurrencyAsync()}! You have {rainTimer} seconds left to join!");
|
||||
}
|
||||
open = false;
|
||||
await PayoutRain(this);
|
||||
await RemoveRain(this);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
public List<Regex> Patterns => [
|
||||
new Regex(@"^rain (?<amount>\d+)$", RegexOptions.IgnoreCase),
|
||||
new Regex(@"^rain (?<amount>\d+\.\d+)$", RegexOptions.IgnoreCase),
|
||||
@@ -169,17 +18,21 @@ public class RainCommand : ICommand
|
||||
|
||||
public string? HelpText => "!rain <amount> to start a rain, !rain to join all active rains";
|
||||
public UserRight RequiredRight => UserRight.Loser;
|
||||
public TimeSpan Timeout => TimeSpan.FromSeconds(30);
|
||||
public TimeSpan Timeout => TimeSpan.FromSeconds(60);
|
||||
public RateLimitOptionsModel? RateLimitOptions => null;
|
||||
|
||||
public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments,
|
||||
CancellationToken ctx)
|
||||
{
|
||||
if (!RainManager.botInstance.Equals(botInstance))
|
||||
{
|
||||
RainManager.botInstance = botInstance;
|
||||
}
|
||||
var cleanupDelay = TimeSpan.FromSeconds(30);
|
||||
if (botInstance.BotServices.KasinoRain == null || !botInstance.BotServices.KasinoRain.IsInitialized())
|
||||
{
|
||||
await botInstance.SendChatMessageAsync($"{user.FormatUsername()}, rain is not available at this time", true,
|
||||
autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
|
||||
var rain = await botInstance.BotServices.KasinoRain.GetRainState();
|
||||
var gambler = await Money.GetGamblerEntityAsync(user.Id, ct: ctx);
|
||||
if (gambler == null)
|
||||
{
|
||||
@@ -187,62 +40,75 @@ public class RainCommand : ICommand
|
||||
}
|
||||
if (!arguments.TryGetValue("amount", out var amount)) //if you're trying to join a rain
|
||||
{
|
||||
if (RainManager.rainLobbies.Count == 0) //if there are no lobbies
|
||||
if (rain == null) //if there are no lobbies
|
||||
{
|
||||
botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()}, there are no rain lobbies currently running. !rain <amount> to start a new rain lobby",
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()}, there's no rain currently running. !rain <amount> to start a new rain",
|
||||
true, autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
else
|
||||
|
||||
if (rain.Participants.Contains(user.Id))
|
||||
{
|
||||
bool success = await RainManager.AddParticipant(gambler, user);
|
||||
if (!success)
|
||||
{
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()}, there are no rain lobbies currently running. !rain <amount> to start a new rain lobby",
|
||||
true, autoDeleteAfter: cleanupDelay);
|
||||
}
|
||||
else
|
||||
{
|
||||
|
||||
string rainCreators = "";
|
||||
await RainManager._lock.WaitAsync();
|
||||
int lobbyCount = 0;
|
||||
try
|
||||
{
|
||||
foreach (var lobby in RainManager.rainLobbies.Values)
|
||||
{
|
||||
rainCreators += (RainManager.rainLobbies.Count > 1 && lobbyCount == RainManager.rainLobbies.Count - 1) ? $"and {lobby.creator.FormatUsername()}": $"{lobby.creator.FormatUsername()}, ";
|
||||
lobbyCount++;
|
||||
}
|
||||
|
||||
}
|
||||
finally
|
||||
{
|
||||
RainManager._lock.Release();
|
||||
}
|
||||
|
||||
string areIs = RainManager.rainLobbies.Count > 1 ? "are" : "is";
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{rainCreators} {areIs} making it rain on {user.FormatUsername()}!",true, autoDeleteAfter: cleanupDelay);
|
||||
|
||||
}
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()}, you're already participating in this rain!", true,
|
||||
autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
|
||||
await botInstance.BotServices.KasinoRain.AddParticipant(user.Id);
|
||||
var pluralSuffix = string.Empty;
|
||||
if (rain.Participants.Count > 0) pluralSuffix = "s";
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"LFG {user.FormatUsername()} is now a participant! There's now {rain.Participants.Count} participant{pluralSuffix}",
|
||||
true, autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
//if you're trying to start the rain
|
||||
decimal decAmount = Convert.ToDecimal(amount.Value);
|
||||
if (decAmount <= 0)
|
||||
{
|
||||
await botInstance.SendChatMessageAsync($"{user.FormatUsername()}, you can't make it rain with nothing.", true, autoDeleteAfter: cleanupDelay);
|
||||
await botInstance.SendChatMessageAsync($"{user.FormatUsername()}, you can't make it rain with nothing.",
|
||||
true, autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
if (gambler.Balance < decAmount)
|
||||
{
|
||||
await botInstance.SendChatMessageAsync($"{user.FormatUsername()}, your balance ${gambler.Balance} KKK is not enough to make it rain for ${decAmount} KKK.", true, autoDeleteAfter: cleanupDelay);
|
||||
await botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()}, your balance {await gambler.Balance.FormatKasinoCurrencyAsync()} is not enough to make it rain for {await decAmount.FormatKasinoCurrencyAsync()}.",
|
||||
true, autoDeleteAfter: cleanupDelay);
|
||||
return;
|
||||
}
|
||||
await RainManager.AddRain(gambler.Id,new RainManager.Rain(user, gambler, decAmount));
|
||||
|
||||
rain = new KasinoRainModel
|
||||
{
|
||||
Participants = [],
|
||||
Creator = user.Id,
|
||||
Started = DateTimeOffset.UtcNow,
|
||||
RainAmount = decAmount,
|
||||
PayoutWhen = DateTimeOffset.MaxValue
|
||||
};
|
||||
var timer = 60;
|
||||
var msg = await botInstance.SendChatMessageAsync(
|
||||
$"{user.FormatUsername()} is making it rain with {await decAmount.FormatKasinoCurrencyAsync()}! Type !rain in the next {timer} seconds to join.",
|
||||
true);
|
||||
var result = await botInstance.WaitForChatMessageAsync(msg, ct: ctx);
|
||||
if (!result)
|
||||
{
|
||||
throw new InvalidOperationException("Failed to send chat message for the rain. Not going to proceed with it");
|
||||
}
|
||||
|
||||
// Wait to set a real payout deadline only when chyat echoes the message out of fairness
|
||||
// (and also so the timer doesn't overlap with the payout deadline)
|
||||
rain.PayoutWhen = DateTimeOffset.UtcNow.AddSeconds(60);
|
||||
await botInstance.BotServices.KasinoRain.SaveRainState(rain);
|
||||
while (timer > 0)
|
||||
{
|
||||
timer--;
|
||||
await botInstance.KfClient.EditMessageAsync(msg.ChatMessageId!.Value,
|
||||
$"{user.FormatUsername()} is making it rain with {await decAmount.FormatKasinoCurrencyAsync()}! Type !rain in the next {timer} seconds to join.");
|
||||
}
|
||||
await botInstance.KfClient.DeleteMessageAsync(msg.ChatMessageId!.Value);
|
||||
// At this point the timer should take care of things but truthfully it's
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user