Salastil/KfChatDotNet
1
0
Fork 0
mirror of https://github.com/barelyprofessional/KfChatDotNet.git synced 2026-05-04 13:22:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Updated session token retrieval system to use Puppeteer to automatically log in and retrieve a token

Browse Source
This commit is contained in:
barelyprofessional
2024-06-13 20:07:41 +08:00
parent 3ef573be5c
commit 37ab1138ef
6 changed files with 129 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.idea/.idea.KfChatDotNet/.idea/sqldialects.xml generated
View File

@@ -1,6 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="SqlDialectMappings">
<file url="file://$PROJECT_DIR$/KfChatDotNetKickBot/Helpers.cs" dialect="SQLite" />
</component>
</project>

36
KfChatDotNetKickBot/Helpers.cs
View File

@@ -1,36 +0,0 @@
using Microsoft.Data.Sqlite;
using NLog;
namespace KfChatDotNetKickBot;
public static class Helpers
{
// This ended up being pretty useless as it turns out Firefox doesn't store session cookies in cookies.sqlite
// But I'll leave it here in case it becomes useful one day
public static async Task<string?> GetXfToken(string cookieName, string cookieDomain, string containerPath)
{
var logger = LogManager.GetCurrentClassLogger();
await using var connection = new SqliteConnection($"Data Source={containerPath}");
await connection.OpenAsync();
logger.Debug($"Opened {containerPath}");
var command = connection.CreateCommand();
command.CommandText = "SELECT value FROM moz_cookies WHERE host = $host AND name = $name ORDER BY creationTime DESC LIMIT 1";
command.Parameters.AddWithValue("$host", cookieDomain);
command.Parameters.AddWithValue("$name", cookieName);
logger.Debug("Created command");
logger.Debug(command.CommandText);
await using var reader = await command.ExecuteReaderAsync();
while (await reader.ReadAsync())
{
logger.Debug("Reading first row, which will be immediately returned anyway");
return reader.GetString(0);
}
logger.Error("Fucked up while retrieving cookie. Cookie doesn't exist?");
return null;
}
}

9
KfChatDotNetKickBot/KfChatDotNetKickBot.csproj
View File

@@ -8,10 +8,11 @@
</PropertyGroup>
<ItemGroup>
<PackageReference Include="Microsoft.Data.Sqlite" Version="8.0.3" />
<PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
<PackageReference Include="NLog" Version="5.2.8" />
<PackageReference Include="NLog" Version="5.3.2" />
<PackageReference Include="PuppeteerSharp" Version="18.0.1" />
<PackageReference Include="Spectre.Console" Version="0.48.0" />
<PackageReference Include="System.Text.Json" Version="8.0.3" />
</ItemGroup>
<ItemGroup>
@@ -31,4 +32,8 @@
</Content>
</ItemGroup>
<ItemGroup>
<Folder Include="Commands\" />
</ItemGroup>
</Project>

54
KfChatDotNetKickBot/KickBot.cs
View File

@@ -1,4 +1,5 @@
using KfChatDotNetWsClient;
using KfChatDotNetKickBot.Services;
using KfChatDotNetWsClient;
using KfChatDotNetWsClient.Models;
using KfChatDotNetWsClient.Models.Events;
using KfChatDotNetWsClient.Models.Json;
@@ -19,6 +20,7 @@ public class KickBot
private Thread _pingThread;
private bool _pingEnabled = true;
private bool _gambaSeshPresent = false;
private string _xfSessionToken;
// Oh no it's an ever expanding list that may never get cleaned up!
// BUY MORE RAM
private List<int> _seenMsgIds = new List<int>();
@@ -37,11 +39,12 @@ public class KickBot
_config = JsonConvert.DeserializeObject<Models.ConfigModel>(File.ReadAllText(configPath)) ??
throw new InvalidOperationException();
RefreshXfToken();
_kfClient = new ChatClient(new ChatClientConfigModel
{
WsUri = _config.KfWsEndpoint,
XfSessionToken = GetXfToken(),
XfSessionToken = _xfSessionToken,
CookieDomain = _config.KfWsEndpoint.Host,
Proxy = _config.KfProxy,
ReconnectTimeout = _config.KfReconnectTimeout
@@ -55,6 +58,7 @@ public class KickBot
_kfClient.OnUsersJoined += OnUsersJoined;
_kfClient.OnWsDisconnection += OnKfWsDisconnected;
_kfClient.OnWsReconnect += OnKfWsReconnected;
_kfClient.OnFailedToJoinRoom += OnFailedToJoinRoom;
_kickClient.OnStreamerIsLive += OnStreamerIsLive;
_kickClient.OnChatMessage += OnKickChatMessage;
@@ -81,24 +85,36 @@ public class KickBot
}
}
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();
_kfClient.UpdateToken(_xfSessionToken);
_logger.Info("Retrieved fresh token. Reconnecting.");
_kfClient.Disconnect();
_kfClient.StartWsClient().Wait();
_logger.Info("Client should be reconnecting now");
}
private void PingThread()
{
while (_pingEnabled)
{
Thread.Sleep(TimeSpan.FromSeconds(15));
_logger.Debug("Pinging KF and Pusher");
AnsiConsole.MarkupLine("[yellow]Pinging KF and Pusher[/]");
_kfClient.SendMessage("/ping");
_kickClient.SendPusherPing();
if (_initialStartCooldown) _initialStartCooldown = false;
}
}
private string GetXfToken()
private void RefreshXfToken()
{
//return Helpers.GetXfToken("xf_session", _config.KfWsEndpoint.Host, _config.FirefoxCookieContainer).Result ??
// throw new InvalidOperationException();
return _config.XfTokenValue;
var cookie = KfTokenService.FetchSessionTokenAsync(_config.KfDomain, _config.KfUsername, _config.KfPassword,
_config.ChromiumPath, _config.KfProxy).Result;
_logger.Debug($"FetchSessionTokenAsync returned {cookie}");
_xfSessionToken = cookie;
}
private void OnStreamerIsLive(object sender, KickModels.StreamerIsLiveEventModel? e)
@@ -135,6 +151,10 @@ public class KickBot
{
_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("!help"))
{
_sendChatMessage("[img]https://i.postimg.cc/fTw6tGWZ/ineedmoneydumbfuck.png[/img]", true);
}
}
else
{
@@ -144,9 +164,9 @@ public class KickBot
}
}
private void _sendChatMessage(string message)
private void _sendChatMessage(string message, bool bypassSeshDetect = false)
{
if (_gambaSeshPresent && _config.EnableGambaSeshDetect)
if (_gambaSeshPresent && _config.EnableGambaSeshDetect && !bypassSeshDetect)
{
AnsiConsole.MarkupLine($"[red]Not sending message '{message.EscapeMarkup()}' as GambaSesh is present[/]");
return;
@@ -196,32 +216,28 @@ public class KickBot
private void OnKfWsDisconnected(object sender, DisconnectionInfo disconnectionInfo)
{
AnsiConsole.MarkupLine($"[red]Sneedchat disconnected due to {disconnectionInfo.Type}[/]");
AnsiConsole.MarkupLine("[yellow]Grabbing fresh token from browser[/]");
var token = GetXfToken();
AnsiConsole.MarkupLine($"[green]Obtained token = {token.EscapeMarkup()}[/]");
_kfClient.UpdateToken(token);
_logger.Error($"Sneedchat disconnected due to {disconnectionInfo.Type}");
}
private void OnKfWsReconnected(object sender, ReconnectionInfo reconnectionInfo)
{
AnsiConsole.MarkupLine($"[red]Sneedchat reconnected due to {reconnectionInfo.Type}[/]");
AnsiConsole.MarkupLine($"[green]Rejoining {_config.KfChatRoomId}[/]");
_logger.Error($"Sneedchat reconnected due to {reconnectionInfo.Type}");
_logger.Info($"Rejoining {_config.KfChatRoomId}");
_kfClient.JoinRoom(_config.KfChatRoomId);
}
private void OnPusherWsReconnected(object sender, ReconnectionInfo reconnectionInfo)
{
AnsiConsole.MarkupLine($"[red]Pusher reconnected due to {reconnectionInfo.Type}[/]");
_logger.Error($"Pusher reconnected due to {reconnectionInfo.Type}");
foreach (var channel in _config.PusherChannels)
{
AnsiConsole.MarkupLine($"[green]Rejoining {channel}[/]");
_logger.Info($"Rejoining {channel}");
_kickClient.SendPusherSubscribe(channel);
}
}
private void OnPusherSubscriptionSucceeded(object sender, PusherModels.BasePusherEventModel? e)
{
AnsiConsole.MarkupLine($"[green]Pusher indicates subscription to {e?.Channel.EscapeMarkup()} was successful[/]");
_logger.Info($"Pusher indicates subscription to {e?.Channel} was successful");
}
}

7
KfChatDotNetKickBot/Models.cs
View File

@@ -18,11 +18,12 @@ public class Models
public string? PusherProxy { get; set; }
public int KfReconnectTimeout { get; set; } = 30;
public int PusherReconnectTimeout { get; set; } = 30;
// Todo: Find a way to extract this from the browser as it's not valid forever
public string? XfTokenValue { get; set; }
// Because his shitty bot crashed and it's annoying not having notifications
public bool EnableGambaSeshDetect { get; set; } = true;
public int GambaSeshUserId { get; set; } = 168162;
public string KickIcon { get; set; } = "https://i.ibb.co/0cqwscx/kick.png";
public string KfDomain { get; set; } = "kiwifarms.st";
public required string KfUsername { get; set; }
public required string KfPassword { get; set; }
public string ChromiumPath { get; set; } = "chromium_install";
}
}

83
KfChatDotNetKickBot/Services/KfTokenService.cs Normal file
View File

@@ -0,0 +1,83 @@
using System.Net;
using System.Text.Json;
using NLog;
using PuppeteerSharp;
namespace KfChatDotNetKickBot.Services;
public class KfTokenService
{
// Shout out Gamba Sesh for open sourcing his token retriever which heavily inspired this implementation
public static async Task<string> FetchSessionTokenAsync(string domain, string username, string password, string browserPath, string? proxy = null)
{
var logger = LogManager.GetCurrentClassLogger();
var browserFetcher = new BrowserFetcher(new BrowserFetcherOptions
{ Browser = SupportedBrowser.Chromium, Path = browserPath });
if (proxy != null)
{
browserFetcher.WebProxy = new WebProxy(proxy);
logger.Debug($"Detected proxy settings for browser download: {proxy}");
}
var installedBrowser = await browserFetcher.DownloadAsync();
logger.Debug("Downloaded browser");
List<string> launchArgs = ["--no-sandbox"];
if (proxy != null)
{
logger.Debug($"Configuring Chromium to use proxy {proxy}");
launchArgs.Add($"--proxy-server=\"{proxy}\"");
}
var launchOptions = new LaunchOptions
{
Headless = false, ExecutablePath = installedBrowser.GetExecutablePath(), UserDataDir = "kf_profile",
Args = launchArgs.ToArray()
};
await using var browser = await Puppeteer.LaunchAsync(launchOptions);
await using var page = await browser.NewPageAsync();
await page.GoToAsync($"https://{domain}/login");
await page.WaitForSelectorAsync("img[alt=\"Kiwi Farms\"]");
if (await page.QuerySelectorAsync("html[data-template=\"login\"]") == null)
{
logger.Debug("Page template is not login. This is expected if we're already logged in. Retrieving cookies");
return await GetXfSessionCookie();
}
var usernameFieldSelector = await page.QuerySelectorAsync("input[autocomplete=\"username\"]");
var passwordFieldSelector = await page.QuerySelectorAsync("input[autocomplete=\"current-password\"]");
var loginButtonSelector = await page.QuerySelectorAsync("div[class=\"formSubmitRow-controls\"] > button[type=\"submit\"]");
if (usernameFieldSelector == null || passwordFieldSelector == null || loginButtonSelector == null)
{
// Realistically this shouldn't happen unless Null changes the login template in a big way
logger.Error("Username/password fields could not be found");
throw new MissingLoginElementsException();
}
await usernameFieldSelector.TypeAsync(username);
await passwordFieldSelector.TypeAsync(password);
await loginButtonSelector.ClickAsync();
logger.Debug("Login fields have been filled out and button clicked. Awaiting page navigation.");
await page.WaitForNavigationAsync();
logger.Debug("Navigation completed. Doing the cookie needful");
return await GetXfSessionCookie();
async Task<string> GetXfSessionCookie()
{
var cookies = await page.GetCookiesAsync();
var xfSession = cookies.FirstOrDefault(x => x.Name == "xf_session");
if (xfSession == null)
{
logger.Error("xf_session cookie not set. Cookie data follows");
logger.Error(JsonSerializer.Serialize(cookies));
throw new MissingSessionCookieException();
}
logger.Debug($"Returning xf_session value: {xfSession.Value}");
return xfSession.Value;
}
}
public class MissingSessionCookieException : Exception;
public class MissingLoginElementsException : Exception;
}