521 lines
22 KiB
C#
521 lines
22 KiB
C#
using Barotrauma.Networking;
|
|
using Microsoft.Xna.Framework;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Barotrauma.Extensions;
|
|
using System;
|
|
|
|
namespace Barotrauma
|
|
{
|
|
partial class Voting
|
|
{
|
|
private struct SubmarineVoteInfo
|
|
{
|
|
public SubmarineInfo SubmarineInfo { get; set; }
|
|
public bool TransferItems { get; set; }
|
|
|
|
public SubmarineVoteInfo(SubmarineInfo submarineInfo, bool transferItems)
|
|
{
|
|
SubmarineInfo = submarineInfo;
|
|
TransferItems = transferItems;
|
|
}
|
|
}
|
|
|
|
private readonly Dictionary<VoteType, int>
|
|
voteCountYes = new Dictionary<VoteType, int>(),
|
|
voteCountNo = new Dictionary<VoteType, int>(),
|
|
voteCountMax = new Dictionary<VoteType, int>();
|
|
|
|
public int GetVoteCountYes(VoteType voteType)
|
|
{
|
|
voteCountYes.TryGetValue(voteType, out int value);
|
|
return value;
|
|
}
|
|
public int GetVoteCountNo(VoteType voteType)
|
|
{
|
|
voteCountNo.TryGetValue(voteType, out int value);
|
|
return value;
|
|
}
|
|
public int GetVoteCountMax(VoteType voteType)
|
|
{
|
|
voteCountMax.TryGetValue(voteType, out int value);
|
|
return value;
|
|
}
|
|
public void SetVoteCountYes(VoteType voteType, int value)
|
|
{
|
|
voteCountYes[voteType] = value;
|
|
}
|
|
public void SetVoteCountNo(VoteType voteType, int value)
|
|
{
|
|
voteCountNo[voteType] = value;
|
|
}
|
|
public void SetVoteCountMax(VoteType voteType, int value)
|
|
{
|
|
voteCountMax[voteType] = value;
|
|
}
|
|
|
|
public void UpdateVoteTexts(IEnumerable<Client> clients, VoteType voteType)
|
|
{
|
|
switch (voteType)
|
|
{
|
|
case VoteType.Sub:
|
|
var subList = GameMain.NetLobbyScreen.SubList;
|
|
|
|
foreach (GUIComponent comp in subList.Content.Children)
|
|
{
|
|
TryRemoveVoteText(comp);
|
|
|
|
var container = comp.GetChild<GUILayoutGroup>();
|
|
var imageFrame = container.GetChild<GUIFrame>();
|
|
var coalIcon = imageFrame.GetChildByUserData(NetLobbyScreen.CoalitionIconUserData);
|
|
var sepIcon = imageFrame.GetChildByUserData(NetLobbyScreen.SeparatistsIconUserData);
|
|
|
|
coalIcon.Enabled = false;
|
|
sepIcon.Enabled = false;
|
|
|
|
TryRemoveVoteText(coalIcon);
|
|
TryRemoveVoteText(sepIcon);
|
|
|
|
static void TryRemoveVoteText(GUIComponent component)
|
|
{
|
|
if (component.FindChild("votes") is GUITextBlock foundText)
|
|
{
|
|
component.RemoveChild(foundText);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (clients == null) { return; }
|
|
|
|
bool isPvP = GameMain.NetLobbyScreen?.SelectedMode == GameModePreset.PvP;
|
|
|
|
if (isPvP)
|
|
{
|
|
var coalitionVoteList = GetVoteCounts<SubmarineInfo>(voteType, clients.Where(static c => c.PreferredTeam is CharacterTeamType.Team1));
|
|
var separatistVoteList = GetVoteCounts<SubmarineInfo>(voteType, clients.Where(static c => c.PreferredTeam is CharacterTeamType.Team2));
|
|
foreach (var (subInfo, amount) in coalitionVoteList)
|
|
{
|
|
SetSubVoteText(subList, subInfo, amount, CharacterTeamType.Team1);
|
|
}
|
|
|
|
foreach (var (subInfo, amount) in separatistVoteList)
|
|
{
|
|
SetSubVoteText(subList, subInfo, amount, CharacterTeamType.Team2);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var subVoteList = GetVoteCounts<SubmarineInfo>(voteType, clients);
|
|
foreach (var (subInfo, amount) in subVoteList)
|
|
{
|
|
SetSubVoteText(subList, subInfo, amount, CharacterTeamType.None);
|
|
}
|
|
}
|
|
|
|
break;
|
|
case VoteType.Mode:
|
|
var modeList = GameMain.NetLobbyScreen.ModeList;
|
|
foreach (GUIComponent comp in modeList.Content.Children)
|
|
{
|
|
if (comp.FindChild("votes") is GUITextBlock voteText)
|
|
{
|
|
comp.RemoveChild(voteText);
|
|
}
|
|
}
|
|
|
|
if (clients == null) { return; }
|
|
|
|
var modeVoteList = GetVoteCounts<GameModePreset>(voteType, clients);
|
|
foreach (var (preset, amount) in modeVoteList)
|
|
{
|
|
SetVoteText(modeList, preset, amount);
|
|
}
|
|
break;
|
|
case VoteType.StartRound:
|
|
if (clients == null) { return; }
|
|
foreach (Client client in clients)
|
|
{
|
|
var clientReady = GameMain.NetLobbyScreen.PlayerList.Content.FindChild(client)?.FindChild("clientready");
|
|
if (clientReady != null)
|
|
{
|
|
clientReady.Visible = client.GetVote<bool>(VoteType.StartRound);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
private void SetSubVoteText(GUIListBox subListBox, SubmarineInfo userData, int votes, CharacterTeamType type)
|
|
{
|
|
GUIComponent subElement = subListBox.Content.GetChildByUserData(userData);
|
|
|
|
if (subElement is null)
|
|
{
|
|
DebugConsole.ThrowError("Failed to find the submarine element in the listbox");
|
|
return;
|
|
}
|
|
var (coalIcon, sepIcon) = GetPvPIcons(subElement);
|
|
|
|
switch (type)
|
|
{
|
|
case CharacterTeamType.None:
|
|
{
|
|
SetVoteText(subListBox, userData, votes);
|
|
break;
|
|
}
|
|
case CharacterTeamType.Team1:
|
|
{
|
|
coalIcon.Enabled = votes > 0;
|
|
CreateSubmarineVoteText(coalIcon, votes);
|
|
break;
|
|
}
|
|
case CharacterTeamType.Team2:
|
|
{
|
|
sepIcon.Enabled = votes > 0;
|
|
CreateSubmarineVoteText(sepIcon, votes);
|
|
break;
|
|
}
|
|
default:
|
|
return;
|
|
}
|
|
|
|
static void CreateSubmarineVoteText(GUIComponent parent, int votes)
|
|
{
|
|
if (parent is null) { return; }
|
|
var voteText = new GUITextBlock(new RectTransform(Vector2.One, parent.RectTransform, Anchor.TopLeft), $"{votes}", textAlignment: Alignment.Center)
|
|
{
|
|
Padding = Vector4.Zero,
|
|
UserData = "votes",
|
|
Shadow = true
|
|
};
|
|
voteText.RectTransform.RelativeOffset = new Vector2(0.33f, 0.33f);
|
|
}
|
|
|
|
static (GUIComponent CoalitionIcon, GUIComponent SeparatistsIcon) GetPvPIcons(GUIComponent child)
|
|
{
|
|
var container = child.GetChild<GUILayoutGroup>();
|
|
var imageFrame = container.GetChild<GUIFrame>();
|
|
var coalIcon = imageFrame.GetChildByUserData(NetLobbyScreen.CoalitionIconUserData);
|
|
var sepIcon = imageFrame.GetChildByUserData(NetLobbyScreen.SeparatistsIconUserData);
|
|
|
|
return (CoalitionIcon: coalIcon, SeparatistsIcon: sepIcon);
|
|
}
|
|
}
|
|
|
|
private void SetVoteText(GUIListBox listBox, object userData, int votes)
|
|
{
|
|
if (userData == null) { return; }
|
|
foreach (GUIComponent comp in listBox.Content.Children)
|
|
{
|
|
if (comp.UserData != userData) { continue; }
|
|
|
|
if (comp.FindChild("votes") is not GUITextBlock voteText)
|
|
{
|
|
voteText = new GUITextBlock(new RectTransform(new Point(GUI.IntScale(30), comp.Rect.Height), comp.RectTransform, Anchor.CenterRight),
|
|
"", textAlignment: Alignment.Center)
|
|
{
|
|
Padding = Vector4.Zero,
|
|
UserData = "votes"
|
|
};
|
|
}
|
|
voteText.Text = votes == 0 ? "" : votes.ToString();
|
|
}
|
|
}
|
|
|
|
public void ResetVotes(IEnumerable<Client> connectedClients)
|
|
{
|
|
foreach (Client client in connectedClients)
|
|
{
|
|
client.ResetVotes();
|
|
}
|
|
|
|
foreach (VoteType voteType in Enum.GetValues(typeof(VoteType)))
|
|
{
|
|
SetVoteCountYes(voteType, 0);
|
|
SetVoteCountNo(voteType, 0);
|
|
SetVoteCountMax(voteType, 0);
|
|
}
|
|
UpdateVoteTexts(connectedClients, VoteType.Mode);
|
|
UpdateVoteTexts(connectedClients, VoteType.Sub);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the given data is valid for the given vote type,
|
|
/// returns false otherwise. If it returns false, the message must
|
|
/// be discarded or reset by the caller, as it is now malformed :)
|
|
/// </summary>
|
|
public bool ClientWrite(IWriteMessage msg, VoteType voteType, object data)
|
|
{
|
|
msg.WriteByte((byte)voteType);
|
|
|
|
switch (voteType)
|
|
{
|
|
case VoteType.Sub:
|
|
if (data is not SubmarineInfo sub) { return false; }
|
|
msg.WriteInt32(sub.EqualityCheckVal);
|
|
if (sub.EqualityCheckVal <= 0)
|
|
{
|
|
//sub doesn't exist client-side, use hash to let the server know which one we voted for
|
|
msg.WriteString(sub.MD5Hash.StringRepresentation);
|
|
}
|
|
break;
|
|
case VoteType.Mode:
|
|
if (data is not GameModePreset gameMode) { return false; }
|
|
msg.WriteIdentifier(gameMode.Identifier);
|
|
break;
|
|
case VoteType.EndRound:
|
|
if (data is not bool endRound) { return false; }
|
|
msg.WriteBoolean(endRound);
|
|
break;
|
|
case VoteType.Kick:
|
|
if (data is not Client votedClient) { return false; }
|
|
|
|
msg.WriteByte(votedClient.SessionId);
|
|
break;
|
|
case VoteType.StartRound:
|
|
if (data is not bool startRound) { return false; }
|
|
msg.WriteBoolean(startRound);
|
|
break;
|
|
case VoteType.PurchaseAndSwitchSub:
|
|
case VoteType.PurchaseSub:
|
|
case VoteType.SwitchSub:
|
|
switch (data)
|
|
{
|
|
case (SubmarineInfo voteSub, bool transferItems):
|
|
//initiate sub vote
|
|
msg.WriteBoolean(true);
|
|
msg.WriteString(voteSub.Name);
|
|
msg.WriteBoolean(transferItems);
|
|
break;
|
|
case int vote:
|
|
// vote
|
|
msg.WriteBoolean(false);
|
|
msg.WriteInt32(vote);
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
break;
|
|
case VoteType.TransferMoney:
|
|
if (data is not int money) { return false; }
|
|
msg.WriteBoolean(false); //not initiating a vote
|
|
msg.WriteInt32(money);
|
|
break;
|
|
case VoteType.Traitor:
|
|
//use 0 to indicate we voted for no-one
|
|
msg.WriteInt32((data as Client)?.SessionId ?? 0);
|
|
break;
|
|
}
|
|
|
|
msg.WritePadBits();
|
|
return true;
|
|
}
|
|
|
|
public void ClientRead(IReadMessage inc)
|
|
{
|
|
GameMain.Client.ServerSettings.AllowSubVoting = inc.ReadBoolean();
|
|
if (GameMain.Client.ServerSettings.AllowSubVoting)
|
|
{
|
|
UpdateVoteTexts(null, VoteType.Sub);
|
|
bool isMultiSub = inc.ReadBoolean();
|
|
int votableCount = inc.ReadByte();
|
|
|
|
List<SubmarineInfo> serversubs = new List<SubmarineInfo>();
|
|
if (GameMain.NetLobbyScreen?.SubList?.Content != null)
|
|
{
|
|
foreach (GUIComponent item in GameMain.NetLobbyScreen.SubList.Content.Children)
|
|
{
|
|
if (item.UserData is SubmarineInfo info)
|
|
{
|
|
serversubs.Add(info);
|
|
}
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < votableCount; i++)
|
|
{
|
|
int votes = inc.ReadByte();
|
|
string subName = inc.ReadString();
|
|
|
|
SubmarineInfo sub = serversubs.FirstOrDefault(s => s.Name == subName);
|
|
SetSubVoteText(GameMain.NetLobbyScreen.SubList, sub, votes, isMultiSub ? CharacterTeamType.Team1 : CharacterTeamType.None);
|
|
}
|
|
|
|
if (isMultiSub)
|
|
{
|
|
int separatistsCount = inc.ReadByte();
|
|
for (int i = 0; i < separatistsCount; i++)
|
|
{
|
|
int votes = inc.ReadByte();
|
|
string subName = inc.ReadString();
|
|
|
|
SubmarineInfo sub = serversubs.FirstOrDefault(s => s.Name == subName);
|
|
SetSubVoteText(GameMain.NetLobbyScreen.SubList, sub, votes, CharacterTeamType.Team2);
|
|
}
|
|
}
|
|
}
|
|
GameMain.Client.ServerSettings.AllowModeVoting = inc.ReadBoolean();
|
|
if (GameMain.Client.ServerSettings.AllowModeVoting)
|
|
{
|
|
UpdateVoteTexts(null, VoteType.Mode);
|
|
int votableCount = inc.ReadByte();
|
|
for (int i = 0; i < votableCount; i++)
|
|
{
|
|
int votes = inc.ReadByte();
|
|
string modeIdentifier = inc.ReadString();
|
|
GameModePreset mode = GameModePreset.List.Find(m => m.Identifier == modeIdentifier);
|
|
SetVoteText(GameMain.NetLobbyScreen.ModeList, mode, votes);
|
|
}
|
|
}
|
|
GameMain.Client.ServerSettings.AllowEndVoting = inc.ReadBoolean();
|
|
if (GameMain.Client.ServerSettings.AllowEndVoting)
|
|
{
|
|
SetVoteCountYes(VoteType.EndRound, inc.ReadByte());
|
|
SetVoteCountMax(VoteType.EndRound, inc.ReadByte());
|
|
}
|
|
GameMain.Client.ServerSettings.AllowVoteKick = inc.ReadBoolean();
|
|
|
|
byte activeVoteStateByte = inc.ReadByte();
|
|
|
|
VoteState activeVoteState = VoteState.None;
|
|
try { activeVoteState = (VoteState)activeVoteStateByte; }
|
|
catch (System.Exception e)
|
|
{
|
|
DebugConsole.ThrowError("Failed to cast vote type \"" + activeVoteStateByte + "\"", e);
|
|
}
|
|
|
|
if (activeVoteState != VoteState.None)
|
|
{
|
|
byte voteTypeByte = inc.ReadByte();
|
|
VoteType voteType = VoteType.Unknown;
|
|
try { voteType = (VoteType)voteTypeByte; }
|
|
catch (System.Exception e)
|
|
{
|
|
DebugConsole.ThrowError("Failed to cast vote type \"" + voteTypeByte + "\"", e);
|
|
}
|
|
|
|
int readVote(int value)
|
|
{
|
|
byte clientCount = inc.ReadByte();
|
|
for (int i = 0; i < clientCount; i++)
|
|
{
|
|
byte clientId = inc.ReadByte();
|
|
var matchingClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == clientId);
|
|
matchingClient?.SetVote(voteType, value);
|
|
}
|
|
|
|
return clientCount;
|
|
}
|
|
|
|
int yesClientCount = readVote(value: 2);
|
|
int noClientCount = readVote(value: 1);
|
|
|
|
byte maxClientCount = inc.ReadByte();
|
|
|
|
SetVoteCountYes(voteType, yesClientCount);
|
|
SetVoteCountNo(voteType, noClientCount);
|
|
SetVoteCountMax(voteType, maxClientCount);
|
|
|
|
switch (activeVoteState)
|
|
{
|
|
case VoteState.Started:
|
|
byte starterID = inc.ReadByte();
|
|
Client starterClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == starterID);
|
|
float timeOut = inc.ReadByte();
|
|
|
|
Client myClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == GameMain.Client.SessionId);
|
|
if (myClient == null || !myClient.InGame) { return; }
|
|
|
|
switch (voteType)
|
|
{
|
|
case VoteType.PurchaseSub:
|
|
case VoteType.PurchaseAndSwitchSub:
|
|
case VoteType.SwitchSub:
|
|
string subName1 = inc.ReadString();
|
|
bool transferItems = inc.ReadBoolean();
|
|
SubmarineInfo info = GameMain.GameSession.OwnedSubmarines.FirstOrDefault(s => s.Name == subName1) ?? GameMain.Client.ServerSubmarines.FirstOrDefault(s => s.Name == subName1);
|
|
if (info == null)
|
|
{
|
|
DebugConsole.ThrowError("Failed to find a matching submarine, vote aborted");
|
|
return;
|
|
}
|
|
GameMain.Client.ShowSubmarineChangeVoteInterface(starterClient, info, voteType, transferItems, timeOut);
|
|
break;
|
|
case VoteType.TransferMoney:
|
|
byte fromClientId = inc.ReadByte();
|
|
byte toClientId = inc.ReadByte();
|
|
int transferAmount = inc.ReadInt32();
|
|
|
|
Client fromClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == fromClientId);
|
|
Client toClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == toClientId);
|
|
GameMain.Client.ShowMoneyTransferVoteInterface(starterClient, fromClient, transferAmount, toClient, timeOut);
|
|
break;
|
|
}
|
|
break;
|
|
case VoteState.Running:
|
|
// Nothing specific
|
|
break;
|
|
case VoteState.Passed:
|
|
case VoteState.Failed:
|
|
bool passed = inc.ReadBoolean();
|
|
SubmarineVoteInfo submarineVoteInfo = default;
|
|
switch (voteType)
|
|
{
|
|
case VoteType.PurchaseSub:
|
|
case VoteType.PurchaseAndSwitchSub:
|
|
case VoteType.SwitchSub:
|
|
string subName2 = inc.ReadString();
|
|
bool transferItems = inc.ReadBoolean();
|
|
if (GameMain.GameSession != null)
|
|
{
|
|
var submarineInfo = GameMain.GameSession.OwnedSubmarines.FirstOrDefault(s => s.Name == subName2) ?? GameMain.Client.ServerSubmarines.FirstOrDefault(s => s.Name == subName2);
|
|
if (submarineInfo == null)
|
|
{
|
|
DebugConsole.ThrowError("Failed to find a matching submarine, vote aborted");
|
|
return;
|
|
}
|
|
submarineVoteInfo = new SubmarineVoteInfo(submarineInfo, transferItems);
|
|
}
|
|
break;
|
|
}
|
|
|
|
GameMain.Client.VotingInterface?.EndVote(passed, yesClientCount, noClientCount);
|
|
if (passed && submarineVoteInfo.SubmarineInfo is { } subInfo)
|
|
{
|
|
switch (voteType)
|
|
{
|
|
case VoteType.PurchaseAndSwitchSub:
|
|
if (GameMain.GameSession.TryPurchaseSubmarine(subInfo))
|
|
{
|
|
GameMain.GameSession.SwitchSubmarine(subInfo, submarineVoteInfo.TransferItems);
|
|
}
|
|
break;
|
|
case VoteType.PurchaseSub:
|
|
GameMain.GameSession.TryPurchaseSubmarine(subInfo);
|
|
break;
|
|
case VoteType.SwitchSub:
|
|
GameMain.GameSession.SwitchSubmarine(subInfo, submarineVoteInfo.TransferItems);
|
|
break;
|
|
}
|
|
|
|
SubmarineSelection.ContentRefreshRequired = true;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
GameMain.NetworkMember.ConnectedClients.ForEach(c => c.SetVote(VoteType.StartRound, false));
|
|
byte readyClientCount = inc.ReadByte();
|
|
for (int i = 0; i < readyClientCount; i++)
|
|
{
|
|
byte clientId = inc.ReadByte();
|
|
var matchingClient = GameMain.NetworkMember.ConnectedClients.Find(c => c.SessionId == clientId);
|
|
matchingClient?.SetVote(VoteType.StartRound, true);
|
|
}
|
|
UpdateVoteTexts(GameMain.NetworkMember.ConnectedClients, VoteType.StartRound);
|
|
|
|
inc.ReadPadBits();
|
|
}
|
|
}
|
|
}
|