mirror of
https://github.com/barelyprofessional/KfChatDotNet.git
synced 2026-05-02 04:22:04 -04:00
Added Shuffle.us
This commit is contained in:
barelyprofessional
@@ -38,6 +38,7 @@ public class BotServices
|
||||
private DLive? _dliveStatusCheck;
|
||||
private PeerTube? _peerTubeStatusCheck;
|
||||
private Owncast? _owncastStatusCheck;
|
||||
private ShuffleDotUs _shuffleDotUs;
|
||||
|
||||
private Task? _websocketWatchdog;
|
||||
private Task? _howlggGetUserTimer;
|
||||
@@ -86,7 +87,8 @@ public class BotServices
|
||||
BuildParti(),
|
||||
BuildDLiveStatusCheck(),
|
||||
BuildPeerTubeLiveStatusCheck(),
|
||||
BuildOwncastLiveStatusCheck()
|
||||
BuildOwncastLiveStatusCheck(),
|
||||
BuildShuffleDotUs()
|
||||
];
|
||||
try
|
||||
{
|
||||
@@ -110,6 +112,14 @@ public class BotServices
|
||||
_shuffle.OnLatestBetUpdated += ShuffleOnLatestBetUpdated;
|
||||
await _shuffle.StartWsClient();
|
||||
}
|
||||
|
||||
private async Task BuildShuffleDotUs()
|
||||
{
|
||||
_logger.Debug("Building Shuffle.us");
|
||||
_shuffleDotUs = new ShuffleDotUs((await SettingsProvider.GetValueAsync(BuiltIn.Keys.Proxy)).Value, _cancellationToken);
|
||||
_shuffleDotUs.OnLatestBetUpdated += ShuffleOnLatestBetUpdated;
|
||||
await _shuffleDotUs.StartWsClient();
|
||||
}
|
||||
|
||||
private async Task BuildDiscord()
|
||||
{
|
||||
@@ -474,6 +484,14 @@ public class BotServices
|
||||
_parti = null!;
|
||||
await BuildParti();
|
||||
}
|
||||
|
||||
if (_shuffleDotUs != null && !_shuffleDotUs.IsConnected())
|
||||
{
|
||||
_logger.Error("Shuffle.us died, recreating it");
|
||||
_shuffleDotUs.Dispose();
|
||||
_shuffleDotUs = null!;
|
||||
await BuildShuffleDotUs();
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
@@ -867,20 +885,35 @@ public class BotServices
|
||||
{
|
||||
var settings = SettingsProvider
|
||||
.GetMultipleValuesAsync([
|
||||
BuiltIn.Keys.ShuffleBmjUsername, BuiltIn.Keys.KiwiFarmsGreenColor, BuiltIn.Keys.KiwiFarmsRedColor
|
||||
BuiltIn.Keys.ShuffleBmjUsername, BuiltIn.Keys.ShuffleDotUsBmjUsername,
|
||||
BuiltIn.Keys.KiwiFarmsGreenColor, BuiltIn.Keys.KiwiFarmsRedColor
|
||||
]).Result;
|
||||
_logger.Trace("Shuffle bet has arrived");
|
||||
if (bet.Username != settings[BuiltIn.Keys.ShuffleBmjUsername].Value)
|
||||
bool isDotUs;
|
||||
if (bet.Username == settings[BuiltIn.Keys.ShuffleBmjUsername].Value)
|
||||
{
|
||||
isDotUs = false;
|
||||
}
|
||||
else if (bet.Username == settings[BuiltIn.Keys.ShuffleDotUsBmjUsername].Value)
|
||||
{
|
||||
isDotUs = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
return;
|
||||
}
|
||||
_logger.Info("ALERT BMJ IS BETTING");
|
||||
_logger.Info($"ALERT BMJ IS BETTING: isDotUs => {isDotUs}");
|
||||
if (CheckBmjIsLive().Result) return;
|
||||
|
||||
var payoutColor = settings[BuiltIn.Keys.KiwiFarmsGreenColor].Value;
|
||||
if (float.Parse(bet.Payout) < float.Parse(bet.Amount)) payoutColor = settings[BuiltIn.Keys.KiwiFarmsRedColor].Value;
|
||||
var preamble = "🚨🚨 Shufflebros! 🚨🚨";
|
||||
if (isDotUs)
|
||||
{
|
||||
preamble = "🦅🦅 Shuffle US! 🦅🦅";
|
||||
}
|
||||
// There will be a check for live status but ignoring that while we deal with an emergency dice situation
|
||||
_chatBot.SendChatMessage($"🚨🚨 Shufflebros! 🚨🚨 {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);
|
||||
_chatBot.SendChatMessage($"{preamble} {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, string channelName, bool isLive)
|
||||
|
||||
224
KfChatDotNetBot/Services/ShuffleDotUs.cs
Normal file
224
KfChatDotNetBot/Services/ShuffleDotUs.cs
Normal file
@@ -0,0 +1,224 @@
|
||||
using System.Net;
|
||||
using System.Net.Http.Json;
|
||||
using System.Net.WebSockets;
|
||||
using System.Text.Json;
|
||||
using KfChatDotNetBot.Models;
|
||||
using NLog;
|
||||
using Websocket.Client;
|
||||
|
||||
namespace KfChatDotNetBot.Services;
|
||||
|
||||
public class ShuffleDotUs : IDisposable
|
||||
{
|
||||
private Logger _logger = LogManager.GetCurrentClassLogger();
|
||||
private WebsocketClient? _wsClient;
|
||||
private Uri _wsUri = new("wss://shuffle.us/main-api/bp-subscription/subscription/graphql");
|
||||
private int _reconnectTimeout = 60;
|
||||
private string? _proxy;
|
||||
public delegate void OnLatestBetUpdatedEventHandler(object sender, ShuffleLatestBetModel bet);
|
||||
public delegate void OnWsDisconnectionEventHandler(object sender, DisconnectionInfo e);
|
||||
public event OnLatestBetUpdatedEventHandler? OnLatestBetUpdated;
|
||||
public event OnWsDisconnectionEventHandler? OnWsDisconnection;
|
||||
private CancellationToken _cancellationToken;
|
||||
private CancellationTokenSource _pingCts = new();
|
||||
private Task _pingTask;
|
||||
|
||||
public ShuffleDotUs(string? proxy = null, CancellationToken cancellationToken = default)
|
||||
{
|
||||
_proxy = proxy;
|
||||
_cancellationToken = cancellationToken;
|
||||
// Moved it up here as I'm concerned about the possibility of reconnections creating multiple ping tasks
|
||||
_pingTask = PeriodicPing();
|
||||
}
|
||||
|
||||
public async Task StartWsClient()
|
||||
{
|
||||
_logger.Debug("StartWsClient() called, creating client");
|
||||
await CreateWsClient();
|
||||
}
|
||||
|
||||
private async Task CreateWsClient()
|
||||
{
|
||||
var factory = new Func<ClientWebSocket>(() =>
|
||||
{
|
||||
var clientWs = new ClientWebSocket();
|
||||
clientWs.Options.AddSubProtocol("graphql-transport-ws");
|
||||
clientWs.Options.SetRequestHeader("Origin", "https://shuffle.us");
|
||||
clientWs.Options.SetRequestHeader("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:126.0) Gecko/20100101 Firefox/126.0");
|
||||
if (_proxy == null) return clientWs;
|
||||
_logger.Debug($"Using proxy address {_proxy}");
|
||||
clientWs.Options.Proxy = new WebProxy(_proxy);
|
||||
return clientWs;
|
||||
});
|
||||
|
||||
var client = new WebsocketClient(_wsUri, factory)
|
||||
{
|
||||
ReconnectTimeout = TimeSpan.FromSeconds(_reconnectTimeout),
|
||||
IsReconnectionEnabled = false // Watchdog will self-destruct this instead
|
||||
};
|
||||
|
||||
client.ReconnectionHappened.Subscribe(WsReconnection);
|
||||
client.MessageReceived.Subscribe(WsMessageReceived);
|
||||
client.DisconnectionHappened.Subscribe(WsDisconnection);
|
||||
|
||||
_wsClient = client;
|
||||
|
||||
_logger.Debug("Websocket client has been built, about to start");
|
||||
await client.Start();
|
||||
_logger.Debug("Websocket client started!");
|
||||
}
|
||||
|
||||
public bool IsConnected()
|
||||
{
|
||||
return _wsClient is { IsRunning: true };
|
||||
}
|
||||
|
||||
private async Task PeriodicPing()
|
||||
{
|
||||
using var timer = new PeriodicTimer(TimeSpan.FromSeconds(10));
|
||||
while (await timer.WaitForNextTickAsync(_pingCts.Token))
|
||||
{
|
||||
if (_wsClient == null)
|
||||
{
|
||||
_logger.Debug("_wsClient doesn't exist yet, not going to try ping");
|
||||
continue;
|
||||
}
|
||||
_logger.Debug("Sending ping to Shuffle.us");
|
||||
_wsClient.Send("{\"type\":\"ping\"}");
|
||||
}
|
||||
}
|
||||
|
||||
private void WsDisconnection(DisconnectionInfo disconnectionInfo)
|
||||
{
|
||||
_logger.Error($"Client disconnected from Shuffle.us (or never successfully connected). Type is {disconnectionInfo.Type}");
|
||||
_logger.Error($"Close Status => {disconnectionInfo.CloseStatus}; Close Status Description => {disconnectionInfo.CloseStatusDescription}");
|
||||
_logger.Error(disconnectionInfo.Exception);
|
||||
OnWsDisconnection?.Invoke(this, disconnectionInfo);
|
||||
}
|
||||
|
||||
private void WsReconnection(ReconnectionInfo reconnectionInfo)
|
||||
{
|
||||
_logger.Error($"Websocket connection dropped and reconnected. Reconnection type is {reconnectionInfo.Type}");
|
||||
_logger.Info("Sending connection_init");
|
||||
var initPayload =
|
||||
"{\"type\":\"connection_init\",\"payload\":{\"x-correlation-id\":\"pdvlnd9tej-di27abvq19-1.30.2-1i0nef1m7-g::anon\",\"authorization\":\"\"}}";
|
||||
_logger.Debug(initPayload);
|
||||
_wsClient?.Send(initPayload);
|
||||
}
|
||||
|
||||
private void WsMessageReceived(ResponseMessage message)
|
||||
{
|
||||
if (message.Text == null)
|
||||
{
|
||||
_logger.Info("Shuffle.us sent a null message");
|
||||
return;
|
||||
}
|
||||
_logger.Trace($"Received event from Shuffle.us: {message.Text}");
|
||||
|
||||
try
|
||||
{
|
||||
var packet = JsonSerializer.Deserialize<JsonElement>(message.Text);
|
||||
var packetType = packet.GetProperty("type").GetString();
|
||||
|
||||
if (packetType == "connection_ack")
|
||||
{
|
||||
_logger.Debug("connection_ack packet, sending subscribe payload");
|
||||
_logger.Info("Sending subscription request");
|
||||
// we're super ghetto today
|
||||
var payload = "{\"id\":\"" + Guid.NewGuid() +
|
||||
"\",\"type\":\"subscribe\",\"payload\":{\"variables\":{},\"extensions\":{},\"operationName\":\"LatestBetUpdated\",\"query\":\"subscription LatestBetUpdated {\\n latestBetUpdated {\\n ...BetActivityFields\\n __typename\\n }\\n}\\n\\nfragment BetActivityFields on BetActivityPayload {\\n id\\n username\\n vipLevel\\n currency\\n amount\\n payout\\n multiplier\\n gameName\\n gameCategory\\n gameSlug\\n __typename\\n}\"}}";
|
||||
_logger.Debug(payload);
|
||||
_wsClient?.SendInstant(payload).Wait(_cancellationToken);
|
||||
return;
|
||||
}
|
||||
|
||||
if (packetType == "pong")
|
||||
{
|
||||
_logger.Info("Shuffle.us pong packet");
|
||||
return;
|
||||
}
|
||||
|
||||
// GAMBA
|
||||
if (packetType == "next")
|
||||
{
|
||||
var bet = packet.GetProperty("payload").GetProperty("data").GetProperty("latestBetUpdated")
|
||||
.Deserialize<ShuffleLatestBetModel>();
|
||||
if (bet == null)
|
||||
{
|
||||
_logger.Error("Caught a null before invoking bet event");
|
||||
throw new NullReferenceException("Caught a null before invoking bet event");
|
||||
}
|
||||
OnLatestBetUpdated?.Invoke(this, bet);
|
||||
return;
|
||||
}
|
||||
_logger.Info("Message from Shuffle was unhandled");
|
||||
_logger.Info(message.Text);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
_logger.Error("Failed to handle message from Shuffle");
|
||||
_logger.Error(e);
|
||||
_logger.Error("--- JSON Payload ---");
|
||||
_logger.Error(message.Text);
|
||||
_logger.Error("--- End of JSON Payload ---");
|
||||
}
|
||||
}
|
||||
|
||||
public async Task<ShuffleUserModel> GetShuffleUser(string username)
|
||||
{
|
||||
var graphQl =
|
||||
"query GetUserProfile($username: String!) {\n user(username: $username) {\n id\n username\n vipLevel\n createdAt\n avatar\n avatarBackground\n bets\n usdWagered\n __typename\n }\n}";
|
||||
_logger.Debug($"Grabbing details for Shuffle.us user {username}");
|
||||
var jsonBody = new Dictionary<string, object>
|
||||
{
|
||||
{ "operationName", "GetUserProfile" },
|
||||
{ "query", graphQl },
|
||||
{ "variables", new Dictionary<string, string> { { "username", username } } }
|
||||
};
|
||||
_logger.Debug("Created dictionary object for the JSON payload, should serialize to following value:");
|
||||
_logger.Debug(JsonSerializer.Serialize(jsonBody));
|
||||
var handler = new HttpClientHandler { AutomaticDecompression = DecompressionMethods.All };
|
||||
if (_proxy != null)
|
||||
{
|
||||
handler.UseProxy = false;
|
||||
handler.Proxy = new WebProxy(_proxy);
|
||||
_logger.Debug($"Configured to use proxy {_proxy}");
|
||||
}
|
||||
|
||||
using var client = new HttpClient(handler);
|
||||
client.DefaultRequestHeaders.Add("content-type", "application/json");
|
||||
var postBody = JsonContent.Create(jsonBody);
|
||||
var response = await client.PostAsync("https://shuffle.us/graphql", postBody, _cancellationToken);
|
||||
var responseContent = await response.Content.ReadFromJsonAsync<JsonElement>(cancellationToken: _cancellationToken);
|
||||
_logger.Debug("Shuffle.us returned following JSON");
|
||||
_logger.Debug(responseContent.GetRawText);
|
||||
var user = responseContent.GetProperty("data").GetProperty("user");
|
||||
if (user.ValueKind == JsonValueKind.Null)
|
||||
{
|
||||
_logger.Debug("data.user was null");
|
||||
throw new ShuffleUserNotFoundException();
|
||||
}
|
||||
|
||||
return user.Deserialize<ShuffleUserModel>() ?? throw new InvalidOperationException();
|
||||
}
|
||||
|
||||
public class ShuffleUserNotFoundException : Exception;
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_wsClient?.Dispose();
|
||||
// Rare bug but has happened at least once
|
||||
try
|
||||
{
|
||||
_pingCts.Cancel();
|
||||
}
|
||||
catch (ObjectDisposedException e)
|
||||
{
|
||||
_logger.Error("Caught object disposed exception when trying to send a cancellation to the ping task");
|
||||
_logger.Error(e);
|
||||
}
|
||||
_pingCts.Dispose();
|
||||
_pingTask.Dispose();
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user