nadekobot/src/NadekoBot/Modules/Gambling/AnimalRacingCommands.cs

using NadekoBot.Modules.Gambling.Common;
using NadekoBot.Modules.Gambling.Common.AnimalRacing;
using NadekoBot.Modules.Gambling.Services;
using NadekoBot.Modules.Gambling.Common.AnimalRacing.Exceptions;
using NadekoBot.Modules.Games.Services;

namespace NadekoBot.Modules.Gambling;

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

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

        private IUserMessage raceMessage = null;

        [NadekoCommand, Aliases]
        [RequireContext(ContextType.Guild)]
        [NadekoOptionsAttribute(typeof(RaceOptions))]
        public Task Race(params string[] args)
        {
            var (options, success) = OptionsParser.ParseFrom(new RaceOptions(), args);

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

            ar.Initialize();

            var count = 0;
            Task _client_MessageReceived(SocketMessage arg)
            {
                var _ = 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 Ar_OnEnded(AnimalRace race)
            {
                _client.MessageReceived -= _client_MessageReceived;
                _service.AnimalRaces.TryRemove(ctx.Guild.Id, out _);
                var winner = race.FinishedUsers[0];
                if (race.FinishedUsers[0].Bet > 0)
                {
                    return SendConfirmAsync(GetText(strs.animal_race),
                        GetText(strs.animal_race_won_money(
                            Format.Bold(winner.Username),
                            winner.Animal.Icon,
                            race.FinishedUsers[0].Bet * (race.Users.Count - 1) + CurrencySign)));
                }
                else
                {
                    return SendConfirmAsync(GetText(strs.animal_race),
                        GetText(strs.animal_race_won(Format.Bold(winner.Username), winner.Animal.Icon)));
                }
            }

            ar.OnStartingFailed += Ar_OnStartingFailed;
            ar.OnStateUpdate += Ar_OnStateUpdate;
            ar.OnEnded += Ar_OnEnded;
            ar.OnStarted += Ar_OnStarted;
            _client.MessageReceived += _client_MessageReceived;

            return SendConfirmAsync(GetText(strs.animal_race), GetText(strs.animal_race_starting(options.StartTime)),
                footer: GetText(strs.animal_race_join_instr(Prefix)));
        }

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

        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 SendConfirmAsync(text)
                    .ConfigureAwait(false);
            else
                await msg.ModifyAsync(x => x.Embed = _eb.Create()
                        .WithTitle(GetText(strs.animal_race))
                        .WithDescription(text)
                        .WithOkColor()
                        .Build())
                    .ConfigureAwait(false);
        }

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

        [NadekoCommand, Aliases]
        [RequireContext(ContextType.Guild)]
        public async Task JoinRace(ShmartNumber amount = default)
        {
            if (!await CheckBetOptional(amount).ConfigureAwait(false))
                return;

            if (!_service.AnimalRaces.TryGetValue(ctx.Guild.Id, out var ar))
            {
                await ReplyErrorLocalizedAsync(strs.race_not_exist).ConfigureAwait(false);
                return;
            }
            try
            {
                var user = await ar.JoinRace(ctx.User.Id, ctx.User.ToString(), amount)
                    .ConfigureAwait(false);
                if (amount > 0)
                    await SendConfirmAsync(GetText(strs.animal_race_join_bet(ctx.User.Mention, user.Animal.Icon, amount + CurrencySign)));
                else
                    await SendConfirmAsync(GetText(strs.animal_race_join(ctx.User.Mention, user.Animal.Icon)));
            }
            catch (ArgumentOutOfRangeException)
            {
                //ignore if user inputed an invalid amount
            }
            catch (AlreadyJoinedException)
            {
                // just ignore this
            }
            catch (AlreadyStartedException)
            {
                //ignore
            }
            catch (AnimalRaceFullException)
            {
                await SendConfirmAsync(GetText(strs.animal_race), GetText(strs.animal_race_full))
                    .ConfigureAwait(false);
            }
            catch (NotEnoughFundsException)
            {
                await SendErrorAsync(GetText(strs.not_enough(CurrencySign)));
            }
        }
    }
}