MafiaTelegramBot/MafiaTelegramBot/Game/GameRooms/GameRoom.GameProcess.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MafiaTelegramBot.Controllers;
using MafiaTelegramBot.CustomCollections.Extensions;
using MafiaTelegramBot.Models;
using MafiaTelegramBot.Resources;
using Telegram.Bot.Types;
using Timer = System.Timers.Timer;

namespace MafiaTelegramBot.Game.GameRooms
{
    public partial class GameRoom
    {
        public void Start()
        {
            new Task(async () =>
            {
                IsRunning = true;
                await FirstNight();
                await FirstDay();
                await GameCycle();
                await EndOfGame();
            }).Start();
        }

        private async Task FirstDay()
        {
            IsDay = true;
            await PlayersCh.SendSticker(Stickers.Sticker["Day"]);
            await PlayersCh.Send(strings.first_day_message);
            await Task.Run(async() =>
            {
                var turnsCount = _turnOrder.Count;
                Player firstPlayer = null;
                for (var i = 0; i < turnsCount; ++i)
                {
                    var player = _turnOrder.Dequeue();
                    if(!player.IsPlaying) continue;
                    await PlayersCh.Send($"{strings.now_turn} ({player.TurnOrder}) {player.NickName}");
                    if (TimerEnabled) await Bot.SendWithMarkdown2(player.ChatId, strings.you_turn_say);
                    await player.CurrentRole.SpeakAction();
                    if (firstPlayer != null && player.IsPlaying) _turnOrder.Enqueue(player);
                    else firstPlayer = player;
                }
                _turnOrder.Enqueue(firstPlayer);
            });
        }

        private async Task FirstNight()
        {
            IsDay = false;
            await Task.Run(async() =>
            {
                await PlayersCh.SendSticker(Stickers.Sticker["Night"]);
                await PlayersCh.Send(strings.city_falls_asleep);
                var mafia = Players.Values.Where(player => player.GetRole() is Roles.Mafia).ToArray();
                var don = Players.Values.FirstOrDefault(player => player.GetRole() is Roles.Don);
                var message = strings.your_teammates;
                if (don != null) message += $"\n({don.TurnOrder}) {don.NickName} - {roles.Don}";
                message = mafia.Aggregate(message, (current, player) => current + $"\n({player.TurnOrder}) {player.NickName}");
                await _mafiaCh.Send(message);
                var resetEvent = new ManualResetEvent(false);
                var timer = new Timer
                {
                    AutoReset = false,
                    Interval = 10 * 1000
                };
                timer.Elapsed += (_, _) => resetEvent.Set();
                timer.Start();
                foreach (var player in Players.Values)
                {
                    player.CurrentRole.KnownRoles.Add(player);
                    if(player.GetRole() is Roles.Don or Roles.Mafia) player.CurrentRole.KnownRoles.AddRange(
                        Players.Values.Where(p=> player.Id != p.Id && p.GetRole() is Roles.Mafia or Roles.Don)
                    );
                }
                resetEvent.WaitOne();
            });
        }

        private async Task GameCycle()
        {
            await Task.Run(async () =>
            {
                var alivePlayers = Players.Values.Where(p => p.IsAlive).ToList();
                var aliveMafia = alivePlayers.Where(p => p.GetRole() is Roles.Don or Roles.Mafia).ToList();
                var gameNotEnded = alivePlayers.Count > 2 * aliveMafia.Count && aliveMafia.Count > 0;
                while (gameNotEnded)
                {
                    await NightPhase();
                    await SummingUpPhase();
                    alivePlayers = Players.Values.Where(p => p.IsAlive).ToList();
                    aliveMafia = alivePlayers.Where(p => p.GetRole() is Roles.Don or Roles.Mafia).ToList();
                    gameNotEnded = alivePlayers.Count > 2 * aliveMafia.Count && aliveMafia.Count > 0;
                    if (!gameNotEnded) break;
                    await DayPhase();
                    await DispatchPhase();
                    alivePlayers = Players.Values.Where(p => p.IsAlive).ToList();
                    aliveMafia = alivePlayers.Where(p => p.GetRole() is Roles.Don or Roles.Mafia).ToList();
                    gameNotEnded = alivePlayers.Count > 2 * aliveMafia.Count && aliveMafia.Count > 0;
                    if (!gameNotEnded) break;
                }
            });
        }

        private async Task DayPhase()
        {
            IsDay = true;
            
            var discussionTimer = new Timer {AutoReset = false, Interval = 60 * 1000};
            var discussionEnded = new ManualResetEvent(false);
            foreach (var player in Players.Values) player.IsSpeaker = true;
            discussionTimer.Elapsed += async (_, _) =>
            {
                foreach (var player in Players.Values) player.IsSpeaker = false;
                await PlayersCh.Send(strings.day_discussion_ended);
                discussionEnded.Set();
            };
            await PlayersCh.Send(strings.disscution_time);
            discussionTimer.Start();
            discussionEnded.WaitOne();
            
            var turnsCount = _turnOrder.Count;
            Player firstPlayer = null;
            for (var i = 0; i < turnsCount; ++i)
            {
                var player = _turnOrder.Dequeue();
                if(!player.IsPlaying || !player.IsAlive) continue;
                if (firstPlayer == null) player.IsFirst = true;
                await PlayersCh.Send($"{strings.now_turn} ({player.TurnOrder}) {player.NickName}");
                if (TimerEnabled) await Bot.SendWithMarkdown2(player.ChatId, strings.you_turn_say);
                await player.CurrentRole.SpeakAction();
                await player.CurrentRole.VotingAction();
                if (player.IsFirst) firstPlayer = player;
                else _turnOrder.Enqueue(player);
            }
            firstPlayer!.IsFirst = false;
            _turnOrder.Enqueue(firstPlayer);
        }

        private async Task NightPhase()
        {
            IsDay = false;
            await PlayersCh.SendSticker(Stickers.Sticker["Night"]);
            await PlayersCh.Send(strings.city_falls_asleep);
            await Task.Run(async () =>
            {
                var resetEvent = new ManualResetEvent(false);
                var timer = new Timer{ Interval = 60*1000, AutoReset = false };
                foreach (var player in Players.Values.Where(p=>p.IsAlive))
                {
                    await player.CurrentRole.NightAction();
                }
                timer.Elapsed += (_, _) => resetEvent.Set();
                timer.Start();
                resetEvent.WaitOne();
            });
        }

        public async Task SummingUpPhase()
        {
            await Task.Run(async () =>
            {
                var beforeKill = Players.Values.ToDictionary(p=>p.Id,p=>p.IsAlive);
                var mafiaNotAgree = "";
                var votes = Players.Values.Where(p=> p.IsAlive && p.CurrentRole.MafiaTargetId != -1 && p.CurrentRole.MafiaTargetId != -2)
                    .GroupBy(p => p.CurrentRole.MafiaTargetId)
                    .Select(item => new {id = item.Key, count = item.Count()})
                    .ToList();
                if (votes.Count == 0) mafiaNotAgree = strings.mafia_not_kill_message;
                else {
                    var max = votes.Max(item => item.count);
                    var maxObjects = votes.Where(item => item.count == max).ToList();
                    if(maxObjects.Count == 1) await Players[maxObjects[0].id].CurrentRole.Kill();
                }   
                foreach (var (role, players) in PlayersRole)
                {
                    if (role is Roles.Mafia)
                        foreach (var mafia in players) await mafia.CurrentRole.ApplyNightActionResult();
                    else if (players.Count == 1)
                    {
                        var player = players[0];
                        if (role != Roles.Doctor) player.CanBeHealed = true;
                        await player.CurrentRole.ApplyNightActionResult();
                    }
                }
                var afterKill = Players.Values.ToDictionary(p=>p.Id,p=>p.IsAlive);
                var message = strings.city_wakes_up;
                if(afterKill.IsEquals(beforeKill)) message += strings.everyone_survived;
                else foreach (var (id, alive) in afterKill)
                        if(beforeKill[id] != alive) {
                            message += alive
                            ? $"\n{Players[id].NickName} {strings.will_be_ressurected}"
                            : $"\n{Players[id].NickName} {strings.will_be_killed}";
                            await PlayersCh.SendTo(id, alive ? strings.you_will_be_ressurected : strings.you_will_be_killed);
                        }
                message += '\n' + mafiaNotAgree;
                await PlayersCh.SendSticker(Stickers.Sticker["Day"]);
                await PlayersCh.Send(message);
            });
        }

        private async Task DefencePhase(List<Player> contenders)
        {
            await Task.Run( async () =>
            {
                foreach (var contender in contenders) await contender.CurrentRole.DefenceAction();
                var votersPlayers = Players.Values.Where(p => p.IsAlive).ToArray();
                Timer votingTimer = new() {Interval = 10 * 1000, AutoReset = false};
                var resetEvent = new ManualResetEvent(false);
                votingTimer.Elapsed += (_, _) => resetEvent.Set();
                List<Message> messagesToDelete = new();
                foreach (var player in votersPlayers)
                    messagesToDelete.Add(await PlayersCh.SendTo(player.ChatId, strings.you_have_ten_seconds_to_vote,  
                        Keyboard.VoteKeyboard(contenders, player.Id, vote: Callback.VoteToKill)));
                VoteUpList.Clear();
                VoteKillList.Clear();
                votingTimer.Start();
                resetEvent.WaitOne();
                votingTimer.Stop();
                foreach (var message in messagesToDelete)
                    await PlayersCh.EditTo(message.Chat.Id, message.MessageId, strings.time_out);
            });
        }

        private async Task DispatchPhase()
        {
            await Task.Run(async () =>
            {
                if (VoteUpList.Count == 1) await VoteUpList[0].CurrentRole.Dispatch();
                else
                {
                    await DefencePhase(VoteUpList);
                    var contenders = await CalculateCandidates();
                    switch (contenders.Count)
                    {
                        case 1:
                            await contenders[0].CurrentRole.Dispatch();
                            break;
                        case > 1:
                            await DefencePhase(contenders);
                            contenders = await CalculateCandidates();
                            if (contenders.Count == 1) await contenders[0].CurrentRole.Dispatch();
                            else await PlayersCh.Send(strings.villagers_could_not_decide);
                            break;
                        default:
                            await PlayersCh.Send(strings.villagers_could_not_decide);
                            break;
                    }
                }
                VoteUpList.Clear();
                VoteKillList.Clear();
            });
        }

        private async Task<List<Player>> CalculateCandidates()
        {
            var message = strings.results_of_voting;
            foreach (var (playerId, targetId) in VoteKillList)
                message += $"\n({Players[playerId].TurnOrder}) {Players[playerId].NickName} {strings.vote_to} " +
                           $"({Players[targetId].TurnOrder}) {Players[targetId].NickName}";
            if (message == strings.results_of_voting) message = strings.no_one_voted;
            await PlayersCh.Send(message);
            
            var votes = VoteUpList.Where(p=> p.IsAlive && p.IsPlaying)
                .GroupBy(p => p.Id)
                .Select(item => new {id = item.Key, count = item.Count()})
                .ToList();
            List<Player> result = new();
            if (votes.Count == 0) return result;
            var max = votes.Max(item => item.count);
            var maxObjects = votes.Where(item => item.count == max);
            result.AddRange(maxObjects.Select(obj => Players[obj.id]));
            VoteUpList.Clear();
            VoteKillList.Clear();
            return result.ToList();
        }

        private async Task PutUpVote(long playerId, long targetId, int messageId, bool toKill = false)
        {
            var player = Players[playerId];
            if (targetId == 0)
            {
                if (!toKill) await PlayersCh.SendExcept(player.ChatId, $"{player.NickName} {strings.skip_vote}");
                await PlayersCh.EditTo(player.Id, messageId, strings.you_skip_vote);
            }
            else
            {
                var target = Players[targetId];
                VoteUpList.Add(target);
                if (toKill) VoteKillList.Add(playerId, targetId);
                if (playerId != targetId)
                {
                    if (!toKill) await PlayersCh.SendExcept(player.ChatId, $"{player.NickName} {strings.put_up_vote_to} {target.NickName}");
                    await PlayersCh.EditTo(player.Id, messageId, $"{strings.you_vote_player} {target.NickName}");
                }
                else
                {
                    if (!toKill) await PlayersCh.SendExcept(player.ChatId, $"{player.NickName} {strings.vote_to_self}");
                    await PlayersCh.EditTo(player.Id, messageId, strings.you_vote_to_self);
                } 
            }
        }

        private async Task EndOfGame()
        {
            await Task.Run(async () =>
            {
                var aliveMafia = Players.Values.Where(p => p.GetRole() is Roles.Don or Roles.Mafia && p.IsAlive).ToList();
                string additionalResult = "";
                foreach (var role in PlayersRole)
                {
                    if (role.Value.Count() != 1) continue;
                    string yellowResult = await role.Value[0].CurrentRole.IsWon();
                    if (yellowResult != "") additionalResult += "\n" + yellowResult;
                }

                if (aliveMafia.Count == 0)
                {
                    await PlayersCh.Send(strings.villagers_won + additionalResult, exceptDied: false);
                    await PlayersCh.SendSticker(Stickers.Sticker["VillagerWins"]);
                }
                else
                {
                    await PlayersCh.Send(strings.mafia_won + additionalResult, exceptDied: false);
                    await PlayersCh.SendSticker(Stickers.Sticker["MafiaWins"]);
                }
                
                var rolesMessage = strings.in_this_game_roles;
                var sortedPLayers = Players.Values.ToList();
                sortedPLayers.Sort((x, y) => x.TurnOrder - y.TurnOrder);
                foreach (var player in sortedPLayers)
                {
                    rolesMessage += $"\n({player.TurnOrder}) {player.NickName} - {player.GetRoleName()}";
                    player.ResetState();
                }

                //var dblist = UserDao.DataBase.Statistics.Where(s => sortedPLayers.Contains(s.UserId, s.Role.ToString())).ToList();
                
                await PlayersCh.Send(rolesMessage);
                IsRunning = false;
                IsDay = false;
                _turnOrder.Clear();
                Settings.Clear();
                foreach (var (_, list) in PlayersRole) list.Clear();
                if (!Players.ContainsKey(Owner.Id) || Players.Count == 0) new Task(async() => 
                    await RoomController.DissolveRoom(RoomEncrypter.GetCode(RoomName))).Start();
                else
                {
                    await Bot.SendWithMarkdown2(Owner.ChatId, strings.thanks_for_game, Keyboard.OwnerGameMenu);
                    await PlayersCh.SendExcept(Owner.Id, strings.thanks_for_game, Keyboard.PlayerGameMenu);
                }
            });
        }
    }
}