MafiaTelegramBot/MafiaTelegramBot/Game/GameRooms/ExtendedGameRoom.cs

using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using MafiaTelegramBot.Resources;

namespace MafiaTelegramBot.Game.GameRooms
{
    public class ExtendedGameRoom : GameRoom
    {
        public override bool IsExtended { get; protected set; } = true;

        public Dictionary<Roles, int> CustomRoomSettings = new();

        public override Dictionary<Roles, List<Player>> PlayersRole { get; set; } = new()
        {
            //passive roles
            [Roles.Elder] = new List<Player>(),
            [Roles.Fool] = new List<Player>(),
            [Roles.Villager] = new List<Player>(),
            //Active roles
            [Roles.Hooker] = new List<Player>(),
            [Roles.Parasite] = new List<Player>(),
            [Roles.Mafia] = new List<Player>(),
            [Roles.Don] = new List<Player>(),
            [Roles.Dame] = new List<Player>(),
            [Roles.Cop] = new List<Player>(),
            [Roles.Journalist] = new List<Player>(),
            [Roles.Detective] = new List<Player>(),
            [Roles.Lawyer] = new List<Player>(),
            [Roles.Bodyguard] = new List<Player>(),
            [Roles.Doctor] = new List<Player>(),
            [Roles.Necromancer] = new List<Player>(),
            [Roles.Werewolf] = new List<Player>(),
        };

        protected override async Task<ResultCode> ReadSettings()
        {
            return await Task.Run(() =>
            {
                var rolesCount = CustomRoomSettings.Values.Sum();
                if (rolesCount != Players.Count) return ResultCode.RolesNotEqualPlayers;
                Settings = CustomRoomSettings.ToDictionary(k=>k.Key, k=>k.Value);
                return ResultCode.CodeOk;
            });
        }

        public async Task InitSettings()
        {
            await Task.Run(() =>
            {
                CustomRoomSettings.Add(Roles.Villager, Players.Count);
                CustomRoomSettings.Add(Roles.Cop, 1);
                if (Players.Count != 8) CustomRoomSettings.Add(Roles.Doctor, 1);
                if (Players.Count % 3 == 0)
                    CustomRoomSettings.Add(Roles.Mafia, Players.Count/3);
                else
                {
                    CustomRoomSettings.Add(Roles.Mafia, Players.Count/3 -1);
                    CustomRoomSettings.Add(Roles.Don, 1);
                }
                foreach (var (key, value) in CustomRoomSettings)
                {
                    if (key != Roles.Villager)
                        CustomRoomSettings[Roles.Villager] -= value;
                }
                if (CustomRoomSettings[Roles.Mafia] < 1) CustomRoomSettings.Remove(Roles.Mafia);
                if (CustomRoomSettings[Roles.Villager] < 1) CustomRoomSettings.Remove(Roles.Villager);
            });
        }
    }
}