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.DataBase.EntityDao;
using MafiaTelegramBot.Game.GameRoles;
using MafiaTelegramBot.Models;
using MafiaTelegramBot.Resources;
using Microsoft.EntityFrameworkCore;
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)
                {
                    if (player.GetRole() is not (Roles.Don or Roles.Mafia or Roles.Dame)) 
                        player.CurrentRole.KnownRoles.Add(player.Id, player.GetRoleName());
                    else
                        foreach (var p in Players.Values
                            .Where(p => p.GetRole() is Roles.Don or Roles.Mafia or Roles.Dame))
                            player.CurrentRole.KnownRoles.Add(p.Id, p.GetRoleName());
                }
                resetEvent.WaitOne();
            });
        }

        private async Task GameCycle()
        {
            await Task.Run(async () =>
            {
                while (GameNotEnded())
                {
                    await NightPhase();
                    await SummingUpPhase();
                    if (!GameNotEnded()) break;
                    await DayPhase();
                    await DispatchPhase();
                    if (!GameNotEnded()) break;
                }
            });
        }

        private bool GameNotEnded()
        {
            var playersCount = Players.Values.Count(p => p.IsAlive);
            var mafiaCount = Players.Values.Count(p => p.IsAlive && p.GetRole() is Roles.Don or Roles.Mafia or Roles.Dame);
            if (PlayersRole.ContainsKey(Roles.Werewolf)
                && PlayersRole[Roles.Werewolf].Count == 1 && PlayersRole[Roles.Werewolf][0].IsAlive
                && ((WerewolfRole) PlayersRole[Roles.Werewolf][0].CurrentRole).IsMafia) mafiaCount++;
            var villagersCount = playersCount - mafiaCount;
            return villagersCount > mafiaCount && mafiaCount > 0;
        }

        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.Where(player => !player.IsBlocked)) 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 toEndQueue = null;
            for (var i = 0; i < turnsCount; ++i)
            {
                var player = _turnOrder.Dequeue();
                if(!Players.ContainsKey(player.Id) || !player.IsAlive) continue;
                if (!player.IsBlocked)
                {
                    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 (Players.ContainsKey(player.Id)) await player.CurrentRole.VotingAction(VoteUpList.Count == 0);
                }
                if (toEndQueue != null) _turnOrder.Enqueue(player);
                else toEndQueue = player;
            }
            _turnOrder.Enqueue(toEndQueue);
        }

        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();
                var doctorId = PlayersRole[Roles.Doctor].Count == 1 ? PlayersRole[Roles.Doctor][0].Id : -1;
                foreach (var player in Players.Values)
                {
                    if(doctorId != player.Id) player.CanBeHealed = true;
                    player.CanBeBlockedNight = true;
                    player.CanBeBlockedDay = true;
                    player.IsBlocked = false;
                }
                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)
                    {
                        if (Players.ContainsKey(maxObjects[0].id))
                            await Players[maxObjects[0].id].CurrentRole.Kill();
                    }
                }   
                foreach (var (role, players) in PlayersRole)
                {
                    if (role is Roles.Mafia)
                        foreach (var mafia in players.Where(p=>p.IsAlive)) await mafia.CurrentRole.ApplyNightActionResult();
                    else if (players.Count == 1 && players[0].IsAlive) await players[0].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 && Players.ContainsKey(id))
                        {
                            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 () =>
            {
                var contendersCopy = contenders.ToList();
                VoteUpList.Clear();
                VoteKillList.Clear();
                foreach (var contender in contendersCopy.Where(p => !p.IsBlocked)) await contender.CurrentRole.DefenceAction();
                var votersPlayers = Players.Values.Where(p => p.IsAlive && !p.IsBlocked).ToArray();
                Timer votingTimer = new() {Interval = 20 * 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(contendersCopy.Where(p=>Players.ContainsKey(p.Id)).ToList(), player.Id, false, Callback.VoteToKill)));
                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 () =>
            {
                switch (VoteUpList.Count)
                {
                    case 0:
                        await PlayersCh.Send(strings.nothing_up_to_vote);
                        break;
                    case 1:
                    {
                        await VoteUpList[0].CurrentRole.Dispatch();
                        break;
                    }
                    default:
                    {
                        await DefencePhase(VoteUpList);
                        VoteUpList = await CalculateCandidates();
                        switch (VoteUpList.Count)
                        {
                            case 0:
                                await PlayersCh.Send(strings.nothing_up_to_dispatch);
                                break;
                            case 1:
                                await VoteUpList[0].CurrentRole.Dispatch();
                                break;
                            case > 1:
                                await DefencePhase(VoteUpList);
                                VoteUpList = await CalculateCandidates();
                                switch (VoteUpList.Count)
                                {
                                    case 0:
                                        await PlayersCh.Send(strings.nothing_up_to_dispatch);
                                        break;
                                    case 1:
                                        await VoteUpList[0].CurrentRole.Dispatch();
                                        break;
                                    default:
                                        await PlayersCh.Send(strings.villagers_could_not_decide);
                                        break;
                                }
                                break;
                        }
                        break;
                    }
                }
                VoteUpList.Clear();
                VoteKillList.Clear();
            });
        }

        private async Task<List<Player>> CalculateCandidates()
        {
            var message = strings.results_of_voting;
            var voteKillCopy = VoteKillList.ToDictionary(e => e.Key, e => e.Value);
            foreach (var (playerId, targetId) in voteKillCopy.Where(item => item.Value != 0))
            {
                message += Players.ContainsKey(playerId) && Players.ContainsKey(targetId)
                    ? $"\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=>Players.ContainsKey(p.Id))
                .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(from obj in maxObjects select Players[obj.id]);
            return result;
        }

        private async Task PutUpVote(long playerId, long targetId, int messageId, bool toKill = false)
        {
            if (Players.ContainsKey(playerId))
            {
                if (targetId == 0 && VoteKillList.AddUniqueByKey(playerId, targetId))
                {
                    if (!toKill) await PlayersCh.SendExcept(playerId, $"{Players[playerId].NickName} {strings.skip_vote}");
                    await PlayersCh.EditTo(playerId, messageId, strings.you_skip_vote);
                }
                else
                {
                    if (Players.ContainsKey(targetId))
                    {
                        if (!toKill)
                        {
                            if (VoteUpList.AddUnique(Players[targetId]))
                            {
                                if (playerId != targetId)
                                {
                                    await PlayersCh.SendExcept(playerId, $"{Players[playerId].NickName} {strings.put_up_vote_to} {Players[targetId].NickName}");
                                    await PlayersCh.EditTo(playerId, messageId, $"{strings.you_vote_player} {Players[targetId].NickName}");
                                }
                                else
                                {
                                    await PlayersCh.SendExcept(playerId, $"{Players[playerId].NickName} {strings.vote_to_self}");
                                    await PlayersCh.EditTo(playerId, messageId, strings.you_vote_to_self);
                                }
                            }
                        } 
                        else if (VoteKillList.AddUniqueByKey(playerId, targetId))
                        {
                            VoteUpList.Add(Players[targetId]);
                            if (playerId != targetId)
                                await PlayersCh.EditTo(playerId, messageId, $"{strings.you_vote_to_kill} {Players[targetId].NickName}");
                            else
                                await PlayersCh.EditTo(playerId, messageId, strings.you_vote_to_kill_self);
                        }
                    }
                }   
            }
            else await PlayersCh.SendTo(playerId, strings.this_player_left_from_game);
        }

        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();
                var additionalResult = "";
                foreach (var (_, value) in PlayersRole)
                {
                    if (value.Count != 1) continue;
                    var yellowResult = await 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"]);
                }
                if (PlayersRole.ContainsKey(Roles.Fool) && PlayersRole[Roles.Fool].Count == 1)
                    await PlayersRole[Roles.Fool][0].CurrentRole.IsWon();
                var rolesMessage = strings.in_this_game_roles;
                var players = Players.Values.ToList();
                
                var statsQuery = "SELECT * FROM mafia.statistics WHERE";
                
                foreach (var player in players)
                {
                    rolesMessage += $"\n({player.TurnOrder}) {player.NickName} - {player.GetRoleName()}";
                    statsQuery += $" id = {player.Id} AND role = '{player.GetRole().ToString()}' OR role = 'All' OR";
                    player.ResetState();
                }

                statsQuery = statsQuery.Substring(0, statsQuery.Length - 2);

                try
                {
                    var statsList = UserDao.DataBase.Statistics.FromSqlRaw(statsQuery).ToArrayAsync().Result;

                    foreach (var player in statsList)
                    {

                        player.Games++;
                        
                        //Yellow get Wins++

                        if (aliveMafia.Count() == 0)
                        {
                            //then civil roles get Wins++;
                        }
                        else
                        {
                            //then civil rols get Wins++;
                        }
                        
                        //Check Fool Wins++
                    }

                    UserDao.DataBase.Statistics.UpdateRange(statsList);

                    await UserDao.DataBase.SaveChangesAsync();
                
                }
                catch (Exception e)
                {
                    await Console.Out.WriteLineAsync(e.Message);
                }
                
                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);
                }
                if (Players.Count <= Constants.PLAYER_DISABLE_TIMER) StartTimer();
            });
        }
    }
}