mirror of
https://github.com/barelyprofessional/KfChatDotNet.git
synced 2026-05-03 04:52:04 -04:00
437 lines
18 KiB
C#
437 lines
18 KiB
C#
using System.Net;
|
||
using System.Text.Json;
|
||
using KfChatDotNetKickBot.Models;
|
||
using KfChatDotNetKickBot.Services;
|
||
using KfChatDotNetWsClient;
|
||
using KfChatDotNetWsClient.Models;
|
||
using KfChatDotNetWsClient.Models.Events;
|
||
using KfChatDotNetWsClient.Models.Json;
|
||
using KickWsClient.Models;
|
||
using NLog;
|
||
using Websocket.Client;
|
||
|
||
namespace KfChatDotNetKickBot;
|
||
|
||
public class KickBot
|
||
{
|
||
private readonly ChatClient _kfClient;
|
||
private readonly KickWsClient.KickWsClient _kickClient;
|
||
private readonly Logger _logger = LogManager.GetCurrentClassLogger();
|
||
private readonly ConfigModel _config;
|
||
private readonly bool _pingEnabled = true;
|
||
private bool _gambaSeshPresent;
|
||
private string _xfSessionToken = null!;
|
||
// Oh no it's an ever expanding list that may never get cleaned up!
|
||
// BUY MORE RAM
|
||
private readonly List<int> _seenMsgIds = [];
|
||
// Suppresses the command handler on initial start, so it doesn't pick up things already handled on restart
|
||
private bool _initialStartCooldown = true;
|
||
private readonly CancellationToken _cancellationToken = new();
|
||
private readonly Twitch _twitch;
|
||
private Shuffle _shuffle;
|
||
private DiscordService _discord;
|
||
private string? _lastDiscordStatus;
|
||
private bool _isBmjLive = false;
|
||
private bool _isBmjLiveSynced = false;
|
||
private Dictionary<string, int> _userIdMapping = new();
|
||
private DateTime _lastKfEvent = DateTime.Now;
|
||
|
||
public KickBot()
|
||
{
|
||
_logger.Info("Bot starting!");
|
||
const string configPath = "config.json";
|
||
if (!Path.Exists(configPath))
|
||
{
|
||
_logger.Error($"{configPath} is missing! Exiting");
|
||
Environment.Exit(1);
|
||
}
|
||
|
||
_config = JsonSerializer.Deserialize<Models.ConfigModel>(File.ReadAllText(configPath)) ??
|
||
throw new InvalidOperationException();
|
||
RefreshXfToken().Wait(_cancellationToken);
|
||
|
||
_kfClient = new ChatClient(new ChatClientConfigModel
|
||
{
|
||
WsUri = _config.KfWsEndpoint,
|
||
XfSessionToken = _xfSessionToken,
|
||
CookieDomain = _config.KfWsEndpoint.Host,
|
||
Proxy = _config.Proxy,
|
||
ReconnectTimeout = _config.KfReconnectTimeout
|
||
});
|
||
|
||
_kickClient = new KickWsClient.KickWsClient(_config.PusherEndpoint.ToString(),
|
||
_config.Proxy, _config.PusherReconnectTimeout);
|
||
|
||
_kfClient.OnMessages += OnKfChatMessage;
|
||
_kfClient.OnUsersParted += OnUsersParted;
|
||
_kfClient.OnUsersJoined += OnUsersJoined;
|
||
_kfClient.OnWsDisconnection += OnKfWsDisconnected;
|
||
_kfClient.OnWsReconnect += OnKfWsReconnected;
|
||
_kfClient.OnFailedToJoinRoom += OnFailedToJoinRoom;
|
||
|
||
_kickClient.OnStreamerIsLive += OnStreamerIsLive;
|
||
_kickClient.OnChatMessage += OnKickChatMessage;
|
||
_kickClient.OnWsReconnect += OnPusherWsReconnected;
|
||
_kickClient.OnPusherSubscriptionSucceeded += OnPusherSubscriptionSucceeded;
|
||
_kickClient.OnStopStreamBroadcast += OnStopStreamBroadcast;
|
||
|
||
_kfClient.StartWsClient().Wait(_cancellationToken);
|
||
|
||
_kickClient.StartWsClient().Wait(_cancellationToken);
|
||
foreach (var channel in _config.PusherChannels)
|
||
{
|
||
_kickClient.SendPusherSubscribe(channel);
|
||
}
|
||
|
||
_logger.Debug("Creating ping thread and starting it");
|
||
var pingThread = new Thread(PingThread);
|
||
pingThread.Start();
|
||
|
||
if (_config.BossmanJackTwitchId != null)
|
||
{
|
||
_logger.Debug("Creating Twitch live stream notification client");
|
||
_twitch = new Twitch([_config.BossmanJackTwitchId.Value], _config.Proxy, _cancellationToken);
|
||
_twitch.OnStreamStateUpdated += OnTwitchStreamStateUpdated;
|
||
_twitch.StartWsClient().Wait(_cancellationToken);
|
||
}
|
||
else
|
||
{
|
||
_logger.Debug("Ignoring Twitch client as TwitchChannels is not defined");
|
||
}
|
||
|
||
BuildShuffle();
|
||
BuildDiscord();
|
||
|
||
_logger.Debug("Blocking the main thread");
|
||
var exitEvent = new ManualResetEvent(false);
|
||
exitEvent.WaitOne();
|
||
}
|
||
|
||
private void BuildShuffle()
|
||
{
|
||
_shuffle = new Shuffle(_config.Proxy, _cancellationToken);
|
||
_shuffle.OnLatestBetUpdated += ShuffleOnLatestBetUpdated;
|
||
_shuffle.OnWsDisconnection += ShuffleOnWsDisconnection;
|
||
_shuffle.StartWsClient().Wait(_cancellationToken);
|
||
}
|
||
|
||
private void BuildDiscord()
|
||
{
|
||
_logger.Debug("Building Discord");
|
||
if (_config.DiscordToken == null)
|
||
{
|
||
_logger.Info("Not building Discord as the token is not configured");
|
||
return;
|
||
}
|
||
_discord = new DiscordService(_config.DiscordToken, _config.Proxy);
|
||
_discord.OnInvalidCredentials += DiscordOnInvalidCredentials;
|
||
_discord.OnWsDisconnection += DiscordOnWsDisconnection;
|
||
_discord.OnMessageReceived += DiscordOnMessageReceived;
|
||
_discord.OnPresenceUpdated += DiscordOnPresenceUpdated;
|
||
_discord.StartWsClient().Wait(_cancellationToken);
|
||
}
|
||
|
||
private void DiscordOnPresenceUpdated(object sender, DiscordPresenceUpdateModel presence)
|
||
{
|
||
if (presence.User.Id != _config.DiscordBmjId)
|
||
{
|
||
return;
|
||
}
|
||
if (_lastDiscordStatus == presence.Status)
|
||
{
|
||
_logger.Debug("Ignoring status update as it's the same as the last one");
|
||
return;
|
||
}
|
||
_lastDiscordStatus = presence.Status;
|
||
var clientStatus = presence.ClientStatus.Keys.Aggregate(string.Empty, (current, device) => current + $"{device} is {presence.ClientStatus[device]}; ");
|
||
_sendChatMessage($"[img]https://i.postimg.cc/cLmQrp89/discord16.png[/img] BossmanJack has updated his Discord presence: {clientStatus}");
|
||
}
|
||
|
||
private void DiscordOnMessageReceived(object sender, DiscordMessageModel message)
|
||
{
|
||
if (message.Author.Id != _config.DiscordBmjId)
|
||
{
|
||
return;
|
||
}
|
||
|
||
var result = $"[img]https://i.postimg.cc/cLmQrp89/discord16.png[/img] BossmanJack: {message.Content}";
|
||
foreach (var attachment in message.Attachments ?? [])
|
||
{
|
||
result += $"[br]Attachment: {attachment.GetProperty("filename").GetString()} {attachment.GetProperty("url").GetString()}";
|
||
}
|
||
_sendChatMessage(result);
|
||
}
|
||
|
||
private void DiscordOnWsDisconnection(object sender, DisconnectionInfo e)
|
||
{
|
||
_logger.Error($"Discord dropped, reason {e.Type}");
|
||
// This is raised when the Websocket client is disposed
|
||
// Attempting to dispose it again while it is being disposed causes a loop and probably eventually a stack overflow
|
||
if (e.Type == DisconnectionType.Exit)
|
||
{
|
||
return;
|
||
}
|
||
_discord.Dispose();
|
||
BuildDiscord();
|
||
}
|
||
|
||
private void DiscordOnInvalidCredentials(object sender, DiscordPacketReadModel packet)
|
||
{
|
||
_logger.Error("Credentials failed to validate. Killing service.");
|
||
_discord.Dispose();
|
||
}
|
||
|
||
private void ShuffleOnWsDisconnection(object sender, DisconnectionInfo e)
|
||
{
|
||
if (e.Type == DisconnectionType.ByServer)
|
||
{
|
||
_shuffle.Dispose();
|
||
BuildShuffle();
|
||
}
|
||
}
|
||
|
||
private void ShuffleOnLatestBetUpdated(object sender, ShuffleLatestBetModel bet)
|
||
{
|
||
_logger.Debug("Shuffle bet has arrived");
|
||
if (bet.Username != "TheBossmanJack")
|
||
{
|
||
_logger.Debug("Ignoring irrelevant user");
|
||
return;
|
||
}
|
||
_logger.Info("ALERT BMJ IS BETTING");
|
||
if (_isBmjLive)
|
||
{
|
||
_logger.Info("Ignoring as BMJ is live");
|
||
return;
|
||
}
|
||
|
||
// Only check once because the bot should be tracking the Twitch stream
|
||
// This is just in case he's already live while the bot starts
|
||
// He was schizo betting on Dice, so I want to avoid a lot of API requests to Twitch in case they rate limit
|
||
if (!_isBmjLiveSynced)
|
||
{
|
||
_isBmjLive = _twitch.IsStreamLive("thebossmanjack").Result;
|
||
_isBmjLiveSynced = true;
|
||
}
|
||
if (_isBmjLive)
|
||
{
|
||
_logger.Info("Double checked and he is really online");
|
||
return;
|
||
}
|
||
|
||
var payoutColor = "green";
|
||
if (float.Parse(bet.Payout) < float.Parse(bet.Amount)) payoutColor = "red";
|
||
// There will be a check for live status but ignoring that while we deal with an emergency dice situation
|
||
_sendChatMessage($"🚨🚨 {bet.Username} just bet {bet.Amount} {bet.Currency} which paid out [color={payoutColor}]{bet.Payout} {bet.Currency}[/color] ({bet.Multiplier}x) on {bet.GameName} 💰💰", true);
|
||
}
|
||
|
||
private void OnTwitchStreamStateUpdated(object sender, int channelId, bool isLive)
|
||
{
|
||
_logger.Info($"BossmanJack stream event came in. isLive => {isLive}");
|
||
if (isLive)
|
||
{
|
||
_sendChatMessage("BossmanJack just went live on Twitch! https://www.twitch.tv/thebossmanjack\r\n" +
|
||
"Ad-free re-stream at https://bossmanjack.tv courtesy of @Kees H");
|
||
_isBmjLive = true;
|
||
return;
|
||
}
|
||
_sendChatMessage("BossmanJack is no longer live! :lossmanjack:");
|
||
_isBmjLive = false;
|
||
}
|
||
|
||
private void OnFailedToJoinRoom(object sender, string message)
|
||
{
|
||
_logger.Error($"Couldn't join the room. KF returned: {message}");
|
||
_logger.Error("This is likely due to the session cookie expiring. Retrieving a new one.");
|
||
RefreshXfToken().Wait(_cancellationToken);
|
||
_kfClient.UpdateToken(_xfSessionToken);
|
||
_logger.Info("Retrieved fresh token. Reconnecting.");
|
||
_kfClient.Disconnect();
|
||
_kfClient.StartWsClient().Wait(_cancellationToken);
|
||
_logger.Info("Client should be reconnecting now");
|
||
}
|
||
|
||
private void PingThread()
|
||
{
|
||
while (_pingEnabled)
|
||
{
|
||
Thread.Sleep(TimeSpan.FromSeconds(15));
|
||
_logger.Debug("Pinging KF and Pusher");
|
||
_kfClient.SendMessage("/ping");
|
||
_kickClient.SendPusherPing();
|
||
if (_initialStartCooldown) _initialStartCooldown = false;
|
||
var inactivityTime = DateTime.Now - _lastKfEvent;
|
||
_logger.Debug($"Last KF event was {inactivityTime:g} ago");
|
||
if (inactivityTime.TotalMinutes > 10)
|
||
{
|
||
_logger.Error("Forcing reconnection as bot is completely dead");
|
||
_kfClient.Reconnect().Wait(_cancellationToken);
|
||
}
|
||
}
|
||
}
|
||
|
||
private async Task RefreshXfToken()
|
||
{
|
||
var cookie = await KfTokenService.FetchSessionTokenAsync(_config.KfDomain, _config.KfUsername, _config.KfPassword,
|
||
_config.ChromiumPath, _config.Proxy);
|
||
_logger.Debug($"FetchSessionTokenAsync returned {cookie}");
|
||
_xfSessionToken = cookie;
|
||
}
|
||
|
||
private void OnStreamerIsLive(object sender, KickModels.StreamerIsLiveEventModel? e)
|
||
{
|
||
if (e == null) return;
|
||
_sendChatMessage($"Dirt Devils LFG! @Juhlonduss is live! {e.Livestream.SessionTitle} https://kick.com/dirtdevil-enjoyer", true);
|
||
}
|
||
|
||
private void OnStopStreamBroadcast(object sender, KickModels.StopStreamBroadcastEventModel? e)
|
||
{
|
||
_sendChatMessage("The stream is so over. :lossmanjack:");
|
||
}
|
||
|
||
private void OnKfChatMessage(object sender, List<MessageModel> messages, MessagesJsonModel jsonPayload)
|
||
{
|
||
_lastKfEvent = DateTime.Now;
|
||
_logger.Debug($"Received {messages.Count} message(s)");
|
||
foreach (var message in messages)
|
||
{
|
||
_logger.Info($"KF ({message.MessageDate.ToLocalTime():HH:mm:ss}) <{message.Author.Username}> {message.Message}");
|
||
if (_config.EnableGambaSeshDetect && !_initialStartCooldown && message.Author.Id == _config.GambaSeshUserId && !_gambaSeshPresent)
|
||
{
|
||
_logger.Info("Received a GambaSesh message after cooldown and while thinking he's not here. Setting the presence flag to avoid spamming chat");
|
||
_gambaSeshPresent = true;
|
||
}
|
||
if (!_seenMsgIds.Contains(message.MessageId) && !_initialStartCooldown)
|
||
{
|
||
if (message.MessageRaw.StartsWith("!time"))
|
||
{
|
||
var bmt = new DateTimeOffset(TimeZoneInfo.ConvertTimeFromUtc(DateTime.UtcNow,
|
||
TimeZoneInfo.FindSystemTimeZoneById("Eastern Standard Time")), TimeZoneInfo.FindSystemTimeZoneById("Eastern Standard Time").BaseUtcOffset);
|
||
_sendChatMessage($"It's currently {bmt:h:mm:ss tt} BMT");
|
||
}
|
||
else if (message.MessageRaw.StartsWith("!twisted"))
|
||
{
|
||
_sendChatMessage("🦍 🗣 GET IT TWISTED 🌪 , GAMBLE ✅ . PLEASE START GAMBLING 👍 . GAMBLING IS AN INVESTMENT 🎰 AND AN INVESTMENT ONLY 👍 . YOU WILL PROFIT 💰 , YOU WILL WIN ❗ ️. YOU WILL DO ALL OF THAT 💯 , YOU UNDERSTAND ⁉ ️ YOU WILL BECOME A BILLIONAIRE 💵 📈 AND REBUILD YOUR FUCKING LIFE 🤯");
|
||
}
|
||
else if (message.MessageRaw.StartsWith("!insanity"))
|
||
{
|
||
// ReSharper disable once StringLiteralTypo
|
||
_sendChatMessage("definition of insanity = doing the same thing over and over and over excecting a different result, and heres my dumbass trying to get rich every day and losing everythign i fucking touch every fucking time FUCK this bullshit FUCK MY LIEFdefinition of insanity = doing the same thing over and over and over excecting a different result, and heres my dumbass trying to get rich every day and losing everythign i fucking touch every fucking time FUCK this bullshit FUCK MY LIEF");
|
||
}
|
||
else if (message.MessageRaw.StartsWith("!helpme"))
|
||
{
|
||
_sendChatMessage("[img]https://i.postimg.cc/fTw6tGWZ/ineedmoneydumbfuck.png[/img]", true);
|
||
}
|
||
else if (message.MessageRaw.StartsWith("!whois"))
|
||
{
|
||
var lookup = WebUtility.HtmlDecode(message.MessageRaw.Replace("!whois ", string.Empty)
|
||
.TrimStart('@').TrimEnd(' ').TrimEnd(','));
|
||
if (_userIdMapping.ContainsKey(lookup))
|
||
{
|
||
_sendChatMessage($"{lookup}'s ID is {_userIdMapping[lookup]}", true);
|
||
}
|
||
else
|
||
{
|
||
_sendChatMessage("Don't know they who they are.", true);
|
||
}
|
||
}
|
||
else if (message.MessageRaw == "!sent")
|
||
{
|
||
_sendChatMessage("[img]https://i.ibb.co/GHq7hb1/4373-g-N5-HEH2-Hkc.png[/img]", true);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
_logger.Debug($"_seenMsgIds check => {!_seenMsgIds.Contains(message.MessageId)}, _initialStartCooldown => {_initialStartCooldown}");
|
||
}
|
||
_seenMsgIds.Add(message.MessageId);
|
||
}
|
||
}
|
||
|
||
private void _sendChatMessage(string message, bool bypassSeshDetect = false)
|
||
{
|
||
if (_config.SuppressChatMessages)
|
||
{
|
||
_logger.Info("Not sending message as SuppressChatMessages is enabled");
|
||
_logger.Info($"Message was: {message}");
|
||
return;
|
||
}
|
||
if (_gambaSeshPresent && _config.EnableGambaSeshDetect && !bypassSeshDetect)
|
||
{
|
||
_logger.Info($"Not sending message '{message}' as GambaSesh is present");
|
||
return;
|
||
}
|
||
|
||
_kfClient.SendMessage(message);
|
||
}
|
||
|
||
private void OnKickChatMessage(object sender, KickModels.ChatMessageEventModel? e)
|
||
{
|
||
if (e == null) return;
|
||
_logger.Info($"Kick ({e.CreatedAt.LocalDateTime.ToLocalTime():HH:mm:ss}) <{e.Sender.Username}> {e.Content}");
|
||
_logger.Debug($"BB Code Translation: {e.Content.TranslateKickEmotes()}");
|
||
|
||
if (e.Sender.Slug != "bossmanjack") return;
|
||
|
||
_logger.Debug("Message from BossmanJack");
|
||
_sendChatMessage($"[img]{_config.KickIcon}[/img] BossmanJack: {e.Content.TranslateKickEmotes()}");
|
||
}
|
||
|
||
private void OnUsersJoined(object sender, List<UserModel> users, UsersJsonModel jsonPayload)
|
||
{
|
||
_lastKfEvent = DateTime.Now;
|
||
_logger.Debug($"Received {users.Count} user join events");
|
||
foreach (var user in users)
|
||
{
|
||
if (user.Id == _config.GambaSeshUserId && _config.EnableGambaSeshDetect)
|
||
{
|
||
_logger.Info("GambaSesh is now present");
|
||
_gambaSeshPresent = true;
|
||
}
|
||
_logger.Info($"{user.Username} joined!");
|
||
if (!_userIdMapping.ContainsKey(user.Username))
|
||
{
|
||
_userIdMapping.Add(user.Username, user.Id);
|
||
}
|
||
}
|
||
}
|
||
|
||
private void OnUsersParted(object sender, List<int> userIds)
|
||
{
|
||
_lastKfEvent = DateTime.Now;
|
||
if (userIds.Contains(_config.GambaSeshUserId) && _config.EnableGambaSeshDetect)
|
||
{
|
||
_logger.Info("GambaSesh is no longer present");
|
||
_gambaSeshPresent = false;
|
||
}
|
||
}
|
||
|
||
private void OnKfWsDisconnected(object sender, DisconnectionInfo disconnectionInfo)
|
||
{
|
||
_logger.Error($"Sneedchat disconnected due to {disconnectionInfo.Type}");
|
||
}
|
||
|
||
private void OnKfWsReconnected(object sender, ReconnectionInfo reconnectionInfo)
|
||
{
|
||
_logger.Error($"Sneedchat reconnected due to {reconnectionInfo.Type}");
|
||
_logger.Info("Resetting GambaSesh presence so it can resync if he crashed while the bot was DC'd");
|
||
_gambaSeshPresent = false;
|
||
_logger.Info($"Rejoining {_config.KfChatRoomId}");
|
||
_kfClient.JoinRoom(_config.KfChatRoomId);
|
||
}
|
||
|
||
private void OnPusherWsReconnected(object sender, ReconnectionInfo reconnectionInfo)
|
||
{
|
||
_logger.Error($"Pusher reconnected due to {reconnectionInfo.Type}");
|
||
foreach (var channel in _config.PusherChannels)
|
||
{
|
||
_logger.Info($"Rejoining {channel}");
|
||
_kickClient.SendPusherSubscribe(channel);
|
||
}
|
||
}
|
||
|
||
private void OnPusherSubscriptionSucceeded(object sender, PusherModels.BasePusherEventModel? e)
|
||
{
|
||
_logger.Info($"Pusher indicates subscription to {e?.Channel} was successful");
|
||
}
|
||
} |