92 lines
No EOL
3.7 KiB
C#
92 lines
No EOL
3.7 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.SendContextualAsync(
|
|
"There are no key roles to list. Add some with `/key-roles add`.");
|
|
|
|
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.SendContextualAsync($"{role.Name} is already a key role.");
|
|
|
|
guildConfig.KeyRoles.Add(role.ID.Value);
|
|
db.Update(guildConfig);
|
|
await db.SaveChangesAsync();
|
|
|
|
return await feedbackService.SendContextualAsync($"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.SendContextualAsync($"{role.Name} is already not a key role.");
|
|
|
|
guildConfig.KeyRoles.Remove(role.ID.Value);
|
|
db.Update(guildConfig);
|
|
await db.SaveChangesAsync();
|
|
|
|
return await feedbackService.SendContextualAsync($"Removed {role.Name} from this server's key roles!");
|
|
}
|
|
} |