113 lines
4 KiB
C#
113 lines
4 KiB
C#
using System.ComponentModel;
|
|
using Catalogger.Backend.Cache.InMemoryCache;
|
|
using Catalogger.Backend.Database;
|
|
using Catalogger.Backend.Database.Queries;
|
|
using Catalogger.Backend.Extensions;
|
|
using Remora.Commands.Attributes;
|
|
using Remora.Commands.Groups;
|
|
using Remora.Discord.API.Abstractions.Objects;
|
|
using Remora.Discord.API.Objects;
|
|
using Remora.Discord.Commands.Attributes;
|
|
using Remora.Discord.Commands.Feedback.Services;
|
|
using Remora.Discord.Commands.Services;
|
|
using Remora.Rest.Core;
|
|
using IResult = Remora.Results.IResult;
|
|
|
|
namespace Catalogger.Backend.Bot.Commands;
|
|
|
|
[Group("key-roles")]
|
|
[DiscordDefaultMemberPermissions(DiscordPermission.ManageGuild)]
|
|
public class KeyRoleCommands(
|
|
DatabaseContext db,
|
|
ContextInjectionService contextInjection,
|
|
IFeedbackService feedbackService,
|
|
GuildCache guildCache,
|
|
RoleCache roleCache
|
|
) : CommandGroup
|
|
{
|
|
[Command("list")]
|
|
[Description("List this server's key roles.")]
|
|
public async Task<IResult> ListKeyRolesAsync()
|
|
{
|
|
var (_, guildId) = contextInjection.GetUserAndGuild();
|
|
if (!guildCache.TryGet(guildId, out var guild))
|
|
throw new CataloggerError("Guild not in cache");
|
|
var guildRoles = roleCache.GuildRoles(guildId).ToList();
|
|
var guildConfig = await db.GetGuildAsync(guildId);
|
|
|
|
if (guildConfig.KeyRoles.Count == 0)
|
|
return await feedbackService.ReplyAsync(
|
|
"There are no key roles to list. Add some with `/key-roles add`.",
|
|
isEphemeral: true
|
|
);
|
|
|
|
var description = string.Join(
|
|
"\n",
|
|
guildConfig.KeyRoles.Select(id =>
|
|
{
|
|
var role = guildRoles.FirstOrDefault(r => r.ID.Value == id);
|
|
return role != null ? $"- {role.Name} <@&{role.ID}>" : $"- unknown role {id}";
|
|
})
|
|
);
|
|
|
|
return await feedbackService.SendContextualEmbedAsync(
|
|
new Embed(
|
|
Title: $"Key roles for {guild.Name}",
|
|
Description: description,
|
|
Colour: DiscordUtils.Purple
|
|
)
|
|
);
|
|
}
|
|
|
|
[Command("add")]
|
|
[Description("Add a new key role.")]
|
|
public async Task<IResult> AddKeyRoleAsync(
|
|
[Description("The role to add.")] [DiscordTypeHint(TypeHint.Role)] Snowflake roleId
|
|
)
|
|
{
|
|
var (_, guildId) = contextInjection.GetUserAndGuild();
|
|
var role = roleCache.GuildRoles(guildId).FirstOrDefault(r => r.ID == roleId);
|
|
if (role == null)
|
|
throw new CataloggerError("Role is not cached");
|
|
|
|
var guildConfig = await db.GetGuildAsync(guildId);
|
|
if (guildConfig.KeyRoles.Any(id => role.ID == id))
|
|
return await feedbackService.ReplyAsync(
|
|
$"{role.Name} is already a key role.",
|
|
isEphemeral: true
|
|
);
|
|
|
|
guildConfig.KeyRoles.Add(role.ID.Value);
|
|
db.Update(guildConfig);
|
|
await db.SaveChangesAsync();
|
|
|
|
return await feedbackService.ReplyAsync($"Added {role.Name} to this server's key roles!");
|
|
}
|
|
|
|
[Command("remove")]
|
|
[Description("Remove a key role.")]
|
|
public async Task<IResult> RemoveKeyRoleAsync(
|
|
[Description("The role to remove.")] [DiscordTypeHint(TypeHint.Role)] Snowflake roleId
|
|
)
|
|
{
|
|
var (_, guildId) = contextInjection.GetUserAndGuild();
|
|
var role = roleCache.GuildRoles(guildId).FirstOrDefault(r => r.ID == roleId);
|
|
if (role == null)
|
|
throw new CataloggerError("Role is not cached");
|
|
|
|
var guildConfig = await db.GetGuildAsync(guildId);
|
|
if (guildConfig.KeyRoles.All(id => role.ID != id))
|
|
return await feedbackService.ReplyAsync(
|
|
$"{role.Name} is already not a key role.",
|
|
isEphemeral: true
|
|
);
|
|
|
|
guildConfig.KeyRoles.Remove(role.ID.Value);
|
|
db.Update(guildConfig);
|
|
await db.SaveChangesAsync();
|
|
|
|
return await feedbackService.ReplyAsync(
|
|
$"Removed {role.Name} from this server's key roles!"
|
|
);
|
|
}
|
|
}
|