mirror of
https://github.com/barelyprofessional/KfChatDotNet.git
synced 2026-05-02 04:22:04 -04:00
Renamed the bot from KickBot -> ChatBot and removed the reference to Kick in the project name
This commit is contained in:
barelyprofessional
257
KfChatDotNetBot/Services/Discord.cs
Normal file
257
KfChatDotNetBot/Services/Discord.cs
Normal file
@@ -0,0 +1,257 @@
|
||||
using System.Net;
|
||||
using System.Net.WebSockets;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Serialization;
|
||||
using NLog;
|
||||
using Websocket.Client;
|
||||
|
||||
namespace KfChatDotNetBot.Services;
|
||||
|
||||
public class DiscordService : IDisposable
|
||||
{
|
||||
private readonly Logger _logger = LogManager.GetCurrentClassLogger();
|
||||
private WebsocketClient _wsClient;
|
||||
private readonly Uri _wsUri = new Uri("wss://gateway.discord.gg/?encoding=json&v=9");
|
||||
// Not sure what a good value for this would be
|
||||
private const int ReconnectTimeout = 60;
|
||||
private readonly string? _proxy;
|
||||
private readonly string _authorization;
|
||||
private int _sequence;
|
||||
private const string UserAgent = "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/115.0";
|
||||
|
||||
public delegate void MessageReceivedEventHandler(object sender, DiscordMessageModel message);
|
||||
public delegate void PresenceUpdateEventHandler(object sender, DiscordPresenceUpdateModel presence);
|
||||
public delegate void WsDisconnectionEventHandler(object sender, DisconnectionInfo e);
|
||||
public delegate void InvalidCredentialsEventHandler(object sender, DiscordPacketReadModel packet);
|
||||
public event MessageReceivedEventHandler OnMessageReceived;
|
||||
public event PresenceUpdateEventHandler OnPresenceUpdated;
|
||||
public event WsDisconnectionEventHandler OnWsDisconnection;
|
||||
public event InvalidCredentialsEventHandler OnInvalidCredentials;
|
||||
|
||||
private readonly CancellationToken _cancellationToken = CancellationToken.None;
|
||||
private readonly CancellationTokenSource _pingCts = new();
|
||||
private Task? _heartbeatTask;
|
||||
// Discord tells us the heartbeat interval to use in the op 10 response so this is just a placeholder
|
||||
private TimeSpan _heartbeatInterval = TimeSpan.FromSeconds(40);
|
||||
|
||||
public DiscordService(string authorization, string? proxy = null, CancellationToken? cancellationToken = null)
|
||||
{
|
||||
_proxy = proxy;
|
||||
if (cancellationToken != null) _cancellationToken = cancellationToken.Value;
|
||||
_authorization = authorization;
|
||||
_logger.Info("Discord Service created");
|
||||
}
|
||||
|
||||
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.SetRequestHeader("Origin", "https://discord.com");
|
||||
clientWs.Options.SetRequestHeader("User-Agent", UserAgent);
|
||||
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
|
||||
};
|
||||
|
||||
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 HeartbeatTimer()
|
||||
{
|
||||
using var timer = new PeriodicTimer(_heartbeatInterval);
|
||||
while (await timer.WaitForNextTickAsync(_pingCts.Token))
|
||||
{
|
||||
if (_wsClient == null)
|
||||
{
|
||||
_logger.Debug("_wsClient doesn't exist yet, not going to try ping");
|
||||
continue;
|
||||
}
|
||||
var heartbeatPacket = JsonSerializer.Serialize(new DiscordPacketWriteModel
|
||||
{
|
||||
OpCode = 1,
|
||||
Sequence = _sequence,
|
||||
});
|
||||
_logger.Debug("Sending heartbeat packet");
|
||||
_logger.Debug(heartbeatPacket);
|
||||
await _wsClient.SendInstant(heartbeatPacket);
|
||||
}
|
||||
}
|
||||
|
||||
private void WsDisconnection(DisconnectionInfo disconnectionInfo)
|
||||
{
|
||||
_logger.Error($"Client disconnected from Discord (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}");
|
||||
}
|
||||
|
||||
private void WsMessageReceived(ResponseMessage message)
|
||||
{
|
||||
if (message.Text == null)
|
||||
{
|
||||
_logger.Info("Discord sent a null message");
|
||||
return;
|
||||
}
|
||||
|
||||
_logger.Debug($"Received event from Discord: {message.Text}");
|
||||
|
||||
try
|
||||
{
|
||||
var packet = JsonSerializer.Deserialize<DiscordPacketReadModel>(message.Text);
|
||||
if (packet == null) throw new InvalidOperationException("Caught a null when deserializing Discord packet");
|
||||
_sequence = packet.Sequence ?? _sequence;
|
||||
if (packet.OpCode == 9)
|
||||
{
|
||||
_logger.Info("Discord sent op code indicating invalid credentials. Raising event.");
|
||||
OnInvalidCredentials?.Invoke(this, packet);
|
||||
return;
|
||||
}
|
||||
|
||||
if (packet.OpCode == 10)
|
||||
{
|
||||
_logger.Info("Discord op code 10 (hello) received. Setting up heartbeat timer and sending init");
|
||||
_logger.Info("Sending connection_init");
|
||||
var initPayload =
|
||||
"{\"op\":2,\"d\":{\"token\":\"" + _authorization + "\",\"capabilities\":30717,\"properties\":" +
|
||||
"{\"os\":\"Linux\",\"browser\":\"Firefox\",\"device\":\"\",\"system_locale\":\"en-US\",\"browser_user_agent\":" +
|
||||
"\"" + UserAgent + "\",\"browser_version\":\"115.0\",\"os_version\":\"\",\"referrer\":\"\",\"referring_domain\":\"\"," +
|
||||
"\"referrer_current\":\"\",\"referring_domain_current\":\"\",\"release_channel\":\"stable\"," +
|
||||
"\"client_build_number\":306208,\"client_event_source\":null,\"design_id\":0},\"presence\":{\"status\":\"unknown\"," +
|
||||
"\"since\":0,\"activities\":[],\"afk\":false},\"compress\":false,\"client_state\":{\"guild_versions\":{}}}}";
|
||||
_logger.Debug(initPayload);
|
||||
_wsClient.SendInstant(initPayload).Wait(_cancellationToken);
|
||||
_heartbeatInterval =
|
||||
TimeSpan.FromMilliseconds(packet.Data.GetProperty("heartbeat_interval").GetInt32());
|
||||
if (_heartbeatTask != null) return;
|
||||
_heartbeatTask = Task.Run(HeartbeatTimer, _cancellationToken);
|
||||
return;
|
||||
}
|
||||
|
||||
if (packet.OpCode == 11)
|
||||
{
|
||||
_logger.Info("Received heartbeat ack from Discord");
|
||||
return;
|
||||
}
|
||||
|
||||
if (packet.OpCode != 0)
|
||||
{
|
||||
_logger.Info($"Op code {packet.OpCode} was unhandled. JSON follows");
|
||||
_logger.Info(message.Text);
|
||||
return;
|
||||
}
|
||||
|
||||
switch (packet.DispatchEvent)
|
||||
{
|
||||
case null:
|
||||
throw new InvalidOperationException("t was null");
|
||||
case "READY":
|
||||
_logger.Info("Discord is now ready");
|
||||
return;
|
||||
case "PRESENCE_UPDATE":
|
||||
OnPresenceUpdated?.Invoke(this,
|
||||
packet.Data.Deserialize<DiscordPresenceUpdateModel>() ?? throw new InvalidOperationException());
|
||||
return;
|
||||
case "MESSAGE_CREATE":
|
||||
OnMessageReceived?.Invoke(this,
|
||||
packet.Data.Deserialize<DiscordMessageModel>() ?? throw new InvalidOperationException());
|
||||
return;
|
||||
default:
|
||||
_logger.Debug($"{packet.DispatchEvent} was unhandled. JSON follows");
|
||||
_logger.Debug(message.Text);
|
||||
break;
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
_logger.Error("Failed to handle message from Discord");
|
||||
_logger.Error(e);
|
||||
_logger.Error("--- JSON Payload ---");
|
||||
_logger.Error(message.Text);
|
||||
_logger.Error("--- End of JSON Payload ---");
|
||||
}
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_logger.Info("Disposing of the Discord service");
|
||||
_wsClient.Dispose();
|
||||
_pingCts.Cancel();
|
||||
_pingCts.Dispose();
|
||||
_heartbeatTask?.Dispose();
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
}
|
||||
|
||||
public class DiscordPacketModel<T>
|
||||
{
|
||||
[JsonPropertyName("t")]
|
||||
public string? DispatchEvent { get; set; }
|
||||
[JsonPropertyName("op")]
|
||||
public int OpCode { get; set; }
|
||||
[JsonPropertyName("d")]
|
||||
public T? Data { get; set; }
|
||||
[JsonPropertyName("s")]
|
||||
public int? Sequence { get; set; }
|
||||
}
|
||||
|
||||
public class DiscordPacketWriteModel : DiscordPacketModel<object>;
|
||||
public class DiscordPacketReadModel : DiscordPacketModel<JsonElement>;
|
||||
|
||||
public class DiscordPresenceUpdateModel
|
||||
{
|
||||
[JsonPropertyName("user")]
|
||||
public required DiscordUserModel User { get; set; }
|
||||
[JsonPropertyName("status")]
|
||||
public required string Status { get; set; }
|
||||
[JsonPropertyName("client_status")]
|
||||
public required Dictionary<string, string> ClientStatus { get; set; }
|
||||
}
|
||||
|
||||
public class DiscordUserModel
|
||||
{
|
||||
[JsonPropertyName("id")]
|
||||
public required string Id { get; set; }
|
||||
[JsonPropertyName("username")]
|
||||
public string? Username { get; set; }
|
||||
}
|
||||
|
||||
public class DiscordMessageModel
|
||||
{
|
||||
[JsonPropertyName("content")]
|
||||
public string? Content { get; set; }
|
||||
[JsonPropertyName("author")]
|
||||
public required DiscordUserModel Author { get; set; }
|
||||
[JsonPropertyName("attachments")]
|
||||
public JsonElement[]? Attachments { get; set; }
|
||||
}
|
||||
Reference in New Issue
Block a user