Permissions Cleanup (#1643)

* fix Permissions.add/remove, by completely changing what they do

* permissions cleanup

Removes overwrite._denied and overwrite._allowed in favor of overwrite.denied.bitfield and overwrite.allowed.bitfield

uses the modified Permissions.add and Permissions.remove to clean up existing code

fixes GuildMember.missingPermissions

changes Permissions add/remove to reverse loops for speed, changes resolve to allow the number 0 as a valid permission.

* Revert createChannel overwrite.allow / overwrite.deny for arrays of {allow:bitfield, deny:bitfield}

Documentation should be improved here, although I would need advice. I believe a overwrite object typedef is needed, to show the structure of the object, and also include that collections may be used for this, rather than arrays.

* api router fix for overwritePermissions

* add Permissions.freeze, and change all returned Permissions to immutable instances

* Make Permissions a permission resolveable

change role.permissions to be an instance of Permissions

* Make permissions.add/remove return a new instance if the instance is frozen

* Fix invalid error

* Update GuildChannel.js

* Update Guild.js

* fix bad merge
This commit is contained in:
bdistin
2017-09-09 15:07:39 -05:00
committed by Crawl
parent b3e5f6271c
commit 98582cd1b7
6 changed files with 74 additions and 82 deletions

View File

@@ -905,8 +905,8 @@ class Guild extends Base {
createChannel(name, type, { overwrites, reason } = {}) { createChannel(name, type, { overwrites, reason } = {}) {
if (overwrites instanceof Collection || overwrites instanceof Array) { if (overwrites instanceof Collection || overwrites instanceof Array) {
overwrites = overwrites.map(overwrite => { overwrites = overwrites.map(overwrite => {
let allow = overwrite.allow || overwrite._allowed; let allow = overwrite.allow || overwrite.allowed.bitfield;
let deny = overwrite.deny || overwrite._denied; let deny = overwrite.deny || overwrite.denied.bitfield;
if (allow instanceof Array) allow = Permissions.resolve(allow); if (allow instanceof Array) allow = Permissions.resolve(allow);
if (deny instanceof Array) deny = Permissions.resolve(deny); if (deny instanceof Array) deny = Permissions.resolve(deny);

View File

@@ -84,34 +84,23 @@ class GuildChannel extends Channel {
permissionsFor(member) { permissionsFor(member) {
member = this.guild.members.resolve(member); member = this.guild.members.resolve(member);
if (!member) return null; if (!member) return null;
if (member.id === this.guild.ownerID) return new Permissions(Permissions.ALL); if (member.id === this.guild.ownerID) return new Permissions(Permissions.ALL).freeze();
let resolved = this.guild.roles.get(this.guild.id).permissions; const roles = member.roles;
const permissions = new Permissions(roles.map(role => role.permissions));
const overwrites = this.overwritesFor(member, true); if (permissions.has(Permissions.FLAGS.ADMINISTRATOR)) return new Permissions(Permissions.ALL).freeze();
if (overwrites.everyone) {
resolved &= ~overwrites.everyone._denied;
resolved |= overwrites.everyone._allowed;
}
let allows = 0; const overwrites = this.overwritesFor(member, true, roles);
let denies = 0;
for (const overwrite of overwrites.roles) {
allows |= overwrite._allowed;
denies |= overwrite._denied;
}
resolved &= ~denies;
resolved |= allows;
if (overwrites.member) { return permissions
resolved &= ~overwrites.member._denied; .remove(overwrites.everyone ? overwrites.everyone.denied : 0)
resolved |= overwrites.member._allowed; .add(overwrites.everyone ? overwrites.everyone.allowed : 0)
} .remove(overwrites.roles.length > 0 ? overwrites.roles.map(role => role.denied) : 0)
.add(overwrites.roles.length > 0 ? overwrites.roles.map(role => role.allowed) : 0)
const admin = Boolean(resolved & Permissions.FLAGS.ADMINISTRATOR); .remove(overwrites.member ? overwrites.member.denied : 0)
if (admin) resolved = Permissions.ALL; .add(overwrites.member ? overwrites.member.allowed : 0)
.freeze();
return new Permissions(resolved);
} }
overwritesFor(member, verified = false, roles = null) { overwritesFor(member, verified = false, roles = null) {
@@ -167,46 +156,43 @@ class GuildChannel extends Channel {
* .catch(console.error); * .catch(console.error);
*/ */
overwritePermissions(userOrRole, options, reason) { overwritePermissions(userOrRole, options, reason) {
const payload = { const allow = new Permissions(0);
allow: 0, const deny = new Permissions(0);
deny: 0, let type;
};
const role = this.guild.roles.get(userOrRole); const role = this.guild.roles.get(userOrRole);
if (role || userOrRole instanceof Role) { if (role || userOrRole instanceof Role) {
userOrRole = role || userOrRole; userOrRole = role || userOrRole;
payload.type = 'role'; type = 'role';
} else { } else {
userOrRole = this.client.users.resolve(userOrRole); userOrRole = this.client.users.resolve(userOrRole);
payload.type = 'member'; type = 'member';
if (!userOrRole) return Promise.reject(new TypeError('INVALID_TYPE', 'parameter', 'User nor a Role', true)); if (!userOrRole) return Promise.reject(new TypeError('INVALID_TYPE', 'parameter', 'User nor a Role', true));
} }
payload.id = userOrRole.id;
const prevOverwrite = this.permissionOverwrites.get(userOrRole.id); const prevOverwrite = this.permissionOverwrites.get(userOrRole.id);
if (prevOverwrite) { if (prevOverwrite) {
payload.allow = prevOverwrite._allowed; allow.add(prevOverwrite.allowed);
payload.deny = prevOverwrite._denied; deny.add(prevOverwrite.denied);
} }
for (const perm in options) { for (const perm in options) {
if (options[perm] === true) { if (options[perm] === true) {
payload.allow |= Permissions.FLAGS[perm] || 0; allow.add(Permissions.FLAGS[perm] || 0);
payload.deny &= ~(Permissions.FLAGS[perm] || 0); deny.remove(Permissions.FLAGS[perm] || 0);
} else if (options[perm] === false) { } else if (options[perm] === false) {
payload.allow &= ~(Permissions.FLAGS[perm] || 0); allow.remove(Permissions.FLAGS[perm] || 0);
payload.deny |= Permissions.FLAGS[perm] || 0; deny.add(Permissions.FLAGS[perm] || 0);
} else if (options[perm] === null) { } else if (options[perm] === null) {
payload.allow &= ~(Permissions.FLAGS[perm] || 0); allow.remove(Permissions.FLAGS[perm] || 0);
payload.deny &= ~(Permissions.FLAGS[perm] || 0); deny.remove(Permissions.FLAGS[perm] || 0);
} }
} }
return this.client.api.channels(this.id).permissions[payload.id] return this.client.api.channels(this.id).permissions[userOrRole.id]
.put({ data: payload, reason }) .put({ data: { id: userOrRole.id, type, allow: allow.bitfield, deny: deny.bitfield }, reason })
.then(() => this); .then(() => this);
} }

View File

@@ -249,13 +249,8 @@ class GuildMember extends Base {
* @readonly * @readonly
*/ */
get permissions() { get permissions() {
if (this.user.id === this.guild.ownerID) return new Permissions(Permissions.ALL); if (this.user.id === this.guild.ownerID) return new Permissions(Permissions.ALL).freeze();
return new Permissions(this.roles.map(role => role.permissions)).freeze();
let permissions = 0;
const roles = this.roles;
for (const role of roles.values()) permissions |= role.permissions;
return new Permissions(permissions);
} }
/** /**
@@ -321,7 +316,7 @@ class GuildMember extends Base {
* @returns {PermissionResolvable[]} * @returns {PermissionResolvable[]}
*/ */
missingPermissions(permissions, explicit = false) { missingPermissions(permissions, explicit = false) {
return permissions.missing(permissions, explicit); return this.permissions.missing(permissions, explicit);
} }
/** /**

View File

@@ -29,20 +29,17 @@ class PermissionOverwrites {
*/ */
this.type = data.type; this.type = data.type;
this._denied = data.deny;
this._allowed = data.allow;
/** /**
* The permissions that are denied for the user or role. * The permissions that are denied for the user or role.
* @type {Permissions} * @type {Permissions}
*/ */
this.denied = new Permissions(this._denied); this.denied = new Permissions(data.deny).freeze();
/** /**
* The permissions that are allowed for the user or role. * The permissions that are allowed for the user or role.
* @type {Permissions} * @type {Permissions}
*/ */
this.allowed = new Permissions(this._allowed); this.allowed = new Permissions(data.allow).freeze();
} }
/** /**

View File

@@ -53,10 +53,10 @@ class Role extends Base {
this.position = data.position; this.position = data.position;
/** /**
* The permissions bitfield of the role * The permissions of the role
* @type {number} * @type {Permissions}
*/ */
this.permissions = data.permissions; this.permissions = new Permissions(data.permissions).freeze();
/** /**
* Whether or not the role is managed by an external service * Whether or not the role is managed by an external service
@@ -139,7 +139,7 @@ class Role extends Base {
* console.log(role.serialize()); * console.log(role.serialize());
*/ */
serialize() { serialize() {
return new Permissions(this.permissions).serialize(); return this.permissions.serialize();
} }
/** /**
@@ -159,7 +159,7 @@ class Role extends Base {
* } * }
*/ */
hasPermission(permission, explicit = false, checkAdmin) { hasPermission(permission, explicit = false, checkAdmin) {
return new Permissions(this.permissions).has( return this.permissions.has(
permission, typeof checkAdmin !== 'undefined' ? checkAdmin : !explicit permission, typeof checkAdmin !== 'undefined' ? checkAdmin : !explicit
); );
} }
@@ -200,7 +200,7 @@ class Role extends Base {
*/ */
edit(data, reason) { edit(data, reason) {
if (data.permissions) data.permissions = Permissions.resolve(data.permissions); if (data.permissions) data.permissions = Permissions.resolve(data.permissions);
else data.permissions = this.permissions; else data.permissions = this.permissions.bitfield;
return this.client.api.guilds[this.guild.id].roles[this.id].patch({ return this.client.api.guilds[this.guild.id].roles[this.id].patch({
data: { data: {
name: data.name || this.name, name: data.name || this.name,
@@ -341,7 +341,7 @@ class Role extends Base {
this.color === role.color && this.color === role.color &&
this.hoist === role.hoist && this.hoist === role.hoist &&
this.position === role.position && this.position === role.position &&
this.permissions === role.permissions && this.permissions.bitfield === role.permissions.bitfield &&
this.managed === role.managed; this.managed === role.managed;
} }

View File

@@ -41,31 +41,43 @@ class Permissions {
} }
/** /**
* Adds permissions to this one, creating a new instance to represent the new bitfield. * Freezes the permission making it immutable.
* @param {...PermissionResolvable} permissions Permissions to add * @returns {Permissions} This permissions
* @returns {Permissions}
*/ */
add(...permissions) { freeze() {
let total = 0; return Object.freeze(this);
for (let p = 0; p < permissions.length; p++) {
const perm = this.constructor.resolve(permissions[p]);
if ((this.bitfield & perm) !== perm) total |= perm;
}
return new this.constructor(this.member, this.bitfield | total);
} }
/** /**
* Removes permissions to this one, creating a new instance to represent the new bitfield. * Adds permissions to this one.
* @param {...PermissionResolvable} permissions Permissions to add
* @returns {Permissions} This permissions or new permissions if the instance is frozen.
*/
add(...permissions) {
let total = 0;
for (let p = permissions.length - 1; p >= 0; p--) {
const perm = this.constructor.resolve(permissions[p]);
if ((this.bitfield & perm) !== perm) total |= perm;
}
if (Object.isFrozen(this)) return new this.constructor(this.bitfield | total);
this.bitfield |= total;
return this;
}
/**
* Removes permissions from this one.
* @param {...PermissionResolvable} permissions Permissions to remove * @param {...PermissionResolvable} permissions Permissions to remove
* @returns {Permissions} * @returns {Permissions} This permissions or new permissions if the instance is frozen.
*/ */
remove(...permissions) { remove(...permissions) {
let total = 0; let total = 0;
for (let p = 0; p < permissions.length; p++) { for (let p = permissions.length - 1; p >= 0; p--) {
const perm = this.constructor.resolve(permissions[p]); const perm = this.constructor.resolve(permissions[p]);
if ((this.bitfield & perm) === perm) total |= perm; if ((this.bitfield & perm) === perm) total |= perm;
} }
return new this.constructor(this.member, this.bitfield & ~total); if (Object.isFrozen(this)) return new this.constructor(this.bitfield & ~total);
this.bitfield &= ~total;
return this;
} }
/** /**
@@ -84,7 +96,8 @@ class Permissions {
* Data that can be resolved to give a permission number. This can be: * Data that can be resolved to give a permission number. This can be:
* * A string (see {@link Permissions.FLAGS}) * * A string (see {@link Permissions.FLAGS})
* * A permission number * * A permission number
* @typedef {string|number} PermissionResolvable * * An instance of Permissions
* @typedef {string|number|Permissions} PermissionResolvable
*/ */
/** /**
@@ -93,10 +106,11 @@ class Permissions {
* @returns {number} * @returns {number}
*/ */
static resolve(permission) { static resolve(permission) {
if (typeof permission === 'number' && permission >= 0) return permission;
if (permission instanceof Permissions) return permission.bitfield;
if (permission instanceof Array) return permission.map(p => this.resolve(p)).reduce((prev, p) => prev | p, 0); if (permission instanceof Array) return permission.map(p => this.resolve(p)).reduce((prev, p) => prev | p, 0);
if (typeof permission === 'string') permission = this.FLAGS[permission]; if (typeof permission === 'string') return this.FLAGS[permission];
if (typeof permission !== 'number' || permission < 1) throw new RangeError('PERMISSION_INVALID'); throw new RangeError('PERMISSIONS_INVALID');
return permission;
} }
} }