mirror of
https://github.com/swc-project/swc.git
synced 2024-12-21 04:32:01 +03:00
325 lines
13 KiB
TypeScript
325 lines
13 KiB
TypeScript
|
// Loaded from https://deno.land/x/discordeno@11.0.0-rc.2/src/util/permissions.ts
|
||
|
|
||
|
|
||
|
import { botId } from "../bot.ts";
|
||
|
import { cacheHandlers } from "../cache.ts";
|
||
|
import { DiscordenoChannel } from "../structures/channel.ts";
|
||
|
import { DiscordenoGuild } from "../structures/guild.ts";
|
||
|
import { DiscordenoMember } from "../structures/member.ts";
|
||
|
import { DiscordenoRole } from "../structures/role.ts";
|
||
|
import { Overwrite } from "../types/channels/overwrite.ts";
|
||
|
import { Errors } from "../types/discordeno/errors.ts";
|
||
|
import { DiscordBitwisePermissionFlags } from "../types/permissions/bitwise_permission_flags.ts";
|
||
|
import type { PermissionStrings } from "../types/permissions/permission_strings.ts";
|
||
|
|
||
|
async function getCached(table: "guilds", key: bigint | DiscordenoGuild): Promise<DiscordenoGuild | undefined>;
|
||
|
async function getCached(table: "channels", key: bigint | DiscordenoChannel): Promise<DiscordenoChannel | undefined>;
|
||
|
async function getCached(table: "members", key: bigint | DiscordenoMember): Promise<DiscordenoMember | undefined>;
|
||
|
async function getCached(
|
||
|
table: "guilds" | "channels" | "members",
|
||
|
key: bigint | DiscordenoGuild | DiscordenoChannel | DiscordenoMember
|
||
|
) {
|
||
|
const cached =
|
||
|
typeof key === "bigint"
|
||
|
? // @ts-ignore TS is wrong here
|
||
|
await cacheHandlers.get(table, key)
|
||
|
: key;
|
||
|
|
||
|
return typeof cached === "bigint" ? undefined : cached;
|
||
|
}
|
||
|
|
||
|
/** Calculates the permissions this member has in the given guild */
|
||
|
export async function calculateBasePermissions(
|
||
|
guildOrId: bigint | DiscordenoGuild,
|
||
|
memberOrId: bigint | DiscordenoMember
|
||
|
) {
|
||
|
const guild = await getCached("guilds", guildOrId);
|
||
|
const member = await getCached("members", memberOrId);
|
||
|
|
||
|
if (!guild || !member) return "8";
|
||
|
|
||
|
let permissions = 0n;
|
||
|
// Calculate the role permissions bits, @everyone role is not in memberRoleIds so we need to pass guildId manualy
|
||
|
permissions |= [...(member.guilds.get(guild.id)?.roles || []), guild.id]
|
||
|
.map((id) => guild.roles.get(id)?.permissions)
|
||
|
// Removes any edge case undefined
|
||
|
.filter((perm) => perm)
|
||
|
.reduce((bits, perms) => {
|
||
|
bits |= BigInt(perms);
|
||
|
return bits;
|
||
|
}, 0n);
|
||
|
|
||
|
// If the memberId is equal to the guild ownerId he automatically has every permission so we add ADMINISTRATOR permission
|
||
|
if (guild.ownerId === member.id) permissions |= 8n;
|
||
|
// Return the members permission bits as a string
|
||
|
return permissions.toString();
|
||
|
}
|
||
|
|
||
|
/** Calculates the permissions this member has for the given Channel */
|
||
|
export async function calculateChannelOverwrites(
|
||
|
channelOrId: bigint | DiscordenoChannel,
|
||
|
memberOrId: bigint | DiscordenoMember
|
||
|
) {
|
||
|
const channel = await getCached("channels", channelOrId);
|
||
|
|
||
|
// This is a DM channel so return ADMINISTRATOR permission
|
||
|
if (!channel?.guildId) return "8";
|
||
|
|
||
|
const member = await getCached("members", memberOrId);
|
||
|
|
||
|
if (!channel || !member) return "8";
|
||
|
|
||
|
// Get all the role permissions this member already has
|
||
|
let permissions = BigInt(await calculateBasePermissions(channel.guildId, member));
|
||
|
|
||
|
// First calculate @everyone overwrites since these have the lowest priority
|
||
|
const overwriteEveryone = channel.permissionOverwrites?.find((overwrite) => overwrite.id === channel.guildId);
|
||
|
if (overwriteEveryone) {
|
||
|
// First remove denied permissions since denied < allowed
|
||
|
permissions &= ~BigInt(overwriteEveryone.deny);
|
||
|
permissions |= BigInt(overwriteEveryone.allow);
|
||
|
}
|
||
|
|
||
|
const overwrites = channel.permissionOverwrites;
|
||
|
|
||
|
// In order to calculate the role permissions correctly we need to temporarily save the allowed and denied permissions
|
||
|
let allow = 0n;
|
||
|
let deny = 0n;
|
||
|
const memberRoles = member.guilds.get(channel.guildId)?.roles || [];
|
||
|
// Second calculate members role overwrites since these have middle priority
|
||
|
for (const overwrite of overwrites || []) {
|
||
|
if (!memberRoles.includes(overwrite.id)) continue;
|
||
|
|
||
|
deny |= BigInt(overwrite.deny);
|
||
|
allow |= BigInt(overwrite.allow);
|
||
|
}
|
||
|
// After role overwrite calculate save allowed permissions first we remove denied permissions since "denied < allowed"
|
||
|
permissions &= ~deny;
|
||
|
permissions |= allow;
|
||
|
|
||
|
// Third calculate member specific overwrites since these have the highest priority
|
||
|
const overwriteMember = overwrites?.find((overwrite) => overwrite.id === member.id);
|
||
|
if (overwriteMember) {
|
||
|
permissions &= ~BigInt(overwriteMember.deny);
|
||
|
permissions |= BigInt(overwriteMember.allow);
|
||
|
}
|
||
|
|
||
|
return permissions.toString();
|
||
|
}
|
||
|
|
||
|
/** Checks if the given permission bits are matching the given permissions. `ADMINISTRATOR` always returns `true` */
|
||
|
export function validatePermissions(permissionBits: string, permissions: PermissionStrings[]) {
|
||
|
if (BigInt(permissionBits) & 8n) return true;
|
||
|
|
||
|
return permissions.every(
|
||
|
(permission) =>
|
||
|
// Check if permission is in permissionBits
|
||
|
BigInt(permissionBits) & BigInt(DiscordBitwisePermissionFlags[permission])
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/** Checks if the given member has these permissions in the given guild */
|
||
|
export async function hasGuildPermissions(
|
||
|
guild: bigint | DiscordenoGuild,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
// First we need the role permission bits this member has
|
||
|
const basePermissions = await calculateBasePermissions(guild, member);
|
||
|
// Second use the validatePermissions function to check if the member has every permission
|
||
|
return validatePermissions(basePermissions, permissions);
|
||
|
}
|
||
|
|
||
|
/** Checks if the bot has these permissions in the given guild */
|
||
|
export function botHasGuildPermissions(guild: bigint | DiscordenoGuild, permissions: PermissionStrings[]) {
|
||
|
// Since Bot is a normal member we can use the hasRolePermissions() function
|
||
|
return hasGuildPermissions(guild, botId, permissions);
|
||
|
}
|
||
|
|
||
|
/** Checks if the given member has these permissions for the given channel */
|
||
|
export async function hasChannelPermissions(
|
||
|
channel: bigint | DiscordenoChannel,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
// First we need the overwrite bits this member has
|
||
|
const channelOverwrites = await calculateChannelOverwrites(channel, member);
|
||
|
// Second use the validatePermissions function to check if the member has every permission
|
||
|
return validatePermissions(channelOverwrites, permissions);
|
||
|
}
|
||
|
|
||
|
/** Checks if the bot has these permissions f0r the given channel */
|
||
|
export function botHasChannelPermissions(channel: bigint | DiscordenoChannel, permissions: PermissionStrings[]) {
|
||
|
// Since Bot is a normal member we can use the hasRolePermissions() function
|
||
|
return hasChannelPermissions(channel, botId, permissions);
|
||
|
}
|
||
|
|
||
|
/** Returns the permissions that are not in the given permissionBits */
|
||
|
export function missingPermissions(permissionBits: string, permissions: PermissionStrings[]) {
|
||
|
if (BigInt(permissionBits) & 8n) return [];
|
||
|
|
||
|
return permissions.filter(
|
||
|
(permission) => !(BigInt(permissionBits) & BigInt(DiscordBitwisePermissionFlags[permission]))
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/** Get the missing Guild permissions this member has */
|
||
|
export async function getMissingGuildPermissions(
|
||
|
guild: bigint | DiscordenoGuild,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
// First we need the role permission bits this member has
|
||
|
const permissionBits = await calculateBasePermissions(guild, member);
|
||
|
// Second return the members missing permissions
|
||
|
return missingPermissions(permissionBits, permissions);
|
||
|
}
|
||
|
|
||
|
/** Get the missing Channel permissions this member has */
|
||
|
export async function getMissingChannelPermissions(
|
||
|
channel: bigint | DiscordenoChannel,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
// First we need the role permissino bits this member has
|
||
|
const permissionBits = await calculateChannelOverwrites(channel, member);
|
||
|
// Second returnn the members missing permissions
|
||
|
return missingPermissions(permissionBits, permissions);
|
||
|
}
|
||
|
|
||
|
/** Throws an error if this member has not all of the given permissions */
|
||
|
export async function requireGuildPermissions(
|
||
|
guild: bigint | DiscordenoGuild,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
const missing = await getMissingGuildPermissions(guild, member, permissions);
|
||
|
if (missing.length) {
|
||
|
// If the member is missing a permission throw an Error
|
||
|
throw new Error(`Missing Permissions: ${missing.join(" & ")}`);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/** Throws an error if the bot does not have all permissions */
|
||
|
export function requireBotGuildPermissions(guild: bigint | DiscordenoGuild, permissions: PermissionStrings[]) {
|
||
|
// Since Bot is a normal member we can use the throwOnMissingGuildPermission() function
|
||
|
return requireGuildPermissions(guild, botId, permissions);
|
||
|
}
|
||
|
|
||
|
/** Throws an error if this member has not all of the given permissions */
|
||
|
export async function requireChannelPermissions(
|
||
|
channel: bigint | DiscordenoChannel,
|
||
|
member: bigint | DiscordenoMember,
|
||
|
permissions: PermissionStrings[]
|
||
|
) {
|
||
|
const missing = await getMissingChannelPermissions(channel, member, permissions);
|
||
|
if (missing.length) {
|
||
|
// If the member is missing a permission throw an Error
|
||
|
throw new Error(`Missing Permissions: ${missing.join(" & ")}`);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/** Throws an error if the bot has not all of the given channel permissions */
|
||
|
export function requireBotChannelPermissions(channel: bigint | DiscordenoChannel, permissions: PermissionStrings[]) {
|
||
|
// Since Bot is a normal member we can use the throwOnMissingChannelPermission() function
|
||
|
return requireChannelPermissions(channel, botId, permissions);
|
||
|
}
|
||
|
|
||
|
/** This function converts a bitwise string to permission strings */
|
||
|
export function calculatePermissions(permissionBits: string) {
|
||
|
return Object.keys(DiscordBitwisePermissionFlags).filter((permission) => {
|
||
|
// Since Object.keys() not only returns the permission names but also the bit values we need to return false if it is a Number
|
||
|
if (Number(permission)) return false;
|
||
|
// Check if permissionBits has this permission
|
||
|
return BigInt(permissionBits) & BigInt(DiscordBitwisePermissionFlags[permission as PermissionStrings]);
|
||
|
}) as PermissionStrings[];
|
||
|
}
|
||
|
|
||
|
/** This function converts an array of permissions into the bitwise string. */
|
||
|
export function calculateBits(permissions: PermissionStrings[]) {
|
||
|
return permissions
|
||
|
.reduce((bits, perm) => {
|
||
|
bits |= BigInt(DiscordBitwisePermissionFlags[perm]);
|
||
|
return bits;
|
||
|
}, 0n)
|
||
|
.toString();
|
||
|
}
|
||
|
|
||
|
/** Internal function to check if the bot has the permissions to set these overwrites */
|
||
|
export async function requireOverwritePermissions(guildOrId: bigint | DiscordenoGuild, overwrites: Overwrite[]) {
|
||
|
let requiredPerms: Set<PermissionStrings> = new Set(["MANAGE_CHANNELS"]);
|
||
|
|
||
|
overwrites?.forEach((overwrite) => {
|
||
|
overwrite.allow.forEach(requiredPerms.add, requiredPerms);
|
||
|
overwrite.deny.forEach(requiredPerms.add, requiredPerms);
|
||
|
});
|
||
|
|
||
|
// MANAGE_ROLES permission can only be set by administrators
|
||
|
if (requiredPerms.has("MANAGE_ROLES")) {
|
||
|
requiredPerms = new Set<PermissionStrings>(["ADMINISTRATOR"]);
|
||
|
}
|
||
|
|
||
|
await requireGuildPermissions(guildOrId, botId, [...requiredPerms]);
|
||
|
}
|
||
|
|
||
|
/** Gets the highest role from the member in this guild */
|
||
|
export async function highestRole(guildOrId: bigint | DiscordenoGuild, memberOrId: bigint | DiscordenoMember) {
|
||
|
const guild = await getCached("guilds", guildOrId);
|
||
|
|
||
|
if (!guild) throw new Error(Errors.GUILD_NOT_FOUND);
|
||
|
|
||
|
// Get the roles from the member
|
||
|
const memberRoles = (await getCached("members", memberOrId))?.guilds.get(guild.id)?.roles;
|
||
|
// This member has no roles so the highest one is the @everyone role
|
||
|
if (!memberRoles) return guild.roles.get(guild.id)!;
|
||
|
|
||
|
let memberHighestRole: DiscordenoRole | undefined;
|
||
|
|
||
|
for (const roleId of memberRoles) {
|
||
|
const role = guild.roles.get(roleId);
|
||
|
// Rare edge case handling if undefined
|
||
|
if (!role) continue;
|
||
|
|
||
|
// If memberHighestRole is still undefined we want to assign the role,
|
||
|
// else we want to check if the current role position is higher than the current memberHighestRole
|
||
|
if (
|
||
|
!memberHighestRole ||
|
||
|
memberHighestRole.position < role.position ||
|
||
|
memberHighestRole.position === role.position
|
||
|
) {
|
||
|
memberHighestRole = role;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// The member has at least one role so memberHighestRole must exist
|
||
|
return memberHighestRole!;
|
||
|
}
|
||
|
|
||
|
/** Checks if the first role is higher than the second role */
|
||
|
export async function higherRolePosition(guildOrId: bigint | DiscordenoGuild, roleId: bigint, otherRoleId: bigint) {
|
||
|
const guild = await getCached("guilds", guildOrId);
|
||
|
|
||
|
if (!guild) return true;
|
||
|
|
||
|
const role = guild.roles.get(roleId);
|
||
|
const otherRole = guild.roles.get(otherRoleId);
|
||
|
if (!role || !otherRole) throw new Error(Errors.ROLE_NOT_FOUND);
|
||
|
|
||
|
// Rare edge case handling
|
||
|
if (role.position === otherRole.position) {
|
||
|
return role.id < otherRole.id;
|
||
|
}
|
||
|
|
||
|
return role.position > otherRole.position;
|
||
|
}
|
||
|
|
||
|
/** Checks if the member has a higher position than the given role */
|
||
|
export async function isHigherPosition(guildOrId: bigint | DiscordenoGuild, memberId: bigint, compareRoleId: bigint) {
|
||
|
const guild = await getCached("guilds", guildOrId);
|
||
|
|
||
|
if (!guild || guild.ownerId === memberId) return true;
|
||
|
|
||
|
const memberHighestRole = await highestRole(guild, memberId);
|
||
|
return higherRolePosition(guild.id, memberHighestRole.id, compareRoleId);
|
||
|
}
|