eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-04 13:06:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix(bot): try fixing unhandled exceptions from interactive prompts

Browse source
This commit is contained in:
alyssa 2024-10-19 07:30:03 +09:00
parent d997f725e2
commit c639d877f8
2 changed files with 17 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
PluralKit.Bot/Interactive/YesNoPrompt.cs
View file

@ -84,20 +84,32 @@ public class YesNoPrompt: BaseInteractive
var queue = _ctx.Services.Resolve<HandlerQueue<MessageCreateEvent>>();
var messageDispatch = queue.WaitFor(MessagePredicate, Timeout, cts.Token);
async Task WaitForMessage()
{
try
{
await queue.WaitFor(MessagePredicate, Timeout, cts.Token);
}
catch (TimeoutException e)
{
if (e.Message != "HandlerQueue#WaitFor timed out")
throw;
}
}
await Start();
cts.Token.Register(() => _tcs.TrySetException(new TimeoutException("Action timed out")));
var messageDispatch = WaitForMessage();
cts.Token.Register(() => _tcs.TrySetException(new TimeoutException("YesNoPrompt timed out")));
try
{
var doneTask = await Task.WhenAny(_tcs.Task, messageDispatch);
if (doneTask == messageDispatch)
await Finish();
}
finally
{
await Finish();
Cleanup();
}
}