github/Discord.Net
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bettah bulk bans (#2915)

Browse Source
This commit is contained in:
Mihail Gribkov
2024-04-23 00:13:31 +03:00
committed by GitHub
parent 27a0785653
commit f85bf9ac48
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/Discord.Net.Core/DiscordConfig.cs
View File

@@ -256,5 +256,10 @@ namespace Discord
/// Returns the maximum number of entitlements that can be gotten per-batch.
/// </summary>
public const int MaxEntitlementsPerBatch = 100;
/// <summary>
/// Returns the maximum number of bans that can be banned in a single bulk request.
/// </summary>
public const int MaxBansPerBulkBatch = 200;
}
}

17
src/Discord.Net.Rest/Entities/Guilds/GuildHelper.cs
View File

@@ -251,9 +251,20 @@ namespace Discord.Rest
public static async Task<BulkBanResult> BulkBanAsync(IGuild guild, BaseDiscordClient client, ulong[] userIds, int? deleteMessageSeconds, RequestOptions options)
{
var model = await client.ApiClient.BulkBanAsync(guild.Id, userIds, deleteMessageSeconds, options);
return new(model.BannedUsers?.ToImmutableArray() ?? ImmutableArray<ulong>.Empty,
model.FailedUsers?.ToImmutableArray() ?? ImmutableArray<ulong>.Empty);
var pos = 0;
var banned = new List<ulong>(userIds.Length);
var failed = new List<ulong>();
while (pos * DiscordConfig.MaxBansPerBulkBatch < userIds.Length)
{
var toBan = userIds
.Skip(pos * DiscordConfig.MaxBansPerBulkBatch)
.Take(DiscordConfig.MaxBansPerBulkBatch);
pos++;
var model = await client.ApiClient.BulkBanAsync(guild.Id, toBan.ToArray(), deleteMessageSeconds, options);
banned.AddRange(model.BannedUsers ?? []);
failed.AddRange(model.FailedUsers ?? []);
}
return new(banned.ToImmutableArray(), failed.ToImmutableArray());
}
#endregion