Joonas Rikkonen
321 lines
14 KiB
C#
321 lines
14 KiB
C#
//#define SERVER_IS_TRAITOR
|
||
//#define ALLOW_SOLO_TRAITOR
|
||
|
||
using System;
|
||
using Barotrauma.Networking;
|
||
using Lidgren.Network;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using Barotrauma.Extensions;
|
||
|
||
namespace Barotrauma
|
||
{
|
||
partial class Traitor
|
||
{
|
||
public class TraitorMission
|
||
{
|
||
private static System.Random random = null;
|
||
|
||
public static void InitializeRandom() => random = new System.Random((int)DateTime.UtcNow.Ticks);
|
||
|
||
// All traitor related functionality should use the following interface for generating random values
|
||
public static int Random(int n) => random.Next(n);
|
||
|
||
// All traitor related functionality should use the following interface for generating random values
|
||
public static double RandomDouble() => random.NextDouble();
|
||
|
||
private static string wordsTxt = Path.Combine("Content", "CodeWords.txt");
|
||
|
||
private readonly List<Objective> allObjectives = new List<Objective>();
|
||
private readonly List<Objective> pendingObjectives = new List<Objective>();
|
||
private readonly List<Objective> completedObjectives = new List<Objective>();
|
||
|
||
public virtual bool IsCompleted => pendingObjectives.Count <= 0;
|
||
|
||
public readonly Dictionary<string, Traitor> Traitors = new Dictionary<string, Traitor>();
|
||
|
||
public string StartText { get; private set; }
|
||
public string CodeWords { get; private set; }
|
||
public string CodeResponse { get; private set; }
|
||
public string EndMessage {
|
||
get
|
||
{
|
||
if (!Traitors.TryGetValue("traitor", out Traitor traitor))
|
||
{
|
||
return "";
|
||
}
|
||
|
||
if (pendingObjectives.Count <= 0)
|
||
{
|
||
if (completedObjectives.Count <= 0) return "";
|
||
return completedObjectives[completedObjectives.Count - 1].EndMessageText;
|
||
}
|
||
else
|
||
{
|
||
return pendingObjectives[0].EndMessageText;
|
||
}
|
||
}
|
||
}
|
||
|
||
public string GlobalEndMessageSuccessTextId { get; private set; }
|
||
public string GlobalEndMessageSuccessDeadTextId { get; private set; }
|
||
public string GlobalEndMessageSuccessDetainedTextId { get; private set; }
|
||
public string GlobalEndMessageFailureTextId { get; private set; }
|
||
public string GlobalEndMessageFailureDeadTextId { get; private set; }
|
||
public string GlobalEndMessageFailureDetainedTextId { get; private set; }
|
||
|
||
private readonly string objectiveGoalInfoFormat = "[index]. [goalinfos]\n";
|
||
|
||
public virtual IEnumerable<string> GlobalEndMessageKeys => new string[] { "[traitorname]", "[traitorgoalinfos]" };
|
||
public virtual IEnumerable<string> GlobalEndMessageValues {
|
||
get {
|
||
var isSuccess = completedObjectives.Count >= allObjectives.Count;
|
||
return new string[] {
|
||
(Traitors.TryGetValue("traitor", out var traitor) ? traitor.Character?.Name : null) ?? "(unknown)",
|
||
(isSuccess ? completedObjectives.LastOrDefault() : pendingObjectives.FirstOrDefault())?.GoalInfos ?? ""
|
||
};
|
||
}
|
||
}
|
||
|
||
public string GlobalEndMessage
|
||
{
|
||
get
|
||
{
|
||
if (!Traitors.TryGetValue("traitor", out Traitor traitor))
|
||
{
|
||
return "";
|
||
}
|
||
|
||
if (allObjectives.Count > 0)
|
||
{
|
||
var isSuccess = completedObjectives.Count >= allObjectives.Count;
|
||
var traitorIsDead = traitor.Character.IsDead;
|
||
var traitorIsDetained = traitor.Character.LockHands;
|
||
var messageId = isSuccess
|
||
? (traitorIsDead ? GlobalEndMessageSuccessDeadTextId : traitorIsDetained ? GlobalEndMessageSuccessDetainedTextId : GlobalEndMessageSuccessTextId)
|
||
: (traitorIsDead ? GlobalEndMessageFailureDeadTextId : traitorIsDetained ? GlobalEndMessageFailureDetainedTextId : GlobalEndMessageFailureTextId);
|
||
return TextManager.FormatServerMessageWithGenderPronouns(traitor.Character?.Info?.Gender ?? Gender.None, messageId, GlobalEndMessageKeys.ToArray(), GlobalEndMessageValues.ToArray());
|
||
}
|
||
return "";
|
||
}
|
||
}
|
||
|
||
public Objective GetCurrentObjective(Traitor traitor)
|
||
{
|
||
return pendingObjectives.Count > 0 ? pendingObjectives[0] : null;
|
||
}
|
||
|
||
protected List<Tuple<Client, Character>> FindTraitorCandidates(GameServer server, Character.TeamType team, params string[] traitorRoles)
|
||
{
|
||
var traitorCandidates = new List<Tuple<Client, Character>>();
|
||
#if SERVER_IS_TRAITOR
|
||
if (server.Character != null)
|
||
{
|
||
traitorCandidates.Add(server.Character);
|
||
}
|
||
else
|
||
#endif
|
||
{
|
||
traitorCandidates.AddRange(server.ConnectedClients.FindAll(c => c.Character != null && !c.Character.IsDead && (team == Character.TeamType.None || c.Character.TeamID == team)).ConvertAll(client => Tuple.Create(client, client.Character)));
|
||
}
|
||
return traitorCandidates;
|
||
}
|
||
|
||
protected List<Character> FindCharacters()
|
||
{
|
||
List<Character> characters = new List<Character>();
|
||
foreach (var character in Character.CharacterList)
|
||
{
|
||
characters.Add(character);
|
||
}
|
||
return characters;
|
||
}
|
||
|
||
public virtual bool CanBeStarted(GameServer server, TraitorManager traitorManager, Character.TeamType team, params string[] traitorRoles)
|
||
{
|
||
var traitorCandidates = FindTraitorCandidates(server, team, traitorRoles);
|
||
if (traitorCandidates.Count <= 0)
|
||
{
|
||
return false;
|
||
}
|
||
var characters = FindCharacters();
|
||
#if !ALLOW_SOLO_TRAITOR
|
||
if (characters.Count < 2)
|
||
{
|
||
return false;
|
||
}
|
||
#endif
|
||
return true;
|
||
}
|
||
|
||
public virtual bool Start(GameServer server, TraitorManager traitorManager, Character.TeamType team, params string[] traitorRoles)
|
||
{
|
||
List<Character> characters = FindCharacters();
|
||
List<Tuple<Client, Character>> traitorCandidates = FindTraitorCandidates(server, team, traitorRoles);
|
||
if (traitorCandidates.Count <= 0)
|
||
{
|
||
return false;
|
||
}
|
||
#if !ALLOW_SOLO_TRAITOR
|
||
if (characters.Count < 2)
|
||
{
|
||
return false;
|
||
}
|
||
#endif
|
||
CodeWords = ToolBox.GetRandomLine(wordsTxt) + ", " + ToolBox.GetRandomLine(wordsTxt);
|
||
CodeResponse = ToolBox.GetRandomLine(wordsTxt) + ", " + ToolBox.GetRandomLine(wordsTxt);
|
||
Traitors.Clear();
|
||
foreach (var role in traitorRoles)
|
||
{
|
||
var candidate = TraitorManager.WeightedRandom(traitorCandidates, Random, t =>
|
||
{
|
||
var previousClient = server.FindPreviousClientData(t.Item1);
|
||
return Math.Max(
|
||
previousClient != null ? traitorManager.GetTraitorCount(previousClient) : 0,
|
||
traitorManager.GetTraitorCount(Tuple.Create(t.Item1.SteamID, t.Item1.Connection?.EndPointString ?? "")));
|
||
}, (t, c) =>
|
||
{
|
||
traitorManager.SetTraitorCount(Tuple.Create(t.Item1.SteamID, t.Item1.Connection?.EndPointString ?? ""), c);
|
||
}, 2, 3);
|
||
traitorCandidates.Remove(candidate);
|
||
|
||
var traitor = new Traitor(this, role, candidate.Item2);
|
||
Traitors.Add(role, traitor);
|
||
}
|
||
|
||
var messages = new Dictionary<Traitor, List<string>>();
|
||
foreach (var traitor in Traitors.Values)
|
||
{
|
||
messages[traitor] = new List<string>();
|
||
if (traitor.CurrentObjective == null) { continue; }
|
||
traitor.Greet(server, CodeWords, CodeResponse, message => messages[traitor].Add(message));
|
||
}
|
||
|
||
messages.ForEach(traitor => traitor.Value.ForEach(message => traitor.Key.SendChatMessage(message)));
|
||
Update(0.0f, GameMain.Server.EndGame);
|
||
messages.ForEach(traitor => traitor.Value.ForEach(message => traitor.Key.SendChatMessageBox(message)));
|
||
#if SERVER
|
||
foreach (var traitor in Traitors.Values)
|
||
{
|
||
GameServer.Log(string.Format("{0} is the traitor and the current goals are:\n{1}", traitor.Character.Name, traitor.CurrentObjective?.GoalInfos != null ? TextManager.GetServerMessage(traitor.CurrentObjective?.GoalInfos) : "(empty)"), ServerLog.MessageType.ServerMessage);
|
||
}
|
||
#endif
|
||
return true;
|
||
}
|
||
|
||
public delegate void TraitorWinHandler();
|
||
|
||
public virtual void Update(float deltaTime, TraitorWinHandler winHandler)
|
||
{
|
||
if (pendingObjectives.Count <= 0 || Traitors.Count <= 0)
|
||
{
|
||
return;
|
||
}
|
||
foreach (var traitor in Traitors.Values)
|
||
{
|
||
if (traitor.Character.IsDead)
|
||
{
|
||
traitor.UpdateCurrentObjective("");
|
||
}
|
||
}
|
||
int previousCompletedCount = completedObjectives.Count;
|
||
int startedCount = 0;
|
||
while (pendingObjectives.Count > 0)
|
||
{
|
||
var objective = pendingObjectives[0];
|
||
if (!objective.IsStarted)
|
||
{
|
||
if (!objective.Start(Traitors["traitor"]))
|
||
{
|
||
pendingObjectives.RemoveAt(0);
|
||
completedObjectives.Add(objective);
|
||
if (pendingObjectives.Count > 0)
|
||
{
|
||
objective.EndMessage();
|
||
}
|
||
continue;
|
||
}
|
||
++startedCount;
|
||
}
|
||
objective.Update(deltaTime);
|
||
if (objective.IsCompleted)
|
||
{
|
||
pendingObjectives.RemoveAt(0);
|
||
completedObjectives.Add(objective);
|
||
if (pendingObjectives.Count > 0)
|
||
{
|
||
objective.EndMessage();
|
||
}
|
||
continue;
|
||
}
|
||
if (!objective.CanBeCompleted)
|
||
{
|
||
objective.EndMessage();
|
||
objective.End(true);
|
||
pendingObjectives.Clear();
|
||
}
|
||
break;
|
||
}
|
||
int completedMax = completedObjectives.Count - 1;
|
||
for (int i = previousCompletedCount; i <= completedMax; ++i)
|
||
{
|
||
var objective = completedObjectives[i];
|
||
objective.End(i < completedMax || pendingObjectives.Count > 0);
|
||
}
|
||
if (pendingObjectives.Count > 0)
|
||
{
|
||
if (startedCount > 0)
|
||
{
|
||
pendingObjectives[0].StartMessage();
|
||
}
|
||
}
|
||
else if (completedObjectives.Count >= allObjectives.Count)
|
||
{
|
||
foreach (var traitor in Traitors)
|
||
{
|
||
SteamAchievementManager.OnTraitorWin(traitor.Value.Character);
|
||
}
|
||
winHandler();
|
||
}
|
||
}
|
||
|
||
public delegate bool CharacterFilter(Character character);
|
||
public Character FindKillTarget(Character traitor, CharacterFilter filter)
|
||
{
|
||
if (traitor == null) { return null; }
|
||
|
||
List<Character> validCharacters = Character.CharacterList.FindAll(c =>
|
||
c.TeamID == traitor.TeamID &&
|
||
c != traitor &&
|
||
!c.IsDead &&
|
||
(filter == null || filter(c)));
|
||
|
||
if (validCharacters.Count > 0)
|
||
{
|
||
return validCharacters[Random(validCharacters.Count)];
|
||
}
|
||
|
||
#if ALLOW_SOLO_TRAITOR
|
||
return traitor;
|
||
#else
|
||
return null;
|
||
#endif
|
||
}
|
||
|
||
public TraitorMission(string startText, string globalEndMessageSuccessTextId, string globalEndMessageSuccessDeadTextId, string globalEndMessageSuccessDetainedTextId, string globalEndMessageFailureTextId, string globalEndMessageFailureDeadTextId, string globalEndMessageFailureDetainedTextId, params Objective[] objectives)
|
||
{
|
||
StartText = startText;
|
||
GlobalEndMessageSuccessTextId = globalEndMessageSuccessTextId;
|
||
GlobalEndMessageSuccessDeadTextId = globalEndMessageSuccessDeadTextId;
|
||
GlobalEndMessageSuccessDetainedTextId = globalEndMessageSuccessDetainedTextId;
|
||
GlobalEndMessageFailureTextId = globalEndMessageFailureTextId;
|
||
GlobalEndMessageFailureDeadTextId = globalEndMessageFailureDeadTextId;
|
||
GlobalEndMessageFailureDetainedTextId = globalEndMessageFailureDetainedTextId;
|
||
allObjectives.AddRange(objectives);
|
||
pendingObjectives.AddRange(objectives);
|
||
}
|
||
}
|
||
}
|
||
}
|