103 lines
No EOL
3.8 KiB
C#
103 lines
No EOL
3.8 KiB
C#
using System.Diagnostics;
|
|
using Catalogger.Backend.Cache;
|
|
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.API.Gateway.Events;
|
|
using Remora.Discord.Extensions.Embeds;
|
|
using Remora.Discord.Gateway.Responders;
|
|
using Remora.Results;
|
|
|
|
namespace Catalogger.Backend.Bot.Responders;
|
|
|
|
public class GuildCreateResponder(
|
|
Config config,
|
|
ILogger logger,
|
|
DatabaseContext db,
|
|
GuildCacheService guildCache,
|
|
ChannelCacheService channelCache,
|
|
WebhookExecutorService webhookExecutor)
|
|
: IResponder<IGuildCreate>, IResponder<GuildDelete>
|
|
{
|
|
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 _))
|
|
{
|
|
_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);
|
|
}
|
|
else if (evt.Guild.TryPickT1(out var unavailableGuild, out _))
|
|
{
|
|
_logger.Verbose("Received guild create for unavailable guild {GuildId}", unavailableGuild.ID);
|
|
guildId = unavailableGuild.ID.ToUlong();
|
|
}
|
|
else throw new UnreachableException();
|
|
|
|
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)
|
|
await webhookExecutor.QueueLogAsync(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 async Task<Result> RespondAsync(GuildDelete evt, CancellationToken ct = default)
|
|
{
|
|
if (evt.IsUnavailable.OrDefault(false))
|
|
{
|
|
_logger.Debug("Guild {GuildId} became unavailable", evt.ID);
|
|
return Result.Success;
|
|
}
|
|
|
|
if (!guildCache.TryGet(evt.ID, out var guild))
|
|
{
|
|
_logger.Information("Left uncached guild {GuildId}", evt.ID);
|
|
return Result.Success;
|
|
}
|
|
|
|
_logger.Information("Left guild {GuildName} / {GuildId}", guild.Name, guild.ID);
|
|
|
|
if (config.Discord.GuildLogId != null)
|
|
await webhookExecutor.QueueLogAsync(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 Result.Success;
|
|
}
|
|
} |