LuaCsForBarotraumaEP/Barotrauma/BarotraumaShared/Source/GameSession/GameModes/TraitorManager.cs

using Barotrauma.Networking;
using System.Collections.Generic;
using System.IO;

namespace Barotrauma
{
    partial class Traitor
    {
        public readonly Character Character;
        public Character TargetCharacter; //TODO: make a modular objective system (similar to crew missions) that allows for things OTHER than assasinations.

        public Traitor(Character character)
        {
            Character = character;
        }

        public void Greet(GameServer server, string codeWords, string codeResponse)
        {
            //Greeting messages TODO: Move this to a function in Traitor class
            string greetingMessage = "You are the Traitor! Your secret task is to assassinate " + TargetCharacter.Name + "! Discretion is an utmost concern; sinking the submarine and killing the entire crew "
            + "will arouse suspicion amongst the Fleet. If possible, make the death look like an accident.";
            string moreAgentsMessage = "It is possible that there are other agents on this submarine. You don't know their names, but you do have a method of communication. "
            + "Use the code words to greet the agent and code response to respond. Disguise such words in a normal-looking phrase so the crew doesn't suspect anything.";
            moreAgentsMessage += "\nThe code words are: " + codeWords + ".";
            moreAgentsMessage += "\nThe code response is: " + codeResponse + ".\n";

            if (server.Character != Character)
            {
                var chatMsg = ChatMessage.Create(
                null,
                greetingMessage + "\n" + moreAgentsMessage,
                (ChatMessageType)ChatMessageType.Server,
                null);

                var msgBox = ChatMessage.Create(
                null,
                "There might be other agents. Use these to communicate with them." +
                "\nThe code words are: " + codeWords + "." +
                "\nThe code response is: " + codeResponse + ".",
                (ChatMessageType)ChatMessageType.MessageBox,
                null);

                Client client = server.ConnectedClients.Find(c => c.Character == Character);
                GameMain.Server.SendChatMessage(chatMsg, client);
                GameMain.Server.SendChatMessage(msgBox, client);
            }

#if CLIENT
            if (server.Character == null)
            {
                new GUIMessageBox("New traitor", Character.Name + " is the traitor and the target is " + TargetCharacter.Name+".");
            }
            else if (server.Character == Character)
            {
                TraitorManager.CreateStartPopUp(TargetCharacter.Name);
                GameMain.NetworkMember.AddChatMessage(moreAgentsMessage, ChatMessageType.Server);
                return;
            }
#endif
        }
    }

    partial class TraitorManager
    {
        private static string wordsTxt = Path.Combine("Content", "CodeWords.txt");

        public List<Traitor> TraitorList
        {
            get { return traitorList; }
        }

        private List<Traitor> traitorList = new List<Traitor>();

        public string codeWords, codeResponse;

        public TraitorManager(GameServer server, int traitorCount)
        {
            if (traitorCount < 1) //what why how
            {
                traitorCount = 1;
                DebugConsole.ThrowError("Traitor Manager: TraitorCount somehow ended up less than 1, setting it to 1.");
            }
            Start(server, traitorCount);
        }

        private void Start(GameServer server, int traitorCount)
        {
            if (server == null) return;

            List<Character> characters = new List<Character>(); //ANYONE can be a target.
            List<Character> traitorCandidates = new List<Character>(); //Keep this to not re-pick traitors twice
            foreach (Client client in server.ConnectedClients)
            {
                if (client.Character != null)
                {
                    characters.Add(client.Character);
                    traitorCandidates.Add(client.Character);
                }
            }

            if (server.Character != null)
            {
                characters.Add(server.Character); //Add host character
                traitorCandidates.Add(server.Character);
            }

            if (characters.Count < 2)
            {
                return;
            }

            codeWords = ToolBox.GetRandomLine(wordsTxt) + ", " + ToolBox.GetRandomLine(wordsTxt);
            codeResponse = ToolBox.GetRandomLine(wordsTxt) + ", " + ToolBox.GetRandomLine(wordsTxt);

            while (traitorCount-- >= 0)
            {
                if (traitorCandidates.Count <= 0)
                    break;

                int traitorIndex = Rand.Int(traitorCandidates.Count);
                Character traitorCharacter = traitorCandidates[traitorIndex];
                traitorCandidates.Remove(traitorCharacter);

                //Add them to the list
                traitorList.Add(new Traitor(traitorCharacter));
            }

            //Now that traitors have been decided, let's do objectives in post for deciding things like Document Exchange.
            foreach (Traitor traitor in traitorList)
            {
                Character traitorCharacter = traitor.Character;
                int targetIndex = Rand.Int(characters.Count);
                while (characters[targetIndex] == traitorCharacter) //Cannot target self
                {
                    targetIndex = Rand.Int(characters.Count);
                }

                Character targetCharacter = characters[targetIndex];
                traitor.TargetCharacter = targetCharacter;
                traitor.Greet(server, codeWords, codeResponse);
            }
        }

        public string GetEndMessage()
        {
            if (GameMain.Server == null || traitorList.Count <= 0) return "";

            string endMessage = "";

            foreach (Traitor traitor in traitorList)
            {
                Character traitorCharacter = traitor.Character;
                Character targetCharacter = traitor.TargetCharacter;
                endMessage += traitorCharacter.Name + " was a traitor! ";
                endMessage += (traitorCharacter.Info.Gender == Gender.Male) ? "His" : "Her";
                endMessage += " task was to assassinate " + targetCharacter.Name;

                if (targetCharacter.IsDead) //Partial or complete mission success
                {
                    endMessage += ". The task was successful";
                    if (traitorCharacter.IsDead)
                    {
                        endMessage += ", but luckily the bastard didn't make it out alive either.";
                    }
                    else if (traitorCharacter.LockHands)
                    {
                        endMessage += ", but ";
                        endMessage += (traitorCharacter.Info.Gender == Gender.Male) ? "he" : "she";
                        endMessage += " was successfuly detained.";
                    }
                    else
                        endMessage += ".";
                }
                else //Partial or complete failure
                {
                    if (traitorCharacter.IsDead)
                    {
                        endMessage += ", but ";
                        endMessage += (traitorCharacter.Info.Gender == Gender.Male) ? "he" : "she";
                        endMessage += " got " + ((traitorCharacter.Info.Gender == Gender.Male) ? "himself" : "herself");
                        endMessage += " killed before completing it.";
                    }
                    else
                    {
                        endMessage += ". The task was unsuccessful";
                        if (traitorCharacter.LockHands)
                        {
                            endMessage += " - ";
                            endMessage += (traitorCharacter.Info.Gender == Gender.Male) ? "he" : "she";
                            endMessage += " was successfuly detained";
                        }
                        if (Submarine.MainSub.AtEndPosition)
                        {
                            endMessage += (traitorCharacter.LockHands ? " and " : " - ");
                            endMessage += "the submarine has reached its destination";
                        }
                        endMessage += ".";
                    }
                }
                endMessage += "\n";
            }

            return endMessage;          
        }

        //public void CharacterLeft(Character character)
        //{
        //    if (character != traitorCharacter && character != targetCharacter) return;

        //    if (character == traitorCharacter)
        //    {
        //        string endMessage = "The traitor has disconnected from the server.";
        //        End(endMessage);
        //    }
        //    else if (character == targetCharacter)
        //    {
        //        string endMessage = "The traitor's target has disconnected from the server.";
        //        End(endMessage);
        //    }
        //}
    }
}