Markus Isberg
110 lines
4.6 KiB
C#
110 lines
4.6 KiB
C#
using Barotrauma.Networking;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Xml.Linq;
|
|
|
|
namespace Barotrauma
|
|
{
|
|
class NPCChangeTeamAction : EventAction
|
|
{
|
|
[Serialize("", IsPropertySaveable.Yes)]
|
|
public Identifier NPCTag { get; set; }
|
|
|
|
[Serialize(CharacterTeamType.None, IsPropertySaveable.Yes)]
|
|
public CharacterTeamType TeamTag { get; set; }
|
|
|
|
[Serialize(false, IsPropertySaveable.Yes)]
|
|
public bool AddToCrew { get; set; }
|
|
|
|
[Serialize(false, IsPropertySaveable.Yes)]
|
|
public bool RemoveFromCrew { get; set; }
|
|
|
|
private bool isFinished = false;
|
|
|
|
public NPCChangeTeamAction(ScriptedEvent parentEvent, ContentXElement element) : base(parentEvent, element)
|
|
{
|
|
var enums = Enum.GetValues(typeof(CharacterTeamType)).Cast<CharacterTeamType>();
|
|
if (!enums.Contains(TeamTag))
|
|
{
|
|
DebugConsole.ThrowError($"Error in {nameof(NPCChangeTeamAction)} in the event {ParentEvent.Prefab.Identifier}. \"{TeamTag}\" is not a valid Team ID. Valid values are {string.Join(',', Enum.GetNames(typeof(CharacterTeamType)))}.");
|
|
}
|
|
}
|
|
|
|
private List<Character> affectedNpcs = null;
|
|
|
|
public override void Update(float deltaTime)
|
|
{
|
|
if (isFinished) { return; }
|
|
|
|
affectedNpcs = ParentEvent.GetTargets(NPCTag).Where(c => c is Character).Select(c => c as Character).ToList();
|
|
foreach (var npc in affectedNpcs)
|
|
{
|
|
// characters will still remain on friendlyNPC team for rest of the tick
|
|
npc.SetOriginalTeam(TeamTag);
|
|
|
|
if (AddToCrew && (TeamTag == CharacterTeamType.Team1 || TeamTag == CharacterTeamType.Team2))
|
|
{
|
|
npc.Info.StartItemsGiven = true;
|
|
GameMain.GameSession.CrewManager.AddCharacter(npc);
|
|
ChangeItemTeam(Submarine.MainSub, true);
|
|
if (GameMain.NetworkMember != null && GameMain.NetworkMember.IsServer)
|
|
{
|
|
GameMain.NetworkMember.CreateEntityEvent(npc, new Character.AddToCrewEventData(TeamTag, npc.Inventory.AllItems));
|
|
}
|
|
}
|
|
else if (RemoveFromCrew && (npc.TeamID == CharacterTeamType.Team1 || npc.TeamID == CharacterTeamType.Team2))
|
|
{
|
|
npc.Info.StartItemsGiven = true;
|
|
GameMain.GameSession.CrewManager.RemoveCharacter(npc, removeInfo: true);
|
|
var sub = Submarine.Loaded.FirstOrDefault(s => s.TeamID == TeamTag);
|
|
ChangeItemTeam(sub, false);
|
|
if (GameMain.NetworkMember != null && GameMain.NetworkMember.IsServer)
|
|
{
|
|
GameMain.NetworkMember.CreateEntityEvent(npc, new Character.RemoveFromCrewEventData(TeamTag, npc.Inventory.AllItems));
|
|
}
|
|
}
|
|
|
|
void ChangeItemTeam(Submarine sub, bool allowStealing)
|
|
{
|
|
foreach (Item item in npc.Inventory.FindAllItems(recursive: true))
|
|
{
|
|
item.AllowStealing = allowStealing;
|
|
if (item.GetComponent<Items.Components.WifiComponent>() is { } wifiComponent)
|
|
{
|
|
wifiComponent.TeamID = TeamTag;
|
|
}
|
|
if (item.GetComponent<Items.Components.IdCard>() is { } idCard)
|
|
{
|
|
idCard.TeamID = TeamTag;
|
|
idCard.SubmarineSpecificID = 0;
|
|
}
|
|
}
|
|
WayPoint subWaypoint =
|
|
WayPoint.WayPointList.Find(wp => wp.Submarine == sub && wp.SpawnType == SpawnType.Human && wp.AssignedJob == npc.Info.Job?.Prefab) ??
|
|
WayPoint.WayPointList.Find(wp => wp.Submarine == sub && wp.SpawnType == SpawnType.Human);
|
|
if (subWaypoint != null)
|
|
{
|
|
npc.GiveIdCardTags(subWaypoint, createNetworkEvent: true);
|
|
}
|
|
}
|
|
}
|
|
isFinished = true;
|
|
}
|
|
|
|
public override bool IsFinished(ref string goTo)
|
|
{
|
|
return isFinished;
|
|
}
|
|
|
|
public override void Reset()
|
|
{
|
|
isFinished = false;
|
|
}
|
|
|
|
public override string ToDebugString()
|
|
{
|
|
return $"{ToolBox.GetDebugSymbol(isFinished)} {nameof(NPCChangeTeamAction)} -> (NPCTag: {NPCTag.ColorizeObject()})";
|
|
}
|
|
}
|
|
} |