elliebot/src/EllieBot/_common/Abstractions/Extensions/SelfAssignableRolesExtensions.cs

using Microsoft.EntityFrameworkCore;
using EllieBot.Db.Models;

namespace EllieBot.Db;

public static class SelfAssignableRolesExtensions
{
    public static bool DeleteByGuildAndRoleId(this DbSet<SelfAssignedRole> roles, ulong guildId, ulong roleId)
    {
        var role = roles.FirstOrDefault(s => s.GuildId == guildId && s.RoleId == roleId);

        if (role is null)
            return false;

        roles.Remove(role);
        return true;
    }

    public static IReadOnlyCollection<SelfAssignedRole> GetFromGuild(
        this DbSet<SelfAssignedRole> roles,
        ulong guildId)
        => roles.AsQueryable().Where(s => s.GuildId == guildId).ToArray();
}