Dekescordes/spacebar
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

⚡ improve invite generation

Browse Source
This commit is contained in:
Flam3rboy 2021-09-03 15:48:09 +02:00
parent d5fcb7f971
commit b86f99c4a5
1 changed files with 41 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
util/src/util/Permissions.ts
View File

@ -1,8 +1,8 @@
// https://github.com/discordjs/discord.js/blob/master/src/util/Permissions.js // https://github.com/discordjs/discord.js/blob/master/src/util/Permissions.js
// Apache License Version 2.0 Copyright 2015 - 2021 Amish Shah // Apache License Version 2.0 Copyright 2015 - 2021 Amish Shah
import { In } from "typeorm";
import { Channel, ChannelPermissionOverwrite, Guild, Member, Role } from "../entities"; import { Channel, ChannelPermissionOverwrite, Guild, Member, Role } from "../entities";
import { BitField } from "./BitField"; import { BitField } from "./BitField";
import "missing-native-js-functions";
// TODO: check role hierarchy permission // TODO: check role hierarchy permission
var HTTPError: any; var HTTPError: any;
@ -205,7 +205,19 @@ export type PermissionCache = {
user_id?: string; user_id?: string;
}; };
export async function getPermission(user_id?: string, guild_id?: string, channel_id?: string) { export async function getPermission(
user_id?: string,
guild_id?: string,
channel_id?: string,
opts: {
guild_select?: (keyof Guild)[];
guild_relations?: string[];
channel_select?: (keyof Channel)[];
channel_relations?: string[];
member_select?: (keyof Member)[];
member_relations?: string[];
} = {}
) {
if (!user_id) throw new HTTPError("User not found"); if (!user_id) throw new HTTPError("User not found");
var channel: Channel | undefined; var channel: Channel | undefined;
var member: Member | undefined; var member: Member | undefined;
@ -214,20 +226,42 @@ export async function getPermission(user_id?: string, guild_id?: string, channel
if (channel_id) { if (channel_id) {
channel = await Channel.findOneOrFail({ channel = await Channel.findOneOrFail({
where: { id: channel_id }, where: { id: channel_id },
relations: ["recipients"], relations: ["recipients", ...(opts.channel_relations || [])],
select: ["id", "recipients", "permission_overwrites", "owner_id", "guild_id"], select: [
"id",
"recipients",
"permission_overwrites",
"owner_id",
"guild_id",
// @ts-ignore
...(opts.channel_select || []),
],
}); });
if (channel.guild_id) guild_id = channel.guild_id; // derive guild_id from the channel if (channel.guild_id) guild_id = channel.guild_id; // derive guild_id from the channel
} }
if (guild_id) { if (guild_id) {
guild = await Guild.findOneOrFail({ where: { id: guild_id }, select: ["id", "owner_id"] }); guild = await Guild.findOneOrFail({
where: { id: guild_id },
select: [
"id",
"owner_id",
// @ts-ignore
...(opts.guild_select || []),
],
relations: opts.guild_relations,
});
if (guild.owner_id === user_id) return new Permissions(Permissions.FLAGS.ADMINISTRATOR); if (guild.owner_id === user_id) return new Permissions(Permissions.FLAGS.ADMINISTRATOR);
member = await Member.findOneOrFail({ member = await Member.findOneOrFail({
where: { guild_id, user_id }, where: { guild_id, user_id },
relations: ["roles"], relations: ["roles", ...(opts.member_relations || [])],
select: ["id", "roles"], select: [
"id",
"roles",
// @ts-ignore
...(opts.member_select || []),
],
}); });
} }