elliebot/src/EllieBot/Modules/Gambling/AnimalRacing/AnimalRacingCommands.cs

#nullable disable
using EllieBot.Common.TypeReaders;
using EllieBot.Modules.Gambling.Common;
using EllieBot.Modules.Gambling.Common.AnimalRacing;
using EllieBot.Modules.Gambling.Common.AnimalRacing.Exceptions;
using EllieBot.Modules.Gambling.Services;
using EllieBot.Modules.Games.Quests;
using EllieBot.Modules.Games.Services;

namespace EllieBot.Modules.Gambling;

// wth is this, needs full rewrite
public partial class Gambling
{
    [Group]
    public partial class AnimalRacingCommands : GamblingModule<AnimalRaceService>
    {
        private readonly ICurrencyService _cs;
        private readonly DiscordSocketClient _client;
        private readonly GamesConfigService _gamesConf;
        private readonly QuestService _quests;

        private IUserMessage raceMessage;

        public AnimalRacingCommands(
            ICurrencyService cs,
            DiscordSocketClient client,
            GamblingConfigService gamblingConf,
            GamesConfigService gamesConf,
            QuestService quests)
            : base(gamblingConf)
        {
            _cs = cs;
            _client = client;
            _gamesConf = gamesConf;
            _quests = quests;
        }

        [Cmd]
        [RequireContext(ContextType.Guild)]
        [EllieOptions<RaceOptions>]
        public Task Race(params string[] args)
        {
            var (options, _) = OptionsParser.ParseFrom(new RaceOptions(), args);

            var ar = new AnimalRace(options, _cs, _gamesConf.Data.RaceAnimals.Shuffle(), _quests);
            if (!_service.AnimalRaces.TryAdd(ctx.Guild.Id, ar))
                return Response()
                    .Error(GetText(strs.animal_race), GetText(strs.animal_race_already_started))
                    .SendAsync();

            ar.Initialize();

            var count = 0;

            Task ClientMessageReceived(SocketMessage arg)
            {
                _ = Task.Run(() =>
                {
                    try
                    {
                        if (arg.Channel.Id == ctx.Channel.Id)
                        {
                            if (ar.CurrentPhase == AnimalRace.Phase.Running && ++count % 9 == 0)
                                raceMessage = null;
                        }
                    }
                    catch
                    {
                    }
                });
                return Task.CompletedTask;
            }

            Task ArOnEnded(AnimalRace race)
            {
                _client.MessageReceived -= ClientMessageReceived;
                _service.AnimalRaces.TryRemove(ctx.Guild.Id, out _);
                var winner = race.FinishedUsers[0];
                if (race.FinishedUsers[0].Bet > 0)
                {
                    return Response()
                        .Embed(CreateEmbed()
                            .WithOkColor()
                            .WithTitle(GetText(strs.animal_race))
                            .WithDescription(GetText(strs.animal_race_won_money(
                                Format.Bold(winner.Username),
                                winner.Animal.Icon,
                                N(race.FinishedUsers[0].Bet * race.Multi))))
                            .WithFooter($"x{race.Multi:F2}"))
                        .SendAsync();
                }

                ar.Dispose();
                return Response()
                    .Confirm(GetText(strs.animal_race),
                        GetText(strs.animal_race_won(Format.Bold(winner.Username), winner.Animal.Icon)))
                    .SendAsync();
            }

            ar.OnStartingFailed += Ar_OnStartingFailed;
            ar.OnStateUpdate += Ar_OnStateUpdate;
            ar.OnEnded += ArOnEnded;
            ar.OnStarted += Ar_OnStarted;
            _client.MessageReceived += ClientMessageReceived;

            return Response()
                .Confirm(GetText(strs.animal_race),
                    GetText(strs.animal_race_starting(options.StartTime)),
                    footer: GetText(strs.animal_race_join_instr(prefix)))
                .SendAsync();
        }

        private Task Ar_OnStarted(AnimalRace race)
        {
            if (race.Users.Count == race.MaxUsers)
                return Response().Confirm(GetText(strs.animal_race), GetText(strs.animal_race_full)).SendAsync();
            return Response()
                .Confirm(GetText(strs.animal_race),
                    GetText(strs.animal_race_starting_with_x(race.Users.Count)))
                .SendAsync();
        }

        private async Task Ar_OnStateUpdate(AnimalRace race)
        {
            var text = $@"|🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁🔚|
{string.Join("\n", race.Users.Select(p =>
{
    var index = race.FinishedUsers.IndexOf(p);
    var extra = index == -1 ? "" : $"#{index + 1} {(index == 0 ? "🏆" : "")}";
    return $"{(int)(p.Progress / 60f * 100),-2}%|{new string('‣', p.Progress) + p.Animal.Icon + extra}";
}))}
|🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁 🏁🔚|";

            var msg = raceMessage;

            if (msg is null)
                raceMessage = await Response().Confirm(text).SendAsync();
            else
            {
                await msg.ModifyAsync(x => x.Embed = CreateEmbed()
                    .WithTitle(GetText(strs.animal_race))
                    .WithDescription(text)
                    .WithOkColor()
                    .Build());
            }
        }

        private Task Ar_OnStartingFailed(AnimalRace race)
        {
            _service.AnimalRaces.TryRemove(ctx.Guild.Id, out _);
            race.Dispose();
            return Response().Error(strs.animal_race_failed).SendAsync();
        }

        [Cmd]
        [RequireContext(ContextType.Guild)]
        public async Task JoinRace([OverrideTypeReader(typeof(BalanceTypeReader))] long amount = default)
        {
            if (!await CheckBetOptional(amount))
                return;

            if (!_service.AnimalRaces.TryGetValue(ctx.Guild.Id, out var ar))
            {
                await Response().Error(strs.race_not_exist).SendAsync();
                return;
            }

            try
            {
                var user = await ar.JoinRace(ctx.User.Id, ctx.User.ToString(), amount);
                if (amount > 0)
                {
                    await Response()
                        .Confirm(GetText(strs.animal_race_join_bet(ctx.User.Mention,
                            user.Animal.Icon,
                            amount + CurrencySign)))
                        .SendAsync();
                }
                else
                    await Response()
                        .Confirm(strs.animal_race_join(ctx.User.Mention, user.Animal.Icon))
                        .SendAsync();
            }
            catch (ArgumentOutOfRangeException)
            {
                //ignore if user inputed an invalid amount
            }
            catch (AlreadyJoinedException)
            {
                // just ignore this
            }
            catch (AlreadyStartedException)
            {
                //ignore
            }
            catch (AnimalRaceFullException)
            {
                await Response().Confirm(GetText(strs.animal_race), GetText(strs.animal_race_full)).SendAsync();
            }
            catch (NotEnoughFundsException)
            {
                await Response().Error(GetText(strs.not_enough(CurrencySign))).SendAsync();
            }
        }
    }
}