mirror of
https://github.com/Gurge44/EndlessHostRoles.git
synced 2026-10-20 01:46:25 +00:00
368 lines
16 KiB
C#
368 lines
16 KiB
C#
using AmongUs.Data;
|
||
using AmongUs.GameOptions;
|
||
using EHR.Modules;
|
||
using EHR.Roles.Crewmate;
|
||
using EHR.Roles.Neutral;
|
||
using HarmonyLib;
|
||
using InnerNet;
|
||
using System;
|
||
using System.Text.RegularExpressions;
|
||
using System.Threading.Tasks;
|
||
using static EHR.Translator;
|
||
|
||
namespace EHR;
|
||
|
||
[HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.OnGameJoined))]
|
||
class OnGameJoinedPatch
|
||
{
|
||
public static void Postfix(AmongUsClient __instance)
|
||
{
|
||
while (!Options.IsLoaded) Task.Delay(1);
|
||
Logger.Info($"{__instance.GameId} joined lobby", "OnGameJoined");
|
||
Main.PlayerVersion = [];
|
||
if (!Main.VersionCheat.Value) RPC.RpcVersionCheck();
|
||
SoundManager.Instance?.ChangeAmbienceVolume(DataManager.Settings.Audio.AmbienceVolume);
|
||
|
||
if (GameStates.IsModHost)
|
||
Main.HostClientId = Utils.GetPlayerById(0)?.GetClientId() ?? -1;
|
||
|
||
ChatUpdatePatch.DoBlockChat = false;
|
||
GameStates.InGame = false;
|
||
ErrorText.Instance?.Clear();
|
||
|
||
if (AmongUsClient.Instance.AmHost)
|
||
{
|
||
GameStartManagerPatch.GameStartManagerUpdatePatch.exitTimer = -1;
|
||
Main.DoBlockNameChange = false;
|
||
Main.NewLobby = true;
|
||
EAC.DeNum = new();
|
||
Main.AllPlayerNames = [];
|
||
|
||
if (Main.NormalOptions?.KillCooldown == 0f)
|
||
Main.NormalOptions.KillCooldown = Main.LastKillCooldown.Value;
|
||
|
||
AURoleOptions.SetOpt(Main.NormalOptions?.Cast<IGameOptions>());
|
||
if (AURoleOptions.ShapeshifterCooldown == 0f)
|
||
AURoleOptions.ShapeshifterCooldown = Main.LastShapeshifterCooldown.Value;
|
||
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (BanManager.CheckEACList(PlayerControl.LocalPlayer.FriendCode, PlayerControl.LocalPlayer.GetClient().GetHashedPuid()) && GameStates.IsOnlineGame)
|
||
{
|
||
AmongUsClient.Instance.ExitGame(DisconnectReasons.Banned);
|
||
SceneChanger.ChangeScene("MainMenu");
|
||
}
|
||
|
||
var client = PlayerControl.LocalPlayer.GetClient();
|
||
Logger.Info($"{client.PlayerName.RemoveHtmlTags()} (ClientID: {client.Id} / FriendCode: {client.FriendCode} / HashPuid: {client.GetHashedPuid()} / Platform: {client.PlatformData.Platform}) Hosted room", "Session");
|
||
}, 1f, "OnGameJoinedPatch");
|
||
|
||
Main.SetRoles = [];
|
||
Main.SetAddOns = [];
|
||
}
|
||
}
|
||
}
|
||
|
||
[HarmonyPatch(typeof(InnerNetClient), nameof(InnerNetClient.DisconnectInternal))]
|
||
class DisconnectInternalPatch
|
||
{
|
||
public static void Prefix( /*InnerNetClient __instance,*/ DisconnectReasons reason, string stringReason)
|
||
{
|
||
ShowDisconnectPopupPatch.Reason = reason;
|
||
ShowDisconnectPopupPatch.StringReason = stringReason;
|
||
ErrorText.Instance.CheatDetected = false;
|
||
ErrorText.Instance.SBDetected = false;
|
||
ErrorText.Instance.Clear();
|
||
Cloud.StopConnect();
|
||
}
|
||
}
|
||
|
||
[HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.OnPlayerJoined))]
|
||
class OnPlayerJoinedPatch
|
||
{
|
||
public static void Postfix( /*AmongUsClient __instance,*/ [HarmonyArgument(0)] ClientData client)
|
||
{
|
||
Logger.Info($"{client.PlayerName} (ClientID: {client.Id} / FriendCode: {client.FriendCode}) joined the lobby", "Session");
|
||
if (AmongUsClient.Instance.AmHost && client.FriendCode == "" && Options.KickPlayerFriendCodeNotExist.GetBool() && !GameStates.IsLocalGame)
|
||
{
|
||
if (!BanManager.TempBanWhiteList.Contains(client.GetHashedPuid()))
|
||
BanManager.TempBanWhiteList.Add(client.GetHashedPuid());
|
||
AmongUsClient.Instance.KickPlayer(client.Id, true);
|
||
Logger.SendInGame(string.Format(GetString("Message.KickedByNoFriendCode"), client.PlayerName));
|
||
Logger.Info($"TempBanned a player {client.PlayerName} without a friend code", "Temp Ban");
|
||
}
|
||
|
||
if (AmongUsClient.Instance.AmHost && client.PlatformData.Platform == (Platforms.Android | Platforms.IPhone) && Options.KickAndroidPlayer.GetBool())
|
||
{
|
||
AmongUsClient.Instance.KickPlayer(client.Id, false);
|
||
string msg = string.Format(GetString("KickAndriodPlayer"), client.PlayerName);
|
||
Logger.SendInGame(msg);
|
||
Logger.Info(msg, "Android Kick");
|
||
}
|
||
|
||
if (AmongUsClient.Instance.AmHost && client.PlayerName.Contains("Silasticm", StringComparison.OrdinalIgnoreCase))
|
||
{
|
||
AmongUsClient.Instance.KickPlayer(client.Id, false);
|
||
Logger.SendInGame("They were probably hacking tbh");
|
||
}
|
||
|
||
if (DestroyableSingleton<FriendsListManager>.Instance.IsPlayerBlockedUsername(client.FriendCode) && AmongUsClient.Instance.AmHost)
|
||
{
|
||
AmongUsClient.Instance.KickPlayer(client.Id, true);
|
||
Logger.Info($"Ban Player ー {client.PlayerName}({client.FriendCode}) has been banned.", "BAN");
|
||
}
|
||
|
||
BanManager.CheckBanPlayer(client);
|
||
BanManager.CheckDenyNamePlayer(client);
|
||
RPC.RpcVersionCheck();
|
||
|
||
if (AmongUsClient.Instance.AmHost)
|
||
{
|
||
if (Main.SayStartTimes.ContainsKey(client.Id)) Main.SayStartTimes.Remove(client.Id);
|
||
if (Main.SayBanwordsTimes.ContainsKey(client.Id)) Main.SayBanwordsTimes.Remove(client.Id);
|
||
}
|
||
}
|
||
}
|
||
|
||
[HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.OnPlayerLeft))]
|
||
class OnPlayerLeftPatch
|
||
{
|
||
public static void Postfix(AmongUsClient __instance, [HarmonyArgument(0)] ClientData data, [HarmonyArgument(1)] DisconnectReasons reason)
|
||
{
|
||
try
|
||
{
|
||
if (GameStates.IsInGame)
|
||
{
|
||
if (data.Character.Is(CustomRoles.Lovers) && !data.Character.Data.IsDead)
|
||
{
|
||
foreach (var lovers in Main.LoversPlayers)
|
||
{
|
||
Main.IsLoversDead = true;
|
||
Main.LoversPlayers.Remove(lovers);
|
||
Main.PlayerStates[lovers.PlayerId].RemoveSubRole(CustomRoles.Lovers);
|
||
}
|
||
}
|
||
|
||
switch (data.Character.GetCustomRole())
|
||
{
|
||
case CustomRoles.Executioner when Executioner.Target.ContainsKey(data.Character.PlayerId):
|
||
Executioner.ChangeRole(data.Character);
|
||
break;
|
||
case CustomRoles.Lawyer when Lawyer.Target.ContainsKey(data.Character.PlayerId):
|
||
Lawyer.ChangeRole(data.Character);
|
||
break;
|
||
case CustomRoles.Pelican:
|
||
Pelican.OnPelicanDied(data.Character.PlayerId);
|
||
break;
|
||
case CustomRoles.Markseeker:
|
||
Markseeker.OnDeath(data.Character);
|
||
break;
|
||
}
|
||
|
||
if (Executioner.Target.ContainsValue(data.Character.PlayerId))
|
||
Executioner.ChangeRoleByTarget(data.Character);
|
||
if (Lawyer.Target.ContainsValue(data.Character.PlayerId))
|
||
Lawyer.ChangeRoleByTarget(data.Character);
|
||
if (Spiritualist.SpiritualistTarget == data.Character.PlayerId)
|
||
Spiritualist.RemoveTarget();
|
||
Postman.CheckAndResetTargets(data.Character);
|
||
GhostRolesManager.AssignedGhostRoles.Remove(data.Character.PlayerId);
|
||
|
||
Utils.AfterPlayerDeathTasks(data.Character, GameStates.IsMeeting);
|
||
|
||
PlayerState state = Main.PlayerStates[data.Character.PlayerId];
|
||
if (state.deathReason == PlayerState.DeathReason.etc) state.deathReason = PlayerState.DeathReason.Disconnected;
|
||
if (!state.IsDead) state.SetDead();
|
||
|
||
NameNotifyManager.Notice.Remove(data.Character.PlayerId);
|
||
data.Character.RpcSetName(data.Character.GetRealName(isMeeting: true));
|
||
AntiBlackout.OnDisconnect(data.Character.Data);
|
||
PlayerGameOptionsSender.RemoveSender(data.Character);
|
||
}
|
||
|
||
if (Main.HostClientId == __instance.ClientId)
|
||
{
|
||
var clientId = -1;
|
||
var player = PlayerControl.LocalPlayer;
|
||
var title = "<color=#aaaaff>" + GetString("DefaultSystemMessageTitle") + "</color>";
|
||
var name = player?.Data?.PlayerName;
|
||
var msg = string.Empty;
|
||
if (GameStates.IsInGame)
|
||
{
|
||
Utils.ErrorEnd("Host Left the Game");
|
||
msg = GetString("Message.HostLeftGameInGame");
|
||
}
|
||
else if (GameStates.IsLobby)
|
||
msg = GetString("Message.HostLeftGameInLobby");
|
||
|
||
player.SetName(title);
|
||
DestroyableSingleton<HudManager>.Instance.Chat.AddChat(player, msg);
|
||
player.SetName(name);
|
||
|
||
var writer = CustomRpcSender.Create("MessagesToSend");
|
||
writer.StartMessage(clientId);
|
||
writer.StartRpc(player.NetId, (byte)RpcCalls.SetName)
|
||
.Write(title)
|
||
.EndRpc();
|
||
writer.StartRpc(player.NetId, (byte)RpcCalls.SendChat)
|
||
.Write(msg)
|
||
.EndRpc();
|
||
writer.StartRpc(player.NetId, (byte)RpcCalls.SetName)
|
||
.Write(player.Data.PlayerName)
|
||
.EndRpc();
|
||
writer.EndMessage();
|
||
writer.SendMessage();
|
||
}
|
||
|
||
// Additional description of the reason for disconnection
|
||
switch (reason)
|
||
{
|
||
case DisconnectReasons.Hacking:
|
||
Logger.SendInGame(string.Format(GetString("PlayerLeftByAU-Anticheat"), data?.PlayerName));
|
||
break;
|
||
case DisconnectReasons.Error:
|
||
Logger.SendInGame(string.Format(GetString("PlayerLeftByError"), data?.PlayerName));
|
||
_ = new LateTask(() =>
|
||
{
|
||
CustomWinnerHolder.ResetAndSetWinner(CustomWinner.Error);
|
||
GameManager.Instance.enabled = false;
|
||
GameManager.Instance?.RpcEndGame(GameOverReason.ImpostorDisconnect, false);
|
||
}, 3f, "Disconnect Error Auto-end");
|
||
|
||
break;
|
||
}
|
||
|
||
Logger.Info($"{data?.PlayerName} - (ClientID: {data?.Id} / FriendCode: {data?.FriendCode}) - Disconnected: {reason}，Ping: ({AmongUsClient.Instance.Ping})", "Session");
|
||
|
||
if (AmongUsClient.Instance.AmHost)
|
||
{
|
||
Main.SayStartTimes.Remove(__instance.ClientId);
|
||
Main.SayBanwordsTimes.Remove(__instance.ClientId);
|
||
Main.PlayerVersion.Remove(data?.Character?.PlayerId ?? byte.MaxValue);
|
||
}
|
||
|
||
Utils.CountAlivePlayers(true);
|
||
|
||
data.Character.Data.Disconnected = true;
|
||
}
|
||
catch (NullReferenceException)
|
||
{
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Logger.Error(ex.ToString(), "OnPlayerLeftPatch.Postfix");
|
||
}
|
||
}
|
||
}
|
||
|
||
[HarmonyPatch(typeof(AmongUsClient), nameof(AmongUsClient.CreatePlayer))]
|
||
class CreatePlayerPatch
|
||
{
|
||
public static void Postfix( /*AmongUsClient __instance,*/ [HarmonyArgument(0)] ClientData client)
|
||
{
|
||
if (!AmongUsClient.Instance.AmHost) return;
|
||
|
||
Logger.Msg($"Create player data： ID {client.Character.PlayerId}: {client.PlayerName}", "CreatePlayer");
|
||
|
||
//规范昵称
|
||
var name = client.PlayerName;
|
||
if (Options.FormatNameMode.GetInt() == 2 && client.Id != AmongUsClient.Instance.ClientId)
|
||
name = Main.Get_TName_Snacks;
|
||
else
|
||
{
|
||
name = name.RemoveHtmlTags().Replace(@"\", string.Empty).Replace("/", string.Empty).Replace("\n", string.Empty).Replace("\r", string.Empty).Replace("<", string.Empty).Replace(">", string.Empty);
|
||
if (name.Length > 10) name = name[..10];
|
||
if (Options.DisableEmojiName.GetBool()) name = Regex.Replace(name, @"\p{Cs}", string.Empty);
|
||
if (Regex.Replace(Regex.Replace(name, @"\s", string.Empty), @"[\x01-\x1F,\x7F]", string.Empty).Length < 1) name = Main.Get_TName_Snacks;
|
||
}
|
||
|
||
Main.AllPlayerNames.Remove(client.Character.PlayerId);
|
||
Main.AllPlayerNames.TryAdd(client.Character.PlayerId, name);
|
||
if (!name.Equals(client.PlayerName))
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (client.Character == null) return;
|
||
Logger.Warn($"Standard nickname：{client.PlayerName} => {name}", "Name Format");
|
||
client.Character.RpcSetName(name);
|
||
}, 1f, "Name Format");
|
||
}
|
||
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (client.Character == null || !GameStates.IsLobby) return;
|
||
OptionItem.SyncAllOptions(client.Id);
|
||
}, 3f, "Sync All Options For New Player");
|
||
|
||
Main.GuessNumber[client.Character.PlayerId] = [-1, 7];
|
||
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (client.Character == null) return;
|
||
if (Main.OverrideWelcomeMsg != string.Empty) Utils.SendMessage(Main.OverrideWelcomeMsg, client.Character.PlayerId);
|
||
else TemplateManager.SendTemplate("welcome", client.Character.PlayerId, true);
|
||
}, 3f, "Welcome Message");
|
||
if (Main.OverrideWelcomeMsg == string.Empty && Main.PlayerStates.Count > 0 && Main.ClientIdList.Contains(client.Id))
|
||
{
|
||
if (Options.AutoDisplayKillLog.GetBool())
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (!AmongUsClient.Instance.IsGameStarted && client.Character != null)
|
||
{
|
||
Main.IsChatCommand = true;
|
||
Utils.ShowKillLog(client.Character.PlayerId);
|
||
}
|
||
}, 1f, "DisplayKillLog");
|
||
}
|
||
|
||
if (Options.AutoDisplayLastAddOns.GetBool())
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (!AmongUsClient.Instance.IsGameStarted && client.Character != null)
|
||
{
|
||
Main.IsChatCommand = true;
|
||
Utils.ShowLastAddOns(client.Character.PlayerId);
|
||
}
|
||
}, 1.1f, "DisplayLastAddOns");
|
||
}
|
||
|
||
if (Options.AutoDisplayLastRoles.GetBool())
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (!AmongUsClient.Instance.IsGameStarted && client.Character != null)
|
||
{
|
||
Main.IsChatCommand = true;
|
||
Utils.ShowLastRoles(client.Character.PlayerId);
|
||
}
|
||
}, 1.2f, "DisplayLastRoles");
|
||
}
|
||
|
||
if (Options.AutoDisplayLastResult.GetBool())
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (!AmongUsClient.Instance.IsGameStarted && client.Character != null)
|
||
{
|
||
Main.IsChatCommand = true;
|
||
Utils.ShowLastResult(client.Character.PlayerId);
|
||
}
|
||
}, 1.3f, "DisplayLastResult");
|
||
}
|
||
|
||
if (PlayerControl.LocalPlayer.FriendCode.GetDevUser().IsUp && Options.EnableUpMode.GetBool())
|
||
{
|
||
_ = new LateTask(() =>
|
||
{
|
||
if (!AmongUsClient.Instance.IsGameStarted && client.Character != null)
|
||
{
|
||
Main.IsChatCommand = true;
|
||
// Utils.SendMessage($"{GetString("Message.YTPlanNotice")} {PlayerControl.LocalPlayer.FriendCode.GetDevUser().UpName}", client.Character.PlayerId);
|
||
}
|
||
}, 1.4f, "DisplayUpWarnning");
|
||
}
|
||
}
|
||
}
|
||
} |