RPCS3/discord-bot
1
0
Fork 0
mirror of https://github.com/RPCS3/discord-bot.git synced 2024-11-26 19:50:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add a hacky way to limit math calculation time as there's no good api

Browse Source
This commit is contained in:
13xforever 2024-06-26 23:29:43 +05:00
parent 10a5a02d80
commit 5c892edd98
No known key found for this signature in database
GPG Key ID: 2B2A36B482FE70C5
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
CompatBot/Commands/BotMath.cs
View File

@ -1,5 +1,6 @@
using System;
using System.Globalization;
using System.Threading;
using System.Threading.Tasks;
using CompatBot.Commands.Attributes;
using DSharpPlus.CommandsNext;
@ -12,6 +13,11 @@ namespace CompatBot.Commands;
[Description("Math, here you go Juhn. Use `math help` for syntax help")]
internal sealed class BotMath : BaseCommandModuleCustom
{
static BotMath()
{
License.iConfirmNonCommercialUse("RPCS3");
}
[GroupCommand, Priority(9)]
public async Task Expression(CommandContext ctx, [RemainingText, Description("Math expression")] string expression)
{
@ -35,8 +41,23 @@ internal sealed class BotMath : BaseCommandModuleCustom
""";
try
{
mXparser.resetCancelCurrentCalculationFlag();
var expr = new Expression(expression);
result = expr.calculate().ToString(CultureInfo.InvariantCulture);
const int timeout = 1_000;
var cts = new CancellationTokenSource(timeout);
// ReSharper disable once MethodSupportsCancellation
var delayTask = Task.Delay(timeout);
var calcTask = Task.Run(() => expr.calculate().ToString(CultureInfo.InvariantCulture), cts.Token);
await Task.WhenAny(calcTask, delayTask).ConfigureAwait(false);
if (calcTask.IsCompletedSuccessfully)
{
result = await calcTask;
}
else
{
mXparser.cancelCurrentCalculation();
result = "Calculation took too much time and all operations were cancelled";
}
}
catch (Exception e)
{