121 lines
4.2 KiB
C#
121 lines
4.2 KiB
C#
using Catalogger.Backend.Cache;
|
|
using Catalogger.Backend.Cache.InMemoryCache;
|
|
using Catalogger.Backend.Database;
|
|
using Catalogger.Backend.Database.Models;
|
|
using Catalogger.Backend.Extensions;
|
|
using Catalogger.Backend.Services;
|
|
using Remora.Discord.API.Abstractions.Gateway.Events;
|
|
using Remora.Discord.Extensions.Embeds;
|
|
using Remora.Discord.Gateway.Responders;
|
|
using Remora.Results;
|
|
|
|
namespace Catalogger.Backend.Bot.Responders.Guilds;
|
|
|
|
public class GuildCreateResponder(
|
|
Config config,
|
|
ILogger logger,
|
|
DatabaseContext db,
|
|
GuildCache guildCache,
|
|
RoleCache roleCache,
|
|
ChannelCache channelCache,
|
|
WebhookExecutorService webhookExecutor,
|
|
IMemberCache memberCache,
|
|
GuildFetchService guildFetchService
|
|
) : IResponder<IGuildCreate>, IResponder<IGuildDelete>
|
|
{
|
|
private readonly ILogger _logger = logger.ForContext<GuildCreateResponder>();
|
|
|
|
public async Task<Result> RespondAsync(IGuildCreate evt, CancellationToken ct = default)
|
|
{
|
|
ulong guildId;
|
|
string? guildName = null;
|
|
if (evt.Guild.TryPickT0(out var guild, out var unavailableGuild))
|
|
{
|
|
_logger.Verbose(
|
|
"Received guild create for available guild {GuildName} / {GuildId})",
|
|
guild.Name,
|
|
guild.ID
|
|
);
|
|
guildId = guild.ID.ToUlong();
|
|
guildName = guild.Name;
|
|
|
|
guildCache.Set(guild);
|
|
foreach (var c in guild.Channels)
|
|
channelCache.Set(c, guild.ID);
|
|
foreach (var r in guild.Roles)
|
|
roleCache.Set(r, guild.ID);
|
|
|
|
if (!await memberCache.IsGuildCachedAsync(guild.ID))
|
|
guildFetchService.EnqueueGuild(guild.ID);
|
|
}
|
|
else
|
|
{
|
|
_logger.Verbose(
|
|
"Received guild create for unavailable guild {GuildId}",
|
|
unavailableGuild.ID
|
|
);
|
|
guildId = unavailableGuild.ID.ToUlong();
|
|
}
|
|
|
|
var tx = await db.Database.BeginTransactionAsync(ct);
|
|
if (await db.Guilds.FindAsync([guildId], ct) != null)
|
|
return Result.Success;
|
|
|
|
db.Add(new Guild { Id = guildId });
|
|
await db.SaveChangesAsync(ct);
|
|
await tx.CommitAsync(ct);
|
|
|
|
_logger.Information("Joined new guild {GuildName} / {GuildId}", guildName, guildId);
|
|
|
|
if (config.Discord.GuildLogId != null && evt.Guild.IsT0)
|
|
webhookExecutor.QueueLog(
|
|
config.Discord.GuildLogId.Value,
|
|
new EmbedBuilder()
|
|
.WithTitle("Joined new guild")
|
|
.WithDescription($"Joined new guild **{guild.Name}**")
|
|
.WithFooter($"ID: {guild.ID}")
|
|
.WithCurrentTimestamp()
|
|
#pragma warning disable CS8604 // Possible null reference argument.
|
|
.WithThumbnailUrl(guild.IconUrl())
|
|
#pragma warning restore CS8604 // Possible null reference argument.
|
|
.Build()
|
|
.GetOrThrow()
|
|
);
|
|
|
|
return Result.Success;
|
|
}
|
|
|
|
public Task<Result> RespondAsync(IGuildDelete evt, CancellationToken ct = default)
|
|
{
|
|
if (evt.IsUnavailable.OrDefault(false))
|
|
{
|
|
_logger.Debug("Guild {GuildId} became unavailable", evt.ID);
|
|
return Task.FromResult(Result.Success);
|
|
}
|
|
|
|
if (!guildCache.TryGet(evt.ID, out var guild))
|
|
{
|
|
_logger.Information("Left uncached guild {GuildId}", evt.ID);
|
|
return Task.FromResult(Result.Success);
|
|
}
|
|
|
|
_logger.Information("Left guild {GuildName} / {GuildId}", guild.Name, guild.ID);
|
|
|
|
if (config.Discord.GuildLogId != null)
|
|
webhookExecutor.QueueLog(
|
|
config.Discord.GuildLogId.Value,
|
|
new EmbedBuilder()
|
|
.WithTitle("Left guild")
|
|
.WithDescription($"Left guild **{guild.Name}**")
|
|
.WithFooter($"ID: {guild.ID}")
|
|
.WithCurrentTimestamp()
|
|
#pragma warning disable CS8604 // Possible null reference argument.
|
|
.WithThumbnailUrl(guild.IconUrl())
|
|
#pragma warning restore CS8604 // Possible null reference argument.
|
|
.Build()
|
|
.GetOrThrow()
|
|
);
|
|
|
|
return Task.FromResult(Result.Success);
|
|
}
|
|
}
|