1
0
Fork 1
mirror of https://github.com/TeamOctolings/Octobot.git synced 2025-01-31 17:19:00 +03:00

Some workarounds...

Signed-off-by: mctaylors <cantsendmails@mctaylors.ru>
This commit is contained in:
Macintxsh 2024-03-26 21:54:31 +03:00
parent 152682e456
commit 5239b82806
Signed by: mctaylors
GPG key ID: 7181BEBE676903C1
5 changed files with 89 additions and 71 deletions

View file

@ -793,6 +793,9 @@
<value>Warn threshold has been exceeded. ({0})</value> <value>Warn threshold has been exceeded. ({0})</value>
</data> </data>
<data name="WarnPunishmentDurationNotSet" xml:space="preserve"> <data name="WarnPunishmentDurationNotSet" xml:space="preserve">
<value>The WarnPunishmentDuration setting is not set for the current punishment type.</value> <value>Warn Punishment Duration is not set for the current punishment type.</value>
</data>
<data name="WarnPunishmentDisabled" xml:space="preserve">
<value>Warn Threshold is set, but the Warn Punishment is not.</value>
</data> </data>
</root> </root>

View file

@ -793,6 +793,9 @@
<value>Превышен порог предупреждений. ({0})</value> <value>Превышен порог предупреждений. ({0})</value>
</data> </data>
<data name="WarnPunishmentDurationNotSet" xml:space="preserve"> <data name="WarnPunishmentDurationNotSet" xml:space="preserve">
<value>Настройка WarnPunishmentDuration не установлена для текущего типа наказания.</value> <value>Длительность наказания предупреждения не установлена для текущего типа наказания.</value>
</data>
<data name="WarnPunishmentDisabled" xml:space="preserve">
<value>Порог предупреждения установлен, но наказание нет.</value>
</data> </data>
</root> </root>

View file

@ -793,6 +793,9 @@
<value>мы дошли до лимита варнов. ({0})</value> <value>мы дошли до лимита варнов. ({0})</value>
</data> </data>
<data name="WarnPunishmentDurationNotSet" xml:space="preserve"> <data name="WarnPunishmentDurationNotSet" xml:space="preserve">
<value>наказание что было установлено хочет установленного WarnPunishmentDuration</value> <value>наказание что было установлено хочет установленную ддлительность в настройках</value>
</data>
<data name="WarnPunishmentDisabled" xml:space="preserve">
<value>лимит стоит, а наказания нет.</value>
</data> </data>
</root> </root>

View file

@ -2,6 +2,7 @@
using System.ComponentModel; using System.ComponentModel;
using System.ComponentModel.DataAnnotations; using System.ComponentModel.DataAnnotations;
using System.Text; using System.Text;
using System.Text.Json.Nodes;
using JetBrains.Annotations; using JetBrains.Annotations;
using Octobot.Data; using Octobot.Data;
using Octobot.Extensions; using Octobot.Extensions;
@ -89,14 +90,8 @@ public class WarnCommandGroup : CommandGroup
var data = await _guildData.GetData(guild.ID, CancellationToken); var data = await _guildData.GetData(guild.ID, CancellationToken);
Messages.Culture = GuildSettings.Language.Get(data.Settings); Messages.Culture = GuildSettings.Language.Get(data.Settings);
return await WarnUserAsync(executor, target, reason, guild, data, channelId, bot, CancellationToken);
}
private async Task<Result> WarnUserAsync(IUser executor, IUser target, string reason, IGuild guild,
GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default)
{
var interactionResult var interactionResult
= await _access.CheckInteractionsAsync(guild.ID, executor.ID, target.ID, "Warn", ct); = await _access.CheckInteractionsAsync(guild.ID, executor.ID, target.ID, "Warn", CancellationToken);
if (!interactionResult.IsSuccess) if (!interactionResult.IsSuccess)
{ {
return ResultExtensions.FromError(interactionResult); return ResultExtensions.FromError(interactionResult);
@ -107,13 +102,41 @@ public class WarnCommandGroup : CommandGroup
var errorEmbed = new EmbedBuilder().WithSmallTitle(interactionResult.Entity, bot) var errorEmbed = new EmbedBuilder().WithSmallTitle(interactionResult.Entity, bot)
.WithColour(ColorsList.Red).Build(); .WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct); return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: CancellationToken);
} }
return await WarnPreparationAsync(executor, target, reason, guild, data, channelId, bot, CancellationToken);
}
private async Task<Result> WarnPreparationAsync(IUser executor, IUser target, string reason, IGuild guild,
GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default)
{
var memberData = data.GetOrCreateMemberData(target.ID); var memberData = data.GetOrCreateMemberData(target.ID);
var warns = memberData.Warns; var warns = memberData.Warns;
var warnThreshold = GuildSettings.WarnsThreshold.Get(data.Settings); var settings = data.Settings;
var warnThreshold = GuildSettings.WarnsThreshold.Get(settings);
var warnPunishment = GuildSettings.WarnPunishment.Get(settings);
var warnDuration = GuildSettings.WarnPunishmentDuration.Get(settings);
if (warnPunishment is "off" or "disable" or "disabled" && warns.Count - 1 >= warnThreshold)
{
var errorEmbed = new EmbedBuilder()
.WithSmallTitle(Messages.WarnPunishmentDisabled, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: CancellationToken);
}
if (warnPunishment is "ban" or "mute" && warnDuration == TimeSpan.Zero)
{
var errorEmbed = new EmbedBuilder()
.WithSmallTitle(Messages.WarnPunishmentDurationNotSet, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: CancellationToken);
}
if (warns.Count >= warnThreshold && warnThreshold is not 0) if (warns.Count >= warnThreshold && warnThreshold is not 0)
{ {
@ -124,6 +147,14 @@ public class WarnCommandGroup : CommandGroup
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct); return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct);
} }
return await WarnUserAsync(executor, target, reason, guild, data, channelId, bot, settings,
warns, warnThreshold, warnPunishment, warnDuration, ct);
}
private async Task<Result> WarnUserAsync(IUser executor, IUser target, string reason, IGuild guild,
GuildData data, Snowflake channelId, IUser bot, JsonNode settings, IList warns, int warnThreshold,
string warnPunishment, TimeSpan warnDuration, CancellationToken ct = default)
{
warns.Add(new Warn warns.Add(new Warn
{ {
WarnedBy = executor.ID.Value, WarnedBy = executor.ID.Value,
@ -153,66 +184,51 @@ public class WarnCommandGroup : CommandGroup
await _channelApi.CreateMessageWithEmbedResultAsync(dmChannel.ID, embedResult: dmEmbed, ct: ct); await _channelApi.CreateMessageWithEmbedResultAsync(dmChannel.ID, embedResult: dmEmbed, ct: ct);
} }
_utility.LogAction( _utility.LogAction(settings, channelId, executor, title, description,
data.Settings, channelId, executor, title, description, target, ColorsList.Yellow, false, ct); target, ColorsList.Yellow, false, ct);
if (warns.Count >= warnThreshold &&
GuildSettings.WarnPunishment.Get(data.Settings) is not "off" and not "disable" and not "disabled")
{
return await PunishUserAsync(target, guild, data, channelId, bot, warns, CancellationToken);
}
var embed = new EmbedBuilder().WithSmallTitle( var embed = new EmbedBuilder().WithSmallTitle(
title, target) title, target)
.WithColour(ColorsList.Green).Build(); .WithColour(ColorsList.Green).Build();
if (warns.Count >= warnThreshold)
{
return await PunishUserAsync(target, guild, data, channelId, bot, warns, warnPunishment, warnDuration, CancellationToken);
}
return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct); return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct);
} }
private async Task<Result> PunishUserAsync(IUser target, IGuild guild, private async Task<Result> PunishUserAsync(IUser target, IGuild guild, GuildData data,
GuildData data, Snowflake channelId, IUser bot, IList warns, CancellationToken ct) Snowflake channelId, IUser bot, IList warns, string punishment, TimeSpan duration, CancellationToken ct)
{ {
var settings = data.Settings; warns.Clear();
var warnPunishment = GuildSettings.WarnPunishment.Get(settings);
var duration = GuildSettings.WarnPunishmentDuration.Get(settings);
if (warnPunishment is "ban" && duration != TimeSpan.Zero) if (punishment is "ban" && duration != TimeSpan.Zero)
{ {
warns.Clear();
var banCommandGroup = new BanCommandGroup( var banCommandGroup = new BanCommandGroup(
_access, _channelApi, _context, _feedback, _guildApi, _guildData, _userApi, _utility); _access, _channelApi, _context, _feedback, _guildApi, _guildData, _userApi, _utility);
await banCommandGroup.BanUserAsync(bot, target, Messages.ReceivedTooManyWarnings, await banCommandGroup.BanUserAsync(bot, target, Messages.ReceivedTooManyWarnings,
duration, guild, data, channelId, bot, ct); duration, guild, data, channelId, bot, ct);
} }
if (warnPunishment is "kick") if (punishment is "kick")
{ {
warns.Clear();
var kickCommandGroup = new KickCommandGroup( var kickCommandGroup = new KickCommandGroup(
_access, _channelApi, _context, _feedback, _guildApi, _guildData, _userApi, _utility); _access, _channelApi, _context, _feedback, _guildApi, _guildData, _userApi, _utility);
await kickCommandGroup.KickUserAsync(bot, target, Messages.ReceivedTooManyWarnings, await kickCommandGroup.KickUserAsync(bot, target, Messages.ReceivedTooManyWarnings,
guild, channelId, data, bot, ct); guild, channelId, data, bot, ct);
} }
if (warnPunishment is "mute" && duration != TimeSpan.Zero) if (punishment is "mute" && duration != TimeSpan.Zero)
{ {
warns.Clear();
var muteCommandGroup = new MuteCommandGroup( var muteCommandGroup = new MuteCommandGroup(
_access, _context, _feedback, _guildApi, _guildData, _userApi, _utility); _access, _context, _feedback, _guildApi, _guildData, _userApi, _utility);
await muteCommandGroup.MuteUserAsync(bot, target, Messages.ReceivedTooManyWarnings, await muteCommandGroup.MuteUserAsync(bot, target, Messages.ReceivedTooManyWarnings,
duration, guild.ID, data, channelId, bot, ct); duration, guild.ID, data, channelId, bot, ct);
} }
if (warnPunishment is not ("ban" or "mute") || duration != TimeSpan.Zero) return Result.FromSuccess();
{
return Result.FromSuccess();
}
var errorEmbed = new EmbedBuilder()
.WithSmallTitle(Messages.WarnPunishmentDurationNotSet, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct);
} }
[Command("unwarn")] [Command("unwarn")]
@ -256,6 +272,21 @@ public class WarnCommandGroup : CommandGroup
var data = await _guildData.GetData(guild.ID, CancellationToken); var data = await _guildData.GetData(guild.ID, CancellationToken);
Messages.Culture = GuildSettings.Language.Get(data.Settings); Messages.Culture = GuildSettings.Language.Get(data.Settings);
var interactionResult
= await _access.CheckInteractionsAsync(guild.ID, executor.ID, target.ID, "Unwarn", CancellationToken);
if (!interactionResult.IsSuccess)
{
return ResultExtensions.FromError(interactionResult);
}
if (interactionResult.Entity is not null)
{
var errorEmbed = new EmbedBuilder().WithSmallTitle(interactionResult.Entity, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: CancellationToken);
}
if (number is not null) if (number is not null)
{ {
return await RemoveUserWarnAsync(executor, target, reason, number.Value, guild, data, channelId, bot, return await RemoveUserWarnAsync(executor, target, reason, number.Value, guild, data, channelId, bot,
@ -268,21 +299,6 @@ public class WarnCommandGroup : CommandGroup
private async Task<Result> RemoveUserWarnAsync(IUser executor, IUser target, string reason, int warnNumber, private async Task<Result> RemoveUserWarnAsync(IUser executor, IUser target, string reason, int warnNumber,
IGuild guild, GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default) IGuild guild, GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default)
{ {
var interactionResult
= await _access.CheckInteractionsAsync(guild.ID, executor.ID, target.ID, "Unwarn", ct);
if (!interactionResult.IsSuccess)
{
return ResultExtensions.FromError(interactionResult);
}
if (interactionResult.Entity is not null)
{
var errorEmbed = new EmbedBuilder().WithSmallTitle(interactionResult.Entity, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct);
}
var memberData = data.GetOrCreateMemberData(target.ID); var memberData = data.GetOrCreateMemberData(target.ID);
var warns = memberData.Warns; var warns = memberData.Warns;
@ -340,21 +356,6 @@ public class WarnCommandGroup : CommandGroup
private async Task<Result> RemoveUserWarnsAsync(IUser executor, IUser target, string reason, private async Task<Result> RemoveUserWarnsAsync(IUser executor, IUser target, string reason,
IGuild guild, GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default) IGuild guild, GuildData data, Snowflake channelId, IUser bot, CancellationToken ct = default)
{ {
var interactionResult
= await _access.CheckInteractionsAsync(guild.ID, executor.ID, target.ID, "Unwarn", ct);
if (!interactionResult.IsSuccess)
{
return ResultExtensions.FromError(interactionResult);
}
if (interactionResult.Entity is not null)
{
var errorEmbed = new EmbedBuilder().WithSmallTitle(interactionResult.Entity, bot)
.WithColour(ColorsList.Red).Build();
return await _feedback.SendContextualEmbedResultAsync(errorEmbed, ct: ct);
}
var memberData = data.GetOrCreateMemberData(target.ID); var memberData = data.GetOrCreateMemberData(target.ID);
var warns = memberData.Warns; var warns = memberData.Warns;

View file

@ -1336,5 +1336,13 @@ namespace Octobot {
return ResourceManager.GetString("WarnPunishmentDurationNotSet", resourceCulture); return ResourceManager.GetString("WarnPunishmentDurationNotSet", resourceCulture);
} }
} }
internal static string WarnPunishmentDisabled
{
get
{
return ResourceManager.GetString("WarnPunishmentDisabled", resourceCulture);
}
}
} }
} }