daan/geekbot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Move the roll command logic into the shared commands project

Browse source
This commit is contained in:
Daan Boerlage 2021-10-31 20:16:56 +01:00
parent 89ea6df6e2
commit 78c139293f
Signed by: daan
GPG key ID: FCE070E1E4956606
3 changed files with 113 additions and 94 deletions
Download patch file Download diff file Expand all files Collapse all files

73
src/Bot/Commands/Games/Roll/Roll.cs
View file

@ -1,14 +1,9 @@
using System;
using System.Linq;
using System.Threading.Tasks;
using Discord.Commands;
using Geekbot.Bot.Utils;
using Geekbot.Commands.Roll;
using Geekbot.Core;
using Geekbot.Core.Database;
using Geekbot.Core.Database.Models;
using Geekbot.Core.ErrorHandling;
using Geekbot.Core.Extensions;
using Geekbot.Core.GuildSettingsManager;
using Geekbot.Core.KvInMemoryStore;
using Geekbot.Core.RandomNumberGenerator;
@ -36,47 +31,14 @@ namespace Geekbot.Bot.Commands.Games.Roll
{
try
{
var inputSpan = Transaction.StartChild("CommandInput");
var number = _randomNumberGenerator.Next(1, 100);
int.TryParse(stuff, out var guess);
inputSpan.Finish();
if (guess <= 100 && guess > 0)
{
var prevRollCheckSpan = Transaction.StartChild("PrevRollCheck");
var kvKey = $"{Context?.Guild?.Id ?? 0}:{Context.User.Id}:RollsPrevious";
var prevRoll = _kvInMemoryStore.Get<RollTimeout>(kvKey);
if (prevRoll?.LastGuess == guess && prevRoll?.GuessedOn.AddDays(1) > DateTime.Now)
{
await ReplyAsync(string.Format(
Localization.Roll.NoPrevGuess,
Context.Message.Author.Mention,
DateLocalization.FormatDateTimeAsRemaining(prevRoll.GuessedOn.AddDays(1))));
Transaction.Status = SpanStatus.InvalidArgument;
return;
}
_kvInMemoryStore.Set(kvKey, new RollTimeout {LastGuess = guess, GuessedOn = DateTime.Now});
prevRollCheckSpan.Finish();
await ReplyAsync(string.Format(Localization.Roll.Rolled, Context.Message.Author.Mention, number, guess));
if (guess == number)
{
var correctGuessSpan = Transaction.StartChild("CorrectGuess");
await ReplyAsync(string.Format(Localization.Roll.Gratz, Context.Message.Author));
var user = await GetUser(Context.User.Id);
user.Rolls += 1;
_database.Rolls.Update(user);
await _database.SaveChangesAsync();
correctGuessSpan.Finish();
}
}
else
{
await ReplyAsync(string.Format(Localization.Roll.RolledNoGuess, Context.Message.Author.Mention, number));
}
var res = await new Geekbot.Commands.Roll.Roll(_kvInMemoryStore, _database, _randomNumberGenerator)
.RunFromGateway(
Context.Guild.Id,
Context.User.Id,
Context.User.Username,
stuff ?? "0"
);
await ReplyAsync(res);
}
catch (Exception e)
{
@ -84,24 +46,5 @@ namespace Geekbot.Bot.Commands.Games.Roll
Transaction.Status = SpanStatus.InternalError;
}
}
private async Task<RollsModel> GetUser(ulong userId)
{
var user = _database.Rolls.FirstOrDefault(u => u.GuildId.Equals(Context.Guild.Id.AsLong()) && u.UserId.Equals(userId.AsLong())) ?? await CreateNewRow(userId);
return user;
}
private async Task<RollsModel> CreateNewRow(ulong userId)
{
var user = new RollsModel()
{
GuildId = Context.Guild.Id.AsLong(),
UserId = userId.AsLong(),
Rolls = 0
};
var newUser = _database.Rolls.Add(user).Entity;
await _database.SaveChangesAsync();
return newUser;
}
}
}